From 1013faac3020f8489f2bb571fb4a71b3d9f66617 Mon Sep 17 00:00:00 2001
From: lipengfei28 <lipengfei28@xiaomi.com>
Date: Wed, 13 Aug 2025 11:18:43 +0800
Subject: [PATCH] pci: epf test default a function

Signed-off-by: lipengfei28 <lipengfei28@xiaomi.com>
---
 drivers/pci/Kconfig        | 7 +++++++
 drivers/pci/pci_epf_test.c | 7 +++----
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/drivers/pci/Kconfig b/drivers/pci/Kconfig
index eb632311bf..23c22e6474 100644
--- a/drivers/pci/Kconfig
+++ b/drivers/pci/Kconfig
@@ -100,5 +100,12 @@ config PCI_EPF_TEST
 	---help---
 		pci epf test driver
 
+config PCI_EPF_TEST_FUNC_COUNT
+	int "EPC Function Number"
+	default 1
+	range 1 8
+	---help---
+		This option set the epc's function number
+
 endif # PCI_ENDPOINT
 
diff --git a/drivers/pci/pci_epf_test.c b/drivers/pci/pci_epf_test.c
index 049959ccc5..fbde50c367 100644
--- a/drivers/pci/pci_epf_test.c
+++ b/drivers/pci/pci_epf_test.c
@@ -38,13 +38,12 @@
 #include <nuttx/pci/pci_ids.h>
 #include <nuttx/pci/pci.h>
 #include <nuttx/wqueue.h>
+#include <nuttx/nuttx.h>
 
 /****************************************************************************
  * Pre-processor Definitions
  ****************************************************************************/
 
-#define PCI_EPF_TEST_FUNCTIONS                 2
-
 #define PCI_EPF_TEST_IRQ_TYPE_LEGACY           0
 #define PCI_EPF_TEST_IRQ_TYPE_MSI              1
 #define PCI_EPF_TEST_IRQ_TYPE_MSIX             2
@@ -716,14 +715,14 @@ int pci_register_epf_test_device(FAR const char *epc_name)
   int ret;
   int i;
 
-  epf = kmm_zalloc(sizeof(*epf) * PCI_EPF_TEST_FUNCTIONS);
+  epf = kmm_zalloc(sizeof(*epf) * CONFIG_PCI_EPF_TEST_FUNC_COUNT);
   if (NULL == epf)
     {
       pcierr("create epf error\n");
       return -ENOMEM;
     }
 
-  for (i = 0; i < PCI_EPF_TEST_FUNCTIONS; i++)
+  for (i = 0; i < CONFIG_PCI_EPF_TEST_FUNC_COUNT; i++)
     {
       epf[i].name = g_pci_epf_test_name[i];
       epf[i].epc_name = epc_name;