95 lines
2.6 KiB
C
95 lines
2.6 KiB
C
#pragma once
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#include "session.h"
|
|
#include "libp2p/conn/session.h"
|
|
#include "libp2p/yamux/yamux.h"
|
|
|
|
// forward declarations
|
|
struct YamuxChannelContext;
|
|
|
|
// NOTE: 'data' is not guaranteed to be preserved when the read_fn
|
|
// handler exists (read: it will be freed).
|
|
struct yamux_stream;
|
|
|
|
typedef void (*yamux_stream_read_fn)(struct yamux_stream* stream, uint32_t data_length, void* data);
|
|
typedef void (*yamux_stream_fin_fn )(struct yamux_stream* stream);
|
|
typedef void (*yamux_stream_rst_fn )(struct yamux_stream* stream);
|
|
typedef void (*yamux_stream_free_fn)(struct yamux_stream* stream);
|
|
|
|
enum yamux_stream_state
|
|
{
|
|
yamux_stream_inited,
|
|
yamux_stream_syn_sent,
|
|
yamux_stream_syn_recv,
|
|
yamux_stream_est,
|
|
yamux_stream_closing,
|
|
yamux_stream_closed
|
|
};
|
|
|
|
struct yamux_stream
|
|
{
|
|
struct yamux_session* session;
|
|
|
|
yamux_stream_read_fn read_fn;
|
|
yamux_stream_fin_fn fin_fn ;
|
|
yamux_stream_rst_fn rst_fn ;
|
|
yamux_stream_free_fn free_fn;
|
|
|
|
struct Stream* stream;
|
|
|
|
enum yamux_stream_state state;
|
|
|
|
yamux_streamid id;
|
|
|
|
uint32_t window_size;
|
|
};
|
|
|
|
/**
|
|
* Create a new stream that has a YamuxChannelContext
|
|
* @param context the Yamux context
|
|
* @param id the stream id
|
|
* @param msg the incoming message
|
|
* @returns a stream that is a Yamux channel
|
|
*/
|
|
struct Stream* yamux_channel_new(struct YamuxContext* context, yamux_streamid id, struct StreamMessage* msg);
|
|
|
|
// not obligatory, SYN is sent by yamux_stream_write when the stream
|
|
// isn't initialised anyway
|
|
ssize_t yamux_stream_init (struct YamuxChannelContext* channel_ctx);
|
|
|
|
/***
|
|
* Closes the stream
|
|
* NOTE: doesn't free the stream, uses FIN
|
|
* @param context the YamuxContext or YamuxChannelContext
|
|
* @returns number of bytes sent
|
|
*/
|
|
ssize_t yamux_stream_close(void* context);
|
|
// uses RST
|
|
ssize_t yamux_stream_reset(struct YamuxChannelContext* stream);
|
|
|
|
void yamux_stream_free(struct yamux_stream* stream);
|
|
|
|
ssize_t yamux_stream_window_update(struct YamuxChannelContext* ctx, int32_t delta);
|
|
ssize_t yamux_stream_write(struct YamuxChannelContext* ctx, uint32_t data_length, void* data);
|
|
|
|
/***
|
|
* process stream
|
|
* @param stream the stream
|
|
* @param frame the frame
|
|
* @param incoming the stream bytes (after the frame)
|
|
* @param incoming_size the size of incoming
|
|
* @returns the number of bytes processed (can be zero) or negative number on error
|
|
*/
|
|
ssize_t yamux_stream_process(struct yamux_stream* stream, struct yamux_frame* frame, const uint8_t* incoming, size_t incoming_size);
|
|
|
|
/**
|
|
* Retrieve the flags for this context
|
|
* @param context the context
|
|
* @returns the correct flag
|
|
*/
|
|
enum yamux_frame_flags get_flags(void* context);
|
|
|