diff mbox series

[FFmpeg-devel,1/2] avformat/file: Add a specialized url_check callback for pipe protocol

Message ID 20220829144320.330685-1-bpeeluk@yahoo.co.uk
State New
Headers show
Series [FFmpeg-devel,1/2] avformat/file: Add a specialized url_check callback for pipe protocol | expand

Checks

Context Check Description
yinshiyou/make_loongarch64 success Make finished
yinshiyou/make_fate_loongarch64 success Make fate finished
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

Neil Roberts Aug. 29, 2022, 2:43 p.m. UTC
Using file_check for the pipe protocol doesn't make sense. For example,
for a URL like “pipe:0” it would end up checking whether the “pipe:0”
file exists. This patch instead makes it check the access modes on the
corresponding file descriptor using fcntl.

Signed-off-by: Neil Roberts <bpeeluk@yahoo.co.uk>
---
 libavformat/file.c | 40 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

Comments

Michael Niedermayer Aug. 29, 2022, 9:54 p.m. UTC | #1
On Mon, Aug 29, 2022 at 04:43:19PM +0200, Neil Roberts wrote:
> Using file_check for the pipe protocol doesn't make sense. For example,
> for a URL like “pipe:0” it would end up checking whether the “pipe:0”
> file exists. This patch instead makes it check the access modes on the
> corresponding file descriptor using fcntl.
> 
> Signed-off-by: Neil Roberts <bpeeluk@yahoo.co.uk>
> ---
>  libavformat/file.c | 40 +++++++++++++++++++++++++++++++++++++---
>  1 file changed, 37 insertions(+), 3 deletions(-)

breaks on mingw64

src/libavformat/file.c: In function ‘pipe_check’:
src/libavformat/file.c:411:24: error: implicit declaration of function ‘fcntl’; did you mean ‘rintl’? [-Werror=implicit-function-declaration]
     int status_flags = fcntl(fd, F_GETFL);
                        ^~~~~
                        rintl
src/libavformat/file.c:411:34: error: ‘F_GETFL’ undeclared (first use in this function)
     int status_flags = fcntl(fd, F_GETFL);
                                  ^~~~~~~
src/libavformat/file.c:411:34: note: each undeclared identifier is reported only once for each function it appears in
cc1: some warnings being treated as errors

[...]
diff mbox series

Patch

diff --git a/libavformat/file.c b/libavformat/file.c
index 98c9e81bcb..f17d219cb2 100644
--- a/libavformat/file.c
+++ b/libavformat/file.c
@@ -375,9 +375,8 @@  const URLProtocol ff_file_protocol = {
 
 #if CONFIG_PIPE_PROTOCOL
 
-static int pipe_open(URLContext *h, const char *filename, int flags)
+static int pipe_get_fd(const char *filename, int flags)
 {
-    FileContext *c = h->priv_data;
     int fd;
     char *final;
     av_strstart(filename, "pipe:", &filename);
@@ -390,6 +389,14 @@  static int pipe_open(URLContext *h, const char *filename, int flags)
             fd = 0;
         }
     }
+
+    return fd;
+}
+
+static int pipe_open(URLContext *h, const char *filename, int flags)
+{
+    FileContext *c = h->priv_data;
+    int fd = pipe_get_fd(filename, flags);
 #if HAVE_SETMODE
     setmode(fd, O_BINARY);
 #endif
@@ -398,13 +405,40 @@  static int pipe_open(URLContext *h, const char *filename, int flags)
     return 0;
 }
 
+static int pipe_check(URLContext *h, int mask)
+{
+    int fd = pipe_get_fd(h->filename, mask);
+    int status_flags = fcntl(fd, F_GETFL);
+    int access;
+
+    if (status_flags == -1)
+        return AVERROR(errno);
+
+    switch (status_flags & O_ACCMODE) {
+    case O_RDONLY:
+        access = AVIO_FLAG_READ;
+        break;
+    case O_WRONLY:
+        access = AVIO_FLAG_WRITE;
+        break;
+    case O_RDWR:
+        access = AVIO_FLAG_READ | AVIO_FLAG_WRITE;
+        break;
+    default:
+        access = 0;
+        break;
+    }
+
+    return access & mask;
+}
+
 const 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,
+    .url_check           = pipe_check,
     .priv_data_size      = sizeof(FileContext),
     .priv_data_class     = &pipe_class,
     .default_whitelist   = "crypto,data"