truncate: always set mode when opening file to avoid fortify errors

[oweals/busybox.git] / coreutils / rm.c

diff --git a/coreutils/rm.c b/coreutils/rm.c
index a84163272e3b8ae9ae9035e85977fb3eba2031b2..d0ad81dfc785160498f907ea19c6b8dc529d73c8 100644 (file)
--- a/coreutils/rm.c
+++ b/coreutils/rm.c
@@ -2,112 +2,67 @@
 /*
  * Mini rm implementation for busybox
  *
+ * Copyright (C) 2001 Matt Kraai <kraai@alumni.carnegiemellon.edu>
  *
- * Copyright (C) 1999,2000,2001 by Lineo, inc.
- * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
 
-#include <stdio.h>
-#include <time.h>
-#include <utime.h>
-#include <dirent.h>
-#include <errno.h>
-#include <unistd.h>
-#include <stdlib.h>
-#include "busybox.h"
+/* BB_AUDIT SUSv3 compliant */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/rm.html */
 
-static int recursiveFlag = FALSE;
-static int forceFlag = FALSE;
-static const char *srcName;
+/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
+ *
+ * Size reduction.
+ */
 
+//usage:#define rm_trivial_usage
+//usage:       "[-irf] FILE..."
+//usage:#define rm_full_usage "\n\n"
+//usage:       "Remove (unlink) FILEs\n"
+//usage:     "\n       -i      Always prompt before removing"
+//usage:     "\n       -f      Never prompt"
+//usage:     "\n       -R,-r   Recurse"
+//usage:
+//usage:#define rm_example_usage
+//usage:       "$ rm -rf /tmp/foo\n"
 
-static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
-{
-       if (unlink(fileName) < 0) {
-               perror_msg("%s", fileName);
-               return (FALSE);
-       }
-       return (TRUE);
-}
+#include "libbb.h"
 
-static int dirAction(const char *fileName, struct stat *statbuf, void* junk)
-{
-       if (recursiveFlag == FALSE) {
-               errno = EISDIR;
-               perror_msg("%s", fileName);
-               return (FALSE);
-       } 
-       if (rmdir(fileName) < 0) {
-               perror_msg("%s", fileName);
-               return (FALSE);
-       }
-       return (TRUE);
-}
+/* This is a NOFORK applet. Be very careful! */
 
-extern int rm_main(int argc, char **argv)
+int rm_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int rm_main(int argc UNUSED_PARAM, char **argv)
 {
-       int status = EXIT_SUCCESS;
-       int stopIt=FALSE;
-       struct stat statbuf;
+       int status = 0;
+       int flags = 0;
+       unsigned opt;
 
-       argc--;
-       argv++;
+       opt_complementary = "f-i:i-f";
+       opt = getopt32(argv, "fiRrv");
+       argv += optind;
+       if (opt & 1)
+               flags |= FILEUTILS_FORCE;
+       if (opt & 2)
+               flags |= FILEUTILS_INTERACTIVE;
+       if (opt & (8|4))
+               flags |= FILEUTILS_RECUR;
+       if ((opt & 16) && FILEUTILS_VERBOSE)
+               flags |= FILEUTILS_VERBOSE;
 
-       /* Parse any options */
-       while (argc > 0 && stopIt == FALSE) {
-               if (**argv == '-') {
-                       while (*++(*argv))
-                               switch (**argv) {
-                                       case 'R':
-                                       case 'r':
-                                               recursiveFlag = TRUE;
-                                               break;
-                                       case 'f':
-                                               forceFlag = TRUE;
-                                               break;
-                                       case '-':
-                                               stopIt = TRUE;
-                                               break;
-                                       default:
-                                               show_usage();
-                               }
-                       argc--;
-                       argv++;
-               }
-               else
-                       break;
-       }
-
-       if (argc < 1 && forceFlag == FALSE) {
-               show_usage();
-       }
+       if (*argv != NULL) {
+               do {
+                       const char *base = bb_get_last_path_component_strip(*argv);
 
-       while (argc-- > 0) {
-               srcName = *(argv++);
-               if (forceFlag == TRUE && lstat(srcName, &statbuf) != 0
-                       && errno == ENOENT) {
-                       /* do not reports errors for non-existent files if -f, just skip them */
-               } else {
-                       if (recursive_action(srcName, recursiveFlag, FALSE,
-                                                               TRUE, fileAction, dirAction, NULL) == FALSE) {
-                               status = EXIT_FAILURE;
+                       if (DOT_OR_DOTDOT(base)) {
+                               bb_error_msg("can't remove '.' or '..'");
+                       } else if (remove_file(*argv, flags) >= 0) {
+                               continue;
                        }
-               }
+                       status = 1;
+               } while (*++argv);
+       } else if (!(flags & FILEUTILS_FORCE)) {
+               bb_show_usage();
        }
+
        return status;
 }