Как получить полный скриншот в сокете Async? - PullRequest
6 голосов
/ 24 июня 2019

У меня есть код Java android, который отправляет данные (изображение или текст) в приложение C#, для получения этих данных я использую Async-сокет. Но существует проблема, связанная с тем, что функция 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

on ServerReceivedImage () метод соответственно.

используя это

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

Я заметил, что получает только 15 КБ (размер файла без использования сжатия).

1 Ответ

1 голос
/ 02 июля 2019

Я писал комментарий, но он не дает мне достаточно места, чтобы выразить свое разочарование по поводу вашего кода.

Мои основные моменты

  • Вы пытаетесь повторно сжатьи идеально сжатое изображение.PNG - это портативная сетевая графика.Он был разработан для сетевых передач.Если это приемлемо, вы должны использовать что-то вроде jpeg.

  • Вы просто декодируете полученный буфер, используя UTF8.GetString, и ищите текст, затем перекодируете эту строку и пытаетесь распаковать и прочитатьизображение из него, начиная с индекса 6, что довольно бессмысленно, учитывая, что вы добавили в начало потока двухбайтовое поле размера, и вы действительно не знаете положение «ЭКРАНА».

  • Вы не проверяете, получили ли вы ВСЕ данные потока.

  • Весь код выглядит так, как будто вы отыграли SO вопросы и ответы и создали копию пасты.

Теперь мои рекомендации.

  • При передаче данных из сети не пытайтесь изобретать колеса.Попробуйте что-то вроде gRPC с пакетами android java и c #.

  • Если вы будете использовать необработанные данные, пожалуйста, знайте свои байты.

  • Я предполагаю, что вы расширите свой код, добавив новые пары команд.Поскольку у вас нет волшебных маркеров какой-либо системы сигналов, вам будет очень трудно отличить данные от заголовка.Для простой реализации добавьте в заголовок какие-то магические данные и найдите их, затем прочитайте заголовок и затем прочитайте данные.Возможно, вам придется читать из сокета снова и снова, пока вы не получите все данные.

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

Этот пример данных показывает, что у нас есть действительный поток, посмотрев на "BJr".Затем прочитайте 2-байтовое целое число без знака, чтобы прочитать размер команды, который равен 6 для ЭКРАНА.Прочитайте команду, а затем прочитайте четыре байта без знака длины для данных команды.Для нашего примера это 36352. На всякий случай я добавил маркер конца команды «rJB».

Чтобы получить бонусный балл, попробуйте уменьшить количество выделенных памяти / копий, посмотрите на System.Span<T>

...