[1711] | 1 | /* |
---|
| 2 | * ORXONOX - the hottest 3D action shooter ever to exist |
---|
| 3 | * > www.orxonox.net < |
---|
| 4 | * |
---|
| 5 | * |
---|
| 6 | * License notice: |
---|
| 7 | * |
---|
| 8 | * This program is free software; you can redistribute it and/or |
---|
| 9 | * modify it under the terms of the GNU General Public License |
---|
| 10 | * as published by the Free Software Foundation; either version 2 |
---|
| 11 | * of the License, or (at your option) any later version. |
---|
| 12 | * |
---|
| 13 | * This program is distributed in the hope that it will be useful, |
---|
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 16 | * GNU General Public License for more details. |
---|
| 17 | * |
---|
| 18 | * You should have received a copy of the GNU General Public License |
---|
| 19 | * along with this program; if not, write to the Free Software |
---|
| 20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
---|
| 21 | * |
---|
| 22 | * Author: |
---|
| 23 | * Oliver Scheuss, (C) 2008 |
---|
| 24 | * Co-authors: |
---|
| 25 | * ... |
---|
| 26 | * |
---|
| 27 | */ |
---|
[1701] | 28 | |
---|
[1711] | 29 | |
---|
[1701] | 30 | #include <enet/enet.h> |
---|
[1711] | 31 | #include <boost/bind.hpp> |
---|
[1701] | 32 | |
---|
| 33 | #include "Packet.h" |
---|
| 34 | #include "network/ConnectionManager.h" |
---|
| 35 | #include "network/ClientInformation.h" |
---|
| 36 | |
---|
| 37 | #include "Acknowledgement.h" |
---|
| 38 | #include "Chat.h" |
---|
| 39 | #include "ClassID.h" |
---|
[1709] | 40 | #include "Gamestate.h" |
---|
[1711] | 41 | #include "Welcome.h" |
---|
[1701] | 42 | #include "network/Host.h" |
---|
[1705] | 43 | #include "core/CoreIncludes.h" |
---|
[1701] | 44 | |
---|
| 45 | namespace network{ |
---|
| 46 | |
---|
| 47 | namespace packet{ |
---|
| 48 | |
---|
[1711] | 49 | #define PACKET_FLAG_DEFAULT ENET_PACKET_FLAG_NO_ALLOCATE |
---|
[1701] | 50 | #define _PACKETID 0 |
---|
| 51 | |
---|
[1711] | 52 | std::map<ENetPacket *, Packet *> Packet::packetMap_; |
---|
[1701] | 53 | |
---|
| 54 | Packet::Packet(){ |
---|
[1711] | 55 | flags_ = PACKET_FLAG_DEFAULT; |
---|
[1701] | 56 | packetDirection_ = ENUM::Outgoing; |
---|
| 57 | clientID_=0; |
---|
[1711] | 58 | data_=0; |
---|
[1701] | 59 | enetPacket_=0; |
---|
| 60 | } |
---|
| 61 | |
---|
[1711] | 62 | Packet::Packet(unsigned char *data, int clientID){ |
---|
| 63 | flags_ = PACKET_FLAG_DEFAULT; |
---|
[1701] | 64 | packetDirection_ = ENUM::Outgoing; |
---|
[1711] | 65 | clientID_=clientID; |
---|
| 66 | data_=data; |
---|
[1701] | 67 | enetPacket_=0; |
---|
| 68 | } |
---|
| 69 | |
---|
| 70 | Packet::Packet(ENetPacket *packet, ENetPeer *peer){ |
---|
| 71 | packetDirection_ = ENUM::Incoming; |
---|
| 72 | enetPacket_ = packet; |
---|
| 73 | clientID_ = ClientInformation::findClient(&peer->address)->getID(); |
---|
[1711] | 74 | data_ = packet->data; |
---|
[1709] | 75 | enet_packet_destroy(packet); |
---|
[1701] | 76 | } |
---|
| 77 | |
---|
[1711] | 78 | Packet::Packet(const Packet &p){ |
---|
| 79 | flags_=p.flags_; |
---|
| 80 | if(p.data_){ |
---|
| 81 | data_ = new unsigned char[p.getSize()]; |
---|
| 82 | memcpy(data_, p.data_, p.getSize()); |
---|
| 83 | }else |
---|
| 84 | data_=0; |
---|
[1701] | 85 | } |
---|
| 86 | |
---|
[1711] | 87 | Packet::~Packet(){ |
---|
| 88 | if(enetPacket_) |
---|
[1705] | 89 | enet_packet_destroy(enetPacket_); |
---|
[1711] | 90 | if(data_) |
---|
| 91 | delete[] data_; |
---|
[1705] | 92 | } |
---|
[1701] | 93 | |
---|
| 94 | bool Packet::send(){ |
---|
[1705] | 95 | if(packetDirection_ != ENUM::Outgoing && packetDirection_ != ENUM::Bidirectional ) |
---|
[1701] | 96 | return false; |
---|
| 97 | if(!enetPacket_){ |
---|
[1711] | 98 | if(!data_) |
---|
[1701] | 99 | return false; |
---|
[1711] | 100 | enetPacket_ = enet_packet_create(getData(), getSize(), getFlags()); |
---|
| 101 | enetPacket_->freeCallback = &Packet::deletePacket; |
---|
| 102 | packetMap_[enetPacket_] = this; |
---|
[1701] | 103 | } |
---|
| 104 | network::Host::addPacket( enetPacket_, clientID_); |
---|
[1705] | 105 | enetPacket_ = 0; // otherwise we have a double free because enet already handles the deallocation of the packet |
---|
[1701] | 106 | return true; |
---|
| 107 | } |
---|
| 108 | |
---|
[1711] | 109 | Packet *Packet::createPacket(ENetPacket *packet, ENetPeer *peer){ |
---|
[1701] | 110 | unsigned char *data = packet->data; |
---|
[1711] | 111 | unsigned int clientID = ClientInformation::findClient(&peer->address)->getID(); |
---|
| 112 | Packet *p; |
---|
[1701] | 113 | switch( *(ENUM::Type *)&data[_PACKETID] ) |
---|
| 114 | { |
---|
| 115 | case ENUM::Acknowledgement: |
---|
[1711] | 116 | p = new Acknowledgement( data, clientID ); |
---|
[1701] | 117 | break; |
---|
| 118 | case ENUM::Chat: |
---|
[1711] | 119 | p = new Chat( data, clientID ); |
---|
[1701] | 120 | break; |
---|
| 121 | case ENUM::ClassID: |
---|
[1711] | 122 | p = new ClassID( data, clientID ); |
---|
[1701] | 123 | break; |
---|
| 124 | case ENUM::Gamestate: |
---|
| 125 | // TODO: remove brackets |
---|
[1711] | 126 | p = new Gamestate( data, true, clientID ); |
---|
[1701] | 127 | break; |
---|
[1711] | 128 | case ENUM::Welcome: |
---|
| 129 | p = new Welcome( data, clientID ); |
---|
[1701] | 130 | default: |
---|
[1710] | 131 | assert(0); //TODO: repair this |
---|
| 132 | break; |
---|
[1701] | 133 | } |
---|
[1711] | 134 | return p; |
---|
[1701] | 135 | } |
---|
| 136 | |
---|
[1711] | 137 | void Packet::deletePacket(ENetPacket *packet){ |
---|
| 138 | assert(packetMap_[packet]); |
---|
| 139 | delete packetMap_[packet]; |
---|
| 140 | } |
---|
| 141 | |
---|
[1701] | 142 | } // namespace packet |
---|
| 143 | |
---|
| 144 | } // namespace network |
---|
| 145 | |
---|