Initial commit

example/server.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+import logging
+
+from os import path
+
+import pyrad.packet
+
+from pyrad import server
+from pyrad.dictionary import Dictionary
+
+logging.basicConfig(filename='pyrad.log', level='DEBUG',
+                    format='%(asctime)s [%(levelname)-8s] %(message)s')
+
+
+def print_attributes(packet):
+    print('Attributes')
+    for key, value in packet.items():
+        print(f'{key}: {value}')
+
+
+class FakeServer(server.Server):
+    def HandleAuthPacket(self, packet):
+        print('Received an authentication request')
+        print_attributes(packet)
+
+        reply = self.CreateReplyPacket(packet, **{
+            'Service-Type': 'Framed-User',
+            'Framed-IP-Address': '192.168.0.1',
+            'Framed-IPv6-Prefix': 'fc66::/64'
+        })
+
+        reply.code = pyrad.packet.AccessAccept
+        self.SendReplyPacket(packet.fd, reply)
+
+    def HandleAcctPacket(self, packet):
+        print('Received an accounting request')
+        print_attributes(packet)
+
+        reply = self.CreateReplyPacket(packet)
+        self.SendReplyPacket(packet.fd, reply)
+
+    def HandleCoaPacket(self, packet):
+        print('Received an coa request')
+        print_attributes(packet)
+
+        reply = self.CreateReplyPacket(packet)
+        self.SendReplyPacket(packet.fd, reply)
+
+    def HandleDisconnectPacket(self, packet):
+        print('Received an disconnect request')
+        print_attributes(packet)
+
+        reply = self.CreateReplyPacket(packet)
+        # COA NAK
+        reply.code = 45
+        self.SendReplyPacket(packet.fd, reply)
+
+
+def main(path_to_dictionary):
+    # create server and read dictionary
+    srv = FakeServer(dict=Dictionary(path_to_dictionary),
+                     coa_enabled=True)
+
+    # add clients (address, secret, name)
+    srv.hosts['127.0.0.1'] = server.RemoteHost(
+            '127.0.0.1',
+            b'Kah3choteereethiejeimaeziecumi',
+            'localhost')
+    srv.BindToAddress('0.0.0.0')
+
+    # start server
+    srv.Run()
+
+
+if __name__ == '__main__':
+    dictionary = path.join(path.dirname(path.abspath(__file__)), 'dictionary')
+    main(dictionary)