lustre/utils/l_getidentity.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   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 version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #include <stdlib.h>
  38 #include <stdint.h>
  39 #include <stdio.h>
  40 #include <unistd.h>
  41 #include <errno.h>
  42 #include <string.h>
  43 #include <fcntl.h>
  44 #include <pwd.h>
  45 #include <grp.h>
  46 #include <stdarg.h>
  47 #include <stddef.h>
  48 #include <libgen.h>
  49 #include <syslog.h>
  50
  51 #include <liblustre.h>
  52 #include <lustre/lustre_user.h>
  53 #include <lustre/lustre_idl.h>
  54
  55 #define PERM_PATHNAME "/etc/lustre/perm.conf"
  56
  57 /*
  58  * permission file format is like this:
  59  * {nid} {uid} {perms}
  60  *
  61  * '*' nid means any nid
  62  * '*' uid means any uid
  63  * the valid values for perms are:
  64  * setuid/setgid/setgrp/rmtacl           -- enable corresponding perm
  65  * nosetuid/nosetgid/nosetgrp/normtacl   -- disable corresponding perm
  66  * they can be listed together, seperated by ',',
  67  * when perm and noperm are in the same line (item), noperm is preferential,
  68  * when they are in different lines (items), the latter is preferential,
  69  * '*' nid is as default perm, and is not preferential.
  70  */
  71
  72 static char *progname;
  73
  74 static void usage(void)
  75 {
  76         fprintf(stderr,
  77                 "\nusage: %s {mdtname} {uid}\n"
  78                 "Normally invoked as an upcall from Lustre, set via:\n"
  79                 "  /proc/fs/lustre/mdt/{mdtname}/identity_upcall\n",
  80                 progname);
  81 }
  82
  83 static int compare_u32(const void *v1, const void *v2)
  84 {
  85         return (*(__u32 *)v1 - *(__u32 *)v2);
  86 }
  87
  88 static void errlog(const char *fmt, ...)
  89 {
  90         va_list args;
  91
  92         openlog(progname, LOG_PERROR, LOG_AUTHPRIV);
  93
  94         va_start(args, fmt);
  95         vsyslog(LOG_NOTICE, fmt, args);
  96         fprintf(stderr, fmt, args);
  97         va_end(args);
  98
  99         closelog();
 100 }
 101
 102 int get_groups_local(struct identity_downcall_data *data)
 103 {
 104         int maxgroups;
 105         gid_t *groups;
 106         unsigned int ngroups = 0;
 107         struct passwd *pw;
 108         struct group *gr;
 109         char *pw_name;
 110         int namelen;
 111         int i;
 112
 113         pw = getpwuid(data->idd_uid);
 114         if (!pw) {
 115                 errlog("no such user %u\n", data->idd_uid);
 116                 data->idd_err = errno ? errno : EIDRM;
 117                 return -1;
 118         }
 119         data->idd_gid = pw->pw_gid;
 120
 121         namelen = sysconf(_SC_LOGIN_NAME_MAX);
 122         if (namelen < _POSIX_LOGIN_NAME_MAX)
 123                 namelen = _POSIX_LOGIN_NAME_MAX;
 124         pw_name = (char *)malloc(namelen);
 125         if (!pw_name) {
 126                 errlog("malloc error\n");
 127                 data->idd_err = errno;
 128                 return -1;
 129         }
 130         memset(pw_name, 0, namelen);
 131         strncpy(pw_name, pw->pw_name, namelen - 1);
 132
 133         maxgroups = sysconf(_SC_NGROUPS_MAX);
 134         if (maxgroups > NGROUPS_MAX)
 135                 maxgroups = NGROUPS_MAX;
 136         groups = data->idd_groups;
 137
 138         groups[ngroups++] = pw->pw_gid;
 139         while ((gr = getgrent())) {
 140                 if (gr->gr_gid == groups[0])
 141                         continue;
 142                 if (!gr->gr_mem)
 143                         continue;
 144                 for (i = 0; gr->gr_mem[i]; i++) {
 145                         if (!strcmp(gr->gr_mem[i], pw_name)) {
 146                                 groups[ngroups++] = gr->gr_gid;
 147                                 break;
 148                         }
 149                 }
 150                 if (ngroups == maxgroups)
 151                         break;
 152         }
 153         endgrent();
 154         qsort(groups, ngroups, sizeof(*groups), compare_u32);
 155         data->idd_ngroups = ngroups;
 156
 157         free(pw_name);
 158         return 0;
 159 }
 160
 161 static inline int comment_line(char *line)
 162 {
 163         char *p = line;
 164
 165         while (*p && (*p == ' ' || *p == '\t')) p++;
 166
 167         if (!*p || *p == '\n' || *p == '#')
 168                 return 1;
 169         return 0;
 170 }
 171
 172 static inline int match_uid(uid_t uid, const char *str)
 173 {
 174         char *end;
 175         uid_t uid2;
 176
 177         if(!strcmp(str, "*"))
 178                 return -1;
 179
 180         uid2 = strtoul(str, &end, 0);
 181         if (*end)
 182                 return 0;
 183
 184         return (uid == uid2);
 185 }
 186
 187 typedef struct {
 188         char   *name;
 189         __u32   bit;
 190 } perm_type_t;
 191
 192 static perm_type_t perm_types[] = {
 193         { "setuid", CFS_SETUID_PERM },
 194         { "setgid", CFS_SETGID_PERM },
 195         { "setgrp", CFS_SETGRP_PERM },
 196         { "rmtacl", CFS_RMTACL_PERM },
 197         { 0 }
 198 };
 199
 200 static perm_type_t noperm_types[] = {
 201         { "nosetuid", CFS_SETUID_PERM },
 202         { "nosetgid", CFS_SETGID_PERM },
 203         { "nosetgrp", CFS_SETGRP_PERM },
 204         { "normtacl", CFS_RMTACL_PERM },
 205         { 0 }
 206 };
 207
 208 int parse_perm(__u32 *perm, __u32 *noperm, char *str)
 209 {
 210         char *start, *end;
 211         char name[64];
 212         perm_type_t *pt;
 213
 214         *perm = 0;
 215         *noperm = 0;
 216         start = str;
 217         while (1) {
 218                 memset(name, 0, sizeof(name));
 219                 end = strchr(start, ',');
 220                 if (!end)
 221                         end = str + strlen(str);
 222                 if (start >= end)
 223                         break;
 224                 strncpy(name, start, end - start);
 225                 for (pt = perm_types; pt->name; pt++) {
 226                         if (!strcasecmp(name, pt->name)) {
 227                                 *perm |= pt->bit;
 228                                 break;
 229                         }
 230                 }
 231
 232                 if (!pt->name) {
 233                         for (pt = noperm_types; pt->name; pt++) {
 234                                 if (!strcasecmp(name, pt->name)) {
 235                                         *noperm |= pt->bit;
 236                                         break;
 237                                 }
 238                         }
 239
 240                         if (!pt->name) {
 241                                 printf("unkown type: %s\n", name);
 242                                 return -1;
 243                         }
 244                 }
 245
 246                 start = end + 1;
 247         }
 248         return 0;
 249 }
 250
 251 int parse_perm_line(struct identity_downcall_data *data, char *line)
 252 {
 253         char uid_str[256], nid_str[256], perm_str[256];
 254         lnet_nid_t nid;
 255         __u32 perm, noperm;
 256         int rc, i;
 257
 258         if (data->idd_nperms >= N_PERMS_MAX) {
 259                 errlog("permission count %d > max %d\n",
 260                         data->idd_nperms, N_PERMS_MAX);
 261                 return -1;
 262         }
 263
 264         rc = sscanf(line, "%s %s %s", nid_str, uid_str, perm_str);
 265         if (rc != 3) {
 266                 errlog("can't parse line %s\n", line);
 267                 return -1;
 268         }
 269
 270         if (!match_uid(data->idd_uid, uid_str))
 271                 return 0;
 272
 273         if (!strcmp(nid_str, "*")) {
 274                 nid = LNET_NID_ANY;
 275         } else {
 276                 nid = libcfs_str2nid(nid_str);
 277                 if (nid == LNET_NID_ANY) {
 278                         errlog("can't parse nid %s\n", nid_str);
 279                         return -1;
 280                 }
 281         }
 282
 283         if (parse_perm(&perm, &noperm, perm_str)) {
 284                 errlog("invalid perm %s\n", perm_str);
 285                 return -1;
 286         }
 287
 288         /* merge the perms with the same nid.
 289          *
 290          * If there is LNET_NID_ANY in data->idd_perms[i].pdd_nid,
 291          * it must be data->idd_perms[0].pdd_nid, and act as default perm.
 292          */
 293         if (nid != LNET_NID_ANY) {
 294                 int found = 0;
 295
 296                 /* search for the same nid */
 297                 for (i = data->idd_nperms - 1; i >= 0; i--) {
 298                         if (data->idd_perms[i].pdd_nid == nid) {
 299                                 data->idd_perms[i].pdd_perm =
 300                                         (data->idd_perms[i].pdd_perm | perm) &
 301                                         ~noperm;
 302                                 found = 1;
 303                                 break;
 304                         }
 305                 }
 306
 307                 /* NOT found, add to tail */
 308                 if (!found) {
 309                         data->idd_perms[data->idd_nperms].pdd_nid = nid;
 310                         data->idd_perms[data->idd_nperms].pdd_perm =
 311                                 perm & ~noperm;
 312                         data->idd_nperms++;
 313                 }
 314         } else {
 315                 if (data->idd_nperms > 0) {
 316                         /* the first one isn't LNET_NID_ANY, need exchange */
 317                         if (data->idd_perms[0].pdd_nid != LNET_NID_ANY) {
 318                                 data->idd_perms[data->idd_nperms].pdd_nid =
 319                                         data->idd_perms[0].pdd_nid;
 320                                 data->idd_perms[data->idd_nperms].pdd_perm =
 321                                         data->idd_perms[0].pdd_perm;
 322                                 data->idd_perms[0].pdd_nid = LNET_NID_ANY;
 323                                 data->idd_perms[0].pdd_perm = perm & ~noperm;
 324                                 data->idd_nperms++;
 325                         } else {
 326                                 /* only fix LNET_NID_ANY item */
 327                                 data->idd_perms[0].pdd_perm =
 328                                         (data->idd_perms[0].pdd_perm | perm) &
 329                                         ~noperm;
 330                         }
 331                 } else {
 332                         /* it is the first one, only add to head */
 333                         data->idd_perms[0].pdd_nid = LNET_NID_ANY;
 334                         data->idd_perms[0].pdd_perm = perm & ~noperm;
 335                         data->idd_nperms = 1;
 336                 }
 337         }
 338
 339         return 0;
 340 }
 341
 342 int get_perms(FILE *fp, struct identity_downcall_data *data)
 343 {
 344         char line[1024];
 345
 346         while (fgets(line, 1024, fp)) {
 347                 if (comment_line(line))
 348                         continue;
 349
 350                 if (parse_perm_line(data, line)) {
 351                         errlog("parse line %s failed!\n", line);
 352                         return -1;
 353                 }
 354         }
 355
 356         return 0;
 357 }
 358
 359 static void show_result(struct identity_downcall_data *data)
 360 {
 361         int i;
 362
 363         if (data->idd_err) {
 364                 errlog("failed to get identity for uid %d: %s\n",
 365                        data->idd_uid, strerror(data->idd_err));
 366                 return;
 367         }
 368
 369         printf("uid=%d gid=", data->idd_uid);
 370         for (i = 0; i < data->idd_ngroups; i++)
 371                 printf("%s%u", i > 0 ? "," : "", data->idd_groups[i]);
 372         printf("\n");
 373         printf("permissions:\n"
 374                "  nid\t\t\tperm\n");
 375         for (i = 0; i < data->idd_nperms; i++) {
 376                 struct perm_downcall_data *pdd;
 377
 378                 pdd = &data->idd_perms[i];
 379
 380                 printf("  "LPX64"\t0x%x\n", pdd->pdd_nid, pdd->pdd_perm);
 381         }
 382         printf("\n");
 383 }
 384
 385 int main(int argc, char **argv)
 386 {
 387         FILE *perms_fp;
 388         char *end;
 389         struct identity_downcall_data *data;
 390         char procname[1024];
 391         unsigned long uid;
 392         int fd, rc;
 393
 394         progname = basename(argv[0]);
 395
 396         if (argc != 3) {
 397                 usage();
 398                 return 1;
 399         }
 400
 401         uid = strtoul(argv[2], &end, 0);
 402         if (*end) {
 403                 errlog("%s: invalid uid '%s'\n", progname, argv[2]);
 404                 usage();
 405                 return 1;
 406         }
 407
 408         data = malloc(sizeof(*data));
 409         if (!data) {
 410                 errlog("malloc identity downcall data(%d) failed!\n",
 411                        sizeof(*data));
 412                 return 1;
 413         }
 414         memset(data, 0, sizeof(*data));
 415         data->idd_magic = IDENTITY_DOWNCALL_MAGIC;
 416         data->idd_uid = uid;
 417
 418         /* get groups for uid */
 419         rc = get_groups_local(data);
 420         if (rc)
 421                 goto downcall;
 422
 423         /* read permission database */
 424         perms_fp = fopen(PERM_PATHNAME, "r");
 425         if (perms_fp) {
 426                 get_perms(perms_fp, data);
 427                 fclose(perms_fp);
 428         } else if (errno != ENOENT) {
 429                 errlog("open %s failed: %s\n",
 430                        PERM_PATHNAME, strerror(errno));
 431         }
 432
 433 downcall:
 434         if (getenv("L_GETIDENTITY_TEST")) {
 435                 show_result(data);
 436                 return 0;
 437         }
 438
 439         snprintf(procname, sizeof(procname),
 440                  "/proc/fs/lustre/mdt/%s/identity_info", argv[1]);
 441         fd = open(procname, O_WRONLY);
 442         if (fd < 0) {
 443                 errlog("can't open file %s: %s\n", procname, strerror(errno));
 444                 return 1;
 445         }
 446
 447         rc = write(fd, data, sizeof(*data));
 448         close(fd);
 449         if (rc != sizeof(*data)) {
 450                 errlog("partial write ret %d: %s\n", rc, strerror(errno));
 451                 return 1;
 452         }
 453
 454         return 0;
 455 }