satip-axe/tools/multicast-rtp

268 lines
6.8 KiB
Plaintext
Raw Normal View History

2016-02-26 20:18:45 +01:00
#!/usr/bin/python3
#
# Tiny SAT>IP client which activates multicast streaming.
#
# Create /etc/sysconfig/multicast file with syntax:
#
# <mport>:<satip-params>
#
# You can modify this file when the program is active and
# reload the configuration using the SIGHUP signal.
#
# Example:
#
# # Channel 1 on multicast port 5554, using first tuner (fe=1)
# 5554:fe=1&src=1&freq=12525&sr=27500&\
# msys=dvbs&mtype=qpsk&pol=v&fec=34&\
# pids=0,1,16,17,18,52,57,100,104,165,299,1029,3002,3003
#
# The SAT>IP parameters should match the SAT>IP specification:
# http://www.satip.info/resources
#
# Note: The multicast address must be specified through minisatip
# option -r <ipv4_mcast> or --remote-rtp <ipv4_mcast>. By default,
# this address is 239.255.255.250.
#
import os
import signal
import functools
import asyncio
import syslog
VERSION='0.1'
SERVER='127.0.0.1:554'
CONFIG='/etc/sysconfig/multicast'
SYSLOG=True
if not os.path.exists('/etc/init.d/axe-settings'):
SERVER='gssbox1:554'
CONFIG='multicast'
SYSLOG=False
#
#
#
def log(msg):
if SYSLOG:
syslog.syslog(msg)
else:
print('LOG:', msg)
#
# Basic RTSP Protocol class
#
class RTSP_Protocol:
def __init__(self, client, params):
self.mport = 0
try:
self.client = client
client.protocol = self
self.mport, self.params = params.split(':')[:2]
self.mport = int(self.mport) & ~1
self.params = self.params.strip().rstrip()
self.action = 'SETUP'
except:
log('Invalid parameters: %s' % params)
client.fatal()
self.timeout = 30
self.session = ''
self.streamID = 0
self.transport = None
self.cseq = 0
def connection_made(self, transport):
self.transport = transport
peer = transport._sock.getpeername()
if self.action == 'SETUP':
s = 'rtsp://%s:%d/?%s' % (peer[0], peer[1], self.params)
log('%d (open): %s' % (self.mport, s))
msg = 'SETUP ' + s + ' RTSP/1.0\r\n'
msg += 'Transport: RTP/AVP;multicast;port=%d-%d\r\n' % (self.mport, self.mport + 1)
msg += 'CSeq: 1\r\n\r\n'
transport.write(msg.encode())
self.action = 'PLAY'
def data_received(self, data):
if not self.transport:
return
lines = data.decode("utf-8").splitlines()
if lines[0] != 'RTSP/1.0 200 OK':
return self.retry()
try:
cseq = 0
streamID = 0
session = ''
timeout = 0
for line in lines[1:]:
if line == '':
break
pos = line.find(':')
if pos >= 0:
header = line[:pos].lower()
data = line[pos+1:].strip().lstrip()
if header == 'cseq':
cseq = int(data)
elif header == 'com.ses.streamid':
streamID = int(data)
elif header == 'session':
a = data.split(';')
session = a[0]
for b in a[1:]:
if b.startswith('timeout='):
timeout = int(b[8:])
if timeout: self.timeout = timeout
if session: self.session = session
if streamID: self.streamID = streamID
except:
self.retry()
return
peer = self.transport._sock.getpeername()
if self.action == 'PLAY':
log('%d (setup): Session %s, streamID %d' % (self.mport, self.session, self.streamID))
self.action = 'OPTIONS'
msg = 'PLAY rtsp://%s:%d/stream=%d RTSP/1.0\r\n' % (peer[0], peer[1], self.streamID)
msg += 'Session: %s\r\n' % self.session
msg += 'CSeq: 2\r\n\r\n'
self.transport.write(msg.encode())
self.client.loop.call_at(loop.time() + self.timeout/2, self.options)
self.cseq = 2
elif self.action == 'OPTIONS' and self.cseq == 2:
log('%d (play): OK' % self.mport)
def options(self):
if not self.transport:
return
peer = self.transport._sock.getpeername()
if self.action == 'OPTIONS':
self.cseq += 1
msg = 'OPTIONS rtsp://%s:%d/stream=%d RTSP/1.0\r\n' % (peer[0], peer[1], self.streamID)
msg += 'Session: %s\r\n' % self.session
msg += 'CSeq: %d\r\n\r\n' % self.cseq
self.transport.write(msg.encode())
self.client.loop.call_at(loop.time() + self.timeout/2, self.options)
def eof_received(self):
self.connection_lost(None)
def connection_lost(self, exc):
if self.transport:
log('%d (connection lost)' % self.mport)
self.client.connection_lost()
self.transport = None
def retry(self):
if self.transport:
self.transport.close()
def fatal(self):
if self.transport:
self.transport.close()
self.client.protocol = None
self.client.fatal()
#
# Basic RTSP Client class
#
class RTSP_Client:
def __init__(self, clients, params):
self.clients = clients
self.loop = clients.loop
self.protocol = None
self.params = params
self.dead = False
self.deleteme = False
log('New multicast client: %s' % (params))
self.create_connection()
def create_connection(self):
srv = SERVER.split(':')
coro = self.loop.create_connection(lambda: RTSP_Protocol(self, self.params),
srv[0], srv[1] and int(srv[1]) or 554)
asyncio.async(coro, loop=self.loop)
def connection_lost(self):
self.protocol = None
self.loop.call_at(loop.time() + 30, self.create_connection)
def fatal(self):
if self.dead:
return
log("Remove multicast client: %s" % self.params)
self.dead = True
if self.protocol:
self.protocol.fatal()
self.clients.remove_client(self)
#
# RTSP clients
#
class RTSP_Clients:
def __init__(self, loop):
self.loop = loop
self.clients = []
self.parse_config_file()
hupfcn = lambda: self.parse_config_file()
loop.add_signal_handler(signal.SIGHUP,
functools.partial(hupfcn))
def remove_client(self, client):
if client in self.clients:
self.clients.remove(client)
client.fatal()
def add_client(self, params):
for c in self.clients:
if c.params == params:
c.deleteme = False
return
self.clients.append(RTSP_Client(self, params))
def parse_config_file(self):
# mark clients
for c in self.clients:
c.deleteme = True
# parse config
lines = []
try:
a = open(CONFIG)
lines = a.readlines()
a.close()
except:
pass
prev = ''
for l in lines:
l = prev + l.strip().rstrip()
prev = ''
if not l or (l[0] in ['#', ';']): continue
if l[-1] == '\\':
prev = l[:-1].rstrip()
else:
self.add_client(l)
# remove marked clients
for c in self.clients:
if c.deleteme:
self.remove_client(c)
#
# configuration file
#
if SYSLOG:
syslog.openlog('multicast-rtp', 0, syslog.LOG_LOCAL7)
log('Version %s' % VERSION)
loop = asyncio.get_event_loop()
clients = RTSP_Clients(loop)
try:
loop.run_forever()
finally:
loop.close()
if SYSLOG:
syslog.closelog()