2007-11-18 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>

Removed.
	* src/PiecedSegment.{h, cc} (operator==)(operator!=)

	Use Segment::operator==()
	* src/HttpConnection.cc
	* src/Segment.h

	Updated usage
	* src/version_usage.cc

	Added EINTR treatment.
	* src/SocketCore.cc
	* src/AbstractDiskWriter.cc

	Rewritten.
	* src/Util.cc (rangedFileCopy)

ChangeLog

@@ -1,3 +1,22 @@
+2007-11-18  Tatsuhiro Tsujikawa  <tujikawa at rednoah dot com>
+
+	Removed.
+	* src/PiecedSegment.{h, cc} (operator==)(operator!=)
+
+	Use Segment::operator==()
+	* src/HttpConnection.cc
+	* src/Segment.h
+
+	Updated usage
+	* src/version_usage.cc
+
+	Added EINTR treatment.
+	* src/SocketCore.cc
+	* src/AbstractDiskWriter.cc
+
+	Rewritten.
+	* src/Util.cc (rangedFileCopy)
+	
 2007-11-17  Tatsuhiro Tsujikawa  <tujikawa at rednoah dot com>
 
 	Fixed typo.

src/AbstractDiskWriter.cc

@@ -99,12 +99,23 @@ void AbstractDiskWriter::createFile(const string& filename, int32_t addFlags)
 
 int32_t AbstractDiskWriter::writeDataInternal(const unsigned char* data, int32_t len)
 {
-  return write(fd, data, len);
+  int32_t writtenLength = 0;
+  while(writtenLength < len) {
+    int32_t ret = 0;
+    while((ret = write(fd, data+writtenLength, len-writtenLength)) == -1 && errno == EINTR);
+    if(ret == -1) {
+      return -1;
+    }
+    writtenLength += ret;
+  }
+  return writtenLength;
 }
 
 int32_t AbstractDiskWriter::readDataInternal(unsigned char* data, int32_t len)
 {
-  return read(fd, data, len);
+  int32_t ret = 0;
+  while((ret = read(fd, data, len)) == -1 && errno == EINTR);
+  return ret;
 }
 
 void AbstractDiskWriter::seek(int64_t offset)

src/BtSetup.cc

@@ -65,8 +65,6 @@ Commands BtSetup::setup(RequestGroup* requestGroup,
   if(btContext.isNull()) {
     return commands;
   }
-  // TODO following process is moved to BtSetup
-
   // commands
   commands.push_back(new TrackerWatcherCommand(CUIDCounterSingletonHolder::instance()->newID(),
 					       requestGroup,

src/ByteArrayDiskWriter.cc

@@ -82,7 +82,6 @@ int32_t ByteArrayDiskWriter::readData(unsigned char* data, int32_t len, int64_t
 {
   buf.seekg(position, ios::beg);
   buf.read(reinterpret_cast<char*>(data), len);
-  // TODO we have to call buf.clear() here? YES
   buf.clear();
   return buf.gcount();
 }

src/DownloadCommand.cc

@@ -79,7 +79,6 @@ DownloadCommand::~DownloadCommand() {
 }
 
 bool DownloadCommand::executeInternal() {
-  // TODO we need to specify the sum of all segmentMan's download speed here.
   if(maxDownloadSpeedLimit > 0 &&
      maxDownloadSpeedLimit < _requestGroup->getSegmentMan()->calculateDownloadSpeed()) {
     e->commands.push_back(this);

src/FileAllocationDispatcherCommand.cc

@@ -57,7 +57,6 @@ bool FileAllocationDispatcherCommand::execute()
   if(!_e->_fileAllocationMan->isFileAllocationBeingExecuted() &&
      _e->_fileAllocationMan->nextFileAllocationEntryExists()) {
     FileAllocationEntryHandle entry = _e->_fileAllocationMan->popNextFileAllocationEntry();
-    // TODO we have to change message
     int32_t newCUID = CUIDCounterSingletonHolder::instance()->newID();
     logger->info(MSG_FILE_ALLOCATION_DISPATCH, newCUID);
     FileAllocationCommand* command =

src/GrowSegment.cc

@@ -57,4 +57,3 @@ PieceHandle GrowSegment::getPiece() const
 {
   return _piece;
 }
-

src/HttpConnection.cc

@@ -129,11 +129,9 @@ bool HttpConnection::isIssued(const SegmentHandle& segment) const
   for(HttpRequestEntries::const_iterator itr = outstandingHttpRequests.begin();
       itr != outstandingHttpRequests.end(); ++itr) {
     HttpRequestHandle httpRequest = (*itr)->getHttpRequest();
-    // TODO fix this using operator==
+    if(httpRequest->getSegment() == segment) {
-    if(httpRequest->getSegment().get() == segment.get()) {
       return true;
     }
   }
   return false;
 }
-

src/PiecedSegment.cc

@@ -95,13 +95,3 @@ PieceHandle PiecedSegment::getPiece() const
 {
   return _piece;
 }
-
-bool PiecedSegment::operator==(const PiecedSegment& segment) const
-{
-  return _piece == segment._piece;
-}
-
-bool PiecedSegment::operator!=(const PiecedSegment& segment) const
-{
-  return !(*this == segment);
-}

src/PiecedSegment.h

@@ -83,11 +83,6 @@ public:
   virtual void clear();
 
   virtual PieceHandle getPiece() const;
-
-  bool operator==(const PiecedSegment& segment) const;
-
-  bool operator!=(const PiecedSegment& segment) const;
-
 };
 
 typedef SharedHandle<PiecedSegment> PiecedSegmentHandle;

src/Segment.h

@@ -65,6 +65,11 @@ public:
   virtual void clear() = 0;
 
   virtual PieceHandle getPiece() const = 0;
+
+  bool operator==(const Segment& segment) const
+  {
+    return getIndex() == segment.getIndex();
+  }
 };
 
 typedef SharedHandle<Segment> SegmentHandle;

src/SocketCore.cc

@@ -95,7 +95,7 @@ void SocketCore::beginListen(int32_t port)
 #ifdef __MINGW32__
     ::closesocket(sockfd);
 #else
-    close(sockfd);
+    while(close(sockfd) == -1 && errno == EINTR);
 #endif // __MINGW32__
     sockfd = -1;
     throw new DlAbortEx(EX_SOCKET_SET_OPT, errorMsg());
@@ -124,10 +124,10 @@ SocketCore* SocketCore::acceptConnection() const
   socklen_t len = sizeof(sockaddr);
   memset((char*)&sockaddr, 0, sizeof(sockaddr));
   int32_t fd;
-  if((fd = accept(sockfd, (struct sockaddr*)&sockaddr, &len)) == -1) {
+  while((fd = accept(sockfd, (struct sockaddr*)&sockaddr, &len)) == -1 && errno == EINTR);
+  if(fd == -1) {
     throw new DlAbortEx(EX_SOCKET_ACCEPT, errorMsg());
   }
-
   SocketCore* s = new SocketCore(fd);
   return s;
 }
@@ -165,7 +165,7 @@ void SocketCore::establishConnection(const string& host, int32_t port)
   }
   SOCKOPT_T sockopt = 1;
   if(setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(socklen_t)) < 0) {
-    close(sockfd);
+    while(close(sockfd) == -1 && errno == EINTR);
     sockfd = -1;
     throw new DlAbortEx(EX_SOCKET_SET_OPT, errorMsg());
   }
@@ -252,7 +252,7 @@ void SocketCore::closeConnection()
 #ifdef __MINGW32__
     ::closesocket(sockfd);
 #else
-    close(sockfd);
+    while(close(sockfd) == -1 && errno == EINTR);
 #endif // __MINGW32__
     sockfd = -1;
   }
@@ -331,7 +331,9 @@ void SocketCore::writeData(const char* data, int32_t len)
   int32_t ret = 0;
 
   if(!secure) {
-    if((ret = send(sockfd, data, (size_t)len, 0)) != len) {
+    while((ret = send(sockfd, data, (size_t)len, 0)) == -1 && errno == EINTR);
+    // TODO assuming Blocking mode.
+    if(ret != len) {
       throw new DlRetryEx(EX_SOCKET_SEND, errorMsg());
     }
   } else {
@@ -355,7 +357,8 @@ void SocketCore::readData(char* data, int32_t& len)
   int32_t ret = 0;
 
   if(!secure) {    
-    if ((ret = recv(sockfd, data, (size_t)len, 0)) < 0) {
+    while((ret = recv(sockfd, data, (size_t)len, 0)) == -1 && errno == EINTR);
+    if(ret == -1) {
       throw new DlRetryEx(EX_SOCKET_RECV, errorMsg());
     }
   } else {
@@ -381,7 +384,8 @@ void SocketCore::peekData(char* data, int32_t& len)
   int32_t ret = 0;
 
   if(!secure) {
-    if ((ret = recv(sockfd, data, (size_t)len, MSG_PEEK)) < 0) {
+    while((ret = recv(sockfd, data, (size_t)len, MSG_PEEK)) == -1 && errno == EINTR);
+    if(ret == -1) {
       throw new DlRetryEx(EX_SOCKET_PEEK, errorMsg());
     }
   } else {

src/Util.cc

@@ -40,6 +40,7 @@
 #include "a2time.h"
 #include "DlAbortEx.h"
 #include "BitfieldMan.h"
+#include "DefaultDiskWriter.h"
 #include <ctype.h>
 #include <errno.h>
 #include <fcntl.h>
@@ -343,51 +344,34 @@ void Util::fileCopy(const string& dest, const string& src) {
 void Util::rangedFileCopy(const string& dest, const string& src, int64_t srcOffset, int64_t length)
 {
   int32_t bufSize = 4096;
-  char buf[bufSize];
+  unsigned char buf[bufSize];
-  int32_t destFd = -1;
+  DefaultDiskWriter srcdw;
-  int32_t srcFd = -1;
+  DefaultDiskWriter destdw;
-  try {
+
-    if((destFd = open(dest.c_str(), O_CREAT|O_WRONLY|O_TRUNC|O_BINARY, OPEN_MODE)) == -1) {
+  srcdw.openExistingFile(src);
-      throw new DlAbortEx(EX_FILE_OPEN, dest.c_str(), strerror(errno));
+  destdw.initAndOpenFile(dest);
-    }
+
-    if((srcFd = open(src.c_str(), O_RDONLY|O_BINARY, OPEN_MODE)) == -1) {
-      throw new DlAbortEx(EX_FILE_OPEN, src.c_str(), strerror(errno));
-    }
-    if(lseek(srcFd, srcOffset, SEEK_SET) != srcOffset) {
-      throw new DlAbortEx(EX_FILE_SEEK, src.c_str(), strerror(errno));
-    }
   int32_t x = length/bufSize;
   int32_t r = length%bufSize;
-    for(int32_t i = 0; i < x; i++) {
+
-      int32_t readLength;
+  int64_t initialOffset = srcOffset;
-      if((readLength = read(srcFd, buf, bufSize)) == -1 || readLength != bufSize) {
+  for(int32_t i = 0; i < x; ++i) {
-	throw new DlAbortEx(EX_FILE_READ, src.c_str(), strerror(errno));
+    int32_t readLength = 0;
-      }
+    while(readLength < bufSize) {
-      if(write(destFd, buf, readLength) == -1) {
+      int32_t ret = srcdw.readData(buf, bufSize-readLength, srcOffset);
-	throw new DlAbortEx(EX_FILE_WRITE, dest.c_str(), strerror(errno));
+      destdw.writeData(buf, ret, srcOffset-initialOffset);
+      srcOffset += ret;
+      readLength += ret;
     }
   }
   if(r > 0) {
-      int32_t readLength;
+    int32_t readLength = 0;
-      if((readLength = read(srcFd, buf, r)) == -1 || readLength != r) {
+    while(readLength < r) { 
-	throw new DlAbortEx(EX_FILE_READ, src.c_str(), strerror(errno));
+      int32_t ret = srcdw.readData(buf, r-readLength, srcOffset);
+      destdw.writeData(buf, ret, srcOffset-initialOffset);
+      srcOffset += ret;
+      readLength += ret;
     }
-      if(write(destFd, buf, r) == -1) {
-	throw new DlAbortEx(EX_FILE_WRITE, dest.c_str(), strerror(errno));
-      }
-    }
-    close(srcFd);
-    close(destFd);
-    srcFd = -1;
-    destFd = -1;
-  } catch(RecoverableException* e) {
-    if(srcFd != -1) {
-      close(srcFd);
-    }
-    if(destFd != -1) {
-      close(destFd);
-    }
-    throw;
   }     
 }
 

src/version_usage.cc

@@ -210,7 +210,8 @@ void showUsage() {
 	    "                              the same used by Netscape and Mozilla.") << endl;
 #if defined ENABLE_BITTORRENT || ENABLE_METALINK
   cout << _(" -S, --show-files             Print file listing of .torrent or .metalink file\n"
-	    "                              and exit.") << endl;
+	    "                              and exit. More detailed information will be listed\n"
+	    "                              in case of torrent file.") << endl;
   cout << _(" --select-file=INDEX...       Set file to download by specifing its index.\n"
 	    "                              You can find the file index using the\n"
 	    "                              --show-files option. Multiple indexes can be\n"
@@ -270,8 +271,12 @@ void showUsage() {
   cout << _(" -h, --help                   Print this message and exit.") << endl;
   cout << endl;
   cout << "URL:" << endl;
-  cout << _(" You can specify multiple URLs. All URLs must point to the same file\n"
+  cout << _(" You can specify multiple URLs. Unless you specify -Z option, all URLs must\n"
-	    " or downloading will fail.") << endl;
+	    " point to the same file or downloading will fail.") << endl;
+  cout << _(" You can specify both torrent file with -T option and URLs. By doing this,\n"
+	    " download a file from both torrent swarm and http/ftp server at the same time,\n"
+	    " while the data from http/ftp are uploaded to the torrent swarm. Note that\n"
+	    " only single file torrent can be integrated with http/ftp.") << endl;
   cout << endl;
   cout << _("Examples:") << endl;
   cout << _(" Download a file using 1 connection:") << endl;
@@ -296,6 +301,8 @@ void showUsage() {
   cout << "  aria2c -T test.torrent dir/file1.zip dir/file2.zip" << endl;
   cout << _(" Print file listing of .torrent file:") << endl;
   cout << "  aria2c -T test.torrent -S" << endl;
+  cout << _(" Download a file using torrent and http/ftp server") << endl;
+  cout << "  aria2c -T test.torrent http://host1/file ftp://host2/file" << endl;
 #endif // ENABLE_BITTORRENT
 #ifdef ENABLE_METALINK
   cout << endl;