Fixed vc1 packetizer.
[vlc.git] / modules / access_output / bonjour.c
1 /*****************************************************************************
2  * bonjour.c
3  *****************************************************************************
4  * Copyright (C) 2005 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Jon Lech Johansen <jon@nanocrew.net>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <vlc_common.h>
33 #include "bonjour.h"
34
35 #ifdef HAVE_AVAHI_CLIENT
36 #include <vlc_sout.h>
37
38 #include <avahi-client/client.h>
39 #include <avahi-client/publish.h>
40 #include <avahi-client/lookup.h>
41 #include <avahi-common/alternative.h>
42 #include <avahi-common/simple-watch.h>
43 #include <avahi-common/malloc.h>
44 #include <avahi-common/error.h>
45
46 /*****************************************************************************
47  * Structures
48  *****************************************************************************/
49 typedef struct poll_thread_t
50 {
51     VLC_COMMON_MEMBERS
52
53     AvahiSimplePoll     *simple_poll;
54 } poll_thread_t;
55
56 typedef struct bonjour_t
57 {
58     vlc_object_t        *p_log;
59
60     poll_thread_t       *poll_thread;
61     AvahiSimplePoll     *simple_poll;
62     AvahiEntryGroup     *group;
63     AvahiClient         *client;
64     char                *psz_stype;
65     char                *psz_name;
66     int                 i_port;
67     char                *psz_txt;
68 } bonjour_t;
69
70 /*****************************************************************************
71  * Prototypes
72  *****************************************************************************/
73 static int create_service( bonjour_t * );
74
75 /*****************************************************************************
76  * entry_group_callback
77  *****************************************************************************/
78 static void entry_group_callback( AvahiEntryGroup *g,
79                                   AvahiEntryGroupState state,
80                                   void *userdata )
81 {
82     (void)g;
83     bonjour_t *p_sys = (bonjour_t *)userdata;
84
85     if( state == AVAHI_ENTRY_GROUP_ESTABLISHED )
86     {
87         msg_Dbg( p_sys->p_log, "service '%s' successfully established",
88                  p_sys->psz_name );
89     }
90     else if( state == AVAHI_ENTRY_GROUP_COLLISION )
91     {
92         char *n;
93
94         n = avahi_alternative_service_name( p_sys->psz_name );
95         avahi_free( p_sys->psz_name );
96         p_sys->psz_name = n;
97
98         create_service( p_sys );
99     }
100 }
101
102 /*****************************************************************************
103  * create_service
104  *****************************************************************************/
105 static int create_service( bonjour_t *p_sys )
106 {
107     int error;
108
109     if( p_sys->group == NULL )
110     {
111         p_sys->group = avahi_entry_group_new( p_sys->client,
112                                               entry_group_callback,
113                                               p_sys );
114         if( p_sys->group == NULL )
115         {
116             msg_Err( p_sys->p_log, "failed to create avahi entry group: %s",
117                      avahi_strerror( avahi_client_errno( p_sys->client ) ) );
118             return VLC_EGENERIC;
119         }
120     }
121
122     error = avahi_entry_group_add_service( p_sys->group, AVAHI_IF_UNSPEC,
123                                            AVAHI_PROTO_UNSPEC, 0, p_sys->psz_name,
124                                            p_sys->psz_stype, NULL, NULL,
125                                            p_sys->i_port,
126                                            p_sys->psz_txt, NULL );
127     if( error < 0 )
128     {
129         msg_Err( p_sys->p_log, "failed to add %s service: %s",
130                  p_sys->psz_stype, avahi_strerror( error ) );
131         return VLC_EGENERIC;
132     }
133
134     error = avahi_entry_group_commit( p_sys->group );
135     if( error < 0 )
136     {
137         msg_Err( p_sys->p_log, "failed to commit entry group: %s",
138                  avahi_strerror( error ) );
139         return VLC_EGENERIC;
140     }
141
142     return VLC_SUCCESS;
143 }
144
145 /*****************************************************************************
146  * client_callback
147  *****************************************************************************/
148 static void client_callback( AvahiClient *c,
149                              AvahiClientState state,
150                              void * userdata )
151 {
152     bonjour_t *p_sys = (bonjour_t *)userdata;
153
154     if( state == AVAHI_CLIENT_S_RUNNING )
155     {
156         p_sys->client = c;
157         create_service( p_sys );
158     }
159     else if( state == AVAHI_CLIENT_S_COLLISION )
160     {
161         if( p_sys->group != NULL )
162             avahi_entry_group_reset( p_sys->group );
163     }
164     else if( state == AVAHI_CLIENT_FAILURE &&
165               (avahi_client_errno(c) == AVAHI_ERR_DISCONNECTED) )
166     {
167         msg_Err( p_sys->p_log, "avahi client disconnected" );
168         avahi_simple_poll_quit( p_sys->simple_poll );
169     }
170 }
171
172 /*****************************************************************************
173  * poll_iterate_thread
174  *****************************************************************************/
175 static void* poll_iterate_thread( vlc_object_t *p_this )
176 {
177     poll_thread_t *p_pt = (poll_thread_t*)p_this;
178     int canc = vlc_savecancel ();
179
180     while( vlc_object_alive (p_pt) )
181         if( avahi_simple_poll_iterate( p_pt->simple_poll, 100 ) != 0 )
182             break;
183
184     vlc_restorecancel (canc);
185     return NULL;
186 }
187
188 /*****************************************************************************
189  * bonjour_start_service
190  *****************************************************************************/
191 void *bonjour_start_service( vlc_object_t *p_log, const char *psz_stype,
192                              const char *psz_name, int i_port, char *psz_txt )
193 {
194     int err;
195
196     bonjour_t* p_sys = calloc( 1, sizeof(*p_sys) );
197     if( p_sys == NULL )
198         return NULL;
199
200     p_sys->p_log = p_log;
201     p_sys->i_port = i_port;
202     p_sys->psz_name = avahi_strdup( psz_name );
203     p_sys->psz_stype = avahi_strdup( psz_stype );
204     if( p_sys->psz_name == NULL || p_sys->psz_stype == NULL )
205         goto error;
206
207     if( psz_txt != NULL )
208     {
209         p_sys->psz_txt = avahi_strdup( psz_txt );
210         if( p_sys->psz_txt == NULL )
211             goto error;
212     }
213
214     p_sys->simple_poll = avahi_simple_poll_new();
215     if( p_sys->simple_poll == NULL )
216     {
217         msg_Err( p_sys->p_log, "failed to create avahi simple pool" );
218         goto error;
219     }
220
221     p_sys->client = avahi_client_new( avahi_simple_poll_get(p_sys->simple_poll),
222                                       0,
223                                       client_callback, p_sys, &err );
224     if( p_sys->client == NULL )
225     {
226         msg_Err( p_sys->p_log, "failed to create avahi client: %s",
227                  avahi_strerror( err ) );
228         goto error;
229     }
230
231     p_sys->poll_thread = vlc_object_create( p_sys->p_log,
232                                             sizeof(poll_thread_t) );
233     if( p_sys->poll_thread == NULL )
234         goto error;
235     p_sys->poll_thread->simple_poll = p_sys->simple_poll;
236
237     if( vlc_thread_create( p_sys->poll_thread, "Avahi Poll Iterate Thread",
238                            poll_iterate_thread,
239                            VLC_THREAD_PRIORITY_HIGHEST, false ) )
240     {
241         msg_Err( p_sys->p_log, "failed to create poll iterate thread" );
242         goto error;
243     }
244
245     return (void *)p_sys;
246
247 error:
248     if( p_sys->poll_thread != NULL )
249         vlc_object_release( p_sys->poll_thread );
250     if( p_sys->client != NULL )
251         avahi_client_free( p_sys->client );
252     if( p_sys->simple_poll != NULL )
253         avahi_simple_poll_free( p_sys->simple_poll );
254     if( p_sys->psz_stype != NULL )
255         avahi_free( p_sys->psz_stype );
256     if( p_sys->psz_name != NULL )
257         avahi_free( p_sys->psz_name );
258     if( p_sys->psz_txt != NULL )
259         avahi_free( p_sys->psz_txt );
260
261     free( p_sys );
262
263     return NULL;
264 }
265
266 /*****************************************************************************
267  * bonjour_stop_service
268  *****************************************************************************/
269 void bonjour_stop_service( void *_p_sys )
270 {
271     bonjour_t *p_sys = (bonjour_t *)_p_sys;
272
273     vlc_object_kill( p_sys->poll_thread );
274     vlc_thread_join( p_sys->poll_thread );
275     vlc_object_release( p_sys->poll_thread );
276
277     if( p_sys->group != NULL )
278         avahi_entry_group_free( p_sys->group );
279
280     avahi_client_free( p_sys->client );
281     avahi_simple_poll_free( p_sys->simple_poll );
282
283     if( p_sys->psz_name != NULL )
284         avahi_free( p_sys->psz_name );
285
286     if( p_sys->psz_txt != NULL )
287         avahi_free( p_sys->psz_txt );
288
289     avahi_free( p_sys->psz_stype );
290
291     free( _p_sys );
292 }
293
294 #endif /* HAVE_AVAHI_CLIENT */