New configuration option "NoZeroConf" to disable ZeroConf registration
[ngircd-alex.git] / src / ngircd / rendezvous.c
1 /*
2  * ngIRCd -- The Next Generation IRC Daemon
3  * Copyright (c)2001-2004 by Alexander Barton (alex@barton.de)
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  * Please read the file COPYING, README and AUTHORS for more information.
10  *
11  * Rendezvous service registration.
12  *
13  * Supported APIs are:
14  *  - Apple Mac OS X
15  *  - Howl
16  */
17
18
19 #include "portab.h"
20
21 #ifdef ZEROCONF
22
23
24 static char UNUSED id[] = "$Id: rendezvous.c,v 1.8 2006/05/10 21:24:01 alex Exp $";
25
26 #include "imp.h"
27 #include <assert.h>
28
29 #include <stdio.h>
30 #include <string.h>
31
32 #ifdef HAVE_MACH_PORT_H
33 #include "mach/port.h"
34 #include "mach/message.h"
35 #endif
36
37 #ifdef HAVE_DNSSERVICEDISCOVERY_DNSSERVICEDISCOVERY_H
38 #include <DNSServiceDiscovery/DNSServiceDiscovery.h>
39 #endif
40
41 #ifdef HAVE_RENDEZVOUS_RENDEZVOUS_H
42 #include <rendezvous/rendezvous.h>
43 #endif
44
45 #include "defines.h"
46 #include "log.h"
47
48 #include "exp.h"
49 #include "rendezvous.h"
50
51
52 #if defined(HAVE_DNSSERVICEREGISTRATIONCREATE)
53 #       define APPLE
54 #elif defined(HAVE_SW_DISCOVERY_INIT)
55 #       define HOWL
56 #else
57 #       error "Can't detect Rendezvous API!?"
58 #endif
59
60
61 #define MAX_RENDEZVOUS 1000
62
63 typedef struct _service
64 {
65         char Desc[CLIENT_ID_LEN];
66 #ifdef APPLE
67         dns_service_discovery_ref Discovery_Ref;
68         mach_port_t Mach_Port;
69 #endif
70 #ifdef HOWL
71         sw_discovery_oid Id;
72 #endif
73 } SERVICE;
74
75 static SERVICE My_Rendezvous[MAX_RENDEZVOUS];
76
77
78 static void Unregister( int Idx );
79
80
81 /* -- Apple API -- */
82
83 #ifdef APPLE
84
85 #define MAX_MACH_MSG_SIZE 512
86
87 static void Registration_Reply_Handler( DNSServiceRegistrationReplyErrorType ErrCode, void *Context );
88
89 #endif /* Apple */
90
91
92 /* -- Howl API -- */
93
94 #ifdef HOWL
95
96 static sw_discovery My_Discovery_Session = NULL;
97 static sw_salt My_Salt;
98
99 static sw_result HOWL_API Registration_Reply_Handler( sw_discovery Session, sw_discovery_publish_status Status, sw_discovery_oid Id, sw_opaque Extra );
100
101 #endif /* Howl */
102
103
104 GLOBAL void Rendezvous_Init( void )
105 {
106         /* Initialize structures */
107
108         int i;
109
110 #ifdef HOWL
111         if( sw_discovery_init( &My_Discovery_Session ) != SW_OKAY )
112         {
113                 Log( LOG_EMERG, "Can't initialize Rendezvous (Howl): sw_discovery_init() failed!" );
114                 Log( LOG_ALERT, "%s exiting due to fatal errors!", PACKAGE_NAME );
115                 exit( 1 );
116         }
117
118         if( sw_discovery_salt( My_Discovery_Session, &My_Salt ) != SW_OKAY )
119         {
120                 Log( LOG_EMERG, "Can't initialize Rendezvous (Howl): sw_discovery_salt() failed!" );
121                 Log( LOG_ALERT, "%s exiting due to fatal errors!", PACKAGE_NAME );
122                 exit( 1 );
123         }
124 #endif
125
126         for( i = 0; i < MAX_RENDEZVOUS; i++ ) My_Rendezvous[i].Desc[0] = '\0';
127 } /* Rendezvous_Init */
128
129
130 GLOBAL void Rendezvous_Exit( void )
131 {
132         /* Clean up & exit module */
133
134         int i;
135
136         for( i = 0; i < MAX_RENDEZVOUS; i++ )
137         {
138                 if( My_Rendezvous[i].Desc[0] ) Unregister( i );
139         }
140
141 #ifdef HOWL
142         sw_discovery_fina( My_Discovery_Session );
143 #endif
144 } /* Rendezvous_Exit */
145
146
147 /**
148  * Register ZeroConf service
149  */
150 GLOBAL bool Rendezvous_Register( char *Name, char *Type, UINT16 Port )
151 {
152         int i;
153
154         if (Conf_NoZeroConf)
155                 return;
156
157         /* Search free port structure */
158         for( i = 0; i < MAX_RENDEZVOUS; i++ ) if( ! My_Rendezvous[i].Desc[0] ) break;
159         if( i >= MAX_RENDEZVOUS )
160         {
161                 Log( LOG_ERR, "Can't register \"%s\" with Rendezvous: limit (%d) reached!", Name, MAX_RENDEZVOUS );
162                 return false;
163         }
164         strlcpy( My_Rendezvous[i].Desc, Name, sizeof( My_Rendezvous[i].Desc ));
165         
166 #ifdef APPLE
167         /* Register new service */
168         My_Rendezvous[i].Discovery_Ref = DNSServiceRegistrationCreate( Name, Type, "", htonl( Port ), "", Registration_Reply_Handler, &My_Rendezvous[i] );
169         if( ! My_Rendezvous[i].Discovery_Ref )
170         {
171                 Log( LOG_ERR, "Can't register \"%s\" with Rendezvous: can't register service!", My_Rendezvous[i].Desc );
172                 My_Rendezvous[i].Desc[0] = '\0';
173                 return false;
174         }
175         
176         /* Get and save the corresponding Mach Port */
177         My_Rendezvous[i].Mach_Port = DNSServiceDiscoveryMachPort( My_Rendezvous[i].Discovery_Ref );
178         if( ! My_Rendezvous[i].Mach_Port )
179         {
180                 Log( LOG_ERR, "Can't register \"%s\" with Rendezvous: got no Mach Port!", My_Rendezvous[i].Desc );
181                 /* Here we actually leek a descriptor :-( */
182                 My_Rendezvous[i].Discovery_Ref = 0;
183                 My_Rendezvous[i].Desc[0] = '\0';
184                 return false;
185         }
186 #endif /* Apple */
187
188 #ifdef HOWL
189         if( sw_discovery_publish( My_Discovery_Session, 0, Name, Type, NULL, NULL, Port, NULL, 0, Registration_Reply_Handler, &My_Rendezvous[i], &My_Rendezvous[i].Id ) != SW_OKAY )
190         {
191                 Log( LOG_ERR, "Can't register \"%s\" with Rendezvous: can't register service!", My_Rendezvous[i].Desc );
192                 My_Rendezvous[i].Desc[0] = '\0';
193                 return false;
194         }
195 #endif /* Howl */
196
197         Log( LOG_DEBUG, "Rendezvous: Registering \"%s\" ...", My_Rendezvous[i].Desc );
198         return true;
199 } /* Rendezvous_Register */
200
201
202 GLOBAL bool Rendezvous_Unregister( char *Name )
203 {
204         /* Unregister service from rendezvous */
205
206         int i;
207         bool ok;
208
209         ok = false;
210         for( i = 0; i < MAX_RENDEZVOUS; i++ )
211         {
212                 if( strcmp( Name, My_Rendezvous[i].Desc ) == 0 )
213                 {
214                         Unregister( i );
215                         ok = true;
216                 }
217         }
218
219         return ok;
220 } /* Rendezvous_Unregister */
221
222
223 GLOBAL void Rendezvous_UnregisterListeners( void )
224 {
225         /* Unregister all our listening sockets from Rendezvous */
226
227         int i;
228
229         for( i = 0; i < MAX_RENDEZVOUS; i++ )
230         {
231                 if( My_Rendezvous[i].Desc[0] ) Unregister( i );
232         }
233 } /* Rendezvous_UnregisterListeners */
234
235
236 GLOBAL void Rendezvous_Handler( void )
237 {
238         /* Handle all Rendezvous stuff; this function must be called
239          * periodically from the run loop of the main program */
240
241 #ifdef APPLE
242         int i;
243         char buffer[MAX_MACH_MSG_SIZE];
244         mach_msg_return_t result;
245         mach_msg_header_t *msg;
246
247         for( i = 0; i < MAX_RENDEZVOUS; i++ )
248         {
249                 if( ! My_Rendezvous[i].Discovery_Ref ) continue;
250
251                 /* Read message from Mach Port */
252                 msg = (mach_msg_header_t *)buffer;
253                 result = mach_msg( msg, MACH_RCV_MSG|MACH_RCV_TIMEOUT, 0, MAX_MACH_MSG_SIZE, My_Rendezvous[i].Mach_Port, 1, 0 );
254
255                 /* Handle message */
256                 if( result == MACH_MSG_SUCCESS ) DNSServiceDiscovery_handleReply( msg );
257 #ifdef DEBUG
258                 else if( result != MACH_RCV_TIMED_OUT ) Log( LOG_DEBUG, "mach_msg(): %ld", (long)result );
259 #endif /* Debug */
260         }
261 #endif /* Apple */
262
263 #ifdef HOWL
264         sw_ulong msecs = 10;
265         sw_salt_step( My_Salt, &msecs );
266 #endif
267 } /* Rendezvous_Handler */
268
269
270 static void Unregister( int Idx )
271 {
272         /* Unregister service */
273
274 #ifdef APPLE
275         DNSServiceDiscoveryDeallocate( My_Rendezvous[Idx].Discovery_Ref );
276 #endif /* Apple */
277
278 #ifdef HOWL
279         if( sw_discovery_cancel( My_Discovery_Session, My_Rendezvous[Idx].Id ) != SW_OKAY )
280         {
281                 Log( LOG_ERR, "Rendezvous: Failed to unregister \"%s\"!", My_Rendezvous[Idx].Desc );
282                 return;
283         }
284 #endif /* Howl */
285         
286         Log( LOG_INFO, "Unregistered \"%s\" from Rendezvous.", My_Rendezvous[Idx].Desc );
287         My_Rendezvous[Idx].Desc[0] = '\0';
288 } /* Unregister */
289
290
291 /* -- Apple API -- */
292
293 #ifdef APPLE
294
295
296 static void Registration_Reply_Handler( DNSServiceRegistrationReplyErrorType ErrCode, void *Context )
297 {
298         SERVICE *s = (SERVICE *)Context;
299         char txt[50];
300
301         if( ErrCode == kDNSServiceDiscoveryNoError )
302         {
303                 /* Success! */
304                 Log( LOG_INFO, "Successfully registered \"%s\" with Rendezvous.", s->Desc );
305                 return;
306         }
307
308         switch( ErrCode )
309         {
310                 case kDNSServiceDiscoveryAlreadyRegistered:
311                         strcpy( txt, "name already registered!" );
312                         break;
313                 case kDNSServiceDiscoveryNameConflict:
314                         strcpy( txt, "name conflict!" );
315                         break;
316                 default:
317                         snprintf(txt, sizeof txt, "error code %ld!",
318                                  (long)ErrCode);
319         }
320
321         Log( LOG_INFO, "Can't register \"%s\" with Rendezvous: %s", s->Desc, txt );
322         s->Desc[0] = '\0';
323 } /* Registration_Reply_Handler */
324
325
326 #endif /* Apple */
327
328
329 /* -- Howl API -- */
330
331 #ifdef HOWL
332
333
334 static sw_result HOWL_API Registration_Reply_Handler( sw_discovery Session, sw_discovery_publish_status Status, UNUSED sw_discovery_oid Id, sw_opaque Extra )
335 {
336         SERVICE *s = (SERVICE *)Extra;
337         char txt[50];
338
339         assert( Session == My_Discovery_Session );
340         assert( Extra != NULL );
341
342         if( Status == SW_DISCOVERY_PUBLISH_STARTED || Status == SW_DISCOVERY_PUBLISH_STOPPED )
343         {
344                 /* Success! */
345                 Log( LOG_INFO, "Successfully registered \"%s\" with Rendezvous.", s->Desc );
346                 return SW_OKAY;
347         }
348                 
349         switch( Status )
350         {
351                 case SW_DISCOVERY_PUBLISH_NAME_COLLISION:
352                         strcpy( txt, "name conflict!" );
353                         break;
354                 default:
355                         snprintf(txt, sizeof txt, "error code %ld!",
356                                  (long)Status);
357         }
358
359         Log( LOG_INFO, "Can't register \"%s\" with Rendezvous: %s", s->Desc, txt );
360         s->Desc[0] = '\0';
361
362         return SW_OKAY;
363 } /* Registration_Reply_Handler */
364
365
366 #endif /* Howl */
367
368
369 #endif  /* ZEROCONF */
370
371
372 /* -eof- */