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

Annotation of /trunk/util.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 370 - (hide annotations)
Mon Oct 27 20:34:27 2008 UTC (11 years, 1 month ago) by siliconforks
File MIME type: text/plain
File size: 12921 byte(s)
Make error messages uniform.
1 siliconforks 2 /*
2     util.c - general purpose utility routines
3 siliconforks 87 Copyright (C) 2007, 2008 siliconforks.com
4 siliconforks 2
5     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
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9    
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13     GNU General Public License for more details.
14    
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18     */
19    
20 siliconforks 106 #define _GNU_SOURCE
21    
22     #include <config.h>
23    
24 siliconforks 2 #include "util.h"
25    
26     #include <assert.h>
27     #include <errno.h>
28 siliconforks 99 #include <limits.h>
29 siliconforks 2 #include <stdarg.h>
30     #include <stdio.h>
31 siliconforks 106 #include <stdint.h>
32 siliconforks 2 #include <string.h>
33     #include <strings.h>
34    
35     #include <dirent.h>
36     #include <libgen.h>
37     #include <sys/stat.h>
38     #include <sys/types.h>
39     #include <unistd.h>
40    
41     const char * program = NULL;
42    
43     void fatal(const char * format, ...) {
44     fprintf(stderr, "%s: ", program);
45     va_list ap;
46     va_start(ap, format);
47     vfprintf(stderr, format, ap);
48     va_end(ap);
49     fputc('\n', stderr);
50 siliconforks 191 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 siliconforks 2 fprintf(stderr, "Try `%s --help' for more information.\n", program);
61     exit(EXIT_FAILURE);
62     }
63    
64 siliconforks 370 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 siliconforks 311 void version(void) {
75     printf("%s %s\n", program, VERSION);
76     printf("Character encoding support: ");
77     #if HAVE_ICONV
78     printf("iconv\n");
79     #elif HAVE_MULTIBYTETOWIDECHAR
80     printf("MultiByteToWideChar\n");
81     #else
82     printf("none\n");
83     #endif
84     exit(EXIT_SUCCESS);
85     }
86    
87 siliconforks 106 size_t addst(size_t x, size_t y) {
88     if (SIZE_MAX - x < y) {
89     fatal("integer overflow");
90     }
91     return x + y;
92     }
93    
94     size_t mulst(size_t x, size_t y) {
95     if (x == 0 || y == 0) {
96     return 0;
97     }
98     if (SIZE_MAX / x < y) {
99     fatal("integer overflow");
100     }
101     return x * y;
102     }
103    
104 siliconforks 2 void * xmalloc(size_t size) {
105     void * result = malloc(size);
106     if (result == NULL) {
107     fatal("out of memory");
108     }
109     return result;
110     }
111    
112 siliconforks 91 void * xrealloc(void * p, size_t size) {
113     void * result = realloc(p, size);
114     if (result == NULL) {
115     fatal("out of memory");
116     }
117     return result;
118     }
119    
120 siliconforks 2 char * xstrdup(const char * s) {
121     char * result = strdup(s);
122     if (result == NULL) {
123     fatal("out of memory");
124     }
125     return result;
126     }
127    
128 siliconforks 106 char * xstrndup(const char * s, size_t size) {
129     char * result = strndup(s, size);
130     if (result == NULL) {
131     fatal("out of memory");
132     }
133     return result;
134     }
135    
136     int xasprintf(char ** s, const char * template, ...) {
137     va_list a;
138     va_start(a, template);
139     int result = vasprintf(s, template, a);
140     va_end(a);
141     if (result < 0) {
142     fatal("out of memory");
143     }
144     return result;
145     }
146    
147 siliconforks 2 char * xgetcwd(void) {
148     char * result = getcwd(NULL, 0);
149     if (result == NULL) {
150     fatal("out of memory");
151     }
152     return result;
153     }
154    
155     FILE * xfopen(const char * file, const char * mode) {
156     FILE * result = fopen(file, mode);
157     if (result == NULL) {
158     fatal("cannot open file: %s", file);
159     }
160     return result;
161     }
162    
163     DIR * xopendir(const char * directory) {
164     DIR * result = opendir(directory);
165     if (result == NULL) {
166     fatal("cannot open directory: %s", directory);
167     }
168     return result;
169     }
170    
171     void xlstat(const char * file, struct stat * buf) {
172     #ifdef _WIN32
173     return xstat(file, buf);
174     #else
175     if (lstat(file, buf) == -1) {
176     fatal("cannot stat file: %s", file);
177     }
178     #endif
179     }
180    
181     void xstat(const char * file, struct stat * buf) {
182     if (stat(file, buf) == -1) {
183     fatal("cannot stat file: %s", file);
184     }
185     }
186    
187     void xmkdir(const char * directory) {
188     int result;
189     #ifdef _WIN32
190     result = mkdir(directory);
191     #else
192     result = mkdir(directory, 0755);
193     #endif
194     if (result == -1) {
195     fatal("cannot create directory: %s", directory);
196     }
197     }
198    
199     void mkdir_if_necessary(const char * directory) {
200     struct stat buf;
201     if (stat(directory, &buf) == 0) {
202     if (! S_ISDIR(buf.st_mode)) {
203     fatal("not a directory: %s", directory);
204     }
205     }
206     else {
207     if (errno == ENOENT) {
208     xmkdir(directory);
209     }
210     else {
211     fatal("cannot stat directory: %s", directory);
212     }
213     }
214     }
215    
216     void mkdirs(const char * directory) {
217     char * d = xmalloc(strlen(directory) + 1);
218     for (const char * p = directory; *p != '\0'; p++) {
219     if (*p == '/' && p > directory) {
220     strncpy(d, directory, p - directory);
221     d[p - directory] = '\0';
222     mkdir_if_necessary(d);
223     }
224     }
225     mkdir_if_necessary(directory);
226     free(d);
227     }
228    
229     void xchdir(const char * directory) {
230     if (chdir(directory) == -1) {
231     fatal("cannot change directory: %s", directory);
232     }
233     }
234    
235 siliconforks 106 bool str_starts_with(const char * string, const char * prefix) {
236     const char * string_ptr = string;
237     const char * prefix_ptr = prefix;
238     while (*string_ptr != '\0' && *prefix_ptr != '\0') {
239     if (*string_ptr != *prefix_ptr) {
240     return false;
241     }
242     string_ptr++;
243     prefix_ptr++;
244     }
245     if (*string_ptr == '\0' && *prefix_ptr != '\0') {
246     return false;
247     }
248     return true;
249     }
250    
251     bool str_ends_with(const char * string, const char * suffix) {
252     size_t string_length = strlen(string);
253     size_t suffix_length = strlen(suffix);
254     if (string_length < suffix_length) {
255     return false;
256     }
257     return strcmp(string + string_length - suffix_length, suffix) == 0;
258     }
259    
260 siliconforks 2 char * make_path(const char * parent, const char * relative_path) {
261     size_t parent_length = strlen(parent);
262     size_t relative_path_length = strlen(relative_path);
263 siliconforks 106 size_t result_length = addst(parent_length, relative_path_length);
264     result_length = addst(result_length, 2);
265     char * result = xmalloc(result_length);
266 siliconforks 2 strcpy(result, parent);
267     result[parent_length] = '/';
268     strcpy(result + parent_length + 1, relative_path);
269     return result;
270     }
271    
272     char * make_canonical_path(const char * relative_path) {
273     char * original_directory = xgetcwd();
274     char * base = make_basename(relative_path);
275     char * dir = make_dirname(relative_path);
276    
277     xchdir(dir);
278     char * canonical_dir = xgetcwd();
279     char * result = make_path(canonical_dir, base);
280    
281     free(canonical_dir);
282     free(base);
283     free(dir);
284     xchdir(original_directory);
285     free(original_directory);
286    
287     return result;
288     }
289    
290     char * make_basename(const char * path) {
291     char * copy = xstrdup(path);
292     char * result = xstrdup(basename(copy));
293     free(copy);
294     return result;
295     }
296    
297     char * make_dirname(const char * path) {
298     char * copy = xstrdup(path);
299     char * result = xstrdup(dirname(copy));
300     free(copy);
301     return result;
302     }
303    
304     int is_same_file(const char * file1, const char * file2) {
305     #ifdef _WIN32
306     #define FILECMP strcasecmp
307     #else
308     #define FILECMP strcmp
309     #endif
310     if (FILECMP(file1, file2) == 0) {
311     return 1;
312     }
313    
314     char * canonical1 = make_canonical_path(file1);
315     char * canonical2 = make_canonical_path(file2);
316     int cmp = FILECMP(canonical1, canonical2);
317     free(canonical1);
318     free(canonical2);
319     if (cmp == 0) {
320     return 1;
321     }
322    
323     #ifndef _WIN32
324     struct stat buf1;
325     if (stat(file1, &buf1) == -1) {
326     if (errno == ENOENT) {
327     return 0;
328     }
329     else {
330     fatal("cannot stat file: %s", file1);
331     }
332     }
333     struct stat buf2;
334     if (stat(file2, &buf2) == -1) {
335     if (errno == ENOENT) {
336     return 0;
337     }
338     else {
339     fatal("cannot stat file: %s", file2);
340     }
341     }
342     if (buf1.st_dev == buf2.st_dev &&
343     buf1.st_ino == buf2.st_ino) {
344     return 1;
345     }
346     #endif
347     return 0;
348     #undef FILECMP
349     }
350    
351     int contains_file(const char * file1, const char * file2) {
352     int result = 0;
353     char * ancestor = make_canonical_path(file1);
354     char * d = make_canonical_path(file2);
355     char * parent = make_dirname(d);
356     while (strcmp(d, parent) != 0) {
357     if (is_same_file(ancestor, parent)) {
358     result = 1;
359     break;
360     }
361     free(d);
362     d = parent;
363     parent = make_dirname(d);
364     }
365     free(d);
366     free(parent);
367     free(ancestor);
368     return result;
369     }
370    
371     void copy_stream(FILE * source, FILE * destination) {
372     unsigned char buffer[8192];
373     for (;;) {
374     int bytes_read = fread(buffer, 1, sizeof(buffer), source);
375     if (bytes_read == 0) {
376     break;
377     }
378     fwrite(buffer, 1, bytes_read, destination);
379     }
380     }
381    
382     void copy_file(const char * source_file, const char * destination_file) {
383     FILE * source = xfopen(source_file, "rb");
384     FILE * destination = xfopen(destination_file, "wb");
385    
386     copy_stream(source, destination);
387    
388 siliconforks 359 #ifndef _WIN32
389     /* copy permissions */
390     struct stat buf;
391     if (fstat(fileno(source), &buf) == -1) {
392     fatal("cannot stat file: %s", source_file);
393     }
394     fchmod(fileno(destination), buf.st_mode);
395     #endif
396    
397 siliconforks 2 fclose(source);
398     fclose(destination);
399     }
400    
401 siliconforks 106 bool directory_is_empty(const char * directory) {
402     bool result = true;
403 siliconforks 2 DIR * dir = xopendir(directory);
404     struct dirent * e;
405     while ((e = readdir(dir)) != NULL) {
406     if (strcmp(e->d_name, ".") != 0 &&
407     strcmp(e->d_name, "..") != 0) {
408 siliconforks 106 result = false;
409     break;
410 siliconforks 2 }
411     }
412     closedir(dir);
413 siliconforks 106 return result;
414 siliconforks 2 }
415    
416     static struct DirListEntry * recursive_dir_list(const char * root, const char * directory_wrt_root, struct DirListEntry * head) {
417     char * directory = directory_wrt_root == NULL? xstrdup(root): make_path(root, directory_wrt_root);
418     DIR * dir = xopendir(directory);
419     struct dirent * e;
420     while ((e = readdir(dir)) != NULL) {
421     if (strcmp(e->d_name, ".") == 0 ||
422     strcmp(e->d_name, "..") == 0) {
423     continue;
424     }
425     char * entry = make_path(directory, e->d_name);
426     char * entry_wrt_root = directory_wrt_root == NULL? xstrdup(e->d_name): make_path(directory_wrt_root, e->d_name);
427     struct stat buf;
428     xlstat(entry, &buf);
429     if (S_ISREG(buf.st_mode)) {
430     struct DirListEntry * p = xmalloc(sizeof(struct DirListEntry));
431     p->name = entry_wrt_root;
432     p->next = head;
433     head = p;
434     }
435     else if (S_ISDIR(buf.st_mode)) {
436     head = recursive_dir_list(root, entry_wrt_root, head);
437     free(entry_wrt_root);
438     }
439 siliconforks 364 #ifndef _WIN32
440 siliconforks 359 else if (S_ISLNK(buf.st_mode)) {
441     /* check what it points to */
442     xstat(entry, &buf);
443     if (S_ISREG(buf.st_mode)) {
444     struct DirListEntry * p = xmalloc(sizeof(struct DirListEntry));
445     p->name = entry_wrt_root;
446     p->next = head;
447     head = p;
448     }
449     else {
450     fatal("refusing to follow symbolic link: %s", entry);
451     }
452     }
453 siliconforks 364 #endif
454 siliconforks 2 else {
455     fatal("unknown file type: %s", entry);
456     }
457     free(entry);
458     }
459     closedir(dir);
460     free(directory);
461     return head;
462     }
463    
464     struct DirListEntry * make_recursive_dir_list(const char * directory) {
465     return recursive_dir_list(directory, NULL, NULL);
466     }
467    
468     void free_dir_list(struct DirListEntry * list) {
469     while (list != NULL) {
470     struct DirListEntry * next = list->next;
471     free(list->name);
472     free(list);
473     list = next;
474     }
475     }
476 siliconforks 99
477 siliconforks 125 #ifndef HAVE_STRNDUP
478     char * strndup(const char * s, size_t size) {
479     size_t length = strlen(s);
480     if (length > size) {
481     char * result = xmalloc(size + 1);
482     strncpy(result, s, size);
483     result[size] = '\0';
484     return result;
485     }
486     else {
487     char * result = xmalloc(length + 1);
488     strcpy(result, s);
489     return result;
490     }
491 siliconforks 106 }
492     #endif
493    
494 siliconforks 99 #ifndef HAVE_VASPRINTF
495     int vasprintf(char ** s, const char * template, va_list a) {
496     int size = 100;
497     *s = malloc(size);
498     if (*s == NULL) {
499     return -1;
500     }
501    
502     va_list copy;
503     va_copy(copy, a);
504     int result = vsnprintf(*s, size, template, copy);
505     if (result >= size) {
506     int new_size = result;
507     if (new_size == INT_MAX) {
508     free(*s);
509 siliconforks 106 return -1;
510 siliconforks 99 }
511     new_size++;
512     char * new_s = realloc(*s, new_size);
513     if (new_s == NULL) {
514     free(*s);
515     return -1;
516     }
517     *s = new_s;
518     size = new_size;
519     va_copy(copy, a);
520     result = vsnprintf(*s, size, template, copy);
521     assert(result == size - 1);
522     }
523     else if (result == -1) {
524     while (result == -1) {
525 siliconforks 106 if (size == INT_MAX) {
526 siliconforks 99 free(*s);
527 siliconforks 106 return -1;
528 siliconforks 99 }
529 siliconforks 106 int new_size;
530     if (size > INT_MAX / 2) {
531     new_size = INT_MAX;
532     }
533     else {
534     new_size = 2 * size;
535     }
536 siliconforks 99 char * new_s = realloc(*s, new_size);
537     if (new_s == NULL) {
538     free(*s);
539     return -1;
540     }
541     *s = new_s;
542     size = new_size;
543     va_copy(copy, a);
544     result = vsnprintf(*s, size, template, copy);
545     }
546     assert(result <= size - 1);
547     }
548    
549     return result;
550     }
551     #endif
552    
553     #ifndef HAVE_ASPRINTF
554     int asprintf(char ** s, const char * template, ...) {
555     va_list a;
556     va_start(a, template);
557     int result = vasprintf(s, template, a);
558     va_end(a);
559     return result;
560     }
561     #endif

  ViewVC Help
Powered by ViewVC 1.1.24