cubixle/radius-rs
1
0
Fork 0
mirror of https://github.com/cubixle/radius-rs.git synced 2026-04-25 00:14:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Support octets type

Browse Source
This commit is contained in:
moznion
2020-11-25 01:13:31 +09:00
parent 426b2faf53
commit cb9876940c
3 changed files with 126 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Makefile
View File

@@ -1,3 +1,6 @@
build:
cargo build
gen:
cat /dev/null > $(PWD)/src/rfc2865.rs
cargo run --bin code_gen $(PWD)/dicts/dictionary.rfc2865 $(PWD)/src/rfc2865.rs

40
src/bin/code_gen.rs
View File

@@ -123,13 +123,16 @@ fn generate_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {
match attr.value_type {
RadiusAttributeValueType::STRING => {
generate_common_attribute_code(w, attr);
generate_string_attribute_code(w, attr)
generate_string_attribute_code(w, attr);
}
RadiusAttributeValueType::USER_PASSWORD => {
generate_common_attribute_code(w, attr);
generate_user_password_attribute_code(w, attr)
generate_user_password_attribute_code(w, attr);
}
RadiusAttributeValueType::OCTETS => {
generate_common_attribute_code(w, attr);
generate_octets_attribute_code(w, attr);
}
// RadiusAttributeValueType::OCTETS => generate_octets_attribute_code(w, attr),
// RadiusAttributeValueType::IPADDR => generate_ipaddr_attribute_code(w, attr),
// RadiusAttributeValueType::INTEGER => generate_integer_attribute_code(w, attr),
// RadiusAttributeValueType::VSA => generate_vsa_attribute_code(w, attr),
@@ -204,13 +207,36 @@ fn generate_user_password_attribute_code(w: &mut BufWriter<File>, attr: &RadiusA
w.write_all(code.as_bytes()).unwrap();
}
fn generate_octets_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {}
fn generate_octets_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {
let attr_name = attr.name.clone();
fn generate_ipaddr_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {}
let type_identifier = format!("{}_TYPE", attr_name.to_screaming_snake_case());
let type_calling = format!("Self::{}", type_identifier);
fn generate_integer_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {}
let code = format!(
"pub fn add_{method_identifier}(packet: &mut Packet, value: &[u8]) {{
let attr = Attribute::from_bytes(value);
packet.add({type_calling}, &attr);
}}
",
method_identifier = attr_name.to_snake_case(),
type_calling = type_calling,
);
fn generate_vsa_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {}
w.write_all(code.as_bytes()).unwrap();
}
fn generate_ipaddr_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {
unimplemented!()
}
fn generate_integer_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {
unimplemented!()
}
fn generate_vsa_attribute_code(w: &mut BufWriter<File>, attr: &RadiusAttribute) {
unimplemented!()
}
type DictParsed = (Vec<RadiusAttribute>, HashMap<String, Vec<RadiusValue>>);

90
src/rfc2865.rs
View File

@@ -38,6 +38,21 @@ impl RFC2865 {
Ok(())
}
pub const CHAP_PASSWORD_TYPE: AVPType = 3;
pub fn delete_chap_password(packet: &mut Packet) {
packet.delete(Self::CHAP_PASSWORD_TYPE);
}
pub fn lookup_chap_password(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::CHAP_PASSWORD_TYPE)
}
pub fn lookup_all_chap_password(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::CHAP_PASSWORD_TYPE)
}
pub fn add_chap_password(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::CHAP_PASSWORD_TYPE, &attr);
}
pub const FILTER_ID_TYPE: AVPType = 11;
pub fn delete_filter_id(packet: &mut Packet) {
packet.delete(Self::FILTER_ID_TYPE);
@@ -113,6 +128,36 @@ impl RFC2865 {
packet.add(Self::FRAMED_ROUTE_TYPE, &attr);
}
pub const STATE_TYPE: AVPType = 24;
pub fn delete_state(packet: &mut Packet) {
packet.delete(Self::STATE_TYPE);
}
pub fn lookup_state(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::STATE_TYPE)
}
pub fn lookup_all_state(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::STATE_TYPE)
}
pub fn add_state(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::STATE_TYPE, &attr);
}
pub const CLASS_TYPE: AVPType = 25;
pub fn delete_class(packet: &mut Packet) {
packet.delete(Self::CLASS_TYPE);
}
pub fn lookup_class(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::CLASS_TYPE)
}
pub fn lookup_all_class(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::CLASS_TYPE)
}
pub fn add_class(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::CLASS_TYPE, &attr);
}
pub const CALLED_STATION_ID_TYPE: AVPType = 30;
pub fn delete_called_station_id(packet: &mut Packet) {
packet.delete(Self::CALLED_STATION_ID_TYPE);
@@ -158,6 +203,21 @@ impl RFC2865 {
packet.add(Self::NAS_IDENTIFIER_TYPE, &attr);
}
pub const PROXY_STATE_TYPE: AVPType = 33;
pub fn delete_proxy_state(packet: &mut Packet) {
packet.delete(Self::PROXY_STATE_TYPE);
}
pub fn lookup_proxy_state(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::PROXY_STATE_TYPE)
}
pub fn lookup_all_proxy_state(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::PROXY_STATE_TYPE)
}
pub fn add_proxy_state(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::PROXY_STATE_TYPE, &attr);
}
pub const LOGIN_LAT_SERVICE_TYPE: AVPType = 34;
pub fn delete_login_lat_service(packet: &mut Packet) {
packet.delete(Self::LOGIN_LAT_SERVICE_TYPE);
@@ -188,6 +248,21 @@ impl RFC2865 {
packet.add(Self::LOGIN_LAT_NODE_TYPE, &attr);
}
pub const LOGIN_LAT_GROUP_TYPE: AVPType = 36;
pub fn delete_login_lat_group(packet: &mut Packet) {
packet.delete(Self::LOGIN_LAT_GROUP_TYPE);
}
pub fn lookup_login_lat_group(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::LOGIN_LAT_GROUP_TYPE)
}
pub fn lookup_all_login_lat_group(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::LOGIN_LAT_GROUP_TYPE)
}
pub fn add_login_lat_group(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::LOGIN_LAT_GROUP_TYPE, &attr);
}
pub const FRAMED_APPLE_TALK_ZONE_TYPE: AVPType = 39;
pub fn delete_framed_apple_talk_zone(packet: &mut Packet) {
packet.delete(Self::FRAMED_APPLE_TALK_ZONE_TYPE);
@@ -203,6 +278,21 @@ impl RFC2865 {
packet.add(Self::FRAMED_APPLE_TALK_ZONE_TYPE, &attr);
}
pub const CHAP_CHALLENGE_TYPE: AVPType = 60;
pub fn delete_chap_challenge(packet: &mut Packet) {
packet.delete(Self::CHAP_CHALLENGE_TYPE);
}
pub fn lookup_chap_challenge(packet: &Packet) -> Option<&Attribute> {
packet.lookup(Self::CHAP_CHALLENGE_TYPE)
}
pub fn lookup_all_chap_challenge(packet: &Packet) -> Vec<&Attribute> {
packet.lookup_all(Self::CHAP_CHALLENGE_TYPE)
}
pub fn add_chap_challenge(packet: &mut Packet, value: &[u8]) {
let attr = Attribute::from_bytes(value);
packet.add(Self::CHAP_CHALLENGE_TYPE, &attr);
}
pub const LOGIN_LAT_PORT_TYPE: AVPType = 63;
pub fn delete_login_lat_port(packet: &mut Packet) {
packet.delete(Self::LOGIN_LAT_PORT_TYPE);