/[jscoverage]/trunk/util.c
ViewVC logotype

Diff of /trunk/util.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 87 by siliconforks, Mon May 5 20:05:27 2008 UTC revision 448 by siliconforks, Tue Aug 11 17:16:32 2009 UTC
# Line 1  Line 1 
1  /*  /*
2      util.c - general purpose utility routines      util.c - general purpose utility routines
3      Copyright (C) 2007, 2008 siliconforks.com      Copyright (C) 2007, 2008, 2009 siliconforks.com
4    
5      This program is free software; you can redistribute it and/or modify      This program is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published by      it under the terms of the GNU General Public License as published by
# Line 17  Line 17 
17      51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.      51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  */  */
19    
20    #define _GNU_SOURCE
21    
22    #include <config.h>
23    
24  #include "util.h"  #include "util.h"
25    
26  #include <assert.h>  #include <assert.h>
27  #include <errno.h>  #include <errno.h>
28    #include <limits.h>
29  #include <stdarg.h>  #include <stdarg.h>
30  #include <stdio.h>  #include <stdio.h>
31    #include <stdint.h>
32  #include <string.h>  #include <string.h>
33  #include <strings.h>  #include <strings.h>
34    
# Line 41  Line 47 
47    vfprintf(stderr, format, ap);    vfprintf(stderr, format, ap);
48    va_end(ap);    va_end(ap);
49    fputc('\n', stderr);    fputc('\n', stderr);
50      exit(EXIT_FAILURE);
51    }
52    
53    void fatal_command_line(const char * format, ...) {
54      fprintf(stderr, "%s: ", program);
55      va_list ap;
56      va_start(ap, format);
57      vfprintf(stderr, format, ap);
58      va_end(ap);
59      fputc('\n', stderr);
60    fprintf(stderr, "Try `%s --help' for more information.\n", program);    fprintf(stderr, "Try `%s --help' for more information.\n", program);
61    exit(EXIT_FAILURE);    exit(EXIT_FAILURE);
62  }  }
63    
64    void fatal_source(const char * source_file, unsigned int line_number, const char * format, ...) {
65      fprintf(stderr, "%s:%s:%u: ", program, source_file, line_number);
66      va_list ap;
67      va_start(ap, format);
68      vfprintf(stderr, format, ap);
69      va_end(ap);
70      fputc('\n', stderr);
71      exit(EXIT_FAILURE);
72    }
73    
74    void warn_source(const char * source_file, unsigned int line_number, const char * format, ...) {
75      fprintf(stderr, "%s:%s:%u: ", program, source_file, line_number);
76      va_list ap;
77      va_start(ap, format);
78      vfprintf(stderr, format, ap);
79      va_end(ap);
80      fputc('\n', stderr);
81    }
82    
83    void version(void) {
84      printf("%s %s\n", program, VERSION);
85      printf("Character encoding support: ");
86    #if HAVE_ICONV
87      printf("iconv\n");
88    #elif HAVE_MULTIBYTETOWIDECHAR
89      printf("MultiByteToWideChar\n");
90    #else
91      printf("none\n");
92    #endif
93      exit(EXIT_SUCCESS);
94    }
95    
96    size_t addst(size_t x, size_t y) {
97      if (SIZE_MAX - x < y) {
98        fatal("integer overflow");
99      }
100      return x + y;
101    }
102    
103    size_t mulst(size_t x, size_t y) {
104      if (x == 0 || y == 0) {
105        return 0;
106      }
107      if (SIZE_MAX / x < y) {
108        fatal("integer overflow");
109      }
110      return x * y;
111    }
112    
113  void * xmalloc(size_t size) {  void * xmalloc(size_t size) {
114    void * result = malloc(size);    void * result = malloc(size);
115    if (result == NULL) {    if (result == NULL) {
# Line 53  Line 118 
118    return result;    return result;
119  }  }
120    
121    void * xrealloc(void * p, size_t size) {
122      void * result = realloc(p, size);
123      if (result == NULL) {
124        fatal("out of memory");
125      }
126      return result;
127    }
128    
129  char * xstrdup(const char * s) {  char * xstrdup(const char * s) {
130    char * result = strdup(s);    char * result = strdup(s);
131    if (result == NULL) {    if (result == NULL) {
# Line 61  Line 134 
134    return result;    return result;
135  }  }
136    
137    char * xstrndup(const char * s, size_t size) {
138      char * result = strndup(s, size);
139      if (result == NULL) {
140        fatal("out of memory");
141      }
142      return result;
143    }
144    
145    int xasprintf(char ** s, const char * template, ...) {
146      va_list a;
147      va_start(a, template);
148      int result = vasprintf(s, template, a);
149      va_end(a);
150      if (result < 0) {
151        fatal("out of memory");
152      }
153      return result;
154    }
155    
156  char * xgetcwd(void) {  char * xgetcwd(void) {
157    char * result = getcwd(NULL, 0);    char * result = getcwd(NULL, 0);
158    if (result == NULL) {    if (result == NULL) {
# Line 149  Line 241 
241    }    }
242  }  }
243    
244    bool str_starts_with(const char * string, const char * prefix) {
245      const char * string_ptr = string;
246      const char * prefix_ptr = prefix;
247      while (*string_ptr != '\0' && *prefix_ptr != '\0') {
248        if (*string_ptr != *prefix_ptr) {
249          return false;
250        }
251        string_ptr++;
252        prefix_ptr++;
253      }
254      if (*string_ptr == '\0' && *prefix_ptr != '\0') {
255        return false;
256      }
257      return true;
258    }
259    
260    bool str_ends_with(const char * string, const char * suffix) {
261      size_t string_length = strlen(string);
262      size_t suffix_length = strlen(suffix);
263      if (string_length < suffix_length) {
264        return false;
265      }
266      return strcmp(string + string_length - suffix_length, suffix) == 0;
267    }
268    
269    static int is_slash(char c) {
270    #ifdef _WIN32
271      return c == '/' || c == '\\';
272    #else
273      return c == '/';
274    #endif
275    }
276    
277  char * make_path(const char * parent, const char * relative_path) {  char * make_path(const char * parent, const char * relative_path) {
278    size_t parent_length = strlen(parent);    size_t parent_length = strlen(parent);
279    size_t relative_path_length = strlen(relative_path);    size_t relative_path_length = strlen(relative_path);
280    char * result = xmalloc(parent_length + relative_path_length + 2);    assert(parent_length > 0);
281      assert(relative_path_length > 0);
282      size_t result_length = addst(parent_length, relative_path_length);
283      int parent_ends_with_slash = is_slash(parent[parent_length - 1]);
284      if (parent_ends_with_slash) {
285        result_length = addst(result_length, 1);
286      }
287      else {
288        result_length = addst(result_length, 2);
289      }
290      char * result = xmalloc(result_length);
291    strcpy(result, parent);    strcpy(result, parent);
292    result[parent_length] = '/';    char * p = result + parent_length;
293    strcpy(result + parent_length + 1, relative_path);    if (! parent_ends_with_slash) {
294        *p = '/';
295        ++p;
296      }
297      strcpy(p, relative_path);
298    return result;    return result;
299  }  }
300    
# Line 275  Line 414 
414    
415    copy_stream(source, destination);    copy_stream(source, destination);
416    
417    #ifndef _WIN32
418      /* copy permissions */
419      struct stat buf;
420      if (fstat(fileno(source), &buf) == -1) {
421        fatal("cannot stat file: %s", source_file);
422      }
423      fchmod(fileno(destination), buf.st_mode);
424    #endif
425    
426    fclose(source);    fclose(source);
427    fclose(destination);    fclose(destination);
428  }  }
429    
430  int directory_is_empty(const char * directory) {  bool directory_is_empty(const char * directory) {
431      bool result = true;
432    DIR * dir = xopendir(directory);    DIR * dir = xopendir(directory);
   int num_entries = 0;  
433    struct dirent * e;    struct dirent * e;
434    while ((e = readdir(dir)) != NULL) {    while ((e = readdir(dir)) != NULL) {
435      if (strcmp(e->d_name, ".") != 0 &&      if (strcmp(e->d_name, ".") != 0 &&
436          strcmp(e->d_name, "..") != 0) {          strcmp(e->d_name, "..") != 0) {
437        num_entries++;        result = false;
438          break;
439      }      }
440    }    }
441    closedir(dir);    closedir(dir);
442    return num_entries == 0;    return result;
443  }  }
444    
445  static struct DirListEntry * recursive_dir_list(const char * root, const char * directory_wrt_root, struct DirListEntry * head) {  static struct DirListEntry * recursive_dir_list(const char * root, const char * directory_wrt_root, struct DirListEntry * head) {
# Line 316  Line 465 
465        head = recursive_dir_list(root, entry_wrt_root, head);        head = recursive_dir_list(root, entry_wrt_root, head);
466        free(entry_wrt_root);        free(entry_wrt_root);
467      }      }
468    #ifndef _WIN32
469        else if (S_ISLNK(buf.st_mode)) {
470          /* check what it points to */
471          xstat(entry, &buf);
472          if (S_ISREG(buf.st_mode)) {
473            struct DirListEntry * p = xmalloc(sizeof(struct DirListEntry));
474            p->name = entry_wrt_root;
475            p->next = head;
476            head = p;
477          }
478          else {
479            fatal("refusing to follow symbolic link: %s", entry);
480          }
481        }
482    #endif
483      else {      else {
484        fatal("unknown file type: %s", entry);        fatal("unknown file type: %s", entry);
485      }      }
# Line 338  Line 502 
502      list = next;      list = next;
503    }    }
504  }  }
505    
506    #ifndef HAVE_STRNDUP
507    char * strndup(const char * s, size_t size) {
508      size_t length = strlen(s);
509      if (length > size) {
510        char * result = xmalloc(size + 1);
511        strncpy(result, s, size);
512        result[size] = '\0';
513        return result;
514      }
515      else {
516        char * result = xmalloc(length + 1);
517        strcpy(result, s);
518        return result;
519      }
520    }
521    #endif
522    
523    #ifndef HAVE_VASPRINTF
524    int vasprintf(char ** s, const char * template, va_list a) {
525      int size = 100;
526      *s = malloc(size);
527      if (*s == NULL) {
528        return -1;
529      }
530    
531      va_list copy;
532      va_copy(copy, a);
533      int result = vsnprintf(*s, size, template, copy);
534      if (result >= size) {
535        int new_size = result;
536        if (new_size == INT_MAX) {
537          free(*s);
538          return -1;
539        }
540        new_size++;
541        char * new_s = realloc(*s, new_size);
542        if (new_s == NULL) {
543          free(*s);
544          return -1;
545        }
546        *s = new_s;
547        size = new_size;
548        va_copy(copy, a);
549        result = vsnprintf(*s, size, template, copy);
550        assert(result == size - 1);
551      }
552      else if (result == -1) {
553        while (result == -1) {
554          if (size == INT_MAX) {
555            free(*s);
556            return -1;
557          }
558          int new_size;
559          if (size > INT_MAX / 2) {
560            new_size = INT_MAX;
561          }
562          else {
563            new_size = 2 * size;
564          }
565          char * new_s = realloc(*s, new_size);
566          if (new_s == NULL) {
567            free(*s);
568            return -1;
569          }
570          *s = new_s;
571          size = new_size;
572          va_copy(copy, a);
573          result = vsnprintf(*s, size, template, copy);
574        }
575        assert(result <= size - 1);
576      }
577    
578      return result;
579    }
580    #endif
581    
582    #ifndef HAVE_ASPRINTF
583    int asprintf(char ** s, const char * template, ...) {
584      va_list a;
585      va_start(a, template);
586      int result = vasprintf(s, template, a);
587      va_end(a);
588      return result;
589    }
590    #endif

Legend:
Removed from v.87  
changed lines
  Added in v.448

  ViewVC Help
Powered by ViewVC 1.1.24