1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright(c) 2010-2014 Intel Corporation
15 #include <rte_lcore.h>
16 #include <rte_memory.h>
18 #include <rte_trace_point.h>
20 #include "eal_internal_cfg.h"
21 #include "eal_private.h"
22 #include "eal_thread.h"
23 #include "eal_trace.h"
25 RTE_DEFINE_PER_LCORE(unsigned int, _lcore_id) = LCORE_ID_ANY;
26 RTE_DEFINE_PER_LCORE(int, _thread_id) = -1;
27 static RTE_DEFINE_PER_LCORE(unsigned int, _socket_id) =
28 (unsigned int)SOCKET_ID_ANY;
29 static RTE_DEFINE_PER_LCORE(rte_cpuset_t, _cpuset);
31 unsigned rte_socket_id(void)
33 return RTE_PER_LCORE(_socket_id);
37 rte_lcore_has_role(unsigned int lcore_id, enum rte_lcore_role_t role)
39 struct rte_config *cfg = rte_eal_get_configuration();
41 if (lcore_id >= RTE_MAX_LCORE)
44 return cfg->lcore_role[lcore_id] == role;
48 eal_cpuset_socket_id(rte_cpuset_t *cpusetp)
51 int socket_id = SOCKET_ID_ANY;
58 if (!CPU_ISSET(cpu, cpusetp))
61 if (socket_id == SOCKET_ID_ANY)
62 socket_id = eal_cpu_socket_id(cpu);
64 sid = eal_cpu_socket_id(cpu);
65 if (socket_id != sid) {
66 socket_id = SOCKET_ID_ANY;
70 } while (++cpu < CPU_SETSIZE);
76 thread_update_affinity(rte_cpuset_t *cpusetp)
78 unsigned int lcore_id = rte_lcore_id();
80 /* store socket_id in TLS for quick access */
81 RTE_PER_LCORE(_socket_id) =
82 eal_cpuset_socket_id(cpusetp);
84 /* store cpuset in TLS for quick access */
85 memmove(&RTE_PER_LCORE(_cpuset), cpusetp,
86 sizeof(rte_cpuset_t));
88 if (lcore_id != (unsigned)LCORE_ID_ANY) {
89 /* EAL thread will update lcore_config */
90 lcore_config[lcore_id].socket_id = RTE_PER_LCORE(_socket_id);
91 memmove(&lcore_config[lcore_id].cpuset, cpusetp,
92 sizeof(rte_cpuset_t));
97 rte_thread_set_affinity(rte_cpuset_t *cpusetp)
99 if (pthread_setaffinity_np(pthread_self(), sizeof(rte_cpuset_t),
101 RTE_LOG(ERR, EAL, "pthread_setaffinity_np failed\n");
105 thread_update_affinity(cpusetp);
110 rte_thread_get_affinity(rte_cpuset_t *cpusetp)
113 memmove(cpusetp, &RTE_PER_LCORE(_cpuset),
114 sizeof(rte_cpuset_t));
118 eal_thread_dump_affinity(char *str, unsigned size)
123 unsigned int out = 0;
125 rte_thread_get_affinity(&cpuset);
127 for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
128 if (!CPU_ISSET(cpu, &cpuset))
131 ret = snprintf(str + out,
132 size - out, "%u,", cpu);
133 if (ret < 0 || (unsigned)ret >= size - out) {
134 /* string will be truncated */
144 /* remove the last separator */
152 __rte_thread_init(unsigned int lcore_id, rte_cpuset_t *cpuset)
154 /* set the lcore ID in per-lcore memory area */
155 RTE_PER_LCORE(_lcore_id) = lcore_id;
157 /* acquire system unique id */
160 thread_update_affinity(cpuset);
162 __rte_trace_mem_per_thread_alloc();
166 __rte_thread_uninit(void)
168 trace_mem_per_thread_free();
170 RTE_PER_LCORE(_lcore_id) = LCORE_ID_ANY;
173 struct rte_thread_ctrl_params {
174 void *(*start_routine)(void *);
176 pthread_barrier_t configured;
179 static void *ctrl_thread_init(void *arg)
182 struct internal_config *internal_conf =
183 eal_get_internal_configuration();
184 rte_cpuset_t *cpuset = &internal_conf->ctrl_cpuset;
185 struct rte_thread_ctrl_params *params = arg;
186 void *(*start_routine)(void *) = params->start_routine;
187 void *routine_arg = params->arg;
189 __rte_thread_init(rte_lcore_id(), cpuset);
191 ret = pthread_barrier_wait(¶ms->configured);
192 if (ret == PTHREAD_BARRIER_SERIAL_THREAD) {
193 pthread_barrier_destroy(¶ms->configured);
197 return start_routine(routine_arg);
201 rte_ctrl_thread_create(pthread_t *thread, const char *name,
202 const pthread_attr_t *attr,
203 void *(*start_routine)(void *), void *arg)
205 struct internal_config *internal_conf =
206 eal_get_internal_configuration();
207 rte_cpuset_t *cpuset = &internal_conf->ctrl_cpuset;
208 struct rte_thread_ctrl_params *params;
211 params = malloc(sizeof(*params));
215 params->start_routine = start_routine;
218 pthread_barrier_init(¶ms->configured, NULL, 2);
220 ret = pthread_create(thread, attr, ctrl_thread_init, (void *)params);
227 ret = rte_thread_setname(*thread, name);
230 "Cannot set name for ctrl thread\n");
233 ret = pthread_setaffinity_np(*thread, sizeof(*cpuset), cpuset);
237 ret = pthread_barrier_wait(¶ms->configured);
238 if (ret == PTHREAD_BARRIER_SERIAL_THREAD) {
239 pthread_barrier_destroy(¶ms->configured);
246 if (PTHREAD_BARRIER_SERIAL_THREAD ==
247 pthread_barrier_wait(¶ms->configured)) {
248 pthread_barrier_destroy(¶ms->configured);
251 pthread_cancel(*thread);
252 pthread_join(*thread, NULL);