#pragma once

#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS

#if HAVE_OPENSSL && NODE_OPENSSL_HAS_QUIC

#include "bindingdata.h"

#include "session.h"

#include "sessionticket.h"

#include "streams.h"

namespace node {

namespace quic {

class Session::Application : public MemoryRetainer {

public:

using Options = Session::Application_Options;

Application(Session* session, const Options& options);

Application(const Application&) = delete;

Application(Application&&) = delete;

Application& operator=(const Application&) = delete;

Application& operator=(Application&&) = delete;

virtual bool Start();

// Session will forward all received stream data immediately on to the

// Application. The only additional processing the Session does is to

// automatically adjust the session-level flow control window. It is up to

// the Application to do the same for the Stream-level flow control.

virtual bool ReceiveStreamData(Stream* stream,

const uint8_t* data,

size_t datalen,

Stream::ReceiveDataFlags flags) = 0;

// Session will forward all data acknowledgements for a stream to the

// Application.

virtual void AcknowledgeStreamData(Stream* stream, size_t datalen);

// Called to determine if a Header can be added to this application.

// Applications that do not support headers will always return false.

virtual bool CanAddHeader(size_t current_count,

size_t current_headers_length,

size_t this_header_length);

// Called to mark the identified stream as being blocked. Not all

// Application types will support blocked streams, and those that do will do

// so differently.

virtual void BlockStream(int64_t id);

// Called when the session determines that there is outbound data available

// to send for the given stream.

virtual void ResumeStream(int64_t id);

// Called when the Session determines that the maximum number of

// remotely-initiated unidirectional streams has been extended. Not all

// Application types will require this notification so the default is to do

// nothing.

virtual void ExtendMaxStreams(EndpointLabel label,

Direction direction,

uint64_t max_streams);

// Called when the Session determines that the flow control window for the

// given stream has been expanded. Not all Application types will require

// this notification so the default is to do nothing.

virtual void ExtendMaxStreamData(Stream* stream, uint64_t max_data);

// Different Applications may wish to set some application data in the

// session ticket (e.g. http/3 would set server settings in the application

// data). By default, there's nothing to set.

virtual void CollectSessionTicketAppData(

SessionTicket::AppData* app_data) const;

// Different Applications may set some application data in the session

// ticket (e.g. http/3 would set server settings in the application data).

// By default, there's nothing to get.

virtual SessionTicket::AppData::Status ExtractSessionTicketAppData(

const SessionTicket::AppData& app_data,

SessionTicket::AppData::Source::Flag flag);

// Notifies the Application that the identified stream has been closed.

virtual void StreamClose(Stream* stream, QuicError error = QuicError());

// Notifies the Application that the identified stream has been reset.

virtual void StreamReset(Stream* stream,

uint64_t final_size,

QuicError error);

// Notifies the Application that the identified stream should stop sending.

virtual void StreamStopSending(Stream* stream, QuicError error);

// Submits an outbound block of headers for the given stream. Not all

// Application types will support headers, in which case this function

// should return false.

virtual bool SendHeaders(const Stream& stream,

HeadersKind kind,

const v8::Local<v8::Array>& headers,

HeadersFlags flags = HeadersFlags::NONE);

// Signals to the Application that it should serialize and transmit any

// pending session and stream packets it has accumulated.

void SendPendingData();

// Set the priority level of the stream if supported by the application. Not

// all applications support priorities, in which case this function is a

// non-op.

virtual void SetStreamPriority(

const Stream& stream,

StreamPriority priority = StreamPriority::DEFAULT,

StreamPriorityFlags flags = StreamPriorityFlags::NONE);

// Get the priority level of the stream if supported by the application. Not

// all applications support priorities, in which case this function returns

// the default stream priority.

virtual StreamPriority GetStreamPriority(const Stream& stream);

protected:

inline Environment* env() const { return session_->env(); }

inline Session& session() { return *session_; }

BaseObjectPtr<Packet> CreateStreamDataPacket();

struct StreamData;

virtual int GetStreamData(StreamData* data) = 0;

virtual bool StreamCommit(StreamData* data, size_t datalen) = 0;

virtual bool ShouldSetFin(const StreamData& data) = 0;

// Write the given stream_data into the buffer.

ssize_t WriteVStream(PathStorage* path,

uint8_t* buf,

ssize_t* ndatalen,

const StreamData& stream_data);

private:

Session* session_;

};

} // namespace quic

} // namespace node

#endif // HAVE_OPENSSL && NODE_OPENSSL_HAS_QUIC

#endif // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS