From d2b619957c42bd1a4e63f9e77586a4325b295de3 Mon Sep 17 00:00:00 2001
From: Sven Peter <sven@svenpeter.dev>
Date: Sat, 29 May 2021 14:35:00 +0200
Subject: [PATCH] tps6598x: add support for the USB PD chip

Signed-off-by: Sven Peter <sven@svenpeter.dev>
---
 Makefile       |  1 +
 src/tps6598x.c | 96 ++++++++++++++++++++++++++++++++++++++++++++++++++
 src/tps6598x.h | 18 ++++++++++
 3 files changed, 115 insertions(+)
 create mode 100644 src/tps6598x.c
 create mode 100644 src/tps6598x.h

diff --git a/Makefile b/Makefile
index 3fec0b18..118bb901 100644
--- a/Makefile
+++ b/Makefile
@@ -60,6 +60,7 @@ OBJECTS := \
 	startup.o \
 	string.o \
 	tunables.o \
+	tps6598x.o \
 	uart.o \
 	uartproxy.o \
 	usb.o usb_dwc3.o \
diff --git a/src/tps6598x.c b/src/tps6598x.c
new file mode 100644
index 00000000..a18b3668
--- /dev/null
+++ b/src/tps6598x.c
@@ -0,0 +1,96 @@
+/* SPDX-License-Identifier: MIT */
+
+#include "tps6598x.h"
+#include "adt.h"
+#include "i2c.h"
+#include "malloc.h"
+#include "types.h"
+#include "utils.h"
+
+#define TPS_REG_CMD1        0x08
+#define TPS_REG_DATA1       0x09
+#define TPS_REG_POWER_STATE 0x20
+#define TPS_CMD_INVALID     0x21434d44 // !CMD
+
+struct tps6598x_dev {
+    i2c_dev_t *i2c;
+    u8 addr;
+};
+
+tps6598x_dev_t *tps6598x_init(const char *adt_node, i2c_dev_t *i2c)
+{
+    int adt_offset;
+    adt_offset = adt_path_offset(adt, adt_node);
+    if (adt_offset < 0) {
+        printf("tps6598x: Error getting %s node\n", adt_node);
+        return NULL;
+    }
+
+    const u8 *iic_addr = adt_getprop(adt, adt_offset, "hpm-iic-addr", NULL);
+    if (iic_addr == NULL) {
+        printf("tps6598x: Error getting %s hpm-iic-addr\n.", adt_node);
+        return NULL;
+    }
+
+    tps6598x_dev_t *dev = malloc(sizeof(*dev));
+    if (!dev)
+        return NULL;
+
+    dev->i2c = i2c;
+    dev->addr = *iic_addr;
+    return dev;
+}
+
+void tps6598x_shutdown(tps6598x_dev_t *dev)
+{
+    free(dev);
+}
+
+int tps6598x_command(tps6598x_dev_t *dev, const char *cmd, const u8 *data_in, size_t len_in,
+                     u8 *data_out, size_t len_out)
+{
+    if (len_in) {
+        if (i2c_smbus_write(dev->i2c, dev->addr, TPS_REG_DATA1, data_in, len_in) < 0)
+            return -1;
+    }
+
+    if (i2c_smbus_write(dev->i2c, dev->addr, TPS_REG_CMD1, (const u8 *)cmd, 4) < 0)
+        return -1;
+
+    u32 cmd_status;
+    do {
+        if (i2c_smbus_read32(dev->i2c, dev->addr, TPS_REG_CMD1, &cmd_status))
+            return -1;
+        if (cmd_status == TPS_CMD_INVALID)
+            return -1;
+    } while (cmd_status != 0);
+
+    if (len_out) {
+        if (i2c_smbus_read(dev->i2c, dev->addr, TPS_REG_DATA1, data_out, len_out) != len_out)
+            return -1;
+    }
+
+    return 0;
+}
+
+int tps6598x_powerup(tps6598x_dev_t *dev)
+{
+    u8 power_state;
+
+    if (i2c_smbus_read8(dev->i2c, dev->addr, TPS_REG_POWER_STATE, &power_state))
+        return -1;
+
+    if (power_state == 0)
+        return 0;
+
+    const u8 data = 0;
+    tps6598x_command(dev, "SSPS", &data, 1, NULL, 0);
+
+    if (i2c_smbus_read8(dev->i2c, dev->addr, TPS_REG_POWER_STATE, &power_state))
+        return -1;
+
+    if (power_state != 0)
+        return -1;
+
+    return 0;
+}
diff --git a/src/tps6598x.h b/src/tps6598x.h
new file mode 100644
index 00000000..2b7efe4f
--- /dev/null
+++ b/src/tps6598x.h
@@ -0,0 +1,18 @@
+/* SPDX-License-Identifier: MIT */
+
+#ifndef TPS6598X_H
+#define TPS6598X_H
+
+#include "i2c.h"
+#include "types.h"
+
+typedef struct tps6598x_dev tps6598x_dev_t;
+
+tps6598x_dev_t *tps6598x_init(const char *adt_path, i2c_dev_t *i2c);
+void tps6598x_shutdown(tps6598x_dev_t *dev);
+
+int tps6598x_command(tps6598x_dev_t *dev, const char *cmd, const u8 *data_in, size_t len_in,
+                     u8 *data_out, size_t len_out);
+int tps6598x_powerup(tps6598x_dev_t *dev);
+
+#endif