X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fconnection.cpp;h=b9c5d2ac8bac413c1264ac6746e873fca0c471e7;hb=8492796a5c45787810aad7fe08fc63cadc0c96b5;hp=0bc8492e6d482b34616bce20ed2e0aeba31b92ab;hpb=a26c92d7dda327f2b1483fe7250cb27580a0a039;p=oweals%2Fminetest.git diff --git a/src/connection.cpp b/src/connection.cpp index 0bc8492e6..b9c5d2ac8 100644 --- a/src/connection.cpp +++ b/src/connection.cpp @@ -20,6 +20,8 @@ with this program; if not, write to the Free Software Foundation, Inc., #include "connection.h" #include "main.h" #include "serialization.h" +#include "log.h" +#include "porting.h" namespace con { @@ -320,7 +322,7 @@ IncomingSplitBuffer::~IncomingSplitBuffer() This will throw a GotSplitPacketException when a full split packet is constructed. */ -void IncomingSplitBuffer::insert(BufferedPacket &p, bool reliable) +SharedBuffer IncomingSplitBuffer::insert(BufferedPacket &p, bool reliable) { u32 headersize = BASE_HEADER_SIZE + 7; assert(p.data.getSize() >= headersize); @@ -363,9 +365,9 @@ void IncomingSplitBuffer::insert(BufferedPacket &p, bool reliable) // Set chunk data in buffer sp->chunks[chunk_num] = chunkdata; - // If not all chunks are received, return + // If not all chunks are received, return empty buffer if(sp->allReceived() == false) - return; + return SharedBuffer(); // Calculate total size u32 totalsize = 0; @@ -392,8 +394,8 @@ void IncomingSplitBuffer::insert(BufferedPacket &p, bool reliable) // Remove sp from buffer m_buf.remove(seqnum); delete sp; - - throw GotSplitPacketException(fulldata); + + return fulldata; } void IncomingSplitBuffer::removeUnreliableTimedOuts(float dtime, float timeout) { @@ -439,16 +441,19 @@ Channel::~Channel() Peer */ -Peer::Peer(u16 a_id, Address a_address) +Peer::Peer(u16 a_id, Address a_address): + address(a_address), + id(a_id), + timeout_counter(0.0), + ping_timer(0.0), + resend_timeout(0.5), + avg_rtt(-1.0), + has_sent_with_id(false), + m_sendtime_accu(0), + m_max_packets_per_second(10), + m_num_sent(0), + m_max_num_sent(0) { - id = a_id; - address = a_address; - timeout_counter = 0.0; - //resend_timeout = RESEND_TIMEOUT_MINIMUM; - ping_timer = 0.0; - resend_timeout = 0.5; - avg_rtt = -1.0; - has_sent_with_id = false; } Peer::~Peer() { @@ -456,6 +461,20 @@ Peer::~Peer() void Peer::reportRTT(float rtt) { + if(rtt >= 0.0){ + if(rtt < 0.01){ + if(m_max_packets_per_second < 100) + m_max_packets_per_second += 10; + } else if(rtt < 0.2){ + if(m_max_packets_per_second < 100) + m_max_packets_per_second += 2; + } else { + m_max_packets_per_second *= 0.8; + if(m_max_packets_per_second < 10) + m_max_packets_per_second = 10; + } + } + if(rtt < -0.999) {} else if(avg_rtt < 0.0) @@ -485,205 +504,862 @@ void Peer::reportRTT(float rtt) Connection */ -Connection::Connection( - u32 protocol_id, - u32 max_packet_size, - float timeout, - PeerHandler *peerhandler -) +Connection::Connection(u32 protocol_id, u32 max_packet_size, float timeout): + m_protocol_id(protocol_id), + m_max_packet_size(max_packet_size), + m_timeout(timeout), + m_peer_id(0), + m_bc_peerhandler(NULL), + m_bc_receive_timeout(0), + m_indentation(0) { - assert(peerhandler != NULL); + m_socket.setTimeoutMs(5); - m_protocol_id = protocol_id; - m_max_packet_size = max_packet_size; - m_timeout = timeout; - m_peer_id = PEER_ID_NEW; - //m_waiting_new_peer_id = false; - m_indentation = 0; - m_peerhandler = peerhandler; + Start(); } -Connection::~Connection() +Connection::Connection(u32 protocol_id, u32 max_packet_size, float timeout, + PeerHandler *peerhandler): + m_protocol_id(protocol_id), + m_max_packet_size(max_packet_size), + m_timeout(timeout), + m_peer_id(0), + m_bc_peerhandler(peerhandler), + m_bc_receive_timeout(0), + m_indentation(0) { - // Clear peers - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - delete peer; - } + m_socket.setTimeoutMs(5); + + Start(); } -void Connection::Serve(unsigned short port) + +Connection::~Connection() { - m_socket.Bind(port); - m_peer_id = PEER_ID_SERVER; + stop(); } -void Connection::Connect(Address address) +/* Internal stuff */ + +void * Connection::Thread() { - core::map::Node *node = m_peers.find(PEER_ID_SERVER); - if(node != NULL){ - throw ConnectionException("Already connected to a server"); - } + ThreadStarted(); + log_register_thread("Connection"); - Peer *peer = new Peer(PEER_ID_SERVER, address); - m_peers.insert(peer->id, peer); - m_peerhandler->peerAdded(peer); - - m_socket.Bind(0); + dout_con<<"Connection thread started"< data(0); - Send(PEER_ID_SERVER, 0, data, true); + u32 curtime = porting::getTimeMs(); + u32 lasttime = curtime; + + while(getRun()) + { + BEGIN_DEBUG_EXCEPTION_HANDLER + + lasttime = curtime; + curtime = porting::getTimeMs(); + float dtime = (float)(curtime - lasttime) / 1000.; + if(dtime > 0.1) + dtime = 0.1; + if(dtime < 0.0) + dtime = 0.0; + + runTimeouts(dtime); + + while(m_command_queue.size() != 0){ + ConnectionCommand c = m_command_queue.pop_front(); + processCommand(c); + } + + send(dtime); + + receive(); + + END_DEBUG_EXCEPTION_HANDLER(derr_con); + } - //m_waiting_new_peer_id = true; + return NULL; } -void Connection::Disconnect() +void Connection::putEvent(ConnectionEvent &e) { - // Create and send DISCO packet - SharedBuffer data(2); - writeU8(&data[0], TYPE_CONTROL); - writeU8(&data[1], CONTROLTYPE_DISCO); - - // Send to all - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - SendAsPacket(peer->id, 0, data, false); - } + assert(e.type != CONNEVENT_NONE); + m_event_queue.push_back(e); } -bool Connection::Connected() +void Connection::processCommand(ConnectionCommand &c) { - if(m_peers.size() != 1) - return false; - - core::map::Node *node = m_peers.find(PEER_ID_SERVER); - if(node == NULL) - return false; - - if(m_peer_id == PEER_ID_NEW) - return false; - - return true; + switch(c.type){ + case CONNCMD_NONE: + dout_con< Channel::ProcessPacket( - SharedBuffer packetdata, - Connection *con, - u16 peer_id, - u8 channelnum, - bool reliable) +void Connection::send(float dtime) { - IndentationRaiser iraiser(&(con->m_indentation)); + for(core::map::Iterator + j = m_peers.getIterator(); + j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + peer->m_sendtime_accu += dtime; + peer->m_num_sent = 0; + peer->m_max_num_sent = peer->m_sendtime_accu * + peer->m_max_packets_per_second; + } + Queue postponed_packets; + while(m_outgoing_queue.size() != 0){ + OutgoingPacket packet = m_outgoing_queue.pop_front(); + Peer *peer = getPeerNoEx(packet.peer_id); + if(!peer) + continue; + if(peer->channels[packet.channelnum].outgoing_reliables.size() >= 5){ + postponed_packets.push_back(packet); + } else if(peer->m_num_sent < peer->m_max_num_sent){ + rawSendAsPacket(packet.peer_id, packet.channelnum, + packet.data, packet.reliable); + peer->m_num_sent++; + } else { + postponed_packets.push_back(packet); + } + } + while(postponed_packets.size() != 0){ + m_outgoing_queue.push_back(postponed_packets.pop_front()); + } + for(core::map::Iterator + j = m_peers.getIterator(); + j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + peer->m_sendtime_accu -= (float)peer->m_num_sent / + peer->m_max_packets_per_second; + if(peer->m_sendtime_accu > 10. / peer->m_max_packets_per_second) + peer->m_sendtime_accu = 10. / peer->m_max_packets_per_second; + } +} - if(packetdata.getSize() < 1) - throw InvalidIncomingDataException("packetdata.getSize() < 1"); +// Receive packets from the network and buffers and create ConnectionEvents +void Connection::receive() +{ + u32 datasize = 100000; + // TODO: We can not know how many layers of header there are. + // For now, just assume there are no other than the base headers. + u32 packet_maxsize = datasize + BASE_HEADER_SIZE; + SharedBuffer packetdata(packet_maxsize); - u8 type = readU8(&packetdata[0]); + bool single_wait_done = false; - if(type == TYPE_CONTROL) + for(;;) { - if(packetdata.getSize() < 2) - throw InvalidIncomingDataException("packetdata.getSize() < 2"); - - u8 controltype = readU8(&packetdata[1]); - - if(controltype == CONTROLTYPE_ACK) + try{ + /* Check if some buffer has relevant data */ { - if(packetdata.getSize() < 4) - throw InvalidIncomingDataException - ("packetdata.getSize() < 4 (ACK header size)"); - - u16 seqnum = readU16(&packetdata[2]); - con->PrintInfo(); - dout_con<<"Got CONTROLTYPE_ACK: channelnum=" - <<((int)channelnum&0xff)<<", peer_id="<PrintInfo(); - outgoing_reliables.print(); - dout_con<PrintInfo(derr_con); - derr_con<<"WARNING: ACKed packet not " - "in outgoing queue" - < resultdata; + bool got = getFromBuffers(peer_id, resultdata); + if(got){ + ConnectionEvent e; + e.dataReceived(peer_id, resultdata); + putEvent(e); + continue; } + } + + if(single_wait_done){ + if(m_socket.WaitData(0) == false) + break; + } + + single_wait_done = true; - throw ProcessedSilentlyException("Got an ACK"); + Address sender; + s32 received_size = m_socket.Receive(sender, *packetdata, packet_maxsize); + + if(received_size < 0) + break; + if(received_size < BASE_HEADER_SIZE) + continue; + if(readU32(&packetdata[0]) != m_protocol_id) + continue; + + u16 peer_id = readPeerId(*packetdata); + u8 channelnum = readChannel(*packetdata); + if(channelnum > CHANNEL_COUNT-1){ + PrintInfo(derr_con); + derr_con<<"Receive(): Invalid channel "<PrintInfo(); - dout_con<<"Got new peer id: "<GetPeerID() != PEER_ID_NEW) + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) { - con->PrintInfo(derr_con); - derr_con<<"WARNING: Not changing" - " existing peer id."<getValue(); + if(peer->has_sent_with_id) + continue; + if(peer->address == sender) + break; + } + + /* + If no peer was found with the same address and port, + we shall assume it is a new peer and create an entry. + */ + if(j.atEnd()) + { + // Pass on to adding the peer } + // Else: A peer was found. else { - dout_con<<"changing."<SetPeerID(peer_id_new); + Peer *peer = j.getNode()->getValue(); + peer_id = peer->id; + PrintInfo(derr_con); + derr_con<<"WARNING: Assuming unknown peer to be " + <<"peer_id="<PrintInfo(); - dout_con<<"PING"<PrintInfo(); - dout_con<<"DISCO: Removing peer "<<(peer_id)<deletePeer(peer_id) == false) + // Somebody wants to make a new connection + + // Get a unique peer id (2 or higher) + u16 peer_id_new = 2; + /* + Find an unused peer id + */ + bool out_of_ids = false; + for(;;) { - con->PrintInfo(derr_con); - derr_con<<"DISCO: Peer not found"<::Node *node = m_peers.find(peer_id); + + if(node == NULL) + { + // Peer not found + // This means that the peer id of the sender is not PEER_ID_INEXISTENT + // and it is invalid. + PrintInfo(derr_con); + derr_con<<"Receive(): Peer not found"<getValue(); + + // Validate peer address + if(peer->address != sender) + { + PrintInfo(derr_con); + derr_con<<"Peer "<timeout_counter = 0.0; + + Channel *channel = &(peer->channels[channelnum]); + + // Throw the received packet to channel->processPacket() + + // Make a new SharedBuffer from the data without the base headers + SharedBuffer strippeddata(received_size - BASE_HEADER_SIZE); + memcpy(*strippeddata, &packetdata[BASE_HEADER_SIZE], + strippeddata.getSize()); + + try{ + // Process it (the result is some data with no headers made by us) + SharedBuffer resultdata = processPacket + (channel, strippeddata, peer_id, channelnum, false); + + PrintInfo(); + dout_con<<"ProcessPacket returned data of size " + < timeouted_peers; + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + + /* + Check peer timeout + */ + peer->timeout_counter += dtime; + if(peer->timeout_counter > m_timeout) + { + PrintInfo(derr_con); + derr_con<<"RunTimeouts(): Peer "<id + <<" has timed out." + <<" (source=peer->timeout_counter)" + <id); + // Don't bother going through the buffers of this one + continue; + } + + float resend_timeout = peer->resend_timeout; + for(u16 i=0; i timed_outs; + core::list::Iterator j; + + Channel *channel = &peer->channels[i]; + + // Remove timed out incomplete unreliable split packets + channel->incoming_splits.removeUnreliableTimedOuts(dtime, m_timeout); + + // Increment reliable packet times + channel->outgoing_reliables.incrementTimeouts(dtime); + + // Check reliable packet total times, remove peer if + // over timeout. + if(channel->outgoing_reliables.anyTotaltimeReached(m_timeout)) + { + PrintInfo(derr_con); + derr_con<<"RunTimeouts(): Peer "<id + <<" has timed out." + <<" (source=reliable packet totaltime)" + <id); + goto nextpeer; + } + + // Re-send timed out outgoing reliables + + timed_outs = channel-> + outgoing_reliables.getTimedOuts(resend_timeout); + + channel->outgoing_reliables.resetTimedOuts(resend_timeout); + + j = timed_outs.begin(); + for(; j != timed_outs.end(); j++) + { + u16 peer_id = readPeerId(*(j->data)); + u8 channel = readChannel(*(j->data)); + u16 seqnum = readU16(&(j->data[BASE_HEADER_SIZE+1])); + + PrintInfo(derr_con); + derr_con<<"RE-SENDING timed-out RELIABLE to "; + j->address.print(&derr_con); + derr_con<<"(t/o="<::Node *node = m_peers.find(PEER_ID_SERVER); + if(node != NULL){ + throw ConnectionException("Already connected to a server"); + } + + Peer *peer = new Peer(PEER_ID_SERVER, address); + m_peers.insert(peer->id, peer); + + // Create event + ConnectionEvent e; + e.peerAdded(peer->id, peer->address); + putEvent(e); + + m_socket.Bind(0); + + // Send a dummy packet to server with peer_id = PEER_ID_INEXISTENT + m_peer_id = PEER_ID_INEXISTENT; + SharedBuffer data(0); + Send(PEER_ID_SERVER, 0, data, true); +} + +void Connection::disconnect() +{ + dout_con< data(2); + writeU8(&data[0], TYPE_CONTROL); + writeU8(&data[1], CONTROLTYPE_DISCO); + + // Send to all + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + rawSendAsPacket(peer->id, 0, data, false); + } +} + +void Connection::sendToAll(u8 channelnum, SharedBuffer data, bool reliable) +{ + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + send(peer->id, channelnum, data, reliable); + } +} + +void Connection::send(u16 peer_id, u8 channelnum, + SharedBuffer data, bool reliable) +{ + dout_con<address, data, + m_protocol_id, m_peer_id, channelnum); + + // Send the packet + rawSend(p); + } +} + +void Connection::rawSend(const BufferedPacket &packet) +{ + try{ + m_socket.Send(packet.address, *packet.data, packet.data.getSize()); + } catch(SendFailedException &e){ + derr_con<<"Connection::rawSend(): SendFailedException: " + <::Node *node = m_peers.find(peer_id); + + if(node == NULL){ + throw PeerNotFoundException("GetPeer: Peer not found (possible timeout)"); + } + + // Error checking + assert(node->getValue()->id == peer_id); + + return node->getValue(); +} + +Peer* Connection::getPeerNoEx(u16 peer_id) +{ + core::map::Node *node = m_peers.find(peer_id); + + if(node == NULL){ + return NULL; + } + + // Error checking + assert(node->getValue()->id == peer_id); + + return node->getValue(); +} + +core::list Connection::getPeers() +{ + core::list list; + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + list.push_back(peer); + } + return list; +} + +bool Connection::getFromBuffers(u16 &peer_id, SharedBuffer &dst) +{ + core::map::Iterator j; + j = m_peers.getIterator(); + for(; j.atEnd() == false; j++) + { + Peer *peer = j.getNode()->getValue(); + for(u16 i=0; ichannels[i]; + SharedBuffer resultdata; + bool got = checkIncomingBuffers(channel, peer_id, resultdata); + if(got){ + dst = resultdata; + return true; + } + } + } + return false; +} + +bool Connection::checkIncomingBuffers(Channel *channel, u16 &peer_id, + SharedBuffer &dst) +{ + u16 firstseqnum = 0; + // Clear old packets from start of buffer + try{ + for(;;){ + firstseqnum = channel->incoming_reliables.getFirstSeqnum(); + if(seqnum_higher(channel->next_incoming_seqnum, firstseqnum)) + channel->incoming_reliables.popFirst(); + else + break; + } + // This happens if all packets are old + }catch(con::NotFoundException) + {} + + if(channel->incoming_reliables.empty() == false) + { + if(firstseqnum == channel->next_incoming_seqnum) + { + BufferedPacket p = channel->incoming_reliables.popFirst(); + + peer_id = readPeerId(*p.data); + u8 channelnum = readChannel(*p.data); + u16 seqnum = readU16(&p.data[BASE_HEADER_SIZE+1]); + + PrintInfo(); + dout_con<<"UNBUFFERING TYPE_RELIABLE" + <<" seqnum="<outgoing_reliables.print(); + dout_con< Channel::ProcessPacket( if(packetdata.getSize() < ORIGINAL_HEADER_SIZE) throw InvalidIncomingDataException ("packetdata.getSize() < ORIGINAL_HEADER_SIZE"); - con->PrintInfo(); + PrintInfo(); dout_con<<"RETURNING TYPE_ORIGINAL to user" < Channel::ProcessPacket( // We have to create a packet again for buffering // This isn't actually too bad an idea. BufferedPacket packet = makePacket( - con->GetPeer(peer_id)->address, + getPeer(peer_id)->address, packetdata, - con->GetProtocolID(), + GetProtocolID(), peer_id, channelnum); - try{ - // Buffer the packet - incoming_splits.insert(packet, reliable); - } - // This exception happens when all the pieces of a packet - // are collected. - catch(GotSplitPacketException &e) + // Buffer the packet + SharedBuffer data = channel->incoming_splits.insert(packet, reliable); + if(data.getSize() != 0) { - con->PrintInfo(); + PrintInfo(); dout_con<<"RETURNING TYPE_SPLIT: Constructed full data, " - <<"size="< Channel::ProcessPacket( u16 seqnum = readU16(&packetdata[1]); - bool is_future_packet = seqnum_higher(seqnum, next_incoming_seqnum); - bool is_old_packet = seqnum_higher(next_incoming_seqnum, seqnum); + bool is_future_packet = seqnum_higher(seqnum, channel->next_incoming_seqnum); + bool is_old_packet = seqnum_higher(channel->next_incoming_seqnum, seqnum); - con->PrintInfo(); + PrintInfo(); if(is_future_packet) dout_con<<"BUFFERING"; else if(is_old_packet) @@ -748,25 +1420,25 @@ SharedBuffer Channel::ProcessPacket( else dout_con<<"RECUR"; dout_con<<" TYPE_RELIABLE seqnum="<incoming_reliables.size() < 100); // Send a CONTROLTYPE_ACK SharedBuffer reply(4); writeU8(&reply[0], TYPE_CONTROL); writeU8(&reply[1], CONTROLTYPE_ACK); writeU16(&reply[2], seqnum); - con->SendAsPacket(peer_id, channelnum, reply, false); + rawSendAsPacket(peer_id, channelnum, reply, false); - //if(seqnum_higher(seqnum, next_incoming_seqnum)) + //if(seqnum_higher(seqnum, channel->next_incoming_seqnum)) if(is_future_packet) { - /*con->PrintInfo(); + /*PrintInfo(); dout_con<<"Buffering reliable packet (seqnum=" < Channel::ProcessPacket( // Actually we have to make a packet to buffer one. // Well, we have all the ingredients, so just do it. BufferedPacket packet = makePacket( - con->GetPeer(peer_id)->address, + getPeer(peer_id)->address, packetdata, - con->GetProtocolID(), + GetProtocolID(), peer_id, channelnum); try{ - incoming_reliables.insert(packet); + channel->incoming_reliables.insert(packet); - /*con->PrintInfo(); + /*PrintInfo(); dout_con<<"INCOMING: "; - incoming_reliables.print(); + channel->incoming_reliables.print(); dout_con< Channel::ProcessPacket( throw ProcessedSilentlyException("Buffered future reliable packet"); } - //else if(seqnum_higher(next_incoming_seqnum, seqnum)) + //else if(seqnum_higher(channel->next_incoming_seqnum, seqnum)) else if(is_old_packet) { // An old packet, dump it throw InvalidIncomingDataException("Got an old reliable packet"); } - next_incoming_seqnum++; - - // Get out the inside packet and re-process it - SharedBuffer payload(packetdata.getSize() - RELIABLE_HEADER_SIZE); - memcpy(*payload, &packetdata[RELIABLE_HEADER_SIZE], payload.getSize()); - - return ProcessPacket(payload, con, peer_id, channelnum, true); - } - else - { - con->PrintInfo(derr_con); - derr_con<<"Got invalid type="<<((int)type&0xff)< Channel::CheckIncomingBuffers(Connection *con, - u16 &peer_id) -{ - u16 firstseqnum = 0; - // Clear old packets from start of buffer - try{ - for(;;){ - firstseqnum = incoming_reliables.getFirstSeqnum(); - if(seqnum_higher(next_incoming_seqnum, firstseqnum)) - incoming_reliables.popFirst(); - else - break; - } - // This happens if all packets are old - }catch(con::NotFoundException) - {} - - if(incoming_reliables.empty() == false) - { - if(firstseqnum == next_incoming_seqnum) - { - BufferedPacket p = incoming_reliables.popFirst(); - - peer_id = readPeerId(*p.data); - u8 channelnum = readChannel(*p.data); - u16 seqnum = readU16(&p.data[BASE_HEADER_SIZE+1]); - - con->PrintInfo(); - dout_con<<"UNBUFFERING TYPE_RELIABLE" - <<" seqnum="< Connection::GetFromBuffers(u16 &peer_id) -{ - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - for(u16 i=0; ichannels[i]; - try{ - SharedBuffer resultdata = channel->CheckIncomingBuffers - (this, peer_id); - - return resultdata; - } - catch(NoIncomingDataException &e) - { - } - catch(InvalidIncomingDataException &e) - { - } - catch(ProcessedSilentlyException &e) - { - } - } - } - throw NoIncomingDataException("No relevant data in buffers"); -} - -u32 Connection::Receive(u16 &peer_id, u8 *data, u32 datasize) -{ - /* - Receive a packet from the network - */ - - // TODO: We can not know how many layers of header there are. - // For now, just assume there are no other than the base headers. - u32 packet_maxsize = datasize + BASE_HEADER_SIZE; - Buffer packetdata(packet_maxsize); - - for(;;) - { - try - { - /* - Check if some buffer has relevant data - */ - try{ - SharedBuffer resultdata = GetFromBuffers(peer_id); - - if(datasize < resultdata.getSize()) - throw InvalidIncomingDataException - ("Buffer too small for received data"); - - memcpy(data, *resultdata, resultdata.getSize()); - return resultdata.getSize(); - } - catch(NoIncomingDataException &e) - { - } - - Address sender; - - s32 received_size = m_socket.Receive(sender, *packetdata, packet_maxsize); - - if(received_size < 0) - throw NoIncomingDataException("No incoming data"); - if(received_size < BASE_HEADER_SIZE) - throw InvalidIncomingDataException("No full header received"); - if(readU32(&packetdata[0]) != m_protocol_id) - throw InvalidIncomingDataException("Invalid protocol id"); - - peer_id = readPeerId(*packetdata); - u8 channelnum = readChannel(*packetdata); - if(channelnum > CHANNEL_COUNT-1){ - PrintInfo(derr_con); - derr_con<<"Receive(): Invalid channel "<::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - if(peer->has_sent_with_id) - continue; - if(peer->address == sender) - break; - } - - /* - If no peer was found with the same address and port, - we shall assume it is a new peer and create an entry. - */ - if(j.atEnd()) - { - // Pass on to adding the peer - } - // Else: A peer was found. - else - { - Peer *peer = j.getNode()->getValue(); - peer_id = peer->id; - PrintInfo(derr_con); - derr_con<<"WARNING: Assuming unknown peer to be " - <<"peer_id="<getValue(); - - // Validate peer address - if(peer->address != sender) - { - PrintInfo(derr_con); - derr_con<<"Peer "<timeout_counter = 0.0; - - Channel *channel = &(peer->channels[channelnum]); - - // Throw the received packet to channel->processPacket() - - // Make a new SharedBuffer from the data without the base headers - SharedBuffer strippeddata(received_size - BASE_HEADER_SIZE); - memcpy(*strippeddata, &packetdata[BASE_HEADER_SIZE], - strippeddata.getSize()); - - try{ - // Process it (the result is some data with no headers made by us) - SharedBuffer resultdata = channel->ProcessPacket - (strippeddata, this, peer_id, channelnum); - - PrintInfo(); - dout_con<<"ProcessPacket returned data of size " - < data, bool reliable) -{ - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) + channel->next_incoming_seqnum++; + + // Get out the inside packet and re-process it + SharedBuffer payload(packetdata.getSize() - RELIABLE_HEADER_SIZE); + memcpy(*payload, &packetdata[RELIABLE_HEADER_SIZE], payload.getSize()); + + return processPacket(channel, payload, peer_id, channelnum, true); + } + else { - Peer *peer = j.getNode()->getValue(); - Send(peer->id, channelnum, data, reliable); + PrintInfo(derr_con); + derr_con<<"Got invalid type="<<((int)type&0xff)< data, bool reliable) +bool Connection::deletePeer(u16 peer_id, bool timeout) { - assert(channelnum < CHANNEL_COUNT); + if(m_peers.find(peer_id) == NULL) + return false; - Peer *peer = GetPeer(peer_id); - Channel *channel = &(peer->channels[channelnum]); + Peer *peer = m_peers[peer_id]; - u32 chunksize_max = m_max_packet_size - BASE_HEADER_SIZE; - if(reliable) - chunksize_max -= RELIABLE_HEADER_SIZE; + // Create event + ConnectionEvent e; + e.peerRemoved(peer_id, timeout, peer->address); + putEvent(e); - core::list > originals; - originals = makeAutoSplitPacket(data, chunksize_max, - channel->next_outgoing_split_seqnum); - - core::list >::Iterator i; - i = originals.begin(); - for(; i != originals.end(); i++) - { - SharedBuffer original = *i; - - SendAsPacket(peer_id, channelnum, original, reliable); - } + delete m_peers[peer_id]; + m_peers.remove(peer_id); + return true; } -void Connection::SendAsPacket(u16 peer_id, u8 channelnum, - SharedBuffer data, bool reliable) -{ - Peer *peer = GetPeer(peer_id); - Channel *channel = &(peer->channels[channelnum]); - - if(reliable) - { - u16 seqnum = channel->next_outgoing_seqnum; - channel->next_outgoing_seqnum++; - - SharedBuffer reliable = makeReliablePacket(data, seqnum); +/* Interface */ - // Add base headers and make a packet - BufferedPacket p = makePacket(peer->address, reliable, - m_protocol_id, m_peer_id, channelnum); - - try{ - // Buffer the packet - channel->outgoing_reliables.insert(p); - } - catch(AlreadyExistsException &e) - { - PrintInfo(derr_con); - derr_con<<"WARNING: Going to send a reliable packet " - "seqnum="<address, data, - m_protocol_id, m_peer_id, channelnum); + return m_event_queue.pop_front(); +} - // Send the packet - RawSend(p); +ConnectionEvent Connection::waitEvent(u32 timeout_ms) +{ + try{ + return m_event_queue.pop_front(timeout_ms); + } catch(ItemNotFoundException &ex){ + ConnectionEvent e; + e.type = CONNEVENT_NONE; + return e; } } -void Connection::RawSend(const BufferedPacket &packet) +void Connection::putCommand(ConnectionCommand &c) { - m_socket.Send(packet.address, *packet.data, packet.data.getSize()); + m_command_queue.push_back(c); } -void Connection::RunTimeouts(float dtime) +void Connection::Serve(unsigned short port) { - core::list timeouted_peers; - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - - /* - Check peer timeout - */ - peer->timeout_counter += dtime; - if(peer->timeout_counter > m_timeout) - { - PrintInfo(derr_con); - derr_con<<"RunTimeouts(): Peer "<id - <<" has timed out." - <<" (source=peer->timeout_counter)" - <id); - // Don't bother going through the buffers of this one - continue; - } - - float resend_timeout = peer->resend_timeout; - for(u16 i=0; i timed_outs; - core::list::Iterator j; - - Channel *channel = &peer->channels[i]; - - // Remove timed out incomplete unreliable split packets - channel->incoming_splits.removeUnreliableTimedOuts(dtime, m_timeout); - - // Increment reliable packet times - channel->outgoing_reliables.incrementTimeouts(dtime); - - // Check reliable packet total times, remove peer if - // over timeout. - if(channel->outgoing_reliables.anyTotaltimeReached(m_timeout)) - { - PrintInfo(derr_con); - derr_con<<"RunTimeouts(): Peer "<id - <<" has timed out." - <<" (source=reliable packet totaltime)" - <id); - goto nextpeer; - } - - // Re-send timed out outgoing reliables - - timed_outs = channel-> - outgoing_reliables.getTimedOuts(resend_timeout); - - channel->outgoing_reliables.resetTimedOuts(resend_timeout); - - j = timed_outs.begin(); - for(; j != timed_outs.end(); j++) - { - u16 peer_id = readPeerId(*(j->data)); - u8 channel = readChannel(*(j->data)); - u16 seqnum = readU16(&(j->data[BASE_HEADER_SIZE+1])); - - PrintInfo(derr_con); - derr_con<<"RE-SENDING timed-out RELIABLE to "; - j->address.print(&derr_con); - derr_con<<"(t/o="<deletingPeer(m_peers[*i], true); - delete m_peers[*i]; - m_peers.remove(*i); - } + core::map::Node *node = m_peers.find(PEER_ID_SERVER); + if(node == NULL) + return false; + + if(m_peer_id == PEER_ID_INEXISTENT) + return false; + + return true; } -Peer* Connection::GetPeer(u16 peer_id) +void Connection::Disconnect() { - core::map::Node *node = m_peers.find(peer_id); + ConnectionCommand c; + c.disconnect(); + putCommand(c); +} - if(node == NULL){ - // Peer not found - throw PeerNotFoundException("Peer not found (possible timeout)"); +u32 Connection::Receive(u16 &peer_id, SharedBuffer &data) +{ + for(;;){ + ConnectionEvent e = waitEvent(m_bc_receive_timeout); + if(e.type != CONNEVENT_NONE) + dout_con<(e.data); + return e.data.getSize(); + case CONNEVENT_PEER_ADDED: { + Peer tmp(e.peer_id, e.address); + if(m_bc_peerhandler) + m_bc_peerhandler->peerAdded(&tmp); + continue; } + case CONNEVENT_PEER_REMOVED: { + Peer tmp(e.peer_id, e.address); + if(m_bc_peerhandler) + m_bc_peerhandler->deletingPeer(&tmp, e.timeout); + continue; } + } } + throw NoIncomingDataException("No incoming data"); +} - // Error checking - assert(node->getValue()->id == peer_id); +void Connection::SendToAll(u8 channelnum, SharedBuffer data, bool reliable) +{ + assert(channelnum < CHANNEL_COUNT); - return node->getValue(); + ConnectionCommand c; + c.sendToAll(channelnum, data, reliable); + putCommand(c); } -Peer* Connection::GetPeerNoEx(u16 peer_id) +void Connection::Send(u16 peer_id, u8 channelnum, + SharedBuffer data, bool reliable) { - core::map::Node *node = m_peers.find(peer_id); + assert(channelnum < CHANNEL_COUNT); - if(node == NULL){ - return NULL; - } + ConnectionCommand c; + c.send(peer_id, channelnum, data, reliable); + putCommand(c); +} - // Error checking - assert(node->getValue()->id == peer_id); +void Connection::RunTimeouts(float dtime) +{ + // No-op +} - return node->getValue(); +Address Connection::GetPeerAddress(u16 peer_id) +{ + JMutexAutoLock peerlock(m_peers_mutex); + return getPeer(peer_id)->address; } -core::list Connection::GetPeers() +float Connection::GetPeerAvgRTT(u16 peer_id) { - core::list list; - core::map::Iterator j; - j = m_peers.getIterator(); - for(; j.atEnd() == false; j++) - { - Peer *peer = j.getNode()->getValue(); - list.push_back(peer); - } - return list; + JMutexAutoLock peerlock(m_peers_mutex); + return getPeer(peer_id)->avg_rtt; } -bool Connection::deletePeer(u16 peer_id) +void Connection::DeletePeer(u16 peer_id) { - if(m_peers.find(peer_id) == NULL) - return false; - m_peerhandler->deletingPeer(m_peers[peer_id], true); - delete m_peers[peer_id]; - m_peers.remove(peer_id); - return true; + ConnectionCommand c; + c.deletePeer(peer_id); + putCommand(c); } void Connection::PrintInfo(std::ostream &out) { - out<