NetAddr.cpp

Go to the documentation of this file.

/*  Project nGenEx
    Destroyer Studios LLC
    Copyright © 1997-2004. All Rights Reserved.

    SUBSYSTEM:    NetEx.lib
    FILE:         NetAddr.cpp
    AUTHOR:       John DiCamillo


    OVERVIEW
    ========
    Network Address
*/


#include "MemDebug.h"
#include "NetAddr.h"
#include "NetHost.h"
#include "NetLayer.h"
#include <ctype.h>

NetAddr::NetAddr(const char* host_name, WORD p)
   : addr(0), port(p)
{
   if (host_name && *host_name) {
      HOSTENT* h = 0;

      if (isdigit(*host_name)) {
         DWORD a = inet_addr(host_name);
         h = gethostbyaddr((const char*) &a, 4, AF_INET);   
      }
      else {
         h = gethostbyname(host_name);
      }

      if (h) {
         if (h->h_addr_list) {
            addr = **(DWORD**) (h->h_addr_list);
         }
      }
   }

   Init();
}

NetAddr::NetAddr(DWORD a, WORD p)
   : addr(a), port(p)
{
   Init();
}

NetAddr::NetAddr(const NetAddr& n)
   : addr(n.addr), port(n.port)
{
   Init();
}

// +--------------------------------------------------------------------+

void
NetAddr::Init()
{
   ZeroMemory(&sadr, sizeof(sadr));

   sadr.sin_family      = AF_INET;
   sadr.sin_port        = ::htons(port);
   sadr.sin_addr.s_addr = addr;
}

void
NetAddr::InitFromSockAddr()
{
   addr = sadr.sin_addr.s_addr;
   port = ::ntohs(sadr.sin_port);
}

// +--------------------------------------------------------------------+

sockaddr*
NetAddr::GetSockAddr() const
{
   return (sockaddr*) &sadr;
}

size_t
NetAddr::GetSockAddrLength() const
{
   return sizeof(sadr);
}

// +--------------------------------------------------------------------+

void
NetAddr::SetSockAddr(sockaddr* s, int size)
{
   if (s) {
      ZeroMemory(&sadr, sizeof(sadr));

      if (size > sizeof(sadr))
         CopyMemory(&sadr, s, sizeof(sadr));
      else
         CopyMemory(&sadr, s, size);

      InitFromSockAddr();
   }
}