| 1 | /** | 
|---|
| 2 | @file  packet.c | 
|---|
| 3 | @brief ENet packet management functions | 
|---|
| 4 | */ | 
|---|
| 5 | #include <string.h> | 
|---|
| 6 | #define ENET_BUILDING_LIB 1 | 
|---|
| 7 | #include "enet/enet.h" | 
|---|
| 8 |  | 
|---|
| 9 | /** @defgroup Packet ENet packet functions | 
|---|
| 10 | @{ | 
|---|
| 11 | */ | 
|---|
| 12 |  | 
|---|
| 13 | /** Creates a packet that may be sent to a peer. | 
|---|
| 14 | @param dataContents initial contents of the packet's data; the packet's data will remain uninitialized if dataContents is NULL. | 
|---|
| 15 | @param dataLength   size of the data allocated for this packet | 
|---|
| 16 | @param flags        flags for this packet as described for the ENetPacket structure. | 
|---|
| 17 | @returns the packet on success, NULL on failure | 
|---|
| 18 | */ | 
|---|
| 19 | ENetPacket * | 
|---|
| 20 | enet_packet_create (const void * data, size_t dataLength, enet_uint32 flags) | 
|---|
| 21 | { | 
|---|
| 22 | ENetPacket * packet = (ENetPacket *) enet_malloc (sizeof (ENetPacket)); | 
|---|
| 23 | if (packet == NULL) | 
|---|
| 24 | return NULL; | 
|---|
| 25 |  | 
|---|
| 26 | if (flags & ENET_PACKET_FLAG_NO_ALLOCATE) | 
|---|
| 27 | packet -> data = (enet_uint8 *) data; | 
|---|
| 28 | else | 
|---|
| 29 | { | 
|---|
| 30 | packet -> data = (enet_uint8 *) enet_malloc (dataLength); | 
|---|
| 31 | if (packet -> data == NULL) | 
|---|
| 32 | { | 
|---|
| 33 | enet_free (packet); | 
|---|
| 34 | return NULL; | 
|---|
| 35 | } | 
|---|
| 36 |  | 
|---|
| 37 | if (data != NULL) | 
|---|
| 38 | memcpy (packet -> data, data, dataLength); | 
|---|
| 39 | } | 
|---|
| 40 |  | 
|---|
| 41 | packet -> referenceCount = 0; | 
|---|
| 42 | packet -> flags = flags; | 
|---|
| 43 | packet -> dataLength = dataLength; | 
|---|
| 44 | packet -> freeCallback = NULL; | 
|---|
| 45 |  | 
|---|
| 46 | return packet; | 
|---|
| 47 | } | 
|---|
| 48 |  | 
|---|
| 49 | /** Destroys the packet and deallocates its data. | 
|---|
| 50 | @param packet packet to be destroyed | 
|---|
| 51 | */ | 
|---|
| 52 | void | 
|---|
| 53 | enet_packet_destroy (ENetPacket * packet) | 
|---|
| 54 | { | 
|---|
| 55 | if (packet -> freeCallback != NULL) | 
|---|
| 56 | (* packet -> freeCallback) (packet); | 
|---|
| 57 | if (! (packet -> flags & ENET_PACKET_FLAG_NO_ALLOCATE)) | 
|---|
| 58 | enet_free (packet -> data); | 
|---|
| 59 | enet_free (packet); | 
|---|
| 60 | } | 
|---|
| 61 |  | 
|---|
| 62 | /** Attempts to resize the data in the packet to length specified in the | 
|---|
| 63 | dataLength parameter | 
|---|
| 64 | @param packet packet to resize | 
|---|
| 65 | @param dataLength new size for the packet data | 
|---|
| 66 | @returns 0 on success, < 0 on failure | 
|---|
| 67 | */ | 
|---|
| 68 | int | 
|---|
| 69 | enet_packet_resize (ENetPacket * packet, size_t dataLength) | 
|---|
| 70 | { | 
|---|
| 71 | enet_uint8 * newData; | 
|---|
| 72 |  | 
|---|
| 73 | if (dataLength <= packet -> dataLength || (packet -> flags & ENET_PACKET_FLAG_NO_ALLOCATE)) | 
|---|
| 74 | { | 
|---|
| 75 | packet -> dataLength = dataLength; | 
|---|
| 76 |  | 
|---|
| 77 | return 0; | 
|---|
| 78 | } | 
|---|
| 79 |  | 
|---|
| 80 | newData = (enet_uint8 *) enet_malloc (dataLength); | 
|---|
| 81 | if (newData == NULL) | 
|---|
| 82 | return -1; | 
|---|
| 83 |  | 
|---|
| 84 | memcpy (newData, packet -> data, packet -> dataLength); | 
|---|
| 85 | enet_free (packet -> data); | 
|---|
| 86 |  | 
|---|
| 87 | packet -> data = newData; | 
|---|
| 88 | packet -> dataLength = dataLength; | 
|---|
| 89 |  | 
|---|
| 90 | return 0; | 
|---|
| 91 | } | 
|---|
| 92 |  | 
|---|
| 93 | static int initializedCRC32 = 0; | 
|---|
| 94 | static enet_uint32 crcTable [256]; | 
|---|
| 95 |  | 
|---|
| 96 | static enet_uint32 | 
|---|
| 97 | reflect_crc (int val, int bits) | 
|---|
| 98 | { | 
|---|
| 99 | int result = 0, bit; | 
|---|
| 100 |  | 
|---|
| 101 | for (bit = 0; bit < bits; bit ++) | 
|---|
| 102 | { | 
|---|
| 103 | if(val & 1) result |= 1 << (bits - 1 - bit); | 
|---|
| 104 | val >>= 1; | 
|---|
| 105 | } | 
|---|
| 106 |  | 
|---|
| 107 | return result; | 
|---|
| 108 | } | 
|---|
| 109 |  | 
|---|
| 110 | static void | 
|---|
| 111 | initialize_crc32 () | 
|---|
| 112 | { | 
|---|
| 113 | int byte; | 
|---|
| 114 |  | 
|---|
| 115 | for (byte = 0; byte < 256; ++ byte) | 
|---|
| 116 | { | 
|---|
| 117 | enet_uint32 crc = reflect_crc (byte, 8) << 24; | 
|---|
| 118 | int offset; | 
|---|
| 119 |  | 
|---|
| 120 | for(offset = 0; offset < 8; ++ offset) | 
|---|
| 121 | { | 
|---|
| 122 | if (crc & 0x80000000) | 
|---|
| 123 | crc = (crc << 1) ^ 0x04c11db7; | 
|---|
| 124 | else | 
|---|
| 125 | crc <<= 1; | 
|---|
| 126 | } | 
|---|
| 127 |  | 
|---|
| 128 | crcTable [byte] = reflect_crc (crc, 32); | 
|---|
| 129 | } | 
|---|
| 130 |  | 
|---|
| 131 | initializedCRC32 = 1; | 
|---|
| 132 | } | 
|---|
| 133 |  | 
|---|
| 134 | enet_uint32 | 
|---|
| 135 | enet_crc32 (const ENetBuffer * buffers, size_t bufferCount) | 
|---|
| 136 | { | 
|---|
| 137 | enet_uint32 crc = 0xFFFFFFFF; | 
|---|
| 138 |  | 
|---|
| 139 | if (! initializedCRC32) initialize_crc32 (); | 
|---|
| 140 |  | 
|---|
| 141 | while (bufferCount -- > 0) | 
|---|
| 142 | { | 
|---|
| 143 | const enet_uint8 * data = (const enet_uint8 *) buffers -> data, | 
|---|
| 144 | * dataEnd = & data [buffers -> dataLength]; | 
|---|
| 145 |  | 
|---|
| 146 | while (data < dataEnd) | 
|---|
| 147 | { | 
|---|
| 148 | crc = (crc >> 8) ^ crcTable [(crc & 0xFF) ^ *data++]; | 
|---|
| 149 | } | 
|---|
| 150 |  | 
|---|
| 151 | ++ buffers; | 
|---|
| 152 | } | 
|---|
| 153 |  | 
|---|
| 154 | return ENET_HOST_TO_NET_32 (~ crc); | 
|---|
| 155 | } | 
|---|
| 156 |  | 
|---|
| 157 | /** @} */ | 
|---|