From c84778a4e1e6a1123d117482a81fe925325ac010 Mon Sep 17 00:00:00 2001 From: Michal Sojka Date: Fri, 9 Apr 2010 16:12:10 +0200 Subject: [PATCH] Prepare for O_DIRECT --- libavformat/avio.h | 1 + libavformat/file.c | 26 +++++++++++++++++++++----- 2 files changed, 22 insertions(+), 5 deletions(-) diff --git a/libavformat/avio.h b/libavformat/avio.h index be02b06f6..3340a12cd 100644 --- a/libavformat/avio.h +++ b/libavformat/avio.h @@ -64,6 +64,7 @@ typedef struct URLPollEntry { #define URL_RDONLY 0 #define URL_WRONLY 1 #define URL_RDWR 2 +#define URL_DIRECT 2 /* Use O_DIRECT for file access */ typedef int URLInterruptCB(void); diff --git a/libavformat/file.c b/libavformat/file.c index da0ce1509..c041e562a 100644 --- a/libavformat/file.c +++ b/libavformat/file.c @@ -19,6 +19,7 @@ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +#define _GNU_SOURCE #include "libavutil/avstring.h" #include "avformat.h" #include @@ -33,8 +34,13 @@ /* standard file protocol */ +struct FileContext { + int fd; +}; + static int file_open(URLContext *h, const char *filename, int flags) { + struct FileContext *s = h->priv_data; int access; int fd; @@ -50,35 +56,45 @@ static int file_open(URLContext *h, const char *filename, int flags) #ifdef O_BINARY access |= O_BINARY; #endif + s = av_malloc(sizeof(struct FileContext)); + if (!s) + return AVERROR(ENOMEM); + if (flags & URL_DIRECT) + access |= O_DIRECT; fd = open(filename, access, 0666); if (fd < 0) return AVERROR(ENOENT); - h->priv_data = (void *) (intptr_t) fd; + s->fd = fd; + h->priv_data = (void *) s; return 0; } static int file_read(URLContext *h, unsigned char *buf, int size) { - int fd = (intptr_t) h->priv_data; + struct FileContext *s = h->priv_data; + int fd = s->fd; return read(fd, buf, size); } static int file_write(URLContext *h, unsigned char *buf, int size) { - int fd = (intptr_t) h->priv_data; + struct FileContext *s = h->priv_data; + int fd = s->fd; 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; + struct FileContext *s = h->priv_data; + int fd = s->fd; return lseek(fd, pos, whence); } static int file_close(URLContext *h) { - int fd = (intptr_t) h->priv_data; + struct FileContext *s = h->priv_data; + int fd = s->fd; return close(fd); } -- 2.39.2