@@ -443,9 +443,12 @@ def packetTest(self):
443
443
444
444
self .sendMsg_TCP (PACKET_CLIENT_JOIN , payload )
445
445
446
-
446
+ def createPacketHeader (self , command , payload ):
447
+ return struct .pack (self .header_format , self .header_size + len (payload ), command )
448
+ def parsePacketHeader (self , header ):
449
+ return struct .unpack (self .header_format , header [:self .header_size ])
447
450
def sendMsg (self , type , command , payload = "" ):
448
- header = struct . pack ( self .header_format , self . header_size + len ( payload ), command )
451
+ header = self .createPacketHeader ( command , payload )
449
452
return self .sendRaw (header + payload , type )
450
453
def sendMsg_TCP (self , * args , ** kwargs ):
451
454
return self .sendMsg (M_TCP , * args , ** kwargs )
@@ -464,8 +467,8 @@ def receiveMsg_UDP(self, datapacket = False):
464
467
return None
465
468
data = self .socket_udp .recv (4096 )
466
469
#print data
467
- [ size , command ], osize = unpackFromExt ( self .header_format , data , 0 )
468
- LOG .debug ("received size: %d/%d , command: %d" % (size , osize , command ))
470
+ size , command = self .parsePacketHeader ( data )
471
+ LOG .debug ("received size: %d, command: %d" % (size , command ))
469
472
content = data [self .header_size :]
470
473
if datapacket :
471
474
return DataPacket (size , command , content )
@@ -488,8 +491,8 @@ def receiveMsg_TCP(self, datapacket = False):
488
491
if readcounter > 1 :
489
492
note += "HEADER SEGMENTED INTO %s SEGMENTS!" % readcounter
490
493
491
- ( size , command ) = struct . unpack ( self .header_format , data )
492
- if not command in [ PACKET_SERVER_FRAME , PACKET_SERVER_SYNC ] :
494
+ size , command = self .parsePacketHeader ( data )
495
+ if not command in ( PACKET_SERVER_FRAME , PACKET_SERVER_SYNC ) :
493
496
if command in packet_names :
494
497
LOG .debug ("received size: %d, command: %s (%d)" % (size , packet_names [command ], command ))
495
498
else :
0 commit comments