nuttx/libs/libc/pthread/pthread_once.c
hujun5 4268b2e297 pthread_once: use rmutex replace sched_[un]lock
sched_[un]lock can not prohibit pre-emption in smp

Signed-off-by: hujun5 <hujun5@xiaomi.com>
2023-09-21 00:42:29 +08:00

98 lines
3.1 KiB
C

/****************************************************************************
* libs/libc/pthread/pthread_once.c
*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership. The
* ASF licenses this file to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*
****************************************************************************/
/****************************************************************************
* Included Files
****************************************************************************/
#include <nuttx/config.h>
#include <assert.h>
#include <errno.h>
#include <stdbool.h>
#include <pthread.h>
#include <nuttx/mutex.h>
#include <debug.h>
/****************************************************************************
* Public Functions
****************************************************************************/
/****************************************************************************
* Name: pthread_once
*
* Description:
* The first call to pthread_once() by any thread with a given
* once_control, will call the init_routine with no arguments. Subsequent
* calls to pthread_once() with the same once_control will have no effect.
* On return from pthread_once(), init_routine will have completed.
*
* Input Parameters:
* once_control - Determines if init_routine should be called.
* once_control should be declared and initialized as follows:
*
* pthread_once_t once_control = PTHREAD_ONCE_INIT;
*
* PTHREAD_ONCE_INIT is defined in pthread.h
* init_routine - The initialization routine that will be called once.
*
* Returned Value:
* 0 (OK) on success or an error number shall be returned to
* indicate the error.
*
* Assumptions:
*
****************************************************************************/
static rmutex_t g_lock = NXRMUTEX_INITIALIZER;
int pthread_once(FAR pthread_once_t *once_control,
CODE void (*init_routine)(void))
{
/* Sanity checks */
if (once_control == NULL || init_routine == NULL)
{
return EINVAL;
}
/* Prohibit pre-emption while we test and set the once_control. */
nxrmutex_lock(&g_lock);
if (!*once_control)
{
*once_control = true;
/* Call the init_routine with pre-emption enabled. */
init_routine();
nxrmutex_unlock(&g_lock);
return OK;
}
/* The init_routine has already been called.
* Restore pre-emption and return.
*/
nxrmutex_unlock(&g_lock);
return OK;
}