Adding asynchronous servers with callbacks

Author: AlexisTM

rawsocketpy/__init__.py

@@ -1,4 +1,11 @@
 # Package init
+from __future__ import print_function
+try:
+    from gevent import monkey; monkey.patch_all()
+    from .asyncserver import RawAsyncServer, RawAsyncServerCallback
+except ImportError:
+    print("Gevent could not be loaded; the sockets will not be cooperative.")
 
 from .util import get_hw, protocol_to_ethertype, to_bytes, to_str
 from .main import RawPacket, RawSocket
+from .server import RawServer, RawRequestHandler, RawServerCallback

rawsocketpy/asyncserver.py

@@ -0,0 +1,11 @@
+from .server import RawServer, RawServerCallback
+from gevent import pool, Greenlet
+
+class RawAsyncServer(RawServer):
+    pool = pool.Pool()
+    def handle_handler(self, handler):
+        self.pool.start(Greenlet(handler.run))
+
+class RawAsyncServerCallback(RawServerCallback, RawAsyncServer):
+    def handle_handler(self, handler):
+        self.pool.start(Greenlet(self.callback, handler, self))

rawsocketpy/main.py

@@ -27,6 +27,8 @@ class RawSocket(object):
     def __init__(self, interface, protocol, sock=None, no_recv_protocol=False):
         if  not 0x0000 < protocol < 0xFFFF:
             raise ValueError("Protocol has to be in the range 0 to 65535")
+        self.no_recv_protocol = no_recv_protocol
+        self.non_processed_protocol = protocol
         self.protocol = socket.htons(protocol)
         self.ethertype = protocol_to_ethertype(protocol)
         self.interface = interface
@@ -35,6 +37,10 @@ def __init__(self, interface, protocol, sock=None, no_recv_protocol=False):
             self.sock = self.sock_create(self.interface, 0, sock)
         else: 
             self.sock = self.sock_create(self.interface, self.protocol, sock)
+        self.close = self.sock.close
+
+    def dup(self):
+        return RawSocket(self.interface, self.non_processed_protocol, self.sock.dup(), self.no_recv_protocol)
 
     @staticmethod
     def sock_create(interface, protocol, sock=None):
@@ -55,17 +61,3 @@ def recv(self):
 
     def __str__(self):
         return self.interface
-
-def main():
-    na = NetworkAdapter("wlp2s0", 0xAA42)
-    i = 0
-    while True:
-        try:
-            na.send(str(i))
-            time.sleep(0.1)
-            i += 1
-        except:
-            break
-
-if __name__ == '__main__':
-    main()

rawsocketpy/server.py

@@ -0,0 +1,56 @@
+from .main import RawPacket, RawSocket
+from .util import get_hw, to_bytes, protocol_to_ethertype
+
+class RawServer(object):
+    def __init__(self, interface, protocol, RequestHandlerClass):
+        self.RequestHandlerClass = RequestHandlerClass
+        self.socket = RawSocket(interface, protocol)
+        self.recv = self.socket.recv
+        self.running = False
+
+    def spin_once(self):
+        packet = self.recv()
+        handler = self.RequestHandlerClass(packet, self)
+        self.handle_handler(handler)
+
+    def handle_handler(self, handler):
+        handler.run()
+
+    def spin(self):
+        self.running = True
+        while self.running:
+            self.spin_once()
+
+
+class RawServerCallback(RawServer):
+    def __init__(self, interface, protocol, RequestHandlerClass, callback):
+        self.callback = callback
+        RawServer.__init__(self, interface, protocol, RequestHandlerClass)
+    
+    def handle_handler(self, handler):
+        self.callback(handler, self)
+
+
+class RawRequestHandler(object):
+    def __init__(self, packet, server):
+        self.packet = packet
+        self.server = server
+        # self.server.socket.sock = low level socket
+
+    def finish(self):
+        pass
+    
+    def setup(self):
+        pass
+
+    def handle(self):
+        print(self.packet)
+    
+    def run(self):
+        try:
+            self.setup()
+            self.handle()
+        except Exception as e:
+            print(e)
+        finally:
+            self.finish()

setup.py

@@ -4,8 +4,8 @@
     long_description = fh.read()
 
 setuptools.setup(name='rawsocketpy',
-      version='0.1',
-      description='This library allows you to implemnet a custom layer 2 communication using raw sockets in Python',
+      version='0.2',
+      description='This library allows you to implemnet a custom layer 2 communication using raw sockets in Python 2 and Python 3, synchronous and asynchronous, with and without callbacks.',
       long_description=long_description,
       long_description_content_type="text/markdown",
       url='https://github.com/AlexisTM/rawsocket_python',