def __forward_packet(self, ttl: str, packet: str) -> None:
""" Forward a supe packet in the net to superpeer
:param ip_sender: ip address of sender host
:param ttl: packet time to live
:param packet: string representing the packet
:return: None
"""
new_ttl = int(ttl) - 1
if new_ttl > 0:
packet = packet[:80] + str(new_ttl).zfill(2) + packet[82:]
superpeer_ip4 = LocalData.get_superpeer_ip4()
superpeer_ip6 = LocalData.get_superpeer_ip6()
superpeer_port = LocalData.get_superpeer_port()
net_utils.send_packet_and_close(superpeer_ip4, superpeer_ip6,
superpeer_port, packet)
def serve(choice: str) -> None:
""" Handle the peer packet
:param choice: the choice to handle
:return: None
"""
sp_ip4 = LocalData.get_superpeer_ip4()
sp_ip6 = LocalData.get_superpeer_ip6()
sp_port = LocalData.get_superpeer_port()
p_ssid = LocalData.session_id
if choice == "ADFF":
# check se shared dir esiste
if not os.path.exists('shared'):
shell.print_red('\nCannot find the shared folder.')
return
# se non ci sono file nella shared dir avvisa ed esce
if not os.scandir('shared'):
shell.print_yellow(
'No file available for sharing. Add files to shared dir to get started.\n'
)
return
temp_files = []
shell.print_blue('\nFiles available for sharing:')
for count, file in enumerate(os.scandir('shared'), 1):
# stampa i risultati della scandir
file_md5 = hasher.get_md5(f'shared/{file.name}')
print(f'{count}] {file.name} | {file_md5}')
temp_files.append((file_md5, file.name))
while True:
index = input('Choose a file to share (q to cancel): ')
if index == "q":
print('\n')
return
try:
index = int(index) - 1
except ValueError:
shell.print_red(
f'\nWrong index: number in range 1 - {len(temp_files)} expected\n'
)
continue
if 0 <= index <= len(temp_files) - 1:
file_name = LocalData.get_shared_file_name(
temp_files[index])
file_md5 = LocalData.get_shared_file_md5(temp_files[index])
# controlla se il file è già in condivisione
if not LocalData.is_shared_file(temp_files[index]):
# crea il pacchetto e lo invia al super peer
packet = choice + p_ssid + file_md5 + file_name.ljust(
100)
try:
net_utils.send_packet_and_close(
sp_ip4, sp_ip6, sp_port, packet)
shell.print_green(
f'Packet sent to supernode: {sp_ip4}|{sp_ip6} [{sp_port}]'
)
except net_utils.socket.error:
shell.print_red(
f'\nError while sending packet to supernode: {sp_ip4}|{sp_ip6} [{sp_port}]\n'
)
# aggiunge il file alla struttura dati ed esce
LocalData.add_shared_file(file_md5, file_name)
shell.print_blue(
f'\nNew shared file added {file_name} | {file_md5}'
)
break
else:
# il file è già in condivisione
shell.print_yellow(
f'\n{file_name} | {file_md5} already in sharing.\n'
)
break
else:
shell.print_red(
f'\nWrong index: number in range 1 - {len(temp_files)} expected\n'
)
continue
elif choice == "DEFF":
# recupera i files in sharing
files = LocalData.get_shared_files()
# check se ci sono file in sharing
if not files:
shell.print_yellow(
'No files currently in sharing. Add files choosing the command from the menu.\n'
)
return
# scelta del file da rimuovere
shell.print_blue('\nFile currently in sharing:')
for count, file in enumerate(files, 1):
print(
f'{count}] {LocalData.get_shared_file_name(file)} | {LocalData.get_shared_file_md5(file)}'
)
while True:
index = input(
'Choose a file to remove from sharing (q to cancel): ')
if index == "q":
print('\n')
return
try:
index = int(index) - 1
except ValueError:
shell.print_red(
f'\nWrong index: number in range 1 - {len(files)} expected\n'
)
continue
if 0 <= index <= len(files) - 1:
# recupera il file dalla DS
file = LocalData.get_shared_file(index)
# crea ed invia il pacchetto al supernode
packet = choice + p_ssid + LocalData.get_shared_file_md5(
file)
try:
net_utils.send_packet_and_close(
sp_ip4, sp_ip6, sp_port, packet)
except net_utils.socket.error:
shell.print_red(
f'\nError while sending packet to supernode: {sp_ip4}|{sp_ip6} [{sp_port}]\n'
)
# rimuove il file dalla DS
LocalData.remove_shared_file(file)
shell.print_green(
f'\n{LocalData.get_shared_file_name(file)} | {LocalData.get_shared_file_md5(file)} removed successfully.'
)
break
else:
shell.print_red(
f'\nWrong index: number in range 1 - {len(files)} expected\n'
)
continue
elif choice == "FIND":
while True:
search = input('\nInsert file\'s name (q to cancel): ')
if search == "q":
print('\n')
return
if not 0 < len(search) <= 20:
shell.print_red(
'\nQuery string must be a valid value (1 - 20 chars).')
continue
break
packet = choice + p_ssid + search.ljust(20)
try:
socket = net_utils.send_packet(sp_ip4, sp_ip6, sp_port, packet)
except (net_utils.socket.error, AttributeError):
shell.print_red(
f'\nError while sending the request to the superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
if socket is not None:
socket.close()
return
spinner = SpinnerThread('Searching files', 'Research done!')
spinner.start()
try:
socket.settimeout(25)
command = socket.recv(4).decode()
spinner.stop()
spinner.join()
print('\033[1A', end='\r')
if command != "AFIN":
shell.print_red(
f'\nReceived a packet with a wrong command ({command}).\n'
)
socket.close()
return
except net_utils.socket.error:
shell.print_red(
f'\nError while receiving the response from the superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
return
except ValueError:
shell.print_red(
f'\nInvalid packet from superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
return
# ('file_name', 'file_md5', owners[])
downloadables = list()
try:
num_downloadables = int(socket.recv(3))
# check se ci sono file da scaricare
if num_downloadables == 0:
shell.print_yellow(f'{search} not found.\n')
socket.close()
return
except net_utils.socket.error:
shell.print_red(
f'\nError while receiving the response from the superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
return
for i in range(num_downloadables):
# ('owner_ip4', 'owner_ip6', 'owner_port')
owners = list()
try:
file_md5 = socket.recv(32).decode()
file_name = socket.recv(100).decode().lstrip().rstrip()
num_copies = int(socket.recv(3))
except net_utils.socket.error:
shell.print_red(
f'\nError while receiving the response from the superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
continue
except ValueError:
shell.print_red(
f'\nInvalid packet from superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
continue
for j in range(num_copies):
try:
(owner_ip4, owner_ip6) = net_utils.get_ip_pair(
socket.recv(55).decode())
owner_port = int(socket.recv(5))
owners.append((owner_ip4, owner_ip6, owner_port))
except net_utils.socket.error:
shell.print_red(
f'\nError while receiving the response from the superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
continue
except ValueError:
shell.print_red(
f'\nInvalid packet from superpeer: {sp_ip4}|{sp_ip6} [{sp_port}].\n'
)
continue
if owners:
# prepara una lista con tutti i file che possono essere scaricati
downloadables.append((file_name, file_md5, owners))
if not downloadables:
shell.print_red(
f'\nSomething went wrong while retrieving {search}\n')
socket.close()
return
shell.print_green(f'\nFiles found:')
for count, downloadable in enumerate(downloadables, 1):
print(f'{count}] {downloadable[0]} | {downloadable[1]}')
while True:
choosed_file_index = input(
'Choose a file to download (q to cancel): ')
if choosed_file_index == "q":
print('\n')
socket.close()
return
try:
choosed_file_index = int(choosed_file_index) - 1
except ValueError:
shell.print_red(
f'\nWrong index: number in range 1 - {len(downloadables)} expected\n'
)
continue
if not 0 <= choosed_file_index <= len(downloadables) - 1:
shell.print_red(
f'\nWrong index: number in range 1 - {len(downloadables)} expected\n'
)
continue
else:
choosed_file_name = downloadables[choosed_file_index][0]
choosed_file_md5 = downloadables[choosed_file_index][1]
choosed_file_owners = downloadables[choosed_file_index][2]
shell.print_green(
f'\nAvailable sources for {choosed_file_name}:')
for count, choosed_file_owner in enumerate(
choosed_file_owners, 1):
print(
f'{count}] {choosed_file_owner[0]}|{choosed_file_owner[1]} [{choosed_file_owner[2]}]'
)
while True:
choosed_owner_index = input(
f'Choose a source for {choosed_file_name} (q to cancel): '
)
if choosed_owner_index == "q":
print('\n')
return
try:
choosed_owner_index = int(choosed_owner_index) - 1
except ValueError:
shell.print_red(
f'\nWrong index: number in range 1 - {len(choosed_file_owners)} expected\n'
)
continue
if 0 <= choosed_owner_index <= len(
choosed_file_owners) - 1:
packet = "RETR" + choosed_file_md5
try:
Downloader(
choosed_file_owners[choosed_owner_index]
[0],
choosed_file_owners[choosed_owner_index]
[1],
choosed_file_owners[choosed_owner_index]
[2], packet, choosed_file_name).start()
shell.print_green(
f'\n{choosed_file_name} downloaded successfully.\n'
)
except OSError:
shell.print_red(
f'\nError while downloading {choosed_file_name}.\n'
)
break
else:
shell.print_red(
f'\nWrong index: number in range 1 - {len(choosed_file_owners)} expected\n'
)
continue
break
elif choice == "SHOWSUPER":
print(
f'\nYour current superpeer is: {sp_ip4}|{sp_ip6} [{sp_port}]\n'
)
elif choice == "LOGO":
packet = choice + p_ssid
sock = net_utils.send_packet(sp_ip4, sp_ip6, sp_port, packet)
try:
command = sock.recv(4).decode()
if command != "ALGO":
shell.print_red(
f'\nWrong response code "{command}": "{choice}" expected.\n'
)
num_files = int(sock.recv(3))
shell.print_blue(
f'\n{num_files} files has been removed from sharing.\n')
except (net_utils.socket.error, AttributeError):
shell.print_red(
f'\nError while receiving the response for "{choice}".\n')
LocalData.clear_backup_data()
else:
pass