rmextattr.c revision 104796 
1/*-
2 * Copyright (c) 2002 Networks Associates Technology, Inc.
3 * Copyright (c) 2002 Poul-Henning Kamp.
4 * Copyright (c) 1999, 2000, 2001, 2002 Robert N. M. Watson
5 * All rights reserved.
6 *
7 * This software was developed for the FreeBSD Project by Poul-Henning
8 * Kamp and Network Associates Laboratories, the Security Research Division
9 * of Network Associates, Inc. under DARPA/SPAWAR contract N66001-01-C-8035
10 * ("CBOSS"), as part of the DARPA CHATS research program
11 *
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
14 * are met:
15 * 1. Redistributions of source code must retain the above copyright
16 *    notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above copyright
18 *    notice, this list of conditions and the following disclaimer in the
19 *    documentation and/or other materials provided with the distribution.
20 * 3. The names of the authors may not be used to endorse or promote
21 *    products derived from this software without specific prior written
22 *    permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 *
36 * $FreeBSD: head/usr.sbin/extattr/rmextattr.c 104796 2002-10-10 16:37:12Z rwatson $
37 */
38
39#include <sys/types.h>
40#include <sys/uio.h>
41#include <sys/extattr.h>
42
43#include <libgen.h>
44#include <libutil.h>
45#include <stdio.h>
46#include <stdlib.h>
47#include <string.h>
48#include <unistd.h>
49#include <vis.h>
50#include <err.h>
51#include <errno.h>
52
53static enum { EADUNNO, EAGET, EASET, EARM, EALS } what = EADUNNO;
54
55static void __dead2
56usage(void)
57{
58
59	switch (what) {
60	case EAGET:
61		fprintf(stderr, "usage: getextattr [-fqsx] attrnamespace");
62		fprintf(stderr, " attrname filename ...\n");
63		exit(-1);
64	case EASET:
65		fprintf(stderr, "usage: setextattr [-fq] attrnamespace");
66		fprintf(stderr, " attrname attrvalue filename ...\n");
67		exit(-1);
68	case EARM:
69		fprintf(stderr, "usage: rmextattr [-fq] attrnamespace");
70		fprintf(stderr, " attrname filename ...\n");
71		exit(-1);
72	case EALS:
73		fprintf(stderr, "usage: lsextattr [-fq] attrnamespace");
74		fprintf(stderr, " filename ...\n");
75		exit(-1);
76	case EADUNNO:
77	default:
78		fprintf(stderr, "usage: (getextattr|lsextattr|rmextattr");
79		fprintf(stderr, "|setextattr)\n");
80		exit (-1);
81	}
82}
83
84static void
85mkbuf(char **buf, int *oldlen, int newlen)
86{
87
88	if (*oldlen >= newlen)
89		return;
90	if (*buf != NULL)
91		free(*buf);
92	*buf = malloc(newlen);
93	if (*buf == NULL)
94		err(1, "malloc");
95	*oldlen = newlen;
96	return;
97}
98
99int
100main(int argc, char *argv[])
101{
102	char	*buf, *visbuf, *p;
103
104	const char *options, *attrname;
105	int	 buflen, visbuflen, ch, error, i, arg_counter, attrnamespace;
106
107	int	flag_force = 0;
108	int	flag_quiet = 0;
109	int	flag_string = 0;
110	int	flag_hex = 0;
111
112	visbuflen = buflen = 0;
113	visbuf = buf = NULL;
114
115	p = basename(argv[0]);
116	if (p == NULL)
117		p = argv[0];
118	if (!strcmp(p, "getextattr")) {
119		what = EAGET;
120		options = "fqsx";
121	} else if (!strcmp(p, "setextattr")) {
122		what = EASET;
123		options = "fq";
124	} else if (!strcmp(p, "rmextattr")) {
125		what = EARM;
126		options = "fq";
127	} else if (!strcmp(p, "lsextattr")) {
128		what = EALS;
129		options = "fq";
130	} else {
131		usage();
132	}
133
134	while ((ch = getopt(argc, argv, options)) != -1) {
135		switch (ch) {
136		case 'f':
137			flag_force = 1;
138			break;
139		case 'q':
140			flag_quiet = 1;
141			break;
142		case 's':
143			flag_string = 1;
144			break;
145		case 'x':
146			flag_hex = 1;
147			break;
148		case '?':
149		default:
150			usage();
151		}
152	}
153
154	argc -= optind;
155	argv += optind;
156
157	if (argc < 2)
158		usage();
159
160	error = extattr_string_to_namespace(argv[0], &attrnamespace);
161	if (error)
162		err(-1, argv[0]);
163	argc--; argv++;
164
165	if (what == EALS) {
166		attrname = "";
167	} else {
168		attrname = argv[0];
169		argc--; argv++;
170	}
171
172	if (what == EASET) {
173		mkbuf(&buf, &buflen, strlen(argv[0]) + 1);
174		strcpy(buf, argv[0]);
175		argc--; argv++;
176	}
177
178	for (arg_counter = 0; arg_counter < argc; arg_counter++) {
179		switch (what) {
180		case EARM:
181			error = extattr_delete_file(argv[arg_counter],
182			    attrnamespace, attrname);
183			if (error >= 0)
184				continue;
185			break;
186		case EASET:
187			error = extattr_set_file(argv[arg_counter],
188			    attrnamespace, attrname, buf, strlen(buf));
189			if (error >= 0)
190				continue;
191			break;
192		case EALS:
193		case EAGET:
194			error = extattr_get_file(argv[arg_counter],
195			    attrnamespace, attrname, NULL, 0);
196			if (error < 0)
197				break;
198			mkbuf(&buf, &buflen, error);
199			error = extattr_get_file(argv[arg_counter],
200			    attrnamespace, attrname, buf, buflen);
201			if (error < 0)
202				break;
203			if (!flag_quiet)
204				printf("%s\t", argv[arg_counter]);
205			if (what == EALS) {
206				for (i = 0; i < error; i += buf[i] + 1)
207				    printf("%s%*.*s", i ? "\t" : "",
208					buf[i], buf[i], buf + i + 1);
209				printf("\n");
210				continue;
211			}
212			if (flag_string) {
213				mkbuf(&visbuf, &visbuflen, error * 4 + 1);
214				strvisx(visbuf, buf, error,
215				    VIS_SAFE | VIS_WHITE);
216				printf("\"%s\"\n", visbuf);
217				continue;
218			} else if (flag_hex) {
219				for (i = 0; i < error; i++)
220					printf("%s%02x", i ? " " : "",
221					    buf[i]);
222				printf("\n");
223				continue;
224			} else {
225				fwrite(buf, buflen, 1, stdout);
226				printf("\n");
227				continue;
228			}
229		default:
230			break;
231		}
232		if (!flag_quiet)
233			warn("%s: failed", argv[arg_counter]);
234		if (flag_force)
235			continue;
236		return(1);
237	}
238	return (0);
239}
240