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
pull/1/head
Tatsuhiro Tsujikawa 2008-11-26 12:50:07 +00:00
parent e074d93348
commit 72d523b990
4 changed files with 40 additions and 19 deletions

View File

@ -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> 2008-11-24 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
* Release 1.0.1 * Release 1.0.1

View File

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

View File

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

View File

@ -9,6 +9,7 @@ class SocketCoreTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(SocketCoreTest); CPPUNIT_TEST_SUITE(SocketCoreTest);
CPPUNIT_TEST(testWriteAndReadDatagram); CPPUNIT_TEST(testWriteAndReadDatagram);
CPPUNIT_TEST(testGetSocketError);
CPPUNIT_TEST_SUITE_END(); CPPUNIT_TEST_SUITE_END();
public: public:
void setUp() {} void setUp() {}
@ -16,6 +17,7 @@ public:
void tearDown() {} void tearDown() {}
void testWriteAndReadDatagram(); 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 } // namespace aria2