Tatsuhiro Tsujikawa 20 år sedan
förälder
incheckning
21ca3f1f6f
2 ändrade filer med 4 tillägg och 2 borttagningar
  1. 3 1
      src/HttpResponseCommand.cc
  2. 1 1
      src/Util.cc

+ 3 - 1
src/HttpResponseCommand.cc

@@ -37,7 +37,7 @@ HttpResponseCommand::~HttpResponseCommand() {
 }
 
 bool HttpResponseCommand::executeInternal(Segment seg) {
-  if(SEGMENT_EQUAL(req->seg, seg) == false) {
+  if(req->seg.sp != seg.sp) {
     e->logger->info(MSG_SEGMENT_CHANGED, cuid);
     return prepareForRetry(0);
   }
@@ -104,6 +104,8 @@ bool HttpResponseCommand::handleDefaultEncoding(const HttpHeader& headers) {
     return prepareForRetry(0);
   } else {
     e->segmentMan->totalSize = size;
+    Segment seg;
+    e->segmentMan->getSegment(seg, cuid);	
     e->diskWriter->initAndOpenFile(e->segmentMan->getFilePath());
     createHttpDownloadCommand();
     return true;

+ 1 - 1
src/Util.cc

@@ -122,7 +122,7 @@ bool Util::endsWith(string target, string part) {
   if(part == "") {
     return true;
   }
-  if(target.find(part) == target.size()-part.size()) {
+  if(target.rfind(part) == target.size()-part.size()) {
     return true;
   } else {
     return false;