diff options
Diffstat (limited to '3rdparty/plibsys/src/psemaphore-sysv.c')
-rw-r--r-- | 3rdparty/plibsys/src/psemaphore-sysv.c | 319 |
1 files changed, 319 insertions, 0 deletions
diff --git a/3rdparty/plibsys/src/psemaphore-sysv.c b/3rdparty/plibsys/src/psemaphore-sysv.c new file mode 100644 index 0000000..34b33fc --- /dev/null +++ b/3rdparty/plibsys/src/psemaphore-sysv.c @@ -0,0 +1,319 @@ +/* + * The MIT License + * + * Copyright (C) 2010-2018 Alexander Saprykin <saprykin.spb@gmail.com> + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * 'Software'), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + */ + +#include "perror.h" +#include "pmem.h" +#include "psemaphore.h" +#include "perror-private.h" +#include "pipc-private.h" + +#include <stdlib.h> +#include <string.h> + +#include <unistd.h> +#include <errno.h> +#include <fcntl.h> +#include <sys/stat.h> +#include <sys/sem.h> +#include <sys/types.h> +#include <sys/ipc.h> + +#define P_SEM_SUFFIX "_p_sem_object" +#define P_SEM_INVALID_HDL -1 + +struct sembuf sem_lock = {0, -1, SEM_UNDO}; +struct sembuf sem_unlock = {0, 1, SEM_UNDO}; + +typedef union p_semun_ { + pint val; + struct semid_ds *buf; + pushort *array; +} p_semun; + +typedef int psem_hdl; + +struct PSemaphore_ { + pboolean file_created; + pboolean sem_created; + key_t unix_key; + pchar *platform_key; + psem_hdl sem_hdl; + PSemaphoreAccessMode mode; + pint init_val; +}; + +static pboolean pp_semaphore_create_handle (PSemaphore *sem, PError **error); +static void pp_semaphore_clean_handle (PSemaphore *sem); + +static pboolean +pp_semaphore_create_handle (PSemaphore *sem, PError **error) +{ + pint built; + p_semun semun_op; + + if (P_UNLIKELY (sem == NULL || sem->platform_key == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + if (P_UNLIKELY ((built = p_ipc_unix_create_key_file (sem->platform_key)) == -1)) { + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to create key file"); + pp_semaphore_clean_handle (sem); + return FALSE; + } else if (built == 0) + sem->file_created = TRUE; + + if (P_UNLIKELY ((sem->unix_key = p_ipc_unix_get_ftok_key (sem->platform_key)) == -1)) { + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to get unique IPC key"); + pp_semaphore_clean_handle (sem); + return FALSE; + } + + if ((sem->sem_hdl = semget (sem->unix_key, + 1, + IPC_CREAT | IPC_EXCL | 0660)) == P_SEM_INVALID_HDL) { + if (p_error_get_last_system () == EEXIST) + sem->sem_hdl = semget (sem->unix_key, 1, 0660); + } else { + sem->sem_created = TRUE; + + /* Maybe key file left after the crash, so take it */ + sem->file_created = (built == 1); + } + + if (P_UNLIKELY (sem->sem_hdl == P_SEM_INVALID_HDL)) { + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to call semget() to create semaphore"); + pp_semaphore_clean_handle (sem); + return FALSE; + } + + if (sem->sem_created == TRUE || sem->mode == P_SEM_ACCESS_CREATE) { + semun_op.val = sem->init_val; + + if (P_UNLIKELY (semctl (sem->sem_hdl, 0, SETVAL, semun_op) == -1)) { + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to set semaphore initial value with semctl()"); + pp_semaphore_clean_handle (sem); + return FALSE; + } + } + + return TRUE; +} + +static void +pp_semaphore_clean_handle (PSemaphore *sem) +{ + if (sem->sem_hdl != P_SEM_INVALID_HDL && + sem->sem_created == TRUE && + semctl (sem->sem_hdl, 0, IPC_RMID) == -1) + P_ERROR ("PSemaphore::pp_semaphore_clean_handle: semctl() with IPC_RMID failed"); + + if (sem->file_created == TRUE && + sem->platform_key != NULL && + unlink (sem->platform_key) == -1) + P_ERROR ("PSemaphore::pp_semaphore_clean_handle: unlink() failed"); + + sem->file_created = FALSE; + sem->sem_created = FALSE; + sem->unix_key = -1; + sem->sem_hdl = P_SEM_INVALID_HDL; +} + +P_LIB_API PSemaphore * +p_semaphore_new (const pchar *name, + pint init_val, + PSemaphoreAccessMode mode, + PError **error) +{ + PSemaphore *ret; + pchar *new_name; + + if (P_UNLIKELY (name == NULL || init_val < 0)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return NULL; + } + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PSemaphore))) == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_NO_RESOURCES, + 0, + "Failed to allocate memory for semaphore"); + return NULL; + } + + if (P_UNLIKELY ((new_name = p_malloc0 (strlen (name) + strlen (P_SEM_SUFFIX) + 1)) == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_NO_RESOURCES, + 0, + "Failed to allocate memory for semaphore"); + p_free (ret); + return NULL; + } + + strcpy (new_name, name); + strcat (new_name, P_SEM_SUFFIX); + + ret->platform_key = p_ipc_get_platform_key (new_name, FALSE); + ret->init_val = init_val; + ret->mode = mode; + + p_free (new_name); + + if (P_UNLIKELY (pp_semaphore_create_handle (ret, error) == FALSE)) { + p_semaphore_free (ret); + return NULL; + } + + return ret; +} + +P_LIB_API void +p_semaphore_take_ownership (PSemaphore *sem) +{ + if (P_UNLIKELY (sem == NULL)) + return; + + sem->sem_created = TRUE; +} + +P_LIB_API pboolean +p_semaphore_acquire (PSemaphore *sem, + PError **error) +{ + pboolean ret; + pint res; + + if (P_UNLIKELY (sem == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + while ((res = semop (sem->sem_hdl, &sem_lock, 1)) == -1 && + p_error_get_last_system () == EINTR) + ; + + ret = (res == 0); + + if (P_UNLIKELY (ret == FALSE && + (p_error_get_last_system () == EIDRM || + p_error_get_last_system () == EINVAL))) { + P_WARNING ("PSemaphore::p_semaphore_acquire: trying to recreate"); + pp_semaphore_clean_handle (sem); + + if (P_UNLIKELY (pp_semaphore_create_handle (sem, error) == FALSE)) + return FALSE; + + while ((res = semop (sem->sem_hdl, &sem_lock, 1)) == -1 && + p_error_get_last_system () == EINTR) + ; + + ret = (res == 0); + } + + if (P_UNLIKELY (ret == FALSE)) + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to call semop() on semaphore"); + + return ret; +} + +P_LIB_API pboolean +p_semaphore_release (PSemaphore *sem, + PError **error) +{ + pboolean ret; + pint res; + + if (P_UNLIKELY (sem == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IPC_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + while ((res = semop (sem->sem_hdl, &sem_unlock, 1)) == -1 && + p_error_get_last_system () == EINTR) + ; + + ret = (res == 0); + + if (P_UNLIKELY (ret == FALSE && + (p_error_get_last_system () == EIDRM || + p_error_get_last_system () == EINVAL))) { + P_WARNING ("PSemaphore::p_semaphore_release: trying to recreate"); + pp_semaphore_clean_handle (sem); + + if (P_UNLIKELY (pp_semaphore_create_handle (sem, error) == FALSE)) + return FALSE; + + return TRUE; + } + + if (P_UNLIKELY (ret == FALSE)) + p_error_set_error_p (error, + (pint) p_error_get_last_ipc (), + p_error_get_last_system (), + "Failed to call semop() on semaphore"); + + return ret; +} + +P_LIB_API void +p_semaphore_free (PSemaphore *sem) +{ + if (P_UNLIKELY (sem == NULL)) + return; + + pp_semaphore_clean_handle (sem); + + if (P_LIKELY (sem->platform_key != NULL)) + p_free (sem->platform_key); + + p_free (sem); +} |