int PacketBuffer_Init (PacketBuffer *buf, PacketRecvInterface *input, PacketPassInterface *output, int num_packets, BPendingGroup *pg) { ASSERT(PacketPassInterface_GetMTU(output) >= PacketRecvInterface_GetMTU(input)) ASSERT(num_packets > 0) // init arguments buf->input = input; buf->output = output; // init input PacketRecvInterface_Receiver_Init(buf->input, (PacketRecvInterface_handler_done)input_handler_done, buf); // set input MTU buf->input_mtu = PacketRecvInterface_GetMTU(buf->input); // init output PacketPassInterface_Sender_Init(buf->output, (PacketPassInterface_handler_done)output_handler_done, buf); // allocate buffer int num_blocks = ChunkBuffer2_calc_blocks(buf->input_mtu, num_packets); if (num_blocks < 0) { goto fail0; } if (!(buf->buf_data = BAllocArray(num_blocks, sizeof(buf->buf_data[0])))) { goto fail0; } // init buffer ChunkBuffer2_Init(&buf->buf, buf->buf_data, num_blocks, buf->input_mtu); // schedule receive PacketRecvInterface_Receiver_Recv(buf->input, buf->buf.input_dest); DebugObject_Init(&buf->d_obj); return 1; fail0: return 0; }
int PacketProtoDecoder_Init (PacketProtoDecoder *enc, StreamRecvInterface *input, PacketPassInterface *output, BPendingGroup *pg, void *user, PacketProtoDecoder_handler_error handler_error) { // init arguments enc->input = input; enc->output = output; enc->user = user; enc->handler_error = handler_error; // init input StreamRecvInterface_Receiver_Init(enc->input, (StreamRecvInterface_handler_done)input_handler_done, enc); // init output PacketPassInterface_Sender_Init(enc->output, (PacketPassInterface_handler_done)output_handler_done, enc); // set output MTU, limit by maximum payload size enc->output_mtu = bmin_int(PacketPassInterface_GetMTU(enc->output), PACKETPROTO_MAXPAYLOAD); // init buffer state enc->buf_size = PACKETPROTO_ENCLEN(enc->output_mtu); enc->buf_start = 0; enc->buf_used = 0; // allocate buffer if (!(enc->buf = (uint8_t *)malloc(enc->buf_size))) { goto fail0; } // start receiving StreamRecvInterface_Receiver_Recv(enc->input, enc->buf, enc->buf_size); DebugObject_Init(&enc->d_obj); return 1; fail0: return 0; }