131 lines
2.4 KiB
C
131 lines
2.4 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "dtypes.h"
|
|
|
|
#ifdef WIN32
|
|
#include <malloc.h>
|
|
#include <sys/timeb.h>
|
|
#include <windows.h>
|
|
#undef NO_ERROR
|
|
#undef MOD_SHIFT
|
|
#undef TRUE
|
|
#undef FALSE
|
|
#undef VOID
|
|
#else
|
|
#include <sys/time.h>
|
|
#include <sys/poll.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "dirpath.h"
|
|
|
|
void get_cwd(char *buf, size_t size)
|
|
{
|
|
#ifndef WIN32
|
|
getcwd(buf, size);
|
|
#else
|
|
GetCurrentDirectory(size, buf);
|
|
#endif
|
|
}
|
|
|
|
int set_cwd(char *buf)
|
|
{
|
|
#ifndef WIN32
|
|
if (chdir(buf) == -1)
|
|
return 1;
|
|
#else
|
|
if (SetCurrentDirectory(buf) == 0)
|
|
return 1;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
// destructively convert path to directory part
|
|
void path_to_dirname(char *path)
|
|
{
|
|
char *sep = strrchr(path, PATHSEP);
|
|
if (sep != NULL) {
|
|
*sep = '\0';
|
|
}
|
|
else {
|
|
path[0] = '\0';
|
|
}
|
|
}
|
|
|
|
#ifdef LINUX
|
|
char *get_exename(char *buf, size_t size)
|
|
{
|
|
char linkname[64]; /* /proc/<pid>/exe */
|
|
pid_t pid;
|
|
ssize_t ret;
|
|
|
|
/* Get our PID and build the name of the link in /proc */
|
|
pid = getpid();
|
|
|
|
if (snprintf(linkname, sizeof(linkname), "/proc/%i/exe", pid) < 0)
|
|
return NULL;
|
|
|
|
/* Now read the symbolic link */
|
|
ret = readlink(linkname, buf, size);
|
|
|
|
/* In case of an error, leave the handling up to the caller */
|
|
if (ret == -1)
|
|
return NULL;
|
|
|
|
/* Report insufficient buffer size */
|
|
if ((size_t)ret >= size)
|
|
return NULL;
|
|
|
|
/* Ensure proper NUL termination */
|
|
buf[ret] = 0;
|
|
|
|
return buf;
|
|
}
|
|
#elif defined(OPENBSD)
|
|
char *get_exename(char *buf, size_t size)
|
|
{
|
|
/* OpenBSD currently has no way of determining a processes pathname */
|
|
return NULL;
|
|
}
|
|
#elif defined(FREEBSD)
|
|
#include <sys/types.h>
|
|
#include <sys/sysctl.h>
|
|
|
|
char *get_exename(char *buf, size_t size)
|
|
{
|
|
int mib[4];
|
|
mib[0] = CTL_KERN;
|
|
mib[1] = KERN_PROC;
|
|
mib[2] = KERN_PROC_PATHNAME;
|
|
mib[3] = -1;
|
|
sysctl(mib, 4, buf, &size, NULL, 0);
|
|
|
|
return buf;
|
|
}
|
|
#elif defined(WIN32)
|
|
char *get_exename(char *buf, size_t size)
|
|
{
|
|
if (GetModuleFileName(NULL, buf, size) == 0)
|
|
return NULL;
|
|
|
|
return buf;
|
|
}
|
|
#elif defined(MACOSX)
|
|
#include <mach-o/dyld.h>
|
|
char *get_exename(char *buf, size_t size)
|
|
{
|
|
uint32_t bufsize = (uint32_t)size;
|
|
if (_NSGetExecutablePath(buf, &bufsize))
|
|
return NULL;
|
|
return buf;
|
|
}
|
|
#endif
|