C   49

xargs.c

Guest on 5th May 2022 10:51:31 PM

  1. /*-
  2.  * Copyright (c) The Regents of the University of California.
  3.  * All rights reserved.
  4.  *
  5.  * This code is derived from software contributed to Berkeley by
  6.  * John B. Roll Jr.
  7.  *
  8.  * Redistribution and use in source and binary forms, with or without
  9.  * modification, are permitted provided that the following conditions
  10.  * are met:
  11.  * 1. Redistributions of source code must retain the above copyright
  12.  *    notice, this list of conditions and the following disclaimer.
  13.  * 2. Redistributions in binary form must reproduce the above copyright
  14.  *    notice, this list of conditions and the following disclaimer in the
  15.  *    documentation and/or other materials provided with the distribution.
  16.  * 3. All advertising materials mentioning features or use of this software
  17.  *    must display the following acknowledgement:
  18.  *      This product includes software developed by the University of
  19.  *      California, Berkeley and its contributors.
  20.  * 4. Neither the name of the University nor the names of its contributors
  21.  *    may be used to endorse or promote products derived from this software
  22.  *    without specific prior written permission.
  23.  *
  24.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34.  * SUCH DAMAGE.
  35.  */
  36.  
  37. #ifndef lint
  38. char copyright[] =
  39. "@(#) Copyright (c) 1990 The Regents of the University of California.\n\
  40. All rights reserved.\n";
  41. #endif /* not lint */
  42.  
  43. #ifndef lint
  44. #if 0
  45. static char sccsid[] = "@(#)xargs.c     5.11 (Berkeley) 6/19/91";
  46. #endif
  47. #endif /* not lint */
  48.  
  49. #include <sys/types.h>
  50. #include <sys/wait.h>
  51. #include <errno.h>
  52. #include <stdio.h>
  53. #include <stdlib.h>
  54. #include <string.h>
  55. #include <unistd.h>
  56. #include <limits.h>
  57. #include <fcntl.h>
  58. #include <stdarg.h>
  59. #if __minix
  60. #define _PATH_ECHO      "/bin/echo"
  61. #else
  62. #include "pathnames.h"
  63. #endif
  64.  
  65. #ifndef ARG_MAX
  66. #define ARG_MAX         (sizeof(int) == 2 ? 4096 : 128 * 1024)
  67. #endif
  68.  
  69. int exit_status = 0;
  70. int tflag;
  71. void err(const char *, ...);
  72. void run(char **argv);
  73. void usage(void);
  74.  
  75. int main(int argc, char **argv)
  76. {
  77.         extern int optind;
  78.         extern char *optarg;
  79.         register int ch;
  80.         register char *p, *bbp, *ebp, **bxp, **exp, **xp;
  81.         int cnt, indouble, insingle, nargs, nflag, nline, xflag, zflag;
  82.         char **av, *argp;
  83.  
  84.         /*
  85.          * POSIX.2 limits the exec line length to ARG_MAX - 2K.  Running that
  86.          * caused some E2BIG errors, so it was changed to ARG_MAX - 4K.  Given
  87.          * that the smallest argument is 2 bytes in length, this means that
  88.          * the number of arguments is limited to:
  89.          *
  90.          *       (ARG_MAX - 4K - LENGTH(utility + arguments)) / 2.
  91.          *
  92.          * We arbitrarily limit the number of arguments to 5000.  This is
  93.          * allowed by POSIX.2 as long as the resulting minimum exec line is
  94.          * at least LINE_MAX.  Realloc'ing as necessary is possible, but
  95.          * probably not worthwhile.
  96.          */
  97. #if !__minix || __minix_vmd
  98.         nargs = 5000;
  99.         nline = ARG_MAX - 4 * 1024;
  100. #else
  101.         /* Things are more cramped under standard Minix. */
  102.         nargs = 80 * sizeof(int);
  103.         nline = ARG_MAX - 512 * sizeof(int);
  104. #endif
  105.         nflag = xflag = zflag = 0;
  106.         while ((ch = getopt(argc, argv, "n:s:tx0")) != EOF)
  107.                 switch(ch) {
  108.                 case 'n':
  109.                         nflag = 1;
  110.                         if ((nargs = atoi(optarg)) <= 0)
  111.                                 err("illegal argument count");
  112.                         break;
  113.                 case 's':
  114.                         nline = atoi(optarg);
  115.                         break;
  116.                 case 't':
  117.                         tflag = 1;
  118.                         break;
  119.                 case 'x':
  120.                         xflag = 1;
  121.                         break;
  122.                 case '0':
  123.                         zflag = 1;
  124.                         break;
  125.                 case '?':
  126.                 default:
  127.                         usage();
  128.         }
  129.         argc -= optind;
  130.         argv += optind;
  131.  
  132.         if (xflag && !nflag)
  133.                 usage();
  134.  
  135.         /*
  136.          * Allocate pointers for the utility name, the utility arguments,
  137.          * the maximum arguments to be read from stdin and the trailing
  138.          * NULL.
  139.          */
  140.         if (!(av = bxp =
  141.             malloc((u_int)(1 + argc + nargs + 1) * sizeof(char **))))
  142.                 err("%s", strerror(errno));
  143.  
  144.         /*
  145.          * Use the user's name for the utility as argv[0], just like the
  146.          * shell.  Echo is the default.  Set up pointers for the user's
  147.          * arguments.
  148.          */
  149.         if (!*argv)
  150.                 cnt = strlen(*bxp++ = _PATH_ECHO);
  151.         else {
  152.                 cnt = 0;
  153.                 do {
  154.                         cnt += strlen(*bxp++ = *argv) + 1;
  155.                 } while (*++argv);
  156.         }
  157.  
  158.         /*
  159.          * Set up begin/end/traversing pointers into the array.  The -n
  160.          * count doesn't include the trailing NULL pointer, so the malloc
  161.          * added in an extra slot.
  162.          */
  163.         exp = (xp = bxp) + nargs;
  164.  
  165.         /*
  166.          * Allocate buffer space for the arguments read from stdin and the
  167.          * trailing NULL.  Buffer space is defined as the default or specified
  168.          * space, minus the length of the utility name and arguments.  Set up
  169.          * begin/end/traversing pointers into the array.  The -s count does
  170.          * include the trailing NULL, so the malloc didn't add in an extra
  171.          * slot.
  172.          */
  173.         nline -= cnt;
  174.         if (nline <= 0)
  175.                 err("insufficient space for command");
  176.  
  177.         if (!(bbp = malloc((u_int)nline + 1)))
  178.                 err("%s", strerror(errno));
  179.         ebp = (argp = p = bbp) + nline - 1;
  180.  
  181.         if (zflag) {
  182.                 /* Read pathnames terminated by null bytes as produced by
  183.                  * find ... -print0.  No comments in this code, see further
  184.                  * below.
  185.                  */
  186.                 for (;;)
  187.                         switch(ch = getchar()) {
  188.                         case EOF:
  189.                                 if (p == bbp)
  190.                                         exit(exit_status);
  191.  
  192.                                 if (argp == p) {
  193.                                         *xp = NULL;
  194.                                         run(av);
  195.                                         exit(exit_status);
  196.                                 }
  197.                                 /*FALL THROUGH*/
  198.                         case '\0':
  199.                                 if (argp == p)
  200.                                         continue;
  201.  
  202.                                 *p = '\0';
  203.                                 *xp++ = argp;
  204.  
  205.                                 if (xp == exp || p == ebp || ch == EOF) {
  206.                                         if (xflag && xp != exp && p == ebp)
  207.                                                 err(
  208.                                            "insufficient space for arguments");
  209.                                         *xp = NULL;
  210.                                         run(av);
  211.                                         if (ch == EOF)
  212.                                                 exit(exit_status);
  213.                                         p = bbp;
  214.                                         xp = bxp;
  215.                                 } else
  216.                                         ++p;
  217.                                 argp = p;
  218.                                 break;
  219.                         default:
  220.                                 if (p < ebp) {
  221.                                         *p++ = ch;
  222.                                         break;
  223.                                 }
  224.  
  225.                                 if (bxp == xp)
  226.                                         err("insufficient space for argument");
  227.                                 if (xflag)
  228.                                         err("insufficient space for arguments");
  229.  
  230.                                 *xp = NULL;
  231.                                 run(av);
  232.                                 xp = bxp;
  233.                                 cnt = ebp - argp;
  234.                                 bcopy(argp, bbp, cnt);
  235.                                 p = (argp = bbp) + cnt;
  236.                                 *p++ = ch;
  237.                                 break;
  238.                         }
  239.                 /* NOTREACHED */
  240.         }
  241.  
  242.         for (insingle = indouble = 0;;)
  243.                 switch(ch = getchar()) {
  244.                 case EOF:
  245.                         /* No arguments since last exec. */
  246.                         if (p == bbp)
  247.                                 exit(exit_status);
  248.  
  249.                         /* Nothing since end of last argument. */
  250.                         if (argp == p) {
  251.                                 *xp = NULL;
  252.                                 run(av);
  253.                                 exit(exit_status);
  254.                         }
  255.                         goto arg1;
  256.                 case ' ':
  257.                 case '\t':
  258.                         /* Quotes escape tabs and spaces. */
  259.                         if (insingle || indouble)
  260.                                 goto addch;
  261.                         goto arg2;
  262.                 case '\n':
  263.                         /* Empty lines are skipped. */
  264.                         if (argp == p)
  265.                                 continue;
  266.  
  267.                         /* Quotes do not escape newlines. */
  268. arg1:                   if (insingle || indouble)
  269.                                  err("unterminated quote");
  270.  
  271. arg2:                   *p = '\0';
  272.                         *xp++ = argp;
  273.  
  274.                         /*
  275.                          * If max'd out on args or buffer, or reached EOF,
  276.                          * run the command.  If xflag and max'd out on buffer
  277.                          * but not on args, object.
  278.                          */
  279.                         if (xp == exp || p == ebp || ch == EOF) {
  280.                                 if (xflag && xp != exp && p == ebp)
  281.                                         err("insufficient space for arguments");
  282.                                 *xp = NULL;
  283.                                 run(av);
  284.                                 if (ch == EOF)
  285.                                         exit(exit_status);
  286.                                 p = bbp;
  287.                                 xp = bxp;
  288.                         } else
  289.                                 ++p;
  290.                         argp = p;
  291.                         break;
  292.                 case '\'':
  293.                         if (indouble)
  294.                                 goto addch;
  295.                         insingle = !insingle;
  296.                         break;
  297.                 case '"':
  298.                         if (insingle)
  299.                                 goto addch;
  300.                         indouble = !indouble;
  301.                         break;
  302.                 case '\\':
  303.                         /* Backslash escapes anything, is escaped by quotes. */
  304.                         if (!insingle && !indouble && (ch = getchar()) == EOF)
  305.                                 err("backslash at EOF");
  306.                         /* FALLTHROUGH */
  307.                 default:
  308. addch:                  if (p < ebp) {
  309.                                 *p++ = ch;
  310.                                 break;
  311.                         }
  312.  
  313.                         /* If only one argument, not enough buffer space. */
  314.                         if (bxp == xp)
  315.                                 err("insufficient space for argument");
  316.                         /* Didn't hit argument limit, so if xflag object. */
  317.                         if (xflag)
  318.                                 err("insufficient space for arguments");
  319.  
  320.                         *xp = NULL;
  321.                         run(av);
  322.                         xp = bxp;
  323.                         cnt = ebp - argp;
  324.                         bcopy(argp, bbp, cnt);
  325.                         p = (argp = bbp) + cnt;
  326.                         *p++ = ch;
  327.                         break;
  328.                 }
  329.         /* NOTREACHED */
  330. }
  331.  
  332. void run(char **argv)
  333. {
  334.         register char **p;
  335.         pid_t pid;
  336.         int noinvoke;
  337.         int status;
  338.         int pfd[2];
  339.  
  340.         if (tflag) {
  341.                 (void)fprintf(stderr, "%s", *argv);
  342.                 for (p = argv + 1; *p; ++p)
  343.                         (void)fprintf(stderr, " %s", *p);
  344.                 (void)fprintf(stderr, "\n");
  345.                 (void)fflush(stderr);
  346.         }
  347.         if (pipe(pfd) < 0) err("pipe: %s", strerror(errno));
  348.  
  349.         switch(pid = fork()) {
  350.         case -1:
  351.                 err("fork: %s", strerror(errno));
  352.         case 0:
  353.                 close(pfd[0]);
  354.                 fcntl(pfd[1], F_SETFD, fcntl(pfd[1], F_GETFD) | FD_CLOEXEC);
  355.  
  356.                 execvp(argv[0], argv);
  357.                 noinvoke = (errno == ENOENT) ? 127 : 126;
  358.                 (void)fprintf(stderr,
  359.                     "xargs: %s exec failed: %s.\n", argv[0], strerror(errno));
  360.  
  361.                 /* Modern way of returning noinvoke instead of a dirty vfork()
  362.                  * trick:                                       (kjb)
  363.                  */
  364.                 write(pfd[1], &noinvoke, sizeof(noinvoke));
  365.                 _exit(-1);
  366.         }
  367.         close(pfd[1]);
  368.         if (read(pfd[0], &noinvoke, sizeof(noinvoke)) < sizeof(noinvoke))
  369.                 noinvoke = 0;
  370.         close(pfd[0]);
  371.  
  372.         pid = waitpid(pid, &status, 0);
  373.         if (pid == -1)
  374.                 err("waitpid: %s", strerror(errno));
  375.  
  376.         /*
  377.          * If we couldn't invoke the utility or the utility didn't exit
  378.          * properly, quit with 127 or 126 respectively.
  379.          */
  380.         if (noinvoke)
  381.                 exit(noinvoke);
  382.  
  383.         /*
  384.          * According to POSIX, we have to exit if the utility exits with
  385.          * a 255 status, or is interrupted by a signal.   xargs is allowed
  386.          * to return any exit status between 1 and 125 in these cases, but
  387.          * we'll use 124 and 125, the same values used by GNU xargs.
  388.          */
  389.         if (WIFEXITED(status)) {
  390.                 if (WEXITSTATUS (status) == 255) {
  391.                         fprintf (stderr, "xargs: %s exited with status 255\n",
  392.                                  argv[0]);
  393.                         exit(124);
  394.                 } else if (WEXITSTATUS (status) != 0) {
  395.                         exit_status = 123;
  396.                 }
  397.         } else if (WIFSTOPPED (status)) {
  398.                 fprintf (stderr, "xargs: %s terminated by signal %d\n",
  399.                          argv[0], WSTOPSIG (status));
  400.                 exit(125);
  401.         } else if (WIFSIGNALED (status)) {
  402.                 fprintf (stderr, "xargs: %s terminated by signal %d\n",
  403.                          argv[0], WTERMSIG (status));
  404.                 exit(125);
  405.         }
  406. }
  407.  
  408. void usage(void)
  409. {
  410.         (void)fprintf(stderr,
  411. "usage: xargs [-t0] [[-x] -n number] [-s size] [utility [argument ...]]\n");
  412.         exit(1);
  413. }
  414.  
  415. void err(const char *fmt, ...)
  416. {
  417.         va_list ap;
  418.  
  419.         va_start(ap, fmt);
  420.         (void)fprintf(stderr, "xargs: ");
  421.         (void)vfprintf(stderr, fmt, ap);
  422.         va_end(ap);
  423.         (void)fprintf(stderr, "\n");
  424.         exit(1);
  425.         /* NOTREACHED */
  426. }

Raw Paste


Login or Register to edit or fork this paste. It's free.