103 lines
3 KiB
C
103 lines
3 KiB
C
#pragma once
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <time.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "config.h"
|
|
#include "frame.h"
|
|
#include "stream.h"
|
|
#include "libp2p/net/stream.h"
|
|
//#include "libp2p/yamux/yamux.h"
|
|
|
|
enum yamux_session_type
|
|
{
|
|
yamux_session_client,
|
|
yamux_session_server
|
|
};
|
|
enum yamux_error
|
|
{
|
|
yamux_error_normal = 0x00,
|
|
yamux_error_protoc = 0x01,
|
|
yamux_error_intern = 0x02
|
|
};
|
|
|
|
struct yamux_session;
|
|
struct yamux_stream;
|
|
|
|
typedef void* (*yamux_session_get_str_ud_fn)(struct yamux_session* session, yamux_streamid newid );
|
|
typedef void (*yamux_session_ping_fn )(struct yamux_session* session, uint32_t val );
|
|
typedef void (*yamux_session_pong_fn )(struct yamux_session* session, uint32_t val, struct timespec dt);
|
|
typedef void (*yamux_session_go_away_fn )(struct yamux_session* session, enum yamux_error err );
|
|
typedef void (*yamux_session_new_stream_fn)(struct yamux_session* session, struct yamux_stream* stream);
|
|
typedef void (*yamux_session_free_fn )(struct yamux_session* sesssion );
|
|
|
|
struct yamux_session_stream
|
|
{
|
|
struct yamux_stream* stream;
|
|
int alive;
|
|
};
|
|
struct yamux_session
|
|
{
|
|
struct yamux_config* config;
|
|
|
|
size_t num_streams;
|
|
size_t cap_streams;
|
|
struct yamux_session_stream* streams;
|
|
|
|
yamux_session_get_str_ud_fn get_str_ud_fn;
|
|
yamux_session_ping_fn ping_fn ;
|
|
yamux_session_pong_fn pong_fn ;
|
|
yamux_session_go_away_fn go_away_fn ;
|
|
yamux_session_new_stream_fn new_stream_fn;
|
|
yamux_session_free_fn free_fn ;
|
|
|
|
void* userdata;
|
|
|
|
struct timespec since_ping;
|
|
|
|
enum yamux_session_type type;
|
|
|
|
struct Stream* parent_stream;
|
|
|
|
yamux_streamid nextid;
|
|
|
|
int closed;
|
|
};
|
|
|
|
/***
|
|
* Create a new yamux session
|
|
* @param config the configuration
|
|
* @param sock the socket
|
|
* @param type session type (yamux_session_server or yamux_session_client)
|
|
* @param userdata user data
|
|
* @returns the yamux_session struct
|
|
*/
|
|
struct yamux_session* yamux_session_new(struct yamux_config* config, struct Stream* parent_stream, enum yamux_session_type type, void* userdata);
|
|
|
|
// does not close the socket, but does close the session
|
|
void yamux_session_free(struct yamux_session* session);
|
|
|
|
// does not free used memory
|
|
ssize_t yamux_session_close(struct yamux_session* session, enum yamux_error err);
|
|
|
|
inline ssize_t yamux_session_go_away(struct yamux_session* session, enum yamux_error err)
|
|
{
|
|
return yamux_session_close(session, err);
|
|
}
|
|
|
|
ssize_t yamux_session_ping(struct yamux_session* session, uint32_t value, int pong);
|
|
|
|
// defers to stream read handlers
|
|
ssize_t yamux_session_read(struct yamux_session* session);
|
|
|
|
struct YamuxChannelContext;
|
|
/**
|
|
* Decode an incoming message
|
|
* @param channel the channel
|
|
* @param incoming the incoming bytes
|
|
* @param incoming_size the size of the incoming bytes
|
|
* @returns true(1) on success, false(0) otherwise
|
|
*/
|
|
int yamux_decode(struct YamuxChannelContext* channel, const uint8_t* incoming, size_t incoming_size);
|