use crate::{Result, YubiKey};
use std::{
borrow::Cow,
ffi::CStr,
sync::{Arc, Mutex},
};
pub type Iter<'ctx> = std::vec::IntoIter<Reader<'ctx>>;
pub struct Context {
ctx: Arc<Mutex<pcsc::Context>>,
reader_names: Vec<u8>,
}
impl Context {
pub fn open() -> Result<Self> {
let ctx = pcsc::Context::establish(pcsc::Scope::System)?;
let reader_names = vec![0u8; ctx.list_readers_len()?];
Ok(Self {
ctx: Arc::new(Mutex::new(ctx)),
reader_names,
})
}
pub fn iter(&mut self) -> Result<Iter<'_>> {
let Self { ctx, reader_names } = self;
let reader_cstrs: Vec<_> = {
let c = ctx.lock().unwrap();
c.is_valid()?;
c.list_readers(reader_names)?.collect()
};
#[allow(clippy::needless_collect)]
let readers: Vec<_> = reader_cstrs
.iter()
.map(|name| Reader::new(name, Arc::clone(ctx)))
.collect();
Ok(readers.into_iter())
}
}
pub struct Reader<'ctx> {
name: &'ctx CStr,
ctx: Arc<Mutex<pcsc::Context>>,
}
impl<'ctx> Reader<'ctx> {
fn new(name: &'ctx CStr, ctx: Arc<Mutex<pcsc::Context>>) -> Self {
Self { name, ctx }
}
pub fn name(&self) -> Cow<'_, str> {
self.name.to_string_lossy()
}
pub fn open(&self) -> Result<YubiKey> {
self.try_into()
}
pub(crate) fn connect(&self) -> Result<pcsc::Card> {
let ctx = self.ctx.lock().unwrap();
Ok(ctx.connect(self.name, pcsc::ShareMode::Shared, pcsc::Protocols::T1)?)
}
}