261 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
			
		
		
	
	
			261 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
| #!/usr/bin/env python
 | |
| 
 | |
| """
 | |
| This is an attempt at bridging matrix clients into a Jitis meet room via Matrix
 | |
| video call.  It uses hard-coded xml strings overg XMPP BOSH. It can display one
 | |
| of the streams from the Jitsi bridge until the second lot of SDP comes down and
 | |
| we set the remote SDP at which point the stream ends. Our video never gets to
 | |
| the bridge.
 | |
| 
 | |
| Requires:
 | |
| npm install jquery jsdom 
 | |
| """
 | |
| 
 | |
| import gevent
 | |
| import grequests
 | |
| from BeautifulSoup import BeautifulSoup
 | |
| import json
 | |
| import urllib
 | |
| import subprocess
 | |
| import time
 | |
| 
 | |
| #ACCESS_TOKEN="" #
 | |
| 
 | |
| MATRIXBASE = 'https://matrix.org/_matrix/client/api/v1/'
 | |
| MYUSERNAME = '@davetest:matrix.org'
 | |
| 
 | |
| HTTPBIND = 'https://meet.jit.si/http-bind'
 | |
| #HTTPBIND = 'https://jitsi.vuc.me/http-bind'
 | |
| #ROOMNAME = "matrix"
 | |
| ROOMNAME = "pibble"
 | |
| 
 | |
| HOST="guest.jit.si"
 | |
| #HOST="jitsi.vuc.me"
 | |
| 
 | |
| TURNSERVER="turn.guest.jit.si"
 | |
| #TURNSERVER="turn.jitsi.vuc.me"
 | |
| 
 | |
| ROOMDOMAIN="meet.jit.si"
 | |
| #ROOMDOMAIN="conference.jitsi.vuc.me"
 | |
| 
 | |
| class TrivialMatrixClient:
 | |
|     def __init__(self, access_token):
 | |
|         self.token = None
 | |
|         self.access_token = access_token
 | |
| 
 | |
|     def getEvent(self):
 | |
|         while True:
 | |
|             url = MATRIXBASE+'events?access_token='+self.access_token+"&timeout=60000"
 | |
|             if self.token:
 | |
|                 url += "&from="+self.token
 | |
|             req = grequests.get(url)
 | |
|             resps = grequests.map([req])
 | |
|             obj = json.loads(resps[0].content)
 | |
|             print "incoming from matrix",obj
 | |
|             if 'end' not in obj:
 | |
|                 continue
 | |
|             self.token = obj['end']
 | |
|             if len(obj['chunk']):
 | |
|                 return obj['chunk'][0]
 | |
| 
 | |
|     def joinRoom(self, roomId):
 | |
|         url = MATRIXBASE+'rooms/'+roomId+'/join?access_token='+self.access_token
 | |
|         print url
 | |
|         headers={ 'Content-Type': 'application/json' }
 | |
|         req = grequests.post(url, headers=headers, data='{}')
 | |
|         resps = grequests.map([req])
 | |
|         obj = json.loads(resps[0].content)
 | |
|         print "response: ",obj
 | |
| 
 | |
|     def sendEvent(self, roomId, evType, event):
 | |
|         url = MATRIXBASE+'rooms/'+roomId+'/send/'+evType+'?access_token='+self.access_token
 | |
|         print url
 | |
|         print json.dumps(event)
 | |
|         headers={ 'Content-Type': 'application/json' }
 | |
|         req = grequests.post(url, headers=headers, data=json.dumps(event))
 | |
|         resps = grequests.map([req])
 | |
|         obj = json.loads(resps[0].content)
 | |
|         print "response: ",obj
 | |
| 
 | |
| 
 | |
| 
 | |
| xmppClients = {}
 | |
| 
 | |
| 
 | |
| def matrixLoop():
 | |
|     while True:
 | |
|         ev = matrixCli.getEvent()
 | |
|         print ev
 | |
|         if ev['type'] == 'm.room.member':
 | |
|             print 'membership event'
 | |
|             if ev['membership'] == 'invite' and ev['state_key'] == MYUSERNAME:
 | |
|                 roomId = ev['room_id']
 | |
|                 print "joining room %s" % (roomId)
 | |
|                 matrixCli.joinRoom(roomId)
 | |
|         elif ev['type'] == 'm.room.message':
 | |
|             if ev['room_id'] in xmppClients:
 | |
|                 print "already have a bridge for that user, ignoring"
 | |
|                 continue
 | |
|             print "got message, connecting"
 | |
|             xmppClients[ev['room_id']] = TrivialXmppClient(ev['room_id'], ev['user_id'])
 | |
|             gevent.spawn(xmppClients[ev['room_id']].xmppLoop)
 | |
|         elif ev['type'] == 'm.call.invite':
 | |
|             print "Incoming call"
 | |
|             #sdp = ev['content']['offer']['sdp']
 | |
|             #print "sdp: %s" % (sdp)
 | |
|             #xmppClients[ev['room_id']] = TrivialXmppClient(ev['room_id'], ev['user_id'])
 | |
|             #gevent.spawn(xmppClients[ev['room_id']].xmppLoop)
 | |
|         elif ev['type'] == 'm.call.answer':
 | |
|             print "Call answered"
 | |
|             sdp = ev['content']['answer']['sdp']
 | |
|             if ev['room_id'] not in xmppClients:
 | |
|                 print "We didn't have a call for that room"
 | |
|                 continue
 | |
|             # should probably check call ID too
 | |
|             xmppCli = xmppClients[ev['room_id']]
 | |
|             xmppCli.sendAnswer(sdp)
 | |
|         elif ev['type'] == 'm.call.hangup':
 | |
|             if ev['room_id'] in xmppClients:
 | |
|                 xmppClients[ev['room_id']].stop()
 | |
|                 del xmppClients[ev['room_id']]
 | |
| 
 | |
| class TrivialXmppClient:
 | |
|     def __init__(self, matrixRoom, userId):
 | |
|         self.rid = 0
 | |
|         self.matrixRoom = matrixRoom
 | |
|         self.userId = userId
 | |
|         self.running = True
 | |
| 
 | |
|     def stop(self):
 | |
|         self.running = False
 | |
| 
 | |
|     def nextRid(self):
 | |
|         self.rid += 1
 | |
|         return '%d' % (self.rid)
 | |
| 
 | |
|     def sendIq(self, xml):
 | |
|         fullXml = "<body rid='%s' xmlns='http://jabber.org/protocol/httpbind' sid='%s'>%s</body>" % (self.nextRid(), self.sid, xml)
 | |
|         #print "\t>>>%s" % (fullXml)
 | |
|         return self.xmppPoke(fullXml)
 | |
| 
 | |
|     def xmppPoke(self, xml):
 | |
|         headers = {'Content-Type': 'application/xml'}
 | |
|         req = grequests.post(HTTPBIND, verify=False, headers=headers, data=xml)
 | |
|         resps = grequests.map([req])
 | |
|         obj = BeautifulSoup(resps[0].content)
 | |
|         return obj
 | |
| 
 | |
|     def sendAnswer(self, answer):
 | |
|         print "sdp from matrix client",answer
 | |
|         p = subprocess.Popen(['node', 'unjingle/unjingle.js', '--sdp'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
 | |
|         jingle, out_err = p.communicate(answer)
 | |
|         jingle = jingle % {
 | |
|             'tojid': self.callfrom,
 | |
|             'action': 'session-accept',
 | |
|             'initiator': self.callfrom,
 | |
|             'responder': self.jid,
 | |
|             'sid': self.callsid
 | |
|         }
 | |
|         print "answer jingle from sdp",jingle
 | |
|         res = self.sendIq(jingle)
 | |
|         print "reply from answer: ",res
 | |
| 
 | |
|         self.ssrcs = {}
 | |
|         jingleSoup = BeautifulSoup(jingle)
 | |
|         for cont in jingleSoup.iq.jingle.findAll('content'):
 | |
|             if cont.description:
 | |
|                 self.ssrcs[cont['name']] = cont.description['ssrc']
 | |
|         print "my ssrcs:",self.ssrcs
 | |
| 
 | |
|         gevent.joinall([
 | |
|                 gevent.spawn(self.advertiseSsrcs)
 | |
|         ])
 | |
| 
 | |
|     def advertiseSsrcs(self):
 | |
|                 time.sleep(7)
 | |
|         print "SSRC spammer started"
 | |
|         while self.running:
 | |
|             ssrcMsg = "<presence to='%(tojid)s' xmlns='jabber:client'><x xmlns='http://jabber.org/protocol/muc'/><c xmlns='http://jabber.org/protocol/caps' hash='sha-1' node='http://jitsi.org/jitsimeet' ver='0WkSdhFnAUxrz4ImQQLdB80GFlE='/><nick xmlns='http://jabber.org/protocol/nick'>%(nick)s</nick><stats xmlns='http://jitsi.org/jitmeet/stats'><stat name='bitrate_download' value='175'/><stat name='bitrate_upload' value='176'/><stat name='packetLoss_total' value='0'/><stat name='packetLoss_download' value='0'/><stat name='packetLoss_upload' value='0'/></stats><media xmlns='http://estos.de/ns/mjs'><source type='audio' ssrc='%(assrc)s' direction='sendre'/><source type='video' ssrc='%(vssrc)s' direction='sendre'/></media></presence>" % { 'tojid': "%s@%s/%s" % (ROOMNAME, ROOMDOMAIN, self.shortJid), 'nick': self.userId, 'assrc': self.ssrcs['audio'], 'vssrc': self.ssrcs['video'] }
 | |
|             res = self.sendIq(ssrcMsg)
 | |
|             print "reply from ssrc announce: ",res
 | |
|             time.sleep(10)
 | |
| 
 | |
| 
 | |
| 
 | |
|     def xmppLoop(self):
 | |
|         self.matrixCallId = time.time()
 | |
|         res = self.xmppPoke("<body rid='%s' xmlns='http://jabber.org/protocol/httpbind' to='%s' xml:lang='en' wait='60' hold='1' content='text/xml; charset=utf-8' ver='1.6' xmpp:version='1.0' xmlns:xmpp='urn:xmpp:xbosh'/>" % (self.nextRid(), HOST))
 | |
| 
 | |
|         print res
 | |
|         self.sid = res.body['sid']
 | |
|         print "sid %s" % (self.sid)
 | |
| 
 | |
|         res = self.sendIq("<auth xmlns='urn:ietf:params:xml:ns:xmpp-sasl' mechanism='ANONYMOUS'/>")
 | |
| 
 | |
|         res = self.xmppPoke("<body rid='%s' xmlns='http://jabber.org/protocol/httpbind' sid='%s' to='%s' xml:lang='en' xmpp:restart='true' xmlns:xmpp='urn:xmpp:xbosh'/>" % (self.nextRid(), self.sid, HOST))
 | |
| 
 | |
|         res = self.sendIq("<iq type='set' id='_bind_auth_2' xmlns='jabber:client'><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></iq>")
 | |
|         print res
 | |
| 
 | |
|         self.jid = res.body.iq.bind.jid.string
 | |
|         print "jid: %s" % (self.jid)
 | |
|         self.shortJid = self.jid.split('-')[0]
 | |
| 
 | |
|         res = self.sendIq("<iq type='set' id='_session_auth_2' xmlns='jabber:client'><session xmlns='urn:ietf:params:xml:ns:xmpp-session'/></iq>")
 | |
| 
 | |
|         #randomthing = res.body.iq['to']
 | |
|         #whatsitpart = randomthing.split('-')[0]
 | |
| 
 | |
|         #print "other random bind thing: %s" % (randomthing)
 | |
| 
 | |
|         # advertise preence to the jitsi room, with our nick
 | |
|         res = self.sendIq("<iq type='get' to='%s' xmlns='jabber:client' id='1:sendIQ'><services xmlns='urn:xmpp:extdisco:1'><service host='%s'/></services></iq><presence to='%s@%s/d98f6c40' xmlns='jabber:client'><x xmlns='http://jabber.org/protocol/muc'/><c xmlns='http://jabber.org/protocol/caps' hash='sha-1' node='http://jitsi.org/jitsimeet' ver='0WkSdhFnAUxrz4ImQQLdB80GFlE='/><nick xmlns='http://jabber.org/protocol/nick'>%s</nick></presence>" % (HOST, TURNSERVER, ROOMNAME, ROOMDOMAIN, self.userId))
 | |
|         self.muc = {'users': []}
 | |
|         for p in res.body.findAll('presence'):
 | |
|             u = {}
 | |
|             u['shortJid'] = p['from'].split('/')[1]
 | |
|             if p.c and p.c.nick:
 | |
|                 u['nick'] = p.c.nick.string
 | |
|             self.muc['users'].append(u)
 | |
|         print "muc: ",self.muc
 | |
| 
 | |
|         # wait for stuff
 | |
|         while True:
 | |
|             print "waiting..."
 | |
|             res = self.sendIq("")
 | |
|             print "got from stream: ",res
 | |
|             if res.body.iq:
 | |
|                 jingles = res.body.iq.findAll('jingle')
 | |
|                 if len(jingles):
 | |
|                     self.callfrom = res.body.iq['from']
 | |
|                     self.handleInvite(jingles[0])
 | |
|             elif 'type' in res.body and res.body['type'] == 'terminate':
 | |
|                 self.running = False
 | |
|                 del xmppClients[self.matrixRoom]
 | |
|                         return
 | |
| 
 | |
|     def handleInvite(self, jingle):
 | |
|         self.initiator = jingle['initiator']
 | |
|         self.callsid = jingle['sid']
 | |
|         p = subprocess.Popen(['node', 'unjingle/unjingle.js', '--jingle'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
 | |
|         print "raw jingle invite",str(jingle)
 | |
|         sdp, out_err = p.communicate(str(jingle))
 | |
|         print "transformed remote offer sdp",sdp
 | |
|         inviteEvent = {
 | |
|             'offer': {
 | |
|                 'type': 'offer',
 | |
|                 'sdp': sdp
 | |
|             },
 | |
|             'call_id': self.matrixCallId,
 | |
|             'version': 0,
 | |
|             'lifetime': 30000
 | |
|         }
 | |
|         matrixCli.sendEvent(self.matrixRoom, 'm.call.invite', inviteEvent)
 | |
| 
 | |
| matrixCli = TrivialMatrixClient(ACCESS_TOKEN)
 | |
| 
 | |
| gevent.joinall([
 | |
|     gevent.spawn(matrixLoop)
 | |
| ])
 | |
| 
 |