1b528cefcSMark Murray /*
2ae771770SStanislav Sedov * Copyright (c) 1997 - 2008 Kungliga Tekniska Högskolan
3b528cefcSMark Murray * (Royal Institute of Technology, Stockholm, Sweden).
4b528cefcSMark Murray * All rights reserved.
5b528cefcSMark Murray *
6b528cefcSMark Murray * Redistribution and use in source and binary forms, with or without
7b528cefcSMark Murray * modification, are permitted provided that the following conditions
8b528cefcSMark Murray * are met:
9b528cefcSMark Murray *
10b528cefcSMark Murray * 1. Redistributions of source code must retain the above copyright
11b528cefcSMark Murray * notice, this list of conditions and the following disclaimer.
12b528cefcSMark Murray *
13b528cefcSMark Murray * 2. Redistributions in binary form must reproduce the above copyright
14b528cefcSMark Murray * notice, this list of conditions and the following disclaimer in the
15b528cefcSMark Murray * documentation and/or other materials provided with the distribution.
16b528cefcSMark Murray *
17b528cefcSMark Murray * 3. Neither the name of the Institute nor the names of its contributors
18b528cefcSMark Murray * may be used to endorse or promote products derived from this software
19b528cefcSMark Murray * without specific prior written permission.
20b528cefcSMark Murray *
21b528cefcSMark Murray * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22b528cefcSMark Murray * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23b528cefcSMark Murray * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24b528cefcSMark Murray * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25b528cefcSMark Murray * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26b528cefcSMark Murray * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27b528cefcSMark Murray * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28b528cefcSMark Murray * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29b528cefcSMark Murray * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30b528cefcSMark Murray * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31b528cefcSMark Murray * SUCH DAMAGE.
32b528cefcSMark Murray */
33b528cefcSMark Murray
34b528cefcSMark Murray #include "iprop.h"
358373020dSJacques Vidrine #include <rtbl.h>
36b528cefcSMark Murray
375e9cd1aeSAssar Westerlund static krb5_log_facility *log_facility;
38b528cefcSMark Murray
39c19800e8SDoug Rabson const char *slave_stats_file;
40c19800e8SDoug Rabson const char *slave_time_missing = "2 min";
41c19800e8SDoug Rabson const char *slave_time_gone = "5 min";
42c19800e8SDoug Rabson
43c19800e8SDoug Rabson static int time_before_missing;
44c19800e8SDoug Rabson static int time_before_gone;
45c19800e8SDoug Rabson
46c19800e8SDoug Rabson const char *master_hostname;
47bbd80c28SJacques Vidrine
48ae771770SStanislav Sedov static krb5_socket_t
make_signal_socket(krb5_context context)49b528cefcSMark Murray make_signal_socket (krb5_context context)
50b528cefcSMark Murray {
51ae771770SStanislav Sedov #ifndef NO_UNIX_SOCKETS
52b528cefcSMark Murray struct sockaddr_un addr;
53c19800e8SDoug Rabson const char *fn;
54ae771770SStanislav Sedov krb5_socket_t fd;
55b528cefcSMark Murray
56c19800e8SDoug Rabson fn = kadm5_log_signal_socket(context);
57c19800e8SDoug Rabson
58b528cefcSMark Murray fd = socket (AF_UNIX, SOCK_DGRAM, 0);
59b528cefcSMark Murray if (fd < 0)
60b528cefcSMark Murray krb5_err (context, 1, errno, "socket AF_UNIX");
61b528cefcSMark Murray memset (&addr, 0, sizeof(addr));
62b528cefcSMark Murray addr.sun_family = AF_UNIX;
63c19800e8SDoug Rabson strlcpy (addr.sun_path, fn, sizeof(addr.sun_path));
64b528cefcSMark Murray unlink (addr.sun_path);
65b528cefcSMark Murray if (bind (fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
66b528cefcSMark Murray krb5_err (context, 1, errno, "bind %s", addr.sun_path);
67b528cefcSMark Murray return fd;
68ae771770SStanislav Sedov #else
69ae771770SStanislav Sedov struct addrinfo *ai = NULL;
70ae771770SStanislav Sedov krb5_socket_t fd;
71ae771770SStanislav Sedov
72ae771770SStanislav Sedov kadm5_log_signal_socket_info(context, 1, &ai);
73ae771770SStanislav Sedov
74ae771770SStanislav Sedov fd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
75ae771770SStanislav Sedov if (rk_IS_BAD_SOCKET(fd))
76ae771770SStanislav Sedov krb5_err (context, 1, rk_SOCK_ERRNO, "socket AF=%d", ai->ai_family);
77ae771770SStanislav Sedov
78ae771770SStanislav Sedov if (rk_IS_SOCKET_ERROR( bind (fd, ai->ai_addr, ai->ai_addrlen) ))
79ae771770SStanislav Sedov krb5_err (context, 1, rk_SOCK_ERRNO, "bind");
80ae771770SStanislav Sedov return fd;
81ae771770SStanislav Sedov #endif
82b528cefcSMark Murray }
83b528cefcSMark Murray
84ae771770SStanislav Sedov static krb5_socket_t
make_listen_socket(krb5_context context,const char * port_str)85c19800e8SDoug Rabson make_listen_socket (krb5_context context, const char *port_str)
86b528cefcSMark Murray {
87ae771770SStanislav Sedov krb5_socket_t fd;
88b528cefcSMark Murray int one = 1;
89b528cefcSMark Murray struct sockaddr_in addr;
90b528cefcSMark Murray
91b528cefcSMark Murray fd = socket (AF_INET, SOCK_STREAM, 0);
92ae771770SStanislav Sedov if (rk_IS_BAD_SOCKET(fd))
93ae771770SStanislav Sedov krb5_err (context, 1, rk_SOCK_ERRNO, "socket AF_INET");
944137ff4cSJacques Vidrine setsockopt (fd, SOL_SOCKET, SO_REUSEADDR, (void *)&one, sizeof(one));
95b528cefcSMark Murray memset (&addr, 0, sizeof(addr));
96b528cefcSMark Murray addr.sin_family = AF_INET;
97c19800e8SDoug Rabson
98c19800e8SDoug Rabson if (port_str) {
995e9cd1aeSAssar Westerlund addr.sin_port = krb5_getportbyname (context,
100c19800e8SDoug Rabson port_str, "tcp",
101c19800e8SDoug Rabson 0);
102c19800e8SDoug Rabson if (addr.sin_port == 0) {
103c19800e8SDoug Rabson char *ptr;
104c19800e8SDoug Rabson long port;
105c19800e8SDoug Rabson
106c19800e8SDoug Rabson port = strtol (port_str, &ptr, 10);
107c19800e8SDoug Rabson if (port == 0 && ptr == port_str)
108c19800e8SDoug Rabson krb5_errx (context, 1, "bad port `%s'", port_str);
109c19800e8SDoug Rabson addr.sin_port = htons(port);
110c19800e8SDoug Rabson }
111c19800e8SDoug Rabson } else {
112c19800e8SDoug Rabson addr.sin_port = krb5_getportbyname (context, IPROP_SERVICE,
113c19800e8SDoug Rabson "tcp", IPROP_PORT);
114c19800e8SDoug Rabson }
115b528cefcSMark Murray if(bind(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
116b528cefcSMark Murray krb5_err (context, 1, errno, "bind");
117b528cefcSMark Murray if (listen(fd, SOMAXCONN) < 0)
118b528cefcSMark Murray krb5_err (context, 1, errno, "listen");
119b528cefcSMark Murray return fd;
120b528cefcSMark Murray }
121b528cefcSMark Murray
1229c91c227SBjoern A. Zeeb static krb5_socket_t
make_listen6_socket(krb5_context context,const char * port_str)1239c91c227SBjoern A. Zeeb make_listen6_socket (krb5_context context, const char *port_str)
1249c91c227SBjoern A. Zeeb {
1259c91c227SBjoern A. Zeeb krb5_socket_t fd;
1269c91c227SBjoern A. Zeeb int one = 1;
1279c91c227SBjoern A. Zeeb struct sockaddr_in6 addr;
1289c91c227SBjoern A. Zeeb
1299c91c227SBjoern A. Zeeb fd = socket (AF_INET6, SOCK_STREAM, 0);
1309c91c227SBjoern A. Zeeb if (rk_IS_BAD_SOCKET(fd))
1319c91c227SBjoern A. Zeeb krb5_err (context, 1, rk_SOCK_ERRNO, "socket AF_INET6");
1329c91c227SBjoern A. Zeeb setsockopt (fd, SOL_SOCKET, SO_REUSEADDR, (void *)&one, sizeof(one));
1339c91c227SBjoern A. Zeeb memset (&addr, 0, sizeof(addr));
1349c91c227SBjoern A. Zeeb addr.sin6_family = AF_INET6;
1359c91c227SBjoern A. Zeeb
1369c91c227SBjoern A. Zeeb if (port_str) {
1379c91c227SBjoern A. Zeeb addr.sin6_port = krb5_getportbyname (context,
1389c91c227SBjoern A. Zeeb port_str, "tcp",
1399c91c227SBjoern A. Zeeb 0);
1409c91c227SBjoern A. Zeeb if (addr.sin6_port == 0) {
1419c91c227SBjoern A. Zeeb char *ptr;
1429c91c227SBjoern A. Zeeb long port;
1439c91c227SBjoern A. Zeeb
1449c91c227SBjoern A. Zeeb port = strtol (port_str, &ptr, 10);
1459c91c227SBjoern A. Zeeb if (port == 0 && ptr == port_str)
1469c91c227SBjoern A. Zeeb krb5_errx (context, 1, "bad port `%s'", port_str);
1479c91c227SBjoern A. Zeeb addr.sin6_port = htons(port);
1489c91c227SBjoern A. Zeeb }
1499c91c227SBjoern A. Zeeb } else {
1509c91c227SBjoern A. Zeeb addr.sin6_port = krb5_getportbyname (context, IPROP_SERVICE,
1519c91c227SBjoern A. Zeeb "tcp", IPROP_PORT);
1529c91c227SBjoern A. Zeeb }
1539c91c227SBjoern A. Zeeb if(bind(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
1549c91c227SBjoern A. Zeeb krb5_err (context, 1, errno, "bind6");
1559c91c227SBjoern A. Zeeb if (listen(fd, SOMAXCONN) < 0)
1569c91c227SBjoern A. Zeeb krb5_err (context, 1, errno, "listen6");
1579c91c227SBjoern A. Zeeb return fd;
1589c91c227SBjoern A. Zeeb }
1599c91c227SBjoern A. Zeeb
1609c91c227SBjoern A. Zeeb #ifndef _SOCKADDR_UNION
1619c91c227SBjoern A. Zeeb #define _SOCKADDR_UNION
1629c91c227SBjoern A. Zeeb union sockaddr_union {
1639c91c227SBjoern A. Zeeb struct sockaddr sa;
1649c91c227SBjoern A. Zeeb struct sockaddr_in sin;
1659c91c227SBjoern A. Zeeb struct sockaddr_in6 sin6;
1669c91c227SBjoern A. Zeeb };
1679c91c227SBjoern A. Zeeb #endif /* _SOCKADDR_UNION */
1689c91c227SBjoern A. Zeeb
169b528cefcSMark Murray struct slave {
170ae771770SStanislav Sedov krb5_socket_t fd;
1719c91c227SBjoern A. Zeeb union sockaddr_union addr;
172b528cefcSMark Murray char *name;
173b528cefcSMark Murray krb5_auth_context ac;
174c19800e8SDoug Rabson uint32_t version;
1758373020dSJacques Vidrine time_t seen;
1768373020dSJacques Vidrine unsigned long flags;
1778373020dSJacques Vidrine #define SLAVE_F_DEAD 0x1
178c19800e8SDoug Rabson #define SLAVE_F_AYT 0x2
179b528cefcSMark Murray struct slave *next;
180b528cefcSMark Murray };
181b528cefcSMark Murray
182b528cefcSMark Murray typedef struct slave slave;
183b528cefcSMark Murray
184b528cefcSMark Murray static int
check_acl(krb5_context context,const char * name)185b528cefcSMark Murray check_acl (krb5_context context, const char *name)
186b528cefcSMark Murray {
187c19800e8SDoug Rabson const char *fn;
188b528cefcSMark Murray FILE *fp;
189b528cefcSMark Murray char buf[256];
190b528cefcSMark Murray int ret = 1;
191ae771770SStanislav Sedov char *slavefile = NULL;
192b528cefcSMark Murray
193ae771770SStanislav Sedov if (asprintf(&slavefile, "%s/slaves", hdb_db_dir(context)) == -1
194ae771770SStanislav Sedov || slavefile == NULL)
195ae771770SStanislav Sedov errx(1, "out of memory");
196c19800e8SDoug Rabson
197c19800e8SDoug Rabson fn = krb5_config_get_string_default(context,
198c19800e8SDoug Rabson NULL,
199c19800e8SDoug Rabson slavefile,
200c19800e8SDoug Rabson "kdc",
201c19800e8SDoug Rabson "iprop-acl",
202c19800e8SDoug Rabson NULL);
203c19800e8SDoug Rabson
204c19800e8SDoug Rabson fp = fopen (fn, "r");
205c19800e8SDoug Rabson free(slavefile);
206b528cefcSMark Murray if (fp == NULL)
207b528cefcSMark Murray return 1;
208b528cefcSMark Murray while (fgets(buf, sizeof(buf), fp) != NULL) {
209c19800e8SDoug Rabson buf[strcspn(buf, "\r\n")] = '\0';
210b528cefcSMark Murray if (strcmp (buf, name) == 0) {
211b528cefcSMark Murray ret = 0;
212b528cefcSMark Murray break;
213b528cefcSMark Murray }
214b528cefcSMark Murray }
215b528cefcSMark Murray fclose (fp);
216b528cefcSMark Murray return ret;
217b528cefcSMark Murray }
218b528cefcSMark Murray
219b528cefcSMark Murray static void
slave_seen(slave * s)2208373020dSJacques Vidrine slave_seen(slave *s)
2218373020dSJacques Vidrine {
222c19800e8SDoug Rabson s->flags &= ~SLAVE_F_AYT;
2238373020dSJacques Vidrine s->seen = time(NULL);
2248373020dSJacques Vidrine }
2258373020dSJacques Vidrine
226c19800e8SDoug Rabson static int
slave_missing_p(slave * s)227c19800e8SDoug Rabson slave_missing_p (slave *s)
2288373020dSJacques Vidrine {
229c19800e8SDoug Rabson if (time(NULL) > s->seen + time_before_missing)
230c19800e8SDoug Rabson return 1;
231c19800e8SDoug Rabson return 0;
232c19800e8SDoug Rabson }
233c19800e8SDoug Rabson
234c19800e8SDoug Rabson static int
slave_gone_p(slave * s)235c19800e8SDoug Rabson slave_gone_p (slave *s)
236c19800e8SDoug Rabson {
237c19800e8SDoug Rabson if (time(NULL) > s->seen + time_before_gone)
238c19800e8SDoug Rabson return 1;
239c19800e8SDoug Rabson return 0;
240c19800e8SDoug Rabson }
241c19800e8SDoug Rabson
242c19800e8SDoug Rabson static void
slave_dead(krb5_context context,slave * s)243c19800e8SDoug Rabson slave_dead(krb5_context context, slave *s)
244c19800e8SDoug Rabson {
245c19800e8SDoug Rabson krb5_warnx(context, "slave %s dead", s->name);
246c19800e8SDoug Rabson
247ae771770SStanislav Sedov if (!rk_IS_BAD_SOCKET(s->fd)) {
248ae771770SStanislav Sedov rk_closesocket (s->fd);
249ae771770SStanislav Sedov s->fd = rk_INVALID_SOCKET;
250bbd80c28SJacques Vidrine }
2518373020dSJacques Vidrine s->flags |= SLAVE_F_DEAD;
2528373020dSJacques Vidrine slave_seen(s);
2538373020dSJacques Vidrine }
2548373020dSJacques Vidrine
2558373020dSJacques Vidrine static void
remove_slave(krb5_context context,slave * s,slave ** root)2568373020dSJacques Vidrine remove_slave (krb5_context context, slave *s, slave **root)
2578373020dSJacques Vidrine {
2588373020dSJacques Vidrine slave **p;
2598373020dSJacques Vidrine
260ae771770SStanislav Sedov if (!rk_IS_BAD_SOCKET(s->fd))
261ae771770SStanislav Sedov rk_closesocket (s->fd);
2628373020dSJacques Vidrine if (s->name)
2638373020dSJacques Vidrine free (s->name);
2648373020dSJacques Vidrine if (s->ac)
2658373020dSJacques Vidrine krb5_auth_con_free (context, s->ac);
2668373020dSJacques Vidrine
2678373020dSJacques Vidrine for (p = root; *p; p = &(*p)->next)
2688373020dSJacques Vidrine if (*p == s) {
2698373020dSJacques Vidrine *p = s->next;
2708373020dSJacques Vidrine break;
2718373020dSJacques Vidrine }
2728373020dSJacques Vidrine free (s);
2738373020dSJacques Vidrine }
2748373020dSJacques Vidrine
2758373020dSJacques Vidrine static void
add_slave(krb5_context context,krb5_keytab keytab,slave ** root,krb5_socket_t fd)276ae771770SStanislav Sedov add_slave (krb5_context context, krb5_keytab keytab, slave **root,
277ae771770SStanislav Sedov krb5_socket_t fd)
278b528cefcSMark Murray {
279b528cefcSMark Murray krb5_principal server;
280b528cefcSMark Murray krb5_error_code ret;
281b528cefcSMark Murray slave *s;
2825e9cd1aeSAssar Westerlund socklen_t addr_len;
283b528cefcSMark Murray krb5_ticket *ticket = NULL;
284b528cefcSMark Murray char hostname[128];
285b528cefcSMark Murray
286b528cefcSMark Murray s = malloc(sizeof(*s));
287b528cefcSMark Murray if (s == NULL) {
288b528cefcSMark Murray krb5_warnx (context, "add_slave: no memory");
289b528cefcSMark Murray return;
290b528cefcSMark Murray }
291b528cefcSMark Murray s->name = NULL;
292b528cefcSMark Murray s->ac = NULL;
293b528cefcSMark Murray
2949c91c227SBjoern A. Zeeb addr_len = sizeof(s->addr.sin6);
2959c91c227SBjoern A. Zeeb s->fd = accept (fd, (struct sockaddr *)&s->addr.sa, &addr_len);
296ae771770SStanislav Sedov if (rk_IS_BAD_SOCKET(s->fd)) {
297ae771770SStanislav Sedov krb5_warn (context, rk_SOCK_ERRNO, "accept");
298b528cefcSMark Murray goto error;
299b528cefcSMark Murray }
300c19800e8SDoug Rabson if (master_hostname)
301c19800e8SDoug Rabson strlcpy(hostname, master_hostname, sizeof(hostname));
302c19800e8SDoug Rabson else
303b528cefcSMark Murray gethostname(hostname, sizeof(hostname));
304c19800e8SDoug Rabson
305b528cefcSMark Murray ret = krb5_sname_to_principal (context, hostname, IPROP_NAME,
306b528cefcSMark Murray KRB5_NT_SRV_HST, &server);
307b528cefcSMark Murray if (ret) {
308b528cefcSMark Murray krb5_warn (context, ret, "krb5_sname_to_principal");
309b528cefcSMark Murray goto error;
310b528cefcSMark Murray }
311b528cefcSMark Murray
312b528cefcSMark Murray ret = krb5_recvauth (context, &s->ac, &s->fd,
3135e9cd1aeSAssar Westerlund IPROP_VERSION, server, 0, keytab, &ticket);
314b528cefcSMark Murray krb5_free_principal (context, server);
315b528cefcSMark Murray if (ret) {
316b528cefcSMark Murray krb5_warn (context, ret, "krb5_recvauth");
317b528cefcSMark Murray goto error;
318b528cefcSMark Murray }
319b528cefcSMark Murray ret = krb5_unparse_name (context, ticket->client, &s->name);
320ae771770SStanislav Sedov krb5_free_ticket (context, ticket);
321b528cefcSMark Murray if (ret) {
322b528cefcSMark Murray krb5_warn (context, ret, "krb5_unparse_name");
323b528cefcSMark Murray goto error;
324b528cefcSMark Murray }
325b528cefcSMark Murray if (check_acl (context, s->name)) {
326b528cefcSMark Murray krb5_warnx (context, "%s not in acl", s->name);
327b528cefcSMark Murray goto error;
328b528cefcSMark Murray }
3298373020dSJacques Vidrine
3308373020dSJacques Vidrine {
3318373020dSJacques Vidrine slave *l = *root;
3328373020dSJacques Vidrine
3338373020dSJacques Vidrine while (l) {
3348373020dSJacques Vidrine if (strcmp(l->name, s->name) == 0)
3358373020dSJacques Vidrine break;
3368373020dSJacques Vidrine l = l->next;
3378373020dSJacques Vidrine }
3388373020dSJacques Vidrine if (l) {
3398373020dSJacques Vidrine if (l->flags & SLAVE_F_DEAD) {
3408373020dSJacques Vidrine remove_slave(context, l, root);
3418373020dSJacques Vidrine } else {
3428373020dSJacques Vidrine krb5_warnx (context, "second connection from %s", s->name);
3438373020dSJacques Vidrine goto error;
3448373020dSJacques Vidrine }
3458373020dSJacques Vidrine }
3468373020dSJacques Vidrine }
3478373020dSJacques Vidrine
3485e9cd1aeSAssar Westerlund krb5_warnx (context, "connection from %s", s->name);
349b528cefcSMark Murray
350b528cefcSMark Murray s->version = 0;
3518373020dSJacques Vidrine s->flags = 0;
3528373020dSJacques Vidrine slave_seen(s);
353b528cefcSMark Murray s->next = *root;
354b528cefcSMark Murray *root = s;
355b528cefcSMark Murray return;
356b528cefcSMark Murray error:
3578373020dSJacques Vidrine remove_slave(context, s, root);
358b528cefcSMark Murray }
359b528cefcSMark Murray
3605e9cd1aeSAssar Westerlund struct prop_context {
3615e9cd1aeSAssar Westerlund krb5_auth_context auth_context;
362ae771770SStanislav Sedov krb5_socket_t fd;
3635e9cd1aeSAssar Westerlund };
3645e9cd1aeSAssar Westerlund
365b528cefcSMark Murray static int
prop_one(krb5_context context,HDB * db,hdb_entry_ex * entry,void * v)366c19800e8SDoug Rabson prop_one (krb5_context context, HDB *db, hdb_entry_ex *entry, void *v)
367b528cefcSMark Murray {
3685e9cd1aeSAssar Westerlund krb5_error_code ret;
369c19800e8SDoug Rabson krb5_storage *sp;
3705e9cd1aeSAssar Westerlund krb5_data data;
371c19800e8SDoug Rabson struct slave *s = (struct slave *)v;
3725e9cd1aeSAssar Westerlund
373c19800e8SDoug Rabson ret = hdb_entry2value (context, &entry->entry, &data);
3745e9cd1aeSAssar Westerlund if (ret)
3755e9cd1aeSAssar Westerlund return ret;
3765e9cd1aeSAssar Westerlund ret = krb5_data_realloc (&data, data.length + 4);
3775e9cd1aeSAssar Westerlund if (ret) {
3785e9cd1aeSAssar Westerlund krb5_data_free (&data);
3795e9cd1aeSAssar Westerlund return ret;
3805e9cd1aeSAssar Westerlund }
3815e9cd1aeSAssar Westerlund memmove ((char *)data.data + 4, data.data, data.length - 4);
382c19800e8SDoug Rabson sp = krb5_storage_from_data(&data);
383c19800e8SDoug Rabson if (sp == NULL) {
384c19800e8SDoug Rabson krb5_data_free (&data);
385c19800e8SDoug Rabson return ENOMEM;
386c19800e8SDoug Rabson }
387c19800e8SDoug Rabson krb5_store_int32(sp, ONE_PRINC);
388c19800e8SDoug Rabson krb5_storage_free(sp);
3895e9cd1aeSAssar Westerlund
390c19800e8SDoug Rabson ret = krb5_write_priv_message (context, s->ac, &s->fd, &data);
3915e9cd1aeSAssar Westerlund krb5_data_free (&data);
3925e9cd1aeSAssar Westerlund return ret;
3935e9cd1aeSAssar Westerlund }
3945e9cd1aeSAssar Westerlund
3955e9cd1aeSAssar Westerlund static int
send_complete(krb5_context context,slave * s,const char * database,uint32_t current_version)3965e9cd1aeSAssar Westerlund send_complete (krb5_context context, slave *s,
397c19800e8SDoug Rabson const char *database, uint32_t current_version)
3985e9cd1aeSAssar Westerlund {
3995e9cd1aeSAssar Westerlund krb5_error_code ret;
400c19800e8SDoug Rabson krb5_storage *sp;
4015e9cd1aeSAssar Westerlund HDB *db;
4025e9cd1aeSAssar Westerlund krb5_data data;
4035e9cd1aeSAssar Westerlund char buf[8];
4045e9cd1aeSAssar Westerlund
4055e9cd1aeSAssar Westerlund ret = hdb_create (context, &db, database);
4065e9cd1aeSAssar Westerlund if (ret)
4075e9cd1aeSAssar Westerlund krb5_err (context, 1, ret, "hdb_create: %s", database);
408c19800e8SDoug Rabson ret = db->hdb_open (context, db, O_RDONLY, 0);
4095e9cd1aeSAssar Westerlund if (ret)
4105e9cd1aeSAssar Westerlund krb5_err (context, 1, ret, "db->open");
4115e9cd1aeSAssar Westerlund
412c19800e8SDoug Rabson sp = krb5_storage_from_mem (buf, 4);
413c19800e8SDoug Rabson if (sp == NULL)
414c19800e8SDoug Rabson krb5_errx (context, 1, "krb5_storage_from_mem");
415c19800e8SDoug Rabson krb5_store_int32 (sp, TELL_YOU_EVERYTHING);
416c19800e8SDoug Rabson krb5_storage_free (sp);
4175e9cd1aeSAssar Westerlund
4185e9cd1aeSAssar Westerlund data.data = buf;
4195e9cd1aeSAssar Westerlund data.length = 4;
4205e9cd1aeSAssar Westerlund
4215e9cd1aeSAssar Westerlund ret = krb5_write_priv_message(context, s->ac, &s->fd, &data);
4225e9cd1aeSAssar Westerlund
4238373020dSJacques Vidrine if (ret) {
4248373020dSJacques Vidrine krb5_warn (context, ret, "krb5_write_priv_message");
425c19800e8SDoug Rabson slave_dead(context, s);
4268373020dSJacques Vidrine return ret;
4278373020dSJacques Vidrine }
4285e9cd1aeSAssar Westerlund
429ae771770SStanislav Sedov ret = hdb_foreach (context, db, HDB_F_ADMIN_DATA, prop_one, s);
4308373020dSJacques Vidrine if (ret) {
4318373020dSJacques Vidrine krb5_warn (context, ret, "hdb_foreach");
432c19800e8SDoug Rabson slave_dead(context, s);
4338373020dSJacques Vidrine return ret;
4348373020dSJacques Vidrine }
4355e9cd1aeSAssar Westerlund
436c19800e8SDoug Rabson (*db->hdb_close)(context, db);
437c19800e8SDoug Rabson (*db->hdb_destroy)(context, db);
438c19800e8SDoug Rabson
439c19800e8SDoug Rabson sp = krb5_storage_from_mem (buf, 8);
440c19800e8SDoug Rabson if (sp == NULL)
441c19800e8SDoug Rabson krb5_errx (context, 1, "krb5_storage_from_mem");
442c19800e8SDoug Rabson krb5_store_int32 (sp, NOW_YOU_HAVE);
443c19800e8SDoug Rabson krb5_store_int32 (sp, current_version);
444c19800e8SDoug Rabson krb5_storage_free (sp);
445c19800e8SDoug Rabson
4465e9cd1aeSAssar Westerlund data.length = 8;
4475e9cd1aeSAssar Westerlund
4488373020dSJacques Vidrine s->version = current_version;
4495e9cd1aeSAssar Westerlund
4508373020dSJacques Vidrine ret = krb5_write_priv_message(context, s->ac, &s->fd, &data);
4518373020dSJacques Vidrine if (ret) {
452c19800e8SDoug Rabson slave_dead(context, s);
4538373020dSJacques Vidrine krb5_warn (context, ret, "krb5_write_priv_message");
4548373020dSJacques Vidrine return ret;
4558373020dSJacques Vidrine }
4568373020dSJacques Vidrine
4578373020dSJacques Vidrine slave_seen(s);
4585e9cd1aeSAssar Westerlund
4595e9cd1aeSAssar Westerlund return 0;
460b528cefcSMark Murray }
461b528cefcSMark Murray
462b528cefcSMark Murray static int
send_are_you_there(krb5_context context,slave * s)463c19800e8SDoug Rabson send_are_you_there (krb5_context context, slave *s)
464b528cefcSMark Murray {
4655e9cd1aeSAssar Westerlund krb5_storage *sp;
466c19800e8SDoug Rabson krb5_data data;
467c19800e8SDoug Rabson char buf[4];
468c19800e8SDoug Rabson int ret;
469c19800e8SDoug Rabson
470c19800e8SDoug Rabson if (s->flags & (SLAVE_F_DEAD|SLAVE_F_AYT))
471c19800e8SDoug Rabson return 0;
472c19800e8SDoug Rabson
473ae771770SStanislav Sedov krb5_warnx(context, "slave %s missing, sending AYT", s->name);
474ae771770SStanislav Sedov
475c19800e8SDoug Rabson s->flags |= SLAVE_F_AYT;
476c19800e8SDoug Rabson
477c19800e8SDoug Rabson data.data = buf;
478c19800e8SDoug Rabson data.length = 4;
479c19800e8SDoug Rabson
480c19800e8SDoug Rabson sp = krb5_storage_from_mem (buf, 4);
481c19800e8SDoug Rabson if (sp == NULL) {
482c19800e8SDoug Rabson krb5_warnx (context, "are_you_there: krb5_data_alloc");
483c19800e8SDoug Rabson slave_dead(context, s);
484c19800e8SDoug Rabson return 1;
485c19800e8SDoug Rabson }
486c19800e8SDoug Rabson krb5_store_int32 (sp, ARE_YOU_THERE);
487c19800e8SDoug Rabson krb5_storage_free (sp);
488c19800e8SDoug Rabson
489c19800e8SDoug Rabson ret = krb5_write_priv_message(context, s->ac, &s->fd, &data);
490c19800e8SDoug Rabson
491c19800e8SDoug Rabson if (ret) {
492c19800e8SDoug Rabson krb5_warn (context, ret, "are_you_there: krb5_write_priv_message");
493c19800e8SDoug Rabson slave_dead(context, s);
494c19800e8SDoug Rabson return 1;
495c19800e8SDoug Rabson }
496c19800e8SDoug Rabson
497c19800e8SDoug Rabson return 0;
498c19800e8SDoug Rabson }
499c19800e8SDoug Rabson
500c19800e8SDoug Rabson static int
send_diffs(krb5_context context,slave * s,int log_fd,const char * database,uint32_t current_version)501c19800e8SDoug Rabson send_diffs (krb5_context context, slave *s, int log_fd,
502c19800e8SDoug Rabson const char *database, uint32_t current_version)
503c19800e8SDoug Rabson {
504c19800e8SDoug Rabson krb5_storage *sp;
505c19800e8SDoug Rabson uint32_t ver;
506b528cefcSMark Murray time_t timestamp;
507b528cefcSMark Murray enum kadm_ops op;
508c19800e8SDoug Rabson uint32_t len;
509b528cefcSMark Murray off_t right, left;
510b528cefcSMark Murray krb5_data data;
511b528cefcSMark Murray int ret = 0;
512b528cefcSMark Murray
513c19800e8SDoug Rabson if (s->version == current_version) {
514c19800e8SDoug Rabson krb5_warnx(context, "slave %s in sync already at version %ld",
515c19800e8SDoug Rabson s->name, (long)s->version);
516b528cefcSMark Murray return 0;
517c19800e8SDoug Rabson }
518b528cefcSMark Murray
5198373020dSJacques Vidrine if (s->flags & SLAVE_F_DEAD)
5208373020dSJacques Vidrine return 0;
5218373020dSJacques Vidrine
522c19800e8SDoug Rabson /* if slave is a fresh client, starting over */
523c19800e8SDoug Rabson if (s->version == 0) {
524c19800e8SDoug Rabson krb5_warnx(context, "sending complete log to fresh slave %s",
525c19800e8SDoug Rabson s->name);
526c19800e8SDoug Rabson return send_complete (context, s, database, current_version);
527c19800e8SDoug Rabson }
528c19800e8SDoug Rabson
529b528cefcSMark Murray sp = kadm5_log_goto_end (log_fd);
5308373020dSJacques Vidrine right = krb5_storage_seek(sp, 0, SEEK_CUR);
531b528cefcSMark Murray for (;;) {
532c19800e8SDoug Rabson ret = kadm5_log_previous (context, sp, &ver, ×tamp, &op, &len);
533c19800e8SDoug Rabson if (ret)
534c19800e8SDoug Rabson krb5_err(context, 1, ret,
535c19800e8SDoug Rabson "send_diffs: failed to find previous entry");
5368373020dSJacques Vidrine left = krb5_storage_seek(sp, -16, SEEK_CUR);
537b528cefcSMark Murray if (ver == s->version)
538b528cefcSMark Murray return 0;
539b528cefcSMark Murray if (ver == s->version + 1)
540b528cefcSMark Murray break;
541c19800e8SDoug Rabson if (left == 0) {
542ae771770SStanislav Sedov krb5_storage_free(sp);
543c19800e8SDoug Rabson krb5_warnx(context,
544c19800e8SDoug Rabson "slave %s (version %lu) out of sync with master "
545c19800e8SDoug Rabson "(first version in log %lu), sending complete database",
546c19800e8SDoug Rabson s->name, (unsigned long)s->version, (unsigned long)ver);
5475e9cd1aeSAssar Westerlund return send_complete (context, s, database, current_version);
548b528cefcSMark Murray }
549c19800e8SDoug Rabson }
550c19800e8SDoug Rabson
551c19800e8SDoug Rabson krb5_warnx(context,
552c19800e8SDoug Rabson "syncing slave %s from version %lu to version %lu",
553c19800e8SDoug Rabson s->name, (unsigned long)s->version,
554c19800e8SDoug Rabson (unsigned long)current_version);
555c19800e8SDoug Rabson
556c19800e8SDoug Rabson ret = krb5_data_alloc (&data, right - left + 4);
557c19800e8SDoug Rabson if (ret) {
558ae771770SStanislav Sedov krb5_storage_free(sp);
559c19800e8SDoug Rabson krb5_warn (context, ret, "send_diffs: krb5_data_alloc");
560c19800e8SDoug Rabson slave_dead(context, s);
561c19800e8SDoug Rabson return 1;
562c19800e8SDoug Rabson }
5638373020dSJacques Vidrine krb5_storage_read (sp, (char *)data.data + 4, data.length - 4);
564b528cefcSMark Murray krb5_storage_free(sp);
565b528cefcSMark Murray
566c19800e8SDoug Rabson sp = krb5_storage_from_data (&data);
567c19800e8SDoug Rabson if (sp == NULL) {
568c19800e8SDoug Rabson krb5_warnx (context, "send_diffs: krb5_storage_from_data");
569c19800e8SDoug Rabson slave_dead(context, s);
570c19800e8SDoug Rabson return 1;
571c19800e8SDoug Rabson }
572c19800e8SDoug Rabson krb5_store_int32 (sp, FOR_YOU);
573c19800e8SDoug Rabson krb5_storage_free(sp);
574b528cefcSMark Murray
5755e9cd1aeSAssar Westerlund ret = krb5_write_priv_message(context, s->ac, &s->fd, &data);
5768373020dSJacques Vidrine krb5_data_free(&data);
577b528cefcSMark Murray
578b528cefcSMark Murray if (ret) {
579c19800e8SDoug Rabson krb5_warn (context, ret, "send_diffs: krb5_write_priv_message");
580c19800e8SDoug Rabson slave_dead(context, s);
581b528cefcSMark Murray return 1;
582b528cefcSMark Murray }
5838373020dSJacques Vidrine slave_seen(s);
5848373020dSJacques Vidrine
585c19800e8SDoug Rabson s->version = current_version;
586c19800e8SDoug Rabson
587b528cefcSMark Murray return 0;
588b528cefcSMark Murray }
589b528cefcSMark Murray
590b528cefcSMark Murray static int
process_msg(krb5_context context,slave * s,int log_fd,const char * database,uint32_t current_version)591b528cefcSMark Murray process_msg (krb5_context context, slave *s, int log_fd,
592c19800e8SDoug Rabson const char *database, uint32_t current_version)
593b528cefcSMark Murray {
594b528cefcSMark Murray int ret = 0;
5955e9cd1aeSAssar Westerlund krb5_data out;
596b528cefcSMark Murray krb5_storage *sp;
597b528cefcSMark Murray int32_t tmp;
598b528cefcSMark Murray
5995e9cd1aeSAssar Westerlund ret = krb5_read_priv_message(context, s->ac, &s->fd, &out);
600b528cefcSMark Murray if(ret) {
6015e9cd1aeSAssar Westerlund krb5_warn (context, ret, "error reading message from %s", s->name);
602b528cefcSMark Murray return 1;
603b528cefcSMark Murray }
604b528cefcSMark Murray
605b528cefcSMark Murray sp = krb5_storage_from_mem (out.data, out.length);
606c19800e8SDoug Rabson if (sp == NULL) {
607c19800e8SDoug Rabson krb5_warnx (context, "process_msg: no memory");
608c19800e8SDoug Rabson krb5_data_free (&out);
609c19800e8SDoug Rabson return 1;
610c19800e8SDoug Rabson }
611c19800e8SDoug Rabson if (krb5_ret_int32 (sp, &tmp) != 0) {
612c19800e8SDoug Rabson krb5_warnx (context, "process_msg: client send too short command");
613c19800e8SDoug Rabson krb5_data_free (&out);
614c19800e8SDoug Rabson return 1;
615c19800e8SDoug Rabson }
616b528cefcSMark Murray switch (tmp) {
617b528cefcSMark Murray case I_HAVE :
618c19800e8SDoug Rabson ret = krb5_ret_int32 (sp, &tmp);
619c19800e8SDoug Rabson if (ret != 0) {
620c19800e8SDoug Rabson krb5_warnx (context, "process_msg: client send too I_HAVE data");
621b528cefcSMark Murray break;
622c19800e8SDoug Rabson }
623c19800e8SDoug Rabson /* new started slave that have old log */
624c19800e8SDoug Rabson if (s->version == 0 && tmp != 0) {
625ae771770SStanislav Sedov if (current_version < (uint32_t)tmp) {
626ae771770SStanislav Sedov krb5_warnx (context, "Slave %s (version %lu) have later version "
627ae771770SStanislav Sedov "the master (version %lu) OUT OF SYNC",
628ae771770SStanislav Sedov s->name, (unsigned long)tmp,
629ae771770SStanislav Sedov (unsigned long)current_version);
630ae771770SStanislav Sedov }
631c19800e8SDoug Rabson s->version = tmp;
632c19800e8SDoug Rabson }
633ae771770SStanislav Sedov if ((uint32_t)tmp < s->version) {
634c19800e8SDoug Rabson krb5_warnx (context, "Slave claims to not have "
635c19800e8SDoug Rabson "version we already sent to it");
636c19800e8SDoug Rabson } else {
637c19800e8SDoug Rabson ret = send_diffs (context, s, log_fd, database, current_version);
638c19800e8SDoug Rabson }
639c19800e8SDoug Rabson break;
640c19800e8SDoug Rabson case I_AM_HERE :
641c19800e8SDoug Rabson break;
642c19800e8SDoug Rabson case ARE_YOU_THERE:
643b528cefcSMark Murray case FOR_YOU :
644b528cefcSMark Murray default :
645b528cefcSMark Murray krb5_warnx (context, "Ignoring command %d", tmp);
646b528cefcSMark Murray break;
647b528cefcSMark Murray }
648b528cefcSMark Murray
649b528cefcSMark Murray krb5_data_free (&out);
650ae771770SStanislav Sedov krb5_storage_free (sp);
6518373020dSJacques Vidrine
6528373020dSJacques Vidrine slave_seen(s);
6538373020dSJacques Vidrine
654b528cefcSMark Murray return ret;
655b528cefcSMark Murray }
656b528cefcSMark Murray
6578373020dSJacques Vidrine #define SLAVE_NAME "Name"
6588373020dSJacques Vidrine #define SLAVE_ADDRESS "Address"
6598373020dSJacques Vidrine #define SLAVE_VERSION "Version"
6608373020dSJacques Vidrine #define SLAVE_STATUS "Status"
6618373020dSJacques Vidrine #define SLAVE_SEEN "Last Seen"
6628373020dSJacques Vidrine
663c19800e8SDoug Rabson static FILE *
open_stats(krb5_context context)664c19800e8SDoug Rabson open_stats(krb5_context context)
665c19800e8SDoug Rabson {
666c19800e8SDoug Rabson char *statfile = NULL;
667c19800e8SDoug Rabson const char *fn;
668c19800e8SDoug Rabson FILE *f;
669c19800e8SDoug Rabson
670c19800e8SDoug Rabson if (slave_stats_file)
671c19800e8SDoug Rabson fn = slave_stats_file;
672c19800e8SDoug Rabson else {
673c19800e8SDoug Rabson asprintf(&statfile, "%s/slaves-stats", hdb_db_dir(context));
674c19800e8SDoug Rabson fn = krb5_config_get_string_default(context,
675c19800e8SDoug Rabson NULL,
676c19800e8SDoug Rabson statfile,
677c19800e8SDoug Rabson "kdc",
678c19800e8SDoug Rabson "iprop-stats",
679c19800e8SDoug Rabson NULL);
680c19800e8SDoug Rabson }
681c19800e8SDoug Rabson f = fopen(fn, "w");
682c19800e8SDoug Rabson if (statfile)
683c19800e8SDoug Rabson free(statfile);
684c19800e8SDoug Rabson
685c19800e8SDoug Rabson return f;
686c19800e8SDoug Rabson }
687c19800e8SDoug Rabson
6888373020dSJacques Vidrine static void
write_master_down(krb5_context context)689c19800e8SDoug Rabson write_master_down(krb5_context context)
690c19800e8SDoug Rabson {
691c19800e8SDoug Rabson char str[100];
692c19800e8SDoug Rabson time_t t = time(NULL);
693c19800e8SDoug Rabson FILE *fp;
694c19800e8SDoug Rabson
695c19800e8SDoug Rabson fp = open_stats(context);
696c19800e8SDoug Rabson if (fp == NULL)
697c19800e8SDoug Rabson return;
698c19800e8SDoug Rabson krb5_format_time(context, t, str, sizeof(str), TRUE);
699c19800e8SDoug Rabson fprintf(fp, "master down at %s\n", str);
700c19800e8SDoug Rabson
701c19800e8SDoug Rabson fclose(fp);
702c19800e8SDoug Rabson }
703c19800e8SDoug Rabson
704c19800e8SDoug Rabson static void
write_stats(krb5_context context,slave * slaves,uint32_t current_version)705c19800e8SDoug Rabson write_stats(krb5_context context, slave *slaves, uint32_t current_version)
7068373020dSJacques Vidrine {
707bbd80c28SJacques Vidrine char str[100];
7088373020dSJacques Vidrine rtbl_t tbl;
7098373020dSJacques Vidrine time_t t = time(NULL);
7108373020dSJacques Vidrine FILE *fp;
7118373020dSJacques Vidrine
712c19800e8SDoug Rabson fp = open_stats(context);
7138373020dSJacques Vidrine if (fp == NULL)
7148373020dSJacques Vidrine return;
7158373020dSJacques Vidrine
716c19800e8SDoug Rabson krb5_format_time(context, t, str, sizeof(str), TRUE);
7178373020dSJacques Vidrine fprintf(fp, "Status for slaves, last updated: %s\n\n", str);
7188373020dSJacques Vidrine
7198373020dSJacques Vidrine fprintf(fp, "Master version: %lu\n\n", (unsigned long)current_version);
7208373020dSJacques Vidrine
7218373020dSJacques Vidrine tbl = rtbl_create();
7228373020dSJacques Vidrine if (tbl == NULL) {
7238373020dSJacques Vidrine fclose(fp);
7248373020dSJacques Vidrine return;
7258373020dSJacques Vidrine }
7268373020dSJacques Vidrine
7278373020dSJacques Vidrine rtbl_add_column(tbl, SLAVE_NAME, 0);
7288373020dSJacques Vidrine rtbl_add_column(tbl, SLAVE_ADDRESS, 0);
7298373020dSJacques Vidrine rtbl_add_column(tbl, SLAVE_VERSION, RTBL_ALIGN_RIGHT);
7308373020dSJacques Vidrine rtbl_add_column(tbl, SLAVE_STATUS, 0);
7318373020dSJacques Vidrine rtbl_add_column(tbl, SLAVE_SEEN, 0);
7328373020dSJacques Vidrine
7338373020dSJacques Vidrine rtbl_set_prefix(tbl, " ");
7348373020dSJacques Vidrine rtbl_set_column_prefix(tbl, SLAVE_NAME, "");
7358373020dSJacques Vidrine
7368373020dSJacques Vidrine while (slaves) {
7378373020dSJacques Vidrine krb5_address addr;
7388373020dSJacques Vidrine krb5_error_code ret;
7398373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_NAME, slaves->name);
7408373020dSJacques Vidrine ret = krb5_sockaddr2address (context,
7419c91c227SBjoern A. Zeeb (struct sockaddr*)&slaves->addr.sa, &addr);
7428373020dSJacques Vidrine if(ret == 0) {
7438373020dSJacques Vidrine krb5_print_address(&addr, str, sizeof(str), NULL);
7448373020dSJacques Vidrine krb5_free_address(context, &addr);
7458373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_ADDRESS, str);
7468373020dSJacques Vidrine } else
7478373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_ADDRESS, "<unknown>");
7488373020dSJacques Vidrine
7498373020dSJacques Vidrine snprintf(str, sizeof(str), "%u", (unsigned)slaves->version);
7508373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_VERSION, str);
7518373020dSJacques Vidrine
7528373020dSJacques Vidrine if (slaves->flags & SLAVE_F_DEAD)
7538373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_STATUS, "Down");
7548373020dSJacques Vidrine else
7558373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_STATUS, "Up");
7568373020dSJacques Vidrine
757c19800e8SDoug Rabson ret = krb5_format_time(context, slaves->seen, str, sizeof(str), TRUE);
758*ed549cb0SCy Schubert if (ret)
759*ed549cb0SCy Schubert rtbl_add_column_entry(tbl, SLAVE_SEEN, "<error-formatting-time>");
760*ed549cb0SCy Schubert else
7618373020dSJacques Vidrine rtbl_add_column_entry(tbl, SLAVE_SEEN, str);
7628373020dSJacques Vidrine
7638373020dSJacques Vidrine slaves = slaves->next;
7648373020dSJacques Vidrine }
7658373020dSJacques Vidrine
7668373020dSJacques Vidrine rtbl_format(tbl, fp);
7678373020dSJacques Vidrine rtbl_destroy(tbl);
7688373020dSJacques Vidrine
7698373020dSJacques Vidrine fclose(fp);
7708373020dSJacques Vidrine }
7718373020dSJacques Vidrine
7728373020dSJacques Vidrine
773ae771770SStanislav Sedov static char sHDB[] = "HDB:";
7745e9cd1aeSAssar Westerlund static char *realm;
7755e9cd1aeSAssar Westerlund static int version_flag;
7765e9cd1aeSAssar Westerlund static int help_flag;
777ae771770SStanislav Sedov static char *keytab_str = sHDB;
7785e9cd1aeSAssar Westerlund static char *database;
779c19800e8SDoug Rabson static char *config_file;
780c19800e8SDoug Rabson static char *port_str;
781ae771770SStanislav Sedov #ifdef SUPPORT_DETACH
782c19800e8SDoug Rabson static int detach_from_console = 0;
783ae771770SStanislav Sedov #endif
7845e9cd1aeSAssar Westerlund
7855e9cd1aeSAssar Westerlund static struct getargs args[] = {
786ae771770SStanislav Sedov { "config-file", 'c', arg_string, &config_file, NULL, NULL },
787ae771770SStanislav Sedov { "realm", 'r', arg_string, &realm, NULL, NULL },
7885e9cd1aeSAssar Westerlund { "keytab", 'k', arg_string, &keytab_str,
7895e9cd1aeSAssar Westerlund "keytab to get authentication from", "kspec" },
7905e9cd1aeSAssar Westerlund { "database", 'd', arg_string, &database, "database", "file"},
791ae771770SStanislav Sedov { "slave-stats-file", 0, arg_string, rk_UNCONST(&slave_stats_file),
792c19800e8SDoug Rabson "file for slave status information", "file"},
793ae771770SStanislav Sedov { "time-missing", 0, arg_string, rk_UNCONST(&slave_time_missing),
794c19800e8SDoug Rabson "time before slave is polled for presence", "time"},
795ae771770SStanislav Sedov { "time-gone", 0, arg_string, rk_UNCONST(&slave_time_gone),
796c19800e8SDoug Rabson "time of inactivity after which a slave is considered gone", "time"},
797c19800e8SDoug Rabson { "port", 0, arg_string, &port_str,
798c19800e8SDoug Rabson "port ipropd will listen to", "port"},
799ae771770SStanislav Sedov #ifdef SUPPORT_DETACH
800c19800e8SDoug Rabson { "detach", 0, arg_flag, &detach_from_console,
801ae771770SStanislav Sedov "detach from console", NULL },
802ae771770SStanislav Sedov #endif
803ae771770SStanislav Sedov { "hostname", 0, arg_string, rk_UNCONST(&master_hostname),
804c19800e8SDoug Rabson "hostname of master (if not same as hostname)", "hostname" },
805ae771770SStanislav Sedov { "version", 0, arg_flag, &version_flag, NULL, NULL },
806ae771770SStanislav Sedov { "help", 0, arg_flag, &help_flag, NULL, NULL }
807b528cefcSMark Murray };
8085e9cd1aeSAssar Westerlund static int num_args = sizeof(args) / sizeof(args[0]);
809b528cefcSMark Murray
810b528cefcSMark Murray int
main(int argc,char ** argv)811b528cefcSMark Murray main(int argc, char **argv)
812b528cefcSMark Murray {
813b528cefcSMark Murray krb5_error_code ret;
814b528cefcSMark Murray krb5_context context;
815b528cefcSMark Murray void *kadm_handle;
816b528cefcSMark Murray kadm5_server_context *server_context;
817b528cefcSMark Murray kadm5_config_params conf;
8189c91c227SBjoern A. Zeeb krb5_socket_t signal_fd, listen_fd, listen6_fd;
819b528cefcSMark Murray int log_fd;
820b528cefcSMark Murray slave *slaves = NULL;
821c19800e8SDoug Rabson uint32_t current_version = 0, old_version = 0;
8225e9cd1aeSAssar Westerlund krb5_keytab keytab;
823c19800e8SDoug Rabson int optidx;
824c19800e8SDoug Rabson char **files;
825b528cefcSMark Murray
826c19800e8SDoug Rabson optidx = krb5_program_setup(&context, argc, argv, args, num_args, NULL);
827b528cefcSMark Murray
828b528cefcSMark Murray if(help_flag)
829b528cefcSMark Murray krb5_std_usage(0, args, num_args);
830b528cefcSMark Murray if(version_flag) {
831b528cefcSMark Murray print_version(NULL);
832b528cefcSMark Murray exit(0);
833b528cefcSMark Murray }
834b528cefcSMark Murray
835c19800e8SDoug Rabson setup_signal();
836c19800e8SDoug Rabson
837c19800e8SDoug Rabson if (config_file == NULL) {
838c19800e8SDoug Rabson asprintf(&config_file, "%s/kdc.conf", hdb_db_dir(context));
839c19800e8SDoug Rabson if (config_file == NULL)
840c19800e8SDoug Rabson errx(1, "out of memory");
841c19800e8SDoug Rabson }
842c19800e8SDoug Rabson
843c19800e8SDoug Rabson ret = krb5_prepend_config_files_default(config_file, &files);
844c19800e8SDoug Rabson if (ret)
845c19800e8SDoug Rabson krb5_err(context, 1, ret, "getting configuration files");
846c19800e8SDoug Rabson
847c19800e8SDoug Rabson ret = krb5_set_config_files(context, files);
848c19800e8SDoug Rabson krb5_free_config_files(files);
849c19800e8SDoug Rabson if (ret)
850c19800e8SDoug Rabson krb5_err(context, 1, ret, "reading configuration files");
851c19800e8SDoug Rabson
852c19800e8SDoug Rabson time_before_gone = parse_time (slave_time_gone, "s");
853c19800e8SDoug Rabson if (time_before_gone < 0)
854c19800e8SDoug Rabson krb5_errx (context, 1, "couldn't parse time: %s", slave_time_gone);
855c19800e8SDoug Rabson time_before_missing = parse_time (slave_time_missing, "s");
856c19800e8SDoug Rabson if (time_before_missing < 0)
857c19800e8SDoug Rabson krb5_errx (context, 1, "couldn't parse time: %s", slave_time_missing);
858c19800e8SDoug Rabson
859ae771770SStanislav Sedov #ifdef SUPPORT_DETACH
860c19800e8SDoug Rabson if (detach_from_console)
861c19800e8SDoug Rabson daemon(0, 0);
862ae771770SStanislav Sedov #endif
8634137ff4cSJacques Vidrine pidfile (NULL);
8645e9cd1aeSAssar Westerlund krb5_openlog (context, "ipropd-master", &log_facility);
8655e9cd1aeSAssar Westerlund krb5_set_warn_dest(context, log_facility);
8665e9cd1aeSAssar Westerlund
8675e9cd1aeSAssar Westerlund ret = krb5_kt_register(context, &hdb_kt_ops);
8685e9cd1aeSAssar Westerlund if(ret)
8695e9cd1aeSAssar Westerlund krb5_err(context, 1, ret, "krb5_kt_register");
8705e9cd1aeSAssar Westerlund
8715e9cd1aeSAssar Westerlund ret = krb5_kt_resolve(context, keytab_str, &keytab);
8725e9cd1aeSAssar Westerlund if(ret)
8735e9cd1aeSAssar Westerlund krb5_err(context, 1, ret, "krb5_kt_resolve: %s", keytab_str);
8745e9cd1aeSAssar Westerlund
875b528cefcSMark Murray memset(&conf, 0, sizeof(conf));
876b528cefcSMark Murray if(realm) {
877b528cefcSMark Murray conf.mask |= KADM5_CONFIG_REALM;
878b528cefcSMark Murray conf.realm = realm;
879b528cefcSMark Murray }
8805e9cd1aeSAssar Westerlund ret = kadm5_init_with_skey_ctx (context,
881b528cefcSMark Murray KADM5_ADMIN_SERVICE,
882b528cefcSMark Murray NULL,
883b528cefcSMark Murray KADM5_ADMIN_SERVICE,
884b528cefcSMark Murray &conf, 0, 0,
885b528cefcSMark Murray &kadm_handle);
886b528cefcSMark Murray if (ret)
887b528cefcSMark Murray krb5_err (context, 1, ret, "kadm5_init_with_password_ctx");
888b528cefcSMark Murray
889b528cefcSMark Murray server_context = (kadm5_server_context *)kadm_handle;
890b528cefcSMark Murray
891b528cefcSMark Murray log_fd = open (server_context->log_context.log_file, O_RDONLY, 0);
892b528cefcSMark Murray if (log_fd < 0)
893b528cefcSMark Murray krb5_err (context, 1, errno, "open %s",
894b528cefcSMark Murray server_context->log_context.log_file);
895b528cefcSMark Murray
896b528cefcSMark Murray signal_fd = make_signal_socket (context);
897c19800e8SDoug Rabson listen_fd = make_listen_socket (context, port_str);
8989c91c227SBjoern A. Zeeb listen6_fd = make_listen6_socket (context, port_str);
899b528cefcSMark Murray
900c19800e8SDoug Rabson kadm5_log_get_version_fd (log_fd, ¤t_version);
9015e9cd1aeSAssar Westerlund
902c19800e8SDoug Rabson krb5_warnx(context, "ipropd-master started at version: %lu",
903c19800e8SDoug Rabson (unsigned long)current_version);
904c19800e8SDoug Rabson
905c19800e8SDoug Rabson while(exit_flag == 0){
906b528cefcSMark Murray slave *p;
907b528cefcSMark Murray fd_set readset;
908b528cefcSMark Murray int max_fd = 0;
909b528cefcSMark Murray struct timeval to = {30, 0};
910c19800e8SDoug Rabson uint32_t vers;
911b528cefcSMark Murray
912ae771770SStanislav Sedov #ifndef NO_LIMIT_FD_SETSIZE
9139c91c227SBjoern A. Zeeb if (signal_fd >= FD_SETSIZE || listen_fd >= FD_SETSIZE ||
9149c91c227SBjoern A. Zeeb listen6_fd >= FD_SETSIZE)
9155e9cd1aeSAssar Westerlund krb5_errx (context, 1, "fd too large");
916ae771770SStanislav Sedov #endif
9175e9cd1aeSAssar Westerlund
918b528cefcSMark Murray FD_ZERO(&readset);
919b528cefcSMark Murray FD_SET(signal_fd, &readset);
920b528cefcSMark Murray max_fd = max(max_fd, signal_fd);
921b528cefcSMark Murray FD_SET(listen_fd, &readset);
922b528cefcSMark Murray max_fd = max(max_fd, listen_fd);
9239c91c227SBjoern A. Zeeb FD_SET(listen6_fd, &readset);
9249c91c227SBjoern A. Zeeb max_fd = max(max_fd, listen6_fd);
925b528cefcSMark Murray
926b528cefcSMark Murray for (p = slaves; p != NULL; p = p->next) {
927bbd80c28SJacques Vidrine if (p->flags & SLAVE_F_DEAD)
928bbd80c28SJacques Vidrine continue;
929b528cefcSMark Murray FD_SET(p->fd, &readset);
930b528cefcSMark Murray max_fd = max(max_fd, p->fd);
931b528cefcSMark Murray }
932b528cefcSMark Murray
933b528cefcSMark Murray ret = select (max_fd + 1,
934b528cefcSMark Murray &readset, NULL, NULL, &to);
935b528cefcSMark Murray if (ret < 0) {
936b528cefcSMark Murray if (errno == EINTR)
937b528cefcSMark Murray continue;
938b528cefcSMark Murray else
939b528cefcSMark Murray krb5_err (context, 1, errno, "select");
940b528cefcSMark Murray }
941b528cefcSMark Murray
942b528cefcSMark Murray if (ret == 0) {
943b528cefcSMark Murray old_version = current_version;
9445e9cd1aeSAssar Westerlund kadm5_log_get_version_fd (log_fd, ¤t_version);
945b528cefcSMark Murray
946c19800e8SDoug Rabson if (current_version > old_version) {
947c19800e8SDoug Rabson krb5_warnx(context,
948c19800e8SDoug Rabson "Missed a signal, updating slaves %lu to %lu",
949c19800e8SDoug Rabson (unsigned long)old_version,
950c19800e8SDoug Rabson (unsigned long)current_version);
951bbd80c28SJacques Vidrine for (p = slaves; p != NULL; p = p->next) {
952bbd80c28SJacques Vidrine if (p->flags & SLAVE_F_DEAD)
953bbd80c28SJacques Vidrine continue;
9545e9cd1aeSAssar Westerlund send_diffs (context, p, log_fd, database, current_version);
955b528cefcSMark Murray }
956bbd80c28SJacques Vidrine }
957c19800e8SDoug Rabson }
958b528cefcSMark Murray
959b528cefcSMark Murray if (ret && FD_ISSET(signal_fd, &readset)) {
960ae771770SStanislav Sedov #ifndef NO_UNIX_SOCKETS
961b528cefcSMark Murray struct sockaddr_un peer_addr;
962ae771770SStanislav Sedov #else
963ae771770SStanislav Sedov struct sockaddr_storage peer_addr;
964ae771770SStanislav Sedov #endif
9655e9cd1aeSAssar Westerlund socklen_t peer_len = sizeof(peer_addr);
966b528cefcSMark Murray
9674137ff4cSJacques Vidrine if(recvfrom(signal_fd, (void *)&vers, sizeof(vers), 0,
968b528cefcSMark Murray (struct sockaddr *)&peer_addr, &peer_len) < 0) {
969b528cefcSMark Murray krb5_warn (context, errno, "recvfrom");
970b528cefcSMark Murray continue;
971b528cefcSMark Murray }
972b528cefcSMark Murray --ret;
973c19800e8SDoug Rabson assert(ret >= 0);
974b528cefcSMark Murray old_version = current_version;
9755e9cd1aeSAssar Westerlund kadm5_log_get_version_fd (log_fd, ¤t_version);
976c19800e8SDoug Rabson if (current_version > old_version) {
977c19800e8SDoug Rabson krb5_warnx(context,
978c19800e8SDoug Rabson "Got a signal, updating slaves %lu to %lu",
979c19800e8SDoug Rabson (unsigned long)old_version,
980c19800e8SDoug Rabson (unsigned long)current_version);
981ae771770SStanislav Sedov for (p = slaves; p != NULL; p = p->next) {
982ae771770SStanislav Sedov if (p->flags & SLAVE_F_DEAD)
983ae771770SStanislav Sedov continue;
9845e9cd1aeSAssar Westerlund send_diffs (context, p, log_fd, database, current_version);
985ae771770SStanislav Sedov }
986c19800e8SDoug Rabson } else {
987c19800e8SDoug Rabson krb5_warnx(context,
988c19800e8SDoug Rabson "Got a signal, but no update in log version %lu",
989c19800e8SDoug Rabson (unsigned long)current_version);
990c19800e8SDoug Rabson }
991b528cefcSMark Murray }
992b528cefcSMark Murray
993c19800e8SDoug Rabson for(p = slaves; p != NULL; p = p->next) {
994bbd80c28SJacques Vidrine if (p->flags & SLAVE_F_DEAD)
995bbd80c28SJacques Vidrine continue;
996c19800e8SDoug Rabson if (ret && FD_ISSET(p->fd, &readset)) {
997adb0ddaeSAssar Westerlund --ret;
998c19800e8SDoug Rabson assert(ret >= 0);
9995e9cd1aeSAssar Westerlund if(process_msg (context, p, log_fd, database, current_version))
1000c19800e8SDoug Rabson slave_dead(context, p);
1001c19800e8SDoug Rabson } else if (slave_gone_p (p))
1002c19800e8SDoug Rabson slave_dead(context, p);
1003ae771770SStanislav Sedov else if (slave_missing_p (p))
1004c19800e8SDoug Rabson send_are_you_there (context, p);
1005b528cefcSMark Murray }
1006b528cefcSMark Murray
10079c91c227SBjoern A. Zeeb if (ret && FD_ISSET(listen6_fd, &readset)) {
10089c91c227SBjoern A. Zeeb add_slave (context, keytab, &slaves, listen6_fd);
10099c91c227SBjoern A. Zeeb --ret;
10109c91c227SBjoern A. Zeeb assert(ret >= 0);
10119c91c227SBjoern A. Zeeb }
1012b528cefcSMark Murray if (ret && FD_ISSET(listen_fd, &readset)) {
10135e9cd1aeSAssar Westerlund add_slave (context, keytab, &slaves, listen_fd);
1014b528cefcSMark Murray --ret;
1015c19800e8SDoug Rabson assert(ret >= 0);
1016b528cefcSMark Murray }
10178373020dSJacques Vidrine write_stats(context, slaves, current_version);
1018b528cefcSMark Murray }
1019b528cefcSMark Murray
1020ae771770SStanislav Sedov if(exit_flag == SIGINT || exit_flag == SIGTERM)
1021c19800e8SDoug Rabson krb5_warnx(context, "%s terminated", getprogname());
1022ae771770SStanislav Sedov #ifdef SIGXCPU
1023ae771770SStanislav Sedov else if(exit_flag == SIGXCPU)
1024ae771770SStanislav Sedov krb5_warnx(context, "%s CPU time limit exceeded", getprogname());
1025ae771770SStanislav Sedov #endif
1026c19800e8SDoug Rabson else
1027ae771770SStanislav Sedov krb5_warnx(context, "%s unexpected exit reason: %ld",
1028ae771770SStanislav Sedov getprogname(), (long)exit_flag);
1029c19800e8SDoug Rabson
1030c19800e8SDoug Rabson write_master_down(context);
1031c19800e8SDoug Rabson
1032b528cefcSMark Murray return 0;
1033b528cefcSMark Murray }
1034