summaryrefslogtreecommitdiff
path: root/util/sbase/rm.c
blob: 3adfc8cdc97bdb71c2506410002b1f61aca59821 (plain)
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
/* See LICENSE file for copyright and license details. */
#include <fcntl.h>
#include  <string.h>

#include "fs.h"
#include "util.h"

static void
usage(void)
{
	eprintf("usage: %s [-f] [-iRr] file ...\n", argv0);
}

static int
forbidden(char *path, struct stat *root)
{
	char *s, *t;
	size_t n;
	struct stat st;
	static int w1, w2;

	n = strlen(path);
	for (t = path + n; t > path && t[-1] == '/'; --t)
		;
	for (s = t; s > path && s[-1] != '/'; --s)
		;
	n = t - s;
	if (n == 1 && *s == '.' || n == 2 && s[0] == '.' && s[1] == '.') {
		if (!w1)
			weprintf("\".\" and \"..\" may not be removed\n");
		w1 = 1;
		return 1;
	}

	if (stat(path, &st) < 0)
		return 0;
	if (st.st_dev == root->st_dev && st.st_ino == root->st_ino) {
		if (!w2)
			weprintf("\"/\" may not be removed\n");
		w2 = 1;
		return 1;
	}

	return 0;
}

int
main(int argc, char *argv[])
{
	char *s;
	struct stat st;
	struct recursor r = { .fn = rm, .maxdepth = 1, .follow = 'P' };

	ARGBEGIN {
	case 'f':
		r.flags |= SILENT | IGNORE;
		break;
	case 'i':
		r.flags |= CONFIRM;
		break;
	case 'R':
	case 'r':
		r.maxdepth = 0;
		break;
	default:
		usage();
	} ARGEND

	if (!argc) {
		if (!(r.flags & IGNORE))
			usage();
		else
			return 0;
	}

	if (stat("/", &st) < 0)
		eprintf("stat root:");
	for (; *argv; argc--, argv++) {
		if (forbidden(*argv, &st)) {
			rm_status = 1;
			continue;
		}
		recurse(AT_FDCWD, *argv, NULL, &r);
	}

	return rm_status || recurse_status;
}