logo

utils-std

Collection of commonly available Unix tools git clone https://anongit.hacktivis.me/git/utils-std.git/

paste.c (6166B)

    

  1. /*-
  2.  * SPDX-License-Identifier: BSD-3-Clause
  3.  *
  4.  * Copyright (c) 1989, 1993
  5.  *	The Regents of the University of California.  All rights reserved.
  6.  *
  7.  * This code is derived from software contributed to Berkeley by
  8.  * Adam S. Moskowitz of Menlo Consulting.
  9.  *
  10.  * Redistribution and use in source and binary forms, with or without
  11.  * modification, are permitted provided that the following conditions
  12.  * are met:
  13.  * 1. Redistributions of source code must retain the above copyright
  14.  *    notice, this list of conditions and the following disclaimer.
  15.  * 2. Redistributions in binary form must reproduce the above copyright
  16.  *    notice, this list of conditions and the following disclaimer in the
  17.  *    documentation and/or other materials provided with the distribution.
  18.  * 3. Neither the name of the University nor the names of its contributors
  19.  *    may be used to endorse or promote products derived from this software
  20.  *    without specific prior written permission.
  21.  *
  22.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  23.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  26.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32.  * SUCH DAMAGE.
  33.  */
  35. #define _POSIX_C_SOURCE 200809L
  37. #include "../libutils/err.h"
  38. #include "../libutils/getopt_nolong.h"
  40. #include <errno.h>
  41. #include <limits.h>
  42. #include <locale.h>
  43. #include <stdint.h> // SIZE_MAX
  44. #include <stdio.h>
  45. #include <stdlib.h>
  46. #include <string.h>
  47. #include <sys/types.h>
  48. #include <unistd.h>
  49. #include <wchar.h>
  51. static wchar_t *delim;
  52. static int delimcnt;
  53. static wint_t linedelim = L'\n';
  55. static wchar_t tab[] = L"\t";
  56. const char *argv0 = "paste";
  58. typedef struct _list
  59. {
  60. 	struct _list *next;
  61. 	FILE *fp;
  62. 	int cnt;
  63. 	char *name;
  64. } LIST;
  66. static int
  67. parallel(char **argv)
  68. {
  69. 	LIST *lp;
  70. 	int cnt;
  71. 	wint_t ich;
  72. 	wchar_t ch;
  73. 	char *p;
  74. 	LIST *head, *tmp;
  75. 	int opencnt, output;
  77. 	for(cnt = 0, head = tmp = NULL; (p = *argv); ++argv, ++cnt)
  78. 	{
  79. 		if((lp = malloc(sizeof(LIST))) == NULL) utils_err(1, NULL);
  80. 		if(p[0] == '-' && !p[1])
  81. 			lp->fp = stdin;
  82. 		else if(!(lp->fp = fopen(p, "r")))
  83. 			utils_err(1, "%s", p);
  84. 		lp->next = NULL;
  85. 		lp->cnt = cnt;
  86. 		lp->name = p;
  87. 		if(!head)
  88. 			head = tmp = lp;
  89. 		else
  90. 		{
  91. 			tmp->next = lp;
  92. 			tmp = lp;
  93. 		}
  94. 	}
  96. 	for(opencnt = cnt; opencnt;)
  97. 	{
  98. 		for(output = 0, lp = head; lp; lp = lp->next)
  99. 		{
  100. 			if(!lp->fp)
  101. 			{
  102. 				if(output && lp->cnt && (ch = delim[(lp->cnt - 1) % delimcnt])) putwchar(ch);
  103. 				continue;
  104. 			}
  105. 			if((ich = getwc(lp->fp)) == WEOF)
  106. 			{
  107. 				if(!--opencnt) break;
  108. 				lp->fp = NULL;
  109. 				if(output && lp->cnt && (ch = delim[(lp->cnt - 1) % delimcnt])) putwchar(ch);
  110. 				continue;
  111. 			}
  112. 			/*
  113. 			 * make sure that we don't print any delimiters
  114. 			 * unless there's a non-empty file.
  115. 			 */
  116. 			if(!output)
  117. 			{
  118. 				output = 1;
  119. 				for(cnt = 0; cnt < lp->cnt; ++cnt)
  120. 					if((ch = delim[cnt % delimcnt])) putwchar(ch);
  121. 			}
  122. 			else if((ch = delim[(lp->cnt - 1) % delimcnt]))
  123. 				putwchar(ch);
  124. 			if(ich == linedelim) continue;
  125. 			do
  126. 			{
  127. 				putwchar(ich);
  128. 			} while((ich = getwc(lp->fp)) != WEOF && ich != linedelim);
  129. 		}
  130. 		if(output) putwchar(linedelim);
  131. 	}
  133. 	return (0);
  134. }
  136. static int
  137. sequential(char **argv)
  138. {
  139. 	FILE *fp;
  140. 	int cnt, failed, needdelim;
  141. 	wint_t ch;
  142. 	char *p;
  144. 	failed = 0;
  145. 	for(; (p = *argv); ++argv)
  146. 	{
  147. 		if(p[0] == '-' && !p[1])
  148. 			fp = stdin;
  149. 		else if(!(fp = fopen(p, "r")))
  150. 		{
  151. 			utils_warn("%s", p);
  152. 			failed = 1;
  153. 			continue;
  154. 		}
  155. 		cnt = needdelim = 0;
  156. 		while((ch = getwc(fp)) != WEOF)
  157. 		{
  158. 			if(needdelim)
  159. 			{
  160. 				needdelim = 0;
  161. 				if(delim[cnt] != '\0') putwchar(delim[cnt]);
  162. 				if(++cnt == delimcnt) cnt = 0;
  163. 			}
  164. 			if(ch != linedelim)
  165. 				putwchar(ch);
  166. 			else
  167. 				needdelim = 1;
  168. 		}
  169. 		if(needdelim) putwchar(linedelim);
  170. 		if(fp != stdin) (void)fclose(fp);
  171. 	}
  173. 	return (failed != 0);
  174. }
  176. static int
  177. tr(wchar_t *arg)
  178. {
  179. 	int cnt;
  180. 	wchar_t ch, *p;
  182. 	for(p = arg, cnt = 0; (ch = *p++); ++arg, ++cnt)
  183. 		if(ch == '\\') switch(ch = *p++)
  184. 			{
  185. 			case 'n':
  186. 				*arg = '\n';
  187. 				break;
  188. 			case 't':
  189. 				*arg = '\t';
  190. 				break;
  191. 			case '0':
  192. 				*arg = '\0';
  193. 				break;
  194. 			default:
  195. 				*arg = ch;
  196. 				break;
  197. 			}
  198. 		else
  199. 			*arg = ch;
  201. 	if(!cnt) utils_errx(1, "no delimiters specified");
  202. 	return (cnt);
  203. }
  205. static void
  206. usage(void)
  207. {
  208. 	(void)fprintf(stderr, "usage: paste [-sz] [-d delimiters] [file...]\n");
  209. 	exit(1);
  210. }
  212. int
  213. main(int argc, char *argv[])
  214. {
  215. 	char *lc_all = setlocale(LC_ALL, "");
  216. 	if(lc_all == NULL)
  217. 	{
  218. 		fprintf(stderr,
  219. 		        "%s: warning: Failed loading locales. setlocale(LC_ALL, \"\"): %s\n",
  220. 		        argv0,
  221. 		        strerror(errno));
  222. 	}
  223. 	errno = 0;
  225. 	int seq = 0;
  227. 	for(int c = -1; (c = getopt_nolong(argc, argv, ":d:sz")) != -1;)
  228. 	{
  229. 		switch(c)
  230. 		{
  231. 		case 'd':
  232. 		{
  233. 			const char *arg = optarg;
  234. 			size_t len = mbsrtowcs(NULL, &arg, 0, NULL);
  235. 			if(len == (size_t)-1) utils_err(1, "delimiters");
  237. 			if(len == SIZE_MAX) utils_err(1, NULL);
  238. 			wchar_t *warg = calloc((len + 1), sizeof(*warg));
  239. 			if(warg == NULL) utils_err(1, NULL);
  241. 			arg = optarg;
  242. 			len = mbsrtowcs(warg, &arg, len + 1, NULL);
  243. 			if(len == (size_t)-1) utils_err(1, "delimiters");
  245. 			delimcnt = tr(delim = warg);
  246. 			break;
  247. 		}
  248. 		case 's':
  249. 			seq = 1;
  250. 			break;
  251. 		case 'z':
  252. 			linedelim = L'\0';
  253. 			break;
  254. 		case ':':
  255. 			fprintf(stderr, "%s: error: Missing operand for option: '-%c'\n", argv0, optopt);
  256. 			usage();
  257. 			return 1;
  258. 		case '?':
  259. 			GETOPT_UNKNOWN_OPT
  260. 			usage();
  261. 			return 1;
  262. 		default:
  263. 			abort();
  264. 		}
  265. 	}
  267. 	argc -= optind;
  268. 	argv += optind;
  270. 	char *dash[] = {(char *)"-", NULL};
  271. 	char **args = argc > 0 ? argv : dash;
  273. 	if(!delim)
  274. 	{
  275. 		delimcnt = 1;
  276. 		delim = tab;
  277. 	}
  279. 	if(seq)
  280. 		return sequential(args);
  281. 	else
  282. 		return parallel(args);
  283. }