rm_u.c

/*	rm - toolbox
	Copyright 2007-2015 PC GO Ld.
	Copyright 2015-2017 Rivoreo

	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.
*/

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <dirent.h>
#include <limits.h>
#include <sys/stat.h>
#include <sys/types.h>

#ifdef _WIN32
#define lstat stat
#endif

#define OPT_RECURSIVE 1
#define OPT_FORCE 2

static void usage() {
	fprintf(stderr,"Usage: rm"
#if defined _WIN32 && !defined _WIN32_WNT_NATIVE
		".exe"
#endif
		" [-rR] [-f] <target> [...]\n");
	//return -1;
}

/* return -1 on failure, with errno set to the first error */
static int unlink_recursive(const char* name, int flags)
{
    struct stat st;
    DIR *dir;
    struct dirent *de;
    int fail = 0;

    /* is it a file or directory? */
    if (stat(name, &st) < 0)
        return ((flags & OPT_FORCE) && errno == ENOENT) ? 0 : -1;

    /* a file, so unlink it */
    if (!S_ISDIR(st.st_mode))
        return unlink(name);

    /* a directory, so open handle */
    dir = opendir(name);
    if (dir == NULL)
        return -1;

    /* recurse over components */
    errno = 0;
    while((de = readdir(dir)) != NULL) {
        char dn[PATH_MAX];
        if(strcmp(de->d_name, "..") == 0 || strcmp(de->d_name, ".") == 0) {
            continue;
	}
        sprintf(dn, "%s/%s", name, de->d_name);
        if(unlink_recursive(dn, flags) < 0) {
            fail = 1;
            break;
        }
        errno = 0;
    }
    /* in case readdir or unlink_recursive failed */
    if (fail || errno) {
        int save = errno;
        closedir(dir);
        errno = save;
        return -1;
    }

    /* close directory handle */
    if (closedir(dir) < 0)
        return -1;

    /* delete target directory */
    return rmdir(name);
}

int rm_main(int argc, char *argv[])
{
	int sr;
	int ret = 0;
	int i;
	int flags = 0;

/*
	if(argc < 2) {
		usage();
		return -1;
	}
*/
	/* check flags */
	while(1) {
		int c = getopt(argc, argv, "frR");
		if(c == -1) break;
		switch (c) {
			case 'f':
				// Don't prompt
				// Don't complain about no files
				// Don't treat ENOENT as an error
				flags |= OPT_FORCE;
				break;
			case 'r':
			case 'R':
				flags |= OPT_RECURSIVE;
				break;
		}
	}

	//printf("optind = %d\n", optind);
	if(optind < 1 || (optind >= argc && !(flags & OPT_FORCE))) {
		usage();
		return -1;
	}

	/* loop over the file/directory args */
	for (i = optind; i < argc; i++) {

		if (flags & OPT_RECURSIVE) {
			sr = unlink_recursive(argv[i], flags);
		} else {
			struct stat st;
			sr = lstat(argv[i], &st);
			if(sr == 0) {
				if(S_ISDIR(st.st_mode)) {
					/* According to POSIX unlink(2) will give EPERM on directories
					 * if the implementation doesn't allow unlinking directories;
					 * since this tool is not designed to remove directory directly
					 * so display an EISDIR error message now */
					fprintf(stderr, "Cannot remove %s, %s\n",
						argv[i], strerror(EISDIR));
					ret = 1;
					continue;
				}
//#ifdef _WIN32
				if(access(argv[i], W_OK) != 0) {
					//if(flags & OPT_FORCE) chmod(argv[i], 666);
					if(!(flags & OPT_FORCE)) {
						//int a;
						printf("remove write-protected file \"%s\"? ", argv[i]);
/*
						a = getchar();
						if(a == 'Y' || a == 'y') chmod(argv[i], 666);
						else if(a == 'N' || a == 'n') return 1;
						else {
							fprintf(stderr, "Illegal input\n");
							return 1;
						}
*/
						switch(getchar()) {
							default:
								fprintf(stderr, "Illegal input\n");
							case 'N':
							case 'n':
								return 1;
							case 'Y':
							case 'y':
#if defined _WIN32 && !defined _WINDOWSNT_NATIVE
								chmod(argv[i], 666);
#endif
								break;
						}
					}
#if defined _WIN32 && !defined _WINDOWSNT_NATIVE
					else chmod(argv[i], 666);
#endif
				}
//#endif
				sr = unlink(argv[i]);
			}
		}

		if(sr < 0 && (errno != ENOENT || !(flags & OPT_FORCE))) {
//#ifdef _WIN32_WCE
//			fprintf(stderr, "rm failed for %s\n", argv[i]);
//#else
			fprintf(stderr, "Cannot remove %s, %s\n", argv[i], strerror(errno));
//#endif
			//return 1;
			ret = 1;
		}
	}

	return ret;
}