81 lines
2.3 KiB
C
81 lines
2.3 KiB
C
#include <stdlib.h>
|
|
|
|
#include "multiaddr/multiaddr.h"
|
|
#include "libp2p/net/stream.h"
|
|
#include "libp2p/net/connectionstream.h"
|
|
#include "libp2p/yamux/yamux.h"
|
|
|
|
int libp2p_stream_default_handle_upgrade(struct Stream* parent_stream, struct Stream* new_stream) {
|
|
return libp2p_net_connection_upgrade(parent_stream, new_stream);
|
|
}
|
|
|
|
struct Stream* libp2p_stream_new() {
|
|
struct Stream* stream = (struct Stream*) malloc(sizeof(struct Stream));
|
|
if (stream != NULL) {
|
|
stream->address = NULL;
|
|
stream->close = NULL;
|
|
stream->parent_stream = NULL;
|
|
stream->peek = NULL;
|
|
stream->read = NULL;
|
|
stream->read_raw = NULL;
|
|
stream->socket_mutex = NULL;
|
|
stream->stream_context = NULL;
|
|
stream->write = NULL;
|
|
stream->handle_upgrade = libp2p_stream_default_handle_upgrade;
|
|
stream->channel = -1;
|
|
}
|
|
return stream;
|
|
}
|
|
|
|
void libp2p_stream_free(struct Stream* stream) {
|
|
if (stream != NULL) {
|
|
if (stream->socket_mutex != NULL) {
|
|
free(stream->socket_mutex);
|
|
stream->socket_mutex = NULL;
|
|
}
|
|
if (stream->address != NULL) {
|
|
multiaddress_free(stream->address);
|
|
stream->address = NULL;
|
|
}
|
|
free(stream);
|
|
}
|
|
}
|
|
|
|
int libp2p_stream_is_open(struct Stream* stream) {
|
|
if (stream == NULL)
|
|
return 0;
|
|
|
|
struct Stream* base_stream = stream;
|
|
while (base_stream->parent_stream != NULL)
|
|
base_stream = base_stream->parent_stream;
|
|
if (base_stream->stream_type == STREAM_TYPE_RAW) {
|
|
struct ConnectionContext* ctx = (struct ConnectionContext*)base_stream->stream_context;
|
|
if (ctx->socket_descriptor > 0)
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// forward declaration
|
|
struct YamuxChannelContext* libp2p_yamux_get_channel_context(void* stream_context);
|
|
|
|
/**
|
|
* Look for the latest stream
|
|
* (properly handles both raw streams and yamux streams)
|
|
* @param in the incoming stream
|
|
* @returns the latest child stream
|
|
*/
|
|
struct Stream* libp2p_stream_get_latest_stream(struct Stream* in) {
|
|
if (in == NULL)
|
|
return NULL;
|
|
if (in->stream_type == STREAM_TYPE_RAW) {
|
|
struct ConnectionContext* ctx = (struct ConnectionContext*)in->stream_context;
|
|
return ctx->session_context->default_stream;
|
|
} else if (in->stream_type == STREAM_TYPE_YAMUX) {
|
|
struct YamuxChannelContext* ctx = libp2p_yamux_get_channel_context(in->stream_context);
|
|
if (ctx != NULL)
|
|
return ctx->child_stream;
|
|
}
|
|
return libp2p_stream_get_latest_stream(in->parent_stream);
|
|
}
|