71 lines
2.4 KiB
Rust
71 lines
2.4 KiB
Rust
use thiserror::Error;
|
|
|
|
mod black_box;
|
|
|
|
#[derive(Error, Debug, PartialEq, Eq)]
|
|
pub enum ConfirmationIdError {
|
|
#[error("Installation ID is too short.")]
|
|
TooShort,
|
|
#[error("Installation ID is too long.")]
|
|
TooLarge,
|
|
#[error("Invalid character in installation ID.")]
|
|
InvalidCharacter,
|
|
#[error("Installation ID checksum failed. Please check that it is typed correctly.")]
|
|
InvalidCheckDigit,
|
|
#[error("Unknown installation ID version.")]
|
|
UnknownVersion,
|
|
#[error("Unable to generate valid confirmation ID.")]
|
|
Unlucky,
|
|
}
|
|
|
|
pub fn generate(installation_id: &str) -> Result<String, ConfirmationIdError> {
|
|
if installation_id.len() < 54 {
|
|
return Err(ConfirmationIdError::TooShort);
|
|
}
|
|
if installation_id.len() > 54 {
|
|
return Err(ConfirmationIdError::TooLarge);
|
|
}
|
|
let inst_id = installation_id.as_bytes();
|
|
let mut conf_id = [0u8; 48];
|
|
let result = black_box::generate(inst_id, &mut conf_id);
|
|
match result {
|
|
0 => {}
|
|
1 => return Err(ConfirmationIdError::TooShort),
|
|
2 => return Err(ConfirmationIdError::TooLarge),
|
|
3 => return Err(ConfirmationIdError::InvalidCharacter),
|
|
4 => return Err(ConfirmationIdError::InvalidCheckDigit),
|
|
5 => return Err(ConfirmationIdError::UnknownVersion),
|
|
6 => return Err(ConfirmationIdError::Unlucky),
|
|
_ => panic!("Unknown error code: {}", result),
|
|
}
|
|
Ok(String::from_utf8_lossy(&conf_id).into())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_generate() {
|
|
assert_eq!(
|
|
generate("334481558826870862843844566221823392794862457401103810").unwrap(),
|
|
"110281-200130-887120-647974-697175-027544-252733"
|
|
);
|
|
assert!(
|
|
generate("33448155882687086284384456622182339279486245740110381")
|
|
.is_err_and(|err| err == ConfirmationIdError::TooShort),
|
|
);
|
|
assert!(
|
|
generate("3344815588268708628438445662218233927948624574011038100")
|
|
.is_err_and(|err| err == ConfirmationIdError::TooLarge),
|
|
);
|
|
assert!(
|
|
generate("33448155882687086284384456622182339279486245740110381!")
|
|
.is_err_and(|err| err == ConfirmationIdError::InvalidCharacter),
|
|
);
|
|
assert!(
|
|
generate("334481558826870862843844566221823392794862457401103811")
|
|
.is_err_and(|err| err == ConfirmationIdError::InvalidCheckDigit),
|
|
);
|
|
}
|
|
}
|