SockAddrIn.h

Go to the documentation of this file.

/*  SockAddrIn.h - Socket address wrapper
    Copyright (C) 2001-2004 Mark Weaver
    Written by Mark Weaver <mark@npsl.co.uk>

    Part of the Open-Win32 library.
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA  02111-1307, USA.
*/

#ifndef OW32_SockAddrIn_h
#define OW32_SockAddrIn_h

#include <OW32/OW32Libs.h>

namespace OW32
{

class OW32_LIB_EXPORT CSockAddrIn : 
    public SOCKADDR_IN
{
public:
    CSockAddrIn(u_short port = 0, u_long addr = INADDR_ANY)
    {
        sin_family = AF_INET;
        sin_port = port;
        sin_addr.s_addr = addr;
    }

    void SetPort(u_short port) { sin_port = port; }

    unsigned long GetPort() { return sin_port; }

    void SetAddr(u_long addr) { sin_addr.s_addr = addr; }

    unsigned long GetAddr() { return sin_addr.s_addr; }
    
    bool SetLiteralIP(const char* literal_ip) 
    {
        // skip spaces (shouldn't be any though)
        while (isspace(*((unsigned char *)literal_ip))) ++literal_ip;

        // skip '[' - this is a common notation from mail days
        if (*literal_ip == '[')
            ++literal_ip;

        // Can't be valid if > 128 characters (IPv6=?)
        char ip_buf[128];
        int pos=0;
        while (pos < sizeof(ip_buf)/sizeof(ip_buf[0]) && *literal_ip != '\0' && *literal_ip != ']') {
            ip_buf[pos++] = *literal_ip++;
        }
        // literal IP too long
        if (pos >= sizeof(ip_buf)/sizeof(ip_buf[0])) return false;
        ip_buf[pos] = '\0';

        // conver the IP, checking for success
        unsigned long addr = inet_addr(ip_buf);
        if (addr == htonl((unsigned long)-1))
            return false;

        // got it!
        SetAddr(addr);
        return true;
    }

    bool SetHostName(const char* hostName) 
    {
        // check for literal ip
        while (isspace(*((unsigned char *)hostName))) ++hostName;
        if (*hostName == '[') {
            return SetLiteralIP(hostName);
        }

        // otherwise, it's actually a "host by name" effort
        struct hostent* hent = gethostbyname(hostName);
        if (hent == NULL)
            return false;

        memmove(&sin_addr, *(hent->h_addr_list), sizeof(sin_addr));
        return true;
    }
};

} // namespace OW32

#endif // OW32_SockAddrIn_h

---