X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=python%2Fovstest%2Fudp.py;h=fa03e74dbfd348ddfdf73f0c054b52084cb13d21;hb=49a4902d76b9149e4fa399bde6d5cb9a91e64c07;hp=e09569db6a49a7fbf91f1a668af06c4b54e17797;hpb=0be6140a9a7de46f07e09d3ba200bd7f0cf73838;p=openvswitch diff --git a/python/ovstest/udp.py b/python/ovstest/udp.py index e09569db..fa03e74d 100644 --- a/python/ovstest/udp.py +++ b/python/ovstest/udp.py @@ -1,4 +1,4 @@ -# Copyright (c) 2011 Nicira Networks +# Copyright (c) 2011, 2012 Nicira, Inc. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -16,9 +16,12 @@ ovsudp contains listener and sender classes for UDP protocol """ +import array +import struct +import time + from twisted.internet.protocol import DatagramProtocol from twisted.internet.task import LoopingCall -import array, struct, time class UdpListener(DatagramProtocol): @@ -28,18 +31,12 @@ class UdpListener(DatagramProtocol): def __init__(self): self.stats = [] - def startProtocol(self): - print "Starting UDP listener" - - def stopProtocol(self): - print "Stopping UDP listener" - def datagramReceived(self, data, (_1, _2)): """This function is called each time datagram is received""" try: self.stats.append(struct.unpack_from("Q", data, 0)) except struct.error: - pass #ignore packets that are less than 8 bytes of size + pass # ignore packets that are less than 8 bytes of size def getResults(self): """Returns number of packets that were actually received""" @@ -51,7 +48,7 @@ class UdpSender(DatagramProtocol): Class that will send UDP packets to UDP Listener """ def __init__(self, host, count, size, duration): - #LoopingCall does not know whether UDP socket is actually writable + # LoopingCall does not know whether UDP socket is actually writable self.looper = None self.host = host self.count = count @@ -61,13 +58,11 @@ class UdpSender(DatagramProtocol): self.data = array.array('c', 'X' * size) def startProtocol(self): - print "Starting UDP sender" self.looper = LoopingCall(self.sendData) period = self.duration / float(self.count) self.looper.start(period , now = False) def stopProtocol(self): - print "Stopping UDP sender" if (self.looper is not None): self.looper.stop() self.looper = None