micro-utils/src/findutils/xargs/xargs.c

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
#include <sys/wait.h>
#include "config.h"

char t_flag;
char r_flag;
int n_flag;
int s_flag;
int P_flag;
int P_flag2;

int args;
char *cmd[NARGS + 1];

enum {
	NONE,
	QUOTE
};

void clear_cmd(void) {
	for (int i = 0; i < args; i++)
		if (cmd[i] != NULL)
			free(cmd[i]);

	args = 0;
}

int add_arg(const char *str, int chars) {
	if (args >= NARGS)
		return 1;

	else if (n_flag > 0 && args > n_flag)
		return 1;

	else if (s_flag > 0 && chars > s_flag)
		return 1;

	cmd[args] = strdup(str);
	args++;

	return 0;
}

int stdin_read(void) {
	char arg[ARG_SIZE + 1];

	/* Word start */
	char *p = arg;

	int chars = 0;
	int args_passed = 0;

	int ret = 0;
	char flag = NONE;
	while (1) {
		int c = getchar();
		if (c == EOF) {
			if (flag == QUOTE) {
				fprintf(stderr, "xargs: unterminated double quote\n");
				ret = 2;
			}

			if (p != arg)
				goto PUSH;

			ret = 1;
			break;
		}

		/* Place str */
		else if ((flag == NONE && isspace(c) && strlen(arg) > 0) || p + 1 == arg + sizeof(arg)) {
PUSH:
			*p = '\0';
			p = arg;

			if (add_arg(arg, chars))
				break;

			args_passed++;
		}

		/* Add char to str */
		else {
			if (c == '"' || c == '\'') {
				if (flag == QUOTE)
					flag = NONE;

				else
					flag = QUOTE;

				continue;
			}

			*p = c;
			if (p + 1 == arg + sizeof(arg))
				continue;

			p++;
			chars++;
		}
	}

	if (args_passed == 0)
		ret = 1;

	return ret;
}

int run(void) {
	if (t_flag) {
		for (int i = 0; i < args; i++)
			fprintf(stderr, "%s ", cmd[i]);

		fputc('\n', stderr);
	}

	pid_t pid;
	if ((pid = fork()) == 0) {
		execvp(cmd[0], cmd);
		fprintf(stderr, "xargs: %s\n", strerror(errno));
		exit(1);
	}

	int status = 0;
	waitpid(pid, &status, 0);

	if (status == 255)
		status = 124;

	else if (status >= 0x180)
		status = 125;

	return status;
}

int spawn(void) {
	if (P_flag == 0)
		return run();

	return 0;
}

void check(int val) {
	if (val <= 0) {
		fprintf(stderr, "xargs: invalid number: %s\n", optarg);
		exit(1);
	}
}

int main(int argc, char **argv) {

	int opt;
	while ((opt = getopt(argc, argv, "tn:s:rP:")) != -1) {
		switch (opt) {
			case 't':
				t_flag = 1;
				break;

			case 'n':
				n_flag = atoi(optarg);
				check(n_flag);
				break;

			case 's':
				s_flag = atoi(optarg);
				check(s_flag);
				break;

			case 'r':
				r_flag = 1;
				break;

			case 'P':
				P_flag = atoi(optarg);
				check(P_flag);
				break;

			default:
				printf("xargs [tnsr] [cmd [arg1] [arg2...]\n\t-t Print the command before start\n\t-n Pass no more than N args\n\t-r Don't run command if input is empty\n\t-s Pass command line of no more than N bytes\n\t-P Run up to N PROGs in parallel\n");
				return 0;
		}
	}

	argv += optind;
	argc -= optind;

	int ret = 0;
	while (1) {

		/* Arg */
		if (argc) {
			for (int i = 0; i < argc; i++)
				if (add_arg(argv[i], 0))
					break;
		}

		else
			add_arg(ECHO_CMD, 0);

		/* Stdin */
		int stdin_stat = stdin_read();

		/* Check NULL */
		for (int i = 0; i < args; i++) {
			if (cmd[i] == NULL) {
				fprintf(stderr, "xargs: strdup failed\n");
				clear_cmd();
				return 1;
			}
		}

		/* Run */
		if (stdin_stat == 1) {
			if (!r_flag)
				ret = spawn();
		}

		else if (stdin_stat == 0)
			ret = spawn();

		/* Clear */
		clear_cmd();
		if (stdin_stat)
			break;
	}

	return ret;
}