bc: convert to "G trick" - this returns bc to zero bss increase
[oweals/busybox.git] / coreutils / dos2unix.c
index 04c56ac9c001a3cc919f2d6f9aa159ada649c8d0..9f098e41e714a46568887656932ae34791807d7e 100644 (file)
 /*
  * dos2unix for BusyBox
  *
- * dos2unix '\n' convertor 0.5.0
+ * dos2unix '\n' converter 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.
-*/
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
+ */
+//config:config DOS2UNIX
+//config:      bool "dos2unix (5.1 kb)"
+//config:      default y
+//config:      help
+//config:      dos2unix is used to convert a text file from DOS format to
+//config:      UNIX format, and vice versa.
+//config:
+//config:config UNIX2DOS
+//config:      bool "unix2dos (5.1 kb)"
+//config:      default y
+//config:      help
+//config:      unix2dos is used to convert a text file from UNIX format to
+//config:      DOS format, and vice versa.
 
-#include "busybox.h"
+//applet:IF_DOS2UNIX(APPLET_NOEXEC(dos2unix, dos2unix, BB_DIR_USR_BIN, BB_SUID_DROP, dos2unix))
+//applet:IF_UNIX2DOS(APPLET_NOEXEC(unix2dos, dos2unix, BB_DIR_USR_BIN, BB_SUID_DROP, unix2dos))
 
-enum ConvType {
+//kbuild:lib-$(CONFIG_DOS2UNIX) += dos2unix.o
+//kbuild:lib-$(CONFIG_UNIX2DOS) += dos2unix.o
+
+//usage:#define dos2unix_trivial_usage
+//usage:       "[-ud] [FILE]"
+//usage:#define dos2unix_full_usage "\n\n"
+//usage:       "Convert FILE in-place from DOS to Unix format.\n"
+//usage:       "When no file is given, use stdin/stdout.\n"
+//usage:     "\n       -u      dos2unix"
+//usage:     "\n       -d      unix2dos"
+//usage:
+//usage:#define unix2dos_trivial_usage
+//usage:       "[-ud] [FILE]"
+//usage:#define unix2dos_full_usage "\n\n"
+//usage:       "Convert FILE in-place from Unix to DOS format.\n"
+//usage:       "When no file is given, use stdin/stdout.\n"
+//usage:     "\n       -u      dos2unix"
+//usage:     "\n       -d      unix2dos"
+
+#include "libbb.h"
+
+/* This is a NOEXEC applet. Be very careful! */
+
+enum {
        CT_UNIX2DOS = 1,
        CT_DOS2UNIX
-} ConvType;
+};
 
 /* if fn is NULL then input is stdin and output is stdout */
-static int convert(char *fn)
+static void convert(char *fn, int conv_type)
 {
        FILE *in, *out;
-       int i;
+       int ch;
+       char *temp_fn = temp_fn; /* for compiler */
+       char *resolved_fn = resolved_fn;
 
+       in = stdin;
+       out = stdout;
        if (fn != NULL) {
-               in = xfopen(fn, "rw");
-               /*
-                  The file is then created with mode read/write and
-                  permissions 0666 for glibc 2.0.6 and earlier or
-                  0600 for glibc 2.0.7 and later.
-                */
-               snprintf(bb_common_bufsiz1, sizeof(bb_common_bufsiz1), "%sXXXXXX", fn);
-               /*
-                  sizeof bb_common_bufsiz1 is 4096, so it should be big enough to
-                  hold the full path.  However if the output is truncated the
-                  subsequent call to mkstemp would fail.
-                */
-               i = mkstemp(&bb_common_bufsiz1[0]);
-               if (i == -1 || chmod(bb_common_bufsiz1, 0600) == -1) {
-                       bb_perror_nomsg_and_die();
-               }
-               out = fdopen(i, "w+");
-               if (!out) {
-                       close(i);
-                       remove(bb_common_bufsiz1);
-               }
-       } else {
-               in = stdin;
-               out = stdout;
+               struct stat st;
+               int fd;
+
+               resolved_fn = xmalloc_follow_symlinks(fn);
+               if (resolved_fn == NULL)
+                       bb_simple_perror_msg_and_die(fn);
+               in = xfopen_for_read(resolved_fn);
+               xfstat(fileno(in), &st, resolved_fn);
+
+               temp_fn = xasprintf("%sXXXXXX", resolved_fn);
+               fd = xmkstemp(temp_fn);
+               if (fchmod(fd, st.st_mode) == -1)
+                       bb_simple_perror_msg_and_die(temp_fn);
+               fchown(fd, st.st_uid, st.st_gid);
+
+               out = xfdopen_for_write(fd);
        }
 
-       while ((i = fgetc(in)) != EOF) {
-               if (i == '\r')
+       while ((ch = fgetc(in)) != EOF) {
+               if (ch == '\r')
                        continue;
-               if (i == '\n') {
-                       if (ConvType == CT_UNIX2DOS)
+               if (ch == '\n')
+                       if (conv_type == CT_UNIX2DOS)
                                fputc('\r', out);
-                       fputc('\n', out);
-                       continue;
-               }
-               fputc(i, out);
+               fputc(ch, out);
        }
 
        if (fn != NULL) {
                if (fclose(in) < 0 || fclose(out) < 0) {
-                       bb_perror_nomsg();
-                       remove(bb_common_bufsiz1);
-                       return -2;
-               }
-               /* Assume they are both on the same filesystem (which
-                * should be true since we put them into the same directory
-                * so we _should_ be ok, but you never know... */
-               if (rename(bb_common_bufsiz1, fn) < 0) {
-                       bb_perror_msg("cannot rename '%s' as '%s'", bb_common_bufsiz1, fn);
-                       return -1;
+                       unlink(temp_fn);
+                       bb_perror_nomsg_and_die();
                }
+               xrename(temp_fn, resolved_fn);
+               free(temp_fn);
+               free(resolved_fn);
        }
-
-       return 0;
 }
 
-int dos2unix_main(int argc, char **argv);
-int dos2unix_main(int argc, char **argv)
+int dos2unix_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int dos2unix_main(int argc UNUSED_PARAM, char **argv)
 {
-       int o;
+       int o, conv_type;
 
        /* See if we are supposed to be doing dos2unix or unix2dos */
-       if (applet_name[0] == 'd') {
-               ConvType = CT_DOS2UNIX; /*2 */
+       if (ENABLE_DOS2UNIX
+        && (!ENABLE_UNIX2DOS || applet_name[0] == 'd')
+       ) {
+               conv_type = CT_DOS2UNIX;
        } else {
-               ConvType = CT_UNIX2DOS; /*1 */
+               conv_type = CT_UNIX2DOS;
        }
-       /* -u and -d are mutally exclusive */
-       opt_complementary = "?:u--d:d--u";
-       /* process parameters */
-       /* -u convert to unix */
-       /* -d convert to dos  */
-       o = getopt32(argc, argv, "du");
+
+       /* -u convert to unix, -d convert to dos */
+       o = getopt32(argv, "^" "du" "\0" "u--d:d--u"); /* mutually exclusive */
 
        /* Do the conversion requested by an argument else do the default
         * conversion depending on our name.  */
        if (o)
-               ConvType = o;
+               conv_type = o;
 
-       if (optind < argc) {
-               while (optind < argc)
-                       if ((o = convert(argv[optind++])) < 0)
-                               break;
-       } else
-               o = convert(NULL);
+       argv += optind;
+       do {
+               /* might be convert(NULL) if there is no filename given */
+               convert(*argv, conv_type);
+       } while (*argv && *++argv);
 
-       return o;
+       return 0;
 }