Method _collect_output

bessctl/test_utils.py:321–358

Collects output on self.sockets[] per ogates. Returns a dictionary indexed by ogate with a list of each packet wrapped by scapy.Ether.

(self, ogates, proto)

        return out_pkts

    def _collect_output(self, ogates, proto):
        """
        Collects output on self.sockets[] per ogates.  Returns a
        dictionary indexed by ogate with a list of each packet
        wrapped by scapy.Ether.
        """
        def get_all_pkts(sock):
            """
            Reads all immediately-available packets.  Wrecks
            the current timeout/nonblock setting.
            """
            ret = []
            sock.settimeout(0.0)
            while True:
                try:
                    received_data = sock.recv(2048)
                    ret.append(proto(received_data))
                # NB: sock.settimeout(0.0) logically should
                # produce a socket.timeout, not a socket.error
                # with EAGAIN, but in fact we get the latter.
                except socket.error as e:
                    if e.errno in (errno.EAGAIN, errno.EWOULDBLOCK):
                        return ret
                    raise
                except socket.timeout:
                    return ret

        timeout = {}
        out_pkts = {}
        for ogate in ogates:
            timeout[ogate] = self.sockets[ogate].gettimeout()
        try:
            for ogate in ogates:
                out_pkts[ogate] = get_all_pkts(self.sockets[ogate])
            return out_pkts
        finally:
            for ogate in ogates:
                self.sockets[ogate].settimeout(timeout[ogate])

    def run_module(self, module, igate, input_pkts, ogates=range(16),
                   time_out=3, proto=scapy.Ether):