drivers: i2c: xec: Handle additional I2C flags for read

I2C restart flag is now handled during I2C reads

Signed-off-by: Jose Alberto Meza <[email protected]>

drivers/i2c/i2c_mchp_xec.c

@@ -261,7 +261,6 @@ static int i2c_xec_poll_write(struct device *dev, struct i2c_msg msg,
 
 	/* Send bytes */
 	for (int i = 0U; i < msg.len; i++) {
-
 		MCHP_I2C_SMB_DATA(ba) = msg.buf[i];
 		ret = wait_completion(ba);
 		if (ret) {
@@ -283,6 +282,7 @@ static int i2c_xec_poll_write(struct device *dev, struct i2c_msg msg,
 			}
 		}
 	}
+
 	return 0;
 }
 
@@ -297,15 +297,17 @@ static int i2c_xec_poll_read(struct device *dev, struct i2c_msg msg,
 	u8_t byte, ctrl;
 	int ret;
 
-	/* Check clock and data lines */
-	if (check_lines(ba)) {
-		return -EBUSY;
-	}
+	if (!(msg.flags & I2C_MSG_RESTART)) {
+		/* Check clock and data lines */
+		if (check_lines(ba)) {
+			return -EBUSY;
+		}
 
-	/* Wait until bus is free */
-	ret = wait_bus_free(ba);
-	if (ret) {
-		return ret;
+		/* Wait until bus is free */
+		ret = wait_bus_free(ba);
+		if (ret) {
+			return ret;
+		}
 	}
 
 	/* Send slave address */
@@ -354,11 +356,6 @@ static int i2c_xec_poll_read(struct device *dev, struct i2c_msg msg,
 		msg.buf[i] = MCHP_I2C_SMB_DATA(ba);
 	}
 
-	/* Check clock and data lines */
-	if (check_lines(ba)) {
-		return -EBUSY;
-	}
-
 	return 0;
 }