1 /* Copyright 1986-1992 Emmet P. Gray.
2 * Copyright 1996-1998,2000-2002,2007,2009 Alain Knaff.
3 * This file is part of mtools.
4 *
5 * Mtools is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
9 *
10 * Mtools is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with Mtools. If not, see <http://www.gnu.org/licenses/>.
17 *
18 * mattrib.c
19 * Change MSDOS file attribute flags
20 */
21
22 #include "sysincludes.h"
23 #include "msdos.h"
24 #include "mtools.h"
25 #include "mainloop.h"
26
27 typedef struct Arg_t {
28 unsigned char add;
29 unsigned char remove;
30 struct MainParam_t mp;
31 int recursive;
32 int doPrintName;
33 } Arg_t;
34
attrib_file(direntry_t * entry,MainParam_t * mp)35 static int attrib_file(direntry_t *entry, MainParam_t *mp)
36 {
37 Arg_t *arg=(Arg_t *) mp->arg;
38
39 if(entry->entry != -3) {
40 /* if not root directory, change it */
41 entry->dir.attr = (entry->dir.attr & arg->remove) | arg->add;
42 dir_write(entry);
43 }
44 return GOT_ONE;
45 }
46
replay_attrib(direntry_t * entry,MainParam_t * mp UNUSEDP)47 static int replay_attrib(direntry_t *entry, MainParam_t *mp UNUSEDP)
48 {
49 if ( (IS_ARCHIVE(entry) && IS_DIR(entry)) ||
50 (!IS_ARCHIVE(entry) && !IS_DIR(entry)) ||
51 IS_SYSTEM(entry) || IS_HIDDEN(entry)) {
52
53 printf("mattrib ");
54
55 if (IS_ARCHIVE(entry) && IS_DIR(entry)) {
56 printf("+a ");
57 }
58
59 if (!IS_ARCHIVE(entry) && !IS_DIR(entry)) {
60 printf("-a ");
61 }
62
63 if (IS_SYSTEM(entry)) {
64 printf("+s ");
65 }
66
67 if (IS_HIDDEN(entry)) {
68 printf("+h ");
69 }
70
71 fprintPwd(stdout, entry, 1);
72 printf("\n");
73 }
74 return GOT_ONE;
75 }
76
77
78
view_attrib(direntry_t * entry,MainParam_t * mp UNUSEDP)79 static int view_attrib(direntry_t *entry, MainParam_t *mp UNUSEDP)
80 {
81 printf(" ");
82 if(IS_ARCHIVE(entry))
83 putchar('A');
84 else
85 putchar(' ');
86 fputs(" ",stdout);
87 if(IS_SYSTEM(entry))
88 putchar('S');
89 else
90 putchar(' ');
91 if(IS_HIDDEN(entry))
92 putchar('H');
93 else
94 putchar(' ');
95 if(IS_READONLY(entry))
96 putchar('R');
97 else
98 putchar(' ');
99 printf(" ");
100 fprintPwd(stdout, entry, 0);
101 printf("\n");
102 return GOT_ONE;
103 }
104
105
concise_view_attrib(direntry_t * entry,MainParam_t * mp)106 static int concise_view_attrib(direntry_t *entry, MainParam_t *mp)
107 {
108 Arg_t *arg=(Arg_t *) mp->arg;
109
110 if(IS_ARCHIVE(entry))
111 putchar('A');
112 if(IS_DIR(entry))
113 putchar('D');
114 if(IS_SYSTEM(entry))
115 putchar('S');
116 if(IS_HIDDEN(entry))
117 putchar('H');
118 if(IS_READONLY(entry))
119 putchar('R');
120 if(arg->doPrintName) {
121 putchar(' ');
122 fprintPwd(stdout, entry, 0);
123 }
124 putchar('\n');
125 return GOT_ONE;
126 }
127
recursive_attrib(direntry_t * entry,MainParam_t * mp)128 static int recursive_attrib(direntry_t *entry, MainParam_t *mp)
129 {
130 mp->callback(entry, mp);
131 return mp->loop(mp->File, mp, "*");
132 }
133
134
135 static void usage(int ret) NORETURN;
usage(int ret)136 static void usage(int ret)
137 {
138 fprintf(stderr, "Mtools version %s, dated %s\n",
139 mversion, mdate);
140 fprintf(stderr,
141 "Usage: %s [-p] [-a|+a] [-h|+h] [-r|+r] [-s|+s] msdosfile [msdosfiles...]\n",
142 progname);
143 exit(ret);
144 }
145
letterToCode(int letter)146 static int letterToCode(int letter)
147 {
148 switch (toupper(letter)) {
149 case 'A':
150 return ATTR_ARCHIVE;
151 case 'H':
152 return ATTR_HIDDEN;
153 case 'R':
154 return ATTR_READONLY;
155 case 'S':
156 return ATTR_SYSTEM;
157 default:
158 usage(1);
159 }
160 }
161
162 void mattrib(int argc, char **argv, int type UNUSEDP) NORETURN;
mattrib(int argc,char ** argv,int type UNUSEDP)163 void mattrib(int argc, char **argv, int type UNUSEDP)
164 {
165 Arg_t arg;
166 int view;
167 int c;
168 int concise;
169 int replay;
170 char *ptr;
171 int wantUsage;
172
173 arg.add = 0;
174 arg.remove = 0xff;
175 arg.recursive = 0;
176 arg.doPrintName = 1;
177 view = 0;
178 concise = 0;
179 replay = 0;
180 wantUsage = 0;
181
182 if(helpFlag(argc, argv))
183 usage(0);
184 while ((c = getopt(argc, argv, "i:/ahrsAHRSXp")) != EOF) {
185 switch (c) {
186 case 'h':
187 wantUsage = 1;
188 /* FALL THROUGH */
189 default:
190 arg.remove &= ~letterToCode(c);
191 break;
192 case 'i':
193 set_cmd_line_image(optarg);
194 break;
195 case 'p':
196 replay = 1;
197 break;
198 case '/':
199 arg.recursive = 1;
200 break;
201 case 'X':
202 concise = 1;
203 break;
204 case '?':
205 usage(1);
206 }
207 }
208
209 if(optind == argc && wantUsage) {
210 usage(0);
211 }
212
213 for(;optind < argc;optind++) {
214 switch(argv[optind][0]) {
215 case '+':
216 for(ptr = argv[optind] + 1; *ptr; ptr++)
217 arg.add |= letterToCode(*ptr);
218 continue;
219 case '-':
220 for(ptr = argv[optind] + 1; *ptr; ptr++)
221 arg.remove &= ~letterToCode(*ptr);
222 continue;
223 }
224 break;
225 }
226
227 if(arg.remove == 0xff && !arg.add)
228 view = 1;
229
230 if (optind >= argc)
231 usage(1);
232
233 init_mp(&arg.mp);
234 if(view){
235 if(concise) {
236 arg.mp.callback = concise_view_attrib;
237 arg.doPrintName = (argc - optind > 1 ||
238 arg.recursive ||
239 strpbrk(argv[optind], "*[?") != 0);
240 } else if (replay) {
241 arg.mp.callback = replay_attrib;
242 } else
243 arg.mp.callback = view_attrib;
244 arg.mp.openflags = O_RDONLY;
245 } else {
246 arg.mp.callback = attrib_file;
247 arg.mp.openflags = O_RDWR;
248 }
249
250 if(arg.recursive)
251 arg.mp.dirCallback = recursive_attrib;
252
253 arg.mp.arg = (void *) &arg;
254 arg.mp.lookupflags = ACCEPT_PLAIN | ACCEPT_DIR;
255 if(arg.recursive)
256 arg.mp.lookupflags |= DO_OPEN_DIRS | NO_DOTS;
257 exit(main_loop(&arg.mp, argv + optind, argc - optind));
258 }
259