h2load: Robust error handling in POST data

de3f2951 · Tatsuhiro Tsujikawa · d00788ce · de3f2951
Commit de3f2951 authored May 14, 2016 by Tatsuhiro Tsujikawa
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

src/h2load_http2_session.cc src/h2load_http2_session.cc +6 -1

No files found.
--- 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;