micro-utils/coreutils/rm.c

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <dirent.h>
#include <unistd.h>
#include "make_path.h"
#include "get_stat.h"
unsigned int f_flag;

int rmtree(const char *path) {
	struct stat stat_path;
	if (mu_get_lstat("rm", path, &stat_path))
		return 1;

	if (!S_ISDIR(stat_path.st_mode) || S_ISLNK(stat_path.st_mode)) {
		if (unlink(path) < 0) {
			if (!f_flag)
				fprintf(stderr, "rm: %s: is not directory\n", path);
			return 1;
		}

		return 0;
	}

	DIR *dir = opendir(path);
	if (dir == NULL) {
		if (!f_flag)
			fprintf(stderr, "rm: %s: Can`t open directory\n", path);
		return 1;
	}

	struct dirent *ep;
	while ((ep = readdir(dir)) != NULL) {
		if (!strcmp(ep->d_name, ".") || !strcmp(ep->d_name, ".."))
			continue;

		char *full_path = mu_make_path("rm", path, ep->d_name);
		if (full_path == NULL)
			continue;

		rmtree(full_path);
		free(full_path);
	}

	closedir(dir);
	if (rmdir(path) < 0)
		if (!f_flag)
			fprintf(stderr, "rm: %s: can`t remove a directory\n", path);

	return 0;
}

int main(const int argc, char **argv) {
	int i;
	for (i = 1; i < argc; i++) {
		if (argv[i][0] != '-')
			break;

		else if (!strcmp(argv[i], "-f"))
			f_flag = 1;

		else if (!strcmp(argv[i], "--help")) {
			printf("rm [-f force] [src1 src2...]\n");
			return 0;
		}
	}

	int status = 0;
	for (; i < argc; i++) {
		if (!strcmp(argv[i], ".") || !strcmp(argv[i], "..")){
			printf("rm: refusing to remove '.' or '..' directory\n");
			break;
		}

		int status = rmtree(argv[i]);
		if (status != 0)
			status = 1;
	}

	return status;
}