[dpdk-dev] [PATCH v4 07/10] app/test: extract test_pci_setup and test_pci_cleanup

[Jan Viktorin]

Signed-off-by: Jan Viktorin <viktorin at rehivetech.com>
---
 app/test/test_pci.c | 47 ++++++++++++++++++++++++++++++++++-------------
 1 file changed, 34 insertions(+), 13 deletions(-)

diff --git a/app/test/test_pci.c b/app/test/test_pci.c
index 8b7c8bb..50078a0 100644
--- a/app/test/test_pci.c
+++ b/app/test/test_pci.c
@@ -148,21 +148,46 @@ static void free_devargs_list(void)
 struct pci_driver_list real_pci_driver_list =
        TAILQ_HEAD_INITIALIZER(real_pci_driver_list);

+static int
+test_pci_setup(void)
+{
+       struct rte_pci_driver *dr;
+
+       /* Unregister original driver list */
+       while (!TAILQ_EMPTY(&pci_driver_list)) {
+               dr = TAILQ_FIRST(&pci_driver_list);
+               rte_eal_pci_unregister(dr);
+               TAILQ_INSERT_TAIL(&real_pci_driver_list, dr, next);
+       }
+
+       return 0;
+}
+
+static int
+test_pci_cleanup(void)
+{
+       struct rte_pci_driver *dr;
+
+       /* Restore original driver list */
+       while (!TAILQ_EMPTY(&real_pci_driver_list)) {
+               dr = TAILQ_FIRST(&real_pci_driver_list);
+               TAILQ_REMOVE(&real_pci_driver_list, dr, next);
+               rte_eal_pci_register(dr);
+       }
+
+       return 0;
+}
+
 int
 test_pci(void)
 {
        struct rte_devargs_list save_devargs_list;
-       struct rte_pci_driver *dr = NULL;

        printf("Dump all devices\n");
        rte_eal_pci_dump(stdout);

-       /* Unregister all previous drivers */
-       while (!TAILQ_EMPTY(&pci_driver_list)) {
-               dr = TAILQ_FIRST(&pci_driver_list);
-               rte_eal_pci_unregister(dr);
-               TAILQ_INSERT_TAIL(&real_pci_driver_list, dr, next);
-       }
+       if (test_pci_setup())
+               return -1;

        rte_eal_pci_register(&my_driver);
        rte_eal_pci_register(&my_driver2);
@@ -199,12 +224,8 @@ test_pci(void)
        rte_eal_pci_unregister(&my_driver);
        rte_eal_pci_unregister(&my_driver2);

-       /* Restore original driver list */
-       while (!TAILQ_EMPTY(&real_pci_driver_list)) {
-               dr = TAILQ_FIRST(&real_pci_driver_list);
-               TAILQ_REMOVE(&real_pci_driver_list, dr, next);
-               rte_eal_pci_register(dr);
-       }
+       if (test_pci_cleanup())
+               return -1;

        return 0;
 }
-- 
2.8.0