void free_up_io (BSocksClient *o)
{
// free sending
BConnection_SendAsync_Free(&o->con);
// free receiving
BConnection_RecvAsync_Free(&o->con);
}
int main ()
{
int ret = 1;
BLog_InitStdout();
// init network
if (!BNetwork_GlobalInit()) {
fprintf(stderr, "BNetwork_GlobalInit failed\n");
goto fail1;
}
// init reactor (event loop)
if (!BReactor_Init(&reactor)) {
fprintf(stderr, "BReactor_Init failed\n");
goto fail1;
}
// init signal handling
sigset_t set;
sigemptyset(&set);
sigaddset(&set, SIGINT);
sigaddset(&set, SIGTERM);
if (!BUnixSignal_Init(&usignal, &reactor, set, signal_handler, NULL)) {
fprintf(stderr, "BUnixSignal_Init failed\n");
goto fail2;
}
// init BConnection object backed by the stdin fd
if (!BConnection_Init(&pipe_con, BCONNECTION_SOURCE_PIPE(0), &reactor, NULL, connection_handler)) {
fprintf(stderr, "BConnection_Init failed\n");
goto fail3;
}
// init connection receive interface
BConnection_RecvAsync_Init(&pipe_con);
source_if = BConnection_RecvAsync_GetIf(&pipe_con);
// init receive done callback
StreamRecvInterface_Receiver_Init(source_if, input_handler_done, NULL);
// receive first chunk
StreamRecvInterface_Receiver_Recv(source_if, buf, BUF_SIZE);
// run event loop
ret = BReactor_Exec(&reactor);
BConnection_RecvAsync_Free(&pipe_con);
BConnection_Free(&pipe_con);
fail3:
BUnixSignal_Free(&usignal, 0);
fail2:
BReactor_Free(&reactor);
fail1:
BLog_Free();
DebugObjectGlobal_Finish();
return ret;
}
static void listener_handler (struct instance *o)
{
ASSERT(!o->dying)
BReactor *reactor = o->i->params->iparams->reactor;
BPendingGroup *pg = BReactor_PendingGroup(reactor);
struct connection *c = malloc(sizeof(*c));
if (!c) {
ModuleLog(o->i, BLOG_ERROR, "malloc failed");
goto fail0;
}
c->inst = o;
LinkedList0_Prepend(&o->connections_list, &c->connections_list_node);
if (!BConnection_Init(&c->con, BConnection_source_listener(&o->listener, &c->addr), reactor, c, (BConnection_handler)connection_con_handler)) {
ModuleLog(o->i, BLOG_ERROR, "BConnection_Init failed");
goto fail1;
}
BConnection_SendAsync_Init(&c->con);
BConnection_RecvAsync_Init(&c->con);
StreamPassInterface *con_send_if = BConnection_SendAsync_GetIf(&c->con);
StreamRecvInterface *con_recv_if = BConnection_RecvAsync_GetIf(&c->con);
PacketPassInterface_Init(&c->recv_if, RECV_MTU, (PacketPassInterface_handler_send)connection_recv_if_handler_send, c, pg);
if (!PacketProtoDecoder_Init(&c->recv_decoder, con_recv_if, &c->recv_if, pg, c, (PacketProtoDecoder_handler_error)connection_recv_decoder_handler_error)) {
ModuleLog(o->i, BLOG_ERROR, "PacketProtoDecoder_Init failed");
goto fail2;
}
PacketStreamSender_Init(&c->send_pss, con_send_if, PACKETPROTO_ENCLEN(SEND_MTU), pg);
PacketPassFifoQueue_Init(&c->send_queue, PacketStreamSender_GetInput(&c->send_pss), pg);
LinkedList0_Init(&c->requests_list);
LinkedList0_Init(&c->replies_list);
c->state = CONNECTION_STATE_RUNNING;
ModuleLog(o->i, BLOG_INFO, "connection initialized");
return;
fail2:
PacketPassInterface_Free(&c->recv_if);
BConnection_RecvAsync_Free(&c->con);
BConnection_SendAsync_Free(&c->con);
BConnection_Free(&c->con);
fail1:
LinkedList0_Remove(&o->connections_list, &c->connections_list_node);
free(c);
fail0:
return;
}
void free_control_io (BSocksClient *o)
{
// free sending
PacketStreamSender_Free(&o->control.send_sender);
BConnection_SendAsync_Free(&o->con);
// free receiving
BConnection_RecvAsync_Free(&o->con);
}
static void connection_free_link (struct connection *c)
{
PacketPassFifoQueue_PrepareFree(&c->send_queue);
LinkedList0Node *ln;
while (ln = LinkedList0_GetFirst(&c->replies_list)) {
struct reply *r = UPPER_OBJECT(ln, struct reply, replies_list_node);
ASSERT(r->con == c)
reply_free(r);
}
PacketPassFifoQueue_Free(&c->send_queue);
PacketStreamSender_Free(&c->send_pss);
PacketProtoDecoder_Free(&c->recv_decoder);
PacketPassInterface_Free(&c->recv_if);
BConnection_RecvAsync_Free(&c->con);
BConnection_SendAsync_Free(&c->con);
BConnection_Free(&c->con);
}
static void connection_free_connection (struct connection *o)
{
// disconnect read instance
if (o->read_inst) {
ASSERT(o->read_inst->con_inst == o)
o->read_inst->con_inst = NULL;
}
// disconnect write instance
if (o->write_inst) {
ASSERT(o->write_inst->con_inst == o)
o->write_inst->con_inst = NULL;
}
// free connection interfaces
BConnection_RecvAsync_Free(&o->connection);
BConnection_SendAsync_Free(&o->connection);
// free connection
BConnection_Free(&o->connection);
// free store
NCDBufStore_Free(&o->store);
}