1 files changed, 33 insertions, 19 deletions

diff --git a/src/soc/amd/cezanne/uart.c b/src/soc/amd/cezanne/uart.c
index 64252cad2b..ba10945bf8 100644
--- a/src/soc/amd/cezanne/uart.c
+++ b/src/soc/amd/cezanne/uart.c
@@ -15,16 +15,18 @@
 
 static const struct {
 	uintptr_t base;
+	unsigned int aoac_device;
+	const char *acpi_name;
 	struct soc_amd_gpio mux[2];
 } uart_info[] = {
-	[0] = { APU_UART0_BASE, {
+	[0] =	{ APU_UART0_BASE, FCH_AOAC_DEV_UART0, "FUR0", {
 			PAD_NF(GPIO_143, UART0_TXD, PULL_NONE),
 			PAD_NF(GPIO_141, UART0_RXD, PULL_NONE),
-	} },
-	[1] = { APU_UART1_BASE, {
+		} },
+	[1] =	{ APU_UART1_BASE, FCH_AOAC_DEV_UART1, "FUR1", {
 			PAD_NF(GPIO_140, UART1_TXD, PULL_NONE),
 			PAD_NF(GPIO_142, UART1_RXD, PULL_NONE),
-	} },
+		} },
 };
 
 uintptr_t get_uart_base(unsigned int idx)
@@ -35,6 +37,28 @@ uintptr_t get_uart_base(unsigned int idx)
 	return uart_info[idx].base;
 }
 
+static enum cb_err get_uart_idx(uintptr_t base, unsigned int *idx)
+{
+	unsigned int i;
+	for (i = 0; i < ARRAY_SIZE(uart_info); i++) {
+		if (base == uart_info[i].base) {
+			*idx = i;
+			return CB_SUCCESS;
+		}
+	}
+	return CB_ERR;
+}
+
+static enum cb_err get_uart_aoac_device(uintptr_t base, unsigned int *aoac_dev)
+{
+	unsigned int idx;
+	if (get_uart_idx(base, &idx) == CB_ERR)
+		return CB_ERR;
+
+	*aoac_dev = uart_info[idx].aoac_device;
+	return CB_SUCCESS;
+}
+
 void clear_uart_legacy_config(void)
 {
 	write16p(FCH_LEGACY_UART_DECODE, 0);
@@ -50,14 +74,11 @@ void set_uart_config(unsigned int idx)
 
 static const char *uart_acpi_name(const struct device *dev)
 {
-	switch (dev->path.mmio.addr) {
-	case APU_UART0_BASE:
-		return "FUR0";
-	case APU_UART1_BASE:
-		return "FUR1";
-	default:
+	unsigned int idx;
+	if (get_uart_idx(dev->path.mmio.addr, &idx) == CB_SUCCESS)
+		return uart_info[idx].acpi_name;
+	else
 		return NULL;
-	}
 }
 
 /* Even though this is called enable, it gets called for both enabled and disabled devices. */
@@ -65,14 +86,7 @@ static void uart_enable(struct device *dev)
 {
 	unsigned int dev_id;
 
-	switch (dev->path.mmio.addr) {
-	case APU_UART0_BASE:
-		dev_id = FCH_AOAC_DEV_UART0;
-		break;
-	case APU_UART1_BASE:
-		dev_id = FCH_AOAC_DEV_UART1;
-		break;
-	default:
+	if (get_uart_aoac_device(dev->path.mmio.addr, &dev_id) == CB_ERR) {
 		printk(BIOS_ERR, "%s: Unknown device: %s\n", __func__, dev_path(dev));
 		return;
 	}