通过TCP实现显示屏截图请求及回传

Posted 2020-10-31 livelyleo

在很多业务场景下，需要监视显示屏画面。在实时性要求不高的情况下，可以通过定时对显示屏进行截图及回传实现。

本文通过C#中提供的TCP通信功能，对该功能的实现进行简单描述。

首先，该功能的实现分为客户端和服务端。其中客户端发送显示屏截图请求；服务端接收截图请求后，进行截图并回传；客户端收到服务端回传的截图后，进行显示处理。

1. 客户端

    1）建立对目标主机的TCP连接

    2）向目标主机上的服务端发送截图指令

    3）接收目标主机上的服务端发回的内存流（代表截图），并解析为截图

    具体代码如下：

// 目标主机网络端点：hostIP表示目标主机IP，hostPort表示端口号
var hostEndPoint = new IPEndPoint(IPAddress.Parse(hostIP), hostPort);

// 连接目标主机的Socket
Socket clientSocket = null;
try
{
    // 创建Socket
    clientSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
    clientSocket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);

    // 连接目标主机
    clientSocket.Connect(hostEndPoint);
}
catch (Exception ex)
{
    // 错误日志
}

// 建立Socket连接失败
if (clientSocket == null || !clientSocket.Connected)
return null;

// 向目标主机发送截图命令，截图命令为字符串常量"GetScreenSnap"
clientSocket.Send(Encoding.ASCII.GetBytes("GetScreenSnap"), "GetScreenSnap".Length, SocketFlags.None);

// 接收并解析截图
MemoryStream snapMemoryStream = null;
Image screenSnap = null;
try
{

    // 每次接收1024字节数据（服务端每次发送1024字节）
    byte[] by = new byte[1024];

    // 重试次数
    int retryCount = 0;

    // 前1024字节数据代表的字符串
    string s = "";
    while (true)
    {

        // 成功接收到前1024字节数据
        if ((clientSocket.Receive(by)) > 0)
        {

            // 转化为字符串
            s = Encoding.Unicode.GetString(by);
            by = new byte[1024];

            // 结束循环
            break;
        }

        // 接收失败后，重试次数加1
        retryCount++;

        // 重试次数超过5次
        if (retryCount > 5)
        {

            // 放弃接收截图
            break;
        }
    }

    // 前1024字节代表的字符串不为空
    if (!s.Equals(""))
    {

        // 解析出截图内存流的总字节长度
        s = s.Split(‘!‘)[0];
        long len = long.Parse(s);

        // 代表截图的内存流

        snapMemoryStream = new MemoryStream();

        // 依次接收字节流

        while (len > 0)
        {
            int number;
            if ((number = clientSocket.Receive(by)) > 0)
            {

               // 写入内存流
                snapMemoryStream.Write(by, 0, number);
                len -= number;
            }
            else
            {
                break;
            }
           by = new byte[1024];
        }

        // 根据内存流创建图片
        snapMemoryStream.Position = 0;
        screenSnap = Image.FromStream(snapMemoryStream);
     }
}
catch (Exception ex)
{
    // 错误日志
}
finally
{
    // 关闭内存流
    if (snapMemoryStream != null)
    snapMemoryStream.Close();

    // 关闭连接
    if (clientSocket != null)
    clientSocket.Close();

}

2. 服务端

    1）创建并启动TCP侦听器，监听客户端截图请求

    2）进行显示屏截图

    3）向客户端发送截图

    具体代码如下：

// 创建监听器并开始监听：localhostIP代表本机IP，port表示要监听的端口号
mSnapRequestListener = new TcpListener(localhostIP, port);
mSnapRequestListener.Start();

// 只要服务端在运行，则一直监听
while (isRunning)
{
    TcpClient tcpClient = null;
    Bitmap screenSnap = null;
    try
    {
        // 获取客户端发送的TcpClient
        tcpClient = mSnapRequestListener.AcceptTcpClient();
        if (tcpClient == null || tcpClient.Client == null)
        {
            return;
        }

        // 解析命令文本
        var inBuffer = new Byte["GetScreenSnap".Length];
        tcpClient.Client.Receive(inBuffer, "GetScreenSnap".Length, SocketFlags.None);
        var commandText = Encoding.ASCII.GetString(inBuffer);
        // 不是截图命令
        if (commandText != "GetScreenSnap")
            return;

// 创建一张空白图片
var primaryScreenSize = Screen.PrimaryScreen.Bounds.Size;
screenSnap = new Bitmap(primaryScreenSize.Width, primaryScreenSize.Height, PixelFormat.Format32bppArgb);
// 将屏幕复制到图片上
using (var grfx = Graphics.FromImage(screenSnap))
{
        grfx.CopyFromScreen(new Point(0, 0), new Point(0, 0), primaryScreenSize);
}

using (var snapStream = new MemoryStream())
{
    // 将屏幕截图保存入内存流，并将内存流位置设为0
    screenSnap.Save(snapStream, ImageFormat.Jpeg);
    snapStream.Position = 0;

    // 截图内存流长度
    var streamLength = snapStream.Length.ToString();
    // 扩充为512位（Unicode编码中为1024字节），右边以‘!‘填充
    streamLength = streamLength.PadRight(512, ‘!‘);

    // 初次发送截图内存流长度
    var snapBytes = Encoding.Unicode.GetBytes(streamLength);

    do
    {
        tcpClient.Client.Send(snapBytes);
    }
    // 从流中读取1024个字节，直到读完为止
    while (snapStream.Read(snapBytes, 0, 1024) > 0);
}

    }catch (Exception ex)
    {
        // 错误日志
    }
    finally
    {
        // 销毁图片
        if (screenSnap != null)
        {
            screenSnap.Dispose();
        }

        // 关闭连接
        if (tcpClient != null && tcpClient.Client != null && tcpClient.Client.Connected)
        {
            tcpClient.Client.Close();
        }
    }
}