From 42f7b969536704e5d65b1ca5075a8fdddd97529e Mon Sep 17 00:00:00 2001
From: Keyun Tong <ktong@fb.com>
Date: Sat, 29 Aug 2020 08:17:09 -0700
Subject: [PATCH] Enable rtmp timeout in decoder

Summary:
* Link libav change into fbcode
* Set rw_timeout value

Differential Revision: D23412524

fbshipit-source-id: 5755950be1b1b4c37cb0c3a69a8c875f8862a92c
---
 torchvision/csrc/cpu/decoder/decoder.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/torchvision/csrc/cpu/decoder/decoder.cpp b/torchvision/csrc/cpu/decoder/decoder.cpp
index e9d1acaa3e0..6281cea3292 100644
--- a/torchvision/csrc/cpu/decoder/decoder.cpp
+++ b/torchvision/csrc/cpu/decoder/decoder.cpp
@@ -303,11 +303,14 @@ bool Decoder::init(
   inputCtx_->flags |= AVFMT_FLAG_NONBLOCK;
 
   AVDictionary* options = nullptr;
-  av_dict_set_int(&options, "analyzeduration", params_.timeoutMs * 1000, 0);
-  av_dict_set_int(&options, "stimeout", params_.timeoutMs * 1000, 0);
   if (params_.listen) {
     av_dict_set_int(&options, "listen", 1, 0);
   }
+  if (params_.timeoutMs > 0) {
+    av_dict_set_int(&options, "analyzeduration", params_.timeoutMs * 1000, 0);
+    av_dict_set_int(&options, "stimeout", params_.timeoutMs * 1000, 0);
+    av_dict_set_int(&options, "rw_timeout", params_.timeoutMs * 1000, 0);
+  }
 
   interrupted_ = false;