receiver_linux.py

# run as root pliz

import asyncio
import websockets
import keyboard
import socket
import time
import subprocess
import pyscreenshot
import base64
from io import BytesIO

# looks the first write doesn't work unless we call a function on the module.
keyboard.is_pressed("e")

def im_2_b64(image):
    buff = BytesIO()
    image.save(buff, format="JPEG")
    img_str = base64.b64encode(buff.getvalue())
    return img_str

async def main():
    async with websockets.connect('ws://192.168.4.148:8766', max_size=None) as websocket:
        await websocket.send("CONNECT_RECEIVER")

        while(True):
            print("waiting for server...")
            server = await websocket.recv()

            if(server.startswith("KEY|")):
                keyboard.write(server[4:])
            
            if(server.startswith("CMD|")):
                output = subprocess.getoutput(server[4:])
                await websocket.send("RECEIVER_OUTPUT|" + output)

            if(server.startswith("SCREEN|")):
                im = pyscreenshot.grab()
                #im.show()
                await websocket.send("RECEIVER_OUTPUT|" + im_2_b64(im).decode("utf-8"))
                
            if(server.startswith("PUT|")):
                try:
                    split = server.split("|")

                    filename = split[1]
                    data = split[2]

                    f = open(filename, 'wb')
                    f.write(base64.b64decode(data.encode("utf-8")))
                    f.close()

                    await websocket.send("RECEIVER_OUTPUT|DONE")
                except:
                    await websocket.send("RECEIVER_OUTPUT|ERROR")

            if(server.startswith("GET|")):
                try:
                    split = server.split("|")

                    source = split[1]

                    f = open(source, "rb")
                    data = base64.b64encode(f.read()).decode("utf-8")
                    await websocket.send("RECEIVER_OUTPUT|" + data)
                except:
                    print("fail")


            print("writing : " + server)


asyncio.get_event_loop().run_until_complete(main())
asyncio.get_event_loop().run_forever()