boinc/api/graphics2_util.C

83 lines
2.0 KiB
C

#ifdef _WIN32
#include "boinc_win.h"
#endif
#include "shmem.h"
#include "filesys.h"
#include "app_ipc.h"
#include "boinc_api.h"
#include "graphics2.h"
#ifdef USE_FILE_MAPPED_SHMEM
#include <sys/stat.h>
#endif
#if (defined(_WIN32) || defined(USE_FILE_MAPPED_SHMEM))
static void get_shmem_name(char* prog_name, char* shmem_name) {
APP_INIT_DATA aid;
boinc_get_init_data(aid);
sprintf(shmem_name, "boinc_%s_%d", prog_name, aid.slot);
}
#else
static key_t get_shmem_name(char* prog_name) {
char cwd[256], path[256];
boinc_getcwd(cwd);
sprintf(path, "%s/init_data.xml", cwd);
return ftok(path, 2);
}
#endif
void* boinc_graphics_make_shmem(char* prog_name, int size) {
#ifdef _WIN32
HANDLE shmem_handle;
char shmem_name[256];
void* p;
get_shmem_name(prog_name, shmem_name);
shmem_handle = create_shmem(shmem_name, size, &p, false);
if (shmem_handle == NULL) return 0;
return p;
#else
void* p;
#ifdef USE_FILE_MAPPED_SHMEM
char shmem_name[256];
get_shmem_name(prog_name, shmem_name);
int retval = create_shmem(shmem_name, size, &p);
#else
key_t key = get_shmem_name(prog_name);
int retval = create_shmem(key, size, 0, &p);
#endif
if (retval) return 0;
return p;
#endif
}
#ifdef _WIN32
void* boinc_graphics_get_shmem(char* prog_name) {
HANDLE shmem_handle;
char shmem_name[256];
void* p;
get_shmem_name(prog_name, shmem_name);
shmem_handle = attach_shmem(shmem_name, &p);
if (shmem_handle == NULL) return 0;
return p;
}
#else
void* boinc_graphics_get_shmem(char* prog_name) {
void* p;
#ifdef USE_FILE_MAPPED_SHMEM
struct stat sbuf;
char shmem_name[256];
get_shmem_name(prog_name, shmem_name);
int retval = stat(shmem_name, &sbuf);
if (retval == 0) {
retval = create_shmem(shmem_name, sbuf.st_size, &p);
}
#else
key_t key = get_shmem_name(prog_name);
int retval = attach_shmem(key, &p);
#endif
if (retval) return 0;
return p;
}
#endif