433567b7745608e6c0822e8991964acaa3d080cb
[oweals/busybox.git] / xargs.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini xargs implementation for busybox
4  *
5  * Copyright (C) 2000 by Lineo, inc.
6  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
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 GNU
16  * General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21  *
22  */
23
24 #include "busybox.h"
25 #include <stdlib.h>
26 #include <stdio.h>
27 #include <string.h>
28 #include <errno.h>
29 #include <getopt.h>
30 #include <ctype.h>
31 #include <sys/types.h>
32 #include <sys/wait.h>
33
34
35 int xargs_main(int argc, char **argv)
36 {
37         char *in_from_stdin = NULL;
38         char *args = NULL;
39         char *cmd_to_be_executed = NULL;
40         char traceflag = 0;
41         int len_args=0, len_cmd_to_be_executed, opt;
42         pid_t pid;
43         int wpid, status;
44
45         /* Note that we do not use getopt here, since
46          * we only want to interpret initial options,
47          * not options passed to commands */
48         while (--argc && **(++argv) == '-') {
49                 while (*++(*argv)) {
50                         switch (**argv) {
51                                 case 't':
52                                         traceflag=1;
53                                         break;
54                                 default:
55                                         fatalError(xargs_usage);
56                         }
57                 }
58         }
59
60         /* Store the command to be executed (taken from the command line) */
61         if (argc == 0) {
62                 len_cmd_to_be_executed=6;
63                 cmd_to_be_executed = xmalloc(len_cmd_to_be_executed);
64                 strcat(cmd_to_be_executed, "echo");
65         } else {
66                 opt=strlen(*argv);
67                 len_cmd_to_be_executed = (opt > 10)? opt : 10;
68                 cmd_to_be_executed = xcalloc(len_cmd_to_be_executed, sizeof(char));
69                 strcat(cmd_to_be_executed, *argv);
70         }
71
72         //args=xrealloc(args, len_args);
73 //      strcpy(args, " ");
74
75         /* Now, read in one line at a time from stdin, and store this 
76          * line to be used later as an argument to the command */
77         in_from_stdin = get_line_from_file(stdin);
78         for (;in_from_stdin!=NULL;) {
79                 char *tmp;
80                 opt = strlen(in_from_stdin);
81                 len_args += opt + 3;
82                 args=xrealloc(args, len_args);
83
84                 /* Strip out the final \n */
85                 in_from_stdin[opt-1]=' ';
86
87                 /* Replace any tabs with spaces */
88                 while( (tmp = strchr(in_from_stdin, '\t')) != NULL )
89                         *tmp=' ';
90
91                 /* Strip out any extra intra-word spaces */
92                 while( (tmp = strstr(in_from_stdin, "  ")) != NULL ) {
93                         opt = strlen(in_from_stdin);
94                         memmove(tmp, tmp+1, opt-(tmp-in_from_stdin));
95                 }
96
97                 /* trim trailing whitespace */
98                 opt = strlen(in_from_stdin) - 1;
99                 while (isspace(in_from_stdin[opt]))
100                         opt--;
101                 in_from_stdin[++opt] = 0;
102
103                 /* Strip out any leading whitespace */
104                 tmp=in_from_stdin;
105                 while(isspace(*tmp))
106                         tmp++;
107
108                 strcat(args, tmp);
109                 strcat(args, " ");
110
111                 free(in_from_stdin);
112                 in_from_stdin = get_line_from_file(stdin);
113         }
114
115         if ((pid = fork()) == 0) {
116                 char *cmd[255];
117                 int i=1;
118
119                 if (traceflag==1) {
120                         fprintf(stderr, "%s ", cmd_to_be_executed);
121                 }
122                 cmd[0] = cmd_to_be_executed;
123                 while (--argc && ++argv && *argv ) {
124                         if (traceflag==1) {
125                                 fprintf(stderr, "%s ", *argv);
126                         }
127                         cmd[i++]=*argv;
128                 }
129                 if (traceflag==1) {
130                         fprintf(stderr, "%s\n", args);
131                 }
132                 cmd[i++] = args;
133                 cmd[i] = NULL;
134                 execvp(cmd_to_be_executed, cmd);
135
136                 /* What?  Still here?  Exit with an error */
137                 fatalError("%s: %s\n", cmd_to_be_executed, strerror(errno));
138         }
139         /* Wait for a child process to exit */
140         wpid = wait(&status);
141
142
143 #ifdef BB_FEATURE_CLEAN_UP
144         free(args);
145         free(cmd_to_be_executed);
146 #endif
147
148         if (wpid > 0)
149                 return (WEXITSTATUS(status));
150         else 
151                 return EXIT_FAILURE;
152 }
153 /*
154 Local Variables:
155 c-file-style: "linux"
156 c-basic-offset: 4
157 tab-width: 4
158 End:
159 */