使用套接字发送多个图像

时间:2019-11-29 22:24:09

标签: python-3.x sockets tcp python-sockets

在用Python制作VNC的雄心勃勃的尝试中,我试图将一个用户(服务器)的屏幕截图连续发送到另一个(客户端)。经过数小时的尝试并达到了鲍尔默的顶峰,我设法做到了。但是,现在我的问题是发送多个图像,连续不断地发送它们。我首先尝试将所有二进制数据写入一个文件,但没有用。当打开第二张图像时,它崩溃了。我认为这可能是因为二进制数据某种程度上已损坏,所以我尝试为每个图像制作一个新文件,但是我遇到了同样的问题。我知道Tcp是恒定的数据流,因此很难知道第一个图像的结束和下一个图像的开始,但是通过创建另一个文件,我认为我会很好。

在解决此问题和/或提高其效率方面的任何帮助都将不胜感激:)

服务器端:

import socket
from PIL import Image, ImageGrab

PORT = 10007
HOST = '127.0.0.1'

with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.bind((HOST, PORT))
    s.listen()
    conn, addr = s.accept()
    with conn:

        counter = 3

        while counter > 0:
            image = ImageGrab.grab(bbox=None)
            image.save('test.png')

            f = open('test.png', 'rb')

            l = f.read(1024)

            while (l):
                conn.send(l)
                l = f.read(1024)

            f.close()
            print('Done sending curr image')
            counter -= 1

conn.close()

客户端:

import socket
from PIL import Image

HOST = '127.0.0.1'
PORT = 10007


with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.connect((HOST, PORT))


    counter = 1

    while counter != 3:
        fname = 'image' + str(counter) + '.png'

        with open(fname, 'wb') as file:

            print('file opened')

            while True:
                data = s.recv(1024)
                if not data:
                    break
                else:
                    file.write(data)

        file.close()
        currImg = Image.open(fname)
        currImg.show()
        counter += 1






s.close()

1 个答案:

答案 0 :(得分:0)

您的接收者不知道一个文件何时结束而下一个文件何时开始。最简单的解决方法是在发送每个文件之前将文件的长度(可能是4字节无符号值)发送给接收方。然后,接收方可以读取长度,读取文件,读取长度,读取文件......

为提高效率,您可以在两端停止将文件数据保存到实际文件中,而是将其保存到内存缓冲区中(显然,也可以从中读取)。有关如何执行此操作的说明,请参见this answer。在Python 3中,看起来您将使用BytesIO模块。

另一个改进是仅发送自上次发送以来已更改的屏幕部分的图像数据。为此,您需要弄清楚如何将当前捕获与上一个捕获进行比较。对于第一遍,您可以使用PIL.ImageChops.difference,然后使用PIL.Image.getbbox,然后仅编码并发送当前捕获的该区域。为此,发送者不仅要告诉接收者PNG的大小,还要告诉输出屏幕图像中应该绘制新图像补丁的位置。因此,除了尺寸和编码图像数据之外,您还需要发送一个位置。