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
127
128
129
130
|
#define _POSIX_C_SOURCE 200112L
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <errno.h>
#include <dirent.h>
#include <limits.h>
#include <sys/types.h>
#include <sys/stat.h>
typedef enum
{
FLAG_FORCE = 1 << 0,
FLAG_INTERACTIVE = 1 << 1,
/* FLAG_INTERACTIVE_ONCE = 1 << 2, */ // TODO
FLAG_RECURSIVE = 1 << 3,
FLAG_DIRECTORY = 1 << 4,
FLAG_VERBOSE = 1 << 5,
} t_flags;
char *g_name = "rm";
void log_errno(char *path)
{
fprintf(stderr, "%s: cannot remove '%s': %s\n", g_name, path, strerror(errno));
}
void fatal_errno(char *path)
{
log_errno(path);
exit(EXIT_FAILURE);
}
int rm_file(char *path, t_flags flags)
{
struct stat statbuf;
char recursive_path[PATH_MAX + 1] = {'\0'};
if (lstat(path, &statbuf) == -1)
fatal_errno(path);
if (flags & FLAG_INTERACTIVE
|| (!(flags & FLAG_FORCE)
&& !((statbuf.st_mode & S_IWUSR && statbuf.st_uid == getuid()) ||
(statbuf.st_mode & S_IWGRP && statbuf.st_gid == getgid()) ||
statbuf.st_mode & S_IWOTH)))
{
printf("%s: remove regular file '%s'? ", g_name, path);
if (tolower(getchar()) != 'y')
return 0;
}
if (S_ISDIR(statbuf.st_mode) && flags & FLAG_RECURSIVE)
{
DIR *directory;
struct dirent *entry;
if ((directory = opendir(path)) == NULL)
{
log_errno(path);
return 0;
}
strcpy(recursive_path, path);
if (path[strlen(path) - 1] != '/')
strcat(recursive_path, "/");
while ((entry = readdir(directory)) != NULL)
{
if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
continue;
strcat(recursive_path, entry->d_name);
rm_file(recursive_path, flags);
strrchr(recursive_path, '/')[1] = '\0';
}
closedir(directory);
if (rmdir(path) == -1)
log_errno(path);
else if (flags & FLAG_VERBOSE)
printf("removed directory '%s'\n", path);
}
else if (S_ISDIR(statbuf.st_mode) && flags & FLAG_DIRECTORY)
{
if (rmdir(path) == -1)
log_errno(path);
else if (flags & FLAG_VERBOSE)
printf("removed directory '%s'\n", path);
}
else
{
if (unlink(path) == -1)
log_errno(path);
else if (flags & FLAG_VERBOSE)
printf("removed '%s'\n", path);
}
return 0;
}
int main(int argc, char **argv)
{
if (argc == 1)
{
fprintf(stderr, "%s: missing operand\n", argv[0]);
return EXIT_FAILURE;
}
g_name = argv[0];
int option;
t_flags flags = 0;
while ((option = getopt(argc, argv, "firRdv")) != -1)
{
switch (option)
{
case 'f': flags |= FLAG_FORCE; break;
case 'i': flags |= FLAG_INTERACTIVE; break;
/* case 'I': flags |= FLAG_INTERACTIVE_ONCE; break; */
case 'd': flags |= FLAG_DIRECTORY; break;
case 'v': flags |= FLAG_VERBOSE; break;
case 'r':
case 'R':
flags |= FLAG_RECURSIVE; break;
break;
}
}
for (; optind < argc; optind++)
rm_file(argv[optind], flags);
return EXIT_SUCCESS;
}
|
