#!/usr/bin/python
import socket
import json
import base64
import time

s = socket.socket()
s.settimeout(2)

def write(data):
    if type(data) is int:
        data = chr(data)
    if type(data) in (list, tuple):
        data = ''.join([chr(x) for x in data])
    s.send(json.dumps({'command':'write', 'argument':base64.b64encode(data)}))
    r = json.loads(s.recv(1024))
    print(r)
    return base64.b64decode(r['data'])

def read(n):
    s.send(json.dumps({'command':'read', 'argument':n}))
    r = json.loads(s.recv(1024))
    print(r)
    return base64.b64decode(r['data'])
    
def cs(b):
    s.send(json.dumps({'command':'select', 'argument':b}))
    r = json.loads(s.recv(1024))
    return r.get('data')

s.connect(('10.79.2.188', 1337))
print(s.recv(1024))


print(cs(False))
print(cs(True))

print(write(0x10))

val = read(1)
print((val, ord(val), hex(ord(val))))

print(cs(False))

s.close()