mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-11-26 19:01:44 +02:00
5b33a55376
Avoid the need to use seeking for getting the file size, use fstat instead, which is significantly faster. See thread: Subject: [FFmpeg-devel] [PATCH] Add support to AVSEEK_SIZE to the file protocol seek callback Date: Fri, 2 Apr 2010 13:13:27 +0200 Originally committed as revision 22799 to svn://svn.ffmpeg.org/ffmpeg/trunk
137 lines
3.2 KiB
C
137 lines
3.2 KiB
C
/*
|
|
* Buffered file io for ffmpeg system
|
|
* Copyright (c) 2001 Fabrice Bellard
|
|
*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg 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.
|
|
*
|
|
* FFmpeg 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 FFmpeg; 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_SETMODE
|
|
#include <io.h>
|
|
#endif
|
|
#include <unistd.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/time.h>
|
|
#include <stdlib.h>
|
|
#include "os_support.h"
|
|
|
|
|
|
/* standard file protocol */
|
|
|
|
static int file_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
int access;
|
|
int fd;
|
|
|
|
av_strstart(filename, "file:", &filename);
|
|
|
|
if (flags & URL_RDWR) {
|
|
access = O_CREAT | O_TRUNC | O_RDWR;
|
|
} else if (flags & URL_WRONLY) {
|
|
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(ENOENT);
|
|
h->priv_data = (void *) (intptr_t) fd;
|
|
return 0;
|
|
}
|
|
|
|
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, unsigned char *buf, int size)
|
|
{
|
|
int fd = (intptr_t) h->priv_data;
|
|
return write(fd, buf, size);
|
|
}
|
|
|
|
/* 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);
|
|
}
|
|
|
|
static int file_get_handle(URLContext *h)
|
|
{
|
|
return (intptr_t) h->priv_data;
|
|
}
|
|
|
|
URLProtocol file_protocol = {
|
|
"file",
|
|
file_open,
|
|
file_read,
|
|
file_write,
|
|
file_seek,
|
|
file_close,
|
|
.url_get_file_handle = file_get_handle,
|
|
};
|
|
|
|
/* 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 & URL_WRONLY) {
|
|
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 pipe_protocol = {
|
|
"pipe",
|
|
pipe_open,
|
|
file_read,
|
|
file_write,
|
|
.url_get_file_handle = file_get_handle,
|
|
};
|