Initial publish

1 files changed, 263 insertions, 0 deletions

diff --git a/general/echo/umdfSocketEcho/Driver/Connection.cpp b/general/echo/umdfSocketEcho/Driver/Connection.cpp
new file mode 100644
index 00000000..e28d0c56
--- /dev/null
+++ b/general/echo/umdfSocketEcho/Driver/Connection.cpp
@@ -0,0 +1,263 @@
+/*++
+
+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;
+    }  
+
+}