git.oblomov.eu Git - wine/blob - tools/winegcc/utils.c

   1 /*
   2  * Useful functions for winegcc/winewrap
   3  *
   4  * Copyright 2000 Francois Gouget
   5  * Copyright 2002 Dimitrie O. Paun
   6  * Copyright 2003 Richard Cohen
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  21  */
  22 #include "config.h"
  23 #include "wine/port.h"
  24
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <stdarg.h>
  28 #include <stdlib.h>
  29 #include <errno.h>
  30
  31 #include "utils.h"
  32
  33 #if !defined(min)
  34 # define min(x,y) (((x) < (y)) ? (x) : (y))
  35 #endif
  36
  37 int verbose = 0;
  38
  39 void error(const char* s, ...)
  40 {
  41     va_list ap;
  42
  43     va_start(ap, s);
  44     fprintf(stderr, "winegcc: ");
  45     vfprintf(stderr, s, ap);
  46     fprintf(stderr, "\n");
  47     va_end(ap);
  48     exit(2);
  49 }
  50
  51 void* xmalloc(size_t size)
  52 {
  53     void* p;
  54
  55     if ((p = malloc (size)) == NULL)
  56         error("Can not malloc %d bytes.", size);
  57
  58     return p;
  59 }
  60
  61 void *xrealloc(void* p, size_t size)
  62 {
  63     void* p2;
  64     if ((p2 = realloc (p, size)) == NULL)
  65         error("Can not realloc %d bytes.", size);
  66
  67     return p2;
  68 }
  69
  70 int strendswith(const char* str, const char* end)
  71 {
  72     int l = strlen(str);
  73     int m = strlen(end);
  74
  75     return l >= m && strcmp(str + l - m, end) == 0;
  76 }
  77
  78 char* strmake(const char* fmt, ...)
  79 {
  80     int n;
  81     size_t size = 100;
  82     char* p;
  83     va_list ap;
  84
  85     p = xmalloc (size);
  86     while (1)
  87     {
  88         va_start(ap, fmt);
  89         n = vsnprintf (p, size, fmt, ap);
  90         va_end(ap);
  91         if (n > -1 && (size_t)n < size) return p;
  92         size = min( size*2, (size_t)n+1 );
  93         p = xrealloc (p, size);
  94     }
  95 }
  96
  97 strarray* strarray_alloc(void)
  98 {
  99     strarray* arr = xmalloc(sizeof(*arr));
 100     arr->maximum = arr->size = 0;
 101     arr->base = NULL;
 102     return arr;
 103 }
 104
 105 void strarray_free(strarray* arr)
 106 {
 107     free(arr->base);
 108     free(arr);
 109 }
 110
 111 void strarray_add(strarray* arr, const char* str)
 112 {
 113     if (arr->size == arr->maximum)
 114     {
 115         arr->maximum += 10;
 116         arr->base = xrealloc(arr->base, sizeof(*(arr->base)) * arr->maximum);
 117     }
 118     arr->base[arr->size++] = str;
 119 }
 120
 121 void strarray_del(strarray* arr, int i)
 122 {
 123     if (i < 0 || i >= arr->size) error("Invalid index i=%d", i);
 124     memmove(&arr->base[i], &arr->base[i + 1], (arr->size - i - 1) * sizeof(arr->base[0]));
 125     arr->size--;
 126 }
 127
 128 void strarray_addall(strarray* arr, const strarray* from)
 129 {
 130     int i;
 131
 132     for (i = 0; i < from->size; i++)
 133         strarray_add(arr, from->base[i]);
 134 }
 135
 136 strarray* strarray_dup(const strarray* arr)
 137 {
 138     strarray* dup = strarray_alloc();
 139     int i;
 140
 141     for (i = 0; i < arr->size; i++)
 142         strarray_add(dup, arr->base[i]);
 143
 144     return dup;
 145 }
 146
 147 strarray* strarray_fromstring(const char* str, const char* delim)
 148 {
 149     strarray* arr = strarray_alloc();
 150     char* buf = strdup(str);
 151     const char* tok;
 152
 153     for(tok = strtok(buf, delim); tok; tok = strtok(0, delim))
 154         strarray_add(arr, strdup(tok));
 155
 156     free(buf);
 157     return arr;
 158 }
 159
 160 char* strarray_tostring(const strarray* arr, const char* sep)
 161 {
 162     char *str, *newstr;
 163     int i;
 164
 165     str = strmake("%s", arr->base[0]);
 166     for (i = 1; i < arr->size; i++)
 167     {
 168         newstr = strmake("%s%s%s", str, sep, arr->base[i]);
 169         free(str);
 170         str = newstr;
 171     }
 172
 173     return str;
 174 }
 175
 176 char* get_basename(const char* file)
 177 {
 178     const char* name;
 179     char *base_name, *p;
 180
 181     if ((name = strrchr(file, '/'))) name++;
 182     else name = file;
 183
 184     base_name = strdup(name);
 185     if ((p = strrchr(base_name, '.'))) *p = 0;
 186
 187     return base_name;
 188 }
 189
 190 void create_file(const char* name, int mode, const char* fmt, ...)
 191 {
 192     va_list ap;
 193     FILE *file;
 194
 195     if (verbose) printf("Creating file %s\n", name);
 196     va_start(ap, fmt);
 197     if ( !(file = fopen(name, "w")) )
 198         error ("Can not create %s.", name);
 199     vfprintf(file, fmt, ap);
 200     va_end(ap);
 201     fclose(file);
 202     chmod(name, mode);
 203 }
 204
 205 file_type get_file_type(const char* filename)
 206 {
 207     /* see tools/winebuild/res32.c: check_header for details */
 208     static const char res_sig[] = { 0,0,0,0, 32,0,0,0, 0xff,0xff, 0,0, 0xff,0xff, 0,0, 0,0,0,0, 0,0, 0,0, 0,0,0,0, 0,0,0,0 };
 209     char buf[sizeof(res_sig)];
 210     int fd, cnt;
 211
 212     fd = open( filename, O_RDONLY );
 213     if (fd == -1) return file_na;
 214     cnt = read(fd, buf, sizeof(buf));
 215     close( fd );
 216     if (cnt == -1) return file_na;
 217
 218     if (cnt == sizeof(res_sig) && !memcmp(buf, res_sig, sizeof(res_sig))) return file_res;
 219     if (strendswith(filename, ".o")) return file_obj;
 220     if (strendswith(filename, ".a")) return file_arh;
 221     if (strendswith(filename, ".res")) return file_res;
 222     if (strendswith(filename, ".so")) return file_so;
 223     if (strendswith(filename, ".dylib")) return file_so;
 224     if (strendswith(filename, ".def")) return file_def;
 225     if (strendswith(filename, ".spec")) return file_spec;
 226     if (strendswith(filename, ".rc")) return file_rc;
 227
 228     return file_other;
 229 }
 230
 231 static char* try_lib_path(const char* dir, const char* pre,
 232                           const char* library, const char* ext,
 233                           file_type expected_type)
 234 {
 235     char *fullname;
 236     file_type type;
 237
 238     fullname = strmake("%s/%s%s%s", dir, pre, library, ext);
 239     if (verbose > 1) fprintf(stderr, "Try %s...", fullname);
 240     type = get_file_type(fullname);
 241     if (verbose > 1) fprintf(stderr, type == expected_type ? "FOUND!\n" : "no\n");
 242     if (type == expected_type) return fullname;
 243     free( fullname );
 244     return 0;
 245 }
 246
 247 static file_type guess_lib_type(const char* dir, const char* library, char** file)
 248 {
 249     /* Unix shared object */
 250     if ((*file = try_lib_path(dir, "lib", library, ".so", file_so)))
 251         return file_so;
 252
 253     /* Mach-O (Darwin/Mac OS X) Dynamic Library behaves mostly like .so */
 254     if ((*file = try_lib_path(dir, "lib", library, ".dylib", file_so)))
 255         return file_so;
 256
 257     /* Windows DLL */
 258     if ((*file = try_lib_path(dir, "lib", library, ".def", file_def)))
 259         return file_dll;
 260     if ((*file = try_lib_path(dir, "", library, ".def", file_def)))
 261         return file_dll;
 262
 263     /* Unix static archives */
 264     if ((*file = try_lib_path(dir, "lib", library, ".a", file_arh)))
 265         return file_arh;
 266
 267     return file_na;
 268 }
 269
 270 file_type get_lib_type(strarray* path, const char* library, char** file)
 271 {
 272     int i;
 273
 274     for (i = 0; i < path->size; i++)
 275     {
 276         file_type type = guess_lib_type(path->base[i], library, file);
 277         if (type != file_na) return type;
 278     }
 279     return file_na;
 280 }
 281
 282 void spawn(const strarray* prefix, const strarray* args)
 283 {
 284     int i, status;
 285     strarray* arr = strarray_dup(args);
 286     const char** argv;
 287     char* prog = 0;
 288
 289     strarray_add(arr, NULL);
 290     argv = arr->base;
 291
 292     if (prefix)
 293     {
 294         for (i = 0; i < prefix->size; i++)
 295         {
 296             const char* p;
 297             struct stat st;
 298
 299             if (!(p = strrchr(argv[0], '/'))) p = argv[0];
 300             free( prog );
 301             prog = strmake("%s/%s", prefix->base[i], p);
 302             if (stat(prog, &st) == 0)
 303             {
 304                 if ((st.st_mode & S_IFREG) && (st.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)))
 305                 {
 306                     argv[0] = prog;
 307                     break;
 308                 }
 309             }
 310         }
 311     }
 312
 313     if (verbose)
 314     {
 315         for(i = 0; argv[i]; i++) printf("%s ", argv[i]);
 316         printf("\n");
 317     }
 318
 319     if ((status = spawnvp( _P_WAIT, argv[0], argv)))
 320     {
 321         if (status > 0) error("%s failed.", argv[0]);
 322         else perror("winegcc");
 323         exit(3);
 324     }
 325
 326     free(prog);
 327     strarray_free(arr);
 328 }