示例#1
0
文件: main.py 项目: mhdr/Thesis
def run_server():
    global HOST
    global PORT
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind((HOST, PORT))
    s.listen(10)

    while True:
        conn, addr = s.accept()
        print("{0} - ".format(get_counter()) + 'Connection from : {0}'.format(addr))

        data_recv=conn.recv(1024)
        data=bytearray()

        conn.settimeout(1)

        while len(data_recv)>0:
            data.extend(data_recv)

            if len(data_recv) < 1024:
                break

            try:
                data_recv=conn.recv(1024)
            except socket.timeout:
                break

        data=str(data)
        if data=="1":
            message=get_message1()
            conn.sendall(message)
        elif data=="2":
            global ifname
            current_mac= MAC.get_local_mac(ifname)
            conn.sendall(current_mac)

        conn.close()