78 lines
2.1 KiB
Python
Executable File
78 lines
2.1 KiB
Python
Executable File
#!/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)
|