aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/lib.rs13
1 files changed, 4 insertions, 9 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 259ff89..471d481 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -679,17 +679,12 @@ pub fn connect() -> Result<UnauthenticatedDevice, CommandError> {
/// }
/// ```
pub fn connect_model(model: Model) -> Result<UnauthenticatedDevice, CommandError> {
- let model_string = match model {
- Model::Storage => "S",
- Model::Pro => "P",
+ let model = match model {
+ Model::Storage => nitrokey_sys::NK_device_model_NK_STORAGE,
+ Model::Pro => nitrokey_sys::NK_device_model_NK_PRO,
};
- let model_cstring = CString::new(model_string);
- if model_cstring.is_err() {
- return Err(CommandError::InvalidString);
- }
- let model = model_cstring.unwrap();
unsafe {
- return match nitrokey_sys::NK_login(model.as_ptr()) {
+ return match nitrokey_sys::NK_login_enum(model) {
1 => Ok(UnauthenticatedDevice {}),
rv => Err(CommandError::from(rv)),
};