82 lines
2.1 KiB
Python
Executable File
82 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)
|