src/gallium/auxiliary/os/os_process.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2013 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL THE AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28
  29 #include "pipe/p_config.h"
  30 #include "os/os_process.h"
  31 #include "util/u_memory.h"
  32
  33 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
  34 #  include <windows.h>
  35 #elif defined(__GLIBC__) || defined(__CYGWIN__)
  36 #  include <errno.h>
  37 #elif defined(PIPE_OS_BSD) || defined(PIPE_OS_APPLE)
  38 #  include <stdlib.h>
  39 #elif defined(PIPE_OS_HAIKU)
  40 #  include <kernel/OS.h>
  41 #  include <kernel/image.h>
  42 #else
  43 #warning unexpected platform in os_process.c
  44 #endif
  45
  46 #if defined(PIPE_OS_LINUX)
  47 #  include <fcntl.h>
  48 #endif
  49
  50
  51 /**
  52  * Return the name of the current process.
  53  * \param procname  returns the process name
  54  * \param size  size of the procname buffer
  55  * \return  TRUE or FALSE for success, failure
  56  */
  57 boolean
  58 os_get_process_name(char *procname, size_t size)
  59 {
  60    const char *name;
  61
  62    /* First, check if the GALLIUM_PROCESS_NAME env var is set to
  63     * override the normal process name query.
  64     */
  65    name = os_get_option("GALLIUM_PROCESS_NAME");
  66
  67    if (!name) {
  68       /* do normal query */
  69
  70 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
  71       char szProcessPath[MAX_PATH];
  72       char *lpProcessName;
  73       char *lpProcessExt;
  74
  75       GetModuleFileNameA(NULL, szProcessPath, ARRAY_SIZE(szProcessPath));
  76
  77       lpProcessName = strrchr(szProcessPath, '\\');
  78       lpProcessName = lpProcessName ? lpProcessName + 1 : szProcessPath;
  79
  80       lpProcessExt = strrchr(lpProcessName, '.');
  81       if (lpProcessExt) {
  82          *lpProcessExt = '\0';
  83       }
  84
  85       name = lpProcessName;
  86
  87 #elif defined(__GLIBC__) || defined(__CYGWIN__)
  88       name = program_invocation_short_name;
  89 #elif defined(PIPE_OS_BSD) || defined(PIPE_OS_APPLE)
  90       /* *BSD and OS X */
  91       name = getprogname();
  92 #elif defined(PIPE_OS_HAIKU)
  93       image_info info;
  94       get_image_info(B_CURRENT_TEAM, &info);
  95       name = info.name;
  96 #else
  97 #warning unexpected platform in os_process.c
  98       return FALSE;
  99 #endif
 100
 101    }
 102
 103    assert(size > 0);
 104    assert(procname);
 105
 106    if (name && procname && size > 0) {
 107       strncpy(procname, name, size);
 108       procname[size - 1] = '\0';
 109       return TRUE;
 110    }
 111    else {
 112       return FALSE;
 113    }
 114 }
 115
 116
 117 /**
 118  * Return the command line for the calling process.  This is basically
 119  * the argv[] array with the arguments separated by spaces.
 120  * \param cmdline  returns the command line string
 121  * \param size  size of the cmdline buffer
 122  * \return  TRUE or FALSE for success, failure
 123  */
 124 boolean
 125 os_get_command_line(char *cmdline, size_t size)
 126 {
 127 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
 128    const char *args = GetCommandLine();
 129    if (args) {
 130       strncpy(cmdline, args, size);
 131       // make sure we terminate the string
 132       cmdline[size - 1] = 0;
 133       return TRUE;
 134    }
 135 #elif defined(PIPE_OS_LINUX)
 136    int f = open("/proc/self/cmdline", O_RDONLY);
 137    if (f) {
 138       const int n = read(f, cmdline, size - 1);
 139       int i;
 140       assert(n < size);
 141       // The arguments are separated by '\0' chars.  Convert them to spaces.
 142       for (i = 0; i < n; i++) {
 143          if (cmdline[i] == 0) {
 144             cmdline[i] = ' ';
 145          }
 146       }
 147       // terminate the string
 148       cmdline[n] = 0;
 149       close(f);
 150       return TRUE;
 151    }
 152 #endif
 153
 154    /* XXX to-do: implement this function for other operating systems */
 155
 156    cmdline[0] = 0;
 157    return FALSE;
 158 }