avio: remove AVIO_* access symbols in favor of new AVIO_FLAG_* symbols
[ffmpeg.git] / libavformat / file.c
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_FLAG_WRITE && flags & AVIO_FLAG_READ) {
64         access = O_CREAT | O_TRUNC | O_RDWR;
65     } else if (flags & AVIO_FLAG_WRITE) {
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_FLAG_READ  : 0;
106     ret |= st.st_mode&S_IWUSR ? mask&AVIO_FLAG_WRITE : 0;
107
108     return ret;
109 }
110
111 URLProtocol ff_file_protocol = {
112     .name                = "file",
113     .url_open            = file_open,
114     .url_read            = file_read,
115     .url_write           = file_write,
116     .url_seek            = file_seek,
117     .url_close           = file_close,
118     .url_get_file_handle = file_get_handle,
119     .url_check           = file_check,
120 };
121
122 #endif /* CONFIG_FILE_PROTOCOL */
123
124 #if CONFIG_PIPE_PROTOCOL
125
126 static int pipe_open(URLContext *h, const char *filename, int flags)
127 {
128     int fd;
129     char *final;
130     av_strstart(filename, "pipe:", &filename);
131
132     fd = strtol(filename, &final, 10);
133     if((filename == final) || *final ) {/* No digits found, or something like 10ab */
134         if (flags & AVIO_FLAG_WRITE) {
135             fd = 1;
136         } else {
137             fd = 0;
138         }
139     }
140 #if HAVE_SETMODE
141     setmode(fd, O_BINARY);
142 #endif
143     h->priv_data = (void *) (intptr_t) fd;
144     h->is_streamed = 1;
145     return 0;
146 }
147
148 URLProtocol ff_pipe_protocol = {
149     .name                = "pipe",
150     .url_open            = pipe_open,
151     .url_read            = file_read,
152     .url_write           = file_write,
153     .url_get_file_handle = file_get_handle,
154     .url_check           = file_check,
155 };
156
157 #endif /* CONFIG_PIPE_PROTOCOL */