c8ce6b2db54e305d6cf57aeb8541584490b44e7f
[nfs-utils.git] / utils / idmapd / idmapd.c
1 /*
2  *  idmapd.c
3  *
4  *  Userland daemon for idmap.
5  *
6  *  Copyright (c) 2002 The Regents of the University of Michigan.
7  *  All rights reserved.
8  *
9  *  Marius Aamodt Eriksen <marius@umich.edu>
10  *
11  *  Redistribution and use in source and binary forms, with or without
12  *  modification, are permitted provided that the following conditions
13  *  are met:
14  *
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. Neither the name of the University nor the names of its
21  *     contributors may be used to endorse or promote products derived
22  *     from this software without specific prior written permission.
23  *
24  *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
25  *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
26  *  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
27  *  DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  *  FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
29  *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
30  *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
31  *  BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
32  *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
33  *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
34  *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35  */
36
37 #include <sys/types.h>
38 #include <sys/time.h>
39 #include <sys/poll.h>
40 #include <sys/socket.h>
41 #include <sys/stat.h>
42 #include <time.h>
43
44 #include "nfs_idmap.h"
45
46 #include <err.h>
47 #include <errno.h>
48 #include <event.h>
49 #include <fcntl.h>
50 #include <dirent.h>
51 #include <unistd.h>
52 #include <netdb.h>
53 #include <signal.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <stdarg.h>
58 #include <syslog.h>
59 #include <pwd.h>
60 #include <grp.h>
61 #include <limits.h>
62 #include <ctype.h>
63 #include <nfsidmap.h>
64
65 #ifdef HAVE_CONFIG_H
66 #include "config.h"
67 #endif /* HAVE_CONFIG_H */
68
69 #include "cfg.h"
70 #include "queue.h"
71 #include "nfslib.h"
72
73 #ifndef PIPEFS_DIR
74 #define PIPEFS_DIR  "/var/lib/nfs/rpc_pipefs/"
75 #endif
76
77 #ifndef NFSD_DIR
78 #define NFSD_DIR  "/proc/net/rpc"
79 #endif
80
81 #ifndef CLIENT_CACHE_TIMEOUT_FILE
82 #define CLIENT_CACHE_TIMEOUT_FILE "/proc/sys/fs/nfs/idmap_cache_timeout"
83 #endif
84
85 #ifndef NFS4NOBODY_USER
86 #define NFS4NOBODY_USER "nobody"
87 #endif
88
89 #ifndef NFS4NOBODY_GROUP
90 #define NFS4NOBODY_GROUP "nobody"
91 #endif
92
93 /* From Niels */
94 #define CONF_SAVE(w, f) do {                    \
95         char *p = f;                            \
96         if (p != NULL)                          \
97                 (w) = p;                        \
98 } while (0)
99
100 #define IC_IDNAME 0
101 #define IC_IDNAME_CHAN  NFSD_DIR "/nfs4.idtoname/channel"
102 #define IC_IDNAME_FLUSH NFSD_DIR "/nfs4.idtoname/flush"
103
104 #define IC_NAMEID 1
105 #define IC_NAMEID_CHAN  NFSD_DIR "/nfs4.nametoid/channel"
106 #define IC_NAMEID_FLUSH NFSD_DIR "/nfs4.nametoid/flush"
107
108 struct idmap_client {
109         short                      ic_which;
110         char                       ic_clid[30];
111         char                      *ic_id;
112         char                       ic_path[PATH_MAX];
113         int                        ic_fd;
114         int                        ic_dirfd;
115         int                        ic_scanned;
116         struct event               ic_event;
117         TAILQ_ENTRY(idmap_client)  ic_next;
118 };
119 static struct idmap_client nfsd_ic[2] = {
120 {IC_IDNAME, "Server", "", IC_IDNAME_CHAN, -1, -1, 0},
121 {IC_NAMEID, "Server", "", IC_NAMEID_CHAN, -1, -1, 0},
122 };
123
124 TAILQ_HEAD(idmap_clientq, idmap_client);
125
126 static void dirscancb(int, short, void *);
127 static void clntscancb(int, short, void *);
128 static void svrreopen(int, short, void *);
129 static int  nfsopen(struct idmap_client *);
130 static void nfscb(int, short, void *);
131 static void nfsdcb(int, short, void *);
132 static int  validateascii(char *, u_int32_t);
133 static int  addfield(char **, ssize_t *, char *);
134 static int  getfield(char **, char *, size_t);
135
136 static void imconv(struct idmap_client *, struct idmap_msg *);
137 static void idtonameres(struct idmap_msg *);
138 static void nametoidres(struct idmap_msg *);
139
140 static int nfsdopen();
141 static int nfsdopenone(struct idmap_client *);
142 static void nfsdreopen(void);
143
144 size_t  strlcat(char *, const char *, size_t);
145 size_t  strlcpy(char *, const char *, size_t);
146 ssize_t atomicio(ssize_t (*)(), int, void *, size_t);
147 void    mydaemon(int, int);
148 void    release_parent();
149
150 static int verbose = 0;
151 #define DEFAULT_IDMAP_CACHE_EXPIRY 600 /* seconds */
152 static int cache_entry_expiration = 0;
153 static char pipefsdir[PATH_MAX];
154 static char *nobodyuser, *nobodygroup;
155 static uid_t nobodyuid;
156 static gid_t nobodygid;
157
158 /* Used by cfg.c */
159 char *conf_path;
160
161 static int
162 flush_nfsd_cache(char *path, time_t now)
163 {
164         int fd;
165         char stime[20];
166
167         sprintf(stime, "%ld\n", now);
168         fd = open(path, O_RDWR);
169         if (fd == -1)
170                 return -1;
171         write(fd, stime, strlen(stime));
172         close(fd);
173         return 0;
174 }
175
176 static int
177 flush_nfsd_idmap_cache(void)
178 {
179         time_t now = time(NULL);
180         int ret;
181
182         ret = flush_nfsd_cache(IC_IDNAME_FLUSH, now);
183         if (ret)
184                 return ret;
185         ret = flush_nfsd_cache(IC_NAMEID_FLUSH, now);
186         return ret;
187 }
188
189 static void
190 msg_format(char *rtnbuff, int rtnbuffsize, int errval,
191            const char *fmt, va_list args)
192 {
193         char buff[1024];
194         int n;
195
196         vsnprintf(buff, sizeof(buff), fmt, args);
197
198         if ((n = strlen(buff)) > 0 && buff[n-1] == '\n')
199                 buff[--n] = '\0';
200
201         snprintf(rtnbuff, rtnbuffsize, "%s: %s", buff, strerror(errval));
202 }
203
204 static void
205 idmapd_warn(const char *fmt, ...)
206 {
207         int errval = errno;     /* save this! */
208         char buff[1024];
209         va_list args;
210
211         va_start(args, fmt);
212         msg_format(buff, sizeof(buff), errval, fmt, args);
213         va_end(args);
214
215         syslog(LOG_WARNING, "%s", buff);
216 }
217
218 static void
219 idmapd_warnx(const char *fmt, ...)
220 {
221         va_list args;
222
223         va_start(args, fmt);
224         vsyslog(LOG_WARNING, fmt, args);
225         va_end(args);
226 }
227
228 static void
229 idmapd_err(int eval, const char *fmt, ...)
230 {
231         int errval = errno;     /* save this! */
232         char buff[1024];
233         va_list args;
234
235         va_start(args, fmt);
236         msg_format(buff, sizeof(buff), errval, fmt, args);
237         va_end(args);
238
239         syslog(LOG_ERR, "%s", buff);
240         exit(eval);
241 }
242
243 static void
244 idmapd_errx(int eval, const char *fmt, ...)
245 {
246         va_list args;
247
248         va_start(args, fmt);
249         vsyslog(LOG_ERR, fmt, args);
250         va_end(args);
251         exit(eval);
252 }
253
254 int
255 main(int argc, char **argv)
256 {
257         int fd = 0, opt, fg = 0, nfsdret = -1;
258         struct idmap_clientq icq;
259         struct event rootdirev, clntdirev, svrdirev;
260         struct event initialize;
261         struct passwd *pw;
262         struct group *gr;
263         struct stat sb;
264         char *xpipefsdir = NULL;
265         int serverstart = 1, clientstart = 1;
266         int ret;
267         char *progname;
268
269         conf_path = _PATH_IDMAPDCONF;
270         nobodyuser = NFS4NOBODY_USER;
271         nobodygroup = NFS4NOBODY_GROUP;
272         strlcpy(pipefsdir, PIPEFS_DIR, sizeof(pipefsdir));
273
274         if ((progname = strrchr(argv[0], '/')))
275                 progname++;
276         else
277                 progname = argv[0];
278         openlog(progname, LOG_PID, LOG_DAEMON);
279
280 #define GETOPTSTR "vfd:p:U:G:c:CS"
281         opterr=0; /* Turn off error messages */
282         while ((opt = getopt(argc, argv, GETOPTSTR)) != -1) {
283                 if (opt == 'c')
284                         conf_path = optarg;
285                 if (opt == '?') {
286                         if (strchr(GETOPTSTR, optopt))
287                                 errx(1, "'-%c' option requires an argument.", optopt);
288                         else
289                                 errx(1, "'-%c' is an invalid argument.", optopt);
290                 }
291         }
292         optind = 1;
293
294         if (stat(conf_path, &sb) == -1 && (errno == ENOENT || errno == EACCES)) {
295                 warn("Skipping configuration file \"%s\"", conf_path);
296                 conf_path = NULL;
297         } else {
298                 conf_init();
299                 verbose = conf_get_num("General", "Verbosity", 0);
300                 cache_entry_expiration = conf_get_num("General",
301                                 "Cache-Expiration", DEFAULT_IDMAP_CACHE_EXPIRY);
302                 CONF_SAVE(xpipefsdir, conf_get_str("General", "Pipefs-Directory"));
303                 if (xpipefsdir != NULL)
304                         strlcpy(pipefsdir, xpipefsdir, sizeof(pipefsdir));
305                 CONF_SAVE(nobodyuser, conf_get_str("Mapping", "Nobody-User"));
306                 CONF_SAVE(nobodygroup, conf_get_str("Mapping", "Nobody-Group"));
307         }
308
309         while ((opt = getopt(argc, argv, GETOPTSTR)) != -1)
310                 switch (opt) {
311                 case 'v':
312                         verbose++;
313                         break;
314                 case 'f':
315                         fg = 1;
316                         break;
317                 case 'p':
318                         strlcpy(pipefsdir, optarg, sizeof(pipefsdir));
319                         break;
320                 case 'd':
321                 case 'U':
322                 case 'G':
323                         errx(1, "the -d, -U, and -G options have been removed;"
324                                 " please use the configuration file instead.");
325                 case 'C':
326                         serverstart = 0;
327                         break;
328                 case 'S':
329                         clientstart = 0;
330                         break;
331                 default:
332                         break;
333                 }
334
335         if (!serverstart && !clientstart)
336                 errx(1, "it is illegal to specify both -C and -S");
337
338         strncat(pipefsdir, "/nfs", sizeof(pipefsdir));
339
340         if ((pw = getpwnam(nobodyuser)) == NULL)
341                 errx(1, "Could not find user \"%s\"", nobodyuser);
342         nobodyuid = pw->pw_uid;
343
344         if ((gr = getgrnam(nobodygroup)) == NULL)
345                 errx(1, "Could not find group \"%s\"", nobodygroup);
346         nobodygid = gr->gr_gid;
347
348 #ifdef HAVE_NFS4_SET_DEBUG
349         nfs4_set_debug(verbose, idmapd_warnx);
350 #endif
351         if (conf_path == NULL)
352                 conf_path = _PATH_IDMAPDCONF;
353         if (nfs4_init_name_mapping(conf_path))
354                 errx(1, "Unable to create name to user id mappings.");
355
356         if (!fg)
357                 mydaemon(0, 0);
358
359         event_init();
360
361         if (verbose > 0)
362                 idmapd_warnx("Expiration time is %d seconds.",
363                              cache_entry_expiration);
364         if (serverstart) {
365                 nfsdret = nfsdopen();
366                 if (nfsdret == 0) {
367                         ret = flush_nfsd_idmap_cache();
368                         if (ret)
369                                 idmapd_errx(1,
370                                         "main: Failed to flush nfsd idmap cache\n");
371                 }
372         }
373
374         if (clientstart) {
375                 struct timeval now = {
376                         .tv_sec = 0,
377                         .tv_usec = 0,
378                 };
379
380                 if (cache_entry_expiration != DEFAULT_IDMAP_CACHE_EXPIRY) {
381                         int timeout_fd, len;
382                         char timeout_buf[12];
383                         if ((timeout_fd = open(CLIENT_CACHE_TIMEOUT_FILE,
384                                                O_RDWR)) == -1) {
385                                 idmapd_warnx("Unable to open '%s' to set "
386                                              "client cache expiration time "
387                                              "to %d seconds\n",
388                                              CLIENT_CACHE_TIMEOUT_FILE,
389                                              cache_entry_expiration);
390                         } else {
391                                 len = snprintf(timeout_buf, sizeof(timeout_buf),
392                                                "%d", cache_entry_expiration);
393                                 if ((write(timeout_fd, timeout_buf, len)) != len)
394                                         idmapd_warnx("Error writing '%s' to "
395                                                      "'%s' to set client "
396                                                      "cache expiration time\n",
397                                                      timeout_buf,
398                                                      CLIENT_CACHE_TIMEOUT_FILE);
399                                 close(timeout_fd);
400                         }
401                 }
402
403                 if ((fd = open(pipefsdir, O_RDONLY)) == -1)
404                         idmapd_err(1, "main: open(%s)", pipefsdir);
405
406                 if (fcntl(fd, F_SETSIG, SIGUSR1) == -1)
407                         idmapd_err(1, "main: fcntl(%s)", pipefsdir);
408
409                 if (fcntl(fd, F_NOTIFY,
410                         DN_CREATE | DN_DELETE | DN_MODIFY | DN_MULTISHOT) == -1)
411                         idmapd_err(1, "main: fcntl(%s)", pipefsdir);
412
413                 TAILQ_INIT(&icq);
414
415                 /* These events are persistent */
416                 signal_set(&rootdirev, SIGUSR1, dirscancb, &icq);
417                 signal_add(&rootdirev, NULL);
418                 signal_set(&clntdirev, SIGUSR2, clntscancb, &icq);
419                 signal_add(&clntdirev, NULL);
420                 signal_set(&svrdirev, SIGHUP, svrreopen, NULL);
421                 signal_add(&svrdirev, NULL);
422
423                 /* Fetch current state */
424                 /* (Delay till start of event_dispatch to avoid possibly losing
425                  * a SIGUSR1 between here and the call to event_dispatch().) */
426                 evtimer_set(&initialize, dirscancb, &icq);
427                 evtimer_add(&initialize, &now);
428         }
429
430         if (nfsdret != 0 && fd == 0)
431                 idmapd_errx(1, "main: Neither NFS client nor NFSd found");
432
433         release_parent();
434
435         if (event_dispatch() < 0)
436                 idmapd_errx(1, "main: event_dispatch returns errno %d (%s)",
437                             errno, strerror(errno));
438         /* NOTREACHED */
439         return 1;
440 }
441
442 static void
443 dirscancb(int fd, short which, void *data)
444 {
445         int nent, i;
446         struct dirent **ents;
447         struct idmap_client *ic;
448         char path[PATH_MAX];
449         struct idmap_clientq *icq = data;
450
451         nent = scandir(pipefsdir, &ents, NULL, alphasort);
452         if (nent == -1) {
453                 idmapd_warn("dirscancb: scandir(%s)", pipefsdir);
454                 return;
455         }
456
457         for (i = 0;  i < nent; i++) {
458                 if (ents[i]->d_reclen > 4 &&
459                     strncmp(ents[i]->d_name, "clnt", 4) == 0) {
460                         TAILQ_FOREACH(ic, icq, ic_next)
461                             if (strcmp(ents[i]->d_name + 4, ic->ic_clid) == 0)
462                                     break;
463                         if (ic != NULL)
464                                 goto next;
465
466                         if ((ic = calloc(1, sizeof(*ic))) == NULL)
467                                 return;
468                         strlcpy(ic->ic_clid, ents[i]->d_name + 4,
469                             sizeof(ic->ic_clid));
470                         path[0] = '\0';
471                         snprintf(path, sizeof(path), "%s/%s",
472                             pipefsdir, ents[i]->d_name);
473
474                         if ((ic->ic_dirfd = open(path, O_RDONLY, 0)) == -1) {
475                                 idmapd_warn("dirscancb: open(%s)", path);
476                                 free(ic);
477                                 return;
478                         }
479
480                         strlcat(path, "/idmap", sizeof(path));
481                         strlcpy(ic->ic_path, path, sizeof(ic->ic_path));
482
483                         if (verbose > 0)
484                                 idmapd_warnx("New client: %s", ic->ic_clid);
485
486                         if (nfsopen(ic) == -1) {
487                                 close(ic->ic_dirfd);
488                                 free(ic);
489                                 return;
490                         }
491
492                         ic->ic_id = "Client";
493
494                         TAILQ_INSERT_TAIL(icq, ic, ic_next);
495
496                 next:
497                         ic->ic_scanned = 1;
498                 }
499         }
500
501         TAILQ_FOREACH(ic, icq, ic_next) {
502                 if (!ic->ic_scanned) {
503                         event_del(&ic->ic_event);
504                         close(ic->ic_fd);
505                         close(ic->ic_dirfd);
506                         TAILQ_REMOVE(icq, ic, ic_next);
507                         if (verbose > 0) {
508                                 idmapd_warnx("Stale client: %s", ic->ic_clid);
509                                 idmapd_warnx("\t-> closed %s", ic->ic_path);
510                         }
511                         free(ic);
512                 } else
513                         ic->ic_scanned = 0;
514         }
515         return;
516 }
517
518 static void
519 svrreopen(int fd, short which, void *data)
520 {
521         nfsdreopen();
522 }
523
524 static void
525 clntscancb(int fd, short which, void *data)
526 {
527         struct idmap_clientq *icq = data;
528         struct idmap_client *ic;
529
530         TAILQ_FOREACH(ic, icq, ic_next)
531                 if (ic->ic_fd == -1 && nfsopen(ic) == -1) {
532                         close(ic->ic_dirfd);
533                         TAILQ_REMOVE(icq, ic, ic_next);
534                         free(ic);
535                 }
536 }
537
538 static void
539 nfsdcb(int fd, short which, void *data)
540 {
541         struct idmap_client *ic = data;
542         struct idmap_msg im;
543         u_char buf[IDMAP_MAXMSGSZ + 1];
544         size_t len, bsiz;
545         char *bp, typebuf[IDMAP_MAXMSGSZ],
546                 buf1[IDMAP_MAXMSGSZ], authbuf[IDMAP_MAXMSGSZ], *p;
547         unsigned long tmp;
548
549         if (which != EV_READ)
550                 goto out;
551
552         if ((len = read(ic->ic_fd, buf, sizeof(buf))) <= 0) {
553                 idmapd_warnx("nfsdcb: read(%s) failed: errno %d (%s)",
554                              ic->ic_path, len?errno:0, 
555                              len?strerror(errno):"End of File");
556                 goto out;
557         }
558
559         /* Get rid of newline and terminate buffer*/
560         buf[len - 1] = '\0';
561         bp = buf;
562
563         memset(&im, 0, sizeof(im));
564
565         /* Authentication name -- ignored for now*/
566         if (getfield(&bp, authbuf, sizeof(authbuf)) == -1) {
567                 idmapd_warnx("nfsdcb: bad authentication name in upcall\n");
568                 return;
569         }
570         if (getfield(&bp, typebuf, sizeof(typebuf)) == -1) {
571                 idmapd_warnx("nfsdcb: bad type in upcall\n");
572                 return;
573         }
574         if (verbose > 0)
575                 idmapd_warnx("nfsdcb: authbuf=%s authtype=%s",
576                              authbuf, typebuf);
577
578         im.im_type = strcmp(typebuf, "user") == 0 ?
579                 IDMAP_TYPE_USER : IDMAP_TYPE_GROUP;
580
581         switch (ic->ic_which) {
582         case IC_NAMEID:
583                 im.im_conv = IDMAP_CONV_NAMETOID;
584                 if (getfield(&bp, im.im_name, sizeof(im.im_name)) == -1) {
585                         idmapd_warnx("nfsdcb: bad name in upcall\n");
586                         return;
587                 }
588                 break;
589         case IC_IDNAME:
590                 im.im_conv = IDMAP_CONV_IDTONAME;
591                 if (getfield(&bp, buf1, sizeof(buf1)) == -1) {
592                         idmapd_warnx("nfsdcb: bad id in upcall\n");
593                         return;
594                 }
595                 tmp = strtoul(buf1, (char **)NULL, 10);
596                 im.im_id = (u_int32_t)tmp;
597                 if ((tmp == ULONG_MAX && errno == ERANGE)
598                                 || (unsigned long)im.im_id != tmp) {
599                         idmapd_warnx("nfsdcb: id '%s' too big!\n", buf1);
600                         return;
601                 }
602                 break;
603         default:
604                 idmapd_warnx("nfsdcb: Unknown which type %d", ic->ic_which);
605                 return;
606         }
607
608         imconv(ic, &im);
609
610         buf[0] = '\0';
611         bp = buf;
612         bsiz = sizeof(buf);
613
614         /* Authentication name */
615         addfield(&bp, &bsiz, authbuf);
616
617         switch (ic->ic_which) {
618         case IC_NAMEID:
619                 /* Type */
620                 p = im.im_type == IDMAP_TYPE_USER ? "user" : "group";
621                 addfield(&bp, &bsiz, p);
622                 /* Name */
623                 addfield(&bp, &bsiz, im.im_name);
624                 /* expiry */
625                 snprintf(buf1, sizeof(buf1), "%lu",
626                          time(NULL) + cache_entry_expiration);
627                 addfield(&bp, &bsiz, buf1);
628                 /* ID */
629                 snprintf(buf1, sizeof(buf1), "%u", im.im_id);
630                 addfield(&bp, &bsiz, buf1);
631
632                 //if (bsiz == sizeof(buf)) /* XXX */
633
634                 bp[-1] = '\n';
635
636                 break;
637         case IC_IDNAME:
638                 /* Type */
639                 p = im.im_type == IDMAP_TYPE_USER ? "user" : "group";
640                 addfield(&bp, &bsiz, p);
641                 /* ID */
642                 snprintf(buf1, sizeof(buf1), "%u", im.im_id);
643                 addfield(&bp, &bsiz, buf1);
644                 /* expiry */
645                 snprintf(buf1, sizeof(buf1), "%lu",
646                          time(NULL) + cache_entry_expiration);
647                 addfield(&bp, &bsiz, buf1);
648                 /* Name */
649                 addfield(&bp, &bsiz, im.im_name);
650
651                 bp[-1] = '\n';
652
653                 break;
654         default:
655                 idmapd_warnx("nfsdcb: Unknown which type %d", ic->ic_which);
656                 return;
657         }
658
659         bsiz = sizeof(buf) - bsiz;
660
661         if (atomicio(write, ic->ic_fd, buf, bsiz) != bsiz)
662                 idmapd_warnx("nfsdcb: write(%s) failed: errno %d (%s)",
663                              ic->ic_path, errno, strerror(errno));
664
665 out:
666         event_add(&ic->ic_event, NULL);
667 }
668
669 static void
670 imconv(struct idmap_client *ic, struct idmap_msg *im)
671 {
672         switch (im->im_conv) {
673         case IDMAP_CONV_IDTONAME:
674                 idtonameres(im);
675                 if (verbose > 1)
676                         idmapd_warnx("%s %s: (%s) id \"%d\" -> name \"%s\"",
677                             ic->ic_id, ic->ic_clid,
678                             im->im_type == IDMAP_TYPE_USER ? "user" : "group",
679                             im->im_id, im->im_name);
680                 break;
681         case IDMAP_CONV_NAMETOID:
682                 if (validateascii(im->im_name, sizeof(im->im_name)) == -1) {
683                         im->im_status |= IDMAP_STATUS_INVALIDMSG;
684                         return;
685                 }
686                 nametoidres(im);
687                 if (verbose > 1)
688                         idmapd_warnx("%s %s: (%s) name \"%s\" -> id \"%d\"",
689                             ic->ic_id, ic->ic_clid,
690                             im->im_type == IDMAP_TYPE_USER ? "user" : "group",
691                             im->im_name, im->im_id);
692                 break;
693         default:
694                 idmapd_warnx("imconv: Invalid conversion type (%d) in message",
695                              im->im_conv);
696                 im->im_status |= IDMAP_STATUS_INVALIDMSG;
697                 break;
698         }
699 }
700
701 static void
702 nfscb(int fd, short which, void *data)
703 {
704         struct idmap_client *ic = data;
705         struct idmap_msg im;
706
707         if (which != EV_READ)
708                 goto out;
709
710         if (atomicio(read, ic->ic_fd, &im, sizeof(im)) != sizeof(im)) {
711                 if (verbose > 0)
712                         idmapd_warn("nfscb: read(%s)", ic->ic_path);
713                 if (errno == EPIPE)
714                         return;
715                 goto out;
716         }
717
718         imconv(ic, &im);
719
720         if (atomicio(write, ic->ic_fd, &im, sizeof(im)) != sizeof(im))
721                 idmapd_warn("nfscb: write(%s)", ic->ic_path);
722 out:
723         event_add(&ic->ic_event, NULL);
724 }
725
726 static void
727 nfsdreopen_one(struct idmap_client *ic)
728 {
729         int fd;
730
731         if (verbose > 0)
732                 idmapd_warnx("ReOpening %s", ic->ic_path);
733
734         if ((fd = open(ic->ic_path, O_RDWR, 0)) != -1) {
735                 if ((ic->ic_event.ev_flags & EVLIST_INIT))
736                         event_del(&ic->ic_event);
737                 if (ic->ic_fd != -1)
738                         close(ic->ic_fd);
739
740                 ic->ic_event.ev_fd = ic->ic_fd = fd;
741                 event_set(&ic->ic_event, ic->ic_fd, EV_READ, nfsdcb, ic);
742                 event_add(&ic->ic_event, NULL);
743         } else {
744                 idmapd_warnx("nfsdreopen: Opening '%s' failed: errno %d (%s)",
745                         ic->ic_path, errno, strerror(errno));
746         }
747 }
748
749 static void
750 nfsdreopen()
751 {
752         nfsdreopen_one(&nfsd_ic[IC_NAMEID]);
753         nfsdreopen_one(&nfsd_ic[IC_IDNAME]);
754         return;
755 }
756
757 static int
758 nfsdopen()
759 {
760         return ((nfsdopenone(&nfsd_ic[IC_NAMEID]) == 0 &&
761                     nfsdopenone(&nfsd_ic[IC_IDNAME]) == 0) ? 0 : -1);
762 }
763
764 static int
765 nfsdopenone(struct idmap_client *ic)
766 {
767         if ((ic->ic_fd = open(ic->ic_path, O_RDWR, 0)) == -1) {
768                 if (verbose > 0)
769                         idmapd_warnx("nfsdopenone: Opening %s failed: "
770                                 "errno %d (%s)",
771                                 ic->ic_path, errno, strerror(errno));
772                 return (-1);
773         }
774
775         event_set(&ic->ic_event, ic->ic_fd, EV_READ, nfsdcb, ic);
776         event_add(&ic->ic_event, NULL);
777
778         if (verbose > 0)
779                 idmapd_warnx("Opened %s", ic->ic_path);
780
781         return (0);
782 }
783
784 static int
785 nfsopen(struct idmap_client *ic)
786 {
787         if ((ic->ic_fd = open(ic->ic_path, O_RDWR, 0)) == -1) {
788                 switch (errno) {
789                 case ENOENT:
790                         fcntl(ic->ic_dirfd, F_SETSIG, SIGUSR2);
791                         fcntl(ic->ic_dirfd, F_NOTIFY,
792                             DN_CREATE | DN_DELETE | DN_MULTISHOT);
793                         break;
794                 default:
795                         idmapd_warn("nfsopen: open(%s)", ic->ic_path);
796                         return (-1);
797                 }
798         } else {
799                 event_set(&ic->ic_event, ic->ic_fd, EV_READ, nfscb, ic);
800                 event_add(&ic->ic_event, NULL);
801                 fcntl(ic->ic_dirfd, F_SETSIG, 0);
802                 fcntl(ic->ic_dirfd, F_NOTIFY, 0);
803                 if (verbose > 0)
804                         idmapd_warnx("Opened %s", ic->ic_path);
805         }
806
807         return (0);
808 }
809
810 static void
811 idtonameres(struct idmap_msg *im)
812 {
813         char domain[NFS4_MAX_DOMAIN_LEN];
814         int ret = 0;
815
816         ret = nfs4_get_default_domain(NULL, domain, sizeof(domain));
817         switch (im->im_type) {
818         case IDMAP_TYPE_USER:
819                 ret = nfs4_uid_to_name(im->im_id, domain, im->im_name,
820                                 sizeof(im->im_name));
821                 if (ret) {
822                         if (strlen(nobodyuser) < sizeof(im->im_name))
823                                 strcpy(im->im_name, nobodyuser);
824                         else
825                                 strcpy(im->im_name, NFS4NOBODY_USER);
826                 }
827                 break;
828         case IDMAP_TYPE_GROUP:
829                 ret = nfs4_gid_to_name(im->im_id, domain, im->im_name,
830                                 sizeof(im->im_name));
831                 if (ret) {
832                         if (strlen(nobodygroup) < sizeof(im->im_name))
833                                 strcpy(im->im_name, nobodygroup);
834                         else
835                                 strcpy(im->im_name, NFS4NOBODY_GROUP);
836                 }
837                 break;
838         }
839         /* XXX Hack? */
840         im->im_status = IDMAP_STATUS_SUCCESS;
841 }
842
843 static void
844 nametoidres(struct idmap_msg *im)
845 {
846         uid_t uid;
847         gid_t gid;
848         int ret = 0;
849
850         /* XXX: nobody fallbacks shouldn't always happen:
851          *      server id -> name should be OK
852          *      client name -> id should be OK
853          * but not otherwise */
854         /* XXX: move nobody stuff to library calls
855          * (nfs4_get_nobody_user(domain), nfs4_get_nobody_group(domain)) */
856         /* XXX: should make this call higher up in the call chain (so we'd
857          * have a chance on looking up server/whatever. */
858         switch (im->im_type) {
859         case IDMAP_TYPE_USER:
860                 ret = nfs4_name_to_uid(im->im_name, &uid);
861                 im->im_id = (u_int32_t) uid;
862                 if (ret)
863                         im->im_id = nobodyuid;
864                 break;
865         case IDMAP_TYPE_GROUP:
866                 ret = nfs4_name_to_gid(im->im_name, &gid);
867                 im->im_id = (u_int32_t) gid;
868                 if (ret)
869                         im->im_id = nobodygid;
870                 break;
871         }
872         /* XXX? */
873         im->im_status = IDMAP_STATUS_SUCCESS;
874 }
875
876 static int
877 validateascii(char *string, u_int32_t len)
878 {
879         int i;
880
881         for (i = 0; i < len; i++) {
882                 if (string[i] == '\0')
883                         break;
884
885                 if (string[i] & 0x80)
886                         return (-1);
887         }
888
889         if (string[i] != '\0')
890                 return (-1);
891
892         return (i + 1);
893 }
894
895 static int
896 addfield(char **bpp, ssize_t *bsizp, char *fld)
897 {
898         char ch, *bp = *bpp;
899         ssize_t bsiz = *bsizp;
900
901         while ((ch = *fld++) != '\0' && bsiz > 0) {
902                 switch(ch) {
903                 case ' ':
904                 case '\t':
905                 case '\n':
906                 case '\\':
907                         if (bsiz >= 4) {
908                                 bp += snprintf(bp, bsiz, "\\%03o", ch);
909                                 bsiz -= 4;
910                         }
911                         break;
912                 default:
913                         *bp++ = ch;
914                         bsiz--;
915                         break;
916                 }
917         }
918
919         if (bsiz < 1 || ch != '\0')
920                 return (-1);
921
922         *bp++ = ' ';
923         bsiz--;
924
925         *bpp = bp;
926         *bsizp = bsiz;
927
928         return (0);
929 }
930
931 static int
932 getfield(char **bpp, char *fld, size_t fldsz)
933 {
934         char *bp;
935         u_int val, n;
936
937         while ((bp = strsep(bpp, " ")) != NULL && bp[0] == '\0')
938                 ;
939
940         if (bp == NULL || bp[0] == '\0' || bp[0] == '\n')
941                 return (-1);
942
943         while (*bp != '\0' && fldsz > 1) {
944                 if (*bp == '\\') {
945                         if ((n = sscanf(bp, "\\%03o", &val)) != 1)
946                                 return (-1);
947                         if (val > (char)-1)
948                                 return (-1);
949                         *fld++ = (char)val;
950                         bp += 4;
951                 } else {
952                         *fld++ = *bp;
953                         bp++;
954                 }
955                 fldsz--;
956         }
957
958         if (*bp != '\0')
959                 return (-1);
960         *fld = '\0';
961
962         return (0);
963 }
964 /*
965  * mydaemon creates a pipe between the partent and child
966  * process. The parent process will wait until the
967  * child dies or writes a '1' on the pipe signaling
968  * that it started successfully.
969  */
970 int pipefds[2] = { -1, -1};
971
972 void
973 mydaemon(int nochdir, int noclose)
974 {
975         int pid, status, tempfd;
976
977         if (pipe(pipefds) < 0)
978                 err(1, "mydaemon: pipe() failed: errno %d", errno);
979
980         if ((pid = fork ()) < 0)
981                 err(1, "mydaemon: fork() failed: errno %d", errno);
982
983         if (pid != 0) {
984                 /*
985                  * Parent. Wait for status from child.
986                  */
987                 close(pipefds[1]);
988                 if (read(pipefds[0], &status, 1) != 1)
989                         exit(1);
990                 exit (0);
991         }
992         /* Child.       */
993         close(pipefds[0]);
994         setsid ();
995         if (nochdir == 0) {
996                 if (chdir ("/") == -1)
997                         err(1, "mydaemon: chdir() failed: errno %d", errno);
998         }
999
1000         while (pipefds[1] <= 2) {
1001                 pipefds[1] = dup(pipefds[1]);
1002                 if (pipefds[1] < 0)
1003                         err(1, "mydaemon: dup() failed: errno %d", errno);
1004         }
1005
1006         if (noclose == 0) {
1007                 tempfd = open("/dev/null", O_RDWR);
1008                 dup2(tempfd, 0);
1009                 dup2(tempfd, 1);
1010                 dup2(tempfd, 2);
1011                 closeall(3);
1012         }
1013
1014         return;
1015 }
1016 void
1017 release_parent()
1018 {
1019         int status;
1020
1021         if (pipefds[1] > 0) {
1022                 write(pipefds[1], &status, 1);
1023                 close(pipefds[1]);
1024                 pipefds[1] = -1;
1025         }
1026 }