소스 검색

Show correct end byte in error message when HTTP response range is not ok

Previously, unless HTTP pipelining is enabled, end byte in that
message is always 0. With this change, it shows correct end byte sent
to the HTTP server.
Tatsuhiro Tsujikawa 12 년 전
부모
커밋
b782a56b1c
3개의 변경된 파일17개의 추가작업 그리고 4개의 파일을 삭제
  1. 4 2
      src/HttpRequest.cc
  2. 7 0
      src/HttpRequest.h
  3. 6 2
      src/HttpRequestCommand.cc

+ 4 - 2
src/HttpRequest.cc

@@ -96,6 +96,8 @@ int64_t HttpRequest::getEndByte() const
       int64_t endByte =
         fileEntry_->gtoloff(segment_->getPosition()+segment_->getLength()-1);
       return std::min(endByte, fileEntry_->getLength()-1);
+    } else if(endOffsetOverride_ > 0) {
+      return endOffsetOverride_ - 1;
     } else {
       return 0;
     }
@@ -201,9 +203,9 @@ std::string HttpRequest::createRequest()
     std::string rangeHeader(fmt("bytes=%" PRId64 "-", getStartByte()));
     if(request_->isPipeliningEnabled()) {
       rangeHeader += util::itos(getEndByte());
-    } else if(getProtocol() != "ftp" && endOffsetOverride_ > 0) {
+    } else if(getEndByte() > 0) {
       // FTP via http proxy does not support endbytes
-      rangeHeader += util::itos(endOffsetOverride_-1);
+      rangeHeader += util::itos(getEndByte());
     }
     builtinHds.push_back(std::make_pair("Range:", rangeHeader));
   }

+ 7 - 0
src/HttpRequest.h

@@ -88,6 +88,13 @@ private:
 
   bool acceptGzip_;
 
+  // Historically, aria2 did not specify end byte marker unless http
+  // pipelining is enabled. Sometimes end byte is known because the
+  // segment/piece ahead of this request was already acquired. In this
+  // case, specifying end byte enables to reuse connection. To achieve
+  // this, if endOffsetOverride_ is more than 0, its value - 1 is used
+  // as an end byte. Please note that FTP protocol cannot specify end
+  // bytes and it is also true if it is used via HTTP proxy.
   int64_t endOffsetOverride_;
 
   std::string ifModSinceHeader_;

+ 6 - 2
src/HttpRequestCommand.cc

@@ -115,7 +115,9 @@ createHttpRequest(const SharedHandle<Request>& req,
   } else {
     httpRequest->disableNoCache();
   }
-  httpRequest->setEndOffsetOverride(endOffset);
+  if(endOffset > 0) {
+    httpRequest->setEndOffsetOverride(endOffset);
+  }
   return httpRequest;
 }
 } // namespace
@@ -181,7 +183,9 @@ bool HttpRequestCommand::executeInternal() {
         const SharedHandle<Segment>& segment = *itr;
         if(!httpConnection_->isIssued(segment)) {
           int64_t endOffset = 0;
-          if(getRequestGroup()->getTotalLength() > 0 && getPieceStorage()) {
+          // FTP via HTTP proxy does not support end byte marker
+          if(getRequest()->getProtocol() != "ftp" &&
+             getRequestGroup()->getTotalLength() > 0 && getPieceStorage()) {
             size_t nextIndex =
               getPieceStorage()->getNextUsedIndex(segment->getIndex());
             endOffset = std::min