source: trunk/third/moira/gen/dhcp.pc @ 24447

Revision 24447, 7.1 KB checked in by broder, 15 years ago (diff)
In moira: * New SVN snapshot.
Line 
1/* $Id: dhcp.pc 3988 2010-02-27 06:24:35Z zacheiss $
2 *
3 * This generates the dhcpd.conf.print and associated files.
4 *
5 * Copyright (C) 1992-1998 by the Massachusetts Institute of Technology.
6 * For copying and distribution information, please see the file
7 * <mit-copyright.h>.
8 */
9
10#include <mit-copyright.h>
11#include <moira.h>
12#include <moira_site.h>
13
14#include <sys/types.h>
15
16#include <ctype.h>
17#include <stdio.h>
18#include <stdlib.h>
19#include <string.h>
20
21#include "util.h"
22
23#define MACHINE_CHWADDR_SIZE 18
24
25EXEC SQL INCLUDE sqlca;
26
27RCSID("$HeadURL: svn+ssh://svn.mit.edu/moira/trunk/moira/gen/dhcp.pc $ $Id: dhcp.pc 3988 2010-02-27 06:24:35Z zacheiss $");
28
29char *whoami = "dhcp.gen";
30char *db = "moira/moira";
31
32void hwcolonify(char *from, char *to);
33
34void sqlerr(void);
35
36int main(int argc, char **argv)
37{
38  EXEC SQL BEGIN DECLARE SECTION;
39  char name[MACHINE_NAME_SIZE], hwaddr[MACHINE_HWADDR_SIZE];
40  char chwaddr[MACHINE_CHWADDR_SIZE];
41  char ohwaddr[MACHINE_HWADDR_SIZE], hwtype[PRINTERS_HWTYPE_SIZE];
42  char addr[MACHINE_ADDRESS_SIZE], location[PRINTERS_LOCATION_SIZE];
43  char contact[PRINTERS_CONTACT_SIZE], logaddr[MACHINE_ADDRESS_SIZE];
44  char modtime[PRINTERS_MODTIME_SIZE], type[PRINTERS_TYPE_SIZE];
45  char *unixtime_fmt = UNIXTIME_FMT;
46  char host[MACHINE_ADDRESS_SIZE], types[SERVERHOSTS_VALUE3_SIZE];
47  int mid, alcount;
48  EXEC SQL END DECLARE SECTION;
49  char shortname[MACHINE_NAME_SIZE], net[MACHINE_ADDRESS_SIZE];
50  char filename[MAXPATHLEN];
51  struct {
52    char types[SERVERHOSTS_VALUE3_SIZE];
53    char host[MACHINE_ADDRESS_SIZE];
54  } *allowlist;
55  char *p, *q;
56  int i, allows, typelen;
57  TARFILE *tf;
58  FILE *out;
59  time_t now = time(NULL);
60
61  EXEC SQL CONNECT :db;
62
63  EXEC SQL WHENEVER SQLERROR DO sqlerr();
64
65  if (argc != 2)
66    {
67      fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
68      exit(MR_ARGS);
69    }
70
71  /* Get print spoolers for allow lists. */
72  EXEC SQL SELECT COUNT(service) INTO :alcount FROM serverhosts sh, machine m
73    WHERE (sh.service = 'PRINT' OR sh.service = 'CUPS-PRINT')
74    AND sh.mach_id = m.mach_id
75    AND (m.status = 1 OR m.status = 2) AND m.address != 'unassigned';
76  allowlist = malloc(alcount * sizeof(*allowlist));
77
78  EXEC SQL DECLARE csr_spool CURSOR FOR
79    SELECT m.address, sh.value3 FROM machine m, serverhosts sh
80    WHERE m.mach_id = sh.mach_id AND
81    (sh.service = 'PRINT' OR sh.service = 'CUPS-PRINT')
82    AND (m.status = 1 OR m.status = 2) AND m.address != 'unassigned';
83  EXEC SQL OPEN csr_spool;
84  for (i = 0; i < alcount; i++)
85    {
86      EXEC SQL FETCH csr_spool INTO :host, :types;
87      if (sqlca.sqlcode)
88        sqlerr();
89      strcpy(allowlist[i].host, strtrim(host));
90      strcpy(allowlist[i].types, strtrim(types));
91    }
92  EXEC SQL CLOSE csr_spool;
93
94  /* Now build the tar file. */
95  tf = tarfile_open(argv[1]);
96
97  /* Build dhcpd.conf.print */
98
99  out = tarfile_start(tf, "/var/boot/dhcpd.conf.print", 0755, 0, 0,
100                      "root", "root", now);
101  ohwaddr[0] = '\0';
102  EXEC SQL DECLARE csr_boot CURSOR FOR
103    SELECT LOWER(m.name), m.hwaddr, m.address, m2.address,
104    pr.location, pr.contact, pr.hwtype
105    FROM printers pr, machine m, machine m2, machine m3
106    WHERE pr.type != 'ALIAS' AND pr.mach_id != 0
107    AND pr.mach_id = m.mach_id AND pr.loghost = m2.mach_id
108    AND pr.status = 1 and pr.mach_id = m.mach_id AND m.status !=3
109    ORDER BY m.hwaddr;
110  EXEC SQL OPEN csr_boot;
111  while (1)
112    {
113      EXEC SQL FETCH csr_boot INTO :name, :hwaddr, :addr, :logaddr,
114        :location, :contact, :hwtype;
115      if (sqlca.sqlcode)
116        break;
117
118      strtrim(hwaddr);
119      if (!*hwaddr || !strcasecmp(hwaddr, "unknown"))
120        continue;
121      if (!strcmp(hwaddr, ohwaddr))
122        {
123          fprintf(stderr, "Ignoring duplicate hwaddr %s\n", hwaddr);
124          continue;
125        }
126      strcpy(ohwaddr, hwaddr);
127
128      hwcolonify(hwaddr, chwaddr);
129
130      strtrim(name);
131      strtrim(addr);
132      strtrim(logaddr);
133      strtrim(location);
134      strtrim(contact);
135      strtrim(hwtype);
136      strcpy(shortname, name);
137      if ((p = strchr(shortname, '.')))
138        *p = '\0';
139
140      if ((p = strchr(addr, '.')) && (q = strchr(++p, '.')))
141        {
142          strncpy(net, p, q - p);
143          net[q - p] = '\0';
144        }
145      else
146        continue;
147
148      fprintf(out, "# %s: %s\n# contact: %s\nhost %s {\n\t"
149              "hardware ethernet %s;\n\tfixed-address %s;\n",
150              shortname, location, contact, name, chwaddr, addr);
151     
152      if (strlen(logaddr) != 0)
153              fprintf(out, "\toption log-servers %s;\n", logaddr);
154             
155      if (!strncmp(hwtype, "HP", 2))
156        fprintf(out, "\toption option-144 \"/hp/%s\";\n", shortname);
157      fprintf(out, "}\n\n");
158    }
159  EXEC SQL CLOSE csr_boot;
160  tarfile_end(tf);
161
162  /* Now generate /var/boot/hp/ files */
163  EXEC SQL DECLARE csr_boot2 CURSOR FOR
164    SELECT LOWER(m.name), m.hwaddr, m2.address, m3.address, pr.type,
165    pr.location, pr.contact, TO_CHAR(pr.modtime, :unixtime_fmt)
166    FROM printers pr, machine m, machine m2, machine m3
167    WHERE pr.hwtype LIKE 'HP%' AND pr.mach_id != 0
168    AND pr.mach_id = m.mach_id AND m.status != 3 AND pr.rm = m2.mach_id
169    AND pr.loghost = m3.mach_id AND pr.type != 'ALIAS' AND pr.status = 1;
170  EXEC SQL OPEN csr_boot2;
171  while (1)
172    {
173      EXEC SQL FETCH csr_boot2 INTO :name, :hwaddr, :addr, :logaddr,
174        :type, :location, :contact, :modtime;
175      if (sqlca.sqlcode)
176        break;
177
178      strtrim(hwaddr);
179      if (!*hwaddr || !strcasecmp(hwaddr, "unknown"))
180        continue;
181
182      strtrim(name);
183      strtrim(addr);
184      strtrim(logaddr);
185      strtrim(type);
186      strtrim(location);
187      strtrim(contact);
188      strcpy(shortname, name);
189      if ((p = strchr(shortname, '.')))
190        *p = '\0';
191
192      /* We create it as foo.new so dhcp.sh can append the passwords
193       * and other data and rename it.
194       */
195      sprintf(filename, "/var/boot/hp/%s.new", shortname);
196      out = tarfile_start(tf, filename, 0755, 0, 0, "root", "root",
197                          unixtime(modtime));
198
199      fprintf(out, "name: %s\nlocation: %s\ncontact: %s\n\n", shortname,
200              *location ? location : "unknown",
201              *contact ? contact : "unknown");
202      if (*logaddr)
203        fprintf(out, "trap-dest: %s\nallow: %s\n", logaddr, logaddr);
204      fprintf(out, "allow: %s\n", addr);
205      typelen = strlen(type);
206      for (i = allows = 0; i < alcount && allows < 9; i++)
207        {
208          char *p;
209
210          /* Don't list the spoolhost twice. */
211          if (!strcmp(allowlist[i].host, addr))
212            continue;
213
214          p = strstr(allowlist[i].types, type);
215          if (!p)
216            continue;
217
218          /* Make sure the match was real, and not just because one type
219           * is a substring of another type.
220           */
221          if (p != allowlist[i].types && *(p - 1) != ',' && *(p - 1) != ' ')
222            continue;
223          p += typelen;
224          if (*p && *p != ',' && *p != ' ')
225            continue;
226
227          fprintf(out, "allow: %s\n", allowlist[i].host);
228          allows++;
229        }
230      /* Rest of data is same for all printers and is appended from a
231       * a file on the boot server.
232       */
233
234      tarfile_end(tf);
235    }
236
237  tarfile_close(tf);
238
239  exit(MR_SUCCESS);
240}
241
242void hwcolonify(char *from, char *to)
243{
244  int f = 0, t = 0;
245  int mod = 2;
246
247  for (f = 0 ; f < MACHINE_HWADDR_SIZE - 1 ; )
248    {
249      to[t++] = from[f++];
250      if (f % mod == 0)
251        to[t++] = ':';
252    }
253       
254  if (f % mod == 0)
255    t--;
256  to[t] = '\0';
257}
258
259void sqlerr(void)
260{
261  db_error(sqlca.sqlcode);
262}
Note: See TracBrowser for help on using the repository browser.