ArtemisProxy.py

import sys
import time
import socket
from Artemis import Decoder
import select
import argparse

parser = argparse.ArgumentParser("ArtemisProxy : proxy between Artemis clients and server, forwards certain events to OSC server")

parser.add_argument("--serverip", type=str, help="Artemis server IP", required=True)
parser.add_argument("--listenip", type=str, help="ip to listen for clients on", required=True)
parser.add_argument("--oscserverip", type=str, help="OSC server IP", default="")
parser.add_argument("--sntfile", type=str, help="snt file of ship being used", required=True)

args = parser.parse_args()




serverip = args.serverip

ktCount = 0
selectionPacketSent = False

serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
serversocket.bind((args.listenip, 2010))
serversocket.listen(1)
print "Waiting for connection from client on %s .." % (sys.argv[2])
serverSock = None

while True:
	(toClientSock, addr) = serversocket.accept()
	print "got connection from ", addr
	break


#packet header string

splitStr = "\xef\xbe\xad\xde"

print "conecting to artemis server at", serverip
toServerSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)



toServerSock.connect((serverip, 2010))

print "..connected"

#set up our decoder, check if OSC is being used first
oscSettings = []
if args.oscserverip != "":
	addrParse = args.oscserverip.split(":")
	addr = addrParse[0]
	port = 12000
	
	if len(addrParse) == 2:
		port = addrParse[1]	
		oscSettings = [addr, int(port)]
		print "using osc!"

	

print "setting up.."

d = Decoder(args.sntfile,  oscSettings)
inputs = [toServerSock, toClientSock]
outputs = []
#data from artemis server to client
buff = ""
#data from artemis client to server
fromClientBuff = ""

#list of packets extracted from stream to client
packets = []
workingPacket = ""
print "..done! Here we go.."

while(True):

	(read, write, fucked) = select.select(inputs, [], [])
	for r in read:
		if r is toServerSock:

			#read the data from the server
			buff = toServerSock.recv(256)
		elif r is toClientSock:
			#read the data from the client
			fromClientBuff = toClientSock.recv(256)
	#scan the buffer for the start string and length
	packets = []
	startPacket = -1
	pktIndex = 0
	while pktIndex < len(buff):
		if buff[pktIndex : pktIndex + 4] == splitStr:
			pktIndex += 4
			if len(workingPacket) > 0:
				packets.append(workingPacket)
				workingPacket = ""
		else:
			workingPacket += buff[pktIndex]
			pktIndex += 1
	
	for p in packets:
		d.processPacket(p)

	#now we've processed it we can forward data in its respective directions
	if len(buff) > 0:
		toClientSock.send(buff)
		buff = ""


	if len(fromClientBuff) > 0:
		toServerSock.send(fromClientBuff)
		fromClientBuff = ""