1 /*
2 * Copyright (c) 2017-2018 The DragonFly Project. All rights reserved.
3 *
4 * This code is derived from software contributed to The DragonFly Project
5 * by Matthew Dillon <dillon@dragonflybsd.org>
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in
15 * the documentation and/or other materials provided with the
16 * distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
21 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
24 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
27 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
28 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #include "hammer2.h"
33
34 static int docleanup(const char *path);
35 static int sameh2prefix(const char *from);
36 static void saveh2prefix(const char *from);
37 static void freeh2prefixes(void);
38
39 int
cmd_cleanup(const char * sel_path)40 cmd_cleanup(const char *sel_path)
41 {
42 struct statfs *fsary;
43 char *fstype;
44 char *path;
45 char *from;
46 int i;
47 int n;
48 int r;
49 int rc;
50
51 if (sel_path)
52 return (docleanup(sel_path));
53
54 n = getmntinfo(&fsary, MNT_NOWAIT);
55 if (n <= 0) {
56 fprintf(stderr, "hammer2 cleanup: no HAMMER2 mounts\n");
57 return 0;
58 }
59 rc = 0;
60 for (i = 0; i < n; ++i) {
61 fstype = fsary[i].f_fstypename;
62 path = fsary[i].f_mntonname;
63 from = fsary[i].f_mntfromname;
64
65 if (strcmp(fstype, "hammer2") != 0)
66 continue;
67 if (sameh2prefix(from)) {
68 printf("hammer2 cleanup \"%s\" (same partition)\n",
69 path);
70 } else {
71 r = docleanup(path);
72 if (r)
73 rc = r;
74 saveh2prefix(from);
75 }
76 }
77 freeh2prefixes();
78
79 return rc;
80 }
81
82 static
83 int
docleanup(const char * path)84 docleanup(const char *path)
85 {
86 int rc;
87
88 printf("hammer2 cleanup \"%s\"\n", path);
89 rc = cmd_bulkfree(path);
90
91 return rc;
92 }
93
94 static char **h2prefixes;
95 static int h2count;
96 static int h2limit;
97
98 static
99 int
sameh2prefix(const char * from)100 sameh2prefix(const char *from)
101 {
102 char *ptr;
103 int rc = 0;
104 int i;
105
106 ptr = strdup(from);
107 if (strchr(ptr, '@'))
108 *strchr(ptr, '@') = 0;
109 for (i = 0; i < h2count; ++i) {
110 if (strcmp(h2prefixes[i], ptr) == 0) {
111 rc = 1;
112 break;
113 }
114 }
115 free(ptr);
116
117 return rc;
118 }
119
120 static
121 void
saveh2prefix(const char * from)122 saveh2prefix(const char *from)
123 {
124 char *ptr;
125
126 if (h2count >= h2limit) {
127 h2limit = (h2limit + 8) << 1;
128 h2prefixes = realloc(h2prefixes, sizeof(char *) * h2limit);
129 }
130 ptr = strdup(from);
131 if (strchr(ptr, '@'))
132 *strchr(ptr, '@') = 0;
133 h2prefixes[h2count++] = ptr;
134 }
135
136 static
137 void
freeh2prefixes(void)138 freeh2prefixes(void)
139 {
140 int i;
141
142 for (i = 0; i < h2count; ++i)
143 free(h2prefixes[i]);
144 free(h2prefixes);
145 h2prefixes = NULL;
146 h2count = 0;
147 h2limit = 0;
148 }
149