如何在设置一个简单的HTTP Server来执行Windows命令和截图

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@file: device_http_server.py
@time: 2022/12/2 13:11
@file_desc: 
"""
import os
from http.server import BaseHTTPRequestHandler, HTTPServer

import pyautogui
import socket

PORT = 9999
IP = socket.gethostbyname(socket.gethostname())
print(IP)


class ScreenHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        cmd = self.path.strip("/")
        if "." not in cmd and "screenshot" != cmd.lower():
            try:
                print(cmd)
                if "%20" in cmd:
                    cmd = cmd.replace("%20", " ")
                cmd_exe = os.popen(cmd)
                output = cmd_exe.read()
                self.send_response(200)  # HTTP Status Code
                self.send_header('Content-Type', 'text/plain')
                self.end_headers()
                self.wfile.write(output.encode("utf-8"))
            except Exception as error:
                pass
        elif "screenshot" == cmd.lower():
            img = pyautogui.screenshot()  # Screenshot
            if img:
                self.send_response(200)  # HTTP Status Code
                self.send_header('Content-Type', 'image/png')
                self.end_headers()
                img.save(self.wfile, 'PNG')  # HTTP Response Stream


def main():
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # check port is open
        result = sock.connect_ex((IP, PORT))
        portopen = result == 0
        sock.close()
        if not portopen:
            # start web server use defined handler
            server = HTTPServer((IP, PORT), ScreenHandler)
            server.serve_forever()  # listen
    except:
        pass


if __name__ == '__main__':
    main()

 

 

实现远程在Windows机器上执行命令和截图。

1. 使用socket和http.server搭建一个简单的web server.

2. pyautogui实现截图功能

3. 通过获取浏览器的path, 截取要执行的命令（注意空格的替换%20）， 通过os.popen执行命令，并将结果写回web页面。


目前将python代码打包成exe放在开机启动项执行（不可以打包成服务在后台运行，否则截图功能没法使用）

具体代码如上。