micro-utils/coreutils/rm.c
2023-10-28 10:02:24 +03:00

89 lines
1.7 KiB
C

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <dirent.h>
#include <sys/stat.h>
#include <unistd.h>
int get_stat(const char *path, struct stat *stat_path) {
if (lstat(path, stat_path)) {
fprintf(stderr, "rm: unable to stat %s: %s\n", path, strerror(errno));
return 1;
}
return 0;
}
char *make_path(const char *src, const char *dst) {
size_t len = strlen(src) + strlen(dst) + 2;
char *full_path = malloc(len + 1);
if (full_path == NULL) {
fprintf(stderr, "rm: malloc() returned NULL\n");
return NULL;
}
snprintf(full_path, len, "%s/%s", src, dst);
return full_path;
}
int rmtree(const char *path) {
struct stat stat_path;
if (get_stat(path, &stat_path))
return 1;
if (!S_ISDIR(stat_path.st_mode) || S_ISLNK(stat_path.st_mode)) {
if (unlink(path) < 0) {
fprintf(stderr, "rm: %s: is not directory\n", path);
return 1;
}
return 0;
}
DIR *dir = opendir(path);
if (dir == NULL) {
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 = make_path(path, ep->d_name);
if (full_path == NULL)
continue;
rmtree(full_path);
free(full_path);
}
closedir(dir);
if (rmdir(path) < 0)
fprintf(stderr, "rm: %s: can`t remove a directory\n", path);
return 0;
}
int main(const int argc, char **argv) {
if (argc == 1 || !strcmp(argv[argc - 1], "--help")) {
printf("rm [src1 src2...]\n");
return 0;
}
int status = 0;
for (int i = 1; i < argc; i++) {
if (argv[i][0] == '-')
continue;
int status = rmtree(argv[i]);
if (status != 0)
status = 1;
}
return status;
}