java - 使用 java 套接字通过网络发送屏幕截图

Question

我正在开发一个套接字程序，该程序应该:1/在某台计算机上运行服务器类，2/运行能够连接、断开连接，然后随时重新连接到服务器以接收服务器屏幕截图图像的客户端类。问题是，在运行服务器然后客户端后，我可以开始接收屏幕截图，但过了一会儿我收到了带有不同流 header @的错误。

我尝试过多种解决方案，但没有机会。

请帮忙。谢谢。

服务器类别:

import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;

import javax.imageio.ImageIO;
class Server {

ServerSocket ssoc;
Socket soc = null;
BufferedImage image;
ByteArrayOutputStream baos;
ObjectOutputStream oos;

Server()throws Exception {

    ssoc = new ServerSocket(9001);
    System.out.println("Start ...");
    while (true) {
        System.out.println("Listening ...");
        soc = ssoc.accept();

        OutputStream os = soc.getOutputStream();
        while(true) {
            try {
                image = new Robot().createScreenCapture(new 
Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
                baos = new ByteArrayOutputStream();
                ImageIO.write(image,"png",baos);
                baos.flush();
                baos.close();
                oos = new ObjectOutputStream(os);
                oos.writeObject(baos.size());
                oos.flush();
                os.write(baos.toByteArray()); 
                os.flush();
            } catch(Exception e) { break; }
        }
    }

}

public static void main(String aerg[])throws Exception {
    new Server();
}

}

客户端类:

import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.net.Socket;

import javax.imageio.ImageIO;
import javax.swing.JFrame;
import javax.swing.JPanel;

class Explorer extends JPanel {

static BufferedImage image;
static Socket soc;
static InputStream is;
ObjectInputStream ois;
ByteArrayOutputStream baos;

Explorer()throws Exception {

    soc = new Socket("192.168.1.212",9001);

    is=soc.getInputStream();
    JFrame frame=new JFrame();
    frame.setSize(500,500);
    frame.setContentPane(this);
    frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    frame.setVisible(true);

    while(true) {

        try {
            ois=new ObjectInputStream(soc.getInputStream());
            int size=Integer.parseInt(ois.readObject().toString());
            baos=new ByteArrayOutputStream(size);
            int sizeread=0,bytesin=0;
            byte[] buffer= new byte[1024];
            while(sizeread<size) {
                bytesin=is.read(buffer);
                sizeread+=bytesin;
                baos.write(buffer,0,bytesin);
            }
            baos.close();
            ByteArrayInputStream bais= new 
ByteArrayInputStream(baos.toByteArray());
            image=ImageIO.read(bais);
            this.repaint();
        }
        catch(IOException e) {
            e.printStackTrace();
        }

    }
}

public static void main(String aerg[])throws Exception {
    new Explorer();
}

public void paintComponent(Graphics g) {
    super.paintComponent(g);
    g.drawImage(image,0,0,null);
}

}

输出:

  at java.io.ObjectInputStream.readStreamHeader(ObjectInputStream.java:866)
  at java.io.ObjectInputStream.<init>(ObjectInputStream.java:358)
  at client.Explorer.<init>(Explorer.java:42)
  at client.Explorer.main(Explorer.java:65)
java.io.StreamCorruptedException: invalid stream header: 0BFF0795
  at java.io.ObjectInputStream.readStreamHeader(ObjectInputStream.java:866)
  at java.io.ObjectInputStream.<init>(ObjectInputStream.java:358)
  at client.Explorer.<init>(Explorer.java:42)
  at client.Explorer.main(Explorer.java:65)
java.io.StreamCorruptedException: invalid stream header: 26A9B121
  at java.io.ObjectInputStream.readStreamHeader(ObjectInputStream.java:866)
  at java.io.ObjectInputStream.<init>(ObjectInputStream.java:358)
  at client.Explorer.<init>(Explorer.java:42)
  at client.Explorer.main(Explorer.java:65)

Answer 1

要通过网络连接发送和接收数据，您必须具有以下方法:

• 将数据转换为字符串(序列化)
• 将字符串转换回您的数据(反序列化)

术语“序列化”和“反序列化”并不是特指 Serialized 接口(interface)，而是指用于将结构化数据与 String 相互转换的任何技术。

流行的格式是 JSON(JavaScript 对象表示法)。 Jackson 库 ( https://github.com/FasterXML/jackson-databind ) 被广泛用于执行此映射，并且非常强大和灵活，同时在简单情况下保持简单易用。

在最简单的情况下，您需要 3 件事:

• 表示数据的方式(即类)
• 转换为字符串的函数
• 转换回数据的函数

示例类可能如下所示:

public class ImageMessage {

    private long timeSent;
    private byte[] imageBytes;

    // getters/setters omitted for brevity
}

即使您只需要使用byte[]，最好将其包装在具有有意义名称的类中； byte[] 可以是任何东西(图像、音频文件、加密 key 等)，而 ImageMes​​sage 是一个 ImageMes​​sage 而不是任何东西否则。

接下来您需要映射函数:

public static String toJson(ImageMessage message) throws JsonProcessingException {
    ObjectMapper mapper = new ObjectMapper();  // from the jackson-databind library, include the maven dependency in your pom
    return mapper.writeValueAsString(message);
}

public static ImageMessage fromJson(String string) throws IOException {
    ObjectMapper mapper = new ObjectMapper();
    return mapper.readValue(string, ImageMessage.class);
}

实现此功能的完整类如下所示:

public class ImageMessage {

        private long timeSent;
        private byte[] imageBytes;

        public String toJson() throws JsonProcessingException {
            return new ObjectMapper().writeValueAsString(this);
        }

        public static ImageMessage fromJson(String json) throws IOException {
            ObjectMapper mapper = new ObjectMapper();
            return mapper.readValue(json, ImageMessage.class);
        }

//        getters and setters


        public long getTimeSent() {
            return timeSent;
        }

        public void setTimeSent(long timeSent) {
            this.timeSent = timeSent;
        }

        public byte[] getImageBytes() {
            return imageBytes;
        }

        public void setImageBytes(byte[] imageBytes) {
            this.imageBytes = imageBytes;
        }
    }

从那里，您需要能够读取和写入这些值。

Socket socket = ...  // created earlier

PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream()));
BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));

ImageMessage toSend = new ImageMessage();  // initialize an object
writer.println(toSend.toJson());  // writes the message to the socket
ImageMessage received = ImageMessage.fromJson(reader.readLine());  // reads the next message from the socket

一些注意事项:

• 上面的代码非常脆弱。没有错误检查、json 数据类型检查等。它旨在说明通过 Socket 以比 Java 序列化更可靠的方式发送数据的基本技术
• 如果您打算在比大学项目更严肃的事情中使用它，请考虑使用框架。可靠的服务器很难编写，让 Spring 或 Apache 或任何为您编写的人吧。我个人在我的很多项目中都使用 Spring，因为编写网络代码需要花费大量的艰苦工作(默认情况下，它可以完成上述所有操作以及更多的工作)
• JSON 只是众多序列化格式中的一种。如果您使用遗留系统，XML 可能是首选。 YAML 也是一个可行的选择(尽管根据我的经验，主要用于静态配置，而不是消息)。 Google Protobuf 对于非常受限的系统很有用，但牺牲了一些灵 active 。一般来说，JSON 适合 99% 的情况，但如果需要，请切换。