eal: fix memory ordering around lcore task accesses

Ensure that the memory operations before the call to
rte_eal_remote_launch are visible to the worker thread.
Use the function pointer to execute in worker thread
as the guard variable.

Ensure that the memory operations in worker thread, that happen
before it returns the status of the assigned function, are
visible to the main thread. Use the variable containing the
lcore's state as the guard variable.

Signed-off-by: Honnappa Nagarahalli <honnappa.nagarahalli@arm.com>
Reviewed-by: Ola Liljedahl <ola.liljedahl@arm.com>
Reviewed-by: Feifei Wang <feifei.wang2@arm.com>
This commit is contained in:
Honnappa Nagarahalli 2021-10-24 23:52:36 -05:00 committed by David Marchand
parent f6c6c686f1
commit 3596537005
4 changed files with 99 additions and 30 deletions

View File

@ -23,14 +23,10 @@
int
rte_eal_wait_lcore(unsigned worker_id)
{
if (lcore_config[worker_id].state == WAIT)
return 0;
while (lcore_config[worker_id].state != WAIT)
while (__atomic_load_n(&lcore_config[worker_id].state,
__ATOMIC_ACQUIRE) != WAIT)
rte_pause();
rte_rmb();
return lcore_config[worker_id].ret;
}

View File

@ -39,11 +39,19 @@ rte_eal_remote_launch(int (*f)(void *), void *arg, unsigned worker_id)
int w2m = lcore_config[worker_id].pipe_worker2main[0];
int rc = -EBUSY;
if (lcore_config[worker_id].state != WAIT)
/* Check if the worker is in 'WAIT' state. Use acquire order
* since 'state' variable is used as the guard variable.
*/
if (__atomic_load_n(&lcore_config[worker_id].state,
__ATOMIC_ACQUIRE) != WAIT)
goto finish;
lcore_config[worker_id].f = f;
lcore_config[worker_id].arg = arg;
/* Ensure that all the memory operations are completed
* before the worker thread starts running the function.
* Use worker thread function as the guard variable.
*/
__atomic_store_n(&lcore_config[worker_id].f, f, __ATOMIC_RELEASE);
/* send message */
n = 0;
@ -100,6 +108,7 @@ eal_thread_loop(__rte_unused void *arg)
/* read on our pipe to get commands */
while (1) {
lcore_function_t *f;
void *fct_arg;
/* wait command */
@ -110,7 +119,11 @@ eal_thread_loop(__rte_unused void *arg)
if (n <= 0)
rte_panic("cannot read on configuration pipe\n");
lcore_config[lcore_id].state = RUNNING;
/* Set the state to 'RUNNING'. Use release order
* since 'state' variable is used as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, RUNNING,
__ATOMIC_RELEASE);
/* send ack */
n = 0;
@ -119,17 +132,29 @@ eal_thread_loop(__rte_unused void *arg)
if (n < 0)
rte_panic("cannot write on configuration pipe\n");
if (lcore_config[lcore_id].f == NULL)
rte_panic("NULL function pointer\n");
/* Load 'f' with acquire order to ensure that
* the memory operations from the main thread
* are accessed only after update to 'f' is visible.
* Wait till the update to 'f' is visible to the worker.
*/
while ((f = __atomic_load_n(&lcore_config[lcore_id].f,
__ATOMIC_ACQUIRE)) == NULL)
rte_pause();
/* call the function and store the return value */
fct_arg = lcore_config[lcore_id].arg;
ret = lcore_config[lcore_id].f(fct_arg);
ret = f(fct_arg);
lcore_config[lcore_id].ret = ret;
lcore_config[lcore_id].f = NULL;
lcore_config[lcore_id].arg = NULL;
rte_wmb();
lcore_config[lcore_id].state = WAIT;
/* Store the state with release order to ensure that
* the memory operations from the worker thread
* are completed before the state is updated.
* Use 'state' as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, WAIT,
__ATOMIC_RELEASE);
}
/* never reached */

View File

@ -39,11 +39,19 @@ rte_eal_remote_launch(int (*f)(void *), void *arg, unsigned int worker_id)
int w2m = lcore_config[worker_id].pipe_worker2main[0];
int rc = -EBUSY;
if (lcore_config[worker_id].state != WAIT)
/* Check if the worker is in 'WAIT' state. Use acquire order
* since 'state' variable is used as the guard variable.
*/
if (__atomic_load_n(&lcore_config[worker_id].state,
__ATOMIC_ACQUIRE) != WAIT)
goto finish;
lcore_config[worker_id].f = f;
lcore_config[worker_id].arg = arg;
/* Ensure that all the memory operations are completed
* before the worker thread starts running the function.
* Use worker thread function pointer as the guard variable.
*/
__atomic_store_n(&lcore_config[worker_id].f, f, __ATOMIC_RELEASE);
/* send message */
n = 0;
@ -100,6 +108,7 @@ eal_thread_loop(__rte_unused void *arg)
/* read on our pipe to get commands */
while (1) {
lcore_function_t *f;
void *fct_arg;
/* wait command */
@ -110,7 +119,11 @@ eal_thread_loop(__rte_unused void *arg)
if (n <= 0)
rte_panic("cannot read on configuration pipe\n");
lcore_config[lcore_id].state = RUNNING;
/* Set the state to 'RUNNING'. Use release order
* since 'state' variable is used as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, RUNNING,
__ATOMIC_RELEASE);
/* send ack */
n = 0;
@ -119,18 +132,29 @@ eal_thread_loop(__rte_unused void *arg)
if (n < 0)
rte_panic("cannot write on configuration pipe\n");
if (lcore_config[lcore_id].f == NULL)
rte_panic("NULL function pointer\n");
/* Load 'f' with acquire order to ensure that
* the memory operations from the main thread
* are accessed only after update to 'f' is visible.
* Wait till the update to 'f' is visible to the worker.
*/
while ((f = __atomic_load_n(&lcore_config[lcore_id].f,
__ATOMIC_ACQUIRE)) == NULL)
rte_pause();
/* call the function and store the return value */
fct_arg = lcore_config[lcore_id].arg;
ret = lcore_config[lcore_id].f(fct_arg);
ret = f(fct_arg);
lcore_config[lcore_id].ret = ret;
lcore_config[lcore_id].f = NULL;
lcore_config[lcore_id].arg = NULL;
rte_wmb();
lcore_config[lcore_id].state = WAIT;
/* Store the state with release order to ensure that
* the memory operations from the worker thread
* are completed before the state is updated.
* Use 'state' as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, WAIT,
__ATOMIC_RELEASE);
}
/* never reached */

View File

@ -29,11 +29,19 @@ rte_eal_remote_launch(lcore_function_t *f, void *arg, unsigned int worker_id)
int m2w = lcore_config[worker_id].pipe_main2worker[1];
int w2m = lcore_config[worker_id].pipe_worker2main[0];
if (lcore_config[worker_id].state != WAIT)
/* Check if the worker is in 'WAIT' state. Use acquire order
* since 'state' variable is used as the guard variable.
*/
if (__atomic_load_n(&lcore_config[worker_id].state,
__ATOMIC_ACQUIRE) != WAIT)
return -EBUSY;
lcore_config[worker_id].f = f;
lcore_config[worker_id].arg = arg;
/* Ensure that all the memory operations are completed
* before the worker thread starts running the function.
* Use worker thread function as the guard variable.
*/
__atomic_store_n(&lcore_config[worker_id].f, f, __ATOMIC_RELEASE);
/* send message */
n = 0;
@ -84,6 +92,7 @@ eal_thread_loop(void *arg __rte_unused)
/* read on our pipe to get commands */
while (1) {
lcore_function_t *f;
void *fct_arg;
/* wait command */
@ -94,7 +103,11 @@ eal_thread_loop(void *arg __rte_unused)
if (n <= 0)
rte_panic("cannot read on configuration pipe\n");
lcore_config[lcore_id].state = RUNNING;
/* Set the state to 'RUNNING'. Use release order
* since 'state' variable is used as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, RUNNING,
__ATOMIC_RELEASE);
/* send ack */
n = 0;
@ -103,18 +116,29 @@ eal_thread_loop(void *arg __rte_unused)
if (n < 0)
rte_panic("cannot write on configuration pipe\n");
if (lcore_config[lcore_id].f == NULL)
rte_panic("NULL function pointer\n");
/* Load 'f' with acquire order to ensure that
* the memory operations from the main thread
* are accessed only after update to 'f' is visible.
* Wait till the update to 'f' is visible to the worker.
*/
while ((f = __atomic_load_n(&lcore_config[lcore_id].f,
__ATOMIC_ACQUIRE)) == NULL)
rte_pause();
/* call the function and store the return value */
fct_arg = lcore_config[lcore_id].arg;
ret = lcore_config[lcore_id].f(fct_arg);
ret = f(fct_arg);
lcore_config[lcore_id].ret = ret;
lcore_config[lcore_id].f = NULL;
lcore_config[lcore_id].arg = NULL;
rte_wmb();
lcore_config[lcore_id].state = WAIT;
/* Store the state with release order to ensure that
* the memory operations from the worker thread
* are completed before the state is updated.
* Use 'state' as the guard variable.
*/
__atomic_store_n(&lcore_config[lcore_id].state, WAIT,
__ATOMIC_RELEASE);
}
}