通过Python tcp套接字传输以阿拉伯语标题的文件时出现ValueError

时间:2019-03-13 07:08:21

标签: python sockets arabic valueerror non-english

我正在构建一个套接字程序来在两台计算机之间传输文件。具有英语标题的文件已成功传输,但是当我尝试发送阿拉伯文标题文件(例如وثيقة。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()

客户:

导入

 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)

这是什么问题?请帮忙。

1 个答案:

答案 0 :(得分:0)

您以标题的unicode字符发送大小,然后尝试以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) 
        ...