git.sesse.net Git - ffmpeg/blob - libavformat/file.c

   1 /*
   2  * buffered file I/O
   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_IO_H
  26 #include <io.h>
  27 #endif
  28 #if HAVE_UNISTD_H
  29 #include <unistd.h>
  30 #endif
  31 #include <sys/stat.h>
  32 #include <stdlib.h>
  33 #include "os_support.h"
  34 #include "url.h"
  35
  36
  37 /* standard file protocol */
  38
  39 static int file_read(URLContext *h, unsigned char *buf, int size)
  40 {
  41     int fd = (intptr_t) h->priv_data;
  42     return read(fd, buf, size);
  43 }
  44
  45 static int file_write(URLContext *h, const unsigned char *buf, int size)
  46 {
  47     int fd = (intptr_t) h->priv_data;
  48     return write(fd, buf, size);
  49 }
  50
  51 static int file_get_handle(URLContext *h)
  52 {
  53     return (intptr_t) h->priv_data;
  54 }
  55
  56 static int file_check(URLContext *h, int mask)
  57 {
  58     struct stat st;
  59     int ret = stat(h->filename, &st);
  60     if (ret < 0)
  61         return AVERROR(errno);
  62
  63     ret |= st.st_mode&S_IRUSR ? mask&AVIO_FLAG_READ  : 0;
  64     ret |= st.st_mode&S_IWUSR ? mask&AVIO_FLAG_WRITE : 0;
  65
  66     return ret;
  67 }
  68
  69 #if CONFIG_FILE_PROTOCOL
  70
  71 static int file_open(URLContext *h, const char *filename, int flags)
  72 {
  73     int access;
  74     int fd;
  75
  76     av_strstart(filename, "file:", &filename);
  77
  78     if (flags & AVIO_FLAG_WRITE && flags & AVIO_FLAG_READ) {
  79         access = O_CREAT | O_TRUNC | O_RDWR;
  80     } else if (flags & AVIO_FLAG_WRITE) {
  81         access = O_CREAT | O_TRUNC | O_WRONLY;
  82     } else {
  83         access = O_RDONLY;
  84     }
  85 #ifdef O_BINARY
  86     access |= O_BINARY;
  87 #endif
  88     fd = open(filename, access, 0666);
  89     if (fd == -1)
  90         return AVERROR(errno);
  91     h->priv_data = (void *) (intptr_t) fd;
  92     return 0;
  93 }
  94
  95 /* XXX: use llseek */
  96 static int64_t file_seek(URLContext *h, int64_t pos, int whence)
  97 {
  98     int fd = (intptr_t) h->priv_data;
  99     if (whence == AVSEEK_SIZE) {
 100         struct stat st;
 101         int ret = fstat(fd, &st);
 102         return ret < 0 ? AVERROR(errno) : st.st_size;
 103     }
 104     return lseek(fd, pos, whence);
 105 }
 106
 107 static int file_close(URLContext *h)
 108 {
 109     int fd = (intptr_t) h->priv_data;
 110     return close(fd);
 111 }
 112
 113 URLProtocol ff_file_protocol = {
 114     .name                = "file",
 115     .url_open            = file_open,
 116     .url_read            = file_read,
 117     .url_write           = file_write,
 118     .url_seek            = file_seek,
 119     .url_close           = file_close,
 120     .url_get_file_handle = file_get_handle,
 121     .url_check           = file_check,
 122 };
 123
 124 #endif /* CONFIG_FILE_PROTOCOL */
 125
 126 #if CONFIG_PIPE_PROTOCOL
 127
 128 static int pipe_open(URLContext *h, const char *filename, int flags)
 129 {
 130     int fd;
 131     char *final;
 132     av_strstart(filename, "pipe:", &filename);
 133
 134     fd = strtol(filename, &final, 10);
 135     if((filename == final) || *final ) {/* No digits found, or something like 10ab */
 136         if (flags & AVIO_FLAG_WRITE) {
 137             fd = 1;
 138         } else {
 139             fd = 0;
 140         }
 141     }
 142 #if HAVE_SETMODE
 143     setmode(fd, O_BINARY);
 144 #endif
 145     h->priv_data = (void *) (intptr_t) fd;
 146     h->is_streamed = 1;
 147     return 0;
 148 }
 149
 150 URLProtocol ff_pipe_protocol = {
 151     .name                = "pipe",
 152     .url_open            = pipe_open,
 153     .url_read            = file_read,
 154     .url_write           = file_write,
 155     .url_get_file_handle = file_get_handle,
 156     .url_check           = file_check,
 157 };
 158
 159 #endif /* CONFIG_PIPE_PROTOCOL */