common/ipsockaddr.cc

Go to the documentation of this file.

// vim:ts=8 sts=4 sw=4
//
// The IPSocketAddress class; holds a SocketAddress for an IP socket.
//
// ***** BEGIN LICENSE BLOCK *****
// Version: MPL 1.1/GPL 2.0/LGPL 2.1
// 
// The contents of this file are subject to the Mozilla Public License
// Version 1.1 (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.mozilla.org/MPL/
// 
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
// the License for the specific language governing rights and limitations
// under the License.
// 
// The Original Code is the consmgr network/serial-line monitoring package.
// 
// The Initial Developer of the Original Code is Chris P. Ross.
// Portions created by the Initial Developer are Copyright (C) 2000-2008
// the Initial Developer.  All Rights Reserved.
// 
// Contributor(s):
//   Geoff Adams
// 
// Alternatively, the contents of this file may be used under the terms of
// either the GNU General Public License Version 2 or later (the "GPL"), or
// the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
// in which case the provisions of the GPL or the LGPL are applicable instead
// of those above. If you wish to allow use of your version of this file only
// under the terms of either the GPL or the LGPL, and not to allow others to
// use your version of this file under the terms of the MPL, indicate your
// decision by deleting the provisions above and replace them with the notice
// and other provisions required by the GPL or the LGPL. If you do not delete
// the provisions above, a recipient may use your version of this file under
// the terms of any one of the MPL, the GPL or the LGPL.
// 
// ***** END LICENSE BLOCK *****

extern "C" {
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
}

#include <iostream>
#include <sstream>

#include "consmgr.h"
#include "ipsockaddr.h"


IPSocketAddress::IPSocketAddress(const std::string &address, int port)
{
    int                         ret;
    struct sockaddr_storage     ss;
    struct sockaddr_in          *sin = reinterpret_cast<struct sockaddr_in*>
                                            (&ss);

#ifdef  INET6
    struct sockaddr_in6         *sin6 = reinterpret_cast<struct sockaddr_in6*>
                                            (&ss);

    /* First, try to parse the supplied address as an IPv6 (or, IPv4
     * mapped) address. */
    ret = inet_pton(AF_INET6, address.c_str(), &sin6->sin6_addr);
    if (ret < 0) {
        throw(fireball(74, "inet_pton(AF_INET6) failed"));
    } else if (ret > 0) {
        ss.ss_family = AF_INET6;
#ifndef NO_SA_LEN
        ss.ss_len = sizeof(struct sockaddr_in6);
#endif
        // XXX - Make port optional?
        sin6->sin6_port = htons(port);

        goto finish;
    } /* else ret == 0, no valid network address (for AF_INET6) seen */
#endif  /* INET6 */

    /* Try to parse the address supplied as an IPv4 address */
    ret = inet_pton(AF_INET, address.c_str(), &sin->sin_addr);
    if (ret < 0) {
        throw(fireball(75, "inet_pton(AF_INET) failed"));
    } else if (ret == 0) {
        throw(CMInvalidParameter("Unable to parse " + address +
                                 " as an address."));
    } /* else ret > 0; success! */

    ss.ss_family = AF_INET;
#ifndef NO_SA_LEN
    ss.ss_len = sizeof(struct sockaddr_in);
#endif
    // XXX - Make port optional?
    sin->sin_port = htons(port);

finish:
    set_addr(reinterpret_cast<struct sockaddr*>(&ss));
}

IPSocketAddress::~IPSocketAddress()
{
}


//
// Inquiry functions
//

u_short
IPSocketAddress::port() const
{
    u_short     pval;

    switch (family()) {
      case AF_INET:
        pval = ((sockaddr_in *)&addr)->sin_port;
        break;
      case AF_INET6:
        pval = ((sockaddr_in6*)&addr)->sin6_port;
        break;
      default:
        throw(CMInvalidParameter("Unknown family seen in IPSocketAddress::port()"));
        /* NOTREACHED */
    }
    return (pval);
}


//
// Generate a string to tell a human who we are...
//
string
IPSocketAddress::get_id(void) const
{
#if MAYBE_A_BAD_IDEA
    // First, check the cache...
    if (! id_cache.empty())
        return id_cache;
#endif
    
    std::ostringstream  id_str;
    // XXX - Can't call get_addr(), cause we don't know if the mutex
    // is already locked.  Just use it directly...
    //sockaddr          *addr = get_addr();
    char                host[64], svc[6];
#ifdef BROKEN_GETNAMEINFO
    sockaddr_in         *a4 = (sockaddr_in *)&addr;
# ifdef INET6
    sockaddr_in6        *a6 = (sockaddr_in6 *)&addr;
# endif /* INET6 */
    void                *aptr;
    u_short             port;
#else
    int                 ret;
#endif

#ifdef  BROKEN_GETNAMEINFO
    // XXX - getnameinfo(3) seems to be broken on the BSD/OS 4.3.1 sparcs.  Odd.
    switch (family()) {
      case AF_INET:
        aptr = (void *)&a4->sin_addr;
        port = a4->sin_port;
        break;
#ifdef  INET6
      case AF_INET6:
        aptr = (void *)&a6->sin6_addr;
        port = a6->sin6_port;
        break;
#endif
      default:
        throw(fireball(19, "Unknown family in current addr"));
    }

    // Make a printable form of the host address (Can't resolve with
    // gethostbyname(), as the appears to be the actual problem
    // within getnameinfo()...)
    if (NULL == inet_ntop(family(), aptr, host, sizeof(host)))
        throw(fireball(19, "Cannot get printable address from inet_ntop"));

    // And a printable form of the svc port number, just to match what
    // getnameinfo() would do...
    snprintf(svc, sizeof(svc), "%d", ntohs(port));
#else

//clog << "IPSocketAddress::get_id(): calling getnameinfo() (thread " << pthread_self() << ")" << endl;
    ret = getnameinfo((sockaddr*)&this->addr, sock_addrlen, host, sizeof(host),
                      svc, sizeof(svc), NI_NUMERICSERV | NI_NUMERICHOST);
//clog << "IPSocketAddress::get_id(): got host '" << host << "', serv " << svc << " (thread " << pthread_self() << ")" << endl;
    if (ret) {
        throw(fireball(19, "unable to getnameinfo() for addr..."));
    }
#endif  /* BROKEN_GETNAMEINFO */

    id_str << "IPSocketAddress[" << host << ":" << svc << "]";

#if MAYBE_A_BAD_IDEA
    // Store this in the cache for later use...
    id_cache = id_str.str();
#endif
    
    return(id_str.str());
}

---