#include "aliased_buffer.h"

#include "debug_utils.h"

#include "memory_tracker-inl.h"

#include "node.h"

#include "node_buffer.h"

#include "node_http2.h"

#include "node_http2_state.h"

#include "node_perf.h"

#include "util-inl.h"

#include <algorithm>

namespace node {

using v8::ArrayBuffer;

using v8::ArrayBufferView;

using v8::Boolean;

using v8::Context;

using v8::Float64Array;

using v8::Function;

using v8::Integer;

using v8::NewStringType;

using v8::Number;

using v8::ObjectTemplate;

using v8::String;

using v8::Uint32;

using v8::Uint32Array;

using v8::Undefined;

using node::performance::PerformanceEntry;

namespace http2 {

namespace {

const char zero_bytes_256[256] = {};

inline Http2Stream* GetStream(Http2Session* session,

int32_t id,

nghttp2_data_source* source) {

Http2Stream* stream = static_cast<Http2Stream*>(source->ptr);

if (stream == nullptr)

stream = session->FindStream(id);

CHECK_NOT_NULL(stream);

CHECK_EQ(id, stream->id());

return stream;

}

} // anonymous namespace

const Http2Session::Callbacks Http2Session::callback_struct_saved[2] = {

Callbacks(false),

Callbacks(true)};

Http2Scope::Http2Scope(Http2Stream* stream) : Http2Scope(stream->session()) {}

Http2Scope::Http2Scope(Http2Session* session) {

}

Http2Scope::~Http2Scope() {

if (session_ == nullptr)

return;

session_->flags_ &= ~SESSION_STATE_HAS_SCOPE;

session_->MaybeScheduleWrite();

}

Http2Options::Http2Options(Environment* env, nghttp2_session_type type) {

}

void Http2Session::Http2Settings::Init() {

AliasedUint32Array& buffer = env()->http2_state()->settings_buffer;

uint32_t flags = buffer[IDX_SETTINGS_COUNT];

size_t n = 0;

#define GRABSETTING(N, trace) \

if (flags & (1 << IDX_SETTINGS_##N)) { \

uint32_t val = buffer[IDX_SETTINGS_##N]; \

if (session_ != nullptr) \

Debug(session_, "setting " trace ": %d\n", val); \

entries_[n++] = \

nghttp2_settings_entry {NGHTTP2_SETTINGS_##N, val}; \

}

GRABSETTING(HEADER_TABLE_SIZE, "header table size");

GRABSETTING(MAX_CONCURRENT_STREAMS, "max concurrent streams");

GRABSETTING(MAX_FRAME_SIZE, "max frame size");

GRABSETTING(INITIAL_WINDOW_SIZE, "initial window size");

GRABSETTING(MAX_HEADER_LIST_SIZE, "max header list size");

GRABSETTING(ENABLE_PUSH, "enable push");

GRABSETTING(ENABLE_CONNECT_PROTOCOL, "enable connect protocol");

#undef GRABSETTING

count_ = n;

}

Http2Session::Http2Settings::Http2Settings(Environment* env,

Http2Session* session,

Local<Object> obj,

uint64_t start_time)

: AsyncWrap(env, obj, PROVIDER_HTTP2SETTINGS),

session_(session),

startTime_(start_time) {

RemoveCleanupHook(); // This object is owned by the Http2Session.

Init();

}

Local<Value> Http2Session::Http2Settings::Pack() {

const size_t len = count_ * 6;

Local<Value> buf = Buffer::New(env(), len).ToLocalChecked();

ssize_t ret =

nghttp2_pack_settings_payload(

reinterpret_cast<uint8_t*>(Buffer::Data(buf)), len,

&entries_[0], count_);

if (ret >= 0)

return buf;

else

return Undefined(env()->isolate());

}

void Http2Session::Http2Settings::Update(Environment* env,

Http2Session* session,

get_setting fn) {

AliasedUint32Array& buffer = env->http2_state()->settings_buffer;

buffer[IDX_SETTINGS_HEADER_TABLE_SIZE] =

fn(**session, NGHTTP2_SETTINGS_HEADER_TABLE_SIZE);

buffer[IDX_SETTINGS_MAX_CONCURRENT_STREAMS] =

fn(**session, NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS);

buffer[IDX_SETTINGS_INITIAL_WINDOW_SIZE] =

fn(**session, NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE);

buffer[IDX_SETTINGS_MAX_FRAME_SIZE] =

fn(**session, NGHTTP2_SETTINGS_MAX_FRAME_SIZE);

buffer[IDX_SETTINGS_MAX_HEADER_LIST_SIZE] =

fn(**session, NGHTTP2_SETTINGS_MAX_HEADER_LIST_SIZE);

buffer[IDX_SETTINGS_ENABLE_PUSH] =

fn(**session, NGHTTP2_SETTINGS_ENABLE_PUSH);

buffer[IDX_SETTINGS_ENABLE_CONNECT_PROTOCOL] =

fn(**session, NGHTTP2_SETTINGS_ENABLE_CONNECT_PROTOCOL);

}

void Http2Session::Http2Settings::RefreshDefaults(Environment* env) {

AliasedUint32Array& buffer = env->http2_state()->settings_buffer;

buffer[IDX_SETTINGS_HEADER_TABLE_SIZE] =

DEFAULT_SETTINGS_HEADER_TABLE_SIZE;

buffer[IDX_SETTINGS_ENABLE_PUSH] =

DEFAULT_SETTINGS_ENABLE_PUSH;

buffer[IDX_SETTINGS_INITIAL_WINDOW_SIZE] =

DEFAULT_SETTINGS_INITIAL_WINDOW_SIZE;

buffer[IDX_SETTINGS_MAX_FRAME_SIZE] =

DEFAULT_SETTINGS_MAX_FRAME_SIZE;

buffer[IDX_SETTINGS_MAX_HEADER_LIST_SIZE] =

DEFAULT_SETTINGS_MAX_HEADER_LIST_SIZE;

buffer[IDX_SETTINGS_COUNT] =

(1 << IDX_SETTINGS_HEADER_TABLE_SIZE) |

(1 << IDX_SETTINGS_ENABLE_PUSH) |

(1 << IDX_SETTINGS_INITIAL_WINDOW_SIZE) |

(1 << IDX_SETTINGS_MAX_FRAME_SIZE) |

(1 << IDX_SETTINGS_MAX_HEADER_LIST_SIZE);

}

void Http2Session::Http2Settings::Send() {

Http2Scope h2scope(session_);

CHECK_EQ(nghttp2_submit_settings(**session_, NGHTTP2_FLAG_NONE,

&entries_[0], count_), 0);

}

void Http2Session::Http2Settings::Done(bool ack) {

uint64_t end = uv_hrtime();

double duration = (end - startTime_) / 1e6;

Local<Value> argv[] = {

Boolean::New(env()->isolate(), ack),

Number::New(env()->isolate(), duration)

};

MakeCallback(env()->ondone_string(), arraysize(argv), argv);

}

Http2Priority::Http2Priority(Environment* env,

}

const char* Http2Session::TypeName() const {

switch (session_type_) {

case NGHTTP2_SESSION_SERVER: return "server";

case NGHTTP2_SESSION_CLIENT: return "client";

default:

// This should never happen

ABORT();

}

}

Headers::Headers(Isolate* isolate,

}

Origins::Origins(Isolate* isolate,

}

Http2Session::Callbacks::Callbacks(bool kHasGetPaddingCallback) {

CHECK_EQ(nghttp2_session_callbacks_new(&callbacks), 0);

nghttp2_session_callbacks_set_on_begin_headers_callback(

callbacks, OnBeginHeadersCallback);

nghttp2_session_callbacks_set_on_header_callback2(

callbacks, OnHeaderCallback);

nghttp2_session_callbacks_set_on_frame_recv_callback(

callbacks, OnFrameReceive);

nghttp2_session_callbacks_set_on_stream_close_callback(

callbacks, OnStreamClose);

nghttp2_session_callbacks_set_on_data_chunk_recv_callback(

callbacks, OnDataChunkReceived);

nghttp2_session_callbacks_set_on_frame_not_send_callback(

callbacks, OnFrameNotSent);

nghttp2_session_callbacks_set_on_invalid_header_callback2(

callbacks, OnInvalidHeader);

nghttp2_session_callbacks_set_error_callback(

callbacks, OnNghttpError);

nghttp2_session_callbacks_set_send_data_callback(

callbacks, OnSendData);

nghttp2_session_callbacks_set_on_invalid_frame_recv_callback(

callbacks, OnInvalidFrame);

nghttp2_session_callbacks_set_on_frame_send_callback(

callbacks, OnFrameSent);

if (kHasGetPaddingCallback) {

nghttp2_session_callbacks_set_select_padding_callback(

callbacks, OnSelectPadding);

}

}

Http2Session::Callbacks::~Callbacks() {

nghttp2_session_callbacks_del(callbacks);

}

class Http2Session::MemoryAllocatorInfo {

public:

explicit MemoryAllocatorInfo(Http2Session* session)

: info({ session, H2Malloc, H2Free, H2Calloc, H2Realloc }) {}

static void* H2Malloc(size_t size, void* user_data) {

return H2Realloc(nullptr, size, user_data);

}

static void* H2Calloc(size_t nmemb, size_t size, void* user_data) {

size_t real_size = MultiplyWithOverflowCheck(nmemb, size);

void* mem = H2Malloc(real_size, user_data);

if (mem != nullptr)

memset(mem, 0, real_size);

return mem;

}

static void H2Free(void* ptr, void* user_data) {

}

static void* H2Realloc(void* ptr, size_t size, void* user_data) {

Http2Session* session = static_cast<Http2Session*>(user_data);

size_t previous_size = 0;

char* original_ptr = nullptr;

// We prepend each allocated buffer with a size_t containing the full

// size of the allocation.

if (size > 0) size += sizeof(size_t);

if (ptr != nullptr) {

// We are free()ing or re-allocating.

original_ptr = static_cast<char*>(ptr) - sizeof(size_t);

previous_size = *reinterpret_cast<size_t*>(original_ptr);

// This means we called StopTracking() on this pointer before.

if (previous_size == 0) {

// Fall back to the standard Realloc() function.

char* ret = UncheckedRealloc(original_ptr, size);

if (ret != nullptr)

ret += sizeof(size_t);

return ret;

}

}

CHECK_GE(session->current_nghttp2_memory_, previous_size);

// TODO(addaleax): Add the following, and handle NGHTTP2_ERR_NOMEM properly

// everywhere:

//

// if (size > previous_size &&

// !session->IsAvailableSessionMemory(size - previous_size)) {

// return nullptr;

//}

char* mem = UncheckedRealloc(original_ptr, size);

if (mem != nullptr) {

// Adjust the memory info counter.

session->current_nghttp2_memory_ += size - previous_size;

*reinterpret_cast<size_t*>(mem) = size;

mem += sizeof(size_t);

} else if (size == 0) {

session->current_nghttp2_memory_ -= previous_size;

}

return mem;

}

static void StopTracking(Http2Session* session, void* ptr) {

}

inline nghttp2_mem* operator*() { return &info; }

nghttp2_mem info;

};

void Http2Session::StopTrackingRcbuf(nghttp2_rcbuf* buf) {

MemoryAllocatorInfo::StopTracking(this, buf);

}

Http2Session::Http2Session(Environment* env,

}

Http2Session::~Http2Session() {

CHECK_EQ(flags_ & SESSION_STATE_HAS_SCOPE, 0);

Debug(this, "freeing nghttp2 session");

for (const auto& iter : streams_)

iter.second->session_ = nullptr;

nghttp2_session_del(session_);

CHECK_EQ(current_nghttp2_memory_, 0);

}

std::string Http2Session::diagnostic_name() const {

return std::string("Http2Session ") + TypeName() + " (" +

std::to_string(static_cast<int64_t>(get_async_id())) + ")";

}

inline bool HasHttp2Observer(Environment* env) {

AliasedUint32Array& observers = env->performance_state()->observers;

return observers[performance::NODE_PERFORMANCE_ENTRY_TYPE_HTTP2] != 0;

}

void Http2Stream::EmitStatistics() {

}

void Http2Session::EmitStatistics() {

}

void Http2Session::Close(uint32_t code, bool socket_closed) {

}

inline Http2Stream* Http2Session::FindStream(int32_t id) {

auto s = streams_.find(id);

return s != streams_.end() ? s->second : nullptr;

}

inline bool Http2Session::CanAddStream() {

}

inline void Http2Session::AddStream(Http2Stream* stream) {

}

inline void Http2Session::RemoveStream(Http2Stream* stream) {

}

ssize_t Http2Session::OnDWordAlignedPadding(size_t frameLen,

}

ssize_t Http2Session::OnMaxFrameSizePadding(size_t frameLen,

}

ssize_t Http2Session::OnCallbackPadding(size_t frameLen,

}

ssize_t Http2Session::Write(const uv_buf_t* bufs, size_t nbufs) {

}

inline int32_t GetFrameID(const nghttp2_frame* frame) {

}

int Http2Session::OnBeginHeadersCallback(nghttp2_session* handle,

}

int Http2Session::OnHeaderCallback(nghttp2_session* handle,

}

int Http2Session::OnFrameReceive(nghttp2_session* handle,

}

int Http2Session::OnInvalidFrame(nghttp2_session* handle,