ccgx: Add function to write to ccgx

Pass writes through via I2C passthrough on the EC

Signed-off-by: Daniel Schaefer <dhs@frame.work>

Author: JohnAZoidberg

framework_lib/src/ccgx/device.rs

@@ -182,6 +182,21 @@ impl PdController {
         )
     }
 
+    pub fn i2c_write(&self, addr: u16, data: &[u8]) -> EcResult<EcI2cPassthruResponse> {
+        trace!(
+            "I2C passthrough from I2C Port {} to I2C Addr {}",
+            self.port.i2c_port()?,
+            self.port.i2c_address()?
+        );
+        i2c_write(
+            &self.ec,
+            self.port.i2c_port()?,
+            self.port.i2c_address()?,
+            addr,
+            data,
+        )
+    }
+
     fn ccgx_read(&self, reg: ControlRegisters, len: u16) -> EcResult<Vec<u8>> {
         let mut data: Vec<u8> = Vec::with_capacity(len.into());
 
@@ -204,6 +219,35 @@ impl PdController {
         Ok(data)
     }
 
+    fn ccgx_write(&self, reg: ControlRegisters, data: &[u8]) -> EcResult<()> {
+        let addr = reg as u16;
+        trace!(
+            "ccgx_write(reg: {:?}, addr: {}, data.len(): {}",
+            reg,
+            addr,
+            data.len()
+        );
+        let mut data_written = 0;
+
+        while data_written < data.len() {
+            let chunk_len = std::cmp::min(MAX_I2C_CHUNK, data.len());
+            let buffer = &data[data_written..data_written + chunk_len];
+            let offset = addr + data_written as u16;
+
+            let i2c_response = self.i2c_write(offset, buffer)?;
+            if let Err(EcError::DeviceError(err)) = i2c_response.is_successful() {
+                return Err(EcError::DeviceError(format!(
+                    "I2C write was not successful: {:?}",
+                    err
+                )));
+            }
+
+            data_written += chunk_len;
+        }
+
+        Ok(())
+    }
+
     pub fn get_silicon_id(&self) -> EcResult<u16> {
         let data = self.ccgx_read(ControlRegisters::SiliconId, 2)?;
         assert_win_len(data.len(), 2);

framework_lib/src/chromium_ec/i2c_passthrough.rs

@@ -155,7 +155,7 @@ pub fn i2c_write(
 
     let data = ec.send_command(EcCommands::I2cPassthrough as u16, 0, &buffer)?;
     let res: _EcI2cPassthruResponse = unsafe { std::ptr::read(data.as_ptr() as *const _) };
-    assert_eq!(data.len(), size_of::<_EcI2cPassthruResponse>()); // No extra data other than the header
+    util::assert_win_len(data.len(), size_of::<_EcI2cPassthruResponse>()); // No extra data other than the header
     debug_assert_eq!(res.messages as usize, messages.len());
     Ok(EcI2cPassthruResponse {
         i2c_status: res.i2c_status,