#include "node_http2.h"

#include "aliased_buffer.h"

#include "aliased_struct-inl.h"

#include "debug_utils-inl.h"

#include "histogram-inl.h"

#include "memory_tracker-inl.h"

#include "node.h"

#include "node_buffer.h"

#include "node_http_common-inl.h"

#include "node_mem-inl.h"

#include "node_perf.h"

#include "node_revert.h"

#include "stream_base-inl.h"

#include "util-inl.h"

#include <algorithm>

namespace node {

using v8::Array;

using v8::ArrayBuffer;

using v8::ArrayBufferView;

using v8::BackingStore;

using v8::Boolean;

using v8::Context;

using v8::EscapableHandleScope;

using v8::False;

using v8::Function;

using v8::FunctionCallbackInfo;

using v8::FunctionTemplate;

using v8::HandleScope;

using v8::Integer;

using v8::Isolate;

using v8::Local;

using v8::MaybeLocal;

using v8::NewStringType;

using v8::Number;

using v8::Object;

using v8::ObjectTemplate;

using v8::String;

using v8::True;

using v8::Uint8Array;

using v8::Undefined;

using v8::Value;

namespace http2 {

namespace {

const char zero_bytes_256[256] = {};

bool HasHttp2Observer(Environment* env) {

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

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

}

} // anonymous namespace

// These configure the callbacks required by nghttp2 itself. There are

// two sets of callback functions, one that is used if a padding callback

// is set, and other that does not include the padding callback.

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

Callbacks(false),

Callbacks(true)};

// The Http2Scope object is used to queue a write to the i/o stream. It is

// used whenever any action is take on the underlying nghttp2 API that may

// push data into nghttp2 outbound data queue.

//

// For example:

//

// Http2Scope h2scope(session);

// nghttp2_submit_ping(session->session(), ... );

//

// When the Http2Scope passes out of scope and is deconstructed, it will

// call Http2Session::MaybeScheduleWrite().

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

Http2Scope::Http2Scope(Http2Session* session) : session_(session) {

if (!session_) return;

// If there is another scope further below on the stack, or

// a write is already scheduled, there's nothing to do.

if (session_->is_in_scope() || session_->is_write_scheduled()) {

session_.reset();

return;

}

session_->set_in_scope();

}

Http2Scope::~Http2Scope() {

if (!session_) return;

session_->set_in_scope(false);

if (!session_->is_write_scheduled())

session_->MaybeScheduleWrite();

}

// The Http2Options object is used during the construction of Http2Session

// instances to configure an appropriate nghttp2_options struct. The class

// uses a single TypedArray instance that is shared with the JavaScript side

// to more efficiently pass values back and forth.

Http2Options::Http2Options(Http2State* http2_state, SessionType type) {

nghttp2_option* option;

CHECK_EQ(nghttp2_option_new(&option), 0);

CHECK_NOT_NULL(option);

options_.reset(option);

// Make sure closed connections aren't kept around, taking up memory.

// Note that this breaks the priority tree, which we don't use.

nghttp2_option_set_no_closed_streams(option, 1);

// We manually handle flow control within a session in order to

// implement backpressure -- that is, we only send WINDOW_UPDATE

// frames to the remote peer as data is actually consumed by user

// code. This ensures that the flow of data over the connection

// does not move too quickly and limits the amount of data we

// are required to buffer.

nghttp2_option_set_no_auto_window_update(option, 1);

// Enable built in support for receiving ALTSVC and ORIGIN frames (but

// only on client side sessions

if (type == NGHTTP2_SESSION_CLIENT) {

nghttp2_option_set_builtin_recv_extension_type(option, NGHTTP2_ALTSVC);

nghttp2_option_set_builtin_recv_extension_type(option, NGHTTP2_ORIGIN);

}

AliasedUint32Array& buffer = http2_state->options_buffer;

uint32_t flags = buffer[IDX_OPTIONS_FLAGS];

if (flags & (1 << IDX_OPTIONS_MAX_DEFLATE_DYNAMIC_TABLE_SIZE)) {

nghttp2_option_set_max_deflate_dynamic_table_size(

option,

buffer[IDX_OPTIONS_MAX_DEFLATE_DYNAMIC_TABLE_SIZE]);

}

if (flags & (1 << IDX_OPTIONS_MAX_RESERVED_REMOTE_STREAMS)) {

nghttp2_option_set_max_reserved_remote_streams(

option,

buffer[IDX_OPTIONS_MAX_RESERVED_REMOTE_STREAMS]);

}

if (flags & (1 << IDX_OPTIONS_MAX_SEND_HEADER_BLOCK_LENGTH)) {

nghttp2_option_set_max_send_header_block_length(

option,

buffer[IDX_OPTIONS_MAX_SEND_HEADER_BLOCK_LENGTH]);

}

// Recommended default

nghttp2_option_set_peer_max_concurrent_streams(option, 100);

if (flags & (1 << IDX_OPTIONS_PEER_MAX_CONCURRENT_STREAMS)) {

nghttp2_option_set_peer_max_concurrent_streams(

option,

buffer[IDX_OPTIONS_PEER_MAX_CONCURRENT_STREAMS]);

}

// The padding strategy sets the mechanism by which we determine how much

// additional frame padding to apply to DATA and HEADERS frames. Currently

// this is set on a per-session basis, but eventually we may switch to

// a per-stream setting, giving users greater control

if (flags & (1 << IDX_OPTIONS_PADDING_STRATEGY)) {

PaddingStrategy strategy =

static_cast<PaddingStrategy>(

buffer.GetValue(IDX_OPTIONS_PADDING_STRATEGY));

set_padding_strategy(strategy);

}

// The max header list pairs option controls the maximum number of

// header pairs the session may accept. This is a hard limit.. that is,

// if the remote peer sends more than this amount, the stream will be

// automatically closed with an RST_STREAM.

if (flags & (1 << IDX_OPTIONS_MAX_HEADER_LIST_PAIRS))

set_max_header_pairs(buffer[IDX_OPTIONS_MAX_HEADER_LIST_PAIRS]);

// The HTTP2 specification places no limits on the number of HTTP2

// PING frames that can be sent. In order to prevent PINGS from being

// abused as an attack vector, however, we place a strict upper limit

// on the number of unacknowledged PINGS that can be sent at any given

// time.

if (flags & (1 << IDX_OPTIONS_MAX_OUTSTANDING_PINGS))

set_max_outstanding_pings(buffer[IDX_OPTIONS_MAX_OUTSTANDING_PINGS]);

// The HTTP2 specification places no limits on the number of HTTP2

// SETTINGS frames that can be sent. In order to prevent PINGS from being

// abused as an attack vector, however, we place a strict upper limit

// on the number of unacknowledged SETTINGS that can be sent at any given

// time.

if (flags & (1 << IDX_OPTIONS_MAX_OUTSTANDING_SETTINGS))

set_max_outstanding_settings(buffer[IDX_OPTIONS_MAX_OUTSTANDING_SETTINGS]);

// The HTTP2 specification places no limits on the amount of memory

// that a session can consume. In order to prevent abuse, we place a

// cap on the amount of memory a session can consume at any given time.

// this is a credit based system. Existing streams may cause the limit

// to be temporarily exceeded but once over the limit, new streams cannot

// created.

// Important: The maxSessionMemory option in javascript is expressed in

// terms of MB increments (i.e. the value 1 == 1 MB)

if (flags & (1 << IDX_OPTIONS_MAX_SESSION_MEMORY))

set_max_session_memory(buffer[IDX_OPTIONS_MAX_SESSION_MEMORY] *

static_cast<uint64_t>(1000000));

if (flags & (1 << IDX_OPTIONS_MAX_SETTINGS)) {

nghttp2_option_set_max_settings(

option,

static_cast<size_t>(buffer[IDX_OPTIONS_MAX_SETTINGS]));

}

}

#define GRABSETTING(entries, count, name) \

do { \

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

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

entries[count++] = \

nghttp2_settings_entry {NGHTTP2_SETTINGS_ ## name, val}; \

} } while (0)

size_t Http2Settings::Init(

Http2State* http2_state,

nghttp2_settings_entry* entries) {

AliasedUint32Array& buffer = http2_state->settings_buffer;

uint32_t flags = buffer[IDX_SETTINGS_COUNT];

size_t count = 0;

#define V(name) GRABSETTING(entries, count, name);

HTTP2_SETTINGS(V)

#undef V

return count;

}

#undef GRABSETTING

// The Http2Settings class is used to configure a SETTINGS frame that is

// to be sent to the connected peer. The settings are set using a TypedArray

// that is shared with the JavaScript side.

Http2Settings::Http2Settings(Http2Session* session,

Local<Object> obj,

Local<Function> callback,

uint64_t start_time)

: AsyncWrap(session->env(), obj, PROVIDER_HTTP2SETTINGS),

session_(session),

startTime_(start_time) {

callback_.Reset(env()->isolate(), callback);

count_ = Init(session->http2_state(), entries_);

}

Local<Function> Http2Settings::callback() const {

return callback_.Get(env()->isolate());

}

void Http2Settings::MemoryInfo(MemoryTracker* tracker) const {

tracker->TrackField("callback", callback_);

}

// Generates a Buffer that contains the serialized payload of a SETTINGS

// frame. This can be used, for instance, to create the Base64-encoded

// content of an Http2-Settings header field.

Local<Value> Http2Settings::Pack() {

return Pack(session_->env(), count_, entries_);

}

Local<Value> Http2Settings::Pack(Http2State* state) {

nghttp2_settings_entry entries[IDX_SETTINGS_COUNT];

size_t count = Init(state, entries);

return Pack(state->env(), count, entries);

}

Local<Value> Http2Settings::Pack(

Environment* env,

size_t count,

const nghttp2_settings_entry* entries) {

EscapableHandleScope scope(env->isolate());

std::unique_ptr<BackingStore> bs;

{

NoArrayBufferZeroFillScope no_zero_fill_scope(env->isolate_data());

bs = ArrayBuffer::NewBackingStore(env->isolate(), count * 6);

}

if (nghttp2_pack_settings_payload(static_cast<uint8_t*>(bs->Data()),

bs->ByteLength(),

entries,

count) < 0) {

return scope.Escape(Undefined(env->isolate()));

}

Local<ArrayBuffer> ab = ArrayBuffer::New(env->isolate(), std::move(bs));

return scope.Escape(Buffer::New(env, ab, 0, ab->ByteLength())

.FromMaybe(Local<Value>()));

}

// Updates the shared TypedArray with the current remote or local settings for

// the session.

void Http2Settings::Update(Http2Session* session, get_setting fn) {

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

#define V(name) \

buffer[IDX_SETTINGS_ ## name] = \

fn(session->session(), NGHTTP2_SETTINGS_ ## name);

HTTP2_SETTINGS(V)

#undef V

}

// Initializes the shared TypedArray with the default settings values.

void Http2Settings::RefreshDefaults(Http2State* http2_state) {

AliasedUint32Array& buffer = http2_state->settings_buffer;

uint32_t flags = 0;

#define V(name) \

do { \

buffer[IDX_SETTINGS_ ## name] = DEFAULT_SETTINGS_ ## name; \

flags |= 1 << IDX_SETTINGS_ ## name; \

} while (0);

HTTP2_SETTINGS(V)

#undef V

buffer[IDX_SETTINGS_COUNT] = flags;

}

void Http2Settings::Send() {

Http2Scope h2scope(session_.get());

CHECK_EQ(nghttp2_submit_settings(

session_->session(),

NGHTTP2_FLAG_NONE,

&entries_[0],

count_), 0);

}

void Http2Settings::Done(bool ack) {

uint64_t end = uv_hrtime();

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

Local<Value> argv[] = {

ack ? True(env()->isolate()) : False(env()->isolate()),

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

};

MakeCallback(callback(), arraysize(argv), argv);

}

// The Http2Priority class initializes an appropriate nghttp2_priority_spec

// struct used when either creating a stream or updating its priority

// settings.

Http2Priority::Http2Priority(Environment* env,

Local<Value> parent,

Local<Value> weight,

Local<Value> exclusive) {

Local<Context> context = env->context();

int32_t parent_ = parent->Int32Value(context).ToChecked();

int32_t weight_ = weight->Int32Value(context).ToChecked();

bool exclusive_ = exclusive->IsTrue();

Debug(env, DebugCategory::HTTP2STREAM,

"Http2Priority: parent: %d, weight: %d, exclusive: %s\n",

parent_, weight_, exclusive_ ? "yes" : "no");

nghttp2_priority_spec_init(this, parent_, weight_, exclusive_ ? 1 : 0);

}

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();

}

}

Origins::Origins(

Environment* env,

Local<String> origin_string,

size_t origin_count)

: count_(origin_count) {

int origin_string_len = origin_string->Length();

if (count_ == 0) {

CHECK_EQ(origin_string_len, 0);

return;

}

{

NoArrayBufferZeroFillScope no_zero_fill_scope(env->isolate_data());

bs_ = ArrayBuffer::NewBackingStore(env->isolate(),

alignof(nghttp2_origin_entry) - 1 +

count_ * sizeof(nghttp2_origin_entry) +

origin_string_len);

}

// Make sure the start address is aligned appropriately for an nghttp2_nv*.

char* start = AlignUp(static_cast<char*>(bs_->Data()),

alignof(nghttp2_origin_entry));

char* origin_contents = start + (count_ * sizeof(nghttp2_origin_entry));

nghttp2_origin_entry* const nva =

reinterpret_cast<nghttp2_origin_entry*>(start);

CHECK_LE(origin_contents + origin_string_len,

static_cast<char*>(bs_->Data()) + bs_->ByteLength());

CHECK_EQ(origin_string->WriteOneByte(

env->isolate(),

reinterpret_cast<uint8_t*>(origin_contents),

0,

origin_string_len,

String::NO_NULL_TERMINATION),

origin_string_len);

size_t n = 0;

char* p;

for (p = origin_contents; p < origin_contents + origin_string_len; n++) {

if (n >= count_) {

static uint8_t zero = '\0';

nva[0].origin = &zero;

nva[0].origin_len = 1;

count_ = 1;

return;

}

nva[n].origin = reinterpret_cast<uint8_t*>(p);

nva[n].origin_len = strlen(p);

p += nva[n].origin_len + 1;

}

}

// Sets the various callback functions that nghttp2 will use to notify us

// about significant events while processing http2 stuff.

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

nghttp2_session_callbacks* callbacks_;

CHECK_EQ(nghttp2_session_callbacks_new(&callbacks_), 0);

callbacks.reset(callbacks_);

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);

}

}

void Http2Session::StopTrackingRcbuf(nghttp2_rcbuf* buf) {

StopTrackingMemory(buf);

}

void Http2Session::CheckAllocatedSize(size_t previous_size) const {

CHECK_GE(current_nghttp2_memory_, previous_size);

}

void Http2Session::IncreaseAllocatedSize(size_t size) {

current_nghttp2_memory_ += size;

}

void Http2Session::DecreaseAllocatedSize(size_t size) {

current_nghttp2_memory_ -= size;

}

Http2Session::Http2Session(Http2State* http2_state,

Local<Object> wrap,

SessionType type)

: AsyncWrap(http2_state->env(), wrap, AsyncWrap::PROVIDER_HTTP2SESSION),

js_fields_(http2_state->env()->isolate()),

session_type_(type),

http2_state_(http2_state) {

MakeWeak();

statistics_.session_type = type;

statistics_.start_time = uv_hrtime();

// Capture the configuration options for this session

Http2Options opts(http2_state, type);

max_session_memory_ = opts.max_session_memory();

uint32_t maxHeaderPairs = opts.max_header_pairs();

max_header_pairs_ =

type == NGHTTP2_SESSION_SERVER

? GetServerMaxHeaderPairs(maxHeaderPairs)

: GetClientMaxHeaderPairs(maxHeaderPairs);

max_outstanding_pings_ = opts.max_outstanding_pings();

max_outstanding_settings_ = opts.max_outstanding_settings();

padding_strategy_ = opts.padding_strategy();

bool hasGetPaddingCallback =

padding_strategy_ != PADDING_STRATEGY_NONE;

auto fn = type == NGHTTP2_SESSION_SERVER ?

nghttp2_session_server_new3 :

nghttp2_session_client_new3;

nghttp2_mem alloc_info = MakeAllocator();

// This should fail only if the system is out of memory, which

// is going to cause lots of other problems anyway, or if any

// of the options are out of acceptable range, which we should

// be catching before it gets this far. Either way, crash if this

// fails.

nghttp2_session* session;

CHECK_EQ(fn(

&session,

callback_struct_saved[hasGetPaddingCallback ? 1 : 0].callbacks.get(),

this,

*opts,

&alloc_info), 0);

session_.reset(session);

outgoing_storage_.reserve(1024);

outgoing_buffers_.reserve(32);

Local<Uint8Array> uint8_arr =

Uint8Array::New(js_fields_.GetArrayBuffer(), 0, kSessionUint8FieldCount);

USE(wrap->Set(env()->context(), env()->fields_string(), uint8_arr));

}

Http2Session::~Http2Session() {

CHECK(!is_in_scope());

Debug(this, "freeing nghttp2 session");

// Explicitly reset session_ so the subsequent

// current_nghttp2_memory_ check passes.

session_.reset();

CHECK_EQ(current_nghttp2_memory_, 0);

}

void Http2Session::MemoryInfo(MemoryTracker* tracker) const {

tracker->TrackField("streams", streams_);

tracker->TrackField("outstanding_pings", outstanding_pings_);

tracker->TrackField("outstanding_settings", outstanding_settings_);

tracker->TrackField("outgoing_buffers", outgoing_buffers_);

tracker->TrackFieldWithSize("stream_buf", stream_buf_.len);

tracker->TrackFieldWithSize("outgoing_storage", outgoing_storage_.size());

tracker->TrackFieldWithSize("pending_rst_streams",

pending_rst_streams_.size() * sizeof(int32_t));

tracker->TrackFieldWithSize("nghttp2_memory", current_nghttp2_memory_);

}

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

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

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

}

MaybeLocal<Object> Http2StreamPerformanceEntryTraits::GetDetails(

Environment* env,

const Http2StreamPerformanceEntry& entry) {

Local<Object> obj = Object::New(env->isolate());

#define SET(name, val) \

if (!obj->Set( \

env->context(), \

env->name(), \

Number::New( \

env->isolate(), \

static_cast<double>(entry.details.val))).IsJust()) { \

return MaybeLocal<Object>(); \

}

SET(bytes_read_string, received_bytes)

SET(bytes_written_string, sent_bytes)

SET(id_string, id)

#undef SET

#define SET(name, val) \

if (!obj->Set( \

env->context(), \

env->name(), \

Number::New( \

env->isolate(), \

(entry.details.val - entry.details.start_time) / 1e6)) \

.IsJust()) { \

return MaybeLocal<Object>(); \

}

SET(time_to_first_byte_string, first_byte)

SET(time_to_first_byte_sent_string, first_byte_sent)

SET(time_to_first_header_string, first_header)

#undef SET

return obj;

}

MaybeLocal<Object> Http2SessionPerformanceEntryTraits::GetDetails(

Environment* env,

const Http2SessionPerformanceEntry& entry) {

Local<Object> obj = Object::New(env->isolate());

#define SET(name, val) \

if (!obj->Set( \

env->context(), \

env->name(), \

Number::New( \

env->isolate(), \

static_cast<double>(entry.details.val))).IsJust()) { \

return MaybeLocal<Object>(); \

}

SET(bytes_written_string, data_sent)

SET(bytes_read_string, data_received)

SET(frames_received_string, frame_count)

SET(frames_sent_string, frame_sent)

SET(max_concurrent_streams_string, max_concurrent_streams)

SET(ping_rtt_string, ping_rtt)

SET(stream_average_duration_string, stream_average_duration)

SET(stream_count_string, stream_count)

if (!obj->Set(

env->context(),

env->type_string(),

OneByteString(

env->isolate(),

(entry.details.session_type == NGHTTP2_SESSION_SERVER)

? "server" : "client")).IsJust()) {

return MaybeLocal<Object>();

}

#undef SET

return obj;

}

void Http2Stream::EmitStatistics() {

CHECK_NOT_NULL(session());

if (LIKELY(!HasHttp2Observer(env())))

return;

double start = statistics_.start_time / 1e6;

double duration = (PERFORMANCE_NOW() / 1e6) - start;

std::unique_ptr<Http2StreamPerformanceEntry> entry =

std::make_unique<Http2StreamPerformanceEntry>(

"Http2Stream",

start - (env()->time_origin() / 1e6),

duration,

statistics_);

env()->SetImmediate([entry = move(entry)](Environment* env) {

if (HasHttp2Observer(env))

entry->Notify(env);

});

}

void Http2Session::EmitStatistics() {

if (LIKELY(!HasHttp2Observer(env())))

return;

double start = statistics_.start_time / 1e6;

double duration = (PERFORMANCE_NOW() / 1e6) - start;

std::unique_ptr<Http2SessionPerformanceEntry> entry =

std::make_unique<Http2SessionPerformanceEntry>(

"Http2Session",

start - (env()->time_origin() / 1e6),

duration,

statistics_);

env()->SetImmediate([entry = std::move(entry)](Environment* env) {

if (HasHttp2Observer(env))

entry->Notify(env);

});

}

// Closes the session and frees the associated resources

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

Debug(this, "closing session");

if (is_closing())

return;

set_closing();

// Stop reading on the i/o stream

if (stream_ != nullptr) {

set_reading_stopped();

stream_->ReadStop();

}

// If the socket is not closed, then attempt to send a closing GOAWAY

// frame. There is no guarantee that this GOAWAY will be received by

// the peer but the HTTP/2 spec recommends sending it anyway. We'll

// make a best effort.

if (!socket_closed) {

Debug(this, "terminating session with code %d", code);

CHECK_EQ(nghttp2_session_terminate_session(session_.get(), code), 0);

SendPendingData();

} else if (stream_ != nullptr) {

stream_->RemoveStreamListener(this);

}

set_destroyed();

// If we are writing we will get to make the callback in OnStreamAfterWrite.

if (!is_write_in_progress()) {

Debug(this, "make done session callback");

HandleScope scope(env()->isolate());

MakeCallback(env()->ondone_string(), 0, nullptr);

if (stream_ != nullptr) {

// Start reading again to detect the other end finishing.

set_reading_stopped(false);

stream_->ReadStart();

}

}

// If there are outstanding pings, those will need to be canceled, do

// so on the next iteration of the event loop to avoid calling out into

// javascript since this may be called during garbage collection.

while (BaseObjectPtr<Http2Ping> ping = PopPing()) {

ping->DetachFromSession();

env()->SetImmediate(

[ping = std::move(ping)](Environment* env) {

ping->Done(false);

});

}

statistics_.end_time = uv_hrtime();

EmitStatistics();

}

// Locates an existing known stream by ID. nghttp2 has a similar method

// but this is faster and does not fail if the stream is not found.

BaseObjectPtr<Http2Stream> Http2Session::FindStream(int32_t id) {

auto s = streams_.find(id);

return s != streams_.end() ? s->second : BaseObjectPtr<Http2Stream>();

}

bool Http2Session::CanAddStream() {

uint32_t maxConcurrentStreams =

nghttp2_session_get_local_settings(

session_.get(), NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS);

size_t maxSize =

std::min(streams_.max_size(), static_cast<size_t>(maxConcurrentStreams));

// We can add a new stream so long as we are less than the current

// maximum on concurrent streams and there's enough available memory

return streams_.size() < maxSize &&

has_available_session_memory(sizeof(Http2Stream));

}

void Http2Session::AddStream(Http2Stream* stream) {

CHECK_GE(++statistics_.stream_count, 0);

streams_[stream->id()] = BaseObjectPtr<Http2Stream>(stream);

size_t size = streams_.size();

if (size > statistics_.max_concurrent_streams)

statistics_.max_concurrent_streams = size;

IncrementCurrentSessionMemory(sizeof(*stream));

}

BaseObjectPtr<Http2Stream> Http2Session::RemoveStream(int32_t id) {

BaseObjectPtr<Http2Stream> stream;

if (streams_.empty())

return stream;

stream = FindStream(id);

if (stream) {

streams_.erase(id);

DecrementCurrentSessionMemory(sizeof(*stream));

}

return stream;

}

// Used as one of the Padding Strategy functions. Will attempt to ensure

// that the total frame size, including header bytes, are 8-byte aligned.

// If maxPayloadLen is smaller than the number of bytes necessary to align,

// will return maxPayloadLen instead.

ssize_t Http2Session::OnDWordAlignedPadding(size_t frameLen,

size_t maxPayloadLen) {

size_t r = (frameLen + 9) % 8;

if (r == 0) return frameLen; // If already a multiple of 8, return.

size_t pad = frameLen + (8 - r);

// If maxPayloadLen happens to be less than the calculated pad length,

// use the max instead, even tho this means the frame will not be

// aligned.

pad = std::min(maxPayloadLen, pad);

Debug(this, "using frame size padding: %d", pad);

return pad;

}

// Used as one of the Padding Strategy functions. Uses the maximum amount

// of padding allowed for the current frame.

ssize_t Http2Session::OnMaxFrameSizePadding(size_t frameLen,

size_t maxPayloadLen) {

Debug(this, "using max frame size padding: %d", maxPayloadLen);

return maxPayloadLen;

}

// Write data received from the i/o stream to the underlying nghttp2_session.

// On each call to nghttp2_session_mem_recv, nghttp2 will begin calling the

// various callback functions. Each of these will typically result in a call

// out to JavaScript so this particular function is rather hot and can be

// quite expensive. This is a potential performance optimization target later.

void Http2Session::ConsumeHTTP2Data() {

CHECK_NOT_NULL(stream_buf_.base);

CHECK_LE(stream_buf_offset_, stream_buf_.len);

size_t read_len = stream_buf_.len - stream_buf_offset_;

// multiple side effects.

Debug(this, "receiving %d bytes [wants data? %d]",

read_len,

nghttp2_session_want_read(session_.get()));

set_receive_paused(false);

custom_recv_error_code_ = nullptr;

ssize_t ret =

nghttp2_session_mem_recv(session_.get(),

reinterpret_cast<uint8_t*>(stream_buf_.base) +

stream_buf_offset_,

read_len);

CHECK_NE(ret, NGHTTP2_ERR_NOMEM);

CHECK_IMPLIES(custom_recv_error_code_ != nullptr, ret < 0);

if (is_receive_paused()) {

CHECK(is_reading_stopped());

CHECK_GT(ret, 0);

CHECK_LE(static_cast<size_t>(ret), read_len);

// Mark the remainder of the data as available for later consumption.

// Even if all bytes were received, a paused stream may delay the

// nghttp2_on_frame_recv_callback which may have an END_STREAM flag.

stream_buf_offset_ += ret;

goto done;

}

// We are done processing the current input chunk.

DecrementCurrentSessionMemory(stream_buf_.len);

stream_buf_offset_ = 0;

stream_buf_ab_.Reset();

stream_buf_allocation_.reset();

stream_buf_ = uv_buf_init(nullptr, 0);

// Send any data that was queued up while processing the received data.

if (ret >= 0 && !is_destroyed()) {

SendPendingData();

}

done:

if (UNLIKELY(ret < 0)) {

Isolate* isolate = env()->isolate();

Debug(this,

"fatal error receiving data: %d (%s)",

ret,

custom_recv_error_code_ != nullptr ?

custom_recv_error_code_ : "(no custom error code)");

Local<Value> args[] = {

Integer::New(isolate, static_cast<int32_t>(ret)),

Null(isolate)

};

if (custom_recv_error_code_ != nullptr) {

args[1] = String::NewFromUtf8(

isolate,

custom_recv_error_code_,

NewStringType::kInternalized).ToLocalChecked();

}

MakeCallback(

env()->http2session_on_error_function(),

arraysize(args),

args);

}

}

int32_t GetFrameID(const nghttp2_frame* frame) {

// If this is a push promise, we want to grab the id of the promised stream

return (frame->hd.type == NGHTTP2_PUSH_PROMISE) ?

frame->push_promise.promised_stream_id :

frame->hd.stream_id;

}

// Called by nghttp2 at the start of receiving a HEADERS frame. We use this

// callback to determine if a new stream is being created or if we are simply

// adding a new block of headers to an existing stream. The header pairs

// themselves are set in the OnHeaderCallback

int Http2Session::OnBeginHeadersCallback(nghttp2_session* handle,

const nghttp2_frame* frame,

void* user_data) {

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

int32_t id = GetFrameID(frame);

Debug(session, "beginning headers for stream %d", id);

BaseObjectPtr<Http2Stream> stream = session->FindStream(id);

// The common case is that we're creating a new stream. The less likely

// case is that we're receiving a set of trailers

if (LIKELY(!stream)) {

if (UNLIKELY(!session->CanAddStream() ||

Http2Stream::New(session, id, frame->headers.cat) ==

nullptr)) {

if (session->rejected_stream_count_++ >

session->js_fields_->max_rejected_streams)

return NGHTTP2_ERR_CALLBACK_FAILURE;

// Too many concurrent streams being opened

nghttp2_submit_rst_stream(

session->session(),

NGHTTP2_FLAG_NONE,

id,

NGHTTP2_ENHANCE_YOUR_CALM);

return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;

}

session->rejected_stream_count_ = 0;

} else if (!stream->is_destroyed()) {

stream->StartHeaders(frame->headers.cat);

}

return 0;

}

// Called by nghttp2 for each header name/value pair in a HEADERS block.

// This had to have been preceded by a call to OnBeginHeadersCallback so

// the Http2Stream is guaranteed to already exist.

int Http2Session::OnHeaderCallback(nghttp2_session* handle,

const nghttp2_frame* frame,

nghttp2_rcbuf* name,

nghttp2_rcbuf* value,

uint8_t flags,

void* user_data) {

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

int32_t id = GetFrameID(frame);

BaseObjectPtr<Http2Stream> stream = session->FindStream(id);

// If stream is null at this point, either something odd has happened

// or the stream was closed locally while header processing was occurring.

// either way, do not proceed and close the stream.

if (UNLIKELY(!stream))

return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;

// If the stream has already been destroyed, ignore.

if (!stream->is_destroyed() && !stream->AddHeader(name, value, flags)) {

// This will only happen if the connected peer sends us more

// than the allowed number of header items at any given time

stream->SubmitRstStream(NGHTTP2_ENHANCE_YOUR_CALM);

return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;

}

return 0;

}

// Called by nghttp2 when a complete HTTP2 frame has been received. There are

// only a handful of frame types that we care about handling here.

int Http2Session::OnFrameReceive(nghttp2_session* handle,

const nghttp2_frame* frame,

void* user_data) {

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

session->statistics_.frame_count++;

Debug(session, "complete frame received: type: %d",

frame->hd.type);

switch (frame->hd.type) {

case NGHTTP2_DATA:

return session->HandleDataFrame(frame);

case NGHTTP2_PUSH_PROMISE:

// Intentional fall-through, handled just like headers frames

case NGHTTP2_HEADERS:

session->HandleHeadersFrame(frame);

break;

case NGHTTP2_SETTINGS:

session->HandleSettingsFrame(frame);

break;

case NGHTTP2_PRIORITY:

session->HandlePriorityFrame(frame);

break;

case NGHTTP2_GOAWAY:

session->HandleGoawayFrame(frame);

break;

case NGHTTP2_PING:

session->HandlePingFrame(frame);

break;

case NGHTTP2_ALTSVC:

session->HandleAltSvcFrame(frame);

break;

case NGHTTP2_ORIGIN:

session->HandleOriginFrame(frame);

break;

default:

break;

}

return 0;

}

int Http2Session::OnInvalidFrame(nghttp2_session* handle,

const nghttp2_frame* frame,

int lib_error_code,

void* user_data) {

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

const uint32_t max_invalid_frames = session->js_fields_->max_invalid_frames;

Debug(session,

"invalid frame received (%u/%u), code: %d",

session->invalid_frame_count_,

max_invalid_frames,

lib_error_code);

if (session->invalid_frame_count_++ > max_invalid_frames) {

session->custom_recv_error_code_ = "ERR_HTTP2_TOO_MANY_INVALID_FRAMES";

return 1;