common/sockconn.cc

Go to the documentation of this file.

//
// The SocketConnector class - establish/maintain a network connection
//
// ***** 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):
// 
// 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 <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
}

#include <sstream>

#include "consmgr.h"
#include "guard.h"
#include "sockconn.h"

SocketConnector::SocketConnector(SocketAddressP &addr)
        : FDConnector(), sock_addr(addr)
{
    // XXX - This may not fire our sub-classes connect() method, if there
    // is one.  This object is not fully constructed.

    // Initiate the connection
    connect();
}


//
// This function actually makes the connection, blocking if need be,
// only returning when complete (successfully.  Throws exceptions
// on failure).  It is intended to be used during construction, so no
// mutex locking is done.
//

void
SocketConnector::connect()
{
    int         ret, s;

    // XXX - Should check that we're not already connected?

    clog << "Making a new connection to " << sock_addr->get_id() << "... ";

    /* Open a socket */
    s = ::socket(sock_addr->family(), SOCK_STREAM, 0);
    if (s < 0) {
        clog << "failed!" << endl;
        throw(fireball(6, "socket(2) failed", true));
    }


    ret = ::connect(s, (sockaddr*)sock_addr->get_addr(), sock_addr->length());
    if ((ret < 0) && (errno != EINPROGRESS)) {
        clog << "failed!" << endl;
        throw(fireball(7, "connect(2) failed", true));
    }

#ifdef SO_KEEPALIVE
    /* Set TCP KEEPALIVEs on this socket... */
    int         flags = 1;

    if (setsockopt(s, SOL_SOCKET, SO_KEEPALIVE, (char *)&flags,
                   sizeof(flags)) < 0) {
        throw(fireball(10, "setsockopt(SO_KEEPALIVE)", true));
    }
#endif

    clog << "successful!" << endl;

    // Set that we're connected, and broadcast the state change
    FDConnector::connect(s);
}

//
// Generate a string to tell a human who we are...
//

std::string
SocketConnector::get_id() const
{
    std::ostringstream  id_str;

    id_str << "SocketConnector[" << sock_addr->get_id() << "]";

    return(id_str.str());
}

---