From 6bb16f3e54bccb1f44c7979d58248d098cde6256 Mon Sep 17 00:00:00 2001
From: mssonicbld <79238446+mssonicbld@users.noreply.github.com>
Date: Thu, 22 Jun 2023 08:27:53 +0800
Subject: [PATCH] [celestica/e1031]: enable emc2305 fan controller timeout
 feature (#14401) (#15571)

---
 .../haliburton/modules/emc2305.c              | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/platform/broadcom/sonic-platform-modules-cel/haliburton/modules/emc2305.c b/platform/broadcom/sonic-platform-modules-cel/haliburton/modules/emc2305.c
index f08033e080ce..c32f5dffbc2d 100644
--- a/platform/broadcom/sonic-platform-modules-cel/haliburton/modules/emc2305.c
+++ b/platform/broadcom/sonic-platform-modules-cel/haliburton/modules/emc2305.c
@@ -407,6 +407,35 @@ emc2305_set_pwm(struct i2c_client *client, int fan_idx, long pwm)
 	mutex_unlock(&data->update_lock);
 	return status;
 }
+
+static int
+emc2305_enable_timeout(struct i2c_client *client, bool enable)
+{
+	struct emc2305_data *data = i2c_get_clientdata(client);
+	int status = 0;
+	u8 conf_val = 0;
+
+	mutex_lock(&data->update_lock);
+
+	status = read_u8_from_i2c(client, REG_CONFIGURATION, &conf_val);
+	if (status < 0) {
+	    mutex_unlock(&data->update_lock);
+	    return status;
+	}
+
+	// Section 6.2: CONFIG REGISTER DIS_TO bit(bit 6)
+	if (enable) {
+	    conf_val &= ~(1 << 6);
+	} else {
+	    conf_val |= (1 << 6);
+	}
+
+	status = i2c_smbus_write_byte_data(client, REG_CONFIGURATION, conf_val);
+
+	mutex_unlock(&data->update_lock);
+	return status;
+}
+
 /*
  * sysfs callback functions
  *
@@ -744,6 +773,9 @@ emc2305_probe(struct i2c_client *client, const struct i2c_device_id *id)
 	i2c_set_clientdata(client, data);
 	mutex_init(&data->update_lock);
 
+	// Enable SMBus timeout feature
+	emc2305_enable_timeout(client, true);
+
 	status = i2c_smbus_read_byte_data(client, REG_PRODUCT_ID);
 	switch (status) {
 	case 0x34: /* EMC2305 */