Fix a memory leak when media library is disabled
[vlc.git] / src / playlist / loadsave.c
1 /*****************************************************************************
2  * loadsave.c : Playlist loading / saving functions
3  *****************************************************************************
4  * Copyright (C) 1999-2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Samuel Hocevar <sam@zoy.org>
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 #ifdef HAVE_CONFIG_H
24 # include "config.h"
25 #endif
26
27 #include <vlc_common.h>
28 #include <vlc_playlist.h>
29 #include <vlc_events.h>
30 #include "playlist_internal.h"
31 #include "config/configuration.h"
32 #include <vlc_charset.h>
33
34 #include <sys/types.h>
35 #include <sys/stat.h>
36 #include <unistd.h>
37 #include <errno.h>
38
39 int playlist_Export( playlist_t * p_playlist, const char *psz_filename ,
40                      playlist_item_t *p_export_root,const char *psz_type )
41 {
42     module_t *p_module;
43     playlist_export_t export;
44
45     if( p_export_root == NULL ) return VLC_EGENERIC;
46
47     msg_Dbg( p_playlist, "saving %s to file %s",
48                     p_export_root->p_input->psz_name, psz_filename );
49
50     /* Prepare the playlist_export_t structure */
51     export.psz_filename = psz_filename ? strdup( psz_filename ) : NULL;
52     export.p_file = utf8_fopen( psz_filename, "wt" );
53     if( export.p_file == NULL )
54     {
55         msg_Err( p_playlist , "could not create playlist file %s (%m)",
56                  psz_filename );
57         free( export.psz_filename );
58         return VLC_EGENERIC;
59     }
60
61     export.p_root = p_export_root;
62
63     playlist_Lock( p_playlist );
64     p_playlist->p_private = (void *)&export;
65
66     /* And call the module ! All work is done now */
67     p_module = module_need( p_playlist, "playlist export", psz_type, true);
68     if( !p_module )
69         msg_Warn( p_playlist, "exporting playlist failed" );
70     else
71         module_unneed( p_playlist , p_module );
72     p_playlist->p_private = NULL;
73     playlist_Unlock( p_playlist );
74
75     /* Clean up */
76     fclose( export.p_file );
77     free( export.psz_filename );
78
79     return p_module ? VLC_SUCCESS : VLC_ENOOBJ;
80 }
81
82 int playlist_Import( playlist_t *p_playlist, const char *psz_file )
83 {
84     input_item_t *p_input;
85     char *psz_uri;
86     const char *const psz_option = "meta-file";
87
88     if( asprintf( &psz_uri, "file/://%s", psz_file ) < 0 )
89         return VLC_EGENERIC;
90
91     p_input = input_item_NewExt( p_playlist, psz_uri, psz_file,
92                                  1, &psz_option, VLC_INPUT_OPTION_TRUSTED, -1 );
93     free( psz_uri );
94
95     playlist_AddInput( p_playlist, p_input, PLAYLIST_APPEND, PLAYLIST_END,
96                        true, false );
97     return input_Read( p_playlist, p_input, true );
98 }
99
100 /*****************************************************************************
101  * A subitem has been added to the Media Library (Event Callback)
102  *****************************************************************************/
103 static void input_item_subitem_added( const vlc_event_t * p_event,
104                                       void * user_data )
105 {
106     playlist_t *p_playlist = user_data;
107     input_item_t *p_item = p_event->u.input_item_subitem_added.p_new_child;
108
109     /* playlist_AddInput() can fail, but we have no way to report that ..
110      * Any way when it has failed, either the playlist is dying, either OOM */
111     playlist_AddInput( p_playlist, p_item, PLAYLIST_APPEND, PLAYLIST_END,
112             false, pl_Unlocked );
113 }
114
115 int playlist_MLLoad( playlist_t *p_playlist )
116 {
117     char *psz_datadir;
118     char *psz_uri = NULL;
119     input_item_t *p_input;
120
121     if( !config_GetInt( p_playlist, "media-library") )
122         return VLC_SUCCESS;
123
124     psz_datadir = config_GetUserDataDir();
125
126     if( !psz_datadir ) /* XXX: This should never happen */
127     {
128         msg_Err( p_playlist, "no data directory, cannot load media library") ;
129         return VLC_EGENERIC;
130     }
131
132     if( asprintf( &psz_uri, "%s" DIR_SEP "ml.xspf", psz_datadir ) == -1 )
133     {
134         psz_uri = NULL;
135         goto error;
136     }
137     struct stat p_stat;
138     /* checks if media library file is present */
139     if( utf8_stat( psz_uri , &p_stat ) )
140         goto error;
141     free( psz_uri );
142
143     /* FIXME: WTF? stat() should never be used right before open()! */
144     if( asprintf( &psz_uri, "file/xspf-open://%s" DIR_SEP "ml.xspf",
145                   psz_datadir ) == -1 )
146     {
147         psz_uri = NULL;
148         goto error;
149     }
150     free( psz_datadir );
151     psz_datadir = NULL;
152
153     const char *const psz_option = "meta-file";
154     /* that option has to be cleaned in input_item_subitem_added() */
155     /* vlc_gc_decref() in the same function */
156     p_input = input_item_NewExt( p_playlist, psz_uri, _("Media Library"),
157                                  1, &psz_option, VLC_INPUT_OPTION_TRUSTED, -1 );
158     if( p_input == NULL )
159         goto error;
160
161     PL_LOCK;
162     if( p_playlist->p_ml_onelevel->p_input )
163         vlc_gc_decref( p_playlist->p_ml_onelevel->p_input );
164     if( p_playlist->p_ml_category->p_input )
165         vlc_gc_decref( p_playlist->p_ml_category->p_input );
166
167     p_playlist->p_ml_onelevel->p_input =
168     p_playlist->p_ml_category->p_input = p_input;
169     /* We save the input at two different place, incref */
170     vlc_gc_incref( p_input );
171     vlc_gc_incref( p_input );
172
173     vlc_event_attach( &p_input->event_manager, vlc_InputItemSubItemAdded,
174                         input_item_subitem_added, p_playlist );
175
176     pl_priv(p_playlist)->b_doing_ml = true;
177     PL_UNLOCK;
178
179     stats_TimerStart( p_playlist, "ML Load", STATS_TIMER_ML_LOAD );
180     input_Read( p_playlist, p_input, true );
181     stats_TimerStop( p_playlist,STATS_TIMER_ML_LOAD );
182
183     PL_LOCK;
184     pl_priv(p_playlist)->b_doing_ml = false;
185     PL_UNLOCK;
186
187     vlc_event_detach( &p_input->event_manager, vlc_InputItemSubItemAdded,
188                         input_item_subitem_added, p_playlist );
189
190     vlc_gc_decref( p_input );
191     free( psz_uri );
192     return VLC_SUCCESS;
193
194 error:
195     free( psz_uri );
196     free( psz_datadir );
197     return VLC_ENOMEM;
198 }
199
200 int playlist_MLDump( playlist_t *p_playlist )
201 {
202     char *psz_datadir;
203
204     if( !config_GetInt( p_playlist, "media-library") )
205         return VLC_SUCCESS;
206
207     psz_datadir = config_GetUserDataDir();
208
209     if( !psz_datadir ) /* XXX: This should never happen */
210     {
211         msg_Err( p_playlist, "no data directory, cannot save media library") ;
212         return VLC_EGENERIC;
213     }
214
215     char psz_dirname[ strlen( psz_datadir ) + sizeof( DIR_SEP "ml.xspf")];
216     strcpy( psz_dirname, psz_datadir );
217     free( psz_datadir );
218     if( config_CreateDir( (vlc_object_t *)p_playlist, psz_dirname ) )
219     {
220         return VLC_EGENERIC;
221     }
222
223     strcat( psz_dirname, DIR_SEP "ml.xspf" );
224
225     stats_TimerStart( p_playlist, "ML Dump", STATS_TIMER_ML_DUMP );
226     playlist_Export( p_playlist, psz_dirname, p_playlist->p_ml_category,
227                      "export-xspf" );
228     stats_TimerStop( p_playlist, STATS_TIMER_ML_DUMP );
229
230     return VLC_SUCCESS;
231 }