3b1ab197be
It is included for the open/read/write/close functions. On MSVC, where this header does not exist, the same functions are provided by io.h, which is already included. On windows, these functions are provided by io.h. Make sure io.h is included if it exists, regardless of the setmode function. Signed-off-by: Martin Storsjö <martin@martin.st>
160 lines
3.9 KiB
C
160 lines
3.9 KiB
C
/*
|
|
* buffered file I/O
|
|
* Copyright (c) 2001 Fabrice Bellard
|
|
*
|
|
* This file is part of Libav.
|
|
*
|
|
* Libav is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* Libav 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with Libav; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "libavutil/avstring.h"
|
|
#include "avformat.h"
|
|
#include <fcntl.h>
|
|
#if HAVE_IO_H
|
|
#include <io.h>
|
|
#endif
|
|
#if HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
#include <sys/stat.h>
|
|
#include <stdlib.h>
|
|
#include "os_support.h"
|
|
#include "url.h"
|
|
|
|
|
|
/* standard file protocol */
|
|
|
|
static int file_read(URLContext *h, unsigned char *buf, int size)
|
|
{
|
|
int fd = (intptr_t) h->priv_data;
|
|
return read(fd, buf, size);
|
|
}
|
|
|
|
static int file_write(URLContext *h, const unsigned char *buf, int size)
|
|
{
|
|
int fd = (intptr_t) h->priv_data;
|
|
return write(fd, buf, size);
|
|
}
|
|
|
|
static int file_get_handle(URLContext *h)
|
|
{
|
|
return (intptr_t) h->priv_data;
|
|
}
|
|
|
|
static int file_check(URLContext *h, int mask)
|
|
{
|
|
struct stat st;
|
|
int ret = stat(h->filename, &st);
|
|
if (ret < 0)
|
|
return AVERROR(errno);
|
|
|
|
ret |= st.st_mode&S_IRUSR ? mask&AVIO_FLAG_READ : 0;
|
|
ret |= st.st_mode&S_IWUSR ? mask&AVIO_FLAG_WRITE : 0;
|
|
|
|
return ret;
|
|
}
|
|
|
|
#if CONFIG_FILE_PROTOCOL
|
|
|
|
static int file_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
int access;
|
|
int fd;
|
|
|
|
av_strstart(filename, "file:", &filename);
|
|
|
|
if (flags & AVIO_FLAG_WRITE && flags & AVIO_FLAG_READ) {
|
|
access = O_CREAT | O_TRUNC | O_RDWR;
|
|
} else if (flags & AVIO_FLAG_WRITE) {
|
|
access = O_CREAT | O_TRUNC | O_WRONLY;
|
|
} else {
|
|
access = O_RDONLY;
|
|
}
|
|
#ifdef O_BINARY
|
|
access |= O_BINARY;
|
|
#endif
|
|
fd = open(filename, access, 0666);
|
|
if (fd == -1)
|
|
return AVERROR(errno);
|
|
h->priv_data = (void *) (intptr_t) fd;
|
|
return 0;
|
|
}
|
|
|
|
/* XXX: use llseek */
|
|
static int64_t file_seek(URLContext *h, int64_t pos, int whence)
|
|
{
|
|
int fd = (intptr_t) h->priv_data;
|
|
if (whence == AVSEEK_SIZE) {
|
|
struct stat st;
|
|
int ret = fstat(fd, &st);
|
|
return ret < 0 ? AVERROR(errno) : st.st_size;
|
|
}
|
|
return lseek(fd, pos, whence);
|
|
}
|
|
|
|
static int file_close(URLContext *h)
|
|
{
|
|
int fd = (intptr_t) h->priv_data;
|
|
return close(fd);
|
|
}
|
|
|
|
URLProtocol ff_file_protocol = {
|
|
.name = "file",
|
|
.url_open = file_open,
|
|
.url_read = file_read,
|
|
.url_write = file_write,
|
|
.url_seek = file_seek,
|
|
.url_close = file_close,
|
|
.url_get_file_handle = file_get_handle,
|
|
.url_check = file_check,
|
|
};
|
|
|
|
#endif /* CONFIG_FILE_PROTOCOL */
|
|
|
|
#if CONFIG_PIPE_PROTOCOL
|
|
|
|
static int pipe_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
int fd;
|
|
char *final;
|
|
av_strstart(filename, "pipe:", &filename);
|
|
|
|
fd = strtol(filename, &final, 10);
|
|
if((filename == final) || *final ) {/* No digits found, or something like 10ab */
|
|
if (flags & AVIO_FLAG_WRITE) {
|
|
fd = 1;
|
|
} else {
|
|
fd = 0;
|
|
}
|
|
}
|
|
#if HAVE_SETMODE
|
|
setmode(fd, O_BINARY);
|
|
#endif
|
|
h->priv_data = (void *) (intptr_t) fd;
|
|
h->is_streamed = 1;
|
|
return 0;
|
|
}
|
|
|
|
URLProtocol ff_pipe_protocol = {
|
|
.name = "pipe",
|
|
.url_open = pipe_open,
|
|
.url_read = file_read,
|
|
.url_write = file_write,
|
|
.url_get_file_handle = file_get_handle,
|
|
.url_check = file_check,
|
|
};
|
|
|
|
#endif /* CONFIG_PIPE_PROTOCOL */
|