successfully wrote data

Signed-off-by: Sean Cross <sean@xobs.io>
This commit is contained in:
2020-04-23 18:43:04 +08:00
parent 05230aa6c4
commit e88650850b
7 changed files with 931 additions and 37 deletions

View File

@ -1,13 +1,15 @@
use ftdi_vcp_sys::{
FT_Close, FT_GetBitMode, FT_GetComPortNumber, FT_OpenEx, FT_Purge, FT_ResetDevice,
FT_SetBitMode, FT_Write, DWORD, FT_HANDLE, FT_OPEN_BY_DESCRIPTION, FT_STATUS, LONG, LPDWORD,
LPVOID, PVOID, UCHAR,
FT_Close, FT_GetBitMode, FT_GetComPortNumber, FT_GetLatencyTimer, FT_OpenEx, FT_Purge, FT_Read,
FT_ResetDevice, FT_SetBitMode, FT_SetLatencyTimer, FT_Write, DWORD, FT_HANDLE,
FT_OPEN_BY_DESCRIPTION, FT_STATUS, LONG, LPDWORD, LPVOID, PVOID, UCHAR,
};
use std::convert::TryInto;
use std::ffi::CString;
use std::io::{Read, Write};
use std::mem::MaybeUninit;
pub mod mpsse;
#[derive(Debug)]
pub enum BitMode {
Reset,
@ -175,6 +177,7 @@ impl VCP {
if result != Error::NoError {
Err(result)
} else {
self.bit_mode = bitmode;
Ok(())
}
}
@ -202,27 +205,126 @@ impl VCP {
}
}
pub fn write(&mut self, out_buffer: &[u8]) -> Result<usize, Error> {
let mut bytes_written = MaybeUninit::<DWORD>::uninit();
let result = Error::from(unsafe {
FT_Write(
self.handle,
out_buffer.as_ptr() as LPVOID,
out_buffer
.len()
.try_into()
.expect("couldn't convert buffer length to DWORD"),
bytes_written.as_mut_ptr() as LPDWORD,
)
});
pub fn latency_timer(&mut self) -> Result<u8, Error> {
let mut latency = MaybeUninit::<UCHAR>::uninit();
let result = Error::from(unsafe { FT_GetLatencyTimer(self.handle, latency.as_mut_ptr()) });
if result != Error::NoError {
Err(result)
} else {
Ok(unsafe { bytes_written.assume_init() }
.try_into()
.expect("invalid number of bytes written"))
Ok(unsafe { latency.assume_init() })
}
}
pub fn set_latency_timer(&mut self, latency: u8) -> Result<(), Error> {
let result = Error::from(unsafe { FT_SetLatencyTimer(self.handle, latency) });
if result != Error::NoError {
Err(result)
} else {
Ok(())
}
}
// pub fn write(&mut self, out_buffer: &[u8]) -> Result<usize, Error> {
// let mut bytes_written = MaybeUninit::<DWORD>::uninit();
// let result = Error::from(unsafe {
// FT_Write(
// self.handle,
// out_buffer.as_ptr() as LPVOID,
// out_buffer
// .len()
// .try_into()
// .expect("couldn't convert buffer length to DWORD"),
// bytes_written.as_mut_ptr() as LPDWORD,
// )
// });
// if result != Error::NoError {
// Err(result)
// } else {
// Ok(unsafe { bytes_written.assume_init() }
// .try_into()
// .expect("invalid number of bytes written"))
// }
// }
pub fn set_gpio(&mut self, value: u8, direction: u8) -> Result<(), Error> {
match self.bit_mode {
BitMode::MPSSE => self
.write_all(&[mpsse::Command::MC_SETB_LOW.to_u8(), value, direction])
.or_else(|_| Err(Error::IoError)),
_ => unimplemented!(),
}
}
pub fn readb_low(&mut self) -> Result<u8, Error> {
self.write_all(&[mpsse::Command::MC_READB_LOW.to_u8()])
.or_else(|_| Err(Error::IoError))?;
let mut result = [0; 1];
self.read(&mut result).or_else(|_| Err(Error::IoError))?;
Ok(result[0])
}
pub fn readb_high(&mut self) -> Result<u8, Error> {
self.write_all(&[mpsse::Command::MC_READB_HIGH.to_u8()])
.or_else(|_| Err(Error::IoError))?;
let mut result = [0; 1];
self.read(&mut result).or_else(|_| Err(Error::IoError))?;
Ok(result[0])
}
/// The purpose of this function is unclear. It appears to send some number
/// of bits out the line.
pub fn xfer_spi_bits(&mut self, data: u8, bits: usize) -> Result<u8, Error> {
if bits < 1 {
return Ok(0);
}
let buffer = &[
/* Input and output, update data on negative edge read on positive, bits. */
0x20 /*MC_DATA_IN*/ | 0x10 /*MC_DATA_OUT*/ | 0x01 /*MC_DATA_OCN*/ | 0x02, /*MC_DATA_BITS*/
bits as u8 - 1,
data,
];
self.write_all(buffer).or_else(|_| Err(Error::IoError))?;
let mut return_val = [0; 1];
self.read_exact(&mut return_val)
.or_else(|_| Err(Error::IoError))?;
Ok(return_val[0])
}
pub fn xfer_spi(&mut self, data: &mut [u8]) -> Result<(), Error> {
if data.is_empty() {
return Ok(());
}
/* Input and output, update data on negative edge read on positive. */
let buffer = &[
0x20 /*MC_DATA_IN*/ | 0x10 /*MC_DATA_OUT*/ | 0x01, /*MC_DATA_OCN*/
(data.len() - 1) as u8,
((data.len() - 1) / 256) as u8,
];
self.write_all(buffer).or_else(|_| Err(Error::IoError))?;
self.write_all(data).or_else(|_| Err(Error::IoError))?;
self.read_exact(data).or_else(|_| Err(Error::IoError))?;
Ok(())
}
pub fn send_spi(&mut self, data: &[u8]) -> Result<(), Error> {
if data.is_empty() {
return Ok(());
}
/* Input and output, update data on negative edge read on positive. */
let buffer = &[
0x10 /*MC_DATA_OUT*/ | 0x01, /*MC_DATA_OCN*/
(data.len() - 1) as u8,
((data.len() - 1) / 256) as u8,
];
self.write_all(buffer).or_else(|_| Err(Error::IoError))?;
self.write_all(data).or_else(|_| Err(Error::IoError))?;
Ok(())
}
}
impl Write for VCP {
@ -241,7 +343,13 @@ impl Write for VCP {
if result != Error::NoError {
Err(std::io::Error::new(std::io::ErrorKind::Other, result))
} else {
Ok(unsafe { bytes_written.assume_init() }
let bytes_written = unsafe { bytes_written.assume_init() };
// println!(
// "Wrote {} bytes (wanted to write {})",
// bytes_written,
// buf.len()
// );
Ok(bytes_written
.try_into()
.expect("invalid number of bytes written"))
}
@ -253,19 +361,23 @@ impl Write for VCP {
impl Read for VCP {
fn read(&mut self, buf: &mut [u8]) -> Result<usize, std::io::Error> {
let mut bytes_written = MaybeUninit::<DWORD>::uninit();
let mut bytes_read = MaybeUninit::<DWORD>::uninit();
let result = Error::from(unsafe {
FT_Write(
FT_Read(
self.handle,
buf.as_ptr() as LPVOID,
1,
bytes_written.as_mut_ptr() as LPDWORD,
buf.as_mut_ptr() as LPVOID,
buf.len()
.try_into()
.expect("couldn't convert buffer length to DWORD"),
bytes_read.as_mut_ptr() as LPDWORD,
)
});
if result != Error::NoError {
Err(std::io::Error::new(std::io::ErrorKind::Other, result))
} else {
Ok(unsafe { bytes_written.assume_init() }
let bytes_read = unsafe { bytes_read.assume_init() };
// println!("Read {} bytes (wanted to read {})", bytes_read, buf.len());
Ok(bytes_read
.try_into()
.expect("invalid number of bytes written"))
}

141
ftdi-vcp-rs/src/mpsse.rs Normal file
View File

@ -0,0 +1,141 @@
// /* Transfer Command bits */
// /* All byte based commands consist of:
// * - Command byte
// * - Length lsb
// * - Length msb
// *
// * If data out is enabled the data follows after the above command bytes,
// * otherwise no additional data is needed.
// * - Data * n
// *
// * All bit based commands consist of:
// * - Command byte
// * - Length
// *
// * If data out is enabled a byte containing bitst to transfer follows.
// * Otherwise no additional data is needed. Only up to 8 bits can be transferred
// * per transaction when in bit mode.
// */
// /* b 0000 0000
// * |||| |||`- Data out negative enable. Update DO on negative clock edge.
// * |||| ||`-- Bit count enable. When reset count represents bytes.
// * |||| |`--- Data in negative enable. Latch DI on negative clock edge.
// * |||| `---- LSB enable. When set clock data out LSB first.
// * ||||
// * |||`------ Data out enable
// * ||`------- Data in enable
// * |`-------- TMS mode enable
// * `--------- Special command mode enable. See mpsse_cmd enum.
// */
// pub enum CommandBits {
// #define MC_DATA_TMS (0x40) /* When set use TMS mode */
// #define MC_DATA_IN (0x20) /* When set read data (Data IN) */
// #define MC_DATA_OUT (0x10) /* When set write data (Data OUT) */
// #define MC_DATA_LSB (0x08) /* When set input/output data LSB first. */
// #define MC_DATA_ICN (0x04) /* When set receive data on negative clock edge */
// #define MC_DATA_BITS (0x02) /* When set count bits not bytes */
// #define MC_DATA_OCN (0x01) /* When set update data on negative clock edge */
// }
/* MPSSE engine command definitions */
#[allow(non_camel_case_types)]
pub enum Command {
// Mode commands
/// Set Data bits LowByte
MC_SETB_LOW,
/// Read Data bits LowByte
MC_READB_LOW,
/// Set Data bits HighByte
MC_SETB_HIGH,
/// Read data bits HighByte
MC_READB_HIGH,
/// Enable loopback
MC_LOOPBACK_EN,
/// Disable loopback
MC_LOOPBACK_DIS,
/// Set clock divisor
MC_SET_CLK_DIV,
/// Flush buffer fifos to the PC.
MC_FLUSH,
/// Wait on GPIOL1 to go high.
MC_WAIT_H,
/// Wait on GPIOL1 to go low.
MC_WAIT_L,
/// Disable /5 div, enables 60MHz master clock
MC_TCK_X5,
/// Enable /5 div, backward compat to FT2232D
MC_TCK_D5,
/// Enable 3 phase clk, DDR I2C
MC_EN_3PH_CLK,
/// Disable 3 phase clk
MC_DIS_3PH_CLK,
/// Clock every bit, used for JTAG
MC_CLK_N,
/// Clock every byte, used for JTAG
MC_CLK_N8,
/// Clock until GPIOL1 goes high
MC_CLK_TO_H,
/// Clock until GPIOL1 goes low
MC_CLK_TO_L,
/// Enable adaptive clocking
MC_EN_ADPT_CLK,
/// Disable adaptive clocking
MC_DIS_ADPT_CLK,
/// Clock until GPIOL1 goes high, count bytes
MC_CLK8_TO_H,
/// Clock until GPIOL1 goes low, count bytes
MC_CLK8_TO_L,
/// Set IO to only drive on 0 and tristate on 1
MC_TRI,
/// CPU mode commands
/// CPUMode read short address
MC_CPU_RS,
/// CPUMode read extended address
MC_CPU_RE,
/// CPUMode write short address
MC_CPU_WS,
/// CPUMode write extended address
MC_CPU_WE,
}
impl Command {
pub fn to_u8(&self) -> u8{
use Command::*;
match *self {
/* Mode commands */
MC_SETB_LOW => 0x80, /* Set Data bits LowByte */
MC_READB_LOW => 0x81, /* Read Data bits LowByte */
MC_SETB_HIGH => 0x82, /* Set Data bits HighByte */
MC_READB_HIGH => 0x83, /* Read data bits HighByte */
MC_LOOPBACK_EN => 0x84, /* Enable loopback */
MC_LOOPBACK_DIS => 0x85, /* Disable loopback */
MC_SET_CLK_DIV => 0x86, /* Set clock divisor */
MC_FLUSH => 0x87, /* Flush buffer fifos to the PC. */
MC_WAIT_H => 0x88, /* Wait on GPIOL1 to go high. */
MC_WAIT_L => 0x89, /* Wait on GPIOL1 to go low. */
MC_TCK_X5 => 0x8A, /* Disable /5 div, enables 60MHz master clock */
MC_TCK_D5 => 0x8B, /* Enable /5 div, backward compat to FT2232D */
MC_EN_3PH_CLK => 0x8C, /* Enable 3 phase clk, DDR I2C */
MC_DIS_3PH_CLK => 0x8D, /* Disable 3 phase clk */
MC_CLK_N => 0x8E, /* Clock every bit, used for JTAG */
MC_CLK_N8 => 0x8F, /* Clock every byte, used for JTAG */
MC_CLK_TO_H => 0x94, /* Clock until GPIOL1 goes high */
MC_CLK_TO_L => 0x95, /* Clock until GPIOL1 goes low */
MC_EN_ADPT_CLK => 0x96, /* Enable adaptive clocking */
MC_DIS_ADPT_CLK => 0x97, /* Disable adaptive clocking */
MC_CLK8_TO_H => 0x9C, /* Clock until GPIOL1 goes high, count bytes */
MC_CLK8_TO_L => 0x9D, /* Clock until GPIOL1 goes low, count bytes */
MC_TRI => 0x9E, /* Set IO to only drive on 0 and tristate on 1 */
/* CPU mode commands */
MC_CPU_RS => 0x90, /* CPUMode read short address */
MC_CPU_RE => 0x91, /* CPUMode read extended address */
MC_CPU_WS => 0x92, /* CPUMode write short address */
MC_CPU_WE => 0x93, /* CPUMode write extended address */
}
}
}