NaxdyOrg/embassy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Switch to ControlHandler owned bufs for control_in()

Browse source
This commit is contained in:
alexmoon 2022-03-28 20:10:13 -04:00 committed by Dario Nieuwenhuis
parent a22639ad92
commit c53bb7394a
3 changed files with 15 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
embassy-usb/src/control.rs
View file

@ -133,8 +133,8 @@ pub enum OutResponse {
#[derive(Copy, Clone, Eq, PartialEq, Debug)]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
pub enum InResponse {
Accepted(usize),
pub enum InResponse<'a> {
Accepted(&'a [u8]),
Rejected,
}
@ -164,8 +164,7 @@ pub trait ControlHandler {
/// # Arguments
///
/// * `req` - The request from the SETUP packet.
/// * `resp` - The buffer for you to write the response.
fn control_in(&mut self, req: Request, resp: &mut [u8]) -> InResponse {
fn control_in(&mut self, req: Request) -> InResponse<'_> {
InResponse::Rejected
}
}

11
embassy-usb/src/lib.rs
View file

@ -284,13 +284,10 @@ impl<'d, D: Driver<'d>> UsbDevice<'d, D> {
.find(|(i, _)| req.index == *i as _)
.map(|(_, h)| h);
match handler {
Some(handler) => {
let mut buf = [0; 128];
match handler.control_in(req, &mut buf) {
InResponse::Accepted(len) => self.control.accept_in(&buf[..len]).await,
InResponse::Rejected => self.control.reject(),
}
}
Some(handler) => match handler.control_in(req) {
InResponse::Accepted(data) => self.control.accept_in(data).await,
InResponse::Rejected => self.control.reject(),
},
None => self.control.reject(),
}
}

14
examples/nrf/src/bin/usb/cdc_acm.rs
View file

@ -66,6 +66,7 @@ pub struct CdcAcmClass<'d, D: Driver<'d>> {
struct Control<'a> {
shared: &'a ControlShared,
buf: [u8; 7],
}
/// Shared data between Control and CdcAcmClass
@ -138,17 +139,17 @@ impl<'a> ControlHandler for Control<'a> {
}
}
fn control_in(&mut self, req: Request, resp: &mut [u8]) -> InResponse {
fn control_in(&mut self, req: Request) -> InResponse<'_> {
match req.request {
// REQ_GET_ENCAPSULATED_COMMAND is not really supported - it will be rejected below.
REQ_GET_LINE_CODING if req.length == 7 => {
info!("Sending line coding");
let coding = self.shared().line_coding.lock(|x| x.get());
resp[0..4].copy_from_slice(&coding.data_rate.to_le_bytes());
resp[4] = coding.stop_bits as u8;
resp[5] = coding.parity_type as u8;
resp[6] = coding.data_bits;
InResponse::Accepted(7)
self.buf[0..4].copy_from_slice(&coding.data_rate.to_le_bytes());
self.buf[4] = coding.stop_bits as u8;
self.buf[5] = coding.parity_type as u8;
self.buf[6] = coding.data_bits;
InResponse::Accepted(&self.buf)
}
_ => InResponse::Rejected,
}
@ -165,6 +166,7 @@ impl<'d, D: Driver<'d>> CdcAcmClass<'d, D> {
) -> Self {
let control = state.control.write(Control {
shared: &state.shared,
buf: [0; 7],
});
let control_shared = &state.shared;