file.c 7.18 KB
Newer Older
1
2
3
/*****************************************************************************
 * file.c
 *****************************************************************************
4
 * Copyright (C) 2001, 2002 the VideoLAN team
5
 * $Id$
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *          Eric Petit <titer@videolan.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * 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
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
dionoea's avatar
dionoea committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23
24
25
26
27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
28
29
30
31
32

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

33
#include <sys/types.h>
34
#include <time.h>
35
#include <fcntl.h>
36
#include <errno.h>
37

38
#include <vlc_common.h>
39
#include <vlc_plugin.h>
zorglub's avatar
zorglub committed
40
41
#include <vlc_sout.h>
#include <vlc_block.h>
42
#include <vlc_fs.h>
ivoire's avatar
ivoire committed
43
#include <vlc_strings.h>
44

45
#if defined( WIN32 ) && !defined( UNDER_CE )
46
#   include <io.h>
47
48
49
#   define lseek _lseeki64
#else
#   include <unistd.h>
gbazin's avatar
   
gbazin committed
50
51
#endif

52
53
54
#ifndef O_LARGEFILE
#   define O_LARGEFILE 0
#endif
55
56
57
58

/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
59
60
61
static int  Open ( vlc_object_t * );
static void Close( vlc_object_t * );

Laurent Aimar's avatar
Laurent Aimar committed
62
#define SOUT_CFG_PREFIX "sout-file-"
zorglub's avatar
zorglub committed
63
64
65
#define APPEND_TEXT N_("Append to file")
#define APPEND_LONGTEXT N_( "Append to file if it exists instead " \
                            "of replacing it.")
Laurent Aimar's avatar
Laurent Aimar committed
66

67
68
69
70
71
72
73
74
vlc_module_begin ()
    set_description( N_("File stream output") )
    set_shortname( N_("File" ))
    set_capability( "sout access", 50 )
    set_category( CAT_SOUT )
    set_subcategory( SUBCAT_SOUT_ACO )
    add_shortcut( "file" )
    add_shortcut( "stream" )
ivoire's avatar
ivoire committed
75
    add_bool( SOUT_CFG_PREFIX "append", false, NULL, APPEND_TEXT,APPEND_LONGTEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
76
              true )
77
78
    set_callbacks( Open, Close )
vlc_module_end ()
79

80
81
82
83

/*****************************************************************************
 * Exported prototypes
 *****************************************************************************/
84
static const char *const ppsz_sout_options[] = {
Laurent Aimar's avatar
Laurent Aimar committed
85
86
87
    "append", NULL
};

88
static ssize_t Write( sout_access_out_t *, block_t * );
89
static int Seek ( sout_access_out_t *, off_t  );
90
static ssize_t Read ( sout_access_out_t *, block_t * );
91
static int Control( sout_access_out_t *, int, va_list );
92

93
struct sout_access_out_sys_t
94
{
gbazin's avatar
   
gbazin committed
95
    int i_handle;
96
};
97
98
99
100
101
102

/*****************************************************************************
 * Open: open the file
 *****************************************************************************/
static int Open( vlc_object_t *p_this )
{
103
    sout_access_out_t   *p_access = (sout_access_out_t*)p_this;
104
    int                 fd;
Laurent Aimar's avatar
Laurent Aimar committed
105

106
    config_ChainParse( p_access, SOUT_CFG_PREFIX, ppsz_sout_options, p_access->p_cfg );
107

108
    if( !p_access->psz_path )
109
110
111
112
    {
        msg_Err( p_access, "no file name specified" );
        return VLC_EGENERIC;
    }
Laurent Aimar's avatar
Laurent Aimar committed
113

114
    bool append = var_GetBool( p_access, SOUT_CFG_PREFIX "append" );
115

116
    if( !strcmp( p_access->psz_path, "-" ) )
117
    {
118
#ifndef UNDER_CE
119
120
#ifdef WIN32
        setmode (fileno (stdout), O_BINARY);
dionoea's avatar
dionoea committed
121
#endif
122
        fd = dup (fileno (stdout));
123
        msg_Dbg( p_access, "using stdout" );
124
125
#else
#warning stdout is not supported on Windows Mobile, but may be used on Windows CE
126
        fd = -1;
127
#endif
128
    }
129
    else
130
    {
131
        char *psz_tmp = str_format( p_access, p_access->psz_path );
132
        path_sanitize( psz_tmp );
133

134
135
        fd = utf8_open( psz_tmp, O_RDWR | O_CREAT | O_LARGEFILE |
                        (append ? 0 : O_TRUNC), 0666 );
136
        free( psz_tmp );
137
    }
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
138

139
140
141
142
    if (fd == -1)
    {
        msg_Err( p_access, "cannot open `%s' (%m)", p_access->psz_path );
        return VLC_EGENERIC;
143
144
    }

gbazin's avatar
   
gbazin committed
145
146
147
    p_access->pf_write = Write;
    p_access->pf_read  = Read;
    p_access->pf_seek  = Seek;
148
    p_access->pf_control = Control;
149
    p_access->p_sys    = (void *)(intptr_t)fd;
150

151
152
153
    msg_Dbg( p_access, "file access output opened (%s)", p_access->psz_path );
    if (append)
        lseek (fd, 0, SEEK_END);
154

155
156
157
158
159
160
161
162
    return VLC_SUCCESS;
}

/*****************************************************************************
 * Close: close the target
 *****************************************************************************/
static void Close( vlc_object_t * p_this )
{
gbazin's avatar
   
gbazin committed
163
    sout_access_out_t *p_access = (sout_access_out_t*)p_this;
164

165
    close( (intptr_t)p_access->p_sys );
166

167
    msg_Dbg( p_access, "file access output closed" );
168
169
}

170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
static int Control( sout_access_out_t *p_access, int i_query, va_list args )
{
    switch( i_query )
    {
        case ACCESS_OUT_CONTROLS_PACE:
        {
            bool *pb = va_arg( args, bool * );
            *pb = strcmp( p_access->psz_access, "stream" );
            break;
        }

        default:
            return VLC_EGENERIC;
    }
    return VLC_SUCCESS;
}

187
188
189
/*****************************************************************************
 * Read: standard read on a file descriptor.
 *****************************************************************************/
190
static ssize_t Read( sout_access_out_t *p_access, block_t *p_buffer )
gbazin's avatar
   
gbazin committed
191
{
192
193
194
195
196
197
198
    ssize_t val;

    do
        val = read( (intptr_t)p_access->p_sys, p_buffer->p_buffer,
                    p_buffer->i_buffer );
    while (val == -1 && errno == EINTR);
    return val;
gbazin's avatar
   
gbazin committed
199
200
201
202
203
}

/*****************************************************************************
 * Write: standard write on a file descriptor.
 *****************************************************************************/
204
static ssize_t Write( sout_access_out_t *p_access, block_t *p_buffer )
205
206
207
{
    size_t i_write = 0;

208
    while( p_buffer )
209
    {
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
        ssize_t val = write ((intptr_t)p_access->p_sys,
                             p_buffer->p_buffer, p_buffer->i_buffer);
        if (val == -1)
        {
            if (errno == EINTR)
                continue;
            block_ChainRelease (p_buffer);
            return -1;
        }

        if ((size_t)val >= p_buffer->i_buffer)
        {
            block_t *p_next = p_buffer->p_next;
            block_Release (p_buffer);
            p_buffer = p_next;
        }
        else
        {
            p_buffer->p_buffer += val;
            p_buffer->i_buffer -= val;
        }
        i_write += val;
232
233
    }
    return i_write;
234
235
236
237
238
}

/*****************************************************************************
 * Seek: seek to a specific location in a file
 *****************************************************************************/
239
static int Seek( sout_access_out_t *p_access, off_t i_pos )
240
{
241
    return lseek( (intptr_t)p_access->p_sys, i_pos, SEEK_SET );
242
}