This commit reworks the unregister API to accept an integer.

Signed-off-by: Harry van Haaren <harry.van.haa...@intel.com>
---
 .../common/include/rte_service_component.h         |  2 +-
 lib/librte_eal/common/rte_service.c                | 25 ++++++----------------
 test/test/test_service_cores.c                     | 10 +++------
 3 files changed, 10 insertions(+), 27 deletions(-)

diff --git a/lib/librte_eal/common/include/rte_service_component.h 
b/lib/librte_eal/common/include/rte_service_component.h
index 5ba5cdf..70cca69 100644
--- a/lib/librte_eal/common/include/rte_service_component.h
+++ b/lib/librte_eal/common/include/rte_service_component.h
@@ -111,7 +111,7 @@ int32_t rte_service_register(const struct rte_service_spec 
*spec,
  * @retval -EBUSY The service is currently running, stop the service before
  *          calling unregister. No action has been taken.
  */
-int32_t rte_service_unregister(struct rte_service_spec *service);
+int32_t rte_service_unregister(uint32_t id);
 
 /**
  * @warning
diff --git a/lib/librte_eal/common/rte_service.c 
b/lib/librte_eal/common/rte_service.c
index 308ec6f..7299514 100644
--- a/lib/librte_eal/common/rte_service.c
+++ b/lib/librte_eal/common/rte_service.c
@@ -259,35 +259,22 @@ rte_service_register(const struct rte_service_spec *spec, 
uint32_t *id_ptr)
 }
 
 int32_t
-rte_service_unregister(struct rte_service_spec *spec)
+rte_service_unregister(uint32_t id)
 {
-       struct rte_service_spec_impl *s = NULL;
-       struct rte_service_spec_impl *spec_impl =
-               (struct rte_service_spec_impl *)spec;
-
        uint32_t i;
-       uint32_t service_id;
-       for (i = 0; i < RTE_SERVICE_NUM_MAX; i++) {
-               if (&rte_services[i] == spec_impl) {
-                       s = spec_impl;
-                       service_id = i;
-                       break;
-               }
-       }
-
-       if (!s)
-               return -EINVAL;
+       struct rte_service_spec_impl *s;
+       SERVICE_VALID_GET_OR_ERR_RET(id, s, -EINVAL);
 
        rte_service_count--;
        rte_smp_wmb();
 
        s->internal_flags &= ~(SERVICE_F_REGISTERED);
 
+       /* clear the run-bit in all cores */
        for (i = 0; i < RTE_MAX_LCORE; i++)
-               lcore_states[i].service_mask &= ~(UINT64_C(1) << service_id);
+               lcore_states[i].service_mask &= ~(UINT64_C(1) << id);
 
-       memset(&rte_services[service_id], 0,
-                       sizeof(struct rte_service_spec_impl));
+       memset(&rte_services[id], 0, sizeof(struct rte_service_spec_impl));
 
        return 0;
 }
diff --git a/test/test/test_service_cores.c b/test/test/test_service_cores.c
index 4ece080..43ad027 100644
--- a/test/test/test_service_cores.c
+++ b/test/test/test_service_cores.c
@@ -131,17 +131,13 @@ static int
 unregister_all(void)
 {
        uint32_t i;
-       struct rte_service_spec *dead = (struct rte_service_spec *)0xdead;
 
-       TEST_ASSERT_EQUAL(-EINVAL, rte_service_unregister(0),
-                       "Unregistered NULL pointer");
-       TEST_ASSERT_EQUAL(-EINVAL, rte_service_unregister(dead),
-                       "Unregistered invalid pointer");
+       TEST_ASSERT_EQUAL(-EINVAL, rte_service_unregister(1000),
+                       "Unregistered invalid service id");
 
        uint32_t c = rte_service_get_count();
        for (i = 0; i < c; i++) {
-               struct rte_service_spec *s = rte_service_get_by_id(i);
-               TEST_ASSERT_EQUAL(0, rte_service_unregister(s),
+               TEST_ASSERT_EQUAL(0, rte_service_unregister(i),
                                "Error unregistering a valid service");
        }
 
-- 
2.7.4

Reply via email to