dataCom/udp_server.py at master · stewberticus/dataCom · 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
__author__ = 'alex.stuart,Lauren.mills,Caleb Stevenson'
import socket               # Import socket module
import thread
import struct

# c is now serv_sock, addr is going to be the address of client
def clientconnection(c):
    while True:
        #receive 1024 bytes for the filename
        #l = c.recv(1024)
        print "Trying to connect"
        l, addr = c.recvfrom(1024)
        print 'Connected to', addr       # Confirm correct client
        if not l:
            break
        #decode the raw byte using UTF8
        filename = l.decode("utf-8")
        print "file:" + filename
        awktidbits = [0] * 256
        filemorsels = [0] * 256
        c.settimeout(1.0)

        #open the file request
        try:
            file = open(filename,'r')

        #read the file from disk in 1024 byte chunks
            l = file.read(1021)
            ll = "000"
            #print "awk num:" + ll

            ll += l
            #print "whole packet:" + ll
            filemorsels.append(l)
        except Exception as e:
	    print e
            #something went wrong with the file need to tell the client
            c.sendto("*****", addr)
            l = False
            file = False
        #while bytes to send
        i = 1
        startwindow = 0
        endwindow = 5
        no_timeout = True
        while(ll):
	    no_timeout = True
            while(l and i < endwindow):
                c.sendto(ll, addr)
                #read next 1024
                l = file.read(1021)
                if(ll):
                    #print type(l)"%02d"%a
                    ll = "%03d"%i
                    #print "chr of i " + ll
                    ll += l
                    filemorsels[i] = l
                print "i", i
                i+=1
                if(i == 256):
                    i = 0
            while(no_timeout):
                try:
                    print "listening"
		            data, addr = c.recvfrom(1024)
		            print data
		            resp_num = int(data)
		            awktidbits[resp_num] = True
		            for g in range(5):
			            if awktidbits[startwindow + g]:
			                newstartwindow = startwindow+g
			            else:
			                break
		            startwindow = newstartwindow
		            endwindow = startwindow + 5
		            print startwindow
		            print endwindow

                except Exception:
                    print "timeout!"
                    no_timeout = False

        if file:
            file.close()

        print "finished sending file"
    print 'Client ' + str(addr) + ' is no longer connected.'

serv_sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)# Create a socket object
this_machine_name = socket.gethostname() # Get local machine ip
#print this_machine_name

port = raw_input("Enter port number: ")
try:
    port = int(port)
    if(port < 0 or port > 65535):
        raise Exception
except Exception:
	print "not a valid port number"
	quit()                # Reserve the passed port for your service.
serv_sock.bind((this_machine_name, port))        # Bind to the port
   # Open a temp file to store the data
#serv_sock.listen(1)                 # Now wait for client connection.


clientconnection(serv_sock)


c.close()