/*++

Copyright (c) Microsoft Corporation. All rights reserved.

Module Name:

    Connection.cpp

Abstract:

    Module for the socket connection specfic routines in the driver. 
    Makes Connection to the server given server host and port address.  
    
Environment:

    User mode only


--*/

#include "internal.h"
#include "connection.tmh"


CConnection::CConnection()
/*++

Routine Description:

  Constructor for connection object     
    
Arguments:
    
 None

Return Value:

    VOID

--*/    
{

  // Initialize the socket member as Invalid 
    Trace(
        TRACE_LEVEL_INFORMATION, 
        "%!FUNC!"
        );
    m_socket = INVALID_SOCKET;
}

HRESULT 
CConnection::Connect(
    IN IWDFDevice *pDevice
   )
/*++

Routine Description:

   This routine is for the initialization of the connection object associated with
   the File Object . It is invoked from the dispatch OnCreateFile on the default
   queue callback of the driver. It socket connection to the client.

Arguments:
    
  pDevice = Wdf Device Object 

Return Value:

    S_OK if success , error HRESULT otherwise 

--*/      
{
        
   Trace(
        TRACE_LEVEL_INFORMATION, 
        "%!FUNC!"
        );
   
    HRESULT hr = S_OK;

    addrinfoW* info = NULL ;

    PWSTR  hostStr = NULL; 

    PWSTR  portStr = NULL;

    //
    // Reads the host and port strings stored in the device context. 
    //

    DeviceContext *pContext = NULL;
 
    hr = pDevice->RetrieveContext((void**)&pContext);

    if ( FAILED(hr) ) 
    {

        Trace(
            TRACE_LEVEL_ERROR,
            L"ERROR: unable to retrieve context  from wdf device object %!hresult!",
            hr
            );
        goto Clean0;
     
    }

     hostStr = pContext->hostStr;
     
     portStr = pContext->portStr;

     //
     // lookup hostname with addrinfo  hints;
     //
    
    addrinfoW hints;

    ZeroMemory(&hints,sizeof(hints));
    
    hints.ai_family = AF_INET;
    
    hints.ai_socktype = SOCK_STREAM;
    
    hints.ai_protocol = IPPROTO_TCP;

    int n = GetAddrInfoW(hostStr, portStr, &hints, &info);

    if (n != 0)
    {
        DWORD err = WSAGetLastError();
        Trace(
            TRACE_LEVEL_ERROR,
            L"ERROR: Unable to find address/port of host %!winerr!",
            err
            );
        hr = HRESULT_FROM_WIN32(err);
        goto Clean0;
    }

    //
    // Create a socket with this infomation recvd in getaddrinfo
    //
    m_socket = socket(info->ai_family,info->ai_socktype,info->ai_protocol);

    if (m_socket == INVALID_SOCKET) 
    {
        DWORD err = WSAGetLastError();

        Trace(
            TRACE_LEVEL_ERROR,
            L"ERROR: Unable to create socket %!winerr!",
            err
            );

        hr = HRESULT_FROM_WIN32(err);

        goto Clean0;
    }

    //
    // If that succeeds , proceed to connect to the socket 
    //

    
    ATLASSERT(info->ai_addrlen <= 0x7fffffff);

    int nret = connect(m_socket,info->ai_addr,(int)info->ai_addrlen);

    if (nret == SOCKET_ERROR)
    {
        DWORD err = WSAGetLastError();
        Trace(
            TRACE_LEVEL_ERROR,
            L"ERROR: Unable to connect to host %!winerr!",
            err
            );
        hr = HRESULT_FROM_WIN32(err);

        goto Clean0;
    }

 
Clean0:
 
    if (info != NULL)
    {
        FreeAddrInfoW(info);

    }

    if (FAILED(hr) && m_socket != INVALID_SOCKET)
    {
        closesocket(m_socket);
        m_socket = INVALID_SOCKET;
    }
      
    return hr;
    
}

HANDLE 
CConnection::GetSocketHandle(
    )
/*++

Routine Description:

  Function returns the socket handle associated with this connection object    
    
Arguments:
    
 None

Return Value:

    Socket handle if valid socket 
    INVALID_HANDLE_VALUE otherwise 

--*/        
{
    Trace(
        TRACE_LEVEL_INFORMATION, 
        "%!FUNC!"
        );
    if ( INVALID_SOCKET != m_socket ) 
    {
        return (HANDLE)m_socket ;
    }
    else 
    {
        return INVALID_HANDLE_VALUE;
    }
        
}


VOID
CConnection::Close()
/*++

Routine Description:

  Closes the socket connection to the server associated with this connection object 
    
Arguments:
    
 None

Return Value:

    None
--*/        
{
    Trace(
        TRACE_LEVEL_INFORMATION, 
        "%!FUNC!"
        );
    if (m_socket != INVALID_SOCKET)
    {
        closesocket(m_socket);
        m_socket = INVALID_SOCKET;
    }  

}