gen/events.pc

   1 /* $Id$
   2  *
   3  * (c) Copyright 2007 by the Massachusetts Institute of Technology.
   4  */
   5
   6 #include <mit-copyright.h>
   7 #include <moira.h>
   8 #include <moira_site.h>
   9
  10 #include <sys/stat.h>
  11
  12 #include <ctype.h>
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16
  17 #include "util.h"
  18
  19 EXEC SQL INCLUDE sqlca;
  20
  21 #define DEFAULT_REALM "@ATHENA.MIT.EDU"
  22
  23 char *whoami = "events.gen";
  24 char *db = "moira/moira";
  25
  26 struct hash *lists;
  27
  28 void output_list(int id, void *list, void *out);
  29
  30 int main(int argc, char **argv)
  31 {
  32   char filename[MAXPATHLEN], *targetfile, *l;
  33   FILE *out = stdout;
  34   int cnt = 0;
  35   EXEC SQL BEGIN DECLARE SECTION;
  36   int lid;
  37   char lname[LIST_NAME_SIZE];
  38   EXEC SQL END DECLARE SECTION;
  39
  40   EXEC SQL CONNECT :db;
  41
  42   if (argc == 2)
  43     {
  44       targetfile = argv[1];
  45       sprintf(filename, "%s~", targetfile);
  46       if (!(out = fopen(filename, "w")))
  47         {
  48           fprintf(stderr, "unable to open %s for output\n", filename);
  49           exit(MR_OCONFIG);
  50         }
  51     }
  52   else if (argc != 1)
  53     {
  54       fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
  55       exit(MR_ARGS);
  56     }
  57
  58   lists = create_hash(15000);
  59
  60   EXEC SQL DECLARE l_cursor CURSOR FOR
  61     SELECT l.list_id, l.name FROM list l
  62     WHERE l.active = 1 and l.maillist = 1;
  63   EXEC SQL OPEN l_cursor;
  64   while (1)
  65     {
  66       EXEC SQL FETCH l_cursor INTO :lid, :lname;
  67       if (sqlca.sqlcode)
  68         break;
  69       l = strdup(strtrim(lname));
  70       if (hash_store(lists, lid, l) < 0)
  71         {
  72           fprintf(stderr, "Out of memory!\n");
  73           exit(MR_NO_MEM);
  74         }
  75       cnt++;
  76     }
  77   EXEC SQL CLOSE l_cursor;
  78   fprintf(stderr, "Loaded %d lists\n", cnt);
  79
  80   hash_step(lists, output_list, out);
  81
  82   if (fclose(out))
  83     {
  84       perror("close failed");
  85       exit(MR_CCONFIG);
  86     }
  87
  88   if (argc == 2)
  89     fix_file(targetfile);
  90   exit(MR_SUCCESS);
  91 }
  92
  93 void output_list(int id, void *list, void *out)
  94 {
  95   EXEC SQL BEGIN DECLARE SECTION;
  96   char *l = list;
  97   int lid = id;
  98   char login[USERS_LOGIN_SIZE];
  99   char principal[STRINGS_STRING_SIZE];
 100   EXEC SQL END DECLARE SECTION;
 101   char *maybecomma = "";
 102
 103   fprintf(out, "%s:", list);
 104
 105   EXEC SQL DECLARE u_cursor CURSOR FOR
 106     SELECT UNIQUE u.login FROM users u, imembers i, list l
 107     WHERE l.list_id = :lid AND l.list_id = i.list_id AND
 108     i.member_type = 'USER' AND i.member_id = u.users_id;
 109   EXEC SQL OPEN u_cursor;
 110   while (1)
 111     {
 112       EXEC SQL FETCH u_cursor INTO :login;
 113       if (sqlca.sqlcode)
 114         break;
 115       fprintf(out, "%s%s", maybecomma, strtrim(login));
 116       maybecomma = ",";
 117     }
 118   EXEC SQL CLOSE u_cursor;
 119
 120   EXEC SQL DECLARE k_cursor CURSOR FOR
 121     SELECT UNIQUE s.string FROM strings s, imembers i, list l
 122     WHERE l.list_id = :lid AND l.list_id = i.list_id AND
 123     i.member_type = 'KERBEROS' AND i.member_id = s.string_id;
 124   EXEC SQL OPEN k_cursor;
 125   while (1)
 126     {
 127       EXEC SQL FETCH k_cursor INTO :principal;
 128       if (sqlca.sqlcode)
 129         break;
 130       if (strstr(principal, DEFAULT_REALM))
 131         {
 132           *strstr(principal, DEFAULT_REALM) = '\0';
 133           fprintf(out, "%s%s", maybecomma, strtrim(principal));
 134           maybecomma = ",";
 135         }
 136     }
 137   EXEC SQL CLOSE k_cursor;
 138
 139   fprintf(out, "\n");
 140 }