def send_file(self):
len_archive = popen("du -hsb " + self.filename).read().split("\t")[0]
print(len_archive)
print(self.filename)
self.file = open(self.filename, "rb")
self.send_initial_pkt()
count_part = 512
while (True):
self.recv_ack_pkt()
if (self.window.can_send_pkt()):
text = self.file.read(512)
count_part += len(text)
if (not text):
break
pkt = make_pkt(self.seq_num, connection_id=self.id, data=text)
self.send_pkt(pkt)
pkt = unpack(pkt)
self.window.buff.append(pkt)
if (self.window.base_equal_next_seq_num()):
pass
self.update_window()
self.file.close()
def run(self):
try:
client_id = 1
udp_port_client = 5000
self.sock = socket(AF_INET, SOCK_DGRAM)
self.sock.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
self.sock.bind(("", self.udp_port))
while (True):
data, addr = self.sock.recvfrom(524)
print("[<==] ", end=" ")
show_pkt(data)
pkt = unpack(data)
if (self.close_connection.is_set()):
break
if (not pkt['SYN']):
continue
hc = HandleClient(client_id, addr, udp_port_client,
self.file_dir)
hc.start()
self.clients.append(hc)
client_id += 1
udp_port_client += 1
except KeyboardInterrupt:
print('\nd Used Ctrl+C.')
print('d Closing server.')
self.close_all_connections()
def recv_ack_pkt(self):
def duplicate_ack(pkt):
if (self.recved_acks.count(pkt['ack_number']) == 3):
return True
return False
def index_pkt_ack(pkt):
for i, p in enumerate(self.window.buff):
if (pkt['ACK'] and is_ack_of(pkt, p)):
self.recved_acks.append(p['ack_number'])
return i + 1
data, _ = self.conn.recvfrom(524)
pkt = unpack(data)
index = index_pkt_ack(pkt)
if (index):
if (duplicate_ack(pkt)):
self.resend_pkt(index + 1)
self.window.ssthresh = self.window.size * 512 // 2
self.window.set_default()
self.c_duplicate_pkt += 1
else: #controle de congestionamento
self.window.base = index
if (self.window.size * 512 < self.window.ssthresh):
self.window.size *= 2
else:
self.window.size += 1
def send_initial_pkt(self):
text = self.file.read(512)
pkt = make_pkt(self.seq_num, self.ack_num, self.id, ACK=1, data=text)
self.send_pkt(pkt)
pkt = unpack(pkt)
self.window.buff.append(pkt)
self.update_window()
def run( self ):
try:
self.conn = socket( AF_INET, SOCK_DGRAM )
self.conn.setsockopt( SOL_SOCKET, SO_REUSEADDR, 1)
self.conn.bind( ( "", self.udp_port ) )
self.send_pkt_syn_ack()
file = open( self.file_dir + str( self.client_id ) + ".file" ,"wb" )
while( not self.close_connection.is_set() ):
self.conn.settimeout(10)
data, _ = self.conn.recvfrom( 524 )
if not data:
self.error(file)
continue
else:
self.conn.settimeout(None)
print("[<==] ",end=" ")
show_pkt(data)
pkt = unpack( data )
if( pkt['FIN'] ):
file.close()
self.close_tcp_connection( pkt )
self.close_connection.set()
break
else:
self.clear_buff( pkt )
if( self.duplicate_ack( pkt ) ):
make_pkt(self.data_to_send_acks[-1])
self.send_pkt( self.data_to_send_acks[-1] )
else:
# self.window.buff.append( pkt )
file.write( pkt['data'] )
self.seq_num = pkt['ack_number']
self.ack_num = pkt['seq_number'] + len_pkt( pkt )
pkt = make_pkt(seq_number=self.seq_num, ack_number=self.ack_num, connection_id=self.client_id, ACK=1)
self.send_pkt( pkt )
pkt = unpack( pkt )
self.data_to_send_acks.append( pkt )
except:
self.close()
def wait_ack_of_fin(self):
while (True):
data, _ = self.conn.recvfrom(524)
pkt = unpack(data)
if (pkt['ACK'] and is_ack_of(pkt, self.window.buff[-1])):
print('Received ACK of FIN.')
break
else:
print('Not received ACK pkt of FIN.')
def close(self):
pkt = make_pkt(self.seq_num,
self.ack_num,
connection_id=self.id,
FIN=1)
self.send_pkt(pkt)
pkt = unpack(pkt)
self.window.buff.append(pkt)
self.wait_ack_of_fin()
self.wait_for_fin()
print("d Number of duplicate packages :", self.c_duplicate_pkt)
def wait_for_fin(self):
self.conn.settimeout(2)
data, _ = self.conn.recvfrom(524)
if data:
pkt = unpack(data)
if (pkt['FIN']):
self.send_pkt_ack_for_fin(pkt)
print('Received pkt fin.')
else:
print('Not received pkt fin.')
else:
print('Timeout')
def wait_for_syn_ack(self):
while (True):
data, addr = self.conn.recvfrom(524)
pkt = unpack(data)
if (pkt['ACK'] and pkt['SYN']):
self.set_connection_options(pkt, addr)
print('Received ack and syn.')
self.ack_num = pkt['seq_number'] + 1
self.seq_num = pkt['ack_number']
break
else:
print('Not received ack and syn.')
def recv_ack_pkt(self):
# print( self.window )
def duplicate_ack(pkt):
if (self.recved_acks.count(pkt['ack_number']) == 4):
return True
return False
def index_pkt_ack(pkt):
for i, p in enumerate(self.window.buff):
if (pkt['ACK'] and is_ack_of(pkt, p)):
self.recved_acks.append(p['ack_number'])
return i + 1
# self.conn.settimeout( 0.1 )
# try:
data, _ = self.conn.recvfrom(524)
# except:
# self.c_rtrss_pkt += 1
# self.conn.settimeout(None)
if (self.c_rtrss_pkt == 5):
print('===== Start retranmission')
for p in self.window.buff[self.window.base:self.window.
next_seq_num]:
pkt = make_pkt(p['seq_number'],
p['ack_number'],
p['connection_id'],
data=p['data'])
self.send_pkt(pkt)
#print( "[<==] ",end=" ")
#show_pkt(data)
pkt = unpack(data)
index = index_pkt_ack(pkt)
if (index):
if (duplicate_ack(pkt)):
self.resend_pkt(index + 1)
self.window.ssthresh = self.window.size * 512 // 2
self.window.set_default()
self.c_duplicate_pkt += 1
else:
self.window.base = index
if (self.window.size * 512 < self.window.ssthresh):
self.window.size *= 2
else:
self.window.size += 1
def send_file(self):
def print_parts_file(number_parts, len_total):
system("clear")
print("d Sending file...")
aux = "#" * number_parts
aux2 = " " * (len_total - number_parts)
print("[" + aux + aux2 + "]")
len_archive = popen("du -hsb " + self.filename).read().split("\t")[0]
print(len_archive)
print(self.filename)
len_part = int(len_archive) // 60
self.file = open(self.filename, "rb")
self.send_initial_pkt()
count_part = 512
while (True):
self.recv_ack_pkt()
if (self.window.can_send_pkt()):
text = self.file.read(512)
count_part += len(text)
print_parts_file(count_part // len_part,
int(len_archive) // len_part)
if (not text):
break
pkt = make_pkt(self.seq_num, connection_id=self.id, data=text)
self.send_pkt(pkt)
pkt = unpack(pkt)
self.window.buff.append(pkt)
if (self.window.base_equal_next_seq_num()):
# start_time()
pass
self.update_window()
self.file.close()
