xwords/xwords4/android/scripts/relay.py
2017-10-22 21:29:16 -07:00

89 lines
2.7 KiB
Python
Executable file

#!/usr/bin/python
import base64, json, mod_python, socket, struct, sys
PROTOCOL_VERSION = 0
PRX_GET_MSGS = 4
try:
from mod_python import apache
apacheAvailable = True
except ImportError:
apacheAvailable = False
def post(req, params, timeoutSecs = 1):
err = 'none'
dataLen = 0
jobj = json.loads(params)
data = base64.b64decode(jobj['data'])
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.settimeout(timeoutSecs) # seconds
addr = ("127.0.0.1", 10997)
sock.sendto(data, addr)
responses = []
while True:
try:
data, server = sock.recvfrom(1024)
responses.append(base64.b64encode(data))
except socket.timeout:
#If data is not received back from server, print it has timed out
err = 'timeout'
break
jobj = {'err' : err, 'data' : responses}
return json.dumps(jobj)
def query(req, ids):
print('ids', ids)
ids = json.loads(ids)
idsLen = 0
for id in ids: idsLen += len(id)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(5) # seconds
sock.connect(('127.0.0.1', 10998))
lenShort = 2 + idsLen + len(ids) + 2
print(lenShort, PROTOCOL_VERSION, PRX_GET_MSGS, len(ids))
header = struct.Struct('!hBBh')
assert header.size == 6
sock.send(header.pack(lenShort, PROTOCOL_VERSION, PRX_GET_MSGS, len(ids)))
for id in ids: sock.send(id + '\n')
unpacker = struct.Struct('!2H') # 2s f')
resLen, nameCount = unpacker.unpack(sock.recv(unpacker.size)) # problem when ids empty
print('resLen:', resLen, 'nameCount:', nameCount)
msgsLists = {}
if nameCount == len(ids):
for ii in range(nameCount):
perGame = []
shortUnpacker = struct.Struct('!H')
countsThisGame, = shortUnpacker.unpack(sock.recv(shortUnpacker.size)) # problem
print('countsThisGame:', countsThisGame)
for jj in range(countsThisGame):
msgLen, = shortUnpacker.unpack(sock.recv(shortUnpacker.size))
print('msgLen:', msgLen)
msgs = []
if msgLen > 0:
msg = sock.recv(msgLen)
print('msg len:', len(msg))
msg = base64.b64encode(msg)
msgs.append(msg)
perGame.append(msgs)
msgsLists[ids[ii]] = perGame
return json.dumps(msgsLists)
def main():
print(query(None, json.dumps(sys.argv[1:])))
# Params = { 'data' : 'V2VkIE9jdCAxOCAwNjowNDo0OCBQRFQgMjAxNwo=' }
# params = json.dumps(params)
# print(post(None, params))
##############################################################################
if __name__ == '__main__':
main()