Statistics
| Branch: | Revision:

ffmpeg / libavformat / file.c @ 31317656

History | View | Annotate | Download (3.94 KB)

1
/*
2
 * Buffered file io for ffmpeg system
3
 * Copyright (c) 2001 Fabrice Bellard
4
 *
5
 * This file is part of Libav.
6
 *
7
 * Libav is free software; you can redistribute it and/or
8
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * Libav is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with Libav; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21

    
22
#include "libavutil/avstring.h"
23
#include "avformat.h"
24
#include <fcntl.h>
25
#if HAVE_SETMODE
26
#include <io.h>
27
#endif
28
#include <unistd.h>
29
#include <sys/stat.h>
30
#include <stdlib.h>
31
#include "os_support.h"
32
#include "url.h"
33

    
34

    
35
/* standard file protocol */
36

    
37
static int file_read(URLContext *h, unsigned char *buf, int size)
38
{
39
    int fd = (intptr_t) h->priv_data;
40
    return read(fd, buf, size);
41
}
42

    
43
static int file_write(URLContext *h, const unsigned char *buf, int size)
44
{
45
    int fd = (intptr_t) h->priv_data;
46
    return write(fd, buf, size);
47
}
48

    
49
static int file_get_handle(URLContext *h)
50
{
51
    return (intptr_t) h->priv_data;
52
}
53

    
54
#if CONFIG_FILE_PROTOCOL
55

    
56
static int file_open(URLContext *h, const char *filename, int flags)
57
{
58
    int access;
59
    int fd;
60

    
61
    av_strstart(filename, "file:", &filename);
62

    
63
    if (flags & AVIO_RDWR) {
64
        access = O_CREAT | O_TRUNC | O_RDWR;
65
    } else if (flags & AVIO_WRONLY) {
66
        access = O_CREAT | O_TRUNC | O_WRONLY;
67
    } else {
68
        access = O_RDONLY;
69
    }
70
#ifdef O_BINARY
71
    access |= O_BINARY;
72
#endif
73
    fd = open(filename, access, 0666);
74
    if (fd == -1)
75
        return AVERROR(errno);
76
    h->priv_data = (void *) (intptr_t) fd;
77
    return 0;
78
}
79

    
80
/* XXX: use llseek */
81
static int64_t file_seek(URLContext *h, int64_t pos, int whence)
82
{
83
    int fd = (intptr_t) h->priv_data;
84
    if (whence == AVSEEK_SIZE) {
85
        struct stat st;
86
        int ret = fstat(fd, &st);
87
        return ret < 0 ? AVERROR(errno) : st.st_size;
88
    }
89
    return lseek(fd, pos, whence);
90
}
91

    
92
static int file_close(URLContext *h)
93
{
94
    int fd = (intptr_t) h->priv_data;
95
    return close(fd);
96
}
97

    
98
static int file_check(URLContext *h, int mask)
99
{
100
    struct stat st;
101
    int ret = stat(h->filename, &st);
102
    if (ret < 0)
103
        return AVERROR(errno);
104

    
105
    ret |= st.st_mode&S_IRUSR ? mask&AVIO_RDONLY : 0;
106
    ret |= st.st_mode&S_IWUSR ? mask&AVIO_WRONLY : 0;
107
    ret |= st.st_mode&S_IWUSR && st.st_mode&S_IRUSR ? mask&AVIO_RDWR : 0;
108

    
109
    return ret;
110
}
111

    
112
URLProtocol ff_file_protocol = {
113
    .name                = "file",
114
    .url_open            = file_open,
115
    .url_read            = file_read,
116
    .url_write           = file_write,
117
    .url_seek            = file_seek,
118
    .url_close           = file_close,
119
    .url_get_file_handle = file_get_handle,
120
    .url_check           = file_check,
121
};
122

    
123
#endif /* CONFIG_FILE_PROTOCOL */
124

    
125
#if CONFIG_PIPE_PROTOCOL
126

    
127
static int pipe_open(URLContext *h, const char *filename, int flags)
128
{
129
    int fd;
130
    char *final;
131
    av_strstart(filename, "pipe:", &filename);
132

    
133
    fd = strtol(filename, &final, 10);
134
    if((filename == final) || *final ) {/* No digits found, or something like 10ab */
135
        if (flags & AVIO_WRONLY) {
136
            fd = 1;
137
        } else {
138
            fd = 0;
139
        }
140
    }
141
#if HAVE_SETMODE
142
    setmode(fd, O_BINARY);
143
#endif
144
    h->priv_data = (void *) (intptr_t) fd;
145
    h->is_streamed = 1;
146
    return 0;
147
}
148

    
149
URLProtocol ff_pipe_protocol = {
150
    .name                = "pipe",
151
    .url_open            = pipe_open,
152
    .url_read            = file_read,
153
    .url_write           = file_write,
154
    .url_get_file_handle = file_get_handle,
155
    .url_check           = file_check,
156
};
157

    
158
#endif /* CONFIG_PIPE_PROTOCOL */