Fix simple http conn

PikaLabs · Jul 2, 2017 · 6de9cab · 6de9cab
1 parent 4333bf6
commit 6de9cab
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/pink/src/simple_http_conn.cc b/pink/src/simple_http_conn.cc
@@ -323,15 +323,16 @@ bool SimpleHTTPConn::BuildRequestHeader() {
   if (!request_->ParseHeadFromArray(rbuf_, header_len_)) {
     return false;  
   }
-  auto iter = request_->headers.begin();
-  iter = request_->headers.find("content-length");
+  auto iter = request_->headers.find("content-length");
   if (iter == request_->headers.end()) {
     remain_packet_len_ = 0;
   } else {
-    int64_t tmp = 0;
-    slash::string2l(iter->second.data(), iter->second.size(),
-        static_cast<long*>(&tmp));
-    remain_packet_len_ = tmp;
+    long tmp = 0;
+    if (slash::string2l(iter->second.data(), iter->second.size(), &tmp)) {
+      remain_packet_len_ = tmp;
+    } else {
+      remain_packet_len_ = 0;
+    }
   }
 
   if (rbuf_pos_ > header_len_) {