mirror of
https://github.com/nillerusr/source-engine.git
synced 2026-08-08 01:39:36 +00:00
1
This commit is contained in:
FluorescentCIAAfricanAmerican
@@ -0,0 +1,162 @@
|
||||
//========= 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;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,40 @@
|
||||
//========= 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: $
|
||||
//=============================================================================//
|
||||
|
||||
#ifndef SIMPLESOCKET_H
|
||||
#define SIMPLESOCKET_H
|
||||
#pragma once
|
||||
|
||||
// opaque socket type
|
||||
typedef struct socket_s socket_t, *HSOCKET;
|
||||
|
||||
// Socket reporting function
|
||||
typedef void ( *REPORTFUNCTION )( HSOCKET socket, const char *pString );
|
||||
|
||||
extern HSOCKET SocketOpen( const char *pServerName, int port );
|
||||
extern void SocketClose( HSOCKET socket );
|
||||
extern void SocketSendString( HSOCKET socket, const char *pString );
|
||||
|
||||
// This should probably return the data so we can handle errors and receive data
|
||||
extern void SocketWait( HSOCKET socket, const char *pString );
|
||||
|
||||
// sets the reporting function
|
||||
extern void SocketReport( REPORTFUNCTION pReportFunction );
|
||||
extern void SocketInit( void );
|
||||
extern void SocketExit( void );
|
||||
|
||||
|
||||
|
||||
|
||||
#endif // SIMPLESOCKET_H
|
||||
@@ -0,0 +1,247 @@
|
||||
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||||
//
|
||||
// Purpose:
|
||||
//
|
||||
// $NoKeywords: $
|
||||
//
|
||||
// SMTP RFC 821 Specifies <CR><LF> to terminate lines.
|
||||
//
|
||||
// <CR> = Carriage Return C: \r Dec: 13 Hex: 0x0d Oct: 0015
|
||||
// <LF> = Line Feed ( Newline ) C: \n Dec: 10 Hex: 0x0a Oct: 0012
|
||||
//
|
||||
//=============================================================================//
|
||||
#pragma warning( disable : 4530 )
|
||||
|
||||
#include <string>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdarg.h>
|
||||
#include <time.h>
|
||||
|
||||
#include "simplesocket.h"
|
||||
|
||||
static int g_ArgCount;
|
||||
static char **g_Args;
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Purpose: simple args API
|
||||
// Input : argc -
|
||||
// *argv[] -
|
||||
//-----------------------------------------------------------------------------
|
||||
void ArgsInit( int argc, char *argv[] )
|
||||
{
|
||||
g_ArgCount = argc;
|
||||
g_Args = argv;
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Purpose: tests for the presence of arg "pName"
|
||||
// Input : *pName -
|
||||
// Output : Returns true on success, false on failure.
|
||||
//-----------------------------------------------------------------------------
|
||||
bool ArgsExist( const char *pName )
|
||||
{
|
||||
int i;
|
||||
|
||||
if ( pName && pName[0] == '-' )
|
||||
{
|
||||
for ( i = 0; i < g_ArgCount; i++ )
|
||||
{
|
||||
// Is this a switch?
|
||||
if ( g_Args[i][0] != '-' )
|
||||
continue;
|
||||
|
||||
if ( !stricmp( pName, g_Args[i] ) )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Purpose: looks for the arg "pName" and returns it's parameter or pDefault otherwise
|
||||
// Input : *pName -
|
||||
// *pDefault -
|
||||
// Output : const char *
|
||||
//-----------------------------------------------------------------------------
|
||||
const char *ArgsGet( const char *pName, const char *pDefault )
|
||||
{
|
||||
int i;
|
||||
|
||||
// don't bother with the last arg, it can't be a switch with a parameter
|
||||
for ( i = 0; i < g_ArgCount-1; i++ )
|
||||
{
|
||||
// Is this a switch?
|
||||
if ( g_Args[i][0] != '-' )
|
||||
continue;
|
||||
|
||||
if ( !stricmp( pName, g_Args[i] ) )
|
||||
{
|
||||
// If the next arg is not a switch, return it
|
||||
if ( g_Args[i+1][0] != '-' )
|
||||
return g_Args[i+1];
|
||||
}
|
||||
}
|
||||
|
||||
return pDefault;
|
||||
}
|
||||
|
||||
|
||||
void Error( const char *pString, ... )
|
||||
{
|
||||
va_list list;
|
||||
va_start( list, pString );
|
||||
vprintf( pString, list );
|
||||
printf("Usage: smtpmail -to <address(-es, ';' separated)> -from <address> -subject \"subject text\" [-verbose] [-server server.name] <FILENAME.TXT>\n" );
|
||||
exit( 1 );
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Purpose: simple routine to printf() all of the socket traffic for -verbose
|
||||
// Input : socket -
|
||||
// *pData -
|
||||
//-----------------------------------------------------------------------------
|
||||
void DumpSocket( HSOCKET, const char *pData )
|
||||
{
|
||||
printf( "%s", pData );
|
||||
fflush( stdout );
|
||||
}
|
||||
|
||||
|
||||
static char *months[] = { "JAN", "FEB", "MAR", "APR", "MAY", "JUN", "JUL", "AUG", "SEP", "OCT", "NOV", "DEC" };
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Purpose: mail a text file using the SMTP mail server connected to socket
|
||||
// Input : socket -
|
||||
// *pFrom - from address
|
||||
// *pTo - to address
|
||||
// *pSubject - subject of the mail
|
||||
// *fp - text mode file opened
|
||||
//-----------------------------------------------------------------------------
|
||||
void MailSendFile( HSOCKET socket, const char *pFrom, const char *pTo[], const char *pSubject, FILE *fp )
|
||||
{
|
||||
int i;
|
||||
char buf[1024];
|
||||
|
||||
SocketSendString( socket, "HELO\r\n" );
|
||||
SocketWait( socket, "\n" );
|
||||
sprintf( buf, "MAIL FROM: <%s>\r\n", pFrom );
|
||||
SocketSendString( socket, buf );
|
||||
SocketWait( socket, "\n" );
|
||||
for (i = 0; pTo[i] != NULL; i++)
|
||||
{
|
||||
sprintf( buf, "RCPT TO: <%s>\r\n", pTo[i] );
|
||||
SocketSendString( socket, buf );
|
||||
SocketWait( socket, "\n" );
|
||||
}
|
||||
SocketSendString( socket, "DATA\r\n" );
|
||||
SocketWait( socket, "\n" );
|
||||
|
||||
time_t currentTime;
|
||||
time( ¤tTime );
|
||||
struct tm *localTime = gmtime( ¤tTime );
|
||||
|
||||
sprintf( buf, "DATE: %02d %s %4d %02d:%02d:%02d\r\n", localTime->tm_mday, months[localTime->tm_mon], localTime->tm_year+1900,
|
||||
localTime->tm_hour, localTime->tm_min, localTime->tm_sec );
|
||||
|
||||
SocketSendString( socket, buf );
|
||||
sprintf( buf, "FROM: %s\r\n", pFrom );
|
||||
SocketSendString( socket, buf );
|
||||
for (i = 0; pTo[i] != NULL; i++)
|
||||
{
|
||||
sprintf( buf, "TO: %s\r\n", pTo[i] );
|
||||
SocketSendString( socket, buf );
|
||||
}
|
||||
sprintf( buf, "SUBJECT: %s\r\n\r\n", pSubject );
|
||||
SocketSendString( socket, buf );
|
||||
|
||||
while ( !feof( fp ) )
|
||||
{
|
||||
fgets( buf, 1024, fp );
|
||||
|
||||
// A period on a line by itself would end the transmission
|
||||
if ( !strcmp( buf, ".\n" ) && !strcmp( buf, ".\r\n" ) )
|
||||
continue;
|
||||
|
||||
SocketSendString( socket, buf );
|
||||
}
|
||||
SocketSendString( socket, "\r\n.\r\n" );
|
||||
SocketWait( socket, "\n" );
|
||||
}
|
||||
|
||||
|
||||
|
||||
int main( int argc, char *argv[] )
|
||||
{
|
||||
|
||||
ArgsInit( argc, argv );
|
||||
|
||||
const char *pServerName = ArgsGet( "-server", "smtp1.valvesoftware.com" );
|
||||
|
||||
const char *pTo[100];
|
||||
pTo[0] = ArgsGet( "-to", NULL );
|
||||
if ( !pTo[0] )
|
||||
{
|
||||
Error( "Must specify a recipient with -to <address(es)>\n" );
|
||||
}
|
||||
|
||||
// split ; separated To string into an array of strings
|
||||
int c = 0;
|
||||
char *where;
|
||||
while (((where = (char *)strchr(pTo[c++],';')) != NULL) && c < 99)
|
||||
{
|
||||
*(where++) = '\0';
|
||||
pTo[c] = where;
|
||||
}
|
||||
pTo[c] = NULL;
|
||||
|
||||
const char *pFrom = ArgsGet( "-from", NULL );
|
||||
if ( !pFrom )
|
||||
{
|
||||
Error( "Must specify a sender with -from <address>\n" );
|
||||
}
|
||||
const char *pSubject = ArgsGet( "-subject", "<NONE>" );
|
||||
int port = atoi( ArgsGet( "-port", "25" ) );
|
||||
|
||||
const char *pFileName = argv[ argc-1 ];
|
||||
|
||||
FILE *fp = fopen( pFileName, "r" );
|
||||
|
||||
if ( !fp )
|
||||
{
|
||||
Error( "Can't open %s\n", pFileName );
|
||||
}
|
||||
|
||||
SocketInit();
|
||||
|
||||
// in verbose mode, printf all of the traffic
|
||||
if ( ArgsExist( "-verbose" ) )
|
||||
{
|
||||
SocketReport( DumpSocket );
|
||||
}
|
||||
|
||||
HSOCKET socket = SocketOpen( pServerName, port );
|
||||
if ( socket )
|
||||
{
|
||||
SocketWait( socket, "\n" );
|
||||
MailSendFile( socket, pFrom, pTo, pSubject, fp );
|
||||
SocketSendString( socket, "quit\r\n" );
|
||||
SocketWait( socket, "\n" );
|
||||
SocketClose( socket );
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("Can't open socket to %s:%d\n", pServerName, port );
|
||||
}
|
||||
|
||||
fclose( fp );
|
||||
|
||||
SocketExit();
|
||||
return 0;
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
//-----------------------------------------------------------------------------
|
||||
// SMTPMAIL.VPC
|
||||
//
|
||||
// Project Script
|
||||
//-----------------------------------------------------------------------------
|
||||
|
||||
$Macro SRCDIR "..\.."
|
||||
$Macro OUTBINDIR "..\bin"
|
||||
|
||||
$Include "$SRCDIR\vpc_scripts\source_exe_con_base.vpc"
|
||||
|
||||
$Configuration
|
||||
{
|
||||
$Compiler
|
||||
{
|
||||
$AdditionalIncludeDirectories "$BASE,..\common,..\vmpi"
|
||||
$PreprocessorDefinitions "$BASE;MACRO_MATHLIB;PROTECTED_THINGS_DISABLE"
|
||||
}
|
||||
|
||||
$Linker
|
||||
{
|
||||
$AdditionalDependencies "$BASE ws2_32.lib odbc32.lib odbccp32.lib"
|
||||
}
|
||||
}
|
||||
|
||||
$Project "Smtpmail"
|
||||
{
|
||||
$Folder "Source Files"
|
||||
{
|
||||
$File "smtpmail.cpp"
|
||||
$File "simplesocket.cpp"
|
||||
}
|
||||
|
||||
$Folder "Header Files"
|
||||
{
|
||||
$File "simplesocket.h"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user