1 /*
2 * Copyright 1999-2008 by Marco d'Itri <md@linux.it>.
3 *
4 * do_nofail and merge_args come from the module-init-tools package.
5 * Copyright 2001 by Rusty Russell.
6 * Copyright 2002, 2003 by Rusty Russell, IBM Corporation.
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program 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
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 */
22
23 /* for strdup */
24 #define _XOPEN_SOURCE 500
25
26 /* System library */
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <stdarg.h>
30 #include <string.h>
31 #include <errno.h>
32
33 /* Application-specific */
34 #include "utils.h"
35
do_nofail(void * ptr,const char * file,const int line)36 void *do_nofail(void *ptr, const char *file, const int line)
37 {
38 if (ptr)
39 return ptr;
40
41 err_quit("Memory allocation failure at %s:%d.", file, line);
42 }
43
44 /* Prepend options from a string. */
merge_args(char * args,char * argv[],int * argc)45 char **merge_args(char *args, char *argv[], int *argc)
46 {
47 char *arg, *argstring;
48 char **newargs = NULL;
49 unsigned int i, num_env = 0;
50
51 if (!args)
52 return argv;
53
54 argstring = NOFAIL(strdup(args));
55 for (arg = strtok(argstring, " "); arg; arg = strtok(NULL, " ")) {
56 num_env++;
57 newargs = NOFAIL(realloc(newargs,
58 sizeof(newargs[0]) * (num_env + *argc + 1)));
59 newargs[num_env] = arg;
60 }
61
62 if (!newargs)
63 return argv;
64
65 /* Append commandline args */
66 newargs[0] = argv[0];
67 for (i = 1; i <= *argc; i++)
68 newargs[num_env + i] = argv[i];
69
70 *argc += num_env;
71 return newargs;
72 }
73
74 /* Error routines */
err_sys(const char * fmt,...)75 void err_sys(const char *fmt, ...)
76 {
77 va_list ap;
78
79 va_start(ap, fmt);
80 vfprintf(stderr, fmt, ap);
81 fprintf(stderr, ": %s\n", strerror(errno));
82 va_end(ap);
83 exit(2);
84 }
85
err_quit(const char * fmt,...)86 void err_quit(const char *fmt, ...)
87 {
88 va_list ap;
89
90 va_start(ap, fmt);
91 vfprintf(stderr, fmt, ap);
92 fputs("\n", stderr);
93 va_end(ap);
94 exit(2);
95 }
96
97