HAL service discovery : fix a memory leak
[vlc.git] / modules / services_discovery / hal.c
1 /*****************************************************************************
2  * hal.c :  HAL interface module
3  *****************************************************************************
4  * Copyright (C) 2004 the VideoLAN team
5  * Copyright © 2006-2007 Rafaël Carré
6  * $Id$
7  *
8  * Authors: Clément Stenac <zorglub@videolan.org>
9  *          Rafaël Carré <funman at videolanorg>
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License as published by
13  * the Free Software Foundation; either version 2 of the License, or
14  * (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24  *****************************************************************************/
25
26 #ifdef HAVE_CONFIG_H
27 # include "config.h"
28 #endif
29
30 #include <vlc_common.h>
31 #include <vlc_plugin.h>
32 #include <vlc_playlist.h>
33 #include <vlc_services_discovery.h>
34
35 #include <vlc_network.h>
36
37 #include <errno.h>                                                 /* ENOMEM */
38
39 #ifdef HAVE_UNISTD_H
40 #    include <unistd.h>
41 #endif
42 #ifdef HAVE_SYS_TIME_H
43 #    include <sys/time.h>
44 #endif
45
46 #include <hal/libhal.h>
47
48 #define MAX_LINE_LENGTH 256
49
50 /*****************************************************************************
51  * Local prototypes
52  *****************************************************************************/
53
54 /* store relation between item id and udi for ejection */
55 struct udi_input_id_t
56 {
57     char            *psz_udi;
58     input_item_t    *p_item;
59 };
60
61 struct services_discovery_sys_t
62 {
63     vlc_thread_t            thread;
64     LibHalContext           *p_ctx;
65     DBusConnection          *p_connection;
66     int                     i_devices_number;
67     struct udi_input_id_t   **pp_devices;
68 };
69 static void *Run ( void * );
70 static int  Open ( vlc_object_t * );
71 static void Close( vlc_object_t * );
72
73 /* HAL callbacks */
74 void DeviceAdded( LibHalContext *p_ctx, const char *psz_udi );
75 void DeviceRemoved( LibHalContext *p_ctx, const char *psz_udi );
76
77 /* to retrieve p_sd in HAL callbacks */
78 services_discovery_t        *p_sd_global;
79
80 /*****************************************************************************
81  * Module descriptor
82  *****************************************************************************/
83 vlc_module_begin ()
84     set_description( N_("HAL devices detection") )
85     set_category( CAT_PLAYLIST )
86     set_subcategory( SUBCAT_PLAYLIST_SD )
87
88     set_capability( "services_discovery", 0 )
89     set_callbacks( Open, Close )
90
91 vlc_module_end ()
92
93
94 /*****************************************************************************
95  * Open: initialize and create stuff
96  *****************************************************************************/
97 static int Open( vlc_object_t *p_this )
98 {
99     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
100     services_discovery_sys_t *p_sys  = malloc(
101                                     sizeof( services_discovery_sys_t ) );
102     if( !p_sys )
103         return VLC_ENOMEM;
104
105     DBusError           dbus_error;
106     DBusConnection      *p_connection;
107
108     p_sd_global = p_sd;
109     p_sys->i_devices_number = 0;
110     p_sys->pp_devices = NULL;
111
112     p_sd->p_sys  = p_sys;
113
114     dbus_error_init( &dbus_error );
115
116     p_sys->p_ctx = libhal_ctx_new();
117     if( !p_sys->p_ctx )
118     {
119         msg_Err( p_sd, "unable to create HAL context") ;
120         free( p_sys );
121         return VLC_EGENERIC;
122     }
123     p_connection = dbus_bus_get( DBUS_BUS_SYSTEM, &dbus_error );
124     if( dbus_error_is_set( &dbus_error ) )
125     {
126         msg_Err( p_sd, "unable to connect to DBUS: %s", dbus_error.message );
127         dbus_error_free( &dbus_error );
128         libhal_ctx_free( p_sys->p_ctx );
129         free( p_sys );
130         return VLC_EGENERIC;
131     }
132     libhal_ctx_set_dbus_connection( p_sys->p_ctx, p_connection );
133     p_sys->p_connection = p_connection;
134     if( !libhal_ctx_init( p_sys->p_ctx, &dbus_error ) )
135     {
136         msg_Err( p_sd, "hal not available : %s", dbus_error.message );
137         goto error;
138     }
139
140     if( !libhal_ctx_set_device_added( p_sys->p_ctx, DeviceAdded ) ||
141             !libhal_ctx_set_device_removed( p_sys->p_ctx, DeviceRemoved ) )
142     {
143         msg_Err( p_sd, "unable to add callback" );
144         goto error;
145     }
146
147     if( vlc_clone( &p_sys->thread, Run, p_this, VLC_THREAD_PRIORITY_LOW ) )
148         goto error;
149
150     return VLC_SUCCESS;
151 error:
152     dbus_error_free( &dbus_error );
153     libhal_ctx_free( p_sys->p_ctx );
154     free( p_sys );
155     return VLC_EGENERIC;
156 }
157
158 /*****************************************************************************
159  * Close:
160  *****************************************************************************/
161 static void Close( vlc_object_t *p_this )
162 {
163     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
164     services_discovery_sys_t *p_sys  = p_sd->p_sys;
165
166     /*vlc_cancel( p_sys->thread );*/
167     vlc_object_kill( p_sd );
168     vlc_join( p_sys->thread, NULL );
169     dbus_connection_unref( p_sys->p_connection );
170     struct udi_input_id_t *p_udi_entry;
171
172     while( p_sys->i_devices_number > 0 )
173     {
174         p_udi_entry = p_sys->pp_devices[0];
175         free( p_udi_entry->psz_udi );
176         TAB_REMOVE( p_sys->i_devices_number, p_sys->pp_devices,
177                 p_sys->pp_devices[0] );
178         free( p_udi_entry );
179     }
180     p_sys->pp_devices = NULL;
181
182     libhal_ctx_free( p_sys->p_ctx );
183
184     free( p_sys );
185 }
186
187 static void AddItem( services_discovery_t *p_sd, input_item_t * p_input,
188                     const char* psz_device )
189 {
190     services_discovery_sys_t *p_sys  = p_sd->p_sys;
191     services_discovery_AddItem( p_sd, p_input, NULL /* no category */ );
192
193     struct udi_input_id_t *p_udi_entry;
194     p_udi_entry = malloc( sizeof( struct udi_input_id_t ) );
195     if( !p_udi_entry )
196         return;
197     p_udi_entry->psz_udi = strdup( psz_device );
198     if( !p_udi_entry->psz_udi )
199     {
200         free( p_udi_entry );
201         return;
202     }
203
204     vlc_gc_incref( p_input );
205     p_udi_entry->p_item = p_input;
206     TAB_APPEND( p_sys->i_devices_number, p_sys->pp_devices, p_udi_entry );
207 }
208
209 static void AddDvd( services_discovery_t *p_sd, const char *psz_device )
210 {
211     char *psz_name;
212     char *psz_uri;
213     char *psz_blockdevice;
214     input_item_t        *p_input;
215
216     psz_name = libhal_device_get_property_string( p_sd->p_sys->p_ctx,
217                                         psz_device, "volume.label", NULL );
218     psz_blockdevice = libhal_device_get_property_string( p_sd->p_sys->p_ctx,
219                                         psz_device, "block.device", NULL );
220
221     if( asprintf( &psz_uri, "dvd://%s", psz_blockdevice ) == -1 )
222         return;
223     /* Create the playlist item here */
224     p_input = input_item_New( p_sd, psz_uri, psz_name );
225     free( psz_uri );
226     if( !p_input )
227     {
228         return;
229     }
230
231     AddItem( p_sd, p_input, psz_device );
232
233     vlc_gc_decref( p_input );
234 }
235
236 static void DelItem( services_discovery_t *p_sd, const char* psz_udi )
237 {
238     services_discovery_sys_t    *p_sys  = p_sd->p_sys;
239
240     int i;
241     for( i = 0; i < p_sys->i_devices_number; i++ )
242     { /*  looks for a matching udi */
243         if( strcmp( psz_udi, p_sys->pp_devices[i]->psz_udi ) == 0 )
244         { /* delete the corresponding item */    
245             services_discovery_RemoveItem( p_sd, p_sys->pp_devices[i]->p_item );
246             vlc_gc_decref( p_sys->pp_devices[i]->p_item );
247             free( p_sys->pp_devices[i]->psz_udi );
248             TAB_REMOVE( p_sys->i_devices_number, p_sys->pp_devices,
249                     p_sys->pp_devices[i] );
250         }
251     }
252 }
253
254 static void AddCdda( services_discovery_t *p_sd, const char *psz_device )
255 {
256     char *psz_uri;
257     char *psz_blockdevice;
258     input_item_t     *p_input;
259
260     psz_blockdevice = libhal_device_get_property_string( p_sd->p_sys->p_ctx,
261                                             psz_device, "block.device", NULL );
262
263     if( asprintf( &psz_uri, "cdda://%s", psz_blockdevice ) == -1 )
264         return;
265     /* Create the item here */
266     p_input = input_item_New( p_sd, psz_uri, "Audio CD" );
267     free( psz_uri );
268     if( !p_input )
269         return;
270
271     AddItem( p_sd, p_input, psz_device );
272
273     vlc_gc_decref( p_input );
274 }
275
276 static void ParseDevice( services_discovery_t *p_sd, const char *psz_device )
277 {
278     char *psz_disc_type;
279     services_discovery_sys_t    *p_sys  = p_sd->p_sys;
280
281     if( !libhal_device_property_exists( p_sys->p_ctx, psz_device,
282                                        "volume.disc.type", NULL ) )
283         return;
284
285     psz_disc_type = libhal_device_get_property_string( p_sys->p_ctx,
286                                                     psz_device,
287                                                     "volume.disc.type",
288                                                     NULL );
289     if( !strncmp( psz_disc_type, "dvd_r", 5 ) )
290     {
291         if (libhal_device_get_property_bool( p_sys->p_ctx, psz_device,
292                                      "volume.disc.is_videodvd", NULL ) )
293         AddDvd( p_sd, psz_device );
294     }
295     else if( !strncmp( psz_disc_type, "cd_r", 4 ) )
296     {
297         if( libhal_device_get_property_bool( p_sys->p_ctx, psz_device,
298                                      "volume.disc.has_audio" , NULL ) )
299             AddCdda( p_sd, psz_device );
300     }
301 }
302
303 /*****************************************************************************
304  * Run: main HAL thread
305  *****************************************************************************/
306 static void *Run( void *data )
307 {
308     services_discovery_t     *p_sd  = data;
309     services_discovery_sys_t *p_sys = p_sd->p_sys;
310     char **devices;
311     int i, i_devices;
312     int canc = vlc_savecancel();
313
314     /* parse existing devices first */
315     if( ( devices = libhal_get_all_devices( p_sys->p_ctx, &i_devices, NULL ) ) )
316     {
317         for( i = 0; i < i_devices; i++ )
318         {
319             ParseDevice( p_sd, devices[ i ] );
320             libhal_free_string( devices[ i ] );
321         }
322         free( devices );
323     }
324
325     /* FIXME: Totally lame. There are DBus watch functions to do this properly.
326      * -- Courmisch, 28/08/2008 */
327     while( vlc_object_alive (p_sd) )
328     {
329         /* look for events on the bus, blocking 1 second */
330         dbus_connection_read_write_dispatch( p_sys->p_connection, 1000 );
331         /* HAL 0.5.8.1 can use libhal_ctx_get_dbus_connection(p_sys->p_ctx) */
332     }
333     vlc_restorecancel (canc);
334     return NULL;
335 }
336
337 void DeviceAdded( LibHalContext *p_ctx, const char *psz_udi )
338 {
339     VLC_UNUSED(p_ctx);
340     ParseDevice( p_sd_global, psz_udi );
341 }
342
343 void DeviceRemoved( LibHalContext *p_ctx, const char *psz_udi )
344 {
345     VLC_UNUSED(p_ctx);
346     DelItem( p_sd_global, psz_udi );
347 }