AppRun.c

   1 /**************************************************************************
   2
   3 Copyright (c) 2004-17 Simon Peter
   4 Portions Copyright (c) 2010 RazZziel
   5
   6 All Rights Reserved.
   7
   8 Permission is hereby granted, free of charge, to any person obtaining a copy
   9 of this software and associated documentation files (the "Software"), to deal
  10 in the Software without restriction, including without limitation the rights
  11 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12 copies of the Software, and to permit persons to whom the Software is
  13 furnished to do so, subject to the following conditions:
  14
  15 The above copyright notice and this permission notice shall be included in
  16 all copies or substantial portions of the Software.
  17
  18 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24 THE SOFTWARE.
  25
  26 **************************************************************************/
  27
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <unistd.h>
  31 #include <libgen.h>
  32 #include <dirent.h>
  33 #include <string.h>
  34
  35 #define die(...)                                    \
  36     do {                                            \
  37         fprintf(stderr, "Error: " __VA_ARGS__);     \
  38         exit(1);                                    \
  39     } while(0);
  40 #define SET_NEW_ENV(str,len,fmt,...)                \
  41     format = fmt;                                   \
  42     length = strlen(format) + (len);                \
  43     char *str = calloc(length, sizeof(char));     \
  44     snprintf(str, length, format, __VA_ARGS__);   \
  45     putenv(str);
  46 #define MAX(a,b)    (a > b ? a : b)
  47 #define bool int
  48 #define false 0
  49 #define true -1
  50
  51 #define LINE_SIZE 255
  52
  53 int filter(const struct dirent *dir) {
  54     char *p = (char*) &dir->d_name;
  55     p = strrchr(p, '.');
  56     return p && !strcmp(p, ".desktop");
  57 }
  58
  59 int main(int argc, char *argv[]) {
  60     char *appdir = dirname(realpath("/proc/self/exe", NULL));
  61     if (!appdir)
  62         die("Could not access /proc/self/exe\n");
  63
  64     int ret;
  65     struct dirent **namelist;
  66     ret = scandir(appdir, &namelist, filter, NULL);
  67
  68     if (ret == 0) {
  69         die("No .desktop files found\n");
  70     } else if(ret == -1) {
  71         die("Could not scan directory %s\n", appdir);
  72     }
  73
  74     /* Extract executable from .desktop file */
  75     char *desktop_file = calloc(LINE_SIZE, sizeof(char));
  76     snprintf(desktop_file, LINE_SIZE, "%s/%s", appdir, namelist[0]->d_name);
  77     FILE *f     = fopen(desktop_file, "r");
  78     char *line  = malloc(LINE_SIZE);
  79     char *exe   = line+5;
  80     size_t n    = LINE_SIZE;
  81
  82     do {
  83         if (getline(&line, &n, f) == -1)
  84             die("Executable not found, make sure there is a line starting with 'Exec='\n");
  85     } while(strncmp(line, "Exec=", 5));
  86     fclose(f);
  87
  88     // parse arguments
  89     bool in_quotes = 0;
  90     for (n = 0; n < LINE_SIZE; n++) {
  91         if (!line[n])         // end of string
  92             break;
  93         else if (line[n] == 10 || line[n] == 13) {
  94             line[n] = '\0';
  95             line[n+1] = '\0';
  96             line[n+2] = '\0';
  97             break;
  98         } else if (line[n] == '"') {
  99             in_quotes = !in_quotes;
 100         } else if (line[n] == ' ' && !in_quotes)
 101             line[n] = '\0';
 102     }
 103
 104     // count arguments
 105     char*   arg         = exe;
 106     int     argcount    = 0;
 107     while ((arg += (strlen(arg)+1)) && *arg)
 108         argcount += 1;
 109
 110     // merge args
 111     char*   outargptrs[argcount + argc + 1];
 112     outargptrs[0] = exe;
 113     int     outargindex = 1;
 114     arg                 = exe;
 115     int     argc_       = argc - 1;     // argv[0] is the filename
 116     char**  argv_       = argv + 1;
 117     while ((arg += (strlen(arg)+1)) && *arg) {
 118         if (arg[0] == '%' || (arg[0] == '"' && arg[1] == '%')) {         // handle desktop file field codes
 119             char code = arg[arg[0] == '%' ? 1 : 2];
 120             switch(code) {
 121                 case 'f':
 122                 case 'u':
 123                     if (argc_ > 0) {
 124                         outargptrs[outargindex++] = *argv_++;
 125                         argc_--;
 126                     }
 127                     break;
 128                 case 'F':
 129                 case 'U':
 130                     while (argc_ > 0) {
 131                         outargptrs[outargindex++] = *argv_++;
 132                         argc_--;
 133                     }
 134                     break;
 135                 case 'i':
 136                 case 'c':
 137                 case 'k':
 138                     fprintf(stderr, "WARNING: Desktop file field code %%%c is not currently supported\n", code);
 139                     break;
 140                 default:
 141                     fprintf(stderr, "WARNING: Invalid desktop file field code %%%c\n", code);
 142                     break;
 143             }
 144         } else {
 145             outargptrs[outargindex++] = arg;
 146         }
 147     }
 148     while (argc_ > 0) {
 149         outargptrs[outargindex++] = *argv_++;
 150         argc_--;
 151     }
 152     outargptrs[outargindex] = '\0';     // trailing null argument required by execvp()
 153
 154     // change directory
 155     size_t appdir_s = strlen(appdir);
 156     char *usr_in_appdir = malloc(appdir_s + 5);
 157     snprintf(usr_in_appdir, appdir_s + 5, "%s/usr", appdir);
 158     ret = chdir(usr_in_appdir);
 159     if (ret != 0)
 160         die("Could not cd into %s\n", usr_in_appdir);
 161
 162     // set environment variables
 163     char *old_env;
 164     size_t length;
 165     const char *format;
 166
 167     /* https://docs.python.org/2/using/cmdline.html#envvar-PYTHONHOME */
 168     SET_NEW_ENV(new_pythonhome, appdir_s, "PYTHONHOME=%s/usr/", appdir);
 169
 170     old_env = getenv("PATH") ?: "";
 171     SET_NEW_ENV(new_path, appdir_s*5 + strlen(old_env), "PATH=%s/usr/bin/:%s/usr/sbin/:%s/usr/games/:%s/bin/:%s/sbin/:%s", appdir, appdir, appdir, appdir, appdir, old_env);
 172
 173     old_env = getenv("LD_LIBRARY_PATH") ?: "";
 174     SET_NEW_ENV(new_ld_library_path, appdir_s*10 + strlen(old_env), "LD_LIBRARY_PATH=%s/usr/lib/:%s/usr/lib/i386-linux-gnu/:%s/usr/lib/x86_64-linux-gnu/:%s/usr/lib32/:%s/usr/lib64/:%s/lib/:%s/lib/i386-linux-gnu/:%s/lib/x86_64-linux-gnu/:%s/lib32/:%s/lib64/:%s", appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, old_env);
 175
 176     old_env = getenv("PYTHONPATH") ?: "";
 177     SET_NEW_ENV(new_pythonpath, appdir_s + strlen(old_env), "PYTHONPATH=%s/usr/share/pyshared/:%s", appdir, old_env);
 178
 179     old_env = getenv("XDG_DATA_DIRS") ?: "";
 180     SET_NEW_ENV(new_xdg_data_dirs, appdir_s + strlen(old_env), "XDG_DATA_DIRS=%s/usr/share/:%s", appdir, old_env);
 181
 182     old_env = getenv("PERLLIB") ?: "";
 183     SET_NEW_ENV(new_perllib, appdir_s*2 + strlen(old_env), "PERLLIB=%s/usr/share/perl5/:%s/usr/lib/perl5/:%s", appdir, appdir, old_env);
 184
 185     /* http://askubuntu.com/questions/251712/how-can-i-install-a-gsettings-schema-without-root-privileges */
 186     old_env = getenv("GSETTINGS_SCHEMA_DIR") ?: "";
 187     SET_NEW_ENV(new_gsettings_schema_dir, appdir_s + strlen(old_env), "GSETTINGS_SCHEMA_DIR=%s/usr/share/glib-2.0/schemas/:%s", appdir, old_env);
 188
 189     old_env = getenv("QT_PLUGIN_PATH") ?: "";
 190     SET_NEW_ENV(new_qt_plugin_path, appdir_s*10 + strlen(old_env), "QT_PLUGIN_PATH=%s/usr/lib/qt4/plugins/:%s/usr/lib/i386-linux-gnu/qt4/plugins/:%s/usr/lib/x86_64-linux-gnu/qt4/plugins/:%s/usr/lib32/qt4/plugins/:%s/usr/lib64/qt4/plugins/:%s/usr/lib/qt5/plugins/:%s/usr/lib/i386-linux-gnu/qt5/plugins/:%s/usr/lib/x86_64-linux-gnu/qt5/plugins/:%s/usr/lib32/qt5/plugins/:%s/usr/lib64/qt5/plugins/:%s", appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, appdir, old_env);
 191
 192     /* Otherwise may get errors because Python cannot write __pycache__ bytecode cache */
 193     putenv("PYTHONDONTWRITEBYTECODE=1");
 194
 195     /* Run */
 196     ret = execvp(exe, outargptrs);
 197
 198     if (ret == -1)
 199         die("Error executing '%s'; return code: %d\n", exe, ret);
 200
 201     free(line);
 202     free(desktop_file);
 203     free(usr_in_appdir);
 204     free(new_pythonhome);
 205     free(new_path);
 206     free(new_ld_library_path);
 207     free(new_pythonpath);
 208     free(new_xdg_data_dirs);
 209     free(new_perllib);
 210     free(new_gsettings_schema_dir);
 211     free(new_qt_plugin_path);
 212     return 0;
 213 }