source/resolve.cpp

   1 #ifdef WIN32
   2 #define _WIN32_WINNT 0x0501
   3 #include <ws2tcpip.h>
   4 #else
   5 #include <netdb.h>
   6 #endif
   7 #include <msp/core/systemerror.h>
   8 #include <msp/strings/format.h>
   9 #include "sockaddr_private.h"
  10 #include "socket.h"
  11 #include "resolve.h"
  12
  13 using namespace std;
  14
  15 namespace Msp {
  16 namespace Net {
  17
  18 SockAddr *resolve(const string &s, Family family)
  19 {
  20         string host, serv;
  21         if(s[0]=='[')
  22         {
  23                 unsigned bracket = s.find(']');
  24                 host = s.substr(1, bracket-1);
  25                 unsigned colon = s.find(':', bracket);
  26                 if(colon!=string::npos)
  27                         serv = s.substr(colon+1);
  28         }
  29         else
  30         {
  31                 unsigned colon = s.find(':');
  32                 if(colon!=string::npos)
  33                 {
  34                         host = s.substr(0, colon);
  35                         serv = s.substr(colon+1);
  36                 }
  37                 else
  38                         host = s;
  39         }
  40
  41         addrinfo hints = { 0, family_to_sys(family), 0, 0, 0, 0, 0, 0 };
  42         addrinfo *res;
  43         const char *chost = (host.empty() ? 0 : host.c_str());
  44         const char *cserv = (serv.empty() ? 0 : serv.c_str());
  45         int err = getaddrinfo(chost, cserv, &hints, &res);
  46         if(err==0)
  47         {
  48                 SockAddr::SysAddr sa;
  49                 sa.size = res->ai_addrlen;
  50                 const char *sptr = reinterpret_cast<const char *>(res->ai_addr);
  51                 char *dptr = reinterpret_cast<char *>(&sa.addr);
  52                 copy(sptr, sptr+res->ai_addrlen, dptr);
  53                 SockAddr *addr = SockAddr::from_sys(sa);
  54                 freeaddrinfo(res);
  55                 return addr;
  56         }
  57         else
  58 #ifdef WIN32
  59                 throw system_error("getaddrinfo", WSAGetLastError());
  60 #else
  61                 throw system_error("getaddrinfo", gai_strerror(err));
  62 #endif
  63 }
  64
  65                 /*sockaddr sa;
  66                 unsigned size = fill_sockaddr(sa);
  67                 char hst[128];
  68                 char srv[128];
  69                 int err = getnameinfo(&sa, size, hst, 128, srv, 128, 0);
  70                 if(err==0)
  71                 {
  72                         host = hst;
  73                         serv = srv;
  74                 }*/
  75
  76 } // namespace Net
  77 } // namespace Msp