168 lines
5.2 KiB
Python
168 lines
5.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from twisted.internet.protocol import ClientFactory
|
|
from twisted.protocols.basic import LineReceiver
|
|
from frn.utility import *
|
|
|
|
|
|
class FRNClient(LineReceiver):
|
|
|
|
client_version = 2010002
|
|
|
|
def connectionMade(self):
|
|
self.login()
|
|
|
|
def ready(self):
|
|
self.status = 'READY'
|
|
self.msgbuffer = []
|
|
self.phase = 0
|
|
self.setRawMode()
|
|
|
|
def startMultiLineMessage(self, msgtype, rest=''):
|
|
self.status = msgtype
|
|
self.phase = None
|
|
self.setLineMode(rest)
|
|
|
|
def collectMultiLineMessage(self, line):
|
|
if self.phase is None:
|
|
while line[0] not in '0123456789': # needed for client list
|
|
line = line[1:]
|
|
self.expected_lines = int(line.strip())
|
|
self.msgbuffer = []
|
|
self.phase = 0
|
|
else:
|
|
self.msgbuffer.append(line)
|
|
self.phase += 1
|
|
if self.phase >= self.expected_lines:
|
|
handler = getattr(self, 'on_'+self.status, self.on_unimplemented)
|
|
message = self.msgbuffer
|
|
self.ready()
|
|
handler(message)
|
|
|
|
def startAudioMessage(self, rest=''):
|
|
self.status = 'AUDIO'
|
|
self.phase = None
|
|
self.msgbuffer = ''
|
|
if len(rest) > 0:
|
|
self.collectAudioMessage(rest)
|
|
|
|
def collectAudioMessage(self, data):
|
|
needed = min(327, max(0, 327-len(self.msgbuffer)))
|
|
if len(data) > 0:
|
|
self.msgbuffer += data[:needed]
|
|
if len(self.msgbuffer) >= 327:
|
|
audio_data = self.msgbuffer
|
|
source = ord(audio_data[0])*256+ord(audio_data[1])
|
|
self.ready()
|
|
self.on_AUDIO(source, audio_data[2:])
|
|
if len(data) > needed:
|
|
self.dataReceived(data[needed:])
|
|
|
|
def lineReceived(self, line):
|
|
if self.status == 'AUTH':
|
|
if self.phase == 0:
|
|
self.latest_client_version = int(line.strip())
|
|
self.phase = 1
|
|
else:
|
|
self.serverdata = parseSimpleXML(line.strip())
|
|
print self.serverdata
|
|
if int(self.serverdata['sv']) > 2009004:
|
|
self.sendLine(makeAuthKey(self.serverdata['kp']))
|
|
self.ready()
|
|
else:
|
|
self.collectMultiLineMessage(line)
|
|
|
|
def rawDataReceived(self, data):
|
|
if self.status == 'READY':
|
|
packet_type = ord(data[0])
|
|
if packet_type == 0: # Keepalive
|
|
self.sendLine('P')
|
|
elif packet_type == 1: # TX ack
|
|
self.status == 'TX'
|
|
self.phase = 0
|
|
if len(data) > 1:
|
|
self.dataReceived(data[1:])
|
|
elif packet_type == 2: # Audio
|
|
self.startAudioMessage(data[1:])
|
|
elif packet_type == 3: # Client list
|
|
self.startMultiLineMessage('CLIENTS', data[1:])
|
|
elif packet_type == 4: # SMS
|
|
self.startMultiLineMessage('SMS', data[1:])
|
|
elif packet_type == 5: # Channel list
|
|
self.startMultiLineMessage('CHANNELS', data[1:])
|
|
else:
|
|
print "Unknown packet type %d" % packet_type
|
|
elif self.status == 'AUDIO':
|
|
self.collectAudioMessage(data)
|
|
elif self.status == 'TX':
|
|
if self.phase == 0:
|
|
self.phase = 1
|
|
self.on_TX(ord(data[0])*256+ord(data[1]))
|
|
|
|
def login(self):
|
|
d = self.factory.client_id
|
|
fields = [
|
|
('VX', self.client_version),
|
|
('EA', d['email']),
|
|
('PW', d['password']),
|
|
('ON', d['operator']),
|
|
('BC', d['transmission']),
|
|
('DS', d['description']),
|
|
('NN', d['country']),
|
|
('CT', d['city']),
|
|
('NT', d['network'])
|
|
]
|
|
ap = "CT:"+formatSimpleXML(fields)
|
|
self.status = 'AUTH'
|
|
self.phase = 0
|
|
self.sendLine(ap)
|
|
#self.request_rx()
|
|
|
|
def set_status(self, status):
|
|
self.sendLine('ST:%s' % str(status))
|
|
|
|
def request_rx(self):
|
|
self.sendLine('RX0')
|
|
|
|
def request_tx(self):
|
|
self.sendLine('TX0')
|
|
|
|
def send_audio(self, frame):
|
|
self.sendLine('TX1')
|
|
self.transport.write(frame)
|
|
|
|
def send_SMS(self, dest, text):
|
|
self.sendLine('TM:'+formatSimpleXML(dict(ID=dest, MS=text)))
|
|
|
|
def on_unimplemented(self, msg):
|
|
print msg
|
|
|
|
def on_AUDIO(self, from_id, frames):
|
|
pass
|
|
|
|
def on_TX(self, my_id):
|
|
pass
|
|
|
|
|
|
class FRNClientFactory(ClientFactory):
|
|
|
|
protocol = FRNClient
|
|
|
|
def __init__(self, **kw):
|
|
self.client_id = kw
|
|
|
|
def startedConnecting(self, connector):
|
|
print 'Started to connect.'
|
|
|
|
def buildProtocol(self, addr):
|
|
print 'Connected.'
|
|
return ClientFactory.buildProtocol(self, addr)
|
|
|
|
def clientConnectionLost(self, connector, reason):
|
|
print 'Lost connection. Reason:', reason
|
|
|
|
def clientConnectionFailed(self, connector, reason):
|
|
print 'Connection failed. Reason:', reason
|
|
|
|
# vim: set et ai sw=4 ts=4 sts=4:
|