Ошибка ValueError при передаче файла с названием на арабском языке через TCP-сокет Python - PullRequest
Новая
       73

Ошибка ValueError при передаче файла с названием на арабском языке через TCP-сокет Python

0 голосов
/ 13 марта 2019

Я создаю программу-сокет для передачи файлов между двумя компьютерами.Файлы с английскими заголовками успешно передаются, но когда я пытаюсь отправить файлы с арабским названием (например, وثيقة. Docx), я получаю длинный список ValueErrors, начиная с: 

invalid literal for int() with base 2: b'.docx000000000000000000010000001'

inva lid literal for int() with base 2: b'10001PK\x03\x04\x14\x00\x08\x08\x08\x00\xe0'

Мой код: Сервер: 

import socket

serversock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = 'localhost'
port = 9000
serversock.bind((host,port))
filename = ""
serversock.listen(10)
print("Waiting for a connection.....")

clientsocket, addr = serversock.accept()
print("Got a connection from %s" % str(addr))

while True:
    try:
        size = clientsocket.recv(16) # Note that you limit your filename length to 255 bytes.
        if not size:
            clientsocket, addr = serversock.accept()
            print("Got a connection from %s" % str(addr))
            continue
        size = int(size, 2)
        print('SIZE', size)
        filename = clientsocket.recv(size)
        print('filename', filename)
        filesize = clientsocket.recv(32)
        print('FILESIZE', filesize, 'TYPE', type(filesize))
        filesize = int(filesize, 2) ##########
        file_to_write = open(filename, 'wb')
        chunksize = 4096
        while filesize > 0:
            if filesize < chunksize:
                chunksize = filesize
            data = clientsocket.recv(chunksize)
            file_to_write.write(data)
            filesize -= len(data)

        file_to_write.close()
        print('File (%s) received successfully' % filename.decode('utf-8'))
    except ValueError as verr:
        print(verr)
        #continue
    except FileNotFoundError:
        print('FileNotFoundError')
        #continue
serversock.close()

Клиент:

import 

 socket
import os
from file_walker import files_to_transmit

def transmit(host, port):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        s.connect((host, port))
        directory = files_to_transmit()
        for file in directory:
            filename = file
            size = len(filename.split('/')[-1]) # split() to get bare file name to transmit to server
            size = bin(size)[2:].zfill(16) # encode filename size as 16 bit binary
            s.send(size.encode('utf-8'))
            s.send(filename.split('/')[-1].encode('utf-8')) # split() to get bare file name

            filename = file
            filesize = os.path.getsize(filename)
            filesize = bin(filesize)[2:].zfill(32) # encode filesize as 32 bit binary
            s.send(filesize.encode('utf-8'))

            file_to_send = open(filename, 'rb')

            l = file_to_send.read()
            s.sendall(l)
            file_to_send.close()
            print('File Sent')

        s.close()
    except ConnectionRefusedError:
        print('ConnectionRefusedError: Server may not be running')

    except ValueError as e:
        print(e)

transmit('localhost', 9000)

В чем здесь проблема?Пожалуйста, помогите.

Ответы [ 2 ]

0 голосов
/ 13 марта 2019

Вы отправляете размер в юникодном символе заголовка, а затем пытаетесь отправить его в кодировке utf-8.Например, в вашем примере: 

title = 'وثيقة.docx'
print(len(title), len(title.encode('utf8')))

дает 

10 15

Пир будет использовать только 10 байтов в качестве имени файла и будет использовать оставшиеся 5 в качестве начала размера файла,И будет подавлен для .docx, не являющегося началом двоичного числа.Что происходит ...

Исправить несложно, создайте строку байтов до вычисления ее длины: 

    ...
    for file in directory:
        filename = file.split('/')[-1].encode('utf_8') # split() to get bare file name
        size = len(filename)

        size = bin(size)[2:].zfill(16) # encode filename size as 16 bit binary
        s.send(size.encode('utf-8'))
        s.send(filename) 
        ...
0 голосов
/ 13 марта 2019

Проблема Simalar обнаружена в моем опыте. Я предлагаю использовать алгоритм для кодирования и декодирования файлов, таких как base64.

Где-то в вашей программе вы читаете файл, чтобы сохранить его в памяти и отправить вторым шагом. Ну вместо того, чтобы просто читать кодировать также 

import base64

with open("yourfile.ext", "rb") as image_file:
    encoded_string = base64.b64encode(image_file.read())

Очевидно, что на другой стороне вам нужно декодировать и записать на диск. Вы также можете кодировать и декодировать заголовок файла, чтобы избежать ValueError

...