server/increment.qc

   1 /*
   2  *      $Source$
   3  *      $Author$
   4  *      $Header$
   5  *
   6  *      Copyright (C) 1989 by the Massachusetts Institute of Technology
   7  *      For copying and distribution information, please see the file
   8  *      <mit-copyright.h>.
   9  *
  10  */
  11
  12 #ifndef lint
  13 static char *rcsid_qrtn_qc = "$Header$";
  14 #endif lint
  15
  16 #include <mit-copyright.h>
  17 #include <moira.h>
  18 #include "query.h"
  19 #include "mr_server.h"
  20
  21 extern char *whoami;
  22 char *malloc();
  23
  24 int inc_pid = 0;
  25 int inc_running = 0;
  26 time_t inc_started;
  27
  28 #define MAXARGC 15
  29
  30 /* structures to save before args */
  31 static char beforeb[MAXARGC][ARGLEN];
  32 static char *before[MAXARGC];
  33 ##char *barg0, *barg1, *barg2, *barg3, *barg4;
  34 ##char *barg5, *barg6, *barg7, *barg8, *barg9;
  35 ##char *barg10, *barg11, *barg12, *barg13, *barg14;
  36 static int beforec;
  37 static char *beforetable;
  38
  39 /* structures to save after args */
  40 static char afterb[MAXARGC][ARGLEN];
  41 static char *after[MAXARGC];
  42 ##char *aarg0, *aarg1, *aarg2, *aarg3, *aarg4;
  43 ##char *aarg5, *aarg6, *aarg7, *aarg8, *aarg9;
  44 ##char *aarg10, *aarg11, *aarg12, *aarg13, *aarg14;
  45 static int afterc;
  46
  47 /* structures to save entire sets of incremental changes */
  48 struct save_queue *incremental_sq = NULL;
  49 struct save_queue *incremental_exec = NULL;
  50 struct iupdate {
  51     char *table;
  52     int beforec;
  53     char **before;
  54     int afterc;
  55     char **after;
  56     char *service;
  57 };
  58
  59
  60 incremental_init()
  61 {
  62     int i;
  63
  64     for (i = 0; i < MAXARGC; i++) {
  65         before[i] = &beforeb[i][0];
  66         after[i] = &afterb[i][0];
  67     }
  68     barg0 = before[0];
  69     barg1 = before[1];
  70     barg2 = before[2];
  71     barg3 = before[3];
  72     barg4 = before[4];
  73     barg5 = before[5];
  74     barg6 = before[6];
  75     barg7 = before[7];
  76     barg8 = before[8];
  77     barg9 = before[9];
  78     barg10 = before[10];
  79     barg11 = before[11];
  80     barg12 = before[12];
  81     barg13 = before[13];
  82     barg14 = before[14];
  83     aarg0 = after[0];
  84     aarg1 = after[1];
  85     aarg2 = after[2];
  86     aarg3 = after[3];
  87     aarg4 = after[4];
  88     aarg5 = after[5];
  89     aarg6 = after[6];
  90     aarg7 = after[7];
  91     aarg8 = after[8];
  92     aarg9 = after[9];
  93     aarg10 = after[10];
  94     aarg11 = after[11];
  95     aarg12 = after[12];
  96     aarg13 = after[13];
  97     aarg14 = after[14];
  98     if (incremental_sq == NULL)
  99       incremental_sq = sq_create();
 100     if (incremental_exec == NULL)
 101       incremental_exec = sq_create();
 102 }
 103
 104
 105 ##incremental_before(table, qual, argv)
 106 ##char *table;
 107 ##char *qual;
 108 char **argv;
 109 ##{
 110 ##  int id;
 111     char buffer[512], *name;
 112
 113     beforetable = table;
 114
 115     if (!strcmp(table, "users")) {
 116 ##      retrieve (barg0 = u.login, barg1 = text(u.uid),
 117 ##                barg2 = u.shell, barg3 = u.last, barg4 = u.first,
 118 ##                barg5 = u.middle, barg6 = text(u.status),
 119 ##                barg7 = u.mit_id, barg8 = u.mit_year)
 120 ##              where qual
 121         beforec = 9;
 122     } else if (!strcmp(table, "machine")) {
 123 ##      retrieve (barg0 = m.name, barg1 = m.type) where qual
 124         beforec = 2;
 125     } else if (!strcmp(table, "cluster")) {
 126 ##      retrieve (barg0 = c.name, barg1 = c.desc, barg2 = c.location)
 127 ##              where qual
 128         beforec = 3;
 129     } else if (!strcmp(table, "mcmap")) {
 130         strcpy(barg0, argv[0]);
 131         strcpy(barg1, argv[1]);
 132         beforec = 2;
 133     } else if (!strcmp(table, "svc")) {
 134         strcpy(barg0, argv[0]);
 135         strcpy(barg1, argv[1]);
 136         strcpy(barg2, argv[2]);
 137         beforec = 3;
 138     } else if (!strcmp(table, "filesys")) {
 139 ##      range of fs is filesys
 140 ##      retrieve (barg0 = fs.label, barg1 = fs.type, barg2 = text(fs.mach_id),
 141 ##                barg3 = fs.name, barg4 = fs.mount, barg5 = fs.access,
 142 ##                barg6 = fs.comments, barg7 = text(fs.owner),
 143 ##                barg8 = text(fs.owners), barg9 = text(fs.createflg),
 144 ##                barg10 = fs.lockertype)
 145 ##        where qual
 146         beforec = 11;
 147     } else if (!strcmp(table, "quota")) {
 148         strcpy(barg0, argv[0]);
 149         if (!strcmp(argv[1], "USER") || !strcmp(argv[1], "GROUP") ||
 150             !strcmp(argv[1], "LIST") || !strcmp(argv[1], "ANY")) {
 151             strcpy(barg1, argv[1]);
 152             strcpy(barg2, argv[2]);
 153         } else {
 154             strcpy(barg1, "USER");
 155             strcpy(barg2, argv[1]);
 156         }
 157         sprintf(buffer, "%s and filesys.filsys_id = q.filsys_id", qual);
 158         qual = buffer;
 159 ##      range of q is quota
 160 ##      retrieve (barg3 = text(q.quota), barg4 = filesys.name) where qual
 161         beforec = 5;
 162     } else if (!strcmp(table, "list")) {
 163 ##      retrieve (barg0 = l.name, barg1 = text(l.active),
 164 ##                barg2 = text(l.public), barg3 = text(l.hidden),
 165 ##                barg4 = text(l.maillist), barg5 = text(l.group),
 166 ##                barg6 = text(l.gid), barg7 = l.acl_type,
 167 ##                barg8 = text(l.acl_id), barg9 = l.desc)
 168 ##        where qual
 169         beforec = 10;
 170     } else if (!strcmp(table, "members")) {
 171         id = (int) argv[0];
 172         name = malloc(0);
 173         id_to_name(id, "LIST", &name);
 174         strcpy(barg0, name);
 175         strcpy(barg1, argv[1]);
 176         id = (int) argv[2];
 177         if (!strcmp(barg1, "USER")) {
 178             id_to_name(id, barg1, &name);
 179         } else if (!strcmp(barg1, "LIST")) {
 180             id_to_name(id, barg1, &name);
 181         } else if (!strcmp(barg1, "STRING")) {
 182             id_to_name(id, barg1, &name);
 183         } else if (!strcmp(barg1, "KERBEROS")) {
 184             id_to_name(id, "STRING", &name);
 185         }
 186         strcpy(barg2, name);
 187         free(name);
 188         beforec = 3;
 189     } /* else
 190       com_err(whoami, 0, "unknown table in incremental_before"); */
 191 ##}
 192
 193
 194 incremental_clear_before()
 195 {
 196     beforec = 0;
 197 }
 198
 199 incremental_clear_after()
 200 {
 201     incremental_after("clear", 0);
 202 }
 203
 204
 205
 206 ##incremental_after(table, qual, argv)
 207 ##char *table;
 208 ##char *qual;
 209 char **argv;
 210 ##{
 211     char buffer[2048], *name;
 212 ##  int id, i;
 213     struct iupdate *iu;
 214     char **copy_argv();
 215
 216     if (!strcmp(table, "users")) {
 217 ##      retrieve (aarg0 = u.login, aarg1 = text(u.uid),
 218 ##                aarg2 = u.shell, aarg3 = u.last, aarg4 = u.first,
 219 ##                aarg5 = u.middle, aarg6 = text(u.status),
 220 ##                aarg7 = u.mit_id, aarg8 = u.mit_year)
 221 ##              where qual
 222         afterc = 9;
 223     } else if (!strcmp(table, "machine")) {
 224 ##      retrieve (aarg0 = m.name, aarg1 = m.type) where qual
 225         afterc = 2;
 226     } else if (!strcmp(table, "cluster")) {
 227 ##      retrieve (aarg0 = c.name, aarg1 = c.desc, aarg2 = c.location)
 228 ##              where qual
 229         afterc = 3;
 230     } else if (!strcmp(table, "mcmap")) {
 231         strcpy(aarg0, argv[0]);
 232         strcpy(aarg1, argv[1]);
 233         afterc = 2;
 234     } else if (!strcmp(table, "svc")) {
 235         strcpy(aarg0, argv[0]);
 236         strcpy(aarg1, argv[1]);
 237         strcpy(aarg2, argv[2]);
 238         afterc = 3;
 239     } else if (!strcmp(table, "filesys")) {
 240 ##      range of fs is filesys
 241 ##      retrieve (aarg0 = fs.label, aarg1 = fs.type,
 242 ##                aarg2 = text(fs.mach_id),
 243 ##                aarg3 = fs.name, aarg4 = fs.mount, aarg5 = fs.access,
 244 ##                aarg6 = fs.comments, aarg7 = text(fs.owner),
 245 ##                aarg8 = text(fs.owners), aarg9 = text(fs.createflg),
 246 ##                aarg10 = fs.lockertype)
 247 ##        where qual
 248         afterc = 11;
 249     } else if (!strcmp(table, "quota")) {
 250         strcpy(aarg0, argv[0]);
 251         strcpy(aarg1, argv[1]);
 252         strcpy(aarg2, argv[2]);
 253         sprintf(buffer, "%s and filesys.filsys_id = q.filsys_id", qual);
 254         qual = buffer;
 255 ##      range of q is quota
 256 ##      retrieve (aarg3 = text(q.quota), aarg4 = filesys.name) where qual
 257         afterc = 5;
 258     } else if (!strcmp(table, "list")) {
 259 ##      retrieve (aarg0 = l.name, aarg1 = text(l.active),
 260 ##                aarg2 = text(l.public), aarg3 = text(l.hidden),
 261 ##                aarg4 = text(l.maillist), aarg5 = text(l.group),
 262 ##                aarg6 = text(l.gid), aarg7 = l.acl_type,
 263 ##                aarg8 = text(l.acl_id), aarg9 = l.desc)
 264 ##        where qual
 265         afterc = 10;
 266     } else if (!strcmp(table, "members")) {
 267         id = (int) argv[0];
 268         name = malloc(0);
 269         id_to_name(id, "LIST", &name);
 270         strcpy(aarg0, name);
 271         strcpy(aarg1, argv[1]);
 272         id = (int) argv[2];
 273         if (!strcmp(aarg1, "USER")) {
 274             id_to_name(id, aarg1, &name);
 275         } else if (!strcmp(aarg1, "LIST")) {
 276             id_to_name(id, aarg1, &name);
 277         } else if (!strcmp(aarg1, "STRING")) {
 278             id_to_name(id, aarg1, &name);
 279         } else if (!strcmp(aarg1, "KERBEROS")) {
 280             id_to_name(id, "STRING", &name);
 281         }
 282         strcpy(aarg2, name);
 283         free(name);
 284         afterc = 3;
 285     } else if (!strcmp(table, "clear")) {
 286         afterc = 0;
 287         table = beforetable;
 288     } /* else
 289       com_err(whoami, 0, "unknown table in incremental_after"); */
 290
 291     iu = (struct iupdate *) malloc(sizeof(struct iupdate));
 292     iu->table = strsave(table);
 293     iu->beforec = beforec;
 294     iu->before = copy_argv(before, beforec);
 295     iu->afterc = afterc;
 296     iu->after = copy_argv(after, afterc);
 297     sq_save_data(incremental_sq, iu);
 298
 299 #ifdef DEBUG
 300     sprintf(buffer, "INCREMENTAL(%s, [", table);
 301     for (i = 0; i < beforec; i++) {
 302         if (i == 0)
 303           strcat(buffer, strtrim(before[0]));
 304         else {
 305             strcat(buffer, ", ");
 306             strcat(buffer, strtrim(before[i]));
 307         }
 308     }
 309     strcat(buffer, "], [");
 310     for (i = 0; i < afterc; i++) {
 311         if (i == 0)
 312           strcat(buffer, strtrim(after[0]));
 313         else {
 314             strcat(buffer, ", ");
 315             strcat(buffer, strtrim(after[i]));
 316         }
 317     }
 318     strcat(buffer, "])");
 319     com_err(whoami, 0, buffer);
 320 #endif DEBUG
 321 ##}
 322
 323
 324 /* Called when the current transaction is committed to start any queued
 325  * incremental updates.  This caches the update table the first time it
 326  * is called.
 327  */
 328
 329 struct inc_cache {
 330     struct inc_cache *next;
 331     char *table;
 332     char *service;
 333 };
 334
 335
 336 incremental_update()
 337 {
 338     static int inited = 0;
 339     static struct inc_cache *cache;
 340     struct inc_cache *c;
 341 ##  char tab[17], serv[17];
 342     struct iupdate *iu;
 343
 344     if (!inited) {
 345         inited++;
 346
 347 ##      retrieve (tab = incremental.table, serv = incremental.service) {
 348             c = (struct inc_cache *)malloc(sizeof(struct inc_cache));
 349             c->next = cache;
 350             c->table = strsave(strtrim(tab));
 351             c->service = strsave(strtrim(serv));
 352             cache = c;
 353 ##      }
 354     }
 355
 356     while (sq_remove_data(incremental_sq, &iu)) {
 357         for (c = cache; c; c = c->next) {
 358             if (!strcmp(c->table, iu->table)) {
 359                 iu->service = c->service;
 360                 sq_save_data(incremental_exec, iu);
 361             }
 362         }
 363     }
 364     if (inc_running == 0)
 365       next_incremental();
 366 }
 367
 368
 369 next_incremental()
 370 {
 371     struct iupdate *iu;
 372     char *argv[MAXARGC * 2 + 4], cafter[3], cbefore[3], prog[BUFSIZ];
 373     int i;
 374
 375     if (incremental_exec == NULL)
 376       incremental_init();
 377
 378     if (sq_empty(incremental_exec) ||
 379         (inc_running && now - inc_started < INC_TIMEOUT))
 380       return;
 381
 382     if (inc_running)
 383       com_err(whoami, 0, "incremental timeout on pid %d", inc_pid);
 384
 385     sq_remove_data(incremental_exec, &iu);
 386     argv[1] = iu->table;
 387     sprintf(cbefore, "%d", iu->beforec);
 388     argv[2] = cbefore;
 389     sprintf(cafter, "%d", iu->afterc);
 390     argv[3] = cafter;
 391     for (i = 0; i < iu->beforec; i++)
 392       argv[4 + i] = iu->before[i];
 393     for (i = 0; i < iu->afterc; i++)
 394       argv[4 + iu->beforec + i] = iu->after[i];
 395
 396     sprintf(prog, "%s/%s.incr", BIN_DIR, iu->service);
 397 #ifdef DEBUG
 398     com_err(whoami, 0, "forking %s", prog);
 399 #endif
 400     argv[0] = prog;
 401     argv[4 + iu->beforec + iu->afterc] = 0;
 402     inc_pid = vfork();
 403     switch (inc_pid) {
 404     case 0:
 405         execv(prog, argv);
 406         exit(1);
 407     case -1:
 408         com_err(whoami, 0, "Failed to start incremental update");
 409         break;
 410     default:
 411         inc_running = 1;
 412         inc_started = now;
 413     }
 414
 415     free_argv(iu->before, iu->beforec);
 416     free_argv(iu->after, iu->afterc);
 417     free(iu->table);
 418     free(iu);
 419
 420 }
 421
 422
 423 /* Called when the current transaction is aborted to throw away any queued
 424  * incremental updates
 425  */
 426
 427 incremental_flush()
 428 {
 429     struct iupdate *iu;
 430
 431     while (sq_get_data(incremental_sq, &iu)) {
 432         free_argv(iu->before, iu->beforec);
 433         free_argv(iu->after, iu->afterc);
 434         free(iu->table);
 435         free(iu);
 436     }
 437     sq_destroy(incremental_sq);
 438     incremental_sq = sq_create();
 439 }
 440
 441
 442 char **copy_argv(argv, argc)
 443 char **argv;
 444 int argc;
 445 {
 446     char **ret = (char **)malloc(sizeof(char *) * argc);
 447     while (--argc >= 0)
 448       ret[argc] = strsave(strtrim(argv[argc]));
 449     return(ret);
 450 }
 451
 452 free_argv(argv, argc)
 453 char **argv;
 454 int argc;
 455 {
 456     while (--argc >= 0)
 457       free(argv[argc]);
 458     free(argv);
 459 }