hdd 0.10.3

hdd: instruments for querying ATA and SCSI disks
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
92
93

use super::parser::{self, Entry};
use super::DriveDB;

use std::fs::File;
use std::io::prelude::*;
use std::io;

use nom;

quick_error! {
	#[derive(Debug)]
	pub enum Error {
		IO(err: io::Error) {
			from()
			display("IO error: {}", err)
			description(err.description())
			cause(err)
		}
		Parse {
			// TODO? Parse(nom::verbose_errors::Err) if dependencies.nom.features = ["verbose-errors"]
			display("Unable to parse the drivedb")
			description("malformed database")
		}
	}
}

fn load(file: &str) -> Result<Vec<Entry>, Error> {
	let mut db = Vec::new();
	File::open(&file)?.read_to_end(&mut db)?;

	match parser::database(&db) {
		nom::IResult::Done(_, entries) => Ok(entries),
		nom::IResult::Error(_) => Err(Error::Parse),
		nom::IResult::Incomplete(_) => unreachable!(), // XXX is it true?
	}
}

/**
Use this helper to load entries from `drivedb.h`.

It is also possible to use `Loader` to create dummy database in case if only user-defined attributes (or none at all) are needed: just call [`db()`](#method.db) right after creating the loader.
*/
#[derive(Debug)]
pub struct Loader {
	entries: Vec<Entry>,
	additional: Vec<Entry>,
}
impl Loader {
	pub fn new() -> Self {
		Loader {
			entries: vec![],
			additional: vec![],
		}
	}
	/**
	Loads entries from main drivedb file.

	Entries from previously loaded main file will be discarded; entries from additional files will not be affected.

	## Errors

	Returns [enum Error](enum.Error.html) if:

	- it encounters any kind of I/O error,
	- drive database is malformed.
	*/
	pub fn load(&mut self, file: &str) -> Result<(), Error> {
		self.entries = load(file)?;
		Ok(())
	}
	/**
	Loads more entries from additional drivedb file. Additional entries always take precedence over the ones from the main file.

	## Errors

	Returns [enum Error](enum.Error.html) if:

	- it encounters any kind of I/O error,
	- drive database is malformed.
	*/
	pub fn load_additional(&mut self, file: &str) -> Result<(), Error> {
		self.entries = load(file)?;
		Ok(())
	}
	/// Returns actual drive database with all entries loaded beforehand.
	pub fn db(self) -> DriveDB {
		let entries: Vec<_> = self.additional.into_iter()
			.chain(self.entries.into_iter())
			.collect();

		DriveDB::new(entries)
	}
}