Add example implementation for server and client

This commit is contained in:
moznion
2020-11-28 04:08:26 +09:00
parent bf412991b0
commit 89f36717e5
2 changed files with 101 additions and 0 deletions

21
examples/client.rs Normal file
View File

@@ -0,0 +1,21 @@
#[macro_use]
extern crate log;
use radius_rs::client::Client;
use radius_rs::code::Code;
use radius_rs::packet::Packet;
use radius_rs::rfc2865;
use std::net::SocketAddr;
#[tokio::main]
async fn main() {
env_logger::init();
let remote_addr: SocketAddr = "127.0.0.1:1812".parse().unwrap();
let mut req_packet = Packet::new(Code::AccessRequest, &b"secret".to_vec());
rfc2865::add_user_name(&mut req_packet, "admin");
rfc2865::add_user_password(&mut req_packet, b"p@ssw0rd").unwrap(); // TODO
let res = Client::send_packet(&remote_addr, &req_packet).await;
info!("response: {:?}", res);
}

80
examples/server.rs Normal file
View File

@@ -0,0 +1,80 @@
#[macro_use]
extern crate log;
use std::io;
use std::net::SocketAddr;
use async_trait::async_trait;
use tokio::net::UdpSocket;
use tokio::signal;
use radius_rs::code::Code;
use radius_rs::request::Request;
use radius_rs::request_handler::RequestHandler;
use radius_rs::rfc2865;
use radius_rs::secret_provider::{SecretProvider, SecretProviderError};
use radius_rs::server::Server;
#[tokio::main]
async fn main() {
env_logger::init();
let server_future = Server::run(
"0.0.0.0",
1812,
1500,
true,
MyRequestHandler {},
MySecretProvider {},
signal::ctrl_c(),
);
let result = server_future.await;
info!("{:?}", result);
}
struct MyRequestHandler {}
#[async_trait]
impl RequestHandler<(), io::Error> for MyRequestHandler {
async fn handle_radius_request(
&self,
conn: &UdpSocket,
req: &Request,
) -> Result<(), io::Error> {
let req_packet = req.get_packet();
let maybe_user_name_attr = rfc2865::lookup_user_name(req_packet);
let maybe_user_password_attr = rfc2865::lookup_user_password(req_packet);
let user_name = maybe_user_name_attr.unwrap().to_string().unwrap();
let user_password = String::from_utf8(
maybe_user_password_attr
.unwrap()
.to_user_password(req_packet.get_secret(), req_packet.get_authenticator())
.unwrap(),
)
.unwrap();
let code = if user_name == "admin" && user_password == "p@ssw0rd" {
Code::AccessAccept
} else {
Code::AccessReject
};
info!("response => {:?} to {}", code, req.get_remote_addr());
conn.send_to(
&req_packet.response(code).encode().unwrap(),
req.get_remote_addr(),
)
.await?;
Ok(())
}
}
struct MySecretProvider {}
impl SecretProvider for MySecretProvider {
fn fetch_secret(&self, _remote_addr: SocketAddr) -> Result<Vec<u8>, SecretProviderError> {
let bs = b"secret".to_vec();
Ok(bs)
}
}