From de3f2951b3f6327c898cf7570ec9a0ea6fdb8109 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Sat, 14 May 2016 00:40:35 +0900
Subject: [PATCH] h2load: Robust error handling in POST data

---
 src/h2load_http2_session.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/h2load_http2_session.cc b/src/h2load_http2_session.cc
index 7883131d..ea71d66a 100644
--- a/src/h2load_http2_session.cc
+++ b/src/h2load_http2_session.cc
@@ -149,8 +149,13 @@ ssize_t file_read_callback(nghttp2_session *session, int32_t stream_id,
 
   req_stat->data_offset += nread;
 
-  if (nread == 0 || req_stat->data_offset == config->data_length) {
+  if (req_stat->data_offset == config->data_length) {
     *data_flags |= NGHTTP2_DATA_FLAG_EOF;
+    return nread;
+  }
+
+  if (req_stat->data_offset > config->data_length || nread == 0) {
+    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
   }
 
   return nread;