summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nitrocli/src/main.rs47
1 files changed, 27 insertions, 20 deletions
diff --git a/nitrocli/src/main.rs b/nitrocli/src/main.rs
index 6b39ad2..32ecb3a 100644
--- a/nitrocli/src/main.rs
+++ b/nitrocli/src/main.rs
@@ -117,6 +117,23 @@ fn receive<P>(handle: &mut libhid::Handle) -> Result<nitrokey::Report<P>>
}
+/// Transmit a HID feature report to the nitrokey and receive a response.
+fn transmit<PS, PR>(handle: &mut libhid::Handle,
+ report: &nitrokey::Report<PS>)
+ -> Result<nitrokey::Report<PR>>
+ where PS: AsRef<[u8]>,
+ PR: AsRef<[u8]> + Default,
+{
+ send(handle, &report)?;
+
+ // We need to give the stick some time to handle the command. If we
+ // don't, we might just receive stale data from before.
+ thread::sleep(time::Duration::from_millis(SEND_RECV_DELAY_MS));
+
+ return receive::<PR>(handle);
+}
+
+
/// Find and open the nitrokey device and execute a function on it.
fn nitrokey_do(function: &NitroFunc) -> Result<()> {
let hid = libhid::init()?;
@@ -138,12 +155,7 @@ fn open() -> Result<()> {
let payload = nitrokey::EnableEncryptedVolumeCommand::new(&passphrase);
let report = nitrokey::Report::from(payload);
- send(handle, &report)?;
- // We need to give the stick some time to handle the command. If we
- // don't, we might just receive stale data from before.
- thread::sleep(time::Duration::from_millis(SEND_RECV_DELAY_MS));
-
- receive::<nitrokey::EmptyPayload>(handle)?;
+ transmit::<_, nitrokey::EmptyPayload>(handle, &report)?;
return Ok(());
});
}
@@ -155,7 +167,7 @@ fn close() -> Result<()> {
let payload = nitrokey::DisableEncryptedVolumeCommand::new();
let report = nitrokey::Report::from(payload);
- send(handle, &report)?;
+ transmit::<_, nitrokey::EmptyPayload>(handle, &report)?;
return Ok(());
});
}
@@ -205,6 +217,8 @@ mod tests {
#[test]
fn wrong_crc() {
+ type Response = nitrokey::Response<nitrokey::DeviceStatusResponse>;
+
nitrokey_do(&|handle| {
let payload = nitrokey::DeviceStatusCommand::new();
let mut report = nitrokey::Report::from(payload);
@@ -213,14 +227,10 @@ mod tests {
// report of the CRC mismatch) repeatedly.
for _ in 0..10 {
report.crc += 1;
- send(handle, &report).unwrap();
- thread::sleep(time::Duration::from_millis(SEND_RECV_DELAY_MS));
- let new_report = receive::<nitrokey::EmptyPayload>(handle).unwrap();
- assert!(new_report.is_valid());
+ let new_report = transmit::<_, nitrokey::EmptyPayload>(handle, &report)?;
+ let response = AsRef::<Response>::as_ref(&new_report.data);
- let response: &nitrokey::Response<nitrokey::DeviceStatusResponse> = new_report.data
- .as_ref();
assert_eq!(response.command, nitrokey::Command::GetDeviceStatus);
assert_eq!(response.command_crc, report.crc);
assert_eq!(response.command_status, nitrokey::CommandStatus::WrongCrc);
@@ -232,17 +242,14 @@ mod tests {
#[test]
fn device_status() {
+ type Response = nitrokey::Response<nitrokey::DeviceStatusResponse>;
+
nitrokey_do(&|handle| {
let payload = nitrokey::DeviceStatusCommand::new();
let report = nitrokey::Report::from(payload);
- send(handle, &report).unwrap();
- thread::sleep(time::Duration::from_millis(SEND_RECV_DELAY_MS));
-
- let new_report = receive::<nitrokey::EmptyPayload>(handle).unwrap();
- assert!(new_report.is_valid());
-
- let response: &nitrokey::Response<nitrokey::DeviceStatusResponse> = new_report.data.as_ref();
+ let report = transmit::<_, nitrokey::EmptyPayload>(handle, &report)?;
+ let response = AsRef::<Response>::as_ref(&report.data);
assert!(response.device_status == nitrokey::StorageStatus::Idle ||
response.device_status == nitrokey::StorageStatus::Okay);