pyWebREPL.py

import struct, socket, time
from random import randint

class WEBREPL:
    def upgrade_http_to_websocket(self):
        frw = self.s.makefile("rwb", 0)
        frw.write(b"""GET / HTTP/1.1\r\nHost: echo.websocket.org\r\nConnection: Upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key: foo\r\n\r\n""")
        while True:
            l = frw.readline()
            if l == b'\r\n': break
            # Response = '''HTTP/1.1 101 Switching Protocols\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: BM0S0+ghftShuFVHQATK/DBiJq8=\r\n\r'''

        buf = b''
        while 1:
            c = self.s.recv(1)
            buf += c
            if c == b':':
                self.s.recv(1)
                break

    def _send(self, payload):
        l = len(payload)
        if l > 125: raise
        key = [randint(0, 255),randint(0, 255),randint(0, 255),randint(0, 255)]
        masked_payload = [ord(c)^key[i%4] for i, c in enumerate(payload) ]
        frame = struct.pack(f">{6+l}B", 0b10000001, 0b10000000 | l, *key,  *masked_payload)
        self.s.send(frame)

    def send(self, code):
        self._send([b'\x05'])
        sec_len = 10 # chop the string if it is longer than 100 char
        for cmd in code.strip().splitlines():
            for i in range(len(cmd)//sec_len+1):
                self._send(cmd[i*sec_len: (i+1)*sec_len])  
                time.sleep(0.03) # This prevents prompt overflow 
            time.sleep(0.01) # This prevents prompt overflow 
            self._send([b'\r']) 
        self._send([b'\x04'])

    def recv(self):
        buf = ''
        while True:
            while True:
                _ = self.s.recv(1)
                if  _ != b'\x81': # ignore nonsense for us
                    break
            if _ == b'\x0d':  # print when receive a Carriage Return even if the buffer is not returnable
                print(buf)
                buf = ''
            if ord(_)>31:     # only buffer readable char
                buf += _.decode()
            if buf[-4:]=='>>> ': # the buffer is now returnable
                return buf

    def soft_reboot(self):
        self._send([b'\x04']) # Send \D
        self.close()

    def write_main(self, main_code):
        self.send('''fw=open('main.py','w')''')
        self.recv()
        for code in main_code.strip().splitlines():
            self.send("""fw.write('''"""+code+"""\\r''')""")
            self.recv() 
        self.send('''fw.flush()\nfw.close()''')
        self.recv()

    def __init__(self, host='192.168.1.1', port=8266, password='123456'):
        self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.s.connect((host, port))
        self.upgrade_http_to_websocket()
        self._send(password)
        self._send('\r')
        print(self.recv())

    def close(self):
        self.s.send(struct.pack(f">6B", 0x88, 0x80, 0x00, 0x00, 0x00, 0x00))