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

extern crate libusb;

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

pub struct RtlSdr {
    usb_ctx: libusb::Context,
    usb_dd: Option<libusb::DeviceDescriptor>
}

impl RtlSdr {
    
    pub fn new() -> RtlSdr {
        RtlSdr {
            usb_ctx: libusb::Context::new().unwrap(),
            usb_dd: None
        }
    }

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

    pub fn find_device(&self) -> Option<libusb::DeviceDescriptor> {
        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(desc)
                }
            }
        }
        None
    }
}

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

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