From patchwork Sun Feb 7 16:41:16 2021 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Zhao Zhili X-Patchwork-Id: 25482 Return-Path: X-Original-To: patchwork@ffaux-bg.ffmpeg.org Delivered-To: patchwork@ffaux-bg.ffmpeg.org Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by ffaux.localdomain (Postfix) with ESMTP id 20FE344B510 for ; Sun, 7 Feb 2021 18:41:34 +0200 (EET) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id E71156881BB; Sun, 7 Feb 2021 18:41:33 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from qq.com (out203-205-221-242.mail.qq.com [203.205.221.242]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id F132F688128 for ; Sun, 7 Feb 2021 18:41:26 +0200 (EET) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=foxmail.com; s=s201512; t=1612716082; bh=7MzByxGElGz7mGQ0GRmYZhiZJ9YeIIQJcyBoneTMbb8=; h=From:To:Cc:Subject:Date; b=yvbH2ry1sLG+/pYicOA+vzERMjw98fQir/3BDoX7RtN+crXoKxuwXRjTCNNNDT9hF FZW9+RboMf0IDC8uX8os9hZOxxJ3gTGbtyAvabTpLYWLo3Kit/aPwC5p22OvvdHrv2 MF5xlU2k+8gVg7yPgahNTNqJuvZ35J1P5rACc4dM= Received: from localhost.localdomain ([27.46.106.137]) by newxmesmtplogicsvrszb6.qq.com (NewEsmtp) with SMTP id A539C29D; Mon, 08 Feb 2021 00:41:19 +0800 X-QQ-mid: xmsmtpt1612716079tp5g6dj8c Message-ID: X-QQ-XMAILINFO: N9C/Vpkk5whXOJwmLFpZ5Des4/kepfDYSfWIQ1XXpmKf6eURnzTDuqB69uFlY9 8be9G3pebNl4jdt7MaFJQ0UoQrY4x7trKszD5k2xKl3Q4Uj1jIm+ciCzgfPvMTf2KzFiqv4CWuz8 zW/1rA/AQXZpmLHlfmkrqH8kyP6siDlKqiBaIdP5M+1cWQTX/kGBCM6+qX2VSswPHpU33Jwr0b5M VR8gFnmPbDwEHgwcVmjn9txo1j6A2R6zGf3y3UX0wNvi3y0lblqyG0i9l0V3HXjfrLkg3j5NX6lO JCNKu1+UDF6GW+Pydyd1wmBJ68eGYUMBlud6ItgKhDvSEindnsnzx6+g7hhF1fCk9B48Q8TXW1f/ /abf9bCNqKYkc401CrdCOy5c28edskcFdF9E/+NWuZKOVy/innSrAPSK9hdYAJ+BvYRSPODkk64x vzYZTkAmaxYM3tcVQ3M4wQJQbRqKPLNMPQtDWGUMkR2ZB91idC6tAPytUnXeWgD53cfvTvNGyp9/ v8z+iwDNS+uUn4UqXKVrzWzJwUHc5LpLo/M7U5/HZhl0DST6dLHxPM/rmNMGehosVM5qEKaohr/d NRypmaheM341BTdApQh03fz3ERrSDeyQXJjeiZoo0H+oA9CR7DJSDU6WHdTUTowDmMiUQnptMWBM nL67UyZ2ragqr8hErVKU1A6IJofOx1A0ZDezKIs3YnWINTIOKyYB8GIpbPFnhKjNEfYxg4KbZmZu Sa8eolLg/5Ec3tSHo5DwgYJHoHf+ejpr3lz/Yi4l1z61LEgHiCtBcQiwFn9516e9fywf3b1tQpJS LX7ZO2uPNGJgaHYgS2zW1S From: Zhao Zhili To: ffmpeg-devel@ffmpeg.org Date: Mon, 8 Feb 2021 00:41:16 +0800 X-OQ-MSGID: <20210207164117.4867-1-quinkblack@foxmail.com> X-Mailer: git-send-email 2.27.0 MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH 1/2] lavf/network: add wait fd callback to ff_network_wait_fd_timeout X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Zhao Zhili Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" --- libavformat/libamqp.c | 4 ++-- libavformat/network.c | 16 ++++++++++++++-- libavformat/network.h | 14 +++++++++++++- libavformat/tcp.c | 6 ++++-- 4 files changed, 33 insertions(+), 7 deletions(-) diff --git a/libavformat/libamqp.c b/libavformat/libamqp.c index c3b9c484ea..ca7d3ab70f 100644 --- a/libavformat/libamqp.c +++ b/libavformat/libamqp.c @@ -241,7 +241,7 @@ static int amqp_proto_write(URLContext *h, const unsigned char *buf, int size) amqp_bytes_t message = { size, (void *)buf }; amqp_basic_properties_t props; - ret = ff_network_wait_fd_timeout(fd, 1, h->rw_timeout, &h->interrupt_callback); + ret = ff_network_wait_fd_timeout(fd, 1, h->rw_timeout, &h->interrupt_callback, NULL); if (ret) return ret; @@ -270,7 +270,7 @@ static int amqp_proto_read(URLContext *h, unsigned char *buf, int size) amqp_rpc_reply_t broker_reply; amqp_envelope_t envelope; - ret = ff_network_wait_fd_timeout(fd, 0, h->rw_timeout, &h->interrupt_callback); + ret = ff_network_wait_fd_timeout(fd, 0, h->rw_timeout, &h->interrupt_callback, NULL); if (ret) return ret; diff --git a/libavformat/network.c b/libavformat/network.c index 0f5a575f77..48a9af012e 100644 --- a/libavformat/network.c +++ b/libavformat/network.c @@ -75,15 +75,27 @@ int ff_network_wait_fd(int fd, int write) return ret < 0 ? ff_neterrno() : p.revents & (ev | POLLERR | POLLHUP) ? 0 : AVERROR(EAGAIN); } -int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, AVIOInterruptCB *int_cb) +static int network_wait_fd_callback(int fd, int write, void *opaque) { + return ff_network_wait_fd(fd, write); +} + +int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, + AVIOInterruptCB *int_cb, const NetworkWaitFdCB *wait_cb) { int ret; int64_t wait_start = 0; + const static NetworkWaitFdCB wait_cb_internal = { + .wait_fd = network_wait_fd_callback, + }; + + if (!wait_cb) + wait_cb = &wait_cb_internal; + while (1) { if (ff_check_interrupt(int_cb)) return AVERROR_EXIT; - ret = ff_network_wait_fd(fd, write); + ret = ff_network_wait_fd_cb(wait_cb, fd, write); if (ret != AVERROR(EAGAIN)) return ret; if (timeout > 0) { diff --git a/libavformat/network.h b/libavformat/network.h index 71347e815b..1a02a6d9ee 100644 --- a/libavformat/network.h +++ b/libavformat/network.h @@ -84,6 +84,16 @@ void ff_network_close(void); int ff_tls_init(void); void ff_tls_deinit(void); +typedef struct NetworkWaitFdCB { + int (*wait_fd)(int /*fd*/, int /*write*/, void* /*opaque*/); + void *opaque; +} NetworkWaitFdCB; + +static av_always_inline int ff_network_wait_fd_cb(const NetworkWaitFdCB *cb, + int fd, int write) { + return cb->wait_fd(fd, write, cb->opaque); +} + int ff_network_wait_fd(int fd, int write); /** @@ -94,9 +104,11 @@ int ff_network_wait_fd(int fd, int write); * @param write Set 1 to wait for socket able to be read, 0 to be written * @param timeout Timeout interval, in microseconds. Actual precision is 100000 mcs, due to ff_network_wait_fd usage * @param int_cb Interrupt callback, is checked before each ff_network_wait_fd call + * @param wait_cb Wait callback, default implementation is used if NULL * @return 0 if data can be read/written, AVERROR(ETIMEDOUT) if timeout expired, or negative error code */ -int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, AVIOInterruptCB *int_cb); +int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, + AVIOInterruptCB *int_cb, const NetworkWaitFdCB *wait_cb); /** * Waits for up to 'timeout' microseconds. If the usert's int_cb is set and diff --git a/libavformat/tcp.c b/libavformat/tcp.c index 2198e0f00e..437eaecbb4 100644 --- a/libavformat/tcp.c +++ b/libavformat/tcp.c @@ -237,7 +237,8 @@ static int tcp_read(URLContext *h, uint8_t *buf, int size) int ret; if (!(h->flags & AVIO_FLAG_NONBLOCK)) { - ret = ff_network_wait_fd_timeout(s->fd, 0, h->rw_timeout, &h->interrupt_callback); + ret = ff_network_wait_fd_timeout(s->fd, 0, h->rw_timeout, + &h->interrupt_callback, NULL); if (ret) return ret; } @@ -253,7 +254,8 @@ static int tcp_write(URLContext *h, const uint8_t *buf, int size) int ret; if (!(h->flags & AVIO_FLAG_NONBLOCK)) { - ret = ff_network_wait_fd_timeout(s->fd, 1, h->rw_timeout, &h->interrupt_callback); + ret = ff_network_wait_fd_timeout(s->fd, 1, h->rw_timeout, + &h->interrupt_callback, NULL); if (ret) return ret; }