Add two new APIs that allow set and get of a name for an lcore that do not explose platform thread details via the public interface.
If the underlying platform thread API does allow set of a name for the lcore thread id it is called as a side-effect of set but lack of support or failure from the platform-specific API does will not cause the call to fail allowing for "debugging" on platforms that happen to support the feature without exposing it as a part of the EAL. Signed-off-by: Tyler Retzlaff <roret...@linux.microsoft.com> --- lib/eal/common/eal_common_lcore.c | 32 +++++++++++++++++++++++++++ lib/eal/common/eal_common_thread.c | 15 ++++++------- lib/eal/common/eal_common_trace.c | 2 +- lib/eal/freebsd/eal.c | 5 ++++- lib/eal/freebsd/eal_thread.c | 6 ++++++ lib/eal/include/rte_lcore.h | 44 ++++++++++++++++++++++++++++++++++++++ lib/eal/include/rte_thread.h | 8 +++++++ lib/eal/linux/eal.c | 5 ++--- lib/eal/linux/eal_thread.c | 19 ++++++++++++++++ lib/eal/version.map | 5 +++++ lib/eal/windows/rte_thread.c | 41 +++++++++++++++++++++++++++++++++++ 11 files changed, 169 insertions(+), 13 deletions(-) diff --git a/lib/eal/common/eal_common_lcore.c b/lib/eal/common/eal_common_lcore.c index 06c594b..700362a 100644 --- a/lib/eal/common/eal_common_lcore.c +++ b/lib/eal/common/eal_common_lcore.c @@ -14,6 +14,38 @@ #include "eal_private.h" #include "eal_thread.h" +static char lcore_names[RTE_MAX_LCORE][RTE_LCORE_NAME_MAX_LEN]; + +int +rte_lcore_set_name(unsigned int lcore_id, const char *name) +{ + if (unlikely(lcore_id >= RTE_MAX_LCORE)) + return -EINVAL; + + if (strlen(name) >= RTE_LCORE_NAME_MAX_LEN) + return -ERANGE; + + (void)strcpy(&lcore_names[lcore_id][0], name); + + rte_thread_set_name((rte_thread_t){lcore_config[lcore_id].thread_id}, name); + + return 0; +} + +int +rte_lcore_get_name(unsigned int lcore_id, char *name, size_t len) +{ + if (unlikely(lcore_id >= RTE_MAX_LCORE)) + return -EINVAL; + + if (len < RTE_LCORE_NAME_MAX_LEN) + return -EINVAL; + + (void)strcpy(name, &lcore_names[lcore_id][0]); + + return 0; +} + unsigned int rte_get_main_lcore(void) { return rte_eal_get_configuration()->main_lcore; diff --git a/lib/eal/common/eal_common_thread.c b/lib/eal/common/eal_common_thread.c index c5d8b43..91066e6 100644 --- a/lib/eal/common/eal_common_thread.c +++ b/lib/eal/common/eal_common_thread.c @@ -272,6 +272,7 @@ static void *ctrl_thread_init(void *arg) const pthread_attr_t *attr, void *(*start_routine)(void *), void *arg) { + rte_thread_t id; struct rte_thread_ctrl_params *params; enum __rte_ctrl_thread_status ctrl_thread_status; int ret; @@ -285,18 +286,16 @@ static void *ctrl_thread_init(void *arg) params->ret = 0; params->ctrl_thread_status = CTRL_THREAD_LAUNCHING; - ret = pthread_create(thread, attr, ctrl_thread_init, (void *)params); + ret = pthread_create((pthread_t *)&id.opaque_id, attr, ctrl_thread_init, params); if (ret != 0) { free(params); return -ret; } - if (name != NULL) { - ret = rte_thread_setname(*thread, name); - if (ret < 0) - RTE_LOG(DEBUG, EAL, - "Cannot set name for ctrl thread\n"); - } + if (name != NULL) + rte_thread_set_name(id, name); + + *thread = (pthread_t)id.opaque_id; /* Wait for the control thread to initialize successfully */ while ((ctrl_thread_status = @@ -312,7 +311,7 @@ static void *ctrl_thread_init(void *arg) /* Check if the control thread encountered an error */ if (ctrl_thread_status == CTRL_THREAD_ERROR) { /* ctrl thread is exiting */ - pthread_join(*thread, NULL); + (void)rte_thread_join(id, NULL); } ret = params->ret; diff --git a/lib/eal/common/eal_common_trace.c b/lib/eal/common/eal_common_trace.c index 5caaac8..03c1055 100644 --- a/lib/eal/common/eal_common_trace.c +++ b/lib/eal/common/eal_common_trace.c @@ -356,7 +356,7 @@ rte_trace_mode rte_trace_mode_get(void) /* Store the thread name */ char *name = header->stream_header.thread_name; memset(name, 0, __RTE_TRACE_EMIT_STRING_LEN_MAX); - rte_thread_getname(pthread_self(), name, + (void)rte_lcore_get_name(rte_lcore_id(), name, __RTE_TRACE_EMIT_STRING_LEN_MAX); trace->lcore_meta[count].mem = header; diff --git a/lib/eal/freebsd/eal.c b/lib/eal/freebsd/eal.c index 607684c..e6a76de 100644 --- a/lib/eal/freebsd/eal.c +++ b/lib/eal/freebsd/eal.c @@ -817,7 +817,10 @@ static void rte_eal_init_alert(const char *msg) /* Set thread_name for aid in debugging. */ snprintf(thread_name, sizeof(thread_name), "rte-worker-%d", i); - rte_thread_setname(lcore_config[i].thread_id, thread_name); + ret = rte_lcore_set_name(i, thread_name); + if (ret != 0) + RTE_LOG(DEBUG, EAL, + "Cannot set name for lcore\n"); ret = pthread_setaffinity_np(lcore_config[i].thread_id, sizeof(rte_cpuset_t), &lcore_config[i].cpuset); diff --git a/lib/eal/freebsd/eal_thread.c b/lib/eal/freebsd/eal_thread.c index ab81b52..af85c48 100644 --- a/lib/eal/freebsd/eal_thread.c +++ b/lib/eal/freebsd/eal_thread.c @@ -32,6 +32,12 @@ int rte_sys_gettid(void) return (int)lwpid; } +void rte_thread_set_name(rte_thread_t id, const char *name) +{ + /* this BSD function returns no error */ + pthread_set_name_np(id.opaque_id, name); +} + int rte_thread_setname(pthread_t id, const char *name) { /* this BSD function returns no error */ diff --git a/lib/eal/include/rte_lcore.h b/lib/eal/include/rte_lcore.h index 6938c3f..251208f 100644 --- a/lib/eal/include/rte_lcore.h +++ b/lib/eal/include/rte_lcore.h @@ -13,6 +13,7 @@ */ #include <stdio.h> +#include <rte_common.h> #include <rte_compat.h> #include <rte_config.h> #include <rte_per_lcore.h> @@ -25,6 +26,7 @@ #endif #define LCORE_ID_ANY UINT32_MAX /**< Any lcore. */ +#define RTE_LCORE_NAME_MAX_LEN RTE_MAX_THREAD_NAME_LEN RTE_DECLARE_PER_LCORE(unsigned, _lcore_id); /**< Per thread "lcore id". */ @@ -90,6 +92,48 @@ enum rte_lcore_role_t { unsigned int rte_get_main_lcore(void); /** + * @warning + * @b EXPERIMENTAL: this API may change without prior notice. + * + * Sets the name of an lcore and (if supported) implicitly sets the + * debugging name of the lcores thread. + * + * @param lcore_id + * The targeted lcore. + * + * @param name + * The name to set for the specified lcore_id. strlen(name) shall be + * no more than RTE_LCORE_NAME_MAX_LEN - 1 long. + * + * @return + * 0 on success or, -errno on failure. + */ +__rte_experimental +int rte_lcore_set_name(unsigned int lcore_id, const char *name); + +/** + * @warning + * @b EXPERIMENTAL: this API may change without prior notice. + * + * Gets the name of an lcore thread. + * + * @param lcore_id + * The targeted lcore. + * + * @param name + * The buffer used to receive the lcore name. + * + * @param len + * The length of the buffer name. The buffer len shall be at + * least RTE_LCORE_NAME_MAX_LEN. + * + * @return + * 0 on success or, -errno on failure. + */ +__rte_experimental +int rte_lcore_get_name(unsigned int lcore_id, char *name, size_t len); + +/** * Return the number of execution units (lcores) on the system. * * @return diff --git a/lib/eal/include/rte_thread.h b/lib/eal/include/rte_thread.h index b9edf70..713330c 100644 --- a/lib/eal/include/rte_thread.h +++ b/lib/eal/include/rte_thread.h @@ -143,6 +143,14 @@ int rte_thread_create(rte_thread_t *thread_id, rte_thread_t rte_thread_self(void); /** + * @internal + * Set the name of the thread. + */ +__rte_internal +void +rte_thread_set_name(rte_thread_t id, const char *name); + +/** * @warning * @b EXPERIMENTAL: this API may change without prior notice. * diff --git a/lib/eal/linux/eal.c b/lib/eal/linux/eal.c index 8c118d0..97e2fec 100644 --- a/lib/eal/linux/eal.c +++ b/lib/eal/linux/eal.c @@ -1255,11 +1255,10 @@ static void rte_eal_init_alert(const char *msg) /* Set thread_name for aid in debugging. */ snprintf(thread_name, sizeof(thread_name), "rte-worker-%d", i); - ret = rte_thread_setname(lcore_config[i].thread_id, - thread_name); + ret = rte_lcore_set_name(i, thread_name); if (ret != 0) RTE_LOG(DEBUG, EAL, - "Cannot set name for lcore thread\n"); + "Cannot set name for lcore\n"); ret = pthread_setaffinity_np(lcore_config[i].thread_id, sizeof(rte_cpuset_t), &lcore_config[i].cpuset); diff --git a/lib/eal/linux/eal_thread.c b/lib/eal/linux/eal_thread.c index 625bde6..2413bc6 100644 --- a/lib/eal/linux/eal_thread.c +++ b/lib/eal/linux/eal_thread.c @@ -10,6 +10,7 @@ #include <rte_eal.h> #include <rte_lcore.h> +#include <rte_log.h> #include <rte_string_fns.h> /* require calling thread tid by gettid() */ @@ -18,6 +19,24 @@ int rte_sys_gettid(void) return (int)syscall(SYS_gettid); } +void rte_thread_set_name(rte_thread_t id, const char *name) +{ + int ret = ENOSYS; +#if defined(__GLIBC__) && defined(__GLIBC_PREREQ) +#if __GLIBC_PREREQ(2, 12) + char truncated[16]; + + strlcpy(truncated, name, sizeof(truncated)); + ret = pthread_setname_np(id.opaque_id, truncated); +#endif +#endif + RTE_SET_USED(id); + RTE_SET_USED(name); + + if (ret != 0) + RTE_LOG(DEBUG, EAL, "Cannot set thread name\n"); +} + int rte_thread_setname(pthread_t id, const char *name) { int ret = ENOSYS; diff --git a/lib/eal/version.map b/lib/eal/version.map index 7ad12a7..4110f75 100644 --- a/lib/eal/version.map +++ b/lib/eal/version.map @@ -440,6 +440,10 @@ EXPERIMENTAL { rte_thread_detach; rte_thread_equal; rte_thread_join; + + # added in 23.03 + rte_lcore_get_name; + rte_lcore_set_name; }; INTERNAL { @@ -483,4 +487,5 @@ INTERNAL { rte_mem_map; rte_mem_page_size; rte_mem_unmap; + rte_thread_set_name; }; diff --git a/lib/eal/windows/rte_thread.c b/lib/eal/windows/rte_thread.c index 1c1e9d0..51028ef 100644 --- a/lib/eal/windows/rte_thread.c +++ b/lib/eal/windows/rte_thread.c @@ -4,7 +4,9 @@ */ #include <errno.h> +#include <wchar.h> +#include <rte_eal.h> #include <rte_common.h> #include <rte_errno.h> #include <rte_thread.h> @@ -305,6 +307,45 @@ struct thread_routine_ctx { return thread_id; } +void +rte_thread_set_name(rte_thread_t id, const char *name) +{ + int ret = 0; + wchar_t wname[RTE_MAX_THREAD_NAME_LEN]; + mbstate_t state = {0}; + size_t rv; + HANDLE thread_handle; + + thread_handle = OpenThread(THREAD_ALL_ACCESS, FALSE, id.opaque_id); + if (thread_handle == NULL) { + ret = thread_log_last_error("OpenThread()"); + goto cleanup; + } + + rv = mbsrtowcs(wname, &name, sizeof(wname) / sizeof(wname[0]), &state); + if (rv == (size_t)-1) { + ret = EILSEQ; + goto cleanup; + } + + if (name != NULL) { + ret = ERANGE; + goto cleanup; + } + + if (FAILED(SetThreadDescription(thread_handle, wname))) { + ret = EINVAL; + goto cleanup; + } + +cleanup: + if (thread_handle != NULL) + CloseHandle(thread_handle); + + if (ret != 0) + RTE_LOG(DEBUG, EAL, "Cannot set name for lcore\n"); +} + int rte_thread_get_priority(rte_thread_t thread_id, enum rte_thread_priority *priority) -- 1.8.3.1