// Copyright 2015 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_CODEGEN_SIGNATURE_H_
#define V8_CODEGEN_SIGNATURE_H_

#include "src/base/hashing.h"
#include "src/base/vector.h"
#include "src/codegen/machine-type.h"
#include "src/sandbox/check.h"
#include "src/zone/zone.h"

namespace v8 {
namespace internal {

template <typename SigT, typename T>
class SignatureBuilder {
 public:
  SignatureBuilder(Zone* zone, size_t return_count, size_t parameter_count)
      : return_count_(return_count),
        parameter_count_(parameter_count),
        rcursor_(0),
        pcursor_(0) {
    // Allocate memory for the signature plus the array backing the
    // signature.
    constexpr size_t padding = sizeof(SigT) % alignof(T);
    using AllocationTypeTag = SignatureBuilder;
    const size_t allocated_bytes =
        sizeof(SigT) + padding + sizeof(T) * (return_count + parameter_count);
    void* memory = zone->Allocate<AllocationTypeTag>(allocated_bytes);
    uint8_t* rep_buffer =
        reinterpret_cast<uint8_t*>(memory) + sizeof(SigT) + padding;
    DCHECK(IsAligned(reinterpret_cast<uintptr_t>(rep_buffer), alignof(T)));
    buffer_ = reinterpret_cast<T*>(rep_buffer);
    sig_ = new (memory) SigT{return_count, parameter_count, buffer_};
  }

  const size_t return_count_;
  const size_t parameter_count_;

  void AddReturn(T val) {
    DCHECK_LT(rcursor_, return_count_);
    buffer_[rcursor_++] = val;
  }

  void AddReturnAt(size_t index, T val) {
    DCHECK_LT(index, return_count_);
    buffer_[index] = val;
    rcursor_ = std::max(rcursor_, index + 1);
  }

  void AddParam(T val) {
    DCHECK_LT(pcursor_, parameter_count_);
    buffer_[return_count_ + pcursor_++] = val;
  }

  void AddParamAt(size_t index, T val) {
    DCHECK_LT(index, parameter_count_);
    buffer_[return_count_ + index] = val;
    pcursor_ = std::max(pcursor_, index + 1);
  }

  SigT* Get() const {
    DCHECK_EQ(rcursor_, return_count_);
    DCHECK_EQ(pcursor_, parameter_count_);
    DCHECK_NOT_NULL(sig_);
    return sig_;
  }

 protected:
  size_t rcursor_;
  size_t pcursor_;
  SigT* sig_;
  T* buffer_;
};

// Describes the inputs and outputs of a function or call.
template <typename T>
class Signature : public ZoneObject {
 public:
  constexpr Signature(size_t return_count, size_t parameter_count,
                      const T* reps)
      : return_count_(return_count),
        parameter_count_(parameter_count),
        reps_(reps) {
    DCHECK_EQ(kReturnCountOffset, offsetof(Signature, return_count_));
    DCHECK_EQ(kParameterCountOffset, offsetof(Signature, parameter_count_));
    DCHECK_EQ(kRepsOffset, offsetof(Signature, reps_));
    static_assert(std::is_standard_layout_v<Signature<T>>);
  }

  size_t return_count() const { return return_count_; }
  size_t parameter_count() const { return parameter_count_; }

  T GetParam(size_t index) const {
    // If heap memory is corrupted, we may get confused about the number of
    // parameters during compilation. These SBXCHECKs defend against that.
    SBXCHECK_LT(index, parameter_count_);
    return reps_[return_count_ + index];
  }

  T GetReturn(size_t index = 0) const {
    SBXCHECK_LT(index, return_count_);
    return reps_[index];
  }

  bool contains(T element) const {
    return std::find(all().cbegin(), all().cend(), element) != all().cend();
  }

  // Iteration support.
  base::Vector<const T> parameters() const {
    return {reps_ + return_count_, parameter_count_};
  }
  base::Vector<const T> returns() const { return {reps_, return_count_}; }
  base::Vector<const T> all() const {
    return {reps_, return_count_ + parameter_count_};
  }

  bool operator==(const Signature& other) const {
    if (this == &other) return true;
    if (parameter_count() != other.parameter_count()) return false;
    if (return_count() != other.return_count()) return false;
    return std::equal(all().begin(), all().end(), other.all().begin());
  }
  bool operator!=(const Signature& other) const { return !(*this == other); }

  // For incrementally building signatures.
  using Builder = SignatureBuilder<Signature<T>, T>;

  static Signature<T>* Build(Zone* zone, std::initializer_list<T> returns,
                             std::initializer_list<T> params) {
    Builder builder(zone, returns.size(), params.size());
    for (T ret : returns) builder.AddReturn(ret);
    for (T param : params) builder.AddParam(param);
    return builder.Get();
  }

  static constexpr size_t kReturnCountOffset = 0;
  static constexpr size_t kParameterCountOffset =
      kReturnCountOffset + kSizetSize;
  static constexpr size_t kRepsOffset = kParameterCountOffset + kSizetSize;

 protected:
  size_t return_count_;
  size_t parameter_count_;
  const T* reps_;
};

using MachineSignature = Signature<MachineType>;

template <typename T>
size_t hash_value(const Signature<T>& sig) {
  // Hash over all contained representations, plus the parameter count to
  // differentiate signatures with the same representation array but different
  // parameter/return count.
  return base::Hasher{}.Add(sig.parameter_count()).AddRange(sig.all()).hash();
}

template <typename T, size_t kNumReturns = 0, size_t kNumParams = 0>
class FixedSizeSignature : public Signature<T> {
 public:
  // Add return types to this signature (only allowed if there are none yet).
  template <typename... ReturnTypes>
  auto Returns(ReturnTypes... return_types) const {
    static_assert(kNumReturns == 0, "Please specify all return types at once");
    return FixedSizeSignature<T, sizeof...(ReturnTypes), kNumParams>{
        std::initializer_list<T>{return_types...}.begin(), reps_};
  }

  // Add parameters to this signature (only allowed if there are none yet).
  template <typename... ParamTypes>
  auto Params(ParamTypes... param_types) const {
    static_assert(kNumParams == 0, "Please specify all parameters at once");
    return FixedSizeSignature<T, kNumReturns, sizeof...(ParamTypes)>{
        reps_, std::initializer_list<T>{param_types...}.begin()};
  }

 private:
  // Other template instantiations can call the private constructor.
  template <typename T2, size_t kNumReturns2, size_t kNumParams2>
  friend class FixedSizeSignature;

  FixedSizeSignature(const T* returns, const T* params)
      : Signature<T>(kNumReturns, kNumParams, reps_) {
    std::copy(returns, returns + kNumReturns, reps_);
    std::copy(params, params + kNumParams, reps_ + kNumReturns);
  }

  T reps_[kNumReturns + kNumParams];
};

// Specialization for zero-sized signatures.
template <typename T>
class FixedSizeSignature<T, 0, 0> : public Signature<T> {
 public:
  constexpr FixedSizeSignature() : Signature<T>(0, 0, nullptr) {}

  // Add return types.
  template <typename... ReturnTypes>
  static auto Returns(ReturnTypes... return_types) {
    return FixedSizeSignature<T, sizeof...(ReturnTypes), 0>{
        std::initializer_list<T>{return_types...}.begin(), nullptr};
  }

  // Add parameters.
  template <typename... ParamTypes>
  static auto Params(ParamTypes... param_types) {
    return FixedSizeSignature<T, 0, sizeof...(ParamTypes)>{
        nullptr, std::initializer_list<T>{param_types...}.begin()};
  }
};

}  // namespace internal
}  // namespace v8

#endif  // V8_CODEGEN_SIGNATURE_H_