Adding listener imports for future development

mandiant · Oct 12, 2017 · 200a3b9 · 200a3b9
1 parent e3b6327
commit 200a3b9
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 1 deletion.
diff --git a/fakenet/listeners/BITSListener.py b/fakenet/listeners/BITSListener.py
@@ -23,7 +23,6 @@
 
 from . import *
 
-
 # BITS Protocol header keys
 K_BITS_SESSION_ID = 'BITS-Session-Id'
 K_BITS_ERROR_CONTEXT = 'BITS-Error-Context'

diff --git a/fakenet/listeners/DNSListener.py b/fakenet/listeners/DNSListener.py
@@ -8,6 +8,8 @@
 import ssl
 import socket
 
+from . import *
+
 class DNSListener():
 
     def taste(self, data, dport):

diff --git a/fakenet/listeners/IRCListener.py b/fakenet/listeners/IRCListener.py
@@ -11,6 +11,8 @@
 
 import BannerFactory
 
+from . import *
+
 RPL_WELCOME = '001'
 SRV_WELCOME = "Welcome to FakeNet."
 

diff --git a/fakenet/listeners/POPListener.py b/fakenet/listeners/POPListener.py
@@ -9,6 +9,8 @@
 import ssl
 import socket
 
+from . import *
+
 EMAIL = """From: "Bob Example" <bob@example.org>
 To: Alice Example <alice@example.com>
 Cc: theboss@example.com

diff --git a/fakenet/listeners/RawListener.py b/fakenet/listeners/RawListener.py
@@ -9,6 +9,7 @@
 import ssl
 import socket
 
+from . import *
 
 class RawListener():