add multicast-rtp package
This commit is contained in:
Jaroslav Kysela
16
Makefile
16
Makefile
@@ -73,6 +73,8 @@ PYTHON3_PACKAGE_NAME=$(PYTHON3)-1
|
||||
PYTHON3_FILENAME=$(PYTHON3).tgz
|
||||
PYTHON3_DOWNLOAD=https://www.python.org/ftp/python/$(PYTHON3_VERSION)/$(PYTHON3_FILENAME)
|
||||
|
||||
MULTICAST_RTP_PACKAGE_NAME=multicast-rtp-1
|
||||
|
||||
TVHEADEND_COMMIT=master
|
||||
|
||||
# 10663 10937
|
||||
@@ -623,6 +625,20 @@ apps/$(PYTHON3)/compiled.stamp: apps/$(PYTHON3)/patch.stamp
|
||||
.PHONY: python3
|
||||
python3: apps/$(PYTHON3)/compiled.stamp
|
||||
|
||||
#
|
||||
# multicast-rtp
|
||||
#
|
||||
|
||||
apps/multicast-rtp/ok.stamp: tools/multicast-rtp
|
||||
mkdir -p apps/multicast-rtp/sbin apps/multicast-rtp/etc/init.extra
|
||||
cp tools/multicast-rtp apps/multicast-rtp/sbin
|
||||
ln -sf ../../sbin/multicast-rtp apps/multicast-rtp/etc/init.extra/multicast-rtp
|
||||
$(call PACKAGE,apps/multicast-rtp,$(MULTICAST_RTP_PACKAGE_NAME),etc sbin)
|
||||
touch apps/multicast-rtp/ok.stamp
|
||||
|
||||
.PHONY: multicast-rtp
|
||||
multicast-rtp: apps/multicast-rtp/ok.stamp
|
||||
|
||||
#
|
||||
# tvheadend
|
||||
#
|
||||
|
||||
267
tools/multicast-rtp
Executable file
267
tools/multicast-rtp
Executable file
@@ -0,0 +1,267 @@
|
||||
#!/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()
|
||||
Reference in New Issue
Block a user