fix(wire): ensure i2c bus is ready

Previously, i2c_master_write and i2c_master_read returned I2C_OK if
first HAL call returned HAL_BUSY which was not correct.
Now make sure the i2c is ready, which guarantees a good
initialization of the read or write sequence.

Fixes stm32duino#1774

Signed-off-by: Frederic Pillon <[email protected]>

Author: fpistm

Diff for: libraries/Wire/src/utility/twi.c

@@ -856,12 +856,20 @@ i2c_status_e i2c_master_write(i2c_t *obj, uint8_t dev_address,
 #if defined(I2C_OTHER_FRAME)
     uint32_t XferOptions = obj->handle.XferOptions; // save XferOptions value, because handle can be modified by HAL, which cause issue in case of NACK from slave
 #endif
-
 #if defined(I2C_OTHER_FRAME)
-    if (HAL_I2C_Master_Seq_Transmit_IT(&(obj->handle), dev_address, data, size, XferOptions) == HAL_OK) {
+    while (HAL_I2C_Master_Seq_Transmit_IT(&(obj->handle), dev_address, data, size, XferOptions) == HAL_BUSY) {
 #else
-    if (HAL_I2C_Master_Transmit_IT(&(obj->handle), dev_address, data, size) == HAL_OK) {
+    while (HAL_I2C_Master_Transmit_IT(&(obj->handle), dev_address, data, size) == HAL_BUSY) {
 #endif
+      // Ensure i2c ready
+      delta = (HAL_GetTick() - tickstart);
+      if (delta > I2C_TIMEOUT_TICK) {
+        ret = I2C_BUSY;
+        break;
+      }
+    }
+    if (ret == I2C_OK) {
+      tickstart = HAL_GetTick();
       // wait for transfer completion
       while ((HAL_I2C_GetState(&(obj->handle)) != HAL_I2C_STATE_READY) && (delta < I2C_TIMEOUT_TICK)) {
         delta = (HAL_GetTick() - tickstart);
@@ -932,10 +940,19 @@ i2c_status_e i2c_master_read(i2c_t *obj, uint8_t dev_address, uint8_t *data, uin
 #endif
 
 #if defined(I2C_OTHER_FRAME)
-  if (HAL_I2C_Master_Seq_Receive_IT(&(obj->handle), dev_address, data, size, XferOptions) == HAL_OK) {
+  while (HAL_I2C_Master_Seq_Receive_IT(&(obj->handle), dev_address, data, size, XferOptions) == HAL_BUSY) {
 #else
-  if (HAL_I2C_Master_Receive_IT(&(obj->handle), dev_address, data, size) == HAL_OK) {
+  while (HAL_I2C_Master_Receive_IT(&(obj->handle), dev_address, data, size) == HAL_BUSY) {
 #endif
+    // Ensure i2c ready
+    delta = (HAL_GetTick() - tickstart);
+    if (delta > I2C_TIMEOUT_TICK) {
+      ret = I2C_BUSY;
+      break;
+    }
+  }
+  if (ret == I2C_OK) {
+    tickstart = HAL_GetTick();
     // wait for transfer completion
     while ((HAL_I2C_GetState(&(obj->handle)) != HAL_I2C_STATE_READY) && (delta < I2C_TIMEOUT_TICK)) {
       delta = (HAL_GetTick() - tickstart);