git.sesse.net Git - vlc/blob - modules/access_output/file.c

   1 /*****************************************************************************
   2  * file.c
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002 VideoLAN
   5  * $Id: file.c,v 1.5 2003/04/29 15:40:31 fenrir Exp $
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Eric Petit <titer@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <stdlib.h>
  29 #include <sys/types.h>
  30 #include <sys/stat.h>
  31 #include <string.h>
  32 #include <errno.h>
  33 #include <fcntl.h>
  34
  35 #include <vlc/vlc.h>
  36 #include <vlc/input.h>
  37 #include <vlc/sout.h>
  38
  39 #ifdef HAVE_UNISTD_H
  40 #   include <unistd.h>
  41 #elif defined( WIN32 ) && !defined( UNDER_CE )
  42 #   include <io.h>
  43 #endif
  44
  45 /*****************************************************************************
  46  * Exported prototypes
  47  *****************************************************************************/
  48 static int     Open   ( vlc_object_t * );
  49 static void    Close  ( vlc_object_t * );
  50
  51 static int     Write( sout_access_out_t *, sout_buffer_t * );
  52 static int     Seek ( sout_access_out_t *, off_t  );
  53
  54 /*****************************************************************************
  55  * Module descriptor
  56  *****************************************************************************/
  57 vlc_module_begin();
  58     set_description( _("File stream ouput") );
  59     set_capability( "sout access", 50 );
  60     add_shortcut( "file" );
  61     set_callbacks( Open, Close );
  62 vlc_module_end();
  63
  64 struct sout_access_out_sys_t
  65 {
  66     FILE *p_file;
  67
  68 };
  69
  70 /*****************************************************************************
  71  * Open: open the file
  72  *****************************************************************************/
  73 static int Open( vlc_object_t *p_this )
  74 {
  75     sout_access_out_t   *p_access = (sout_access_out_t*)p_this;
  76
  77     if( !( p_access->p_sys = malloc( sizeof( sout_access_out_sys_t ) ) ) )
  78     {
  79         msg_Err( p_access, "out of memory" );
  80         return( VLC_EGENERIC );
  81     }
  82
  83     if( !p_access->psz_name )
  84     {
  85         msg_Err( p_access, "no file name specified" );
  86         return VLC_EGENERIC;
  87     }
  88     if( !strcmp( p_access->psz_name, "-" ) )
  89     {
  90         p_access->p_sys->p_file = stdout;
  91         msg_Dbg( p_access, "using stdout" );
  92     }
  93     else if( !( p_access->p_sys->p_file = fopen( p_access->psz_name, "wb" ) ) )
  94     {
  95         msg_Err( p_access, "cannot open `%s'", p_access->psz_name );
  96         free( p_access->p_sys );
  97         return( VLC_EGENERIC );
  98     }
  99
 100     p_access->pf_write        = Write;
 101     p_access->pf_seek         = Seek;
 102
 103     msg_Info( p_access, "Open: name:`%s'", p_access->psz_name );
 104     return VLC_SUCCESS;
 105 }
 106
 107 /*****************************************************************************
 108  * Close: close the target
 109  *****************************************************************************/
 110 static void Close( vlc_object_t * p_this )
 111 {
 112     sout_access_out_t   *p_access = (sout_access_out_t*)p_this;
 113
 114     if( strcmp( p_access->psz_name, "-" ) )
 115     {
 116         if( p_access->p_sys->p_file )
 117         {
 118             fclose( p_access->p_sys->p_file );
 119         }
 120     }
 121     free( p_access->p_sys );
 122
 123     msg_Info( p_access, "Close" );
 124 }
 125
 126 /*****************************************************************************
 127  * Read: standard read on a file descriptor.
 128  *****************************************************************************/
 129 static int Write( sout_access_out_t *p_access, sout_buffer_t *p_buffer )
 130 {
 131     size_t i_write = 0;
 132
 133     do
 134     {
 135         sout_buffer_t *p_next;
 136
 137         i_write += fwrite( p_buffer->p_buffer, 1, p_buffer->i_size,
 138                            p_access->p_sys->p_file );
 139         p_next = p_buffer->p_next;
 140         sout_BufferDelete( p_access->p_sout, p_buffer );
 141         p_buffer = p_next;
 142
 143     } while( p_buffer );
 144
 145     return( i_write );
 146 }
 147
 148 /*****************************************************************************
 149  * Seek: seek to a specific location in a file
 150  *****************************************************************************/
 151 static int Seek( sout_access_out_t *p_access, off_t i_pos )
 152 {
 153
 154     msg_Dbg( p_access, "Seek: pos:"I64Fd, (int64_t)i_pos );
 155     if( strcmp( p_access->psz_name, "-" ) )
 156     {
 157         return( fseek( p_access->p_sys->p_file, i_pos, SEEK_SET ) );
 158     }
 159     else
 160     {
 161         msg_Err( p_access, "cannot seek while using stdout" );
 162         return VLC_EGENERIC;
 163     }
 164 }
 165
 166
 167