Rev 3376 | Rev 4327 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
3296 | post | 1 | /* Copyright (c) 2008, Tim Post <tinkertim@gmail.com> |
2 | * All rights reserved. |
||
3277 | post | 3 | * |
3296 | post | 4 | * Redistribution and use in source and binary forms, with or without |
5 | * modification, are permitted provided that the following conditions are met: |
||
6 | * |
||
7 | * Redistributions of source code must retain the above copyright notice, this |
||
8 | * list of conditions and the following disclaimer. |
||
9 | * |
||
10 | * Redistributions in binary form must reproduce the above copyright notice, |
||
11 | * this list of conditions and the following disclaimer in the documentation |
||
12 | * and/or other materials provided with the distribution. |
||
13 | * |
||
14 | * Neither the name of the original program's authors nor the names of its |
||
15 | * contributors may be used to endorse or promote products derived from this |
||
16 | * software without specific prior written permission. |
||
17 | * |
||
18 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||
19 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||
20 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||
21 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
||
22 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
||
23 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
||
24 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
||
25 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
||
26 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
||
27 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
||
28 | * POSSIBILITY OF SUCH DAMAGE. |
||
29 | */ |
||
3277 | post | 30 | |
31 | #include <stdio.h> |
||
32 | #include <stdlib.h> |
||
3296 | post | 33 | #include <unistd.h> |
34 | #include <fcntl.h> |
||
35 | #include <dirent.h> |
||
3337 | post | 36 | #include <assert.h> |
37 | #include <getopt.h> |
||
38 | |||
3296 | post | 39 | #include "config.h" |
40 | #include "errors.h" |
||
41 | #include "util.h" |
||
3277 | post | 42 | #include "entry.h" |
43 | #include "rm.h" |
||
44 | #include "cmds.h" |
||
45 | |||
46 | static char *cmdname = "rm"; |
||
3337 | post | 47 | #define RM_VERSION "0.0.1" |
3277 | post | 48 | |
3337 | post | 49 | static rm_job_t rm; |
50 | |||
3375 | post | 51 | static struct option const long_options[] = { |
3338 | post | 52 | { "help", no_argument, 0, 'h' }, |
53 | { "version", no_argument, 0, 'v' }, |
||
54 | { "recursive", no_argument, 0, 'r' }, |
||
55 | { "force", no_argument, 0, 'f' }, |
||
56 | { "safe", no_argument, 0, 's' }, |
||
3337 | post | 57 | { 0, 0, 0, 0 } |
58 | }; |
||
59 | |||
3364 | post | 60 | static unsigned int rm_start(rm_job_t *rm) |
3337 | post | 61 | { |
62 | rm->recursive = 0; |
||
63 | rm->force = 0; |
||
64 | rm->safe = 0; |
||
65 | |||
66 | /* Make sure we can allocate enough memory to store |
||
67 | * what is needed in the job structure */ |
||
68 | if (NULL == (rm->nwd = (char *) malloc(PATH_MAX))) |
||
69 | return 0; |
||
70 | memset(rm->nwd, 0, sizeof(rm->nwd)); |
||
71 | |||
72 | if (NULL == (rm->owd = (char *) malloc(PATH_MAX))) |
||
73 | return 0; |
||
74 | memset(rm->owd, 0, sizeof(rm->owd)); |
||
75 | |||
76 | if (NULL == (rm->cwd = (char *) malloc(PATH_MAX))) |
||
77 | return 0; |
||
78 | memset(rm->cwd, 0, sizeof(rm->cwd)); |
||
79 | |||
80 | chdir("."); |
||
81 | |||
82 | if (NULL == (getcwd(rm->owd, PATH_MAX))) |
||
83 | return 0; |
||
84 | |||
85 | return 1; |
||
86 | } |
||
87 | |||
3364 | post | 88 | static void rm_end(rm_job_t *rm) |
3337 | post | 89 | { |
90 | if (NULL != rm->nwd) |
||
91 | free(rm->nwd); |
||
92 | |||
93 | if (NULL != rm->owd) |
||
94 | free(rm->owd); |
||
95 | |||
96 | if (NULL != rm->cwd) |
||
97 | free(rm->cwd); |
||
98 | |||
99 | return; |
||
100 | } |
||
101 | |||
3364 | post | 102 | static unsigned int rm_recursive(const char *path) |
3296 | post | 103 | { |
3337 | post | 104 | int rc; |
105 | |||
106 | /* First see if it will just go away */ |
||
107 | rc = rmdir(path); |
||
108 | if (rc == 0) |
||
109 | return 0; |
||
110 | |||
111 | /* Its not empty, recursively scan it */ |
||
112 | cli_error(CL_ENOTSUP, |
||
113 | "Can not remove %s, directory not empty", path); |
||
3296 | post | 114 | return 1; |
115 | } |
||
116 | |||
3364 | post | 117 | static unsigned int rm_single(const char *path) |
3296 | post | 118 | { |
119 | if (unlink(path)) { |
||
120 | cli_error(CL_EFAIL, "rm: could not remove file %s", path); |
||
121 | return 1; |
||
122 | } |
||
123 | return 0; |
||
124 | } |
||
125 | |||
3364 | post | 126 | static unsigned int rm_scope(const char *path) |
3296 | post | 127 | { |
128 | int fd; |
||
129 | DIR *dirp; |
||
130 | |||
131 | dirp = opendir(path); |
||
132 | if (dirp) { |
||
133 | closedir(dirp); |
||
134 | return RM_DIR; |
||
135 | } |
||
136 | |||
137 | fd = open(path, O_RDONLY); |
||
138 | if (fd > 0) { |
||
139 | close(fd); |
||
140 | return RM_FILE; |
||
141 | } |
||
142 | |||
143 | return RM_BOGUS; |
||
144 | } |
||
145 | |||
3277 | post | 146 | /* Dispays help for rm in various levels */ |
147 | void * help_cmd_rm(unsigned int level) |
||
148 | { |
||
3296 | post | 149 | if (level == HELP_SHORT) { |
150 | printf("`%s' removes files and directories.\n", cmdname); |
||
151 | } else { |
||
152 | help_cmd_rm(HELP_SHORT); |
||
3337 | post | 153 | printf( |
154 | "Usage: %s [options] <path>\n" |
||
155 | "Options:\n" |
||
156 | " -h, --help A short option summary\n" |
||
157 | " -v, --version Print version information and exit\n" |
||
158 | " -r, --recursive Recursively remove sub directories\n" |
||
159 | " -f, --force Do not prompt prior to removing files\n" |
||
160 | " -s, --safe Stop if directories change during removal\n\n" |
||
161 | "Currently, %s is under development, some options don't work.\n", |
||
162 | cmdname, cmdname); |
||
3296 | post | 163 | } |
164 | return CMD_VOID; |
||
3277 | post | 165 | } |
166 | |||
167 | /* Main entry point for rm, accepts an array of arguments */ |
||
168 | int * cmd_rm(char **argv) |
||
169 | { |
||
170 | unsigned int argc; |
||
3296 | post | 171 | unsigned int i, scope, ret = 0; |
3337 | post | 172 | int c, opt_ind; |
173 | size_t len; |
||
174 | char *buff = NULL; |
||
3277 | post | 175 | |
3376 | post | 176 | argc = cli_count_args(argv); |
177 | |||
3277 | post | 178 | if (argc < 2) { |
3296 | post | 179 | cli_error(CL_EFAIL, |
3337 | post | 180 | "%s: insufficient arguments. Try %s --help", cmdname, cmdname); |
3296 | post | 181 | return CMD_FAILURE; |
3277 | post | 182 | } |
183 | |||
3337 | post | 184 | if (!rm_start(&rm)) { |
185 | cli_error(CL_ENOMEM, "%s: could not initialize", cmdname); |
||
186 | rm_end(&rm); |
||
187 | return CMD_FAILURE; |
||
188 | } |
||
189 | |||
190 | for (c = 0, optind = 0, opt_ind = 0; c != -1;) { |
||
191 | c = getopt_long(argc, argv, "hvrfs", long_options, &opt_ind); |
||
192 | switch (c) { |
||
193 | case 'h': |
||
194 | help_cmd_rm(HELP_LONG); |
||
195 | return CMD_SUCCESS; |
||
196 | case 'v': |
||
197 | printf("%s\n", RM_VERSION); |
||
198 | return CMD_SUCCESS; |
||
199 | case 'r': |
||
200 | rm.recursive = 1; |
||
201 | break; |
||
202 | case 'f': |
||
203 | rm.force = 1; |
||
204 | break; |
||
205 | case 's': |
||
206 | rm.safe = 1; |
||
207 | break; |
||
208 | } |
||
209 | } |
||
210 | |||
211 | if (optind == argc) { |
||
212 | cli_error(CL_EFAIL, |
||
213 | "%s: insufficient arguments. Try %s --help", cmdname, cmdname); |
||
214 | rm_end(&rm); |
||
215 | return CMD_FAILURE; |
||
216 | } |
||
217 | |||
218 | i = optind; |
||
219 | while (NULL != argv[i]) { |
||
220 | len = strlen(argv[i]) + 2; |
||
221 | buff = (char *) realloc(buff, len); |
||
222 | assert(buff != NULL); |
||
223 | memset(buff, 0, sizeof(buff)); |
||
224 | snprintf(buff, len, argv[i]); |
||
225 | |||
3296 | post | 226 | scope = rm_scope(buff); |
227 | switch (scope) { |
||
3337 | post | 228 | case RM_BOGUS: /* FIXME */ |
3296 | post | 229 | case RM_FILE: |
230 | ret += rm_single(buff); |
||
231 | break; |
||
232 | case RM_DIR: |
||
3337 | post | 233 | if (! rm.recursive) { |
234 | printf("%s is a directory, use -r to remove it.\n", buff); |
||
235 | ret ++; |
||
236 | } else { |
||
237 | ret += rm_recursive(buff); |
||
238 | } |
||
3296 | post | 239 | break; |
240 | } |
||
3337 | post | 241 | i++; |
3296 | post | 242 | } |
3277 | post | 243 | |
3337 | post | 244 | if (NULL != buff) |
245 | free(buff); |
||
246 | |||
247 | rm_end(&rm); |
||
248 | |||
3296 | post | 249 | if (ret) |
250 | return CMD_FAILURE; |
||
251 | else |
||
252 | return CMD_SUCCESS; |
||
3277 | post | 253 | } |
254 |