From 89f36717e53f6e5642ecded9fd23a663c4b22f64 Mon Sep 17 00:00:00 2001
From: moznion <moznion@gmail.com>
Date: Sat, 28 Nov 2020 04:08:26 +0900
Subject: [PATCH] Add example implementation for server and client

---
 examples/client.rs | 21 ++++++++++++
 examples/server.rs | 80 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 101 insertions(+)
 create mode 100644 examples/client.rs
 create mode 100644 examples/server.rs

diff --git a/examples/client.rs b/examples/client.rs
new file mode 100644
index 0000000..1a3cda7
--- /dev/null
+++ b/examples/client.rs
@@ -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);
+}
diff --git a/examples/server.rs b/examples/server.rs
new file mode 100644
index 0000000..abb840b
--- /dev/null
+++ b/examples/server.rs
@@ -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)
+    }
+}