cdez/CS460-Projects
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Try the HW5 code out?

Browse source
This commit is contained in:
Charles Descamps 2025-04-21 18:59:31 -07:00
parent cbcb35a158
commit 3206565d55
2 changed files with 136 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

46
chatserver/client.py
View file

@ -1,3 +1,4 @@
import os
import socket
import sys
import select
@ -15,13 +16,25 @@ def main():
SERVER_HOST = sys.argv[1]
SERVER_PORT = int(sys.argv[2])
# Prompt user for a local directory path for file operations
while True:
local_path = input("Enter a local directory path for sending/receiving files: ").strip()
if os.path.isdir(local_path):
print(f"Using directory: {local_path}")
break
else:
print("Invalid directory. Please enter a valid path.")
# create a socket object
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
# connect to the server
client_socket.connect((SERVER_HOST, SERVER_PORT))
print(f"Connected to chat server at {SERVER_HOST}:{SERVER_PORT}")
# Send the local path to the server
client_socket.send(f"SET_PATH {local_path}".encode('utf-8'))
while True:
# use select to wait for either server messages or user input
read_sockets, _, _ = select.select([client_socket, sys.stdin], [], [])
@ -33,8 +46,35 @@ def main():
if not server_message:
print("Disconnected from server")
return
print(f"\n{server_message}")
print("> ", end="", flush=True)
if server_message.startswith("FILE_REQUEST"):
# Handle file request
_, sending_username, filename = server_message.split()
file_path = os.path.join(local_path, filename)
if os.path.isfile(file_path):
file_size = os.path.getsize(file_path)
with open(file_path, "rb") as f:
file_bytes = f.read()
filetype = "File"
client_socket.send(
f"@{sending_username}@CLIENT -f {filename} {filetype} {file_size} {file_bytes.decode('latin1')} ENDFILE".encode('utf-8')
)
else:
client_socket.send(
f"@{sending_username}@CLIENT -f {filename} NOTFOUND".encode('utf-8')
)
elif server_message.startswith("FILE_TRANSFER"):
# Handle file transfer
_, sending_username, filename, filetype, filelength = server_message.split()
file_bytes = client_socket.recv(int(filelength))
end_marker = client_socket.recv(BUFFER_SIZE).decode('utf-8')
if end_marker == "ENDFILE":
file_path = os.path.join(local_path, filename)
with open(file_path, "wb") as f:
f.write(file_bytes)
print(f"Received file '{filename}' from {sending_username}.")
else:
print(f"\n{server_message}")
print("> ", end="", flush=True)
else:
# prompt for user input
client_message = input()