openstack/deb-ryu
Code Issues Proposed changes

packet lib: calculate the udp length

Browse Source 
If the length is not given, calculate it.

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
FUJITA Tomonori 2012-08-27 11:37:26 +09:00
parent 79ae375136
commit 4e887d03c8
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ryu/lib/packet/udp.py
View File

@ -20,13 +20,12 @@ from . import packet_base
class udp(packet_base.PacketBase): class udp(packet_base.PacketBase):
_PACK_STR = '!HHHH' _PACK_STR = '!HHHH'
def __init__(self, src_port, dst_port, length, csum=0, data=None): def __init__(self, src_port, dst_port, length, csum=0):
super(udp, self).__init__() super(udp, self).__init__()
self.src_port = src_port self.src_port = src_port
self.dst_port = dst_port self.dst_port = dst_port
self.length = length self.length = length
self.csum = csum self.csum = csum
self.data = data
@classmethod @classmethod
def parser(cls, buf): def parser(cls, buf):
@ -39,5 +38,7 @@ class udp(packet_base.PacketBase):
return msg, None return msg, None
def serialize(self, payload, prev): def serialize(self, payload, prev):
if self.length == 0:
self.length = struct.calcsize(udp._PACK_STR) + len(payload)
return struct.pack(udp._PACK_STR, self.src_port, self.dst_port, return struct.pack(udp._PACK_STR, self.src_port, self.dst_port,
self.length, self.csum) self.length, self.csum)