busybox/coreutils/expand.c

/* expand - convert tabs to spaces
 * unexpand - convert spaces to tabs
 *
 * Copyright (C) 89, 91, 1995-2006 Free Software Foundation, Inc.
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 *
 * David MacKenzie <djm@gnu.ai.mit.edu>
 *
 * Options for expand:
 * -t num  --tabs=NUM      Convert tabs to num spaces (default 8 spaces).
 * -i      --initial       Only convert initial tabs on each line to spaces.
 *
 * Options for unexpand:
 * -a      --all           Convert all blanks, instead of just initial blanks.
 * -f      --first-only    Convert only leading sequences of blanks (default).
 * -t num  --tabs=NUM      Have tabs num characters apart instead of 8.
 *
 *  Busybox version (C) 2007 by Tito Ragusa <farmatito@tiscali.it>
 *
 *  Caveat: this versions of expand and unexpand don't accept tab lists.
 */

#include "libbb.h"

enum {
	OPT_INITIAL     = 1 << 0,
	OPT_TABS        = 1 << 1,
	OPT_ALL         = 1 << 2,
};

static void xputchar(char c)
{
	if (putchar(c) < 0)
		bb_error_msg_and_die(bb_msg_write_error);
}

#if ENABLE_EXPAND
static void expand(FILE *file, unsigned tab_size, unsigned opt)
{
	char *line;
	char *ptr;
	int convert;
	unsigned pos;

	/* Increment tab_size by 1 locally.*/
	tab_size++;

	while ((line = xmalloc_fgets(file)) != NULL) {
		convert = 1;
		pos = 0;
		ptr = line;
		while (*line) {
			pos++;
			if (*line == '\t' && convert) {
				for (; pos < tab_size; pos++) {
					xputchar(' ');
				}
			} else {
				if ((opt & OPT_INITIAL) && !isblank(*line)) {
					convert = 0;
				}
				xputchar(*line);
			}
			if (pos == tab_size) {
				pos = 0;
			}
			line++;
		}
		free(ptr);
	}
}
#endif

#if ENABLE_UNEXPAND
static void unexpand(FILE *file, unsigned int tab_size, unsigned opt)
{
	char *line;
	char *ptr;
	int convert;
	int pos;
	int i = 0;
	unsigned column = 0;

	while ((line = xmalloc_fgets(file)) != NULL) {
		convert = 1;
		pos = 0;
		ptr = line;
		while (*line) {
			while ((*line == ' ' || *line == '\t') && convert) {
				pos += (*line == ' ') ? 1 : tab_size;
				line++;
				column++;
				if ((opt & OPT_ALL) && column == tab_size) {
					column = 0;
					goto put_tab;
				}
			}
			if (pos) {
				i = pos / tab_size;
				if (i) {
					for (; i > 0; i--) {
 put_tab:
						xputchar('\t');
					}
				} else {
					for (i = pos % tab_size; i > 0; i--) {
						xputchar(' ');
					}
				}
				pos = 0;
			} else {
				if (opt & OPT_INITIAL) {
					convert = 0;
				}
				if (opt & OPT_ALL) {
					column++;
				}
				xputchar(*line);
				line++;
			}
		}
		free(ptr);
	}
}
#endif

int expand_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int expand_main(int argc UNUSED_PARAM, char **argv)
{
	/* Default 8 spaces for 1 tab */
	const char *opt_t = "8";
	FILE *file;
	unsigned tab_size;
	unsigned opt;
	int exit_status = EXIT_SUCCESS;

#if ENABLE_FEATURE_EXPAND_LONG_OPTIONS
	static const char expand_longopts[] ALIGN1 =
		/* name, has_arg, val */
		"initial\0"          No_argument       "i"
		"tabs\0"             Required_argument "t"
	;
#endif
#if ENABLE_FEATURE_UNEXPAND_LONG_OPTIONS
	static const char unexpand_longopts[] ALIGN1 =
		/* name, has_arg, val */
		"first-only\0"       No_argument       "i"
		"tabs\0"             Required_argument "t"
		"all\0"              No_argument       "a"
	;
#endif

	if (ENABLE_EXPAND && (!ENABLE_UNEXPAND || applet_name[0] == 'e')) {
		USE_FEATURE_EXPAND_LONG_OPTIONS(applet_long_options = expand_longopts);
		opt = getopt32(argv, "it:", &opt_t);
	} else {
		USE_FEATURE_UNEXPAND_LONG_OPTIONS(applet_long_options = unexpand_longopts);
		/* -t NUM sets also -a */
		opt_complementary = "ta";
		opt = getopt32(argv, "ft:a", &opt_t);
		/* -f --first-only is the default */
		if (!(opt & OPT_ALL)) opt |= OPT_INITIAL;
	}
	tab_size = xatou_range(opt_t, 1, UINT_MAX);

	argv += optind;

	if (!*argv) {
		*--argv = (char*)bb_msg_standard_input;
	}
	do {
		file = fopen_or_warn_stdin(*argv);
		if (!file) {
			exit_status = EXIT_FAILURE;
			continue;
		}

		if (ENABLE_EXPAND && (!ENABLE_UNEXPAND || applet_name[0] == 'e'))
			USE_EXPAND(expand(file, tab_size, opt));
		else
			USE_UNEXPAND(unexpand(file, tab_size, opt));

		/* Check and close the file */
		if (fclose_if_not_stdin(file)) {
			bb_simple_perror_msg(*argv);
			exit_status = EXIT_FAILURE;
		}
		/* If stdin also clear EOF */
		if (file == stdin)
			clearerr(file);
	} while (*++argv);

	/* Now close stdin also */
	/* (if we didn't read from it, it's a no-op) */
	if (fclose(stdin))
		bb_perror_msg_and_die(bb_msg_standard_input);

	fflush_stdout_and_exit(exit_status);
}
1	niro	816	/* expand - convert tabs to spaces
2			* unexpand - convert spaces to tabs
3			*
4			* Copyright (C) 89, 91, 1995-2006 Free Software Foundation, Inc.
5			*
6			* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
7			*
8			* David MacKenzie <djm@gnu.ai.mit.edu>
9			*
10			* Options for expand:
11			* -t num --tabs=NUM Convert tabs to num spaces (default 8 spaces).
12			* -i --initial Only convert initial tabs on each line to spaces.
13			*
14			* Options for unexpand:
15			* -a --all Convert all blanks, instead of just initial blanks.
16			* -f --first-only Convert only leading sequences of blanks (default).
17			* -t num --tabs=NUM Have tabs num characters apart instead of 8.
18			*
19			* Busybox version (C) 2007 by Tito Ragusa <farmatito@tiscali.it>
20			*
21			* Caveat: this versions of expand and unexpand don't accept tab lists.
22			*/
23
24			#include "libbb.h"
25
26			enum {
27			OPT_INITIAL = 1 << 0,
28			OPT_TABS = 1 << 1,
29			OPT_ALL = 1 << 2,
30			};
31
32			static void xputchar(char c)
33			{
34			if (putchar(c) < 0)
35			bb_error_msg_and_die(bb_msg_write_error);
36			}
37
38			#if ENABLE_EXPAND
39			static void expand(FILE *file, unsigned tab_size, unsigned opt)
40			{
41			char *line;
42			char *ptr;
43			int convert;
44			unsigned pos;
45
46			/* Increment tab_size by 1 locally.*/
47			tab_size++;
48
49			while ((line = xmalloc_fgets(file)) != NULL) {
50			convert = 1;
51			pos = 0;
52			ptr = line;
53			while (*line) {
54			pos++;
55			if (*line == '\t' && convert) {
56			for (; pos < tab_size; pos++) {
57			xputchar(' ');
58			}
59			} else {
60			if ((opt & OPT_INITIAL) && !isblank(*line)) {
61			convert = 0;
62			}
63			xputchar(*line);
64			}
65			if (pos == tab_size) {
66			pos = 0;
67			}
68			line++;
69			}
70			free(ptr);
71			}
72			}
73			#endif
74
75			#if ENABLE_UNEXPAND
76			static void unexpand(FILE *file, unsigned int tab_size, unsigned opt)
77			{
78			char *line;
79			char *ptr;
80			int convert;
81			int pos;
82			int i = 0;
83			unsigned column = 0;
84
85			while ((line = xmalloc_fgets(file)) != NULL) {
86			convert = 1;
87			pos = 0;
88			ptr = line;
89			while (*line) {
90			while ((line == ' ' \|\| line == '\t') && convert) {
91			pos += (*line == ' ') ? 1 : tab_size;
92			line++;
93			column++;
94			if ((opt & OPT_ALL) && column == tab_size) {
95			column = 0;
96			goto put_tab;
97			}
98			}
99			if (pos) {
100			i = pos / tab_size;
101			if (i) {
102			for (; i > 0; i--) {
103			put_tab:
104			xputchar('\t');
105			}
106			} else {
107			for (i = pos % tab_size; i > 0; i--) {
108			xputchar(' ');
109			}
110			}
111			pos = 0;
112			} else {
113			if (opt & OPT_INITIAL) {
114			convert = 0;
115			}
116			if (opt & OPT_ALL) {
117			column++;
118			}
119			xputchar(*line);
120			line++;
121			}
122			}
123			free(ptr);
124			}
125			}
126			#endif
127
128			int expand_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
129			int expand_main(int argc UNUSED_PARAM, char **argv)
130			{
131			/* Default 8 spaces for 1 tab */
132			const char *opt_t = "8";
133			FILE *file;
134			unsigned tab_size;
135			unsigned opt;
136			int exit_status = EXIT_SUCCESS;
137
138			#if ENABLE_FEATURE_EXPAND_LONG_OPTIONS
139			static const char expand_longopts[] ALIGN1 =
140			/* name, has_arg, val */
141			"initial\0" No_argument "i"
142			"tabs\0" Required_argument "t"
143			;
144			#endif
145			#if ENABLE_FEATURE_UNEXPAND_LONG_OPTIONS
146			static const char unexpand_longopts[] ALIGN1 =
147			/* name, has_arg, val */
148			"first-only\0" No_argument "i"
149			"tabs\0" Required_argument "t"
150			"all\0" No_argument "a"
151			;
152			#endif
153
154			if (ENABLE_EXPAND && (!ENABLE_UNEXPAND \|\| applet_name[0] == 'e')) {
155			USE_FEATURE_EXPAND_LONG_OPTIONS(applet_long_options = expand_longopts);
156			opt = getopt32(argv, "it:", &opt_t);
157			} else {
158			USE_FEATURE_UNEXPAND_LONG_OPTIONS(applet_long_options = unexpand_longopts);
159			/* -t NUM sets also -a */
160			opt_complementary = "ta";
161			opt = getopt32(argv, "ft:a", &opt_t);
162			/* -f --first-only is the default */
163			if (!(opt & OPT_ALL)) opt \|= OPT_INITIAL;
164			}
165			tab_size = xatou_range(opt_t, 1, UINT_MAX);
166
167			argv += optind;
168
169			if (!*argv) {
170			--argv = (char)bb_msg_standard_input;
171			}
172			do {
173			file = fopen_or_warn_stdin(*argv);
174			if (!file) {
175			exit_status = EXIT_FAILURE;
176			continue;
177			}
178
179			if (ENABLE_EXPAND && (!ENABLE_UNEXPAND \|\| applet_name[0] == 'e'))
180			USE_EXPAND(expand(file, tab_size, opt));
181			else
182			USE_UNEXPAND(unexpand(file, tab_size, opt));
183
184			/* Check and close the file */
185			if (fclose_if_not_stdin(file)) {
186			bb_simple_perror_msg(*argv);
187			exit_status = EXIT_FAILURE;
188			}
189			/* If stdin also clear EOF */
190			if (file == stdin)
191			clearerr(file);
192			} while (*++argv);
193
194			/* Now close stdin also */
195			/* (if we didn't read from it, it's a no-op) */
196			if (fclose(stdin))
197			bb_perror_msg_and_die(bb_msg_standard_input);
198
199			fflush_stdout_and_exit(exit_status);
200			}