/*****************************************************************************
* demuxdump.c : Pseudo demux module for vlc (dump raw stream)
*****************************************************************************
- * Copyright (C) 2001 VideoLAN
- * $Id: demuxdump.c,v 1.3 2003/01/25 16:58:34 fenrir Exp $
+ * Copyright (C) 2001-2004 the VideoLAN team
+ * $Id$
*
* Authors: Laurent Aimar <fenrir@via.ecp.fr>
*
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
- *
+ *
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
*****************************************************************************/
/*****************************************************************************
* Preamble
*****************************************************************************/
-#include <stdlib.h> /* malloc(), free() */
-#include <string.h> /* strdup() */
-#include <errno.h>
-#include <vlc/vlc.h>
-#include <vlc/input.h>
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
-#include <sys/types.h>
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_demux.h>
+#include <vlc_fs.h>
/*****************************************************************************
- * Local prototypes
+ * Module descriptor
*****************************************************************************/
-static int Activate ( vlc_object_t * );
-static int Demux ( input_thread_t * );
-static void Desactivate ( vlc_object_t * );
+#define FILE_TEXT N_("Dump filename")
+#define FILE_LONGTEXT N_( \
+ "Name of the file to which the raw stream will be dumped." )
+#define APPEND_TEXT N_("Append to existing file")
+#define APPEND_LONGTEXT N_( \
+ "If the file already exists, it will not be overwritten." )
+
+static int Open( vlc_object_t * );
+static void Close ( vlc_object_t * );
+
+vlc_module_begin ()
+ set_shortname("Dump")
+ set_category( CAT_INPUT )
+ set_subcategory( SUBCAT_INPUT_DEMUX )
+ set_description( N_("File dumper") )
+ set_capability( "demux", 0 )
+ add_file( "demuxdump-file", "stream-demux.dump", NULL, FILE_TEXT,
+ FILE_LONGTEXT, false )
+ add_bool( "demuxdump-append", false, NULL, APPEND_TEXT, APPEND_LONGTEXT,
+ false )
+ set_callbacks( Open, Close )
+ add_shortcut( "dump" )
+vlc_module_end ()
-#define DUMP_BLOCKSIZE 16384
/*****************************************************************************
- * Module descriptor
+ * Local prototypes
*****************************************************************************/
-vlc_module_begin();
- set_description( _("Dump Demux input") );
- set_capability( "demux", 0 );
- add_category_hint( "File", NULL );
- add_string( "demuxdump-file", NULL, NULL,
- "dump file name",
- "file name for dumping raw stream read by demux" );
- set_callbacks( Activate, Desactivate );
- add_shortcut( "dump" );
-vlc_module_end();
+static int Demux( demux_t * );
+static int Control( demux_t *, int,va_list );
+
+#define DUMP_BLOCKSIZE 16384
struct demux_sys_t
{
- char *psz_name;
+ char *psz_file;
FILE *p_file;
uint64_t i_write;
- void *p_demux_data_sav;
+ uint8_t buffer[DUMP_BLOCKSIZE];
};
/*
*/
/*****************************************************************************
- * Activate: initializes dump structures
+ * Open: initializes dump structures
*****************************************************************************/
-static int Activate( vlc_object_t * p_this )
+static int Open( vlc_object_t * p_this )
{
- input_thread_t *p_input = (input_thread_t *)p_this;
- demux_sys_t *p_demux;
-
- char *psz_name;
-
- /* Set the demux function */
- p_input->pf_demux = Demux;
-
- /* Initialize access plug-in structures. */
- if( p_input->i_mtu == 0 )
- {
- /* Improve speed. */
- p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
- }
-
- psz_name = config_GetPsz( p_input, "demuxdump-file" );
- if( !psz_name || !*psz_name )
- {
- psz_name = strdup( "stream-demux.dump" );
- }
+ demux_t *p_demux = (demux_t*)p_this;
+ demux_sys_t *p_sys;
+ const char *psz_mode;
+ bool b_append;
+
+ /* Accept only if forced */
+ if( !p_demux->b_force )
+ return VLC_EGENERIC;
+
+ p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
+ if( !p_sys )
+ return VLC_ENOMEM;
+
+ b_append = var_CreateGetBool( p_demux, "demuxdump-append" );
+ if ( b_append )
+ psz_mode = "ab";
+ else
+ psz_mode = "wb";
- p_demux = malloc( sizeof( demux_sys_t ) );
- memset( p_demux, 0, sizeof( demux_sys_t ) );
+ p_demux->pf_demux = Demux;
+ p_demux->pf_control = Control;
- if( !( p_demux->p_file = fopen( psz_name, "wb" ) ) )
- {
- msg_Err( p_input,
- "cannot create `%s' for writing",
- psz_name );
- free( p_demux );
- return( -1 );
- }
- else
+ p_sys->i_write = 0;
+ p_sys->p_file = NULL;
+ p_sys->psz_file = var_CreateGetString( p_demux, "demuxdump-file" );
+ if( *p_sys->psz_file == '\0' )
{
- msg_Info( p_input,
- "dumping raw stream to file `%s'",
- psz_name );
- p_demux->psz_name = psz_name;
+ msg_Warn( p_demux, "no dump file name given" );
+ free( p_sys->psz_file );
+ free( p_sys );
+ return VLC_EGENERIC;
}
- p_demux->i_write = 0;
- p_demux->p_demux_data_sav = p_input->p_demux_data;
-
- if( p_input->stream.p_selected_program != NULL )
+ if( !strcmp( p_sys->psz_file, "-" ) )
{
- /* workaround for dvd access */
- msg_Warn( p_input, "demux data already initializated (by access?)" );
+ msg_Info( p_demux, "dumping raw stream to standard output" );
+ p_sys->p_file = stdout;
}
- else
+ else if( ( p_sys->p_file = vlc_fopen( p_sys->psz_file, psz_mode ) ) == NULL )
{
-
- if( input_InitStream( p_input, 0 ) == -1 )
- {
- fclose( p_demux->p_file );
- free( p_demux );
- return( -1 );
- }
- input_AddProgram( p_input, 0, 0 );
- p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
-
- vlc_mutex_lock( &p_input->stream.stream_lock );
- p_input->stream.p_selected_area->i_tell = 0;
- vlc_mutex_unlock( &p_input->stream.stream_lock );
+ msg_Err( p_demux, "cannot create `%s' for writing", p_sys->psz_file );
+ free( p_sys->psz_file );
+ free( p_sys );
+ return VLC_EGENERIC;
}
+ msg_Info( p_demux, "%s raw stream to file `%s'",
+ b_append ? "appending" : "dumping", p_sys->psz_file );
- p_input->p_demux_data = p_demux;
-
- vlc_mutex_lock( &p_input->stream.stream_lock );
- p_input->stream.p_selected_program->b_is_ok = 1;
- vlc_mutex_unlock( &p_input->stream.stream_lock );
-
- return( 0 );
+ return VLC_SUCCESS;
}
/*****************************************************************************
- * Desctivate: initializes dump structures
+ * Close:
*****************************************************************************/
-static void Desactivate ( vlc_object_t *p_this )
+static void Close( vlc_object_t *p_this )
{
- input_thread_t *p_input = (input_thread_t *)p_this;
- demux_sys_t *p_demux = (demux_sys_t*)p_input->p_demux_data;
- msg_Info( p_input,
- "closing %s ("I64Fd" Kbytes dumped)",
- p_demux->psz_name,
- p_demux->i_write / 1024 );
+ demux_t *p_demux = (demux_t*)p_this;
+ demux_sys_t *p_sys = p_demux->p_sys;
- if( p_demux->p_file )
- {
- fclose( p_demux->p_file );
- p_demux->p_file = NULL;
- }
- if( p_demux->psz_name )
+ msg_Info( p_demux ,"closing %s (%"PRId64" KiB dumped)", p_sys->psz_file,
+ p_sys->i_write / 1024 );
+
+ if( p_sys->p_file != stdout )
{
- free( p_demux->psz_name );
+ fclose( p_sys->p_file );
}
- p_input->p_demux_data = p_demux->p_demux_data_sav;
- free( p_demux );
+ free( p_sys->psz_file );
+ free( p_sys );
}
/*****************************************************************************
*****************************************************************************
* Returns -1 in case of error, 0 in case of EOF, 1 otherwise
*****************************************************************************/
-static int Demux( input_thread_t * p_input )
+static int Demux( demux_t *p_demux )
{
- demux_sys_t *p_demux = (demux_sys_t*)p_input->p_demux_data;
+ demux_sys_t *p_sys = p_demux->p_sys;
- ssize_t i_read;
- data_packet_t * p_data;
- int i_write;
+ int i_data;
- p_input->p_demux_data = p_demux->p_demux_data_sav;
- i_read = input_SplitBuffer( p_input, &p_data, DUMP_BLOCKSIZE );
- p_input->p_demux_data = p_demux;
-
- if ( i_read <= 0 )
- {
- return i_read;
- }
+ i_data = stream_Read( p_demux->s, p_sys->buffer, DUMP_BLOCKSIZE );
+ if ( i_data <= 0 )
+ return i_data;
- i_write = fwrite( p_data->p_payload_start,
- 1,
- i_read,
- p_demux->p_file );
-
- input_DeletePacket( p_input->p_method_data, p_data );
+ i_data = fwrite( p_sys->buffer, 1, i_data, p_sys->p_file );
- if( i_write < 0 )
+ if( i_data == 0 )
{
- msg_Err( p_input,
- "failed to write %d bytes",
- i_write );
- return( -1 );
- }
- else
- {
- msg_Dbg( p_input,
- "dumped %d bytes",
- i_write );
- p_demux->i_write += i_write;
+ msg_Err( p_demux, "failed to write data" );
+ return -1;
}
+#if 0
+ msg_Dbg( p_demux, "dumped %d bytes", i_data );
+#endif
+ p_sys->i_write += i_data;
- if( (p_input->stream.p_selected_program->i_synchro_state == SYNCHRO_REINIT)
- | (input_ClockManageControl( p_input,
- p_input->stream.p_selected_program,
- (mtime_t)0 ) == PAUSE_S) )
- {
- msg_Warn( p_input, "synchro reinit" );
- p_input->stream.p_selected_program->i_synchro_state = SYNCHRO_OK;
- }
+ return 1;
+}
- return( 1 );
+/*****************************************************************************
+ * Demux: reads and demuxes data packets
+ *****************************************************************************/
+static int Control( demux_t *p_demux, int i_query, va_list args )
+{
+ return demux_vaControlHelper( p_demux->s, 0, -1, 0, 1, i_query, args );
}