usb_port_power: Allow variable number of ports

Currently, usb_port_power_dumb allows fixed number of ports (2)
and usb_port_power_smart allows less than 4 ports. This change
removes the limitation and makes them accept variable number of
ports.

BUG=chromium:715178
BRANCH=none
TEST=Verified usbchargemode returns consistent charge_mode across
sysjumps on Electro.

Change-Id: I7675c65ba03d3e6f7f53b2fe48cbe510231d945a
Signed-off-by: Daisuke Nojiri <dnojiri@chromium.org>
Reviewed-on: https://chromium-review.googlesource.com/481077
Reviewed-by: Randall Spangler <rspangler@chromium.org>

1 files changed, 14 insertions, 31 deletions

diff --git a/common/usb_port_power_smart.c b/common/usb_port_power_smart.c
index 86836df96d..fb9e1e1230 100644
--- a/common/usb_port_power_smart.c
+++ b/common/usb_port_power_smart.c
@@ -25,13 +25,11 @@
 #define CONFIG_USB_PORT_POWER_SMART_DEFAULT_MODE USB_CHARGE_MODE_SDP2
 #endif
 
-/* The previous USB port state before sys jump */
-struct usb_state {
-	uint8_t port_mode[CONFIG_USB_PORT_POWER_SMART_PORT_COUNT];
-};
-
 static uint8_t charge_mode[CONFIG_USB_PORT_POWER_SMART_PORT_COUNT];
 
+/* GPIOs to enable/disable USB ports. Board specific. */
+extern const int usb_port_enable[CONFIG_USB_PORT_POWER_SMART_PORT_COUNT];
+
 static void usb_charge_set_control_mode(int port_id, int mode)
 {
 #ifdef CONFIG_USB_PORT_POWER_SMART_SIMPLE
@@ -57,18 +55,7 @@ static void usb_charge_set_control_mode(int port_id, int mode)
 static void usb_charge_set_enabled(int port_id, int en)
 {
 	ASSERT(port_id < CONFIG_USB_PORT_POWER_SMART_PORT_COUNT);
-#if CONFIG_USB_PORT_POWER_SMART_PORT_COUNT >= 1
-	if (port_id == 0)
-		gpio_set_level(GPIO_USB1_ENABLE, en);
-#endif
-#if CONFIG_USB_PORT_POWER_SMART_PORT_COUNT >= 2
-	if (port_id == 1)
-		gpio_set_level(GPIO_USB2_ENABLE, en);
-#endif
-#if CONFIG_USB_PORT_POWER_SMART_PORT_COUNT >= 3
-	if (port_id == 2)
-		gpio_set_level(GPIO_USB3_ENABLE, en);
-#endif
+	gpio_set_level(usb_port_enable[port_id], en);
 }
 
 static void usb_charge_set_ilim(int port_id, int sel)
@@ -195,31 +182,27 @@ DECLARE_HOST_COMMAND(EC_CMD_USB_CHARGE_SET_MODE,
 
 static void usb_charge_preserve_state(void)
 {
-	struct usb_state state;
-	int i;
-
-	for (i = 0; i < CONFIG_USB_PORT_POWER_SMART_PORT_COUNT; i++)
-		state.port_mode[i] = charge_mode[i];
-
 	system_add_jump_tag(USB_SYSJUMP_TAG, USB_HOOK_VERSION,
-			    sizeof(state), &state);
+			    sizeof(charge_mode), charge_mode);
 }
 DECLARE_HOOK(HOOK_SYSJUMP, usb_charge_preserve_state, HOOK_PRIO_DEFAULT);
 
 static void usb_charge_init(void)
 {
-	const struct usb_state *prev;
+	const uint8_t *prev;
 	int version, size, i;
 
-	prev = (const struct usb_state *)system_get_jump_tag(USB_SYSJUMP_TAG,
-							     &version, &size);
+	prev = (const uint8_t *)system_get_jump_tag(USB_SYSJUMP_TAG,
+						    &version, &size);
 
-	if (prev && version == USB_HOOK_VERSION && size == sizeof(*prev)) {
-		for (i = 0; i < CONFIG_USB_PORT_POWER_SMART_PORT_COUNT; i++)
-			usb_charge_set_mode(i, prev->port_mode[i]);
-	} else {
+	if (!prev || version != USB_HOOK_VERSION ||
+			size != sizeof(charge_mode)) {
 		usb_charge_all_ports_ctrl(USB_CHARGE_MODE_DISABLED);
+		return;
 	}
+
+	for (i = 0; i < CONFIG_USB_PORT_POWER_SMART_PORT_COUNT; i++)
+		usb_charge_set_mode(i, prev[i]);
 }
 DECLARE_HOOK(HOOK_INIT, usb_charge_init, HOOK_PRIO_DEFAULT);