forked from TeaPackCZ/heidi
-
Notifications
You must be signed in to change notification settings - Fork 0
/
paveproxy.py
56 lines (48 loc) · 1.46 KB
/
paveproxy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#!/usr/bin/python
"""
Socket for filtering PaVE (Parrot Video Encapsulation) packets
./paveproxy.py <input file>
"""
import sys
import struct
import socket
from threading import Thread,Event,Lock
from pave import PaVE
HOST = 'localhost' # access URL for cv2.VideoCapture
PORT = 50007
BUFFER_SIZE = 1024
class PaVEProxy( Thread ):
def __init__( self, readCbk ):
Thread.__init__( self )
self.setDaemon(True)
self.lock = Lock()
self.shouldIRun = Event()
self.shouldIRun.set()
self.serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket, self.addr = None, None
self.readCbk = readCbk
self.pave = PaVE()
self.start()
def run( self ):
"wait only for the first client - delayed binding"
self.serverSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.serverSocket.bind((HOST, PORT))
self.serverSocket.listen(1)
self.socket, self.addr = self.serverSocket.accept()
while self.shouldIRun.isSet():
self.pave.append( self.readCbk( BUFFER_SIZE ) )
header,payload = self.pave.extract()
if payload:
self.socket.send( payload )
def term( self ):
self.shouldIRun.clear()
if self.socket==None:
print "KILLING run()"
tmpSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tmpSocket.connect((HOST, PORT))
tmpSocket.close()
print "DONE"
if __name__ == "__main__":
if len(sys.argv) < 2:
print __doc__
sys.exit(2)