168 lines
4.5 KiB
C++
168 lines
4.5 KiB
C++
/*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* Copyright 2011 Danny Robson <danny@nerdcruft.net>
|
|
*/
|
|
|
|
#include "address.hpp"
|
|
|
|
#include "except.hpp"
|
|
#include "../debug.hpp"
|
|
#include "../endian.hpp"
|
|
#include "../types.hpp"
|
|
#include "../raii.hpp"
|
|
|
|
|
|
#ifdef __WIN32
|
|
#else
|
|
#include <netdb.h>
|
|
#include <arpa/inet.h>
|
|
#endif
|
|
|
|
//-----------------------------------------------------------------------------
|
|
using namespace net;
|
|
using namespace std;
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
#ifdef __WIN32
|
|
const char* inet_ntop(int af, const void* src, char* dst, int size){
|
|
struct sockaddr_in srcaddr;
|
|
memset(&srcaddr, 0, sizeof(struct sockaddr_in));
|
|
memcpy(&(srcaddr.sin_addr), src, sizeof(srcaddr.sin_addr));
|
|
srcaddr.sin_family = af;
|
|
|
|
if (WSAAddressToString ((struct sockaddr*) &srcaddr, sizeof (struct sockaddr_in), 0, dst, (LPDWORD) &size) != 0)
|
|
net::error::throw_code ();
|
|
|
|
return dst;
|
|
}
|
|
#endif
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <domain D>
|
|
typename address<D>::port_type
|
|
address<D>::port (void) const
|
|
{ return m_port; }
|
|
|
|
|
|
template <domain D>
|
|
void
|
|
address<D>::set_port (const port_type &_port)
|
|
{ m_port = _port; }
|
|
|
|
|
|
template <domain D>
|
|
typename address<D>::ip_type
|
|
address<D>::resolve (const std::string &str) {
|
|
addrinfo hint;
|
|
memset (&hint, 0, sizeof (hint));
|
|
hint.ai_family = static_cast<int> (D);
|
|
|
|
addrinfo* resolved;
|
|
int err = getaddrinfo (str.c_str (), nullptr, nullptr, &resolved);
|
|
if (err)
|
|
net::error::throw_code ();
|
|
|
|
auto deletor = [] (addrinfo *a) { freeaddrinfo (a); };
|
|
std::unique_ptr<addrinfo, decltype(deletor)> raii(resolved, deletor);
|
|
return ip_type (reinterpret_cast<sockaddr_type*> (resolved->ai_addr)->sin_addr.s_addr);
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
namespace net {
|
|
template <>
|
|
address<domain::INET>::address (const sockaddr_type &addr):
|
|
m_ip (addr.sin_addr.s_addr),
|
|
m_mask (0),
|
|
m_port (ntoh (addr.sin_port))
|
|
{
|
|
CHECK (addr.sin_family == (int)domain::INET);
|
|
}
|
|
|
|
|
|
template <domain D>
|
|
address<D>::address (const std::string &_addr,
|
|
port_type _port):
|
|
m_ip (resolve (_addr)),
|
|
m_mask ( 0),
|
|
m_port (_port)
|
|
{ ; }
|
|
|
|
|
|
template <>
|
|
address<domain::INET>::sockaddr_type
|
|
address<domain::INET>::to_sockaddr (void) const {
|
|
sockaddr_type addr;
|
|
|
|
addr.sin_family = (int)domain::INET;
|
|
addr.sin_port = hton (m_port);
|
|
addr.sin_addr.s_addr = m_ip.m_integer;
|
|
|
|
return addr;
|
|
}
|
|
|
|
|
|
template <>
|
|
std::string
|
|
address<domain::INET>::to_string (void) const {
|
|
char dest[INET_ADDRSTRLEN + 1];
|
|
sockaddr_type addr = to_sockaddr ();
|
|
|
|
if (NULL == inet_ntop ((int)domain::INET, &addr.sin_addr, dest, sizeof (dest)))
|
|
net::error::throw_code ();
|
|
return dest;
|
|
}
|
|
|
|
|
|
template <>
|
|
bool
|
|
address<domain::INET>::operator ==(const address<domain::INET> &rhs) {
|
|
return m_ip == rhs.m_ip &&
|
|
m_mask == rhs.m_mask &&
|
|
m_port == rhs.m_port;
|
|
}
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
std::ostream&
|
|
net::operator<< (std::ostream &os, const address<net::domain::INET> &addr) {
|
|
os << addr.to_string () << ":" << addr.port ();
|
|
return os;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <> const address<domain::INET>
|
|
address<domain::INET>::LOOPBACK ("127.0.0.1", 0);
|
|
|
|
template <> const address<domain::INET>
|
|
address<domain::INET>::ANY ("0.0.0.0", 0);
|
|
|
|
namespace net {
|
|
template class address<domain::INET>;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//template <> const address<domain::INET6>
|
|
//address<domain::INET6>::LOOPBACK ("::1", 0);
|
|
//
|
|
//template <> const address<domain::INET6>
|
|
//address<domain::INET6>::ANY ("::0", 0);
|
|
//
|
|
//template class address<domain::INET6>;
|
|
|