Prechádzať zdrojové kódy

2008-11-26 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>

	Applied aria2-1.0.1-mingw.patch.
	Use SOCKET_ERRNO instead of errno if the code is used both linux
	and mingw.
	Collected 4th and 5th argument types for getsockopt and
	setsockopt.
	* src/SocketCore.cc
	* src/a2netcompat.h
	* test/SocketCoreTest.cc
Tatsuhiro Tsujikawa 17 rokov pred
rodič
commit
72d523b990
4 zmenil súbory, kde vykonal 40 pridanie a 19 odobranie
  1. 10 0
      ChangeLog
  2. 18 17
      src/SocketCore.cc
  3. 2 2
      src/a2netcompat.h
  4. 10 0
      test/SocketCoreTest.cc

+ 10 - 0
ChangeLog

@@ -1,3 +1,13 @@
+2008-11-26  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
+
+	Applied aria2-1.0.1-mingw.patch.
+	Use SOCKET_ERRNO instead of errno if the code is used both linux and
+	mingw.
+	Collected 4th and 5th argument types for getsockopt and setsockopt.
+	* src/SocketCore.cc
+	* src/a2netcompat.h
+	* test/SocketCoreTest.cc
+
 2008-11-24  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
 
 	* Release 1.0.1

+ 18 - 17
src/SocketCore.cc

@@ -174,8 +174,8 @@ void SocketCore::bind(uint16_t port)
     if(fd == -1) {
       continue;
     }
-    SOCKOPT_T sockopt = 1;
-    if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(socklen_t)) < 0) {
+    int sockopt = 1;
+    if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (a2_sockopt_t) &sockopt, sizeof(sockopt)) < 0) {
       CLOSE(fd);
       continue;
     }
@@ -204,7 +204,7 @@ SocketCore* SocketCore::acceptConnection() const
   struct sockaddr_storage sockaddr;
   socklen_t len = sizeof(sockaddr);
   sock_t fd;
-  while((fd = accept(sockfd, reinterpret_cast<struct sockaddr*>(&sockaddr), &len)) == -1 && errno == EINTR);
+  while((fd = accept(sockfd, reinterpret_cast<struct sockaddr*>(&sockaddr), &len)) == -1 && SOCKET_ERRNO == EINTR);
   if(fd == -1) {
     throw DlAbortEx(StringFormat(EX_SOCKET_ACCEPT, errorMsg()).str());
   }
@@ -256,8 +256,8 @@ void SocketCore::establishConnection(const std::string& host, uint16_t port)
     if(fd == -1) {
       continue;
     }
-    SOCKOPT_T sockopt = 1;
-    if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(socklen_t)) < 0) {
+    int sockopt = 1;
+    if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (a2_sockopt_t) &sockopt, sizeof(sockopt)) < 0) {
       CLOSE(fd);
       continue;
     }
@@ -498,9 +498,9 @@ ssize_t SocketCore::writeData(const char* data, size_t len)
   _wantWrite = false;
 
   if(!secure) {
-    while((ret = send(sockfd, data, len, 0)) == -1 && errno == EINTR);
+    while((ret = send(sockfd, data, len, 0)) == -1 && SOCKET_ERRNO == EINTR);
     if(ret == -1) {
-      if(errno == EAGAIN) {
+      if(SOCKET_ERRNO == EAGAIN) {
 	_wantWrite = true;
 	ret = 0;
       } else {
@@ -546,10 +546,10 @@ void SocketCore::readData(char* data, size_t& len)
   _wantWrite = false;
 
   if(!secure) {    
-    while((ret = recv(sockfd, data, len, 0)) == -1 && errno == EINTR);
+    while((ret = recv(sockfd, data, len, 0)) == -1 && SOCKET_ERRNO == EINTR);
     
     if(ret == -1) {
-      if(errno == EAGAIN) {
+      if(SOCKET_ERRNO == EAGAIN) {
 	_wantRead = true;
 	ret = 0;
       } else {
@@ -597,9 +597,9 @@ void SocketCore::peekData(char* data, size_t& len)
   _wantWrite = false;
 
   if(!secure) {
-    while((ret = recv(sockfd, data, len, MSG_PEEK)) == -1 && errno == EINTR);
+    while((ret = recv(sockfd, data, len, MSG_PEEK)) == -1 && SOCKET_ERRNO == EINTR);
     if(ret == -1) {
-      if(errno == EAGAIN) {
+      if(SOCKET_ERRNO == EAGAIN) {
 	_wantRead = true;
 	ret = 0;
       } else {
@@ -984,11 +984,11 @@ ssize_t SocketCore::writeData(const char* data, size_t len,
   struct addrinfo* rp;
   ssize_t r = -1;
   for(rp = res; rp; rp = rp->ai_next) {
-    while((r = sendto(sockfd, data, len, 0, rp->ai_addr, rp->ai_addrlen)) == -1 && EINTR == errno);
+    while((r = sendto(sockfd, data, len, 0, rp->ai_addr, rp->ai_addrlen)) == -1 && EINTR == SOCKET_ERRNO);
     if(r == static_cast<ssize_t>(len)) {
       break;
     }
-    if(r == -1 && errno == EAGAIN) {
+    if(r == -1 && SOCKET_ERRNO == EAGAIN) {
       _wantWrite = true;
       r = 0;
       break;
@@ -1012,9 +1012,9 @@ ssize_t SocketCore::readDataFrom(char* data, size_t len,
   struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
   ssize_t r;
   while((r = recvfrom(sockfd, data, len, 0, addrp, &sockaddrlen)) == -1 &&
-	EINTR == errno);
+	EINTR == SOCKET_ERRNO);
   if(r == -1) {
-    if(errno == EAGAIN) {
+    if(SOCKET_ERRNO == EAGAIN) {
       _wantRead = true;
       r = 0;
     } else {
@@ -1029,9 +1029,10 @@ ssize_t SocketCore::readDataFrom(char* data, size_t len,
 
 std::string SocketCore::getSocketError() const
 {
-  SOCKOPT_T error;
+  int error;
   socklen_t optlen = sizeof(error);
-  if(getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &optlen) == -1) {
+
+  if(getsockopt(sockfd, SOL_SOCKET, SO_ERROR, (a2_sockopt_t) &error, &optlen) == -1) {
     throw DlAbortEx(StringFormat("Failed to get socket error: %s",
 				 errorMsg()).str());
   }

+ 2 - 2
src/a2netcompat.h

@@ -50,12 +50,12 @@
 #endif // !__CYGWIN__
 
 #ifdef __MINGW32__
-# define SOCKOPT_T char
+# define a2_sockopt_t char *
 # define HAVE_GETADDRINFO
 # undef HAVE_GAI_STRERROR
 # undef gai_strerror
 #else
-# define SOCKOPT_T int
+# define a2_sockopt_t int *
 #endif // __MINGW32__
 
 #ifdef HAVE_NETDB_H

+ 10 - 0
test/SocketCoreTest.cc

@@ -9,6 +9,7 @@ class SocketCoreTest:public CppUnit::TestFixture {
 
   CPPUNIT_TEST_SUITE(SocketCoreTest);
   CPPUNIT_TEST(testWriteAndReadDatagram);
+  CPPUNIT_TEST(testGetSocketError);
   CPPUNIT_TEST_SUITE_END();
 public:
   void setUp() {}
@@ -16,6 +17,7 @@ public:
   void tearDown() {}
 
   void testWriteAndReadDatagram();
+  void testGetSocketError();
 };
 
 
@@ -59,4 +61,12 @@ void SocketCoreTest::testWriteAndReadDatagram()
   }
 }
 
+void SocketCoreTest::testGetSocketError()
+{
+  SocketCore s;
+  s.bind(0);
+  // See there is no error at this point
+  CPPUNIT_ASSERT_EQUAL(std::string(""), s.getSocketError());
+}
+
 } // namespace aria2