| 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 |  | 
|---|
| 24 |     if (flags & ENET_PACKET_FLAG_NO_ALLOCATE) | 
|---|
| 25 |       packet -> data = (enet_uint8 *) data; | 
|---|
| 26 |     else | 
|---|
| 27 |     { | 
|---|
| 28 |        packet -> data = (enet_uint8 *) enet_malloc (dataLength); | 
|---|
| 29 |        if (packet -> data == NULL) | 
|---|
| 30 |        { | 
|---|
| 31 |           enet_free (packet); | 
|---|
| 32 |           return NULL; | 
|---|
| 33 |        } | 
|---|
| 34 |  | 
|---|
| 35 |        if (data != NULL) | 
|---|
| 36 |          memcpy (packet -> data, data, dataLength); | 
|---|
| 37 |     } | 
|---|
| 38 |  | 
|---|
| 39 |     packet -> referenceCount = 0; | 
|---|
| 40 |     packet -> flags = flags; | 
|---|
| 41 |     packet -> dataLength = dataLength; | 
|---|
| 42 |     packet -> freeCallback = NULL; | 
|---|
| 43 |  | 
|---|
| 44 |     return packet; | 
|---|
| 45 | } | 
|---|
| 46 |  | 
|---|
| 47 | /** Destroys the packet and deallocates its data. | 
|---|
| 48 |     @param packet packet to be destroyed | 
|---|
| 49 | */ | 
|---|
| 50 | void | 
|---|
| 51 | enet_packet_destroy (ENetPacket * packet) | 
|---|
| 52 | { | 
|---|
| 53 |     if (packet -> freeCallback != NULL) | 
|---|
| 54 |       (* packet -> freeCallback) (packet); | 
|---|
| 55 |     if (! (packet -> flags & ENET_PACKET_FLAG_NO_ALLOCATE)) | 
|---|
| 56 |       enet_free (packet -> data); | 
|---|
| 57 |     enet_free (packet); | 
|---|
| 58 | } | 
|---|
| 59 |  | 
|---|
| 60 | /** Attempts to resize the data in the packet to length specified in the  | 
|---|
| 61 |     dataLength parameter  | 
|---|
| 62 |     @param packet packet to resize | 
|---|
| 63 |     @param dataLength new size for the packet data | 
|---|
| 64 |     @returns 0 on success, < 0 on failure | 
|---|
| 65 | */ | 
|---|
| 66 | int | 
|---|
| 67 | enet_packet_resize (ENetPacket * packet, size_t dataLength) | 
|---|
| 68 | { | 
|---|
| 69 |     enet_uint8 * newData; | 
|---|
| 70 |     | 
|---|
| 71 |     if (dataLength <= packet -> dataLength || (packet -> flags & ENET_PACKET_FLAG_NO_ALLOCATE)) | 
|---|
| 72 |     { | 
|---|
| 73 |        packet -> dataLength = dataLength; | 
|---|
| 74 |  | 
|---|
| 75 |        return 0; | 
|---|
| 76 |     } | 
|---|
| 77 |  | 
|---|
| 78 |     newData = (enet_uint8 *) enet_malloc (dataLength); | 
|---|
| 79 |     if (newData == NULL) | 
|---|
| 80 |       return -1; | 
|---|
| 81 |  | 
|---|
| 82 |     memcpy (newData, packet -> data, packet -> dataLength); | 
|---|
| 83 |     enet_free (packet -> data); | 
|---|
| 84 |      | 
|---|
| 85 |     packet -> data = newData; | 
|---|
| 86 |     packet -> dataLength = dataLength; | 
|---|
| 87 |  | 
|---|
| 88 |     return 0; | 
|---|
| 89 | } | 
|---|
| 90 |  | 
|---|
| 91 | static int initializedCRC32 = 0; | 
|---|
| 92 | static enet_uint32 crcTable [256]; | 
|---|
| 93 |  | 
|---|
| 94 | static void initialize_crc32 () | 
|---|
| 95 | { | 
|---|
| 96 |     int byte; | 
|---|
| 97 |  | 
|---|
| 98 |     for (byte = 0; byte < 256; ++ byte) | 
|---|
| 99 |     { | 
|---|
| 100 |         enet_uint32 crc = byte << 24; | 
|---|
| 101 |         int offset; | 
|---|
| 102 |  | 
|---|
| 103 |         for(offset = 0; offset < 8; ++ offset) | 
|---|
| 104 |         { | 
|---|
| 105 |             if (crc & 0x80000000) | 
|---|
| 106 |                 crc = (crc << 1) ^ 0x04c11db7; | 
|---|
| 107 |             else | 
|---|
| 108 |                 crc <<= 1; | 
|---|
| 109 |         } | 
|---|
| 110 |  | 
|---|
| 111 |         crcTable [byte] = crc; | 
|---|
| 112 |     } | 
|---|
| 113 |  | 
|---|
| 114 |     initializedCRC32 = 1; | 
|---|
| 115 | } | 
|---|
| 116 |      | 
|---|
| 117 | enet_uint32 | 
|---|
| 118 | enet_crc32 (const ENetBuffer * buffers, size_t bufferCount) | 
|---|
| 119 | { | 
|---|
| 120 |     enet_uint32 crc = 0xFFFFFFFF; | 
|---|
| 121 |      | 
|---|
| 122 |     if (! initializedCRC32) initialize_crc32 (); | 
|---|
| 123 |  | 
|---|
| 124 |     while (bufferCount -- > 0) | 
|---|
| 125 |     { | 
|---|
| 126 |         const enet_uint8 * data = (const enet_uint8 *) buffers -> data, | 
|---|
| 127 |                          * dataEnd = & data [buffers -> dataLength]; | 
|---|
| 128 |  | 
|---|
| 129 |         while (data < dataEnd) | 
|---|
| 130 |         { | 
|---|
| 131 |             crc = ((crc << 8) | * data ++) ^ crcTable [crc >> 24];         | 
|---|
| 132 |         } | 
|---|
| 133 |  | 
|---|
| 134 |         ++ buffers; | 
|---|
| 135 |     } | 
|---|
| 136 |  | 
|---|
| 137 |     return ENET_HOST_TO_NET_32 (~ crc); | 
|---|
| 138 | } | 
|---|
| 139 |  | 
|---|
| 140 | /** @} */ | 
|---|