Fix errno undefined error
[vlc.git] / modules / access_filter / record.c
1 /*****************************************************************************
2  * record.c
3  *****************************************************************************
4  * Copyright (C) 2005 VideoLAN
5  * $Id: demux.c 7546 2004-04-29 13:53:29Z gbazin $
6  *
7  * Author: Laurent Aimar <fenrir@via.ecp.fr>
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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <stdlib.h>
28
29 #include <vlc/vlc.h>
30 #include <vlc/input.h>
31 #include <vlc/vout.h>
32
33 #include "vlc_keys.h"
34 #include <osd.h>
35 #include <errno.h>
36
37 /*****************************************************************************
38  * Module descriptor
39  *****************************************************************************/
40
41 #define RECORD_PATH_TXT N_("Record directory")
42 #define RECORD_PATH_LONGTXT N_( \
43     "Allows you to specify the directory where the record will be stored" )
44
45 static int  Open ( vlc_object_t * );
46 static void Close( vlc_object_t * );
47
48 vlc_module_begin();
49     set_shortname( _("Record") );
50     set_description( _("Record") );
51     set_category( CAT_INPUT );
52     set_subcategory( SUBCAT_INPUT_ACCESS_FILTER );
53     set_capability( "access_filter", 0 );
54     add_shortcut( "record" );
55
56     add_string( "record-path", NULL, NULL,
57                 RECORD_PATH_TXT, RECORD_PATH_LONGTXT, VLC_TRUE );
58
59     set_callbacks( Open, Close );
60
61 vlc_module_end();
62
63 /*****************************************************************************
64  * Local prototypes
65  *****************************************************************************/
66
67 static block_t *Block  ( access_t * );
68 static int      Read   ( access_t *, uint8_t *, int );
69 static int      Control( access_t *, int i_query, va_list args );
70 static int      Seek   ( access_t *, int64_t );
71
72 static void Dump( access_t *, uint8_t *, int );
73
74 static int EventKey( vlc_object_t *, char const *,
75                      vlc_value_t, vlc_value_t, void * );
76
77 struct access_sys_t
78 {
79     vlc_bool_t b_dump;
80
81     char *psz_path;
82     char *psz_ext;
83     char *psz_file;
84     int64_t i_size;
85     FILE *f;
86
87     vout_thread_t *p_vout;
88     int            i_vout_chan;
89 };
90
91 /*****************************************************************************
92  * Open:
93  *****************************************************************************/
94 static int Open( vlc_object_t *p_this )
95 {
96     access_t *p_access = (access_t*)p_this;
97     access_t *p_src = p_access->p_source;
98     access_sys_t *p_sys;
99     char *psz;
100
101     /* */
102     p_access->pf_read  = p_src->pf_read  ? Read : NULL;
103     p_access->pf_block = p_src->pf_block ? Block : NULL;
104     p_access->pf_seek  = p_src->pf_seek  ? Seek : NULL;
105     p_access->pf_control = Control;
106
107     /* */
108     p_access->info = p_src->info;
109
110     /* */
111     p_access->p_sys = p_sys = malloc( sizeof( access_t ) );
112
113     /* */
114     p_sys->f = NULL;
115     p_sys->i_size = 0;
116     p_sys->psz_file = NULL;
117     p_sys->psz_ext = "dat";
118     p_sys->b_dump = VLC_FALSE;
119     p_sys->p_vout = NULL;
120     p_sys->i_vout_chan = -1;
121
122     if( !strncasecmp( p_src->psz_access, "dvb", 3 ) ||
123         !strncasecmp( p_src->psz_access, "udp", 3 ) )
124         p_sys->psz_ext = "ts";
125
126     psz = var_CreateGetString( p_access, "record-path" );
127     if( *psz == '\0' )
128     {
129         free( psz );
130         if( p_access->p_vlc->psz_homedir )
131             psz = strdup( p_access->p_vlc->psz_homedir );
132     }
133     p_sys->psz_path = psz;
134     msg_Dbg( p_access, "Record access filter path %s", psz );
135
136     /* catch all key event */
137     var_AddCallback( p_access->p_vlc, "key-pressed", EventKey, p_access );
138
139     return VLC_SUCCESS;
140 }
141
142 /*****************************************************************************
143  * Close:
144  *****************************************************************************/
145 static void Close( vlc_object_t *p_this )
146 {
147     access_t     *p_access = (access_t*)p_this;
148     access_sys_t *p_sys = p_access->p_sys;
149
150     var_DelCallback( p_access->p_vlc, "key-pressed", EventKey, p_access );
151
152     if( p_sys->f )
153     {
154         fclose( p_sys->f );
155         free( p_sys->psz_file );
156     }
157
158     free( p_sys->psz_path );
159     free( p_sys );
160 }
161
162 /*****************************************************************************
163  *
164  *****************************************************************************/
165 static block_t *Block( access_t *p_access )
166 {
167     access_t     *p_src = p_access->p_source;
168     block_t      *p_block;
169
170     /* */
171     p_block = p_src->pf_block( p_src );
172     if( p_block && p_block->i_buffer )
173         Dump( p_access, p_block->p_buffer, p_block->i_buffer );
174
175     /* */
176     p_access->info = p_src->info;
177
178     return p_block;
179 }
180
181 /*****************************************************************************
182  *
183  *****************************************************************************/
184 static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
185 {
186     access_t     *p_src = p_access->p_source;
187     int i_ret;
188
189     i_ret = p_src->pf_read( p_src, p_buffer, i_len );
190
191     if( i_ret > 0 )
192         Dump( p_access, p_buffer, i_ret );
193
194     /* */
195     p_access->info = p_src->info;
196
197     return i_ret;
198 }
199
200 /*****************************************************************************
201  *
202  *****************************************************************************/
203 static int Control( access_t *p_access, int i_query, va_list args )
204 {
205     access_t     *p_src = p_access->p_source;
206     int i_ret;
207
208     i_ret = p_src->pf_control( p_src, i_query, args );
209
210     /* */
211     p_access->info = p_src->info;
212
213     return i_ret;
214 }
215
216 /*****************************************************************************
217  *
218  *****************************************************************************/
219 static int Seek( access_t *p_access, int64_t i_pos )
220 {
221     access_t     *p_src = p_access->p_source;
222     int i_ret;
223
224     i_ret = p_src->pf_seek( p_src, i_pos );
225
226     /* */
227     p_access->info = p_src->info;
228
229     return i_ret;
230 }
231
232 /*****************************************************************************
233  *
234  *****************************************************************************/
235 static int EventKey( vlc_object_t *p_this, char const *psz_var,
236                      vlc_value_t oldval, vlc_value_t newval, void *p_data )
237 {
238     access_t     *p_access = p_data;
239     access_sys_t *p_sys = p_access->p_sys;
240
241     struct hotkey *p_hotkeys = p_access->p_vlc->p_hotkeys;
242     int i_action = -1, i;
243
244     for( i = 0; p_hotkeys[i].psz_action != NULL; i++ )
245     {
246         if( p_hotkeys[i].i_key == newval.i_int )
247         {
248             i_action = p_hotkeys[i].i_action;
249         }
250     }
251
252     if( i_action == ACTIONID_RECORD )
253     {
254         if( p_sys->b_dump )
255             p_sys->b_dump = VLC_FALSE;
256         else
257             p_sys->b_dump = VLC_TRUE;
258     }
259
260     return VLC_SUCCESS;
261 }
262
263 /*****************************************************************************
264  *
265  *****************************************************************************/
266 static void Notify( access_t *p_access, vlc_bool_t b_dump )
267 {
268     access_sys_t *p_sys = p_access->p_sys;
269     vout_thread_t *p_vout;
270
271     p_vout = vlc_object_find( p_access, VLC_OBJECT_VOUT, FIND_ANYWHERE );
272
273     if( p_vout != p_sys->p_vout )
274     {
275         p_sys->p_vout = p_vout;
276         if( spu_Control( p_vout->p_spu, SPU_CHANNEL_REGISTER,
277                          &p_sys->i_vout_chan  ) )
278             p_sys->i_vout_chan = -1;
279     }
280
281     if( p_sys->i_vout_chan != -1 )
282     {
283         if( b_dump )
284             vout_OSDMessage( p_vout, p_sys->i_vout_chan, "Recording" );
285         else
286             vout_OSDMessage( p_vout, p_sys->i_vout_chan, "Recording done" );
287     }
288     vlc_object_release( p_vout );
289 }
290
291 /*****************************************************************************
292  *
293  *****************************************************************************/
294 static void Dump( access_t *p_access, uint8_t *p_buffer, int i_buffer )
295 {
296     access_sys_t *p_sys = p_access->p_sys;
297     int i_write;
298
299     /* */
300     if( !p_sys->b_dump )
301     {
302         if( p_sys->f )
303         {
304             msg_Dbg( p_access, "dumped "I64Fd" kb (%s)",
305                      p_sys->i_size/1024, p_sys->psz_file );
306
307             Notify( p_access, VLC_FALSE );
308
309             fclose( p_sys->f );
310             p_sys->f = NULL;
311
312             free( p_sys->psz_file );
313             p_sys->psz_file = NULL;
314
315             p_sys->i_size = 0;
316         }
317         return;
318     }
319
320     /* */
321     if( !p_sys->f )
322     {
323         input_thread_t *p_input;
324         char *psz_name = NULL;
325         time_t t = time(NULL);
326         struct tm l;
327
328         if( !localtime_r( &t, &l ) )
329             memset( &l, 0, sizeof(l) );
330
331         p_input = vlc_object_find( p_access, VLC_OBJECT_INPUT, FIND_PARENT );
332         if( p_input )
333         {
334             vlc_mutex_lock( &p_input->input.p_item->lock );
335             if( p_input->input.p_item->psz_name &&
336                 strlen( p_input->input.p_item->psz_name ) < 64 )
337                 psz_name = strdup( p_input->input.p_item->psz_name );
338             vlc_mutex_unlock( &p_input->input.p_item->lock );
339
340             vlc_object_release( p_input );
341         }
342         if( psz_name == NULL )
343             psz_name = strdup( "Unknown" );
344
345         asprintf( &p_sys->psz_file, "%s/%s %d-%d-%d %.2dh%.2dm%.2ds.%s",
346                   p_sys->psz_path, psz_name,
347                   l.tm_mday, l.tm_mon+1, l.tm_year+1900,
348                   l.tm_hour, l.tm_min, l.tm_sec,
349                   p_sys->psz_ext );
350
351         free( psz_name );
352
353         msg_Dbg( p_access, "dump in file '%s'", p_sys->psz_file );
354
355         p_sys->f = fopen( p_sys->psz_file, "wb" );
356         if( p_sys->f == NULL )
357         {
358             msg_Err( p_access, "cannot open file '%s' (%s)",
359                      p_sys->psz_file, strerror(errno) );
360             free( p_sys->psz_file );
361             p_sys->psz_file = NULL;
362             p_sys->b_dump = VLC_FALSE;
363             return;
364         }
365
366         Notify( p_access, VLC_TRUE );
367
368         p_sys->i_size = 0;
369     }
370
371     /* */
372     if( ( i_write = fwrite( p_buffer, 1, i_buffer, p_sys->f ) ) > 0 )
373         p_sys->i_size += i_write;
374 }
375