source-engine/devtools/smtpmail/simplesocket.cpp

163 lines
4.1 KiB
C++
Raw Permalink Normal View History

2020-04-22 16:56:21 +00:00
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: simple TCP socket API for communicating as a TCP client over a TEXT
// connection
//
// $Workfile: $
// $Date: $
//
//-----------------------------------------------------------------------------
// $Log: $
//
// $NoKeywords: $
//=============================================================================//
#include <winsock.h>
#include "simplesocket.h"
static REPORTFUNCTION g_SocketReport = NULL;
//-----------------------------------------------------------------------------
// Purpose: intialize sockets
//-----------------------------------------------------------------------------
void SocketInit( void )
{
WSADATA wsData;
WORD wVersionRequested = MAKEWORD(1, 1);
WSAStartup(wVersionRequested, &wsData);
}
//-----------------------------------------------------------------------------
// Purpose: cleanup all socket resources
//-----------------------------------------------------------------------------
void SocketExit( void )
{
WSACleanup();
}
//-----------------------------------------------------------------------------
// Purpose: sets up a reporting function
// Input : *pReportFunction -
//-----------------------------------------------------------------------------
void SocketReport( REPORTFUNCTION pReportFunction )
{
g_SocketReport = pReportFunction;
}
//-----------------------------------------------------------------------------
// Purpose: Open a TCP socket & connect to a given server
// Input : *pServerName - server name (text or ip)
// port - port number of the server
// Output : HSOCKET
//-----------------------------------------------------------------------------
HSOCKET SocketOpen( const char *pServerName, int port )
{
SOCKADDR_IN sockAddr;
SOCKET s;
memset(&sockAddr,0,sizeof(sockAddr));
s = socket( AF_INET, SOCK_STREAM, 0 );
sockAddr.sin_family = AF_INET;
sockAddr.sin_addr.s_addr = inet_addr(pServerName);
if (sockAddr.sin_addr.s_addr == INADDR_NONE)
{
LPHOSTENT lphost;
lphost = gethostbyname(pServerName);
if (lphost != NULL)
{
sockAddr.sin_addr.s_addr = ((LPIN_ADDR)lphost->h_addr)->s_addr;
}
else
{
WSASetLastError(WSAEINVAL);
return FALSE;
}
}
sockAddr.sin_port = htons((u_short)port);
if ( connect( s, (SOCKADDR *)&sockAddr, sizeof(sockAddr) ) == SOCKET_ERROR )
{
// printf("Socket error:%d\n", WSAGetLastError()) ;
return NULL;
}
return (HSOCKET)s;
}
//-----------------------------------------------------------------------------
// Purpose: close the socket opened with SocketOpen()
// Input : socket -
//-----------------------------------------------------------------------------
void SocketClose( HSOCKET socket )
{
SOCKET s = (SOCKET)socket;
closesocket( s );
}
//-----------------------------------------------------------------------------
// Purpose: Write a string to the socket. String is NULL terminated on input,
// but terminator is NOT written to the socket
// Input : socket -
// *pString - string to write
//-----------------------------------------------------------------------------
void SocketSendString( HSOCKET socket, const char *pString )
{
if ( !pString )
return;
int len = (int)strlen( pString );
if ( !len )
return;
if ( send( (SOCKET)socket, pString, len, 0 ) != SOCKET_ERROR )
{
if ( g_SocketReport )
{
g_SocketReport( socket, pString );
}
}
else
{
// printf("Send failed\n");
}
}
//-----------------------------------------------------------------------------
// Purpose: receive input from a socket until a certain string is received
// ASSUME: socket data is all text
// Input : socket -
// *pString - string to match, if NULL, just poll the socket once
//-----------------------------------------------------------------------------
void SocketWait( HSOCKET socket, const char *pString )
{
char buf[1024];
bool done = false;
while ( !done )
{
int len = recv( (SOCKET)socket, buf, sizeof(buf)-1, 0 );
buf[len] = 0;
if ( g_SocketReport )
{
g_SocketReport( socket, buf );
}
if ( !pString || strstr( buf, pString ) )
return;
}
}