foundationdb/flow/ThreadPrimitives.cpp

/*
 * ThreadPrimitives.cpp
 *
 * This source file is part of the FoundationDB open source project
 *
 * Copyright 2013-2018 Apple Inc. and the FoundationDB project authors
 *
 * 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.
 */

#include "flow/ThreadPrimitives.h"
#include "flow/Trace.h"
#include <stdint.h>
#include <iostream>
#include <errno.h>
#include <string.h>
#include <stdio.h>

#ifdef _WIN32
#include <windows.h>
#undef min
#undef max
#endif

extern std::string format( const char *form, ... );

Event::Event() {
#ifdef _WIN32
	ev = CreateEvent(NULL, FALSE, FALSE, NULL);
#elif defined(__linux__)
	int result = sem_init(&sem, 0, 0);
	if (result)
		criticalError(FDB_EXIT_INIT_SEMAPHORE, "UnableToInitializeSemaphore", format("Could not initialize semaphore - %s", strerror(errno)).c_str());
#elif defined(__APPLE__)
	self = mach_task_self();
	kern_return_t ret = semaphore_create(self, &sem, SYNC_POLICY_FIFO, 0);
	if(ret != KERN_SUCCESS)
		criticalError(FDB_EXIT_INIT_SEMAPHORE, "UnableToInitializeSemaphore", format("Could not initialize semaphore - %s", strerror(errno)).c_str());
#else
#error Port me!
#endif
}

Event::~Event() {
#ifdef _WIN32
	CloseHandle(ev);
#elif defined(__linux__)
	sem_destroy(&sem);
#elif defined(__APPLE__)
	semaphore_destroy(self, sem);
#else
#error Port me!
#endif
}

void Event::set() {
#ifdef _WIN32
	SetEvent(ev);
#elif defined(__linux__)
	sem_post(&sem);
#elif defined(__APPLE__)
	semaphore_signal(sem);
#else
#error Port me!
#endif
}

void Event::block() {
#ifdef _WIN32
	WaitForSingleObject(ev, INFINITE);
#elif defined(__linux__)
	int ret;
	do {
		ret = sem_wait(&sem);
	} while (ret != 0 && errno == EINTR);
#elif defined(__APPLE__)
	kern_return_t ret;
	do {
		ret = semaphore_wait(sem);
	} while ((ret != KERN_SUCCESS) && (ret != KERN_TERMINATED));
#else
#error Port me!
#endif
}

// Mutex::impl is allocated from the heap so that its size doesn't need to be in the header.  Is there a better way?
Mutex::Mutex() {
	impl = new CRITICAL_SECTION;
	InitializeCriticalSection((CRITICAL_SECTION*)impl);
}
Mutex::~Mutex() {
	DeleteCriticalSection((CRITICAL_SECTION*)impl);
	delete (CRITICAL_SECTION*)impl;
}
void Mutex::enter() {
	EnterCriticalSection((CRITICAL_SECTION*)impl);
}
void Mutex::leave() {
	LeaveCriticalSection((CRITICAL_SECTION*)impl);
}