Макаров М.И. ПИ20-5

ftp-client.py

@@ -1,17 +1,83 @@
 import socket
+import re
+import os
 
 HOST = 'localhost'
-PORT = 6666
+PORT = 8080
+END_FLAG = b"$$STREAM_FILE_END_FLAG$$"
+FAIL_FLAG = b'$FAILED$'
+login = input("Введите логин: ")
+password = input("Введите пароль: ")
+# login = "admin"
+# password = "admin"
+current_directory = "\\"
 
-while True:
-    request = input('>')
-
-    sock = socket.socket()
-    sock.connect((HOST, PORT))
+def creator(message, size=0):
+    global login, password, current_directory
+    return f"{login}=login{password}=password{current_directory}=cur_dir{size}=file_size{message}".encode()
+
+
+def sending(request):
+    global sock, FAIL_FLAG, END_FLAG
 
-    sock.send(request.encode())
+    flag_finder = sock.recv(1024)
+    if FAIL_FLAG in flag_finder:
+        print((flag_finder.replace(FAIL_FLAG, b"")).decode())
+    else:
+        filename = re.split("[ \\/]+", request)[-1]
+        with open (filename, "wb") as bytefile:
+            while True:
+                if END_FLAG in flag_finder:
+                    flag_finder, end_flag_msg = flag_finder.split(END_FLAG)
+                    bytefile.write(flag_finder.replace(END_FLAG, b""))
+                    break
+                else:
+                    bytefile.write(flag_finder)
+                    flag_finder = sock.recv(1024)
+    print(sock.recv(1024).decode())
 
-    response = sock.recv(1024).decode()
-    print(response)
+
+def receiving(request):
+    global sock, END_FLAG
+    filename = re.split("[ \\/]+", request)[-1]
+    if os.path.exists(filename):
+        size = os.path.getsize(filename)
+        sock.send(creator(request, size))
+        enought_flag = sock.recv(1024).decode()
+        if enought_flag != '$ENOUGHT$':
+            print(enought_flag)
+            return False
+        with open(filename, "rb") as bytefile:
 
-    sock.close()
+            while read_bytes := bytefile.read(1024):
+                sock.send(read_bytes)
+        sock.send(END_FLAG)
+        print(sock.recv(1024).decode())
+        return True
+    else:
+        print("Нет такого файла")
+        sock.send(creator("error"))
+        return False
+
+while True:
+    request = input(current_directory+'>')
+    if request == "exit":
+        print("goodbye")
+        break
+    arrRequest = request.split(" ")
+    sock = socket.socket()
+    sock.connect((HOST, PORT))
+    if arrRequest[0] == "get_file":
+        if not receiving(request):
+            continue
+    else:
+        sock.send(creator(request))
+        if arrRequest[0] == "send_file":
+            sending(request)
+        else:
+            response = sock.recv(1024).decode()
+            if arrRequest[0] == "cd":
+                current_directory = response
+            else:
+                print(response)
+    sock.close()

ftp-server.py

@@ -1,35 +1,229 @@
 import socket
 import os
+import shutil
+import csv
+import re
+
 '''
 pwd - показывает название рабочей директории
 ls - показывает содержимое текущей директории
+cd <path> - перемещение по директориям
+mkdir <dirname> - создать директорию (рекурсивно)
+rmtree <dirname> - удалить директорию (рекурсивно)
+touch <filename> - создать пустой файл
+remove <filename> - удалить файл
 cat <filename> - отправляет содержимое файла
+get_file <filename> - отправляет файл в текущую директорию у клиента
+send_file <filename> - принимает файл с текущей директории клиента
 '''
 
-dirname = os.path.join(os.getcwd(), 'docs')
+END_FLAG = b"$$STREAM_FILE_END_FLAG$$"
+FAIL_FLAG = b'$FAILED$'
+PORT = 8080
+global_root = os.path.join(os.getcwd())
+usersfile = os.path.join(global_root, "users.csv")
+log_file = os.path.join(global_root, "log.txt")
+
+def get_size(start_path):
+    total_size = 0
+    for dirpath, dirnames, filenames in os.walk(start_path):
+        for f in filenames:
+            fp = os.path.join(dirpath, f)
+            if not os.path.islink(fp):
+                total_size += os.path.getsize(fp)
+    return total_size
+
+def log_print(*strings):
+    print(*strings)
+    with open(log_file, "a") as logfile:
+        logfile.write(" ".join([str(item) for item in strings]) + "\n")
+
+
+def authorization(message):
+    global usersfile, global_root
+    login, message = message.split("=login", 1)
+    password, message = message.split("=password", 1)
+    current_directory, message = message.split("=cur_dir", 1)
+    size, message = message.split("=file_size", 1)
+    if login == password == "admin":
+         user_root = global_root
+    else:
+        user_root =  os.path.join(global_root, login)
+        with open(usersfile, "a+", newline = "") as csvfile:
+            csvfile.seek(0,0)
+            reader = csv.reader(csvfile, delimiter = ";")
+            for line in reader:
+                if line[0] == login:
+                    if line[1] == password:
+                        break
+                    else:
+                        return None
+                else:
+                    writer = csv.writer(csvfile, delimiter = ";")
+                    writer.writerow([login, password])
+        try: 
+            os.makedirs(user_root)
+        except FileExistsError:
+            pass
+
+    return user_root, current_directory, message, size
+
 
 def process(req):
-    if req == 'pwd':
-        return dirname
-    elif req == 'ls':
-        return '; '.join(os.listdir(dirname))
-    return 'bad request'
+    req = authorization(req)
+    if req:
+        user_root, current_directory, req, size = req
+        req, *dir_ = req.split()
+        path = [path_decoder(user_root, current_directory, item) for item in dir_]
+        print(req, path)
+        if not path:
+            path = [""]
+        if req == 'pwd':
+            return pwd(current_directory)
+        elif req == 'ls':
+            return ls(os.path.join(user_root, current_directory[1:]))
+        elif req == "cd":
+            return cd(path[0], current_directory, user_root)
+        elif req == 'mkdir':
+            return mkdir(path[0])
+        elif req == 'rmdir':
+            return rmdir(path[0])
+        elif req == 'touch':
+            return touch(path[0])
+        elif req == 'remove':
+            return remove(path[0])
+        elif req == 'cat':
+            return cat(path[0])
+        elif req == 'rename':
+            return rename(*path[:2])
+        elif req == "get_file":
+            return get_file(path[0], user_root, size)
+        elif req == "send_file":
+            return send_file(path[0])
+        else:
+            return 'bad request'
+    else:
+        return "Incorrect password"
+
+def path_decoder(root, current, dir_):
+    if current == "\\" and dir_[:2] == "..":
+
+        return root + dir_[2:]
+    elif dir_[0] in ["\\", "/"]:
+        dir_ = re.sub(r"^[\\/]+", "", dir_)
+        log_print(dir_)
+        return os.path.join(root, dir_)
+    else:
+        return os.path.join(root, current[1:], dir_)
+
+def try_decorator(path_func):
+    def wrapper(*path):
+        try:
+            returned = path_func(*path)
+            if returned == None:
+                return "successfully"
+            else:
+                return returned
+        except FileNotFoundError:
+            return (f'Invalid path')
+        except FileExistsError:
+            return (f'Has already exist') 
+        except PermissionError:
+            return f"Permission denied"
+    return wrapper
+
+def pwd(dirname):
+    return os.path.join(dirname)
 
+def ls(path):
+    return '\n\r'.join(os.listdir(path))
+
+def cd(path, current, root):
+    try:
+        os.chdir(path)
+    except:
+        return current
+    return os.getcwd().replace(root,"")+"\\"
+
+
+@try_decorator
+def mkdir(path):
+    os.makedirs(path)
+
+@try_decorator
+def rmdir(path):
+    shutil.rmtree(path)
+
+@try_decorator
+def remove(path):
+    os.remove(path)
+
+@try_decorator
+def touch(path):
+    with open(path, 'x'):
+        pass
+
+@try_decorator
+def cat(path):
+    with open(path, "r") as file:
+        return "\n\r".join(file.readlines())
+
+@try_decorator
+def rename(path1, path2):
+    os.rename(path1, path2)
+
+def send_file(path):
+    global conn, END_FLAG, FAIL_FLAG
+    try:
+        with open(path, "rb") as bytefile:
+            while read_bytes := bytefile.read(1024):
+                conn.send(read_bytes)
+    except FileNotFoundError:
+        returned = b'Invalid path'+FAIL_FLAG
+    except PermissionError:
+        returned = b"Permission denied"+FAIL_FLAG
+    else:
+        conn.send(END_FLAG)
+        returned = "uploaded successfully"
+        log_print("Has been get from user")
+    return returned
+
+def get_file(path, root, size):
+    global conn, END_FLAG, FAIL_FLAG
+    available = pow(2,20)*10 - get_size(root) #10Mb for each user
+    if available < int(size):
+        return "Not enough disk space!"
+    else:
+        conn.send(b"$ENOUGHT$")
+    flag_finder = conn.recv(1024)
+    with open (path, "wb") as bytefile:
+        while True:
+            if END_FLAG in flag_finder:
+                bytefile.write(flag_finder.replace(END_FLAG, b""))
+                break
+            else:
+                bytefile.write(flag_finder)
+                flag_finder = conn.recv(1024)
+    log_print("Has been sent for user")
+    return "get is successfull"
 
-PORT = 6666
 
 sock = socket.socket()
 sock.bind(('', PORT))
 sock.listen()
-print("Прослушиваем порт", PORT)
+log_print("Прослушиваем порт", PORT)
 
 while True:
     conn, addr = sock.accept()
 
     request = conn.recv(1024).decode()
-    print(request)
+    log_print("request:", request)
 
     response = process(request)
-    conn.send(response.encode())
-
-conn.close()
+    if not response:
+        response = "\00"
+    try:
+        conn.send(response.encode())
+    except AttributeError:
+        conn.send(response)
+    conn.close()