Initial commit

example/acct.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+import random
+import socket
+import sys
+
+from os import path
+
+import pyrad.packet
+
+from pyrad.client import Client
+from pyrad.dictionary import Dictionary
+
+
+def send_accounting_packet(srv, req):
+    try:
+        srv.SendPacket(req)
+    except pyrad.client.Timeout:
+        print('RADIUS server does not reply')
+        sys.exit(1)
+    except socket.error as error:
+        print('Network error: ' + error[1])
+        sys.exit(1)
+
+
+def main(path_to_dictionary):
+    srv = Client(server='127.0.0.1',
+                 secret=b'Kah3choteereethiejeimaeziecumi',
+                 dict=Dictionary(path_to_dictionary))
+
+    req = srv.CreateAcctPacket(**{
+        'User-Name': 'wichert',
+        'NAS-IP-Address': '192.168.1.10',
+        'NAS-Port': 0,
+        'NAS-Identifier': 'trillian',
+        'Called-Station-Id': '00-04-5F-00-0F-D1',
+        'Calling-Station-Id': '00-01-24-80-B3-9C',
+        'Framed-IP-Address': '10.0.0.100',
+    })
+
+    print('Sending accounting start packet')
+    req['Acct-Status-Type'] = 'Start'
+    send_accounting_packet(srv, req)
+
+    print('Sending accounting stop packet')
+    req['Acct-Status-Type'] = 'Stop'
+    req['Acct-Input-Octets'] = random.randrange(2**10, 2**30)
+    req['Acct-Output-Octets'] = random.randrange(2**10, 2**30)
+    req['Acct-Session-Time'] = random.randrange(120, 3600)
+    req['Acct-Terminate-Cause'] = random.choice(['User-Request', 'Idle-Timeout'])
+    send_accounting_packet(srv, req)
+
+
+if __name__ == '__main__':
+    dictionary = path.join(path.dirname(path.abspath(__file__)), 'dictionary')
+    main(dictionary)