1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
use chbs::scheme::ToScheme as _;
use rand::seq::SliceRandom as _;
const SYMBOLS: &[u8] = b"!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~";
const NUMBERS: &[u8] = b"0123456789";
const LETTERS: &[u8] =
b"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
const NONCONFUSABLES: &[u8] = b"34678abcdefhjkmnpqrtuwxy";
#[derive(Debug, Eq, PartialEq, Copy, Clone)]
pub enum Type {
AllChars,
NoSymbols,
Numbers,
NonConfusables,
Diceware,
}
pub fn pwgen(ty: Type, len: usize) -> String {
if ty == Type::Diceware {
let config = chbs::config::BasicConfig {
words: len,
capitalize_first: chbs::probability::Probability::Never,
capitalize_words: chbs::probability::Probability::Never,
..chbs::config::BasicConfig::default()
};
return config.to_scheme().generate();
}
let alphabet = match ty {
Type::AllChars => {
let mut v = vec![];
v.extend(SYMBOLS.iter().copied());
v.extend(NUMBERS.iter().copied());
v.extend(LETTERS.iter().copied());
v
}
Type::NoSymbols => {
let mut v = vec![];
v.extend(NUMBERS.iter().copied());
v.extend(LETTERS.iter().copied());
v
}
Type::Numbers => {
let mut v = vec![];
v.extend(NUMBERS.iter().copied());
v
}
Type::NonConfusables => {
let mut v = vec![];
v.extend(NONCONFUSABLES.iter().copied());
v
}
Type::Diceware => unreachable!(),
};
let mut rng = rand::thread_rng();
let mut pass = vec![];
pass.extend(
std::iter::repeat_with(|| alphabet.choose(&mut rng).unwrap())
.take(len),
);
// unwrap is safe because the method of generating passwords guarantees
// valid utf8
String::from_utf8(pass).unwrap()
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn test_pwgen() {
let pw = pwgen(Type::AllChars, 50);
assert_eq!(pw.len(), 50);
// technically this could fail, but the chances are incredibly low
// (around 0.000009%)
assert_duplicates(&pw);
let pw = pwgen(Type::AllChars, 100);
assert_eq!(pw.len(), 100);
assert_duplicates(&pw);
let pw = pwgen(Type::NoSymbols, 100);
assert_eq!(pw.len(), 100);
assert_duplicates(&pw);
let pw = pwgen(Type::Numbers, 100);
assert_eq!(pw.len(), 100);
assert_duplicates(&pw);
let pw = pwgen(Type::NonConfusables, 100);
assert_eq!(pw.len(), 100);
assert_duplicates(&pw);
}
#[track_caller]
fn assert_duplicates(s: &str) {
let mut set = std::collections::HashSet::new();
for c in s.chars() {
set.insert(c);
}
assert!(set.len() < s.len())
}
}
|