dummy_s.py

import socket
import argparse
import time
import os
from threading import Thread


class DummyS():

    def __init__(self, port, peer_list):
        self.port = port
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        self.sock.bind(('', self.port))
        self.peer_list = peer_list
        print("I'm a DummyS")

    def receive(self):
        try:
            msg, address = self.sock.recvfrom(5)
        except socket.timeout:
            msg = b'0'
            address = ()
        return (msg.decode('utf8'), address)

    def send(self, data, address):
        self.sock.sendto(str(data).encode('utf8'), address)

    def listen_to_the_team(self):
        print("Listening to the team")
        msg, address = self.receive()
        if int(msg) > 0:
            print("\033[91m MP detected in round {} by {} \033[0m"
                  .format(msg, address))
            os._exit(1)

    def run(self):
        data = 0
        _round = 0
        while True:
            _round += 1
            print("Round {}".format(_round))
            for p in self.peer_list:
                data += 1
                self.send(data, p)
                print("{} sent to {}".format(data, p))
                time.sleep(0.004*len(peer_list))


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-p", "--port", type=int,
                        help="UDP port used by peers")
    parser.add_argument("-z", "--size", type=int,
                        help="Team size (without splitter)")
    parser.add_argument("-e", "--extra_peers", default=0, type=int,
                        help="Peers out of the SDN")
    parser.add_argument("--split", default=False,
                        action='store_true',
                        help="Distribute the team in 2 switches")
    args = parser.parse_args()

    peer_list = []
    hosts = args.size+1
    if args.split:
        for p in range(0, hosts//2):
            peer_list.append(("10.0.0."+str(p+1), args.port))
        for p in range(hosts//2, hosts-1):
            peer_list.append(("11.0.0."+str(p+1), args.port))
        for p in range(hosts, hosts + args.extra_peers):
            peer_list.append(("172.31.31."+str(p+1), args.port))
    else:
        for p in range(0, hosts-1):
            peer_list.append(("10.0.0."+str(p+1), args.port))

    print("\nPeer List:{}".format(peer_list))
    peer = DummyS(args.port, peer_list)
    Thread(target=peer.listen_to_the_team, args=[]).start()
    peer.run()