Socket.cpp

Go to the documentation of this file.

/* 
 * Confdump-Agent - Dump static and runtime system configuration
 * Copyright (C) 2009-2012  Straton IT, SAS
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 * 
 * This program 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <winsock2.h>
#include <ws2tcpip.h>
#include <iphlpapi.h>

#include "QVariant"

#include "boost/scoped_array.hpp"

#include "Confdump/System/Win32Dumper.hpp"
#include "Confdump/System/Win32/Socket.hpp"
#include "Confdump/System/Win32/errors.hpp"

namespace Confdump
{

namespace System
{

namespace Win32
{

Result BasicTcp4Socket::allSystemInstances()
{
    Result table;

    // First call to get the proper size
    DWORD size = 0;
    DWORD rc = GetTcpTable( NULL, &size, TRUE );
    if ( rc != ERROR_INSUFFICIENT_BUFFER )
        throwOnWin32Error( rc, "get the tcp connection table size" );

    // Second call to get the data
    boost::scoped_array<char> mibtablebuf( new char[size] );
    MIB_TCPTABLE *mibtable = reinterpret_cast<MIB_TCPTABLE*>( mibtablebuf.get() );
    rc = GetTcpTable( mibtable, &size, false );
    throwOnWin32Error( rc, "get the tcp connection table" );

    // Convert to objects
    for ( size_t i = 0; i < mibtable->dwNumEntries; ++i )
    {
        boost::shared_ptr<BasicTcp4Socket> socket( new BasicTcp4Socket( mibtable->table[i] ) );
        table.push_back( socket );
    }

    return table;
}

BasicTcp4Socket::BasicTcp4Socket( const MIB_TCPROW &row )
{
    in_addr addr;

    setProperty( "protocolName", "tcp" );

    addr.S_un.S_addr = row.dwLocalAddr;
    setProperty( "localAddressString", QString( "%1:%2" )
        .arg( inet_ntoa( addr ) )
        .arg( ntohs( static_cast<quint16>( row.dwLocalPort ) ) ) );

    addr.S_un.S_addr = row.dwRemoteAddr;
    setProperty( "remoteAddressString", QString( "%1:%2" )
        .arg( inet_ntoa( addr ) )
        .arg( ntohs( static_cast<quint16>( row.dwRemotePort ) ) ) );

    QString state;
    switch ( row.dwState )
    {
    case MIB_TCP_STATE_CLOSED:     state = "CLOSE";       break;
    case MIB_TCP_STATE_LISTEN:     state = "LISTEN";      break;
    case MIB_TCP_STATE_SYN_SENT:   state = "SYN_SENT";    break;
    case MIB_TCP_STATE_SYN_RCVD:   state = "SYN_RECV";    break;
    case MIB_TCP_STATE_ESTAB:      state = "ESTABLISHED"; break;
    case MIB_TCP_STATE_FIN_WAIT1:  state = "FIN_WAIT1";   break;
    case MIB_TCP_STATE_FIN_WAIT2:  state = "FIN_WAIT2";   break;
    case MIB_TCP_STATE_CLOSE_WAIT: state = "CLOSE_WAIT";  break;
    case MIB_TCP_STATE_CLOSING:    state = "CLOSING";     break;
    case MIB_TCP_STATE_LAST_ACK:   state = "LAST_ACK";    break;
    case MIB_TCP_STATE_TIME_WAIT:  state = "TIME_WAIT";   break;
    case MIB_TCP_STATE_DELETE_TCB: state = "DELETE_TCB";  break;
    }
    setProperty( "stateString", state );
}

BasicTcp4Socket::~BasicTcp4Socket()
{
}

}
}
}