如何在异步套接字中接收完整的屏幕截图?

时间:2019-06-24 18:23:08

标签: c# asyncsocket beginreceive

我有一个Java android代码,该数据将数据(图像或文本)发送到C#应用程序,以接收这些数据(我正在使用异步套接字)。但是存在一个与 BeginReceive() 函数相关的问题,当发送图像时,该函数没有接收完整的数据。那么,我如何制作一种“循环”来接收完整的数据并在 Picturebox 上显示图像后(例如)?

表格

private Listener listener;
private Thread startListen;

private Bitmap _buffer;

public frmMain()
{
  InitializeComponent();
}

private void serverReceivedImage(Client client, byte[] image)
{
    try
    {
        byte[] newImage = new byte[image.Length - 6];

        Array.Copy(image, 6, newImage, 0, newImage.Length);

            using (var stream = new MemoryStream(newImage))
            {
                using (var msInner = new MemoryStream())
                {

                  stream.Seek(2, SeekOrigin.Begin);

                  using (DeflateStream z = new DeflateStream(stream, CompressionMode.Decompress))
                  {
                    z.CopyTo(msInner);
                  }

                  msInner.Seek(0, SeekOrigin.Begin);

                  var bitmap = new Bitmap(msInner);
                  Invoke(new frmMain.ImageCompleteDelegate(ImageComplete), new object[] { bitmap });
                }
            }
    }
     catch (Exception)
     {
        System.Diagnostics.Process.GetCurrentProcess().Kill();
     }
}

private delegate void ImageCompleteDelegate(Bitmap bitmap);
private void ImageComplete(Bitmap bitmap)
{
   if (_buffer != null)
       _buffer.Dispose();

       _buffer = new Bitmap(bitmap);
       pictureBox1.Size = _buffer.Size;
       pictureBox1.Invalidate();
}

private void pictureBox1_Paint(object sender, PaintEventArgs e)
{
  if (_buffer == null) return;
      e.Graphics.DrawImage(_buffer, 0, 0);
}

private void startToolStripMenuItem_Click(object sender, EventArgs e)
{
  startListen = new Thread(listen);
  startListen.Start();
}

private void listen()
{
  listener = new Listener();
  listener.BeginListen(101);
  listener.receivedImage += new Listener.ReceivedImageEventHandler(serverReceivedImage);

  startToolStripMenuItem.Enabled = false;
}

监听器

class Listener
{

    private Socket s;
    public List<Client> clients;

    public delegate void ReceivedImageEventHandler(Client client, byte[] image);
    public event ReceivedImageEventHandler receivedImage;

    private bool listening = false;

    public Listener()
    {
        clients = new List<Client>();
        s = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
    }

    public bool Running
    {
        get { return listening; }
    }

    public void BeginListen(int port)
    {
        s.Bind(new IPEndPoint(IPAddress.Any, port));
        s.Listen(100);
        s.BeginAccept(new AsyncCallback(AcceptCallback), s);
        listening = true;
    }

    public void StopListen()
    {
        if (listening == true)
        {
            s.Close();
            listening = false;
        }
    }

    void AcceptCallback(IAsyncResult ar)
    {
        Socket handler = (Socket)ar.AsyncState;
        Socket sock = handler.EndAccept(ar);
        Client client = new Client(sock);
        clients.Add(client);

        sock.BeginReceive(client.buffer, 0, client.buffer.Length, SocketFlags.None, new AsyncCallback(ReadCallback), client);

        client.Send("REQUEST_PRINT" + Environment.NewLine); 

        handler.BeginAccept(new AsyncCallback(AcceptCallback), handler);
    }

    void ReadCallback(IAsyncResult ar)
    {

        Client client = (Client)ar.AsyncState;
        try
        {
            int rec = client.sock.EndReceive(ar);
            if (rec != 0)
            {
                string data = Encoding.UTF8.GetString(client.buffer, 0, rec);

                if (data.Contains("SCREEN"))
                { 
                    byte[] bytes = Encoding.UTF8.GetBytes(data);
                    receivedImage(client, bytes);
                }
                else // not is a image, is a text
                {
                    // prepare text to show in TextBox
                }
            }
            else
            {
                Disconnected(client);
                return;
            }

            client.sock.BeginReceive(client.buffer, 0, client.buffer.Length, SocketFlags.None, new AsyncCallback(ReadCallback), client);
        }
        catch
        {
            Disconnected(client);
            client.sock.Close();
            clients.Remove(client);
        }
    }

}

客户

class Client
{
    public Socket sock;
    public byte[] buffer = new byte[8192];

    public Client(Socket sock)
    {
        this.sock = sock;
    }

    public void Send(string data)
    {
        byte[] buffer = Encoding.ASCII.GetBytes(data);
        sock.BeginSend(buffer, 0, buffer.Length, SocketFlags.None, new AsyncCallback((ar) =>
        {
            sock.EndSend(ar);
        }), buffer);
    }
}

Android代码

private byte[] compress(byte[] data) {

    Deflater deflater = new Deflater();
    deflater.setInput(data);
    ByteArrayOutputStream outputStream = new ByteArrayOutputStream(data.length);
    deflater.finish();
    byte[] buffer = new byte[1024];
    while (!deflater.finished()) {
        int count = deflater.deflate(buffer);
        outputStream.write(buffer, 0, count);
    }
    outputStream.close();
    byte[] output = outputStream.toByteArray();

    return output;
}

public static DataOutputStream dos;
public static byte[] array;

ByteArrayOutputStream bos = new ByteArrayOutputStream();
bitmap.compress(Bitmap.CompressFormat.PNG, 100, bos);
array = compress(bos.toByteArray());

//...

dos = new DataOutputStream(SocketBackgroundService.clientSocket.getOutputStream());
byte[] header = ("SCREEN").getBytes(StandardCharsets.UTF_8);
byte[] dataToSend = new byte[header.length + array.length];
System.arraycopy(header, 0, dataToSend, 0, header.length);
System.arraycopy(array, 0, dataToSend, header.length, array.length);
dos.writeInt(dataToSend.length);
dos.write(dataToSend, 0, dataToSend.length);

dos.flush();

EDITION

我总是在此行中得到错误无效参数

var bitmap = new Bitmap(msInner);

在这里使用压缩也一样

z.CopyTo(msInner);

IvalidDataException

分别在 ServerReceivedImage()方法上

使用此

File.WriteAllBytes(Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyPictures), "image.png"), newImage);

我注意到它仅接收15KB(不使用压缩的文件大小)。

1 个答案:

答案 0 :(得分:1)

我正在写评论,但是它没有足够的空间来表达您对代码的不满。

我的主要观点是

  • 您尝试重新压缩并完美压缩图像。 PNG是便携式网络图形。它是专为网络传输而设计的。如果可以接受,则应使用类似jpeg的名称。

  • 您只需使用UTF8.GetString解码接收到的缓冲区并搜索文本,然后重新编码该字符串并尝试从索引6开始解压缩并从中读取图像,考虑到您添加的索引,这是毫无意义的流的开头有两个字节大小的字段,您真的不知道“ SCREEN”的位置。

  • 您不会检查是否已收到所有流数据。

  • 所有代码看起来像您已经搜查过SO问题和答案并创建了复制面食。

现在我的建议。

  • 从网络传输数据时,请勿尝试发明轮子。尝试使用类似gRPC的东西,它同时具有android java和c#软件包。

  • 如果您要使用原始数据,请知道您的字节数。

  • 我假设您将通过添加新的命令对来扩展代码。由于您没有某种信号系统的魔力标记,因此很难将数据与标头区分开。对于简单的实现,将某种魔术数据添加到标头中并搜索该数据,然后读取标头,然后读取数据。您可能需要一次又一次地从套接字读取数据,直到收到所有数据为止。

    424A72 0600 53435245454E 008E0005   .....  724A42
     B J r    6  S C R E E N    36352   .....  rJB
    

此样本数据通过查看“ BJr”表明我们具有有效的流。然后读取2字节无符号整数以读取命令大小(对于SCREEN)为6。读取命令,然后读取四个字节的无符号长度的命令数据。对于我们的示例,它是36352。为安全起见,我添加了命令标记“ rJB”的结尾。

要获得奖励积分,请尝试减少内存分配/副本,您可以查看System.Span<T>