UDP-socket/socket_server.py at master · csecrestjr/UDP-socket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import socket
import sys
import select
import os
from geopy import geocoders
from datetime import datetime
import pytz

class Socket_Server:

    def __init__(self,type,port):

        self.server_address = (socket.gethostname(), port)
        self.server_ip = socket.gethostbyname(socket.getfqdn())
        self.unknown_command_error = "Unknown command"

        if type.upper() == "UDP":
            # Create UDP socket
            self.sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM) # UDP
            self.sock.bind(self.server_address)
            self.manage_conections_udp()

    def list_files(self):
        dirs = os.listdir(os.path.dirname(os.path.abspath(__file__)))
        files = " ".join(str(x) for x in dirs)
        return files

    def get_sendPacket(data,sock,data_size):
        packets = ["%s"%data[i:i+data_size] for i in range(0,len(data),data_size)]
        packets[-1] = packets[-1] + "\x00"*(len(data)%data_size)
        for p in packets:
        sock.sendto(p,*ADDRINFO)
        print(sendPacket("data",sock,120))
        now = datetime.now(pytz.timezone(timezone.zone))
        return now.ctime()

    def get_command_and_param(self,data):
        param = data.decode().rsplit(None, 2)[1:]
        param = " ".join(str(x) for x in param)
        data = data.decode().rsplit(None, 2)[0]
        return (data,param)

    def manage_conections_udp(self):
        while True:
            data, address = self.sock.recvfrom(1024)
            data,param = self.get_command_and_param(data)
            if data == "hostname":
                self.sock.sendto(socket.gethostname().encode(encoding='utf_4'), address)
            elif data == "server-ip":
                self.sock.sendto(self.server_ip.encode(encoding='utf_2'), address)
            elif data == "list-messages":
                self.sock.sendto(self.list_messages().encode(encoding='utf_8'), address)
            elif data == "checksum":
                self.sock.sendto(self.get_checksum(param).encode(encoding='utf_4'), address)
            elif data == "time":
                self.sock.sendto(self.get_time(param).encode(encoding='utf_8'), address)
            elif data == "timestamp":
                self.sock.sendto(self.get_(param).encode(encoding='utf_8'), address)
                print("goit")
                # data, addr = self.sock.recvfrom(1024)
                # f = open(data.decode(), 'wb')
                #
                # data, addr = self.sock.recvfrom(1024)
                # print("goit3")
                # try:
                #     while (data):
                #         f.write(data)
                #         self.sock.settimeout(2)
                #         data, addr = self.sock.recvfrom(1024)
                #     print("Message Sent")
                # except socket.timeout:
                #     f.close()
                #     self.sock.close()
                #     print("Message Recieved")


    def manage_conections_udp(self):

        while True:
            # Wait for a connection
            #print ('waiting for a connection')
            connection, client_address = self.sock.accept()
            try:
                #print ('connection from', client_address)

                # Receive the data in small chunks and retransmit it
                while True:
                    data = connection.recv(1024)
                    data, param = self.get_command_and_param(data)
                    if data == "hostname":
                        connection.sendall(socket.gethostname().encode(encoding='utf_4'))
                    elif data == "server-ip":
                        connection.sendall(self.server_ip.encode(encoding='utf_2'))
                    elif data == "list-messages":
                        connection.sendall(self.list_messages().encode(encoding='utf_8'))
                    elif data == "checksum":
                        connection.sendall(self.get_checksum(param).encode(encoding='utf_4')
                    elif data == "time":
                        connection.sendall(self.get_time(param).encode(encoding='utf_8'))
                    else:
                        connection.sendall(self.unknown_command_error.encode(encoding='utf_8'))
                        break

            finally:
                # Clean up the connection
                connection.close()


#newSoc = Socket_Server("UDP",6868);