umount.nfs: Refactor remount logic in umount.nfs
[nfs-utils.git] / utils / mount / nfsumount.c
1 /*
2  * nfsumount.c -- Linux NFS umount
3  * Copyright (C) 2006 Amit Gud <agud@redhat.com>
4  *
5  * - Basic code and wrapper around NFS umount code originally
6  *   in util-linux/mount/nfsmount.c
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2, or (at your option)
11  * any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  */
19
20 #include <unistd.h>
21 #include <stdio.h>
22 #include <errno.h>
23 #include <getopt.h>
24 #include <mntent.h>
25 #include <sys/mount.h>
26 #include <ctype.h>
27 #include <pwd.h>
28
29 #include "xcommon.h"
30 #include "fstab.h"
31 #include "nls.h"
32
33 #include "mount_constants.h"
34 #include "mount.h"
35 #include "error.h"
36 #include "network.h"
37
38 #if !defined(MNT_FORCE)
39 /* dare not try to include <linux/mount.h> -- lots of errors */
40 #define MNT_FORCE 1
41 #endif
42
43 #if !defined(MNT_DETACH)
44 #define MNT_DETACH 2
45 #endif
46
47 extern char *progname;
48 extern int nomtab;
49 extern int verbose;
50 int force;
51 int lazy;
52 int remount;
53
54
55 static int try_remount(const char *spec, const char *node)
56 {
57         int res;
58
59         res = mount(spec, node, NULL,
60                     MS_MGC_VAL | MS_REMOUNT | MS_RDONLY, NULL);
61         if (res == 0) {
62                 struct mntent remnt;
63                 nfs_error(_("%s: %s busy - remounted read-only"),
64                                 progname, spec);
65                 remnt.mnt_type = remnt.mnt_fsname = NULL;
66                 remnt.mnt_dir = xstrdup(node);
67                 remnt.mnt_opts = xstrdup("ro");
68                 if (!nomtab)
69                         update_mtab(node, &remnt);
70         } else if (errno != EBUSY) {    /* hmm ... */
71                 perror(_("remount"));
72                 nfs_error(_("%s: could not remount %s read-only"),
73                                 progname, spec);
74         }
75         return res;
76 }
77
78 static int del_mtab(const char *spec, const char *node)
79 {
80         int umnt_err, res;
81
82         umnt_err = 0;
83         if (lazy) {
84                 res = umount2 (node, MNT_DETACH);
85                 if (res < 0)
86                         umnt_err = errno;
87                 goto writemtab;
88         }
89
90         if (force) {
91                 res = umount2 (node, MNT_FORCE);
92                 if (res == -1) {
93                         int errsv = errno;
94                         perror(_("umount2"));
95                         errno = errsv;
96                         if (errno == ENOSYS) {
97                                 if (verbose)
98                                         printf(_("no umount2, trying umount...\n"));
99                                 res = umount (node);
100                         }
101                 }
102         } else
103                 res = umount (node);
104
105         if (res < 0 && remount && errno == EBUSY && spec) {
106                 res = try_remount(spec, node);
107                 if (!res)
108                         return 0;
109         }
110
111         if (res >= 0) {
112                 /* Umount succeeded */
113                 if (verbose)
114                         printf(_("%s umounted\n"), spec ? spec : node);
115         }
116
117  writemtab:
118         if (!nomtab &&
119             (umnt_err == 0 || umnt_err == EINVAL || umnt_err == ENOENT)) {
120                 update_mtab(node, NULL);
121         }
122
123         if (res >= 0)
124                 return 0;
125
126         if (umnt_err)
127                 umount_error(umnt_err, node);
128         return EX_FILEIO;
129 }
130
131 /*
132  * Pick up certain mount options used during the original mount
133  * from /etc/mtab.  The basics include the server's IP address and
134  * the server pathname of the share to unregister.
135  *
136  * These options might also describe the mount port, mount protocol
137  * version, and transport protocol used to punch through a firewall.
138  * We will need this information to get through the firewall again
139  * to do the umount.
140  */
141 static int do_nfs_umount(const char *spec, char *opts)
142 {
143         char *hostname;
144         char *dirname;
145         clnt_addr_t mnt_server = { &hostname, };
146         struct mntent mnt = { .mnt_opts = opts };
147         struct pmap *pmap = &mnt_server.pmap;
148         char *p;
149
150         if (spec == NULL) {
151                 nfs_error(_("%s: No NFS export name was provided"),
152                                 progname);
153                 return EX_USAGE;
154         }
155         
156         p = strchr(spec, ':');
157         if (p == NULL) {
158                 nfs_error(_("%s: '%s' is not a legal NFS export name"),
159                                 progname, spec);
160                 return EX_USAGE;
161         }
162         hostname = xstrndup(spec, p - spec);
163         dirname = xstrdup(p + 1);
164 #ifdef NFS_MOUNT_DEBUG
165         printf(_("host: %s, directory: %s\n"), hostname, dirname);
166 #endif
167
168         if (opts && (p = strstr(opts, "addr="))) {
169                 char *q;
170
171                 free(hostname);
172                 p += 5;
173                 q = p;
174                 while (*q && *q != ',') q++;
175                 hostname = xstrndup(p,q-p);
176         }
177
178         if (opts && (p = strstr(opts, "mounthost="))) {
179                 char *q;
180
181                 free(hostname);
182                 p += 10;
183                 q = p;
184                 while (*q && *q != ',') q++;
185                 hostname = xstrndup(p,q-p);
186         }
187
188         pmap->pm_prog = MOUNTPROG;
189         pmap->pm_vers = MOUNTVERS_NFSV3;
190         if (opts && (p = strstr(opts, "mountprog=")) && isdigit(*(p+10)))
191                 pmap->pm_prog = atoi(p+10);
192         if (opts && (p = strstr(opts, "mountport=")) && isdigit(*(p+10)))
193                 pmap->pm_port = atoi(p+10);
194         if (opts && hasmntopt(&mnt, "v2"))
195                 pmap->pm_vers = nfsvers_to_mnt(2);
196         if (opts && hasmntopt(&mnt, "v3"))
197                 pmap->pm_vers = nfsvers_to_mnt(3);
198         if (opts && hasmntopt(&mnt, "v4"))
199                 pmap->pm_vers = nfsvers_to_mnt(4);
200         if (opts && (p = strstr(opts, "vers=")) && isdigit(*(p+5)))
201                 pmap->pm_vers = nfsvers_to_mnt(atoi(p+5));
202         if (opts && (p = strstr(opts, "mountvers=")) && isdigit(*(p+10)))
203                 pmap->pm_vers = atoi(p+10);
204         if (opts && (hasmntopt(&mnt, "udp") || hasmntopt(&mnt, "proto=udp")))
205                 pmap->pm_prot = IPPROTO_UDP;
206         if (opts && (hasmntopt(&mnt, "tcp") || hasmntopt(&mnt, "proto=tcp")))
207                 pmap->pm_prot = IPPROTO_TCP;
208
209         if (!nfs_gethostbyname(hostname, &mnt_server.saddr)) {
210                 nfs_error(_("%s: '%s' does not contain a recognized hostname"),
211                                 progname, spec);
212                 return EX_USAGE;
213         }
214
215         if (!nfs_call_umount(&mnt_server, &dirname)) {
216                 nfs_error(_("%s: Server failed to unmount '%s'"),
217                                 progname, spec);
218                 return EX_USAGE;
219         }
220
221         return EX_SUCCESS;
222 }
223
224 static struct option umount_longopts[] =
225 {
226   { "force", 0, 0, 'f' },
227   { "help", 0, 0, 'h' },
228   { "no-mtab", 0, 0, 'n' },
229   { "verbose", 0, 0, 'v' },
230   { "read-only", 0, 0, 'r' },
231   { NULL, 0, 0, 0 }
232 };
233
234 static void umount_usage(void)
235 {
236         printf(_("usage: %s dir [-fvnrlh]\n"), progname);
237         printf(_("options:\n\t-f\t\tforce unmount\n"));
238         printf(_("\t-v\tverbose\n"));
239         printf(_("\t-n\tDo not update /etc/mtab\n"));
240         printf(_("\t-r\tremount\n"));
241         printf(_("\t-l\tlazy unmount\n"));
242         printf(_("\t-h\tprint this help\n\n"));
243 }
244
245 int nfsumount(int argc, char *argv[])
246 {
247         int c, ret;
248         char *spec;
249         struct mntentchn *mc;
250
251         if (argc < 2) {
252                 umount_usage();
253                 return EX_USAGE;
254         }
255
256         spec = argv[1];
257
258         argv += 1;
259         argc -= 1;
260
261         argv[0] = argv[-1]; /* So that getopt error messages are correct */
262         while ((c = getopt_long (argc, argv, "fvnrlh",
263                                 umount_longopts, NULL)) != -1) {
264
265                 switch (c) {
266                 case 'f':
267                         ++force;
268                         break;
269                 case 'v':
270                         ++verbose;
271                         break;
272                 case 'n':
273                         ++nomtab;
274                         break;
275                 case 'r':
276                         ++remount;
277                         break;
278                 case 'l':
279                         ++lazy;
280                         break;
281                 case 'h':
282                 default:
283                         umount_usage();
284                         return EX_USAGE;
285                 }
286         }
287         if (optind != argc) {
288                 umount_usage();
289                 return EX_USAGE;
290         }
291         
292         if (spec == NULL || (*spec != '/' && strchr(spec,':') == NULL)) {
293                 nfs_error(_("%s: %s: not found\n"), progname, spec);
294                 return EX_USAGE;
295         }
296
297         if (*spec == '/')
298                 mc = getmntdirbackward(spec, NULL);
299         else
300                 mc = getmntdevbackward(spec, NULL);
301         if (!mc && verbose)
302                 printf(_("Could not find %s in mtab\n"), spec);
303
304         if (mc && strcmp(mc->m.mnt_type, "nfs") != 0 &&
305             strcmp(mc->m.mnt_type, "nfs4") != 0) {
306                 nfs_error(_("%s: %s on %s is not an NFS filesystem"),
307                                 progname, mc->m.mnt_fsname, mc->m.mnt_dir);
308                 return EX_USAGE;
309         }
310
311         if (getuid() != 0) {
312                 /* only permitted if "user=" or "users" is in mount options */
313                 if (!mc) {
314                         /* umount might call us twice.  The second time there will
315                          * be no entry in mtab and we should just exit quietly
316                          */
317                         return EX_SUCCESS;
318
319                 only_root:
320                         nfs_error(_("%s: You are not permitted to unmount %s"),
321                                         progname, spec);
322                         return EX_USAGE;
323                 }
324                 if (hasmntopt(&mc->m, "users") == NULL) {
325                         char *opt = hasmntopt(&mc->m, "user");
326                         struct passwd *pw;
327                         char *comma;
328                         int len;
329                         if (!opt)
330                                 goto only_root;
331                         if (opt[4] != '=')
332                                 goto only_root;
333                         comma = strchr(opt, ',');
334                         if (comma)
335                                 len = comma - (opt + 5);
336                         else
337                                 len = strlen(opt+5);
338                         pw = getpwuid(getuid());
339                         if (pw == NULL || strlen(pw->pw_name) != len
340                             || strncmp(pw->pw_name, opt+5, len) != 0)
341                                 goto only_root;
342                 }
343         }
344
345         ret = 0;
346         if (mc) {
347                 if (!lazy)
348                         ret = do_nfs_umount(mc->m.mnt_fsname, mc->m.mnt_opts);
349                 if (!ret || force)
350                         ret = del_mtab(mc->m.mnt_fsname, mc->m.mnt_dir);
351         } else if (*spec != '/') {
352                 if (!lazy)
353                         ret = do_nfs_umount(spec, "tcp,v3");
354         } else
355                 ret = del_mtab(NULL, spec);
356
357         return ret;
358 }