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