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); }