Skip to content
This repository

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
branch: master
Fetching contributors…

Octocat-spinner-32-eaf2f5

Cannot retrieve contributors at this time

file 127 lines (108 sloc) 2.791 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
#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>

#define OPT_RECURSIVE 1
#define OPT_FORCE 2

static int usage()
{
    fprintf(stderr,"Usage: rm [-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 (lstat(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, "..") || !strcmp(de->d_name, "."))
            continue;
        sprintf(dn, "%s/%s", name, de->d_name);
        if (unlink_recursive(dn, flags) < 0) {
            if (!(flags & OPT_FORCE)) {
                fail = 1;
                break;
            }
        }
        errno = 0;
    }
    /* in case readdir or unlink_recursive failed */
    if (fail || errno < 0) {
        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 ret;
    int i, c;
    int flags = 0;
    int something_failed = 0;

    if (argc < 2)
        return usage();

    /* check flags */
    do {
        c = getopt(argc, argv, "frR");
        if (c == EOF)
            break;
        switch (c) {
        case 'f':
            flags |= OPT_FORCE;
            break;
        case 'r':
        case 'R':
            flags |= OPT_RECURSIVE;
            break;
        }
    } while (1);

    if (optind < 1 || optind >= argc) {
        usage();
        return -1;
    }

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

        if (flags & OPT_RECURSIVE) {
            ret = unlink_recursive(argv[i], flags);
        } else {
            ret = unlink(argv[i]);
            if (ret < 0 && errno == ENOENT && (flags & OPT_FORCE)) {
                continue;
            }
        }

        if (ret < 0) {
            fprintf(stderr, "rm failed for %s, %s\n", argv[i], strerror(errno));
            if (!(flags & OPT_FORCE)) {
                return -1;
            } else {
                something_failed = 1;
            }
        }
    }

    return something_failed;
}

Something went wrong with that request. Please try again.