1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) 2005 the VideoLAN team
5 * $Id: demux.c 7546 2004-04-29 13:53:29Z gbazin $
7 * Author: Laurent Aimar <fenrir@via.ecp.fr>
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.
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.
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 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
30 #include <vlc/input.h>
38 /*****************************************************************************
40 *****************************************************************************/
42 #define RECORD_PATH_TXT N_("Record directory")
43 #define RECORD_PATH_LONGTXT N_( \
44 "Allows you to specify the directory where the record will be stored" )
46 static int Open ( vlc_object_t * );
47 static void Close( vlc_object_t * );
50 set_shortname( _("Record") );
51 set_description( _("Record") );
52 set_category( CAT_INPUT );
53 set_subcategory( SUBCAT_INPUT_ACCESS_FILTER );
54 set_capability( "access_filter", 0 );
55 add_shortcut( "record" );
57 add_directory( "record-path", NULL, NULL,
58 RECORD_PATH_TXT, RECORD_PATH_LONGTXT, VLC_TRUE );
60 set_callbacks( Open, Close );
64 /*****************************************************************************
66 *****************************************************************************/
68 static block_t *Block ( access_t * );
69 static int Read ( access_t *, uint8_t *, int );
70 static int Control( access_t *, int i_query, va_list args );
71 static int Seek ( access_t *, int64_t );
73 static void Dump( access_t *, uint8_t *, int );
75 static int EventKey( vlc_object_t *, char const *,
76 vlc_value_t, vlc_value_t, void * );
88 vout_thread_t *p_vout;
92 /*****************************************************************************
94 *****************************************************************************/
95 static int Open( vlc_object_t *p_this )
97 access_t *p_access = (access_t*)p_this;
98 access_t *p_src = p_access->p_source;
103 p_access->pf_read = p_src->pf_read ? Read : NULL;
104 p_access->pf_block = p_src->pf_block ? Block : NULL;
105 p_access->pf_seek = p_src->pf_seek ? Seek : NULL;
106 p_access->pf_control = Control;
109 p_access->info = p_src->info;
112 p_access->p_sys = p_sys = malloc( sizeof( access_t ) );
117 p_sys->psz_file = NULL;
118 p_sys->psz_ext = "dat";
119 p_sys->b_dump = VLC_FALSE;
120 p_sys->p_vout = NULL;
121 p_sys->i_vout_chan = -1;
123 if( !strncasecmp( p_src->psz_access, "dvb", 3 ) ||
124 !strncasecmp( p_src->psz_access, "udp", 3 ) )
125 p_sys->psz_ext = "ts";
127 psz = var_CreateGetString( p_access, "record-path" );
131 if( p_access->p_vlc->psz_homedir )
132 psz = strdup( p_access->p_vlc->psz_homedir );
134 p_sys->psz_path = psz;
135 msg_Dbg( p_access, "Record access filter path %s", psz );
137 /* catch all key event */
138 var_AddCallback( p_access->p_vlc, "key-pressed", EventKey, p_access );
143 /*****************************************************************************
145 *****************************************************************************/
146 static void Close( vlc_object_t *p_this )
148 access_t *p_access = (access_t*)p_this;
149 access_sys_t *p_sys = p_access->p_sys;
151 var_DelCallback( p_access->p_vlc, "key-pressed", EventKey, p_access );
156 free( p_sys->psz_file );
159 free( p_sys->psz_path );
163 /*****************************************************************************
165 *****************************************************************************/
166 static block_t *Block( access_t *p_access )
168 access_t *p_src = p_access->p_source;
172 p_block = p_src->pf_block( p_src );
173 if( p_block && p_block->i_buffer )
174 Dump( p_access, p_block->p_buffer, p_block->i_buffer );
177 p_access->info = p_src->info;
182 /*****************************************************************************
184 *****************************************************************************/
185 static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
187 access_t *p_src = p_access->p_source;
190 i_ret = p_src->pf_read( p_src, p_buffer, i_len );
193 Dump( p_access, p_buffer, i_ret );
196 p_access->info = p_src->info;
201 /*****************************************************************************
203 *****************************************************************************/
204 static int Control( access_t *p_access, int i_query, va_list args )
206 access_t *p_src = p_access->p_source;
209 i_ret = p_src->pf_control( p_src, i_query, args );
212 p_access->info = p_src->info;
217 /*****************************************************************************
219 *****************************************************************************/
220 static int Seek( access_t *p_access, int64_t i_pos )
222 access_t *p_src = p_access->p_source;
225 i_ret = p_src->pf_seek( p_src, i_pos );
228 p_access->info = p_src->info;
233 /*****************************************************************************
235 *****************************************************************************/
236 static int EventKey( vlc_object_t *p_this, char const *psz_var,
237 vlc_value_t oldval, vlc_value_t newval, void *p_data )
239 access_t *p_access = p_data;
240 access_sys_t *p_sys = p_access->p_sys;
242 struct hotkey *p_hotkeys = p_access->p_vlc->p_hotkeys;
243 int i_action = -1, i;
245 for( i = 0; p_hotkeys[i].psz_action != NULL; i++ )
247 if( p_hotkeys[i].i_key == newval.i_int )
249 i_action = p_hotkeys[i].i_action;
253 if( i_action == ACTIONID_RECORD )
256 p_sys->b_dump = VLC_FALSE;
258 p_sys->b_dump = VLC_TRUE;
264 /*****************************************************************************
266 *****************************************************************************/
267 static void Notify( access_t *p_access, vlc_bool_t b_dump )
269 access_sys_t *p_sys = p_access->p_sys;
270 vout_thread_t *p_vout;
272 p_vout = vlc_object_find( p_access, VLC_OBJECT_VOUT, FIND_ANYWHERE );
274 if( p_vout != p_sys->p_vout )
276 p_sys->p_vout = p_vout;
277 if( spu_Control( p_vout->p_spu, SPU_CHANNEL_REGISTER,
278 &p_sys->i_vout_chan ) )
279 p_sys->i_vout_chan = -1;
282 if( p_sys->i_vout_chan != -1 )
285 vout_OSDMessage( p_vout, p_sys->i_vout_chan, "Recording" );
287 vout_OSDMessage( p_vout, p_sys->i_vout_chan, "Recording done" );
289 vlc_object_release( p_vout );
292 /*****************************************************************************
294 *****************************************************************************/
295 static void Dump( access_t *p_access, uint8_t *p_buffer, int i_buffer )
297 access_sys_t *p_sys = p_access->p_sys;
305 msg_Dbg( p_access, "dumped "I64Fd" kb (%s)",
306 p_sys->i_size/1024, p_sys->psz_file );
308 Notify( p_access, VLC_FALSE );
313 free( p_sys->psz_file );
314 p_sys->psz_file = NULL;
324 input_thread_t *p_input;
325 char *psz_name = NULL;
326 time_t t = time(NULL);
329 #ifdef HAVE_LOCALTIME_R
330 if( !localtime_r( &t, &l ) ) memset( &l, 0, sizeof(l) );
334 struct tm *p_l = localtime( &t );
336 else memset( &l, 0, sizeof(l) );
340 p_input = vlc_object_find( p_access, VLC_OBJECT_INPUT, FIND_PARENT );
343 vlc_mutex_lock( &p_input->input.p_item->lock );
344 if( p_input->input.p_item->psz_name )
346 char *p = strrchr( p_input->input.p_item->psz_name, '/' );
348 p = strrchr( p_input->input.p_item->psz_name, '\\' );
351 psz_name = strdup( p_input->input.p_item->psz_name );
352 else if( p[1] != '\0' )
353 psz_name = strdup( &p[1] );
355 vlc_mutex_unlock( &p_input->input.p_item->lock );
357 vlc_object_release( p_input );
360 if( psz_name == NULL )
361 psz_name = strdup( "Unknown" );
363 asprintf( &p_sys->psz_file, "%s/%s %d-%d-%d %.2dh%.2dm%.2ds.%s",
364 p_sys->psz_path, psz_name,
365 l.tm_mday, l.tm_mon+1, l.tm_year+1900,
366 l.tm_hour, l.tm_min, l.tm_sec,
371 msg_Dbg( p_access, "dump in file '%s'", p_sys->psz_file );
373 p_sys->f = fopen( p_sys->psz_file, "wb" );
374 if( p_sys->f == NULL )
376 msg_Err( p_access, "cannot open file '%s' (%s)",
377 p_sys->psz_file, strerror(errno) );
378 free( p_sys->psz_file );
379 p_sys->psz_file = NULL;
380 p_sys->b_dump = VLC_FALSE;
384 Notify( p_access, VLC_TRUE );
390 if( ( i_write = fwrite( p_buffer, 1, i_buffer, p_sys->f ) ) > 0 )
391 p_sys->i_size += i_write;