238 lines
6.4 KiB
C
238 lines
6.4 KiB
C
/*
|
|
* Copyright © 2003 Felix Kuehling
|
|
* Copyright © 2018 Advanced Micro Devices, Inc.
|
|
* All Rights Reserved.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sub license, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NON-INFRINGEMENT. IN NO EVENT SHALL THE COPYRIGHT HOLDERS, AUTHORS
|
|
* AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|
* USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*
|
|
* The above copyright notice and this permission notice (including the
|
|
* next paragraph) shall be included in all copies or substantial portions
|
|
* of the Software.
|
|
*/
|
|
|
|
#include "u_process.h"
|
|
#include "detect_os.h"
|
|
#include "macros.h"
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
|
|
#undef GET_PROGRAM_NAME
|
|
|
|
#if DETECT_OS_WINDOWS
|
|
#include <windows.h>
|
|
#else
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#if DETECT_OS_APPLE
|
|
#include <mach-o/dyld.h>
|
|
#endif
|
|
|
|
#if DETECT_OS_FREEBSD
|
|
#include <sys/types.h>
|
|
#include <sys/sysctl.h>
|
|
#endif
|
|
|
|
#if defined(__linux__) && defined(HAVE_PROGRAM_INVOCATION_NAME)
|
|
|
|
static char *path = NULL;
|
|
|
|
static void __freeProgramPath()
|
|
{
|
|
free(path);
|
|
path = NULL;
|
|
}
|
|
|
|
static const char *
|
|
__getProgramName()
|
|
{
|
|
char * arg = strrchr(program_invocation_name, '/');
|
|
if (arg) {
|
|
/* If the / character was found this is likely a linux path or
|
|
* an invocation path for a 64-bit wine program.
|
|
*
|
|
* However, some programs pass command line arguments into argv[0].
|
|
* Strip these arguments out by using the realpath only if it was
|
|
* a prefix of the invocation name.
|
|
*/
|
|
if (!path) {
|
|
path = realpath("/proc/self/exe", NULL);
|
|
atexit(__freeProgramPath);
|
|
}
|
|
|
|
if (path && strncmp(path, program_invocation_name, strlen(path)) == 0) {
|
|
/* This shouldn't be null because path is a a prefix,
|
|
* but check it anyway since path is static. */
|
|
char * name = strrchr(path, '/');
|
|
if (name)
|
|
return name + 1;
|
|
}
|
|
|
|
return arg+1;
|
|
}
|
|
|
|
/* If there was no '/' at all we likely have a windows like path from
|
|
* a wine application.
|
|
*/
|
|
arg = strrchr(program_invocation_name, '\\');
|
|
if (arg)
|
|
return arg+1;
|
|
|
|
return program_invocation_name;
|
|
}
|
|
# define GET_PROGRAM_NAME() __getProgramName()
|
|
#elif defined(HAVE_PROGRAM_INVOCATION_NAME)
|
|
# define GET_PROGRAM_NAME() program_invocation_short_name
|
|
#elif defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__) || defined(ANDROID)
|
|
# define GET_PROGRAM_NAME() getprogname()
|
|
#elif defined(__NetBSD__)
|
|
# include <sys/param.h>
|
|
# if defined(__NetBSD_Version__) && (__NetBSD_Version__ >= 106000100)
|
|
# define GET_PROGRAM_NAME() getprogname()
|
|
# endif
|
|
#elif defined(__sun)
|
|
/* Solaris has getexecname() which returns the full path - return just
|
|
the basename to match BSD getprogname() */
|
|
# include <libgen.h>
|
|
|
|
static const char *
|
|
__getProgramName()
|
|
{
|
|
static const char *progname;
|
|
|
|
if (progname == NULL) {
|
|
const char *e = getexecname();
|
|
if (e != NULL) {
|
|
/* Have to make a copy since getexecname can return a readonly
|
|
string, but basename expects to be able to modify its arg. */
|
|
char *n = strdup(e);
|
|
if (n != NULL) {
|
|
progname = basename(n);
|
|
}
|
|
}
|
|
}
|
|
return progname;
|
|
}
|
|
|
|
# define GET_PROGRAM_NAME() __getProgramName()
|
|
#elif defined(WIN32)
|
|
static const char *
|
|
__getProgramName()
|
|
{
|
|
static const char *progname;
|
|
if (progname == NULL) {
|
|
static char buf[MAX_PATH];
|
|
GetModuleFileNameA(NULL, buf, sizeof(buf));
|
|
progname = strrchr(buf, '\\');
|
|
if (progname)
|
|
progname++;
|
|
else
|
|
progname = buf;
|
|
}
|
|
return progname;
|
|
}
|
|
# define GET_PROGRAM_NAME() __getProgramName()
|
|
#elif defined(__HAIKU__)
|
|
# include <libgen.h>
|
|
extern char **__libc_argv;
|
|
extern int __libc_argc;
|
|
|
|
static const char *
|
|
__getProgramName()
|
|
{
|
|
static const char *progname;
|
|
|
|
if (progname == NULL) {
|
|
char *n = strdup(__libc_argv[0]);
|
|
if (n != NULL) {
|
|
progname = basename(n);
|
|
}
|
|
}
|
|
return progname;
|
|
}
|
|
# define GET_PROGRAM_NAME() __getProgramName()
|
|
#endif
|
|
|
|
#if !defined(GET_PROGRAM_NAME)
|
|
# if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__UCLIBC__) || defined(ANDROID)
|
|
/* This is a hack. It's said to work on OpenBSD, NetBSD and GNU.
|
|
* Rogelio M.Serrano Jr. reported it's also working with UCLIBC. It's
|
|
* used as a last resort, if there is no documented facility available. */
|
|
static const char *
|
|
__getProgramName()
|
|
{
|
|
extern const char *__progname;
|
|
char * arg = strrchr(__progname, '/');
|
|
if (arg)
|
|
return arg+1;
|
|
else
|
|
return __progname;
|
|
}
|
|
# define GET_PROGRAM_NAME() __getProgramName()
|
|
# else
|
|
# define GET_PROGRAM_NAME() ""
|
|
# pragma message ( "Warning: Per application configuration won't work with your OS version." )
|
|
# endif
|
|
#endif
|
|
|
|
const char *
|
|
util_get_process_name(void)
|
|
{
|
|
return GET_PROGRAM_NAME();
|
|
}
|
|
|
|
size_t
|
|
util_get_process_exec_path(char* process_path, size_t len)
|
|
{
|
|
#if DETECT_OS_WINDOWS
|
|
return GetModuleFileNameA(NULL, process_path, len);
|
|
#elif DETECT_OS_APPLE
|
|
uint32_t bufSize = len;
|
|
int result = _NSGetExecutablePath(process_path, &bufSize);
|
|
|
|
return (result == 0) ? strlen(process_path) : 0;
|
|
#elif DETECT_OS_FREEBSD
|
|
int mib[4] = { CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1 };
|
|
|
|
(void) sysctl(mib, 4, process_path, &len, NULL, 0);
|
|
process_path[len - 1] = '\0';
|
|
|
|
return len;
|
|
#elif DETECT_OS_UNIX
|
|
ssize_t r;
|
|
|
|
if ((r = readlink("/proc/self/exe", process_path, len)) > 0)
|
|
goto success;
|
|
if ((r = readlink("/proc/curproc/exe", process_path, len)) > 0)
|
|
goto success;
|
|
if ((r = readlink("/proc/curproc/file", process_path, len)) > 0)
|
|
goto success;
|
|
|
|
return 0;
|
|
success:
|
|
if (r == len)
|
|
return 0;
|
|
|
|
process_path[r] = '\0';
|
|
return r;
|
|
|
|
#endif
|
|
return 0;
|
|
}
|