blob: dcd8b00b788867c228896574b247e5ca67b02b50 [file] [log] [blame]
/* Copyright (C) 2025 Free Software Foundation, Inc.
Contributed by Thomas Koenig, Nicolas Koenig, Andre Vehreschild
This file is part of the GNU Fortran Shmem Coarray Library (caf_shmem).
Caf_shmem is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3, or (at your option)
any later version.
Caf_shmem is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
Under Section 7 of GPL version 3, you are granted additional
permissions described in the GCC Runtime Library Exception, version
3.1, as published by the Free Software Foundation.
You should have received a copy of the GNU General Public License and
a copy of the GCC Runtime Library Exception along with this program;
see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
<http://www.gnu.org/licenses/>. */
#include "thread_support.h"
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#if !defined(WIN32) && !defined(__CYGWIN__)
#include <pthread.h>
#define ERRCHECK(a) \
do \
{ \
int rc = a; \
if (rc) \
{ \
errno = rc; \
perror (#a " failed"); \
exit (1); \
} \
} \
while (0)
void
initialize_shared_mutex (caf_shmem_mutex *mutex)
{
pthread_mutexattr_t mattr;
ERRCHECK (pthread_mutexattr_init (&mattr));
ERRCHECK (pthread_mutexattr_setpshared (&mattr, PTHREAD_PROCESS_SHARED));
#ifdef PTHREAD_MUTEX_ROBUST
ERRCHECK (pthread_mutexattr_setrobust (&mattr, PTHREAD_MUTEX_ROBUST));
#endif
ERRCHECK (pthread_mutex_init (mutex, &mattr));
ERRCHECK (pthread_mutexattr_destroy (&mattr));
}
void
initialize_shared_errorcheck_mutex (caf_shmem_mutex *mutex)
{
pthread_mutexattr_t mattr;
ERRCHECK (pthread_mutexattr_init (&mattr));
ERRCHECK (pthread_mutexattr_settype (&mattr, PTHREAD_MUTEX_ERRORCHECK));
ERRCHECK (pthread_mutexattr_setpshared (&mattr, PTHREAD_PROCESS_SHARED));
#ifdef PTHREAD_MUTEX_ROBUST
ERRCHECK (pthread_mutexattr_setrobust (&mattr, PTHREAD_MUTEX_ROBUST));
#endif
ERRCHECK (pthread_mutex_init (mutex, &mattr));
ERRCHECK (pthread_mutexattr_destroy (&mattr));
}
void
initialize_shared_condition (caf_shmem_condvar *cond, const int)
{
pthread_condattr_t cattr;
ERRCHECK (pthread_condattr_init (&cattr));
ERRCHECK (pthread_condattr_setpshared (&cattr, PTHREAD_PROCESS_SHARED));
ERRCHECK (pthread_cond_init (cond, &cattr));
ERRCHECK (pthread_condattr_destroy (&cattr));
}
#else
#include "../caf_error.h"
#include "supervisor.h"
#include "teams_mgmt.h"
#include <windows.h>
#include <assert.h>
static HANDLE *handles = NULL;
static size_t cap_handles = 0;
static const int ULONGBITS = sizeof (unsigned long) << 3; // *8
static size_t
smax (size_t a, size_t b)
{
return a < b ? b : a;
}
static HANDLE
get_handle (const size_t id, const char t)
{
const int add = t == 'c' ? 1 : 0;
while (id + add >= cap_handles)
{
cap_handles += 1024;
if (handles)
handles = realloc (handles, sizeof (HANDLE) * cap_handles);
else
handles = malloc (sizeof (HANDLE) * cap_handles);
if (!handles)
caf_runtime_error (
"can not get buffer for synchronication objects, aborting");
memset (&handles[cap_handles - 1024], 0, sizeof (HANDLE) * 1024);
}
if (!handles[id])
{
static char *pid = NULL;
char name[MAX_PATH];
if (!pid)
pid = shared_memory_get_env ();
snprintf (name, MAX_PATH, "Global_gfortran-%s-%c-%zd", pid, t, id);
switch (t)
{
case 'm':
handles[id] = CreateMutex (NULL, false, name);
break;
case 'c':
{
handles[id] = CreateSemaphore (NULL, 0, __INT_MAX__, name);
snprintf (name, MAX_PATH, "Global_gfortran-%s-%c-%zd_lock", pid, t,
id);
handles[id + 1] = CreateSemaphore (NULL, 1, 1, name);
this_image.supervisor->global_used_handles
= smax (this_image.supervisor->global_used_handles, id + 2);
break;
}
default:
caf_runtime_error ("Unknown handle type %c", t);
exit (1);
}
if (handles[id] == NULL)
{
caf_runtime_error (
"Could not create synchronisation object, error: %d",
GetLastError ());
return NULL;
}
this_image.supervisor->global_used_handles
= smax (this_image.supervisor->global_used_handles, id + 1);
}
return handles[id];
}
static HANDLE
get_mutex (caf_shmem_mutex *m)
{
return get_handle (m->id, 'm');
}
static HANDLE
get_condvar (caf_shmem_condvar *cv)
{
return get_handle (cv->id, 'c');
}
void
thread_support_init_supervisor (void)
{
if (local->total_num_images > ULONGBITS * MAX_NUM_SIGNALED)
caf_runtime_error ("Maximum number of supported images is %zd.",
ULONGBITS * MAX_NUM_SIGNALED);
this_image.supervisor->global_used_handles = 0;
}
int
caf_shmem_mutex_lock (caf_shmem_mutex *m)
{
HANDLE mutex = get_mutex (m);
DWORD res = WaitForSingleObject (mutex, INFINITE);
/* Return zero on success. */
return res != WAIT_OBJECT_0;
}
int
caf_shmem_mutex_trylock (caf_shmem_mutex *m)
{
HANDLE mutex = get_mutex (m);
DWORD res = WaitForSingleObject (mutex, 0);
return res == WAIT_OBJECT_0 ? 0 : EBUSY;
}
int
caf_shmem_mutex_unlock (caf_shmem_mutex *m)
{
HANDLE mutex = get_mutex (m);
BOOL res = ReleaseMutex (mutex);
if (!res)
{
LPVOID lpMsgBuf;
DWORD dw = GetLastError ();
if (FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER
| FORMAT_MESSAGE_FROM_SYSTEM
| FORMAT_MESSAGE_IGNORE_INSERTS,
NULL, dw, MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR) &lpMsgBuf, 0, NULL)
== 0)
{
fprintf (stderr, "%d: formatting the error message failed.\n",
this_image.image_num);
ExitProcess (dw);
}
fprintf (stderr, "%d: unlock mutex failed: %d, %s\n",
this_image.image_num, dw, (LPCTSTR) lpMsgBuf);
LocalFree (lpMsgBuf);
}
return res ? 0 : EPERM;
}
static bool
bm_is_set (volatile unsigned long mask[], const int b)
{
return (mask[b / ULONGBITS] & (1UL << (b % ULONGBITS))) != 0;
}
static void
bm_clear_bit (volatile unsigned long mask[], const int b)
{
mask[b / ULONGBITS] &= ~(1UL << (b % ULONGBITS));
}
static void
bm_set_mask (volatile unsigned long mask[], const int size)
{
const int entries = size / ULONGBITS;
const int rem = size % ULONGBITS;
int i = 0;
assert (entries >= 0);
for (; i < entries; ++i)
mask[i] = ~0UL;
if (rem != 0)
mask[i] = ~0UL >> (ULONGBITS - rem);
}
__attribute__ ((used)) static bool
bm_is_none (volatile unsigned long mask[], const int size)
{
const int entries = size / ULONGBITS;
const int rem = size % ULONGBITS;
int i = 0;
for (; i < entries; ++i)
if (mask[i] != 0)
return false;
return rem == 0 || ((mask[i] & (~0UL >> (ULONGBITS - rem))) == 0);
}
void
caf_shmem_cond_wait (caf_shmem_condvar *cv, caf_shmem_mutex *m)
{
HANDLE mutex = get_mutex (m), condvar = get_condvar (cv),
lock = get_handle (cv->id + 1, 'c');
HANDLE entry[3] = {mutex, condvar, lock};
int res;
WaitForSingleObject (lock, INFINITE);
for (;;)
{
if (bm_is_set (cv->signaled, this_image.image_num) || cv->any)
{
break;
}
ReleaseMutex (mutex);
ReleaseSemaphore (lock, 1, NULL);
res = WaitForMultipleObjects (3, entry, true, INFINITE);
if (res != WAIT_OBJECT_0)
{
fprintf (stderr, "%d: failed to get all wait for: %d\n",
this_image.image_num, res);
fflush (stderr);
}
ReleaseSemaphore (condvar, 1, NULL);
}
res = WaitForSingleObject (condvar, INFINITE);
if (res != WAIT_OBJECT_0)
{
fprintf (stderr, "%d: failed to get condvar: %d\n", this_image.image_num,
res);
fflush (stderr);
}
bm_clear_bit (cv->signaled, this_image.image_num);
cv->any = 0;
ReleaseSemaphore (lock, 1, NULL);
}
void
caf_shmem_cond_broadcast (caf_shmem_condvar *cv)
{
HANDLE condvar = get_condvar (cv), lock = get_handle (cv->id + 1, 'c');
WaitForSingleObject (lock, INFINITE);
bm_set_mask (cv->signaled, cv->size);
bm_clear_bit (cv->signaled, this_image.image_num);
ReleaseSemaphore (condvar, cv->size, NULL);
ReleaseSemaphore (lock, 1, NULL);
}
void
caf_shmem_cond_signal (caf_shmem_condvar *cv)
{
HANDLE condvar = get_condvar (cv), lock = get_handle (cv->id + 1, 'c');
if (caf_current_team)
{
WaitForSingleObject (lock, INFINITE);
}
else
return;
/* The first image is zero, which wouldn't allow it to signal. */
cv->any = this_image.image_num + 1;
ReleaseSemaphore (condvar, 1, NULL);
ReleaseSemaphore (lock, 1, NULL);
}
void
caf_shmem_cond_update_count (caf_shmem_condvar *cv, int val)
{
cv->size += val;
}
void
initialize_shared_mutex (caf_shmem_mutex *m)
{
*m = (caf_shmem_mutex) {this_image.supervisor->global_used_handles};
get_mutex (m);
}
void
initialize_shared_errorcheck_mutex (caf_shmem_mutex *m)
{
*m = (caf_shmem_mutex) {this_image.supervisor->global_used_handles};
get_mutex (m);
}
void
initialize_shared_condition (caf_shmem_condvar *cv, const int size)
{
*cv = (caf_shmem_condvar) {this_image.supervisor->global_used_handles,
0,
size,
{}};
memset ((void *) cv->signaled, 0, sizeof (unsigned long) * MAX_NUM_SIGNALED);
get_condvar (cv);
assert (bm_is_none (cv->signaled, cv->size));
}
void
thread_support_cleanup (void)
{
for (size_t i = 0; i < this_image.supervisor->global_used_handles; ++i)
if (handles[i])
CloseHandle (handles[i]);
}
#endif