hwaddr 0.1.7

MAC address handling.
Documentation 
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

//            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
//                    Version 2, December 2004
//
// Copyleft (ↄ) meh. <meh@schizofreni.co> | http://meh.schizofreni.co
//
// Everyone is permitted to copy and distribute verbatim or modified
// copies of this license document, and changing it is allowed as long
// as the name is changed.
//
//            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
//   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
//
//  0. You just DO WHAT THE FUCK YOU WANT TO.

use std::env;

#[cfg(features = "database")]
use std::{io::{Read, Write, BufWriter}, fs::File, path::Path, collections::HashSet};
#[cfg(features = "database")]
use url::Url;
#[cfg(features = "database")]
use regex::Regex;

fn main() {
	if env::var("CARGO_FEATURE_DATABASE").is_err() {
		return;
	}

	#[cfg(features = "database")]
	smol::block_on(async {
		let mut content = String::new();

		if let Ok(value) = env::var("HWADDR_DATABASE") {
			File::open(value).unwrap().read_to_string(&mut content).unwrap();
		}
		else if let Ok(url) = env::var("HWADDR_DOWNLOAD") {
			content.push_str(&if Url::parse(&url).is_ok() {
				reqwest::get(&url).await.unwrap()
			}
			else {
				reqwest::get("http://standards.ieee.org/develop/regauth/oui/oui.txt").await.unwrap()
			}.text().await.unwrap());
		}
		else {
			File::open("database.txt").unwrap().read_to_string(&mut content).unwrap();
		}

		let     path    = Path::new(&env::var("OUT_DIR").unwrap()).join("database.rs");
		let mut file    = BufWriter::new(File::create(&path).unwrap());
		let mut builder = phf_codegen::Map::new();

		write!(&mut file, "pub static DATABASE: ::phf::Map<[u8; 3], Producer> = ").unwrap();

		let mut lines   = content.lines();
		let mut entries = HashSet::new();
		let     start   = Regex::new(r#"^(?P<prefix>[[:xdigit:]]{6})\s+\(base 16\)\s*(?P<name>.+)$"#).unwrap();

		while let Some(line) = lines.next() {
			if let Some(captures) = start.captures(line) {
				let prefix = captures["prefix"].to_owned();
				let prefix = [
					u8::from_str_radix(&prefix[0 .. 2], 16).unwrap(),
					u8::from_str_radix(&prefix[2 .. 4], 16).unwrap(),
					u8::from_str_radix(&prefix[4 .. 6], 16).unwrap() ];

				let name    = captures["name"].to_owned();
				let street  = lines.next().unwrap().trim().to_owned();
				let city    = lines.next().unwrap().trim().to_owned();
				let country = lines.next().unwrap().trim().to_owned();

				if !entries.contains(&prefix) {
					entries.insert(prefix);
					builder.entry(prefix, &format!(r#"
						Producer {{
							prefix:  {:?},
							name:    {:?},
							address: Address {{
								street:  {:?},
								city:    {:?},
								country: {:?},
							}},
						}}
					"#, prefix, name, street, city, country));
				}
			}
		}

		write!(&mut file, "{}", builder.build()).unwrap();
		write!(&mut file, ";\n").unwrap();
	});
}