busybox/coreutils/dos2unix.c

/* vi: set sw=4 ts=4: */
/*
 * dos2unix for BusyBox
 *
 * dos2unix '\n' convertor 0.5.0
 * based on Unix2Dos 0.9.0 by Peter Hanecak (made 19.2.1997)
 * Copyright 1997,.. by Peter Hanecak <hanecak@megaloman.sk>.
 * All rights reserved.
 *
 * dos2unix filters reading input from stdin and writing output to stdout.
 *
 * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
*/

#include "libbb.h"

enum {
	CT_UNIX2DOS = 1,
	CT_DOS2UNIX
};

/* if fn is NULL then input is stdin and output is stdout */
static void convert(char *fn, int conv_type)
{
	FILE *in, *out;
	int i;
	char *temp_fn = temp_fn; /* for compiler */
	char *resolved_fn = resolved_fn;

	in = stdin;
	out = stdout;
	if (fn != NULL) {
		struct stat st;

		resolved_fn = xmalloc_follow_symlinks(fn);
		if (resolved_fn == NULL)
			bb_simple_perror_msg_and_die(fn);
		in = xfopen_for_read(resolved_fn);
		fstat(fileno(in), &st);

		temp_fn = xasprintf("%sXXXXXX", resolved_fn);
		i = mkstemp(temp_fn);
		if (i == -1
		 || fchmod(i, st.st_mode) == -1
		) {
			bb_simple_perror_msg_and_die(temp_fn);
		}
		out = xfdopen_for_write(i);
	}

	while ((i = fgetc(in)) != EOF) {
		if (i == '\r')
			continue;
		if (i == '\n')
			if (conv_type == CT_UNIX2DOS)
				fputc('\r', out);
		fputc(i, out);
	}

	if (fn != NULL) {
		if (fclose(in) < 0 || fclose(out) < 0) {
			unlink(temp_fn);
			bb_perror_nomsg_and_die();
		}
		xrename(temp_fn, resolved_fn);
		free(temp_fn);
		free(resolved_fn);
	}
}

int dos2unix_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int dos2unix_main(int argc UNUSED_PARAM, char **argv)
{
	int o, conv_type;

	/* See if we are supposed to be doing dos2unix or unix2dos */
	conv_type = CT_UNIX2DOS;
	if (applet_name[0] == 'd') {
		conv_type = CT_DOS2UNIX;
	}

	/* -u convert to unix, -d convert to dos */
	opt_complementary = "u--d:d--u"; /* mutually exclusive */
	o = getopt32(argv, "du");

	/* Do the conversion requested by an argument else do the default
	 * conversion depending on our name.  */
	if (o)
		conv_type = o;

	argv += optind;
	do {
		/* might be convert(NULL) if there is no filename given */
		convert(*argv, conv_type);
	} while (*++argv);

	return 0;
}
1	/* vi: set sw=4 ts=4: */
2	/*
3	* dos2unix for BusyBox
4	*
5	* dos2unix '\n' convertor 0.5.0
6	* based on Unix2Dos 0.9.0 by Peter Hanecak (made 19.2.1997)
7	* Copyright 1997,.. by Peter Hanecak <hanecak@megaloman.sk>.
8	* All rights reserved.
9	*
10	* dos2unix filters reading input from stdin and writing output to stdout.
11	*
12	* Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
13	*/
14
15	#include "libbb.h"
16
17	enum {
18	CT_UNIX2DOS = 1,
19	CT_DOS2UNIX
20	};
21
22	/* if fn is NULL then input is stdin and output is stdout */
23	static void convert(char *fn, int conv_type)
24	{
25	FILE in, out;
26	int i;
27	char temp_fn = temp_fn; / for compiler */
28	char *resolved_fn = resolved_fn;
29
30	in = stdin;
31	out = stdout;
32	if (fn != NULL) {
33	struct stat st;
34
35	resolved_fn = xmalloc_follow_symlinks(fn);
36	if (resolved_fn == NULL)
37	bb_simple_perror_msg_and_die(fn);
38	in = xfopen_for_read(resolved_fn);
39	fstat(fileno(in), &st);
40
41	temp_fn = xasprintf("%sXXXXXX", resolved_fn);
42	i = mkstemp(temp_fn);
43	if (i == -1
44	\|\| fchmod(i, st.st_mode) == -1
45	) {
46	bb_simple_perror_msg_and_die(temp_fn);
47	}
48	out = xfdopen_for_write(i);
49	}
50
51	while ((i = fgetc(in)) != EOF) {
52	if (i == '\r')
53	continue;
54	if (i == '\n')
55	if (conv_type == CT_UNIX2DOS)
56	fputc('\r', out);
57	fputc(i, out);
58	}
59
60	if (fn != NULL) {
61	if (fclose(in) < 0 \|\| fclose(out) < 0) {
62	unlink(temp_fn);
63	bb_perror_nomsg_and_die();
64	}
65	xrename(temp_fn, resolved_fn);
66	free(temp_fn);
67	free(resolved_fn);
68	}
69	}
70
71	int dos2unix_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
72	int dos2unix_main(int argc UNUSED_PARAM, char **argv)
73	{
74	int o, conv_type;
75
76	/* See if we are supposed to be doing dos2unix or unix2dos */
77	conv_type = CT_UNIX2DOS;
78	if (applet_name[0] == 'd') {
79	conv_type = CT_DOS2UNIX;
80	}
81
82	/* -u convert to unix, -d convert to dos */
83	opt_complementary = "u--d:d--u"; /* mutually exclusive */
84	o = getopt32(argv, "du");
85
86	/* Do the conversion requested by an argument else do the default
87	* conversion depending on our name. */
88	if (o)
89	conv_type = o;
90
91	argv += optind;
92	do {
93	/* might be convert(NULL) if there is no filename given */
94	convert(*argv, conv_type);
95	} while (*++argv);
96
97	return 0;
98	}