static TCPChild* _tcpchild_new(TCP* tcp, TCP* parent, in_addr_t peerIP, in_port_t peerPort) {
MAGIC_ASSERT(tcp);
MAGIC_ASSERT(parent);
TCPChild* child = g_new0(TCPChild, 1);
MAGIC_INIT(child);
/* my parent can find me by my key */
child->key = utility_ipPortHash(peerIP, peerPort);
descriptor_ref(tcp);
child->tcp = tcp;
descriptor_ref(parent);
child->parent = parent;
child->state = TCPCS_INCOMPLETE;
socket_setPeerName(&(child->tcp->super), peerIP, peerPort);
/* the child is bound to the parent server's address, because all packets
* coming from the child should appear to be coming from the server itself */
socket_setSocketName(&(child->tcp->super), socket_getBinding(&(parent->super)),
parent->super.boundPort);
return child;
}
static in_addr_t tcp_getIP(TCP* tcp) {
in_addr_t ip = 0;
if(tcp->server) {
ip = socket_getBinding(&(tcp->super));
if(ip == 0) {
ip = tcp->server->lastIP;
}
} else if(tcp->child) {
ip = socket_getBinding(&(tcp->child->parent->super));
if(ip == 0) {
ip = tcp->child->parent->server->lastIP;
}
} else {
ip = socket_getBinding(&(tcp->super));
}
return ip;
}
/*
* this function builds a UDP packet and sends to the virtual node given by the
* ip and port parameters. this function assumes that the socket is already
* bound to a local port, no matter if that happened explicitly or implicitly.
*/
gssize udp_sendUserData(UDP* udp, gconstpointer buffer, gsize nBytes, in_addr_t ip, in_port_t port) {
MAGIC_ASSERT(udp);
gsize space = socket_getOutputBufferSpace(&(udp->super));
if(space < nBytes) {
/* not enough space to buffer the data */
return -1;
}
/* break data into segments and send each in a packet */
gsize maxPacketLength = CONFIG_DATAGRAM_MAX_SIZE;
gsize remaining = nBytes;
gsize offset = 0;
/* create as many packets as needed */
while(remaining > 0) {
gsize copyLength = MIN(maxPacketLength, remaining);
/* use default destination if none was specified */
in_addr_t destinationIP = (ip != 0) ? ip : udp->super.peerIP;
in_port_t destinationPort = (port != 0) ? port : udp->super.peerPort;
/* create the UDP packet */
Packet* packet = packet_new(buffer + offset, copyLength);
packet_setUDP(packet, PUDP_NONE, socket_getBinding(&(udp->super)),
udp->super.boundPort, destinationIP, destinationPort);
/* buffer it in the transport layer, to be sent out when possible */
gboolean success = socket_addToOutputBuffer((Socket*) udp, packet);
/* counter maintenance */
if(success) {
remaining -= copyLength;
offset += copyLength;
} else {
warning("unable to send UDP packet");
break;
}
}
/* update the tracker output buffer stats */
Tracker* tracker = node_getTracker(worker_getPrivate()->cached_node);
Socket* socket = (Socket* )udp;
Descriptor* descriptor = (Descriptor *)socket;
gsize outLength = socket_getOutputBufferLength(socket);
gsize outSize = socket_getOutputBufferSize(socket);
tracker_updateSocketOutputBuffer(tracker, descriptor->handle, outLength, outSize);
debug("buffered %"G_GSIZE_FORMAT" outbound UDP bytes from user", offset);
return (gssize) offset;
}
