mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
383 lines
8.9 KiB
383 lines
8.9 KiB
#include "precomp.h"
|
|
DEBUG_FILEZONE(ZONE_T120_MSMCSTCP);
|
|
/* Socket.cpp
|
|
*
|
|
* Copyright (c) 1996 by Microsoft Corporation
|
|
*
|
|
* Abstract:
|
|
* This is the implementation of our socket constructor/destructor functions.
|
|
*
|
|
*/
|
|
#include "socket.h"
|
|
#include "plgxprt.h"
|
|
|
|
/* Size of listen queue */
|
|
#define LISTEN_QUEUE_SIZE 3
|
|
|
|
/* External definitions */
|
|
extern HWND TCP_Window_Handle;
|
|
extern PTransportInterface g_Transport;
|
|
|
|
/*
|
|
* void CreateAndConfigureListenSocket (VOID)
|
|
*
|
|
* Functional Description
|
|
* This function sets up a listening socket.
|
|
* returns INVALID_SOCKET if there is any error.
|
|
*/
|
|
SOCKET CreateAndConfigureListenSocket (VOID)
|
|
{
|
|
SOCKADDR_IN socket_control;
|
|
SOCKET Socket;
|
|
|
|
// Create the listening socket.
|
|
Socket = socket (AF_INET, SOCK_STREAM, 0);
|
|
|
|
if (Socket == INVALID_SOCKET) {
|
|
WARNING_OUT (("Socket: error creating listening socket (errno = %d)", WSAGetLastError()));
|
|
goto Error;
|
|
}
|
|
|
|
// The listen socket only waits for FD_ACCEPT msgs.
|
|
ASSERT(TCP_Window_Handle);
|
|
if (WSAAsyncSelect (Socket,
|
|
TCP_Window_Handle,
|
|
WM_SOCKET_NOTIFICATION,
|
|
FD_ACCEPT) != 0) {
|
|
|
|
WARNING_OUT (("CreateAndConfigureListenSocket: Error on WSAAsyncSelect = %d", WSAGetLastError()));
|
|
goto Error;
|
|
}
|
|
|
|
/*
|
|
* Load the socket control structure with the parameters necessary.
|
|
* - Internet socket
|
|
* - Let it assign any address to this socket
|
|
* - Assign our port number
|
|
*/
|
|
socket_control.sin_family = AF_INET;
|
|
socket_control.sin_addr.s_addr = INADDR_ANY;
|
|
socket_control.sin_port = htons ( TCP_PORT_NUMBER );
|
|
|
|
/* Issue the bind call */
|
|
if (bind (Socket, (LPSOCKADDR) &socket_control, sizeof(SOCKADDR_IN)) != 0) {
|
|
WARNING_OUT (("Socket::Listen: bind failed: Unable to use WinSock"));
|
|
goto Error;
|
|
}
|
|
|
|
/*
|
|
* Issue a listen to WinSock to tell it we are willing to accept calls.
|
|
* This is a non-blocking listen, therefore we will receive FD_ACCEPT
|
|
* if someone is trying to call us.
|
|
*/
|
|
if (listen (Socket, LISTEN_QUEUE_SIZE) != 0) {
|
|
WARNING_OUT (("Socket::Listen: listen failed: Unable to use WinSock"));
|
|
goto Error;
|
|
}
|
|
ASSERT(Socket != INVALID_SOCKET);
|
|
|
|
return Socket;
|
|
|
|
Error:
|
|
|
|
if (INVALID_SOCKET != Socket)
|
|
{
|
|
::closesocket(Socket);
|
|
}
|
|
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
|
|
/*
|
|
* PSocket newSocket (SOCKET socket_number)
|
|
*
|
|
* Functional Description:
|
|
* This is a constructor for the Socket object. It allocates the
|
|
* send and receive buffers and sets up internal variables.
|
|
*/
|
|
PSocket newSocket(TransportConnection XprtConn, PSecurityContext pSC)
|
|
{
|
|
if (IS_SOCKET(XprtConn))
|
|
{
|
|
return ::newSocketEx(XprtConn, pSC);
|
|
}
|
|
return g_pSocketList->FindByTransportConnection(XprtConn, TRUE);
|
|
}
|
|
|
|
|
|
PSocket newPluggableSocket(TransportConnection XprtConn)
|
|
{
|
|
PSocket pSocket = ::newSocketEx(XprtConn, NULL);
|
|
if (NULL != pSocket)
|
|
{
|
|
g_pSocketList->SafeAppend(pSocket);
|
|
}
|
|
return pSocket;
|
|
}
|
|
|
|
|
|
PSocket newSocketEx(TransportConnection XprtConn, PSecurityContext pSC)
|
|
{
|
|
BOOL fRet;
|
|
DBG_SAVE_FILE_LINE
|
|
PSocket pSocket = new CSocket(&fRet, XprtConn, pSC);
|
|
if (NULL != pSocket)
|
|
{
|
|
if (fRet)
|
|
{
|
|
return pSocket;
|
|
}
|
|
pSocket->Release();
|
|
}
|
|
ERROR_OUT(("newSocket: Unable to allocate memory for Socket struct, pSocket=0x%x", pSocket));
|
|
return NULL;
|
|
}
|
|
|
|
|
|
CSocket::CSocket(BOOL *_pfRet, TransportConnection _XprtConn, PSecurityContext _pSC)
|
|
:
|
|
CRefCount(MAKE_STAMP_ID('S','o','c','k')),
|
|
State(IS_SOCKET(_XprtConn) ? NOT_CONNECTED : SOCKET_CONNECTED),
|
|
SecState((NULL == _pSC) ? SC_UNDETERMINED : SC_SECURE),
|
|
pSC(_pSC),
|
|
Max_Packet_Length(DEFAULT_MAX_X224_SIZE),
|
|
Current_Length(0),
|
|
Data_Indication_Buffer(NULL),
|
|
Data_Indication_Length(0),
|
|
Read_State(READ_HEADER),
|
|
X224_Length(0),
|
|
bSpaceAllocated(FALSE),
|
|
Data_Memory(NULL),
|
|
fExtendedX224(FALSE),
|
|
fIncomingSecure(FALSE),
|
|
XprtConn(_XprtConn)
|
|
{
|
|
// assume failure
|
|
*_pfRet = FALSE;
|
|
|
|
// zero out sub structures
|
|
::ZeroMemory(&X224_Header, sizeof(X224_Header));
|
|
::ZeroMemory(&Retry_Info, sizeof(Retry_Info));
|
|
Remote_Address[0] = '\0';
|
|
|
|
if (IS_SOCKET(XprtConn))
|
|
{
|
|
if (INVALID_SOCKET == XprtConn.nLogicalHandle)
|
|
{
|
|
/* Create a STREAM socket (fully reliable, full duplex, and sequenced) */
|
|
if ((XprtConn.nLogicalHandle = ::socket(AF_INET, SOCK_STREAM, 0))
|
|
== INVALID_SOCKET)
|
|
{
|
|
ERROR_OUT (("CSocket: error acquiring INET socket # (errno = %d)", WSAGetLastError()));
|
|
return;
|
|
}
|
|
}
|
|
|
|
/* Enable Tx and Rx messages to the window */
|
|
ASSERT(TCP_Window_Handle);
|
|
if (::WSAAsyncSelect(XprtConn.nLogicalHandle, TCP_Window_Handle,
|
|
WM_SOCKET_NOTIFICATION,
|
|
FD_READ | FD_WRITE | FD_CLOSE | FD_CONNECT) != 0)
|
|
{
|
|
WARNING_OUT (("CSocket: Error on WSAAsyncSelect = %d", WSAGetLastError()));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ASSERT(IS_PLUGGABLE(XprtConn));
|
|
CPluggableConnection *p = ::GetPluggableConnection(this);
|
|
if (NULL == p)
|
|
{
|
|
ERROR_OUT(("newSocket: Unable to find plugable transport (%d, %d)",
|
|
XprtConn.eType, XprtConn.nLogicalHandle));
|
|
return;
|
|
}
|
|
}
|
|
|
|
// success
|
|
*_pfRet = TRUE;
|
|
}
|
|
|
|
|
|
/*
|
|
* void freeSocket (PSocket, TransportConnection)
|
|
*
|
|
* Functional Description:
|
|
* This is a destructor for the Socket object. It frees the send
|
|
* and receive buffers and connection structure.
|
|
* It will also cleanup the listening socket. In this case,
|
|
* "pSocket" is set to NULL and "trash_packets" should be set to TRUE.
|
|
*/
|
|
void freeSocket(PSocket pSocket, TransportConnection XprtConn)
|
|
{
|
|
if (IS_SOCKET(XprtConn))
|
|
{
|
|
if (NULL != g_pSocketList)
|
|
{
|
|
g_pSocketList->SafeRemove(pSocket);
|
|
}
|
|
freeSocketEx(pSocket, XprtConn);
|
|
}
|
|
}
|
|
|
|
|
|
void freeListenSocket(TransportConnection XprtConn)
|
|
{
|
|
ASSERT(IS_SOCKET(XprtConn));
|
|
freeSocketEx(NULL, XprtConn);
|
|
}
|
|
|
|
|
|
void freePluggableSocket(PSocket pSocket)
|
|
{
|
|
freeSocketEx(pSocket, pSocket->XprtConn);
|
|
if (NULL != g_pSocketList)
|
|
{
|
|
g_pSocketList->SafeRemove(pSocket);
|
|
}
|
|
}
|
|
|
|
|
|
void freeSocketEx(PSocket pSocket, TransportConnection XprtConn)
|
|
{
|
|
// Either "pSocket" is NULL, or the socket is not invalid.
|
|
#ifdef _DEBUG
|
|
if (IS_SOCKET(XprtConn))
|
|
{
|
|
if (NULL != pSocket)
|
|
{
|
|
ASSERT(INVALID_SOCKET != pSocket->XprtConn.nLogicalHandle);
|
|
}
|
|
else
|
|
{
|
|
// it is a listen socket
|
|
ASSERT(INVALID_SOCKET != XprtConn.nLogicalHandle);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
// Determine the socket number to use... Either the socket is the
|
|
// socket indicated in the PSocket structure, or it is a structure-less
|
|
// listen socket. Note: both cannot be valid!
|
|
|
|
if (IS_SOCKET(XprtConn))
|
|
{
|
|
SOCKET socket = (pSocket) ? pSocket->XprtConn.nLogicalHandle : XprtConn.nLogicalHandle;
|
|
XprtConn.nLogicalHandle = socket;
|
|
|
|
/* Disable notifications to our window */
|
|
if (::IsWindow(TCP_Window_Handle))
|
|
{
|
|
::WSAAsyncSelect(socket, TCP_Window_Handle, 0, 0);
|
|
}
|
|
}
|
|
|
|
if (pSocket != NULL)
|
|
{
|
|
pSocket->Release();
|
|
}
|
|
else
|
|
{
|
|
// This is the listening socket
|
|
::ShutdownAndClose (XprtConn, FALSE, 0);
|
|
}
|
|
}
|
|
|
|
|
|
CSocket::~CSocket(void)
|
|
{
|
|
switch (State)
|
|
{
|
|
case SOCKET_CONNECTED:
|
|
// case WAITING_FOR_DISCONNECT:
|
|
/* All physically connected states issue a shutdown() first */
|
|
::ShutdownAndClose(XprtConn, TRUE, SD_BOTH);
|
|
break;
|
|
|
|
case X224_CONNECTED:
|
|
// Shutdown disable reception only.
|
|
::ShutdownAndClose(XprtConn, TRUE, SD_RECEIVE);
|
|
break;
|
|
|
|
default:
|
|
::ShutdownAndClose(XprtConn, FALSE, 0);
|
|
break;
|
|
}
|
|
|
|
/* Free the structures */
|
|
FreeTransportBuffer();
|
|
delete pSC;
|
|
}
|
|
|
|
|
|
void CSocket::FreeTransportBuffer(void)
|
|
{
|
|
if (NULL != Data_Memory)
|
|
{
|
|
::FreeMemory(Data_Memory);
|
|
Data_Memory = NULL;
|
|
Data_Indication_Buffer = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
void CSocketList::SafeAppend(PSocket pSocket)
|
|
{
|
|
::EnterCriticalSection(&g_csTransport);
|
|
if (! Find(pSocket))
|
|
{
|
|
Append(pSocket);
|
|
}
|
|
::LeaveCriticalSection(&g_csTransport);
|
|
}
|
|
|
|
|
|
BOOL CSocketList::SafeRemove(PSocket pSocket)
|
|
{
|
|
::EnterCriticalSection(&g_csTransport);
|
|
BOOL fRet = Remove(pSocket);
|
|
::LeaveCriticalSection(&g_csTransport);
|
|
return fRet;
|
|
}
|
|
|
|
|
|
PSocket CSocketList::FindByTransportConnection(TransportConnection XprtConn, BOOL fNoAddRef)
|
|
{
|
|
PSocket pSocket;
|
|
::EnterCriticalSection(&g_csTransport);
|
|
Reset();
|
|
while (NULL != (pSocket = Iterate()))
|
|
{
|
|
if (IS_SAME_TRANSPORT_CONNECTION(pSocket->XprtConn, XprtConn))
|
|
{
|
|
if (! fNoAddRef)
|
|
{
|
|
pSocket->AddRef();
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
::LeaveCriticalSection(&g_csTransport);
|
|
return pSocket;
|
|
}
|
|
|
|
|
|
PSocket CSocketList::RemoveByTransportConnection(TransportConnection XprtConn)
|
|
{
|
|
PSocket pSocket;
|
|
::EnterCriticalSection(&g_csTransport);
|
|
Reset();
|
|
while (NULL != (pSocket = Iterate()))
|
|
{
|
|
if (IS_SAME_TRANSPORT_CONNECTION(pSocket->XprtConn, XprtConn))
|
|
{
|
|
Remove(pSocket);
|
|
break;
|
|
}
|
|
}
|
|
::LeaveCriticalSection(&g_csTransport);
|
|
return pSocket;
|
|
}
|
|
|