FreeRDP/winpr/libwinpr/synch/semaphore.c

265 lines
5.6 KiB
C
Raw Normal View History

2012-05-29 22:14:26 +04:00
/**
* WinPR: Windows Portable Runtime
* Synchronization Functions
*
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
*
* Licensed 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.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
2012-05-29 22:14:26 +04:00
#include <winpr/synch.h>
#include "synch.h"
2012-05-29 22:14:26 +04:00
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifndef _WIN32
2012-05-29 22:14:26 +04:00
2015-03-11 19:57:01 +03:00
#include <errno.h>
2013-05-17 01:32:58 +04:00
#include "../handle/handle.h"
2014-08-18 19:22:22 +04:00
#include "../log.h"
#define TAG WINPR_TAG("synch.semaphore")
2015-03-11 17:11:17 +03:00
static BOOL SemaphoreCloseHandle(HANDLE handle);
static BOOL SemaphoreIsHandled(HANDLE handle)
{
2019-11-06 17:24:51 +03:00
WINPR_TIMER* pSemaphore = (WINPR_TIMER*)handle;
2015-03-11 17:11:17 +03:00
if (!pSemaphore || (pSemaphore->Type != HANDLE_TYPE_SEMAPHORE))
2015-03-11 17:11:17 +03:00
{
SetLastError(ERROR_INVALID_HANDLE);
return FALSE;
}
return TRUE;
}
2015-03-11 19:57:01 +03:00
static int SemaphoreGetFd(HANDLE handle)
{
WINPR_SEMAPHORE* sem = (WINPR_SEMAPHORE*)handle;
2015-03-11 19:57:01 +03:00
if (!SemaphoreIsHandled(handle))
return -1;
return sem->pipe_fd[0];
}
static DWORD SemaphoreCleanupHandle(HANDLE handle)
2015-03-11 17:11:17 +03:00
{
2015-03-11 19:57:01 +03:00
int length;
WINPR_SEMAPHORE* sem = (WINPR_SEMAPHORE*)handle;
2015-03-11 19:57:01 +03:00
if (!SemaphoreIsHandled(handle))
return WAIT_FAILED;
length = read(sem->pipe_fd[0], &length, 1);
if (length != 1)
{
WLog_ERR(TAG, "semaphore read() failure [%d] %s", errno, strerror(errno));
return WAIT_FAILED;
}
return WAIT_OBJECT_0;
2015-03-11 17:11:17 +03:00
}
BOOL SemaphoreCloseHandle(HANDLE handle)
{
2019-11-06 17:24:51 +03:00
WINPR_SEMAPHORE* semaphore = (WINPR_SEMAPHORE*)handle;
2015-03-11 17:11:17 +03:00
if (!SemaphoreIsHandled(handle))
return FALSE;
#ifdef WINPR_PIPE_SEMAPHORE
if (semaphore->pipe_fd[0] != -1)
{
close(semaphore->pipe_fd[0]);
semaphore->pipe_fd[0] = -1;
if (semaphore->pipe_fd[1] != -1)
{
close(semaphore->pipe_fd[1]);
semaphore->pipe_fd[1] = -1;
}
}
#else
#if defined __APPLE__
2019-11-06 17:24:51 +03:00
semaphore_destroy(mach_task_self(), *((winpr_sem_t*)semaphore->sem));
2015-03-11 17:11:17 +03:00
#else
2019-11-06 17:24:51 +03:00
sem_destroy((winpr_sem_t*)semaphore->sem);
2015-03-11 17:11:17 +03:00
#endif
#endif
free(semaphore);
return TRUE;
}
2013-05-17 01:32:58 +04:00
2019-11-06 17:24:51 +03:00
static HANDLE_OPS ops = { SemaphoreIsHandled,
SemaphoreCloseHandle,
SemaphoreGetFd,
SemaphoreCleanupHandle,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL };
HANDLE CreateSemaphoreW(LPSECURITY_ATTRIBUTES lpSemaphoreAttributes, LONG lInitialCount,
LONG lMaximumCount, LPCWSTR lpName)
2012-05-29 22:14:26 +04:00
{
HANDLE handle;
2014-08-18 21:34:47 +04:00
WINPR_SEMAPHORE* semaphore;
2019-11-06 17:24:51 +03:00
semaphore = (WINPR_SEMAPHORE*)calloc(1, sizeof(WINPR_SEMAPHORE));
2012-05-29 22:14:26 +04:00
2013-08-29 12:42:08 +04:00
if (!semaphore)
return NULL;
semaphore->pipe_fd[0] = -1;
2017-10-28 11:59:23 +03:00
semaphore->pipe_fd[1] = -1;
2019-11-06 17:24:51 +03:00
semaphore->sem = (winpr_sem_t*)NULL;
semaphore->ops = &ops;
#ifdef WINPR_PIPE_SEMAPHORE
if (pipe(semaphore->pipe_fd) < 0)
{
WLog_ERR(TAG, "failed to create semaphore");
free(semaphore);
return NULL;
}
while (lInitialCount > 0)
{
if (write(semaphore->pipe_fd[1], "-", 1) != 1)
{
close(semaphore->pipe_fd[0]);
close(semaphore->pipe_fd[1]);
2013-08-29 17:30:22 +04:00
free(semaphore);
return NULL;
}
lInitialCount--;
}
#else
2019-11-06 17:24:51 +03:00
semaphore->sem = (winpr_sem_t*)malloc(sizeof(winpr_sem_t));
if (!semaphore->sem)
{
WLog_ERR(TAG, "failed to allocate semaphore memory");
free(semaphore);
return NULL;
}
2012-05-29 22:14:26 +04:00
#if defined __APPLE__
2019-11-06 17:24:51 +03:00
if (semaphore_create(mach_task_self(), semaphore->sem, SYNC_POLICY_FIFO, lMaximumCount) !=
KERN_SUCCESS)
2012-05-29 22:14:26 +04:00
#else
if (sem_init(semaphore->sem, 0, lMaximumCount) == -1)
2012-05-29 22:14:26 +04:00
#endif
{
WLog_ERR(TAG, "failed to create semaphore");
free(semaphore->sem);
free(semaphore);
return NULL;
}
#endif
WINPR_HANDLE_SET_TYPE_AND_MODE(semaphore, HANDLE_TYPE_SEMAPHORE, WINPR_FD_READ);
2019-11-06 17:24:51 +03:00
handle = (HANDLE)semaphore;
return handle;
2012-05-29 22:14:26 +04:00
}
HANDLE CreateSemaphoreA(LPSECURITY_ATTRIBUTES lpSemaphoreAttributes, LONG lInitialCount,
LONG lMaximumCount, LPCSTR lpName)
2012-05-29 22:14:26 +04:00
{
return CreateSemaphoreW(lpSemaphoreAttributes, lInitialCount, lMaximumCount, NULL);
2012-05-29 22:14:26 +04:00
}
HANDLE OpenSemaphoreW(DWORD dwDesiredAccess, BOOL bInheritHandle, LPCWSTR lpName)
2012-05-29 22:14:26 +04:00
{
2014-08-18 19:22:22 +04:00
WLog_ERR(TAG, "not implemented");
2012-05-29 22:14:26 +04:00
return NULL;
}
HANDLE OpenSemaphoreA(DWORD dwDesiredAccess, BOOL bInheritHandle, LPCSTR lpName)
2012-05-29 22:14:26 +04:00
{
2014-08-18 19:22:22 +04:00
WLog_ERR(TAG, "not implemented");
2012-05-29 22:14:26 +04:00
return NULL;
}
BOOL ReleaseSemaphore(HANDLE hSemaphore, LONG lReleaseCount, LPLONG lpPreviousCount)
{
ULONG Type;
2015-07-03 10:36:58 +03:00
WINPR_HANDLE* Object;
2014-08-18 21:34:47 +04:00
WINPR_SEMAPHORE* semaphore;
if (!winpr_Handle_GetInfo(hSemaphore, &Type, &Object))
return FALSE;
if (Type == HANDLE_TYPE_SEMAPHORE)
{
2019-11-06 17:24:51 +03:00
semaphore = (WINPR_SEMAPHORE*)Object;
#ifdef WINPR_PIPE_SEMAPHORE
if (semaphore->pipe_fd[0] != -1)
{
while (lReleaseCount > 0)
{
if (write(semaphore->pipe_fd[1], "-", 1) != 1)
return FALSE;
lReleaseCount--;
}
}
#else
while (lReleaseCount > 0)
{
2012-05-29 22:14:26 +04:00
#if defined __APPLE__
2019-11-06 17:24:51 +03:00
semaphore_signal(*((winpr_sem_t*)semaphore->sem));
2012-05-29 22:14:26 +04:00
#else
2019-11-06 17:24:51 +03:00
sem_post((winpr_sem_t*)semaphore->sem);
#endif
}
2012-05-29 22:14:26 +04:00
#endif
return TRUE;
}
2012-05-29 22:14:26 +04:00
WLog_ERR(TAG, "calling %s on a handle that is not a semaphore", __FUNCTION__);
return FALSE;
2012-05-29 22:14:26 +04:00
}
#endif