src/lib.rs


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

extern crate libusb;

const KNOWN_DEVICES: [(u16, u16, &str); 2] = [
    (0x0bda, 0x2832, "Generic RTL2832U"),
    (0x0bda, 0x2838, "Generic RTL2832U OEM")
];

struct Usb<'a> {
    ctx: libusb::Context,
    dev: Option<libusb::DeviceHandle<'a>>
}

pub struct RtlSdr<'a> {
    usb: Usb<'a>
}

impl<'a> RtlSdr<'a> {
    
    pub fn new() -> RtlSdr<'a> {
        RtlSdr {
            usb: Usb {
                ctx: libusb::Context::new().unwrap(),
                dev: None
            }
        }
    }

    pub fn init(&mut self) {
        self.usb.dev = self.find_device();
    }

    pub fn find_device(&self) -> Option<libusb::DeviceHandle<'a>> {
        for mut dev in self.usb.ctx.devices().unwrap().iter() {
            let desc = dev.device_descriptor().unwrap();
            let vid = desc.vendor_id();
            let pid = desc.product_id();

            for kd in KNOWN_DEVICES.iter() {
                if kd.0 == vid && kd.1 == pid {
                    return Some(dev.open().unwrap())
                }
            }
        }
        None
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_init() {
        let mut rtlsdr = RtlSdr::new();
        assert!(rtlsdr.usb.dev.is_none());
        rtlsdr.init();
        assert!(rtlsdr.usb.dev.is_some());
    }
}