]> andersk Git - moira.git/blobdiff - incremental/afs.c
Suppress warning about ownership changes - administrators have been warned.
[moira.git] / incremental / afs.c
index 9ab13f57da91fd8ce828ef1a0c5d4b10985c3687..4171c2a76c5cac4c1a0e7c9004fecdb9b48857b6 100644 (file)
@@ -2,30 +2,68 @@
  *
  * Do AFS incremental updates
  *
- * Copyright (C) 1989 by the Massachusetts Institute of Technology
+ * Copyright (C) 1989,1992 by the Massachusetts Institute of Technology
  * for copying and distribution information, please see the file
  * <mit-copyright.h>.
  */
 
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/file.h>
+#include <strings.h>
+
+#include <krb.h>
 #include <moira.h>
 #include <moira_site.h>
-#include <sys/file.h>
 
+#include <afs/param.h>
+#include <afs/cellconfig.h>
+#include <afs/venus.h>
+#include <afs/ptclient.h>
+#include <afs/pterror.h>
+
+#define STOP_FILE "/moira/afs/noafs"
 #define file_exists(file) (access((file), F_OK) == 0)
 
-char pts[64], fs[64];
+#if defined(vax) && !defined(__STDC__)
+#define volatile
+#endif
 
 char *whoami;
 
+/* Main stub routines */
+int do_user();
+int do_list();
+int do_member();
+int do_filesys();
+int do_quota();
+
+/* Support stub routines */
+int run_cmd();
+int add_user_lists();
+int get_members();
+int edit_group();
+int pr_try();
+int check_afs();
+
+/* libprot.a routines */
+extern long pr_Initialize();
+extern long pr_CreateUser();
+extern long pr_CreateGroup();
+extern long pr_DeleteByID();
+extern long pr_ChangeEntry();
+extern long pr_SetFieldsEntry();
+extern long pr_AddToGroup();
+extern long pr_RemoveUserFromGroup();
+
+static char tbl_buf[1024];
+
 main(argc, argv)
 char **argv;
 int argc;
 {
     int beforec, afterc, i;
     char *table, **before, **after;
-#ifdef DEBUG
-    char buf[1024];
-#endif
 
     for (i = getdtablesize() - 1; i > 2; i--)
       close(i);
@@ -37,27 +75,28 @@ int argc;
     after = &argv[4 + beforec];
     whoami = argv[0];
 
-#ifdef DEBUG
-    sprintf(buf, "%s (", table);
+    setlinebuf(stdout);
+
+    strcpy(tbl_buf, table);
+    strcat(tbl_buf, " (");
     for (i = 0; i < beforec; i++) {
        if (i > 0)
-         strcat(buf, ",");
-       strcat(buf, before[i]);
+         strcat(tbl_buf, ",");
+       strcat(tbl_buf, before[i]);
     }
-    strcat(buf, ")->(");
+    strcat(tbl_buf, ")->(");
     for (i = 0; i < afterc; i++) {
        if (i > 0)
-         strcat(buf, ",");
-       strcat(buf, after[i]);
+         strcat(tbl_buf, ",");
+       strcat(tbl_buf, after[i]);
     }
-    strcat(buf, ")\n");
-    write(1,buf,strlen(buf));
+    strcat(tbl_buf, ")");
+#ifdef DEBUG
+    printf("%s\n", tbl_buf);
 #endif
 
     initialize_sms_error_table();
     initialize_krb_error_table();
-    sprintf(pts, "%s/pts", BIN_DIR);
-    sprintf(fs, "%s/fs", BIN_DIR);
 
     if (!strcmp(table, "users")) {
        do_user(before, beforec, after, afterc);
@@ -74,105 +113,191 @@ int argc;
 }
 
 
-do_cmd(cmd)
-char *cmd;
-{
-    int success = 0, tries = 0;
-
-    while (success == 0 && tries < 3) {
-       com_err(whoami, 0, "Executing command: %s", cmd);
-       if (system(cmd) == 0)
-         success++;
-       if (!success) {
-           tries++;
-           sleep(5 * 60);
-       }
-    }
-    if (!success)
-      critical_alert("incremental", "failed command: %s", cmd);
-}
-
-
 do_user(before, beforec, after, afterc)
 char **before;
 int beforec;
 char **after;
 int afterc;
 {
-    int bstate, astate;
-    char cmd[512];
-
-    cmd[0] = bstate = astate = 0;
-    if (afterc > U_STATE)
-      astate = atoi(after[U_STATE]);
-    if (beforec > U_STATE)
-      bstate = atoi(before[U_STATE]);
+    int astate, bstate, auid, buid, code;
+    char hostname[64];
+    char *av[2];
+
+    auid = buid = astate = bstate = 0;
+    if (afterc > U_STATE) astate = atoi(after[U_STATE]);
+    if (beforec > U_STATE) bstate = atoi(before[U_STATE]);
+    if (afterc > U_UID) auid = atoi(after[U_UID]);
+    if (beforec > U_UID) buid = atoi(before[U_UID]);
+
+    /* We consider "half-registered" users to be active */
     if (astate == 2) astate = 1;
     if (bstate == 2) bstate = 1;
 
-    if (astate != 1 && bstate != 1)
-      return;
-    if (astate == 1 && bstate != 1) {
-       sprintf(cmd, "%s createuser -name %s -id %s",
-               pts, after[U_NAME], after[U_UID]);
-       do_cmd(cmd);
+    if (astate != 1 && bstate != 1)            /* inactive user */
        return;
-    } else if (astate != 1 && bstate == 1) {
-       sprintf(cmd, "%s delete -name %s", pts, before[U_NAME]);
-       do_cmd(cmd);
+
+    if (astate == bstate && auid == buid && 
+       !strcmp(before[U_NAME], after[U_NAME]))
+       /* No AFS related attributes have changed */
        return;
-    }
 
-    if (beforec > U_UID && afterc > U_UID &&
-       strcmp(before[U_UID], after[U_UID])) {
-       /* change UID, & possibly user name here */
-       exit(1);
+    if (astate == bstate) {
+       /* Only a modify has to be done */
+       code = pr_try(pr_ChangeEntry, before[U_NAME], after[U_NAME], auid, "");
+       if (code) {
+           critical_alert("incremental",
+                          "Couldn't change user %s (id %d) to %s (id %d): %s",
+                          before[U_NAME], buid, after[U_NAME], auid,
+                          error_message(code));
+       }
+       return;
     }
+    if (bstate == 1) {
+       code = pr_try(pr_DeleteByID, buid);
+       if (code && code != PRNOENT) {
+           critical_alert("incremental",
+                          "Couldn't delete user %s (id %d): %s",
+                          before[U_NAME], buid, error_message(code));
+       }
+       return;
+    }
+    if (astate == 1) {
+       code = pr_try(pr_CreateUser, after[U_NAME], &auid);
+       if (code) {
+           critical_alert("incremental",
+                          "Couldn't create user %s (id %d): %s",
+                          after[U_NAME], auid, error_message(code));
+           return;
+       }
 
-    if (beforec > U_NAME && afterc > U_NAME &&
-       strcmp(before[U_NAME], after[U_NAME])) {
-       sprintf(cmd, "%s rename -oldname %s -newname %s",
-               pts, before[U_NAME], after[U_NAME]);
-       do_cmd(cmd);
+       if (bstate != 0) {
+           /* Reactivating a user; get his group list */
+           gethostname(hostname, sizeof(hostname));
+           code = mr_connect(hostname);
+           if (!code) code = mr_auth("afs.incr");
+           if (code) {
+               critical_alert("incremental",
+                              "Error contacting Moira server to retrieve grouplist of user %s: %s",
+                              after[U_NAME], error_message(code));
+               return;
+           }
+           av[0] = "ruser";
+           av[1] = after[U_NAME];
+           code = mr_query("get_lists_of_member", 2, av,
+                           add_user_lists, after[U_NAME]);
+           if (code && code != MR_NO_MATCH)
+               critical_alert("incremental",
+                              "Couldn't retrieve membership of user %s: %s",
+                              after[U_NAME], error_message(code));
+           mr_disconnect();
+       }
+       return;
     }
 }
 
 
-
 do_list(before, beforec, after, afterc)
 char **before;
 int beforec;
 char **after;
 int afterc;
 {
-    char cmd[512];
-    int agid, bgid;
-
-    cmd[0] = agid = bgid = 0;
-    if (beforec > L_GID && atoi(before[L_ACTIVE]) && atoi(before[L_GROUP]))
-      bgid = atoi(before[L_GID]);
-    if (afterc > L_GID && atoi(after[L_ACTIVE]) && atoi(after[L_GROUP]))
-      agid = atoi(after[L_GID]);
-
-    if (bgid == 0 && agid != 0) {
-       sprintf(cmd,
-               "%s creategroup -name system:%s -owner system:administrators -id -%s",
-               pts, after[L_NAME], after[L_GID]);
-       do_cmd(cmd);
+    register int agid, bgid;
+    int ahide, bhide;
+    long code, id;
+    char hostname[64];
+    char g1[PR_MAXNAMELEN], g2[PR_MAXNAMELEN];
+    char *av[2];
+
+    agid = bgid = 0;
+    if (beforec > L_GID && atoi(before[L_ACTIVE]) && atoi(before[L_GROUP])) {
+       bgid = atoi(before[L_GID]);
+       bhide = atoi(before[L_HIDDEN]);
+    }
+    if (afterc > L_GID && atoi(after[L_ACTIVE]) && atoi(after[L_GROUP])) {
+       agid = atoi(after[L_GID]);
+       ahide = atoi(after[L_HIDDEN]);
+    }
+
+    if (agid == 0 && bgid == 0)                        /* Not active groups */
+       return;
+
+    if (agid && bgid) {
+       if (strcmp(after[L_NAME], before[L_NAME])) {
+           /* Only a modify is required */
+           strcpy(g1, "system:");
+           strcpy(g2, "system:");
+           strcat(g1, before[L_NAME]);
+           strcat(g2, after[L_NAME]);
+           code = pr_try(pr_ChangeEntry, g1, g2, -agid, "");
+           if (code) {
+               critical_alert("incremental",
+                              "Couldn't change group %s (id %d) to %s (id %d): %s",
+                              before[L_NAME], -bgid, after[L_NAME], -agid,
+                              error_message(code));
+           }
+       }
+       if (ahide != bhide) {
+           code = pr_try(pr_SetFieldsEntry, -agid, PR_SF_ALLBITS,
+                         (ahide ? PRP_STATUS_ANY : PRP_GROUP_DEFAULT) >>PRIVATE_SHIFT,
+                         0 /*ngroups*/, 0 /*nusers*/);
+           if (code) {
+               critical_alert("incremental",
+                              "Couldn't set flags of group %s: %s",
+                              after[L_NAME], error_message(code));
+           }
+       }
        return;
     }
-    if (agid == 0 && bgid != 0) {
-       sprintf(cmd, "%s delete -name system:%s", pts, before[L_NAME]);
-       do_cmd(cmd);
+    if (bgid) {
+       code = pr_try(pr_DeleteByID, -bgid);
+       if (code && code != PRNOENT) {
+           critical_alert("incremental",
+                          "Couldn't delete group %s (id %d): %s",
+                          before[L_NAME], -bgid, error_message(code));
+       }
        return;
     }
-    if (agid == 0 && bgid == 0)
-      return;
-    if (strcmp(before[L_NAME], after[L_NAME])) {
-       sprintf(cmd,
-               "%s rename -oldname system:%s -newname system:%s",
-               pts, before[L_NAME], after[L_NAME]);
-       do_cmd(cmd);
+    if (agid) {
+       strcpy(g1, "system:");
+       strcat(g1, after[L_NAME]);
+       strcpy(g2, "system:administrators");
+       id = -agid;
+       code = pr_try(pr_CreateGroup, g1, g2, &id);
+       if (code) {
+           critical_alert("incremental",
+                          "Couldn't create group %s (id %d): %s",
+                          after[L_NAME], id, error_message(code));
+           return;
+       }
+       if (ahide) {
+           code = pr_try(pr_SetFieldsEntry, -agid, PR_SF_ALLBITS,
+                         (ahide ? PRP_STATUS_ANY : PRP_GROUP_DEFAULT) >>PRIVATE_SHIFT,
+                         0 /*ngroups*/, 0 /*nusers*/);
+           if (code) {
+               critical_alert("incremental",
+                              "Couldn't set flags of group %s: %s",
+                              after[L_NAME], error_message(code));
+           }
+       }
+
+       /* We need to make sure the group is properly populated */
+       if (beforec < L_ACTIVE || atoi(before[L_ACTIVE]) == 0) return;
+
+       gethostname(hostname, sizeof(hostname));
+       code = mr_connect(hostname);
+       if (!code) code = mr_auth("afs.incr");
+       if (code) {
+           critical_alert("incremental",
+                          "Error contacting Moira server to resolve %s: %s",
+                          after[L_NAME], error_message(code));
+           return;
+       }
+       av[0] = "LIST";
+       av[1] = after[L_NAME];
+       get_members(2, av, after[L_NAME]);
+
+       mr_disconnect();
        return;
     }
 }
@@ -184,39 +309,17 @@ int beforec;
 char **after;
 int afterc;
 {
-    char cmd[512], *p;
-
+    int code;
+    char *p;
+    
     if ((beforec < 4 || !atoi(before[LM_END])) &&
        (afterc < 4 || !atoi(after[LM_END])))
-      return;
-    if (beforec == 0 && !strcmp(after[LM_TYPE], "USER")) {
-       sprintf(cmd, "%s adduser -user %s -group system:%s",
-               pts, after[LM_MEMBER], after[LM_LIST]);
-       do_cmd(cmd);
-       return;
-    }
-    if (afterc == 0 && !strcmp(before[LM_TYPE], "USER")) {
-       sprintf(cmd, "%s removeuser -user %s -group system:%s",
-               pts, before[LM_MEMBER], before[LM_LIST]);
-       do_cmd(cmd);
-       return;
-    }
-    if (beforec == 0 && !strcmp(after[LM_TYPE], "KERBEROS")) {
-       p = index(after[LM_MEMBER], '@');
-       if (p) *p = 0;
-       sprintf(cmd, "%s adduser -user %s -group system:%s",
-               pts, after[LM_MEMBER], after[LM_LIST]);
-       do_cmd(cmd);
-       return;
-    }
-    if (afterc == 0 && !strcmp(before[LM_TYPE], "KERBEROS")) {
-       p = index(before[LM_MEMBER], '@');
-       if (p) *p = 0;
-       sprintf(cmd, "%s removeuser -user %s -group system:%s",
-               pts, before[LM_MEMBER], before[LM_LIST]);
-       do_cmd(cmd);
        return;
-    }
+
+    if (afterc) 
+       edit_group(1, after[LM_LIST], after[LM_TYPE], after[LM_MEMBER]);
+    if (beforec)
+       edit_group(0, before[LM_LIST], before[LM_TYPE], before[LM_MEMBER]);
 }
 
 
@@ -226,29 +329,69 @@ int beforec;
 char **after;
 int afterc;
 {
-    char cmd[512];
+    char cmd[1024];
+    int acreate, atype, btype;
+
+    if (afterc < FS_CREATE) {
+       atype = acreate = 0;
+    } else {
+       atype = !strcmp(after[FS_TYPE], "AFS");
+       acreate = atoi(after[FS_CREATE]);
+    }
 
-    if (afterc < FS_CREATE ||
-       strcmp("AFS", after[FS_TYPE]))
-      return;
     if (beforec < FS_CREATE) {
+       if (acreate == 0 || atype == 0) return;
+
        /* new locker creation */
-       sprintf(cmd, "%s/afs_create.pl %s %s %s %s %s %s %s %s %s",
-               BIN_DIR, after[FS_LABEL], after[FS_LOCKERTYPE],
-               after[FS_MACHINE], after[FS_PACK], 2200, after[FS_OWNER],
-               0, after[FS_OWNERS], 0);
-       do_cmd(cmd);
+       sprintf(cmd, "%s/perl -I%s %s/afs_create.pl %s %s %s %s %s %s",
+               BIN_DIR, BIN_DIR, BIN_DIR,
+               after[FS_NAME], after[FS_L_TYPE], after[FS_MACHINE],
+               after[FS_PACK], after[FS_OWNER], after[FS_OWNERS]);
+       run_cmd(cmd);
+       return;
+    }
+    
+    btype = !strcmp(before[FS_TYPE], "AFS");
+    if (afterc < FS_CREATE) {
+       if (btype)
+           critical_alert("incremental",
+                          "Cannot delete AFS filesystem %s: Operation not supported",
+                          before[FS_NAME]);
        return;
-    } else if (strcmp(before[FS_TYPE], "AFS")) {
-       /* moved from non-AFS */
-    } else {
-       /* other modification */
-       !strncmp("/afs/", after[FS_PACK], 5) &&
-       atoi(after[FS_CREATE]) &&
-       !file_exists(after[FS_PACK])) {
-       critical_alert("incremental", "unable to create locker %s",
-                      after[FS_PACK]);
     }
+
+    if (!acreate)
+       return;
+
+    /* Are we dealing with AFS lockers (could be type ERR lockers) */
+    if (!atype && !btype)
+       if (strcmp(before[FS_TYPE], "ERR") || strcmp(after[FS_TYPE], "ERR"))
+           return;
+                                                   
+    /* By now, we know we are simply changing AFS filesystem attributes.
+     * Operations supported:
+     *    Name change:  rename/remount
+     *    Path change:  remount
+     *    Type change:  ERR<-->AFS
+     */
+
+#if 0
+    if (strcmp(before[FS_OWNER], after[FS_OWNER]) ||
+       strcmp(before[FS_OWNERS], after[FS_OWNERS]))
+    {
+       critical_alert("incremental",
+                      "Cannot change ownership of filesystem %s: Operation not yet supported",
+                      after[FS_NAME]);
+    }
+#endif
+
+    sprintf(cmd, "%s/perl -I%s %s/afs_rename.pl %s %s %s %s %s %s %s %s %s %s",
+           BIN_DIR, BIN_DIR, BIN_DIR,
+           before[FS_NAME], before[FS_MACHINE], before[FS_TYPE],
+           before[FS_L_TYPE], before[FS_PACK],
+           after[FS_NAME], after[FS_MACHINE], after[FS_TYPE],
+           after[FS_L_TYPE], after[FS_PACK]);
+    run_cmd(cmd);
 }
 
 
@@ -258,14 +401,182 @@ int beforec;
 char **after;
 int afterc;
 {
-    char cmd[512];
+    char cmd[1024];
 
     if (afterc < Q_DIRECTORY || strcmp("ANY", after[Q_TYPE]) ||
-       strncmp("/afs", after[Q_DIRECTORY], 4))
-      return;
+       strncmp("/afs/", after[Q_DIRECTORY], 5))
+       return;
 
-    sprintf(cmd, "%s setquota -dir %s -max %s", fs,
+    sprintf(cmd, "%s/perl -I%s %s/afs_quota.pl %s %s",
+           BIN_DIR, BIN_DIR, BIN_DIR,
            after[Q_DIRECTORY], after[Q_QUOTA]);
-    do_cmd(cmd);
+    run_cmd(cmd);
     return;
 }
+
+
+run_cmd(cmd)
+char *cmd;
+{
+    int success=0, tries=0;
+
+    check_afs();
+    
+    while (success == 0 && tries < 2) {
+       if (tries++)
+           sleep(90);
+       com_err(whoami, 0, "Executing command: %s", cmd);
+       if (system(cmd) == 0)
+           success++;
+    }
+    if (!success)
+       critical_alert("incremental", "failed command: %s", cmd);
+}
+
+
+add_user_lists(ac, av, user)
+    int ac;
+    char *av[];
+    char *user;
+{
+    if (atoi(av[5]))
+       edit_group(1, av[0], "USER", user);
+}
+
+
+get_members(ac, av, group)
+    int ac;
+    char *av[];
+    char *group;
+{
+    int code=0;
+
+    if (strcmp(av[0], "LIST")) {
+       edit_group(1, group, av[0], av[1]);
+    } else {
+       code = mr_query("get_end_members_of_list", 1, &av[1],
+                       get_members, group);
+       if (code)
+           critical_alert("incremental",
+                          "Couldn't retrieve full membership of %s: %s",
+                          group, error_message(code));
+    }
+    return code;
+}
+
+
+edit_group(op, group, type, member)
+    int op;
+    char *group;
+    char *type;
+    char *member;
+{
+    char *p = 0;
+    char buf[PR_MAXNAMELEN];
+    int code;
+    static char local_realm[REALM_SZ+1] = "";
+
+    /* The following KERBEROS code allows for the use of entities
+     * user@foreign_cell.
+     */
+    if (!local_realm[0])
+       krb_get_lrealm(local_realm, 1);
+    if (!strcmp(type, "KERBEROS")) {
+       p = index(member, '@');
+       if (p && !strcasecmp(p+1, local_realm))
+           *p = 0;
+    } else if (strcmp(type, "USER"))
+       return;                                 /* invalid type */
+
+    strcpy(buf, "system:");
+    strcat(buf, group);
+    code=pr_try(op ? pr_AddToGroup : pr_RemoveUserFromGroup, member, buf);
+    if (code) {
+       if (op==0 && code == PRNOENT) return;
+       if (op==1 && code == PRIDEXIST) return;
+       if (strcmp(type, "KERBEROS") || code != PRNOENT) {
+           critical_alert("incremental",
+                          "Couldn't %s %s %s %s: %s",
+                          op ? "add" : "remove", member,
+                          op ? "to" : "from", buf,
+                          error_message(code));
+       }
+    }
+}
+
+
+long pr_try(fn, a1, a2, a3, a4, a5, a6, a7, a8)
+    long (*fn)();
+    char *a1, *a2, *a3, *a4, *a5, *a6, *a7, *a8;
+{
+    static int initd=0;
+    volatile register long code;
+    register int tries = 0;
+#ifdef DEBUG
+    char fname[64];
+#endif
+
+    check_afs();
+    
+    if (!initd) {
+       code=pr_Initialize(1, AFSCONF_CLIENTNAME, 0);
+       if (code) {
+           critical_alert("incremental", "Couldn't initialize libprot: %s",
+                          error_message(code));
+           return;
+       }
+       initd = 1;
+    }
+    sleep(1);                                  /* give ptserver room */
+
+    while (code = (*fn)(a1, a2, a3, a4, a5, a6, a7, a8)) {
+#ifdef DEBUG
+       long t;
+       t = time(0);
+       if (fn == pr_AddToGroup) strcpy(fname, "pr_AddToGroup");
+       else if (fn == pr_RemoveUserFromGroup)
+           strcpy(fname, "pr_RemoveUserFromGroup");
+       else if (fn == pr_CreateUser) strcpy(fname, "pr_CreateUser");
+       else if (fn == pr_CreateGroup) strcpy(fname, "pr_CreateGroup");
+       else if (fn == pr_DeleteByID) strcpy(fname, "pr_DeleteByID");
+       else if (fn == pr_ChangeEntry) strcpy(fname, "pr_ChangeEntry");
+       else if (fn == pr_SetFieldsEntry) strcpy(fname, "pr_SetFieldsEntry");
+       else if (fn == pr_AddToGroup) strcpy(fname, "pr_AddToGroup");
+       else
+           sprintf(fname, "pr_??? (0x%08x)", (long)fn);
+
+       com_err(whoami, code, "- %s failed (try %d @%u)", fname, tries+1, t);
+#endif
+       if (++tries > 2) break;         /* 3 tries */
+       
+       if (code == UNOQUORUM) sleep(90);
+       else sleep(15);
+
+       /* Re-initialize the prdb connection */
+       code=pr_Initialize(0, AFSCONF_CLIENTNAME, 0);
+       if (!code) code=pr_Initialize(1, AFSCONF_CLIENTNAME, 0);
+       if (code) {
+           critical_alert("incremental", "Couldn't re-initialize libprot: %s",
+                          error_message(code));
+           initd = 0;                          /* we lost */
+           break;
+       }
+    }
+    return code;
+}
+
+
+check_afs()
+{
+    int i;
+    
+    for (i=0; file_exists(STOP_FILE); i++) {
+       if (i > 30) {
+           critical_alert("incremental",
+                          "AFS incremental failed (%s exists): %s",
+                          STOP_FILE, tbl_buf);
+           exit(1);
+       }
+       sleep(60);
+    }
+}
This page took 0.266955 seconds and 4 git commands to generate.