java - 使用java套接字发送时图像不完整

Question

我想将图像发送到套接字服务器，但它没有完全到达

首先，我创建一个屏幕截图，应使用以下代码通过套接字将其发送到服务器:

try {
    Robot robot = new Robot();
    Rectangle rectangle = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
    File file = new File("original.png");
    BufferedImage bufferedImage = robot.createScreenCapture(rectangle);
    try {
        ImageIO.write(bufferedImage, "png", file);
    } catch (IOException e) {
        e.printStackTrace();
    }
    socket(robot.createScreenCapture(rectangle));
} catch (AWTException ex) {
    System.err.println(ex);
}

void socket(BufferedImage bufferedImage) {
    try {
        Socket socket = new Socket("localhost", 20000);
        OutputStream outputStream = socket.getOutputStream();

        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
        ImageIO.write(bufferedImage, "jpg", byteArrayOutputStream);

        byte[] size = ByteBuffer.allocate(4).putInt(byteArrayOutputStream.size()).array();
        outputStream.write(size);
        outputStream.write(byteArrayOutputStream.toByteArray());
        outputStream.flush();

        System.out.println(byteArrayOutputStream.size());

        try {
            sleep(1200);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        socket.close();
        System.exit(0);
    } catch (IOException e) {
        e.printStackTrace();
    }
}

服务器应该使用以下代码保存它:

try {
    ServerSocket serverSocket = new ServerSocket(20000);
    System.out.println("-------------------------------------------------------");
    while ( true ) {
        Socket clientSocket = null;
        try {
            clientSocket = serverSocket.accept();
            handleConnection(clientSocket);
        }
        catch (IOException e) {
            e.printStackTrace();
        }
        finally {
            if (clientSocket != null) {
                try {
                    clientSocket.close();
                } catch (IOException e) {
                }
            }
        }
    }
} catch (UnknownHostException uhe) {
    System.out.println(uhe);
} catch (IOException ioe) {
    System.out.println(ioe);
}
static void handleConnection(Socket clientSocket) {
    try {
        OutputStream socketoutstr = clientSocket.getOutputStream();
        OutputStreamWriter osr = new OutputStreamWriter(socketoutstr);
        BufferedWriter bw = new BufferedWriter(osr);

        InputStream socketinstr = clientSocket.getInputStream();
        InputStreamReader isr = new InputStreamReader(socketinstr);
        BufferedReader br = new BufferedReader(isr);


        byte[] sizeAr = new byte[4];
        socketinstr.read(sizeAr);
        int size = ByteBuffer.wrap(sizeAr).asIntBuffer().get();

        byte[] imageAr = new byte[size];
        socketinstr.read(imageAr);

        BufferedImage image = ImageIO.read(new ByteArrayInputStream(imageAr));
        ImageIO.write(image, "png", new File("send.png"));

        System.out.println("111");

        bw.write("");
        bw.newLine();
        bw.flush();

        bw.close();
        br.close();
        clientSocket.close();
    } catch (IOException e) {
        e.printStackTrace();
    }
}

发送的图片需要更多空间:

有人知道如何在不更改任何内容的情况下发送图片的解决方案吗？

Answer 1

我看到两个问题

写入套接字的格式是 jpg，而不是 png。因此比较文件的大小总是会有所不同。

@joachim-sauer 提到了另一个 - read(byte[]) 需要处于循环中:

int read = 0;
while(read<size)
{
  read += socketinstr.read(imageAr,read,size-read); 
}