pci: allow to override sysfs path

[dpdk.git] / app / test / test_pci.c

diff --git a/app/test/test_pci.c b/app/test/test_pci.c
index 28d710b..362ae3e 100644 (file)
--- a/app/test/test_pci.c
+++ b/app/test/test_pci.c
@@ -43,6 +43,7 @@
 #include <rte_devargs.h>
 
 #include "test.h"
+#include "resource.h"
 
 /* Generic maximum number of drivers to have room to allocate all drivers */
 #define NUM_MAX_DRIVERS 256
@@ -144,37 +145,90 @@ static void free_devargs_list(void)
        }
 }
 
-/* backup real drivers (not used for testing) */
+/* backup real devices & drivers (not used for testing) */
 struct pci_driver_list real_pci_driver_list =
        TAILQ_HEAD_INITIALIZER(real_pci_driver_list);
+struct pci_device_list real_pci_device_list =
+       TAILQ_HEAD_INITIALIZER(real_pci_device_list);
+
+REGISTER_LINKED_RESOURCE(test_pci_sysfs);
 
 static int
 test_pci_setup(void)
 {
+       struct rte_pci_device *dev;
        struct rte_pci_driver *dr;
+       const struct resource *r;
+       int ret;
+
+       r = resource_find("test_pci_sysfs");
+       TEST_ASSERT_NOT_NULL(r, "missing resource test_pci_sysfs");
+
+       ret = resource_untar(r);
+       TEST_ASSERT_SUCCESS(ret, "failed to untar %s", r->name);
 
-       /* Unregister original driver list */
+       ret = setenv("SYSFS_PCI_DEVICES", "test_pci_sysfs/bus/pci/devices", 1);
+       TEST_ASSERT_SUCCESS(ret, "failed to setenv");
+
+       /* Unregister original devices & drivers lists */
        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);
        }
 
+       while (!TAILQ_EMPTY(&pci_device_list)) {
+               dev = TAILQ_FIRST(&pci_device_list);
+               TAILQ_REMOVE(&pci_device_list, dev, next);
+               TAILQ_INSERT_TAIL(&real_pci_device_list, dev, next);
+       }
+
+       ret = rte_eal_pci_scan();
+       TEST_ASSERT_SUCCESS(ret, "failed to scan PCI bus");
+       rte_eal_pci_dump(stdout);
+
        return 0;
 }
 
 static int
 test_pci_cleanup(void)
 {
+       struct rte_pci_device *dev;
        struct rte_pci_driver *dr;
+       const struct resource *r;
+       int ret;
+
+       unsetenv("SYSFS_PCI_DEVICES");
+
+       r = resource_find("test_pci_sysfs");
+       TEST_ASSERT_NOT_NULL(r, "missing resource test_pci_sysfs");
+
+       ret = resource_rm_by_tar(r);
+       TEST_ASSERT_SUCCESS(ret, "Failed to delete resource %s", r->name);
 
-       /* Restore original driver list */
+       /*
+        * FIXME: there is no API in DPDK to free a rte_pci_device so we
+        * cannot free the devices in the right way. Let's assume that we
+        * don't care for tests.
+        */
+       while (!TAILQ_EMPTY(&pci_device_list)) {
+               dev = TAILQ_FIRST(&pci_device_list);
+               TAILQ_REMOVE(&pci_device_list, dev, next);
+       }
+
+       /* Restore original devices & drivers lists */
        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);
        }
 
+       while (!TAILQ_EMPTY(&real_pci_device_list)) {
+               dev = TAILQ_FIRST(&real_pci_device_list);
+               TAILQ_REMOVE(&real_pci_device_list, dev, next);
+               TAILQ_INSERT_TAIL(&pci_device_list, dev, next);
+       }
+
        return 0;
 }
 
@@ -224,9 +278,37 @@ test_pci_blacklist(void)
        return 0;
 }
 
+static int test_pci_sysfs(void)
+{
+       const char *orig;
+       const char *newpath;
+       int ret;
+
+       orig = pci_get_sysfs_path();
+       ret = setenv("SYSFS_PCI_DEVICES", "My Documents", 1);
+       TEST_ASSERT_SUCCESS(ret, "Failed setenv to My Documents");
+
+       newpath = pci_get_sysfs_path();
+       TEST_ASSERT(!strcmp(newpath, "My Documents"),
+                       "pci_get_sysfs_path() should return 'My Documents' "
+                       "but gives %s", newpath);
+
+       ret = setenv("SYSFS_PCI_DEVICES", orig, 1);
+       TEST_ASSERT_SUCCESS(ret, "Failed setenv back to '%s'", orig);
+
+       newpath = pci_get_sysfs_path();
+       TEST_ASSERT(!strcmp(orig, newpath),
+                       "pci_get_sysfs_path returned unexpected path: "
+                       "%s (expected: %s)", newpath, orig);
+       return 0;
+}
+
 int
 test_pci(void)
 {
+       if (test_pci_sysfs())
+               return -1;
+
        if (test_pci_setup())
                return -1;