|
|
|
@ -1,9 +1,10 @@ |
|
|
|
//! WchLink device type
|
|
|
|
|
|
|
|
use log::info; |
|
|
|
use rusb::{DeviceHandle, UsbContext}; |
|
|
|
|
|
|
|
use crate::{ |
|
|
|
commands::{ChipId, Response}, |
|
|
|
commands::{ChipId, RawCommand, Response}, |
|
|
|
transport::Transport, |
|
|
|
Result, RiscvChip, |
|
|
|
}; |
|
|
|
@ -14,6 +15,12 @@ const PRODUCT_ID: u16 = 0x8010; |
|
|
|
const ENDPOINT_OUT: u8 = 0x01; |
|
|
|
const ENDPOINT_IN: u8 = 0x81; |
|
|
|
|
|
|
|
const VENDOR_ID_DAP: u16 = 0x1a86; |
|
|
|
const PRODUCT_ID_DAP: u16 = 0x8012; |
|
|
|
|
|
|
|
const ENDPOINT_OUT_DAP: u8 = 0x02; |
|
|
|
// const ENDPOINT_IN_DAP: u8 = 0x83;
|
|
|
|
|
|
|
|
/// Attached chip information
|
|
|
|
#[derive(Debug)] |
|
|
|
pub struct ChipInfo { |
|
|
|
@ -29,8 +36,8 @@ pub struct ChipInfo { |
|
|
|
pub memory_start_addr: u32, |
|
|
|
// Fields for ROM/RAM split
|
|
|
|
pub sram_code_mode: u8, |
|
|
|
pub(crate) rom_kb: u32, |
|
|
|
pub(crate) ram_kb: u32, |
|
|
|
//pub(crate) rom_kb: u32,
|
|
|
|
//pub(crate) ram_kb: u32,
|
|
|
|
} |
|
|
|
|
|
|
|
#[derive(Debug)] |
|
|
|
@ -54,18 +61,15 @@ impl WchLink { |
|
|
|
.unwrap_or(false) |
|
|
|
}) |
|
|
|
.nth(nth) |
|
|
|
.map_or( |
|
|
|
Err(crate::error::Error::Custom("No such device".to_string())), |
|
|
|
Ok, |
|
|
|
)?; |
|
|
|
.map_or(Err(crate::error::Error::ProbeNotFound), Ok)?; |
|
|
|
|
|
|
|
let mut device_handle = device.open()?; |
|
|
|
|
|
|
|
let config = device.active_config_descriptor()?; |
|
|
|
|
|
|
|
let descriptor = device.device_descriptor()?; |
|
|
|
// let descriptor = device.device_descriptor()?;
|
|
|
|
|
|
|
|
log::trace!("Device descriptor: {:?}", &descriptor); |
|
|
|
log::trace!("Device: {:?}", &device); |
|
|
|
|
|
|
|
device_handle.claim_interface(0)?; |
|
|
|
|
|
|
|
@ -114,3 +118,85 @@ impl Drop for WchLink { |
|
|
|
let _ = self.device_handle.release_interface(0); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/// Switch from DAP mode to RV mode
|
|
|
|
// ref: https://github.com/cjacker/wchlinke-mode-switch/blob/main/main.c
|
|
|
|
pub fn try_switch_from_rv_to_dap(nth: usize) -> Result<()> { |
|
|
|
let dev = open_usb_device(VENDOR_ID, PRODUCT_ID, nth)?; |
|
|
|
info!("Switch mode WCH-LinkRV {:?}", dev.device()); |
|
|
|
|
|
|
|
let mut dev = WchLink { |
|
|
|
device_handle: dev, |
|
|
|
chip: None, |
|
|
|
}; |
|
|
|
let info = dev.probe_info()?; |
|
|
|
info!("probe info: {:?}", info); |
|
|
|
|
|
|
|
let _ = dev.send_command(RawCommand::<0xff>(vec![0x41])); |
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
/// Switch from RV mode to DAP mode
|
|
|
|
pub fn try_switch_from_dap_to_rv(nth: usize) -> Result<()> { |
|
|
|
let dev = open_usb_device(VENDOR_ID_DAP, PRODUCT_ID_DAP, nth)?; |
|
|
|
info!("Switch mode for WCH-LinkDAP {:?}", dev.device()); |
|
|
|
|
|
|
|
let buf = [0x81, 0xff, 0x01, 0x52]; |
|
|
|
log::trace!("send {} {}", hex::encode(&buf[..3]), hex::encode(&buf[3..])); |
|
|
|
let _ = dev.write_bulk( |
|
|
|
ENDPOINT_OUT_DAP, |
|
|
|
&buf, |
|
|
|
std::time::Duration::from_millis(5000), |
|
|
|
); |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
/// Check connected USB device
|
|
|
|
pub fn check_usb_device() -> Result<()> { |
|
|
|
let context = rusb::Context::new()?; |
|
|
|
log::trace!("Acquired libusb context."); |
|
|
|
|
|
|
|
for device in context.devices()?.iter() { |
|
|
|
let desc = device.device_descriptor()?; |
|
|
|
if desc.vendor_id() == VENDOR_ID && desc.product_id() == PRODUCT_ID { |
|
|
|
log::info!("Found WCH-LinkRV, {:?}", device); |
|
|
|
} else if desc.vendor_id() == VENDOR_ID_DAP && desc.product_id() == PRODUCT_ID_DAP { |
|
|
|
log::info!("Found WCH-LinkDAP, {:?}", device); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
fn open_usb_device( |
|
|
|
vendor_id: u16, |
|
|
|
produce_id: u16, |
|
|
|
nth: usize, |
|
|
|
) -> Result<DeviceHandle<rusb::Context>> { |
|
|
|
let context = rusb::Context::new()?; |
|
|
|
log::trace!("Acquired libusb context."); |
|
|
|
|
|
|
|
let device = context |
|
|
|
.devices()? |
|
|
|
.iter() |
|
|
|
.filter(|device| { |
|
|
|
device |
|
|
|
.device_descriptor() |
|
|
|
.map(|desc| desc.vendor_id() == vendor_id && desc.product_id() == produce_id) |
|
|
|
.unwrap_or(false) |
|
|
|
}) |
|
|
|
.nth(nth) |
|
|
|
.map_or(Err(crate::error::Error::ProbeNotFound), Ok)?; |
|
|
|
|
|
|
|
let mut device_handle = device.open()?; |
|
|
|
|
|
|
|
device_handle.claim_interface(0)?; |
|
|
|
|
|
|
|
log::trace!("Claimed interface 0 of USB device."); |
|
|
|
|
|
|
|
// TODO: endpoint check
|
|
|
|
// let config = device.active_config_descriptor()?;
|
|
|
|
|
|
|
|
Ok(device_handle) |
|
|
|
} |
|
|
|
|
Andelf
1 year ago