From 804d65c07ff6c332b786ba18885daea3ac632563 Mon Sep 17 00:00:00 2001
From: phil <phil@secdev.org>
Date: Thu, 16 Mar 2017 18:00:06 +0100
Subject: [PATCH] Added TCPConnectPipe and TCPListenPipe
---
scapy/scapypipes.py | 50 +++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 50 insertions(+)
diff --git a/scapy/scapypipes.py b/scapy/scapypipes.py
index 9af8193e..fbd4747e 100644
--- a/scapy/scapypipes.py
+++ b/scapy/scapypipes.py
@@ -124,3 +124,53 @@ class UDPDrain(Drain):
p = IP(dst=self.ip)/UDP(sport=1234,dport=self.port)/msg
self._send(p)
+
+class FDSourceSink(Source):
+ def __init__(self, fd, name=None):
+ Source.__init__(self, name=name)
+ self.fd = fd
+ def push(self, msg):
+ self.fd.write(msg)
+ def fileno(self):
+ return self.fd.fileno()
+ def deliver(self):
+ self._send(self.fd.read())
+
+
+class TCPConnectPipe(Source):
+ def __init__(self, addr="", port=0, name=None):
+ Source.__init__(self, name=name)
+ self.addr = addr
+ self.port = port
+ self.fd = None
+ def start(self):
+ self.fd = socket.socket()
+ self.fd.connect((self.addr,self.port))
+ def stop(self):
+ self.fd.close()
+ def push(self, msg):
+ self.fd.send(msg)
+ def fileno(self):
+ return self.fd.fileno()
+ def deliver(self):
+ self._send(self.fd.recv(65536))
+
+class TCPListenPipe(TCPConnectPipe):
+ def __init__(self, addr="", port=0, name=None):
+ TCPConnectPipe.__init__(self, addr, port, name)
+ self.connected = False
+ def start(self):
+ self.connected = False
+ self.fd = socket.socket()
+ self.fd.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+ self.fd.bind((self.addr,self.port))
+ self.fd.listen(1)
+ def deliver(self):
+ if self.connected:
+ self._send(self.fd.recv(65536))
+ else:
+ fd,frm = self.fd.accept()
+ self._high_send(repr(frm))
+ self.fd.close()
+ self.fd = fd
+ self.connected = True
--
GitLab