github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity

Use sockaddr_union instead of sockaddr_storage.

pull/2/head
Tatsuhiro Tsujikawa 2011-10-20 00:06:44 +09:00
parent c01c34f24a
commit 9b4855a5f0
4 changed files with 54 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/BtSetup.cc
View File

@ -229,14 +229,12 @@ void BtSetup::setup(std::vector<Command*>& commands,
initialized = true;
}
} else {
std::vector<std::pair<sockaddr_storage, socklen_t> > ifAddrs;
std::vector<std::pair<sockaddr_union, socklen_t> > ifAddrs;
getInterfaceAddress(ifAddrs, lpdInterface, AF_INET, AI_NUMERICHOST);
for(std::vector<std::pair<sockaddr_storage, socklen_t> >::const_iterator
for(std::vector<std::pair<sockaddr_union, socklen_t> >::const_iterator
i = ifAddrs.begin(), eoi = ifAddrs.end(); i != eoi; ++i) {
sockaddr_in addr;
memcpy(&addr, &(*i).first, (*i).second);
char host[NI_MAXHOST];
if(inetNtop(AF_INET, &addr.sin_addr, host, sizeof(host)) == 0 &&
if(inetNtop(AF_INET, &(*i).first.in, host, sizeof(host)) == 0 &&
receiver->init(host)) {
initialized = true;
break;

11
src/FtpConnection.cc
View File

@ -221,21 +221,20 @@ SharedHandle<SocketCore> FtpConnection::createServerSocket()
bool FtpConnection::sendEprt(const SharedHandle<SocketCore>& serverSocket)
{
if(socketBuffer_.sendBufferIsEmpty()) {
struct sockaddr_storage sockaddr;
sockaddr_union sockaddr;
socklen_t len = sizeof(sockaddr);
serverSocket->getAddrInfo(sockaddr, len);
std::pair<std::string, uint16_t> addrinfo = util::getNumericNameInfo
(reinterpret_cast<const struct sockaddr*>(&sockaddr), len);
std::pair<std::string, uint16_t> addrinfo =
util::getNumericNameInfo(&sockaddr.sa, len);
std::string request = "EPRT ";
request += "|";
request += util::itos(sockaddr.ss_family == AF_INET?1:2);
request += util::itos(sockaddr.storage.ss_family == AF_INET ? 1 : 2);
request += "|";
request += addrinfo.first;
request += "|";
request += util::uitos(addrinfo.second);
request += "|\r\n";
A2_LOG_INFO(fmt(MSG_SENDING_REQUEST,
cuid_, request.c_str()));
A2_LOG_INFO(fmt(MSG_SENDING_REQUEST, cuid_, request.c_str()));
socketBuffer_.pushStr(request);
}
socketBuffer_.send();

98
src/SocketCore.cc
View File

@ -118,7 +118,7 @@ std::string errorMsg(int errNum)
int SocketCore::protocolFamily_ = AF_UNSPEC;
std::vector<std::pair<struct sockaddr_storage, socklen_t> >
std::vector<std::pair<sockaddr_union, socklen_t> >
SocketCore::bindAddrs_;
#ifdef ENABLE_SSL
@ -280,14 +280,12 @@ void SocketCore::bind
sockfd_ = fd;
}
} else {
for(std::vector<std::pair<struct sockaddr_storage, socklen_t> >::
for(std::vector<std::pair<sockaddr_union, socklen_t> >::
const_iterator i = bindAddrs_.begin(), eoi = bindAddrs_.end();
i != eoi; ++i) {
char host[NI_MAXHOST];
int s;
s = getnameinfo(reinterpret_cast<const struct sockaddr*>(&(*i).first),
(*i).second,
host, NI_MAXHOST, 0, 0,
s = getnameinfo(&(*i).first.sa, (*i).second, host, NI_MAXHOST, 0, 0,
NI_NUMERICHOST);
if(s) {
error = gai_strerror(s);
@ -336,11 +334,10 @@ void SocketCore::beginListen()
SocketCore* SocketCore::acceptConnection() const
{
struct sockaddr_storage sockaddr;
sockaddr_union sockaddr;
socklen_t len = sizeof(sockaddr);
sock_t fd;
while((fd = accept(sockfd_, reinterpret_cast<struct sockaddr*>(&sockaddr),
&len)) == (sock_t) -1 &&
while((fd = accept(sockfd_, &sockaddr.sa, &len)) == (sock_t) -1 &&
SOCKET_ERRNO == A2_EINTR);
int errNum = SOCKET_ERRNO;
if(fd == (sock_t) -1) {
@ -351,18 +348,15 @@ SocketCore* SocketCore::acceptConnection() const
void SocketCore::getAddrInfo(std::pair<std::string, uint16_t>& addrinfo) const
{
struct sockaddr_storage sockaddr;
sockaddr_union sockaddr;
socklen_t len = sizeof(sockaddr);
getAddrInfo(sockaddr, len);
addrinfo = util::getNumericNameInfo
(reinterpret_cast<const struct sockaddr*>(&sockaddr), len);
addrinfo = util::getNumericNameInfo(&sockaddr.sa, len);
}
void SocketCore::getAddrInfo
(struct sockaddr_storage& sockaddr, socklen_t& len) const
void SocketCore::getAddrInfo(sockaddr_union& sockaddr, socklen_t& len) const
{
struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
if(getsockname(sockfd_, addrp, &len) == -1) {
if(getsockname(sockfd_, &sockaddr.sa, &len) == -1) {
int errNum = SOCKET_ERRNO;
throw DL_ABORT_EX(fmt(EX_SOCKET_GET_NAME, errorMsg(errNum).c_str()));
}
@ -370,22 +364,21 @@ void SocketCore::getAddrInfo
int SocketCore::getAddressFamily() const
{
struct sockaddr_storage sockaddr;
sockaddr_union sockaddr;
socklen_t len = sizeof(sockaddr);
getAddrInfo(sockaddr, len);
return sockaddr.ss_family;
return sockaddr.storage.ss_family;
}
void SocketCore::getPeerInfo(std::pair<std::string, uint16_t>& peerinfo) const
{
struct sockaddr_storage sockaddr;
sockaddr_union sockaddr;
socklen_t len = sizeof(sockaddr);
struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
if(getpeername(sockfd_, addrp, &len) == -1) {
if(getpeername(sockfd_, &sockaddr.sa, &len) == -1) {
int errNum = SOCKET_ERRNO;
throw DL_ABORT_EX(fmt(EX_SOCKET_GET_NAME, errorMsg(errNum).c_str()));
}
peerinfo = util::getNumericNameInfo(addrp, len);
peerinfo = util::getNumericNameInfo(&sockaddr.sa, len);
}
void SocketCore::establishConnection(const std::string& host, uint16_t port)
@ -421,11 +414,10 @@ void SocketCore::establishConnection(const std::string& host, uint16_t port)
}
if(!bindAddrs_.empty()) {
bool bindSuccess = false;
for(std::vector<std::pair<struct sockaddr_storage, socklen_t> >::
for(std::vector<std::pair<sockaddr_union, socklen_t> >::
const_iterator i = bindAddrs_.begin(), eoi = bindAddrs_.end();
i != eoi; ++i) {
if(::bind(fd,reinterpret_cast<const struct sockaddr*>(&(*i).first),
(*i).second) == -1) {
if(::bind(fd, &(*i).first.sa, (*i).second) == -1) {
errNum = SOCKET_ERRNO;
error = errorMsg(errNum);
A2_LOG_DEBUG(fmt(EX_SOCKET_BIND, error.c_str()));
@ -1056,12 +1048,11 @@ ssize_t SocketCore::readDataFrom(char* data, size_t len,
{
wantRead_ = false;
wantWrite_ = false;
struct sockaddr_storage sockaddr;
socklen_t sockaddrlen = sizeof(struct sockaddr_storage);
struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
sockaddr_union sockaddr;
socklen_t sockaddrlen = sizeof(sockaddr);
ssize_t r;
while((r = recvfrom(sockfd_, data, len, 0, addrp, &sockaddrlen)) == -1 &&
A2_EINTR == SOCKET_ERRNO);
while((r = recvfrom(sockfd_, data, len, 0, &sockaddr.sa, &sockaddrlen)) == -1
&& A2_EINTR == SOCKET_ERRNO);
int errNum = SOCKET_ERRNO;
if(r == -1) {
if(A2_WOULDBLOCK(errNum)) {
@ -1071,7 +1062,7 @@ ssize_t SocketCore::readDataFrom(char* data, size_t len,
throw DL_RETRY_EX(fmt(EX_SOCKET_RECV, errorMsg(errNum).c_str()));
}
} else {
sender = util::getNumericNameInfo(addrp, sockaddrlen);
sender = util::getNumericNameInfo(&sockaddr.sa, sockaddrlen);
}
return r;
@ -1108,23 +1099,19 @@ bool SocketCore::wantWrite() const
void SocketCore::bindAddress(const std::string& iface)
{
std::vector<std::pair<struct sockaddr_storage, socklen_t> > bindAddrs;
std::vector<std::pair<sockaddr_union, socklen_t> > bindAddrs;
getInterfaceAddress(bindAddrs, iface, protocolFamily_);
if(bindAddrs.empty()) {
throw DL_ABORT_EX
(fmt(MSG_INTERFACE_NOT_FOUND,
iface.c_str(),
"not available"));
(fmt(MSG_INTERFACE_NOT_FOUND, iface.c_str(), "not available"));
} else {
bindAddrs_ = bindAddrs;
for(std::vector<std::pair<struct sockaddr_storage, socklen_t> >::
bindAddrs_.swap(bindAddrs);
for(std::vector<std::pair<sockaddr_union, socklen_t> >::
const_iterator i = bindAddrs_.begin(), eoi = bindAddrs_.end();
i != eoi; ++i) {
char host[NI_MAXHOST];
int s;
s = getnameinfo(reinterpret_cast<const struct sockaddr*>(&(*i).first),
(*i).second,
host, NI_MAXHOST, 0, 0,
s = getnameinfo(&(*i).first.sa, (*i).second, host, NI_MAXHOST, 0, 0,
NI_NUMERICHOST);
if(s == 0) {
A2_LOG_DEBUG(fmt("Sockets will bind to %s", host));
@ -1134,7 +1121,7 @@ void SocketCore::bindAddress(const std::string& iface)
}
void getInterfaceAddress
(std::vector<std::pair<struct sockaddr_storage, socklen_t> >& ifAddrs,
(std::vector<std::pair<sockaddr_union, socklen_t> >& ifAddrs,
const std::string& iface, int family, int aiFlags)
{
A2_LOG_DEBUG(fmt("Finding interface %s", iface.c_str()));
@ -1146,8 +1133,8 @@ void getInterfaceAddress
A2_LOG_INFO(fmt(MSG_INTERFACE_NOT_FOUND,
iface.c_str(), errorMsg(errNum).c_str()));
} else {
auto_delete<struct ifaddrs*> ifaddrDeleter(ifaddr, freeifaddrs);
for(struct ifaddrs* ifa = ifaddr; ifa; ifa = ifa->ifa_next) {
auto_delete<ifaddrs*> ifaddrDeleter(ifaddr, freeifaddrs);
for(ifaddrs* ifa = ifaddr; ifa; ifa = ifa->ifa_next) {
if(!ifa->ifa_addr) {
continue;
}
@ -1168,37 +1155,36 @@ void getInterfaceAddress
continue;
}
if(std::string(ifa->ifa_name) == iface) {
socklen_t bindAddrLen = iffamily == AF_INET?sizeof(struct sockaddr_in):
sizeof(struct sockaddr_in6);
struct sockaddr_storage bindAddr;
socklen_t bindAddrLen =
iffamily == AF_INET ? sizeof(sockaddr_in) : sizeof(sockaddr_in6);
sockaddr_union bindAddr;
memset(&bindAddr, 0, sizeof(bindAddr));
memcpy(&bindAddr, ifa->ifa_addr, bindAddrLen);
memcpy(&bindAddr.storage, ifa->ifa_addr, bindAddrLen);
ifAddrs.push_back(std::make_pair(bindAddr, bindAddrLen));
}
}
}
#endif // HAVE_GETIFADDRS
if(ifAddrs.empty()) {
struct addrinfo* res;
addrinfo* res;
int s;
s = callGetaddrinfo(&res, iface.c_str(), 0, family, SOCK_STREAM, aiFlags,0);
if(s) {
A2_LOG_INFO(fmt(MSG_INTERFACE_NOT_FOUND, iface.c_str(), gai_strerror(s)));
} else {
WSAAPI_AUTO_DELETE<struct addrinfo*> resDeleter(res, freeaddrinfo);
struct addrinfo* rp;
WSAAPI_AUTO_DELETE<addrinfo*> resDeleter(res, freeaddrinfo);
addrinfo* rp;
for(rp = res; rp; rp = rp->ai_next) {
socklen_t bindAddrLen = rp->ai_addrlen;
struct sockaddr_storage bindAddr;
memset(&bindAddr, 0, sizeof(bindAddr));
memcpy(&bindAddr, rp->ai_addr, rp->ai_addrlen);
// Try to bind socket with this address. If it fails, the
// address is not for this machine.
try {
SocketCore socket;
socket.bind
(reinterpret_cast<const struct sockaddr*>(&bindAddr), bindAddrLen);
ifAddrs.push_back(std::make_pair(bindAddr, bindAddrLen));
//socket.bind(&bindAddr.sa, bindAddrLen);
socket.bind(rp->ai_addr, rp->ai_addrlen);
sockaddr_union bindAddr;
memset(&bindAddr, 0, sizeof(bindAddr));
memcpy(&bindAddr.storage, rp->ai_addr, rp->ai_addrlen);
ifAddrs.push_back(std::make_pair(bindAddr, rp->ai_addrlen));
} catch(RecoverableException& e) {
continue;
}

8
src/SocketCore.h
View File

@ -76,7 +76,7 @@ private:
static int protocolFamily_;
static std::vector<std::pair<struct sockaddr_storage, socklen_t> > bindAddrs_;
static std::vector<std::pair<sockaddr_union, socklen_t> > bindAddrs_;
bool blocking_;
int secure_;
@ -159,7 +159,7 @@ public:
* in len.
*/
void getAddrInfo
(struct sockaddr_storage& sockaddr, socklen_t& len) const;
(sockaddr_union& sockaddr, socklen_t& len) const;
/**
* Returns address family of this socket.
@ -344,7 +344,7 @@ public:
static void bindAddress(const std::string& iface);
friend void getInterfaceAddress
(std::vector<std::pair<struct sockaddr_storage, socklen_t> >& ifAddrs,
(std::vector<std::pair<sockaddr_union, socklen_t> >& ifAddrs,
const std::string& iface, int family, int aiFlags);
};
@ -366,7 +366,7 @@ int callGetaddrinfo
// to collect using family argument. aiFlags is passed to
// getaddrinfo() as hints.ai_flags. No throw.
void getInterfaceAddress
(std::vector<std::pair<struct sockaddr_storage, socklen_t> >& ifAddrs,
(std::vector<std::pair<sockaddr_union, socklen_t> >& ifAddrs,
const std::string& iface, int family = AF_UNSPEC, int aiFlags = 0);
// Provides functionality of inet_ntop using getnameinfo. The return