Microsoft/Windows2003-3790
Microsoft/Windows2003-3790
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
50969b0ea2
Windows2003-3790/inetsrv/pop3/setup/pop2exch/p2eworker.cpp

1082 lines
38 KiB
C++
Raw Normal View History

First commit
2001-01-01 00:00:00 +01:00
// P2EWorker.cpp: implementation of the CP2EWorker class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "P2EWorker.h"
#include "resource.h"
#include "resource2.h"
#include <sbs6base.h>
#include <P3Admin.h>
#include <pop3server.h>
#include <logutil.h>
#include <mailbox.h>
#include "UserInfo.h"
#include <AuthID.h>
#include <FormatMessage.h>
#include <atlbase.h>
#include <comdef.h>
#include <dsgetdc.h>
#include <lm.h>
#include <tchar.h>
#include <stdio.h>
#include <ws2tcpip.h>
//#include <mswsock.h>
#include <AuthUtil.h>
#ifndef ASSERT
#define ASSERT assert
#endif
LOGFILE(L"Pop2ExchLog.txt");
#define SMTP_BUFFERSIZE 2048
#define SMTP_QUIT "QUIT\r\n"
#define SMTP_DATA "DATA\r\n"
#define SMTP_DATA_END "\r\n.\r\n"
#define SMTP_RSET "RSET\r\n"
#define SMTP_MSG220 "220" // 220 <domain> Service ready
#define SMTP_MSG221 "221" // 221 <domain> Service ready
#define SMTP_MSG250 "250" // 250 Requested mail action okay, completed
#define SMTP_MSG354 "354" // 354 Start mail input; end with <CRLF>.<CRLF>
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CP2EWorker::CP2EWorker() :
m_bSuppressPrintError( false )
{
}
CP2EWorker::~CP2EWorker()
{
}
//////////////////////////////////////////////////////////////////////
// Implementation : public
//////////////////////////////////////////////////////////////////////
int CP2EWorker::CreateUser(int argc, wchar_t *argv[], const bool bCreateUser, const bool bCreateMailbox)
{
USES_CONVERSION;
HRESULT hr;
VARIANT v;
BSTR bstrName, bstrEmailName;
_bstr_t _bstrDomainName;
CComPtr<IP3Config> spIConfig;
CComPtr<IP3Domains> spIDomains;
IP3Domain *pIDomain = NULL;
CComPtr<IEnumVARIANT> spIEnumVARIANT;
bool bErrors = false, bIgnoreErrors = false, bUnregisterDependencies = false;
char sPassword[MAX_PATH];
CUserInfo userInfoX;
if ( 4 != argc && 3 != argc )
return -1;
if ( 2 < argc ) // Since domain name is now required this is no longer a valid option
{ // But I'm going to leave the code in case we ever reverse this decision
if ( 0 == wcscmp( L"/i", argv[2] ) || ( 0 == wcscmp( L"-i", argv[2] )))
bIgnoreErrors = true;
else
_bstrDomainName = argv[2];
}
if ( 4 == argc )
{
if ( bIgnoreErrors )
_bstrDomainName = argv[3];
else if ( 0 == wcscmp( L"/i", argv[3] ) || ( 0 == wcscmp( L"-i", argv[3] )))
bIgnoreErrors = true;
else
return -1;
}
if ( 0 == _bstrDomainName.length() )
return -1;
{ // Log the command
LPWSTR ps2 = argv[2], ps3 = argv[3];
if ( NULL == ps2 )
{
ps2 = L"";
ps3 = NULL;
}
if ( NULL == ps3 )
{
ps3 = L"";
}
LOG( L"%s %s %s %s", argv[0], argv[1], ps2, ps3 );
}
VariantInit( &v );
hr = CoCreateInstance( __uuidof( P3Config ), NULL, CLSCTX_ALL, __uuidof( IP3Config ),reinterpret_cast<LPVOID *>( &spIConfig ));
if ( REGDB_E_CLASSNOTREG == hr )
{
hr = RegisterDependencies();
if ( S_OK == hr )
hr = CoCreateInstance( __uuidof( P3Config ), NULL, CLSCTX_ALL, __uuidof( IP3Config ),reinterpret_cast<LPVOID *>( &spIConfig ));
bUnregisterDependencies = true;
}
if ( S_OK == hr )
{
if ( bCreateUser ) // Only valid for the MD5 Authentication method
{
CComPtr<IAuthMethods> spIAuthMethods;
CComPtr<IAuthMethod> spIAuthMethod;
BSTR bstrID;
_variant_t _v;
long lValue;
WCHAR sBuffer[MAX_PATH];
hr = spIConfig->get_Authentication( &spIAuthMethods );
if ( REGDB_E_CLASSNOTREG == hr )
{
PrintMessage( IDS_ERROR_POP3REQUIRED );
m_bSuppressPrintError = true;
}
if ( S_OK == hr )
hr = spIAuthMethods->get_CurrentAuthMethod( &_v );
if ( S_OK == hr )
hr = spIAuthMethods->get_Item( _v, &spIAuthMethod );
if ( S_OK == hr )
hr = spIAuthMethod->get_ID( &bstrID );
if ( S_OK == hr )
{
if ( 0 != _wcsicmp( bstrID, SZ_AUTH_ID_MD5_HASH ))
hr = HRESULT_FROM_WIN32( ERROR_DS_INAPPROPRIATE_AUTH );
SysFreeString( bstrID );
}
}
}
if ( S_OK == hr )
hr = spIConfig->get_Domains( &spIDomains );
if ( S_OK == hr )
{
CComPtr<IP3Users> spIUsers;
CComPtr<IP3User> spIUser;
if ( 0 == _bstrDomainName.length() )
{ // For each domain
hr = spIDomains->get__NewEnum( &spIEnumVARIANT );
if ( S_OK == hr )
{
hr = spIEnumVARIANT->Next( 1, &v, NULL );
}
}
else
{ // Just for this domain
_variant_t _v( _bstrDomainName );
hr = spIDomains->get_Item( _v, &pIDomain );
}
while ( S_OK == hr )
{
if ( NULL != spIEnumVARIANT.p )
{
if ( VT_DISPATCH != V_VT( &v ))
hr = E_UNEXPECTED;
if ( S_OK == hr )
hr = V_DISPATCH( &v )->QueryInterface( __uuidof( IP3Domain ), reinterpret_cast<void**>( &pIDomain ));
VariantClear( &v );
}
if ( S_OK == hr )
hr = pIDomain->get_Name( &bstrName );
if ( S_OK == hr )
{
LOG( L"Domain: %s ", bstrName );
SysFreeString( bstrName );
}
// Enumerate the users
if ( S_OK == hr )
{
hr = pIDomain->get_Users( &spIUsers );
if ( S_OK == hr )
{ // List users
CComPtr<IEnumVARIANT> spIEnumVARIANT2;
hr = spIUsers->get__NewEnum( &spIEnumVARIANT2 );
if ( S_OK == hr )
hr = spIEnumVARIANT2->Next( 1, &v, NULL );
while ( S_OK == hr )
{
if ( VT_DISPATCH != V_VT( &v ))
hr = E_UNEXPECTED;
if ( S_OK == hr )
hr = V_DISPATCH( &v )->QueryInterface( __uuidof( IP3User ), reinterpret_cast<void**>( &spIUser ));
VariantClear( &v );
hr = spIUser->get_Name( &bstrName );
if ( S_OK == hr )
{
if ( S_OK == hr )
hr = userInfoX.SetUserName( bstrName );
if ( S_OK == hr )
{
hr = spIUser->get_EmailName( &bstrEmailName );
if ( S_OK == hr )
{
hr = GetMD5Password( bstrEmailName, sPassword );
if ( S_OK == hr )
hr = userInfoX.SetPassword( A2W( sPassword ));
}
}
if ( S_OK == hr )
{
if ( bCreateUser )
{ // Create the user
hr = userInfoX.CreateAccount();
if ( S_OK == hr )
LOG( L"Create user: %s succeeded.", bstrName );
else
{
LOG( L"ERROR!: Create user %s failed. %s", bstrName, CFormatMessage(hr).c_str() );
if ( bIgnoreErrors )
{
hr = S_OK;
bErrors = true;
}
}
}
if ( bCreateMailbox )
{ // Create the mailbox
hr = userInfoX.CreateMailbox();
if ( S_OK == hr )
LOG( L"Create mailbox: %s succeeded.", bstrName );
else
{
LOG( L"ERROR!: Create mailbox %s failed. %s", bstrName, CFormatMessage(hr).c_str() );
if ( bIgnoreErrors )
{
hr = S_OK;
bErrors = true;
}
}
}
}
SysFreeString( bstrName );
}
if ( S_OK == hr )
hr = spIEnumVARIANT2->Next( 1, &v, NULL );
}
if ( S_FALSE == hr )
hr = S_OK; // Reached end of user enumeration
}
}
if ( S_OK == hr )
{
if ( NULL != spIEnumVARIANT.p )
{
hr = spIEnumVARIANT->Next( 1, &v, NULL );
if ( S_OK == hr )
pIDomain->Release();
}
else
hr = S_FALSE; // Were done.
}
}
if ( S_FALSE == hr )
hr = S_OK; // Reached end of domain enumeration
}
if ( NULL != pIDomain )
pIDomain->Release();
if ( bCreateUser )
{
if ( S_OK == hr )
{
PrintMessage( IDS_SUCCESS_CREATEUSER );
if ( bErrors )
PrintMessage( IDS_WARNING_ERRORSIGNORED );
}
else
PrintMessage( IDS_ERROR_CREATEUSER_FAILED );
}
if ( bCreateMailbox )
{
if ( S_OK == hr )
{
PrintMessage( IDS_SUCCESS_CREATEMAILBOX );
if ( bErrors )
PrintMessage( IDS_WARNING_ERRORSIGNORED );
}
else
PrintMessage( IDS_ERROR_CREATEMAILBOX_FAILED );
}
if ( bUnregisterDependencies )
UnRegisterDependencies();
return hr;
}
#define LOCALHOST L"localhost"
int CP2EWorker::Mail(int argc, wchar_t *argv[], bool bDelete /*= FALSE*/ )
{
USES_CONVERSION;
HRESULT hr = S_OK;
VARIANT v;
DWORD dwCount, dwIndex;
BSTR bstrName, bstrEmailName;
WSTRING sEmailDomain;
ASTRING sASCFROM;
_bstr_t _bstrDomainName;
CComPtr<IP3Config> spIConfig;
CComPtr<IP3Domains> spIDomains;
IP3Domain *pIDomain = NULL;
CComPtr<IEnumVARIANT> spIEnumVARIANT;
bool bErrors = false, bIgnoreErrors = false, bLocked = false, bUnregisterDependencies = false;
CMailBox mailboxX;
WCHAR sFilename[MAX_PATH];
ASTRING sASCComputerName;
WSTRING sComputerName;
DWORD dwFileNameSize = MAX_PATH;
LPWSTR psArgv;
LPSTR pszComputerName;
if ( 5 < argc || 3 > argc )
return -1;
for ( int i = 2; i < argc; i++ )
{
psArgv = argv[i];
if ( 0 == wcscmp( L"/i", psArgv ) || ( 0 == wcscmp( L"-i", psArgv )))
{
if ( bIgnoreErrors )
return -1;
bIgnoreErrors = true;
}
else if ( 0 == _wcsnicmp( L"/S:", psArgv, 3 ) || 0 == _wcsnicmp( L"-S:", psArgv, 3 ))
{
if ( 3 < wcslen( psArgv ))
{
if ( sComputerName.empty() )
sComputerName = psArgv + 3;
else
return -1;
}
else
return -1;
}
else
{
if ( 0 == _bstrDomainName.length() )
_bstrDomainName = psArgv;
else
return -1;
}
}
if ( 0 == _bstrDomainName.length() )
return -1;
if ( sComputerName.empty() )
sComputerName = LOCALHOST;
sASCComputerName = W2A(sComputerName.c_str());
{
LPWSTR ps2 = argv[2], ps3 = argv[3], ps4 = argv[4];
if ( NULL == ps2 )
{
ps2 = L"";
ps3 = L"";
ps4 = L"";
}
if ( NULL == ps3 )
{
ps3 = L"";
ps4 = L"";
}
if ( NULL == ps4 )
ps4 = L"";
LOG( L"%s %s %s %s %s", argv[0], argv[1], ps2, ps3, ps4 );
}
VariantInit( &v );
// Socket Initialization
WSADATA wsaData;
SOCKET socket = INVALID_SOCKET;
int iErr, iSize;
LPSTR psSendBuffer = new CHAR[SMTP_BUFFERSIZE];
if ( NULL == psSendBuffer )
return E_OUTOFMEMORY;
iErr = WSAStartup( MAKEWORD( 2, 2 ), &wsaData );
if( 0 != iErr )
{
wsaData.wVersion = 0;
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: WSAStartup. %s", CFormatMessage(iErr).c_str() );
}
else
{ // Confirm that the WinSock DLL supports 2.2.
// Note that if the DLL supports versions greater than 2.2 in addition to 2.2, it will still return
// 2.2 in wVersion since that is the version we requested.
if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 2 )
{ /* Tell the user that we could not find a usable WinSock DLL. */
LOG( L"ERROR!: WSAStartup. Unexpected version returned" );
hr = E_UNEXPECTED;
}
}
// Connect to port 25
if ( S_OK == hr )
{
socket = WSASocket( AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, 0 );
if( INVALID_SOCKET == socket )
{
iErr = WSAGetLastError();
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: WSASocket. %s", CFormatMessage(iErr).c_str() );
}
else
{
addrinfo stHints;
addrinfo *pstAIHead = NULL, *pstAINext;
sockaddr_in stSAInServer;
ZeroMemory( &stSAInServer, sizeof( stSAInServer ));
ZeroMemory( &stHints, sizeof( stHints ));
stHints.ai_family = PF_INET;
stHints.ai_flags = AI_CANONNAME;
iErr = getaddrinfo( sASCComputerName.c_str(), "SMTP", &stHints, &pstAIHead );
if( 0 != iErr )
{
iErr = WSAGetLastError();
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: getaddrinfo. %s", CFormatMessage(iErr).c_str() );
}
else
{
memcpy( &stSAInServer, pstAIHead->ai_addr, sizeof( stSAInServer ));
pstAINext = pstAIHead;
while ( NULL != pstAINext )
{
pstAINext = pstAIHead->ai_next;
freeaddrinfo( pstAIHead );
pstAIHead = pstAINext;
}
iErr = connect( socket, reinterpret_cast<sockaddr*>( &stSAInServer ), sizeof( stSAInServer ));
if( 0 != iErr )
{
iErr = WSAGetLastError();
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: connect. %s", CFormatMessage(iErr).c_str() );
}
hr = RecvResp( socket, SMTP_MSG220 );
}
}
}
// Get the domain name we are sending to
if ( S_OK == hr )
{ // Get domain controller name
PDOMAIN_CONTROLLER_INFO pstDomainControllerInfo = NULL;
LPCWSTR psComputerName = ( sComputerName == LOCALHOST ) ? NULL:sComputerName.c_str();
DWORD dwRC = DsGetDcName( psComputerName, NULL, NULL, NULL, DS_RETURN_DNS_NAME, &pstDomainControllerInfo );
if ( NO_ERROR != dwRC )
{
hr = HRESULT_FROM_WIN32( dwRC );
LOG( L"ERROR!: DsGetDcName. %s", CFormatMessage(dwRC).c_str() );
}
ASSERT(hr == S_OK && "DsGetDcName failed");
// cache the domain
if ( pstDomainControllerInfo )
{
sEmailDomain = pstDomainControllerInfo->DomainName;
// free memory
NetApiBufferFree( pstDomainControllerInfo );
}
}
// SMTP:HELO
if ( S_OK == hr )
{
iSize = _snprintf( psSendBuffer, 2048, "HELO %s\r\n", W2A( sEmailDomain.c_str() ));
if ( 0 < iSize )
hr = SendRecv( socket, psSendBuffer, iSize, SMTP_MSG250 );
else
hr = E_UNEXPECTED;
}
// Process domains and mailboxes
if ( S_OK == hr )
hr = CoCreateInstance( __uuidof( P3Config ), NULL, CLSCTX_ALL, __uuidof( IP3Config ),reinterpret_cast<LPVOID *>( &spIConfig ));
if ( REGDB_E_CLASSNOTREG == hr )
{
hr = RegisterDependencies();
if ( S_OK == hr )
hr = CoCreateInstance( __uuidof( P3Config ), NULL, CLSCTX_ALL, __uuidof( IP3Config ),reinterpret_cast<LPVOID *>( &spIConfig ));
bUnregisterDependencies = true;
}
if ( S_OK == hr )
hr = spIConfig->get_Domains( &spIDomains );
if ( S_OK == hr )
{
CComPtr<IP3Users> spIUsers;
CComPtr<IP3User> spIUser;
if ( 0 == _bstrDomainName.length() )
{ // For each domain
hr = spIDomains->get__NewEnum( &spIEnumVARIANT );
if ( S_OK == hr )
{
hr = spIEnumVARIANT->Next( 1, &v, NULL );
}
}
else
{ // Just for this domain
_variant_t _v( _bstrDomainName );
hr = spIDomains->get_Item( _v, &pIDomain );
}
while ( S_OK == hr )
{
if ( NULL != spIEnumVARIANT.p )
{
if ( VT_DISPATCH != V_VT( &v ))
hr = E_UNEXPECTED;
if ( S_OK == hr )
hr = V_DISPATCH( &v )->QueryInterface( __uuidof( IP3Domain ), reinterpret_cast<void**>( &pIDomain ));
VariantClear( &v );
}
if ( S_OK == hr )
hr = pIDomain->get_Name( &bstrName );
if ( S_OK == hr )
{
LOG( L"Domain: %s ", bstrName );
SysFreeString( bstrName );
}
// Enumerate the users
if ( S_OK == hr )
{
hr = pIDomain->get_Users( &spIUsers );
if ( S_OK == hr )
{ // List users
CComPtr<IEnumVARIANT> spIEnumVARIANT2;
hr = spIUsers->get__NewEnum( &spIEnumVARIANT2 );
if ( S_OK == hr )
hr = spIEnumVARIANT2->Next( 1, &v, NULL );
while ( S_OK == hr )
{
if ( VT_DISPATCH != V_VT( &v ))
hr = E_UNEXPECTED;
if ( S_OK == hr )
hr = V_DISPATCH( &v )->QueryInterface( __uuidof( IP3User ), reinterpret_cast<void**>( &spIUser ));
VariantClear( &v );
if ( S_OK == hr )
hr = spIUser->get_Name( &bstrName );
if ( S_OK == hr )
{
hr = spIUser->get_EmailName( &bstrEmailName );
if ( S_OK == hr )
{
hr = mailboxX.OpenMailBox( bstrEmailName ) ? S_OK : E_UNEXPECTED;
if ( S_OK == hr )
{
bLocked = mailboxX.LockMailBox() ? true : false;
hr = mailboxX.EnumerateMailBox() ? S_OK : E_UNEXPECTED;
}
if ( S_OK == hr )
dwCount = mailboxX.GetMailCount();
for ( dwIndex = 0; (S_OK == hr) && (dwIndex < dwCount); dwIndex++ )
{
if ( !mailboxX.GetMailFileName( dwIndex, sFilename, dwFileNameSize ))
hr = E_UNEXPECTED;
if ( S_OK == hr )
{
hr = GetMailFROM( sFilename, sASCFROM );
if ( S_OK != hr )
sASCFROM.erase();
}
if ( S_OK == hr )
{ // SMTP:MAIL
iSize = _snprintf( psSendBuffer, SMTP_BUFFERSIZE, "MAIL FROM:<%s>\r\n", sASCFROM.c_str() );
if ( 0 < iSize )
hr = SendRecv( socket, psSendBuffer, iSize, SMTP_MSG250 );
else
hr = E_UNEXPECTED;
}
if ( S_OK == hr )
{ // SMTP:RCPT
iSize = _snprintf( psSendBuffer, SMTP_BUFFERSIZE, "RCPT TO:%s@%s\r\n", W2A( bstrName ), W2A( sEmailDomain.c_str() ));
if ( 0 < iSize )
hr = SendRecv( socket, psSendBuffer, iSize, SMTP_MSG250 );
else
hr = E_UNEXPECTED;
}
if ( S_OK == hr )
{ // SMTP:DATA
hr = SendRecv( socket, SMTP_DATA, iSize, SMTP_MSG354 );
}
if ( S_OK == hr )
{
iErr = mailboxX.TransmitMail( socket, dwIndex );
if ( ERROR_SUCCESS != iErr )
{
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: TransmitMail failed. %s", CFormatMessage(iErr).c_str() );
// Data probably wasn't terminated since TransmitMail failed.
SendRecv( socket, SMTP_DATA_END, iSize, SMTP_MSG250 );
}
else
{ // There should be a 250 OK reply waiting.
hr = RecvResp( socket, SMTP_MSG250 );
}
}
if ( S_OK != hr && bIgnoreErrors )
{
SendRecv( socket, SMTP_RSET, strlen( SMTP_RSET ), SMTP_MSG250 );
hr = S_OK;
bErrors = true;
}
}
if ( bLocked )
{
mailboxX.UnlockMailBox();
bLocked = false;
}
SysFreeString( bstrEmailName );
mailboxX.QuitAndClose();
}
SysFreeString( bstrName );
}
if ( S_OK == hr )
hr = spIEnumVARIANT2->Next( 1, &v, NULL );
}
if ( S_FALSE == hr )
hr = S_OK; // Reached end of user enumeration
}
}
if ( S_OK == hr )
{
if ( NULL != spIEnumVARIANT.p )
{
hr = spIEnumVARIANT->Next( 1, &v, NULL );
if ( S_OK == hr )
pIDomain->Release();
}
else
hr = S_FALSE; // Were done.
}
}
if ( S_FALSE == hr )
hr = S_OK; // Reached end of domain enumeration
}
if ( NULL != pIDomain )
pIDomain->Release();
// Cleanup
if( INVALID_SOCKET != socket )
{
// SMTP:QUIT
HRESULT hr2 = SendRecv( socket, SMTP_QUIT, strlen( SMTP_QUIT ), SMTP_MSG221 );
if ( S_OK == hr ) hr = hr2;
iErr = closesocket( socket );
if ( 0 != iErr ) LOG( L"ERROR!: closesocket. %s", CFormatMessage(iErr).c_str() );
assert( 0 == iErr );
}
if ( 0 != wsaData.wVersion )
{
iErr = WSACleanup( );
if ( 0 != iErr ) LOG( L"ERROR!: WSACleanup. %s", CFormatMessage(iErr).c_str() );
assert( 0 == iErr );
}
if ( S_OK == hr )
{
PrintMessage( IDS_SUCCESS_SENDMAIL );
if ( bErrors )
PrintMessage( IDS_WARNING_ERRORSIGNORED );
}
else
PrintMessage( IDS_ERROR_SENDMAIL_FAILED );
if ( bUnregisterDependencies )
UnRegisterDependencies();
return hr;
}
#define FROMFIELD "\r\nfrom:"
#define FROMFIELDEND "\r\n"
#define EOH "\r\n\r\n"
HRESULT CP2EWorker::GetMailFROM( LPCWSTR sFileName, ASTRING &sFrom )
{
HRESULT hr = S_OK;
bool isThereMoreToRead = true, isFromComplete = false;
char sBuffer[LOCAL_FILE_BUFFER_SIZE+1];
DWORD dwBytesToRead, dwBytesRead;
LPSTR ps, psEOH, psFrom;
DWORD dwStart;
ASTRING sFromFULL;
HANDLE hFile = CreateFile( sFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL );
if ( INVALID_HANDLE_VALUE == hFile )
hr = HRESULT_FROM_WIN32( GetLastError() );
// Read till we find the "\r\nFrom:"
dwStart = strlen( FROMFIELD );
dwBytesToRead = LOCAL_FILE_BUFFER_SIZE - dwStart;
if( ReadFile( hFile, sBuffer, LOCAL_FILE_BUFFER_SIZE, &dwBytesRead, NULL) )
sBuffer[dwBytesRead] = 0;
else
hr = HRESULT_FROM_WIN32( GetLastError() );
while( isThereMoreToRead && S_OK == hr )
{
if( dwBytesRead < dwBytesToRead )
isThereMoreToRead = false;
_strlwr( sBuffer );
ps = sBuffer;
psEOH = strstr( ps, EOH );
if ( NULL != psEOH )
{
*psEOH = 0;
isThereMoreToRead = false;
}
psFrom = strstr( ps, FROMFIELD );
if ( NULL != psFrom )
{ // Found the From Field, do we have it all?
psFrom += strlen( FROMFIELD );
ps = strstr( psFrom, FROMFIELDEND );
if ( NULL != ps )
{
*ps = 0;
isFromComplete = true;
}
sFromFULL = psFrom;
isThereMoreToRead = false;
}
if ( isThereMoreToRead )
{ // Copy end of buffer to beginning (to handle wrap around) then continue
memcpy( sBuffer, &(sBuffer[dwBytesToRead]), dwStart );
if( ReadFile( hFile, &(sBuffer[dwStart]), dwBytesToRead, &dwBytesRead, NULL) )
sBuffer[dwBytesRead + dwStart] = 0;
else
hr = HRESULT_FROM_WIN32( GetLastError() );
}
}
if ( !isFromComplete && NULL != psFrom )
{ // Should be able to get rest of FROM with next read
// Copy end of buffer to beginning (to handle wrap around) then continue
memcpy( sBuffer, &(sBuffer[dwBytesToRead]), dwStart );
if( ReadFile( hFile, &(sBuffer[dwStart]), dwBytesToRead, &dwBytesRead, NULL) )
{
sBuffer[dwBytesRead + dwStart] = 0;
ps = strstr( sBuffer, FROMFIELDEND );
if ( NULL != ps )
{
*ps = 0;
if ( ps > &(sBuffer[dwStart]) )
sFromFULL += sBuffer[dwStart];
else
{ // cleanup sFrom
ASTRING::size_type iPos = sFromFULL.find( '\r' );
if ( ASTRING::npos != iPos )
sFromFULL.resize( iPos );
}
}
}
else
hr = HRESULT_FROM_WIN32( GetLastError() );
}
if ( !sFromFULL.empty() && S_OK == hr )
{ // Look for the open bracket
ASTRING::size_type iStart = sFromFULL.find( '<' );
ASTRING::size_type iEnd = sFromFULL.find( '>' );
if ( WSTRING::npos != iStart )
{
if ( ASTRING::npos != iEnd )
sFrom = sFromFULL.substr( iStart + 1, iEnd - iStart - 1 );
}
else
sFrom = sFromFULL;
}
if ( INVALID_HANDLE_VALUE != hFile )
CloseHandle( hFile );
return hr;
}
void CP2EWorker::PrintError( int iRC )
{
if ( m_bSuppressPrintError || (E_FAIL == iRC) )
{
m_bSuppressPrintError = false;
return;
}
LPVOID lpMsgBuf;
if ( FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, iRC, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), reinterpret_cast<LPWSTR>( &lpMsgBuf ), 0, NULL ))
{
wprintf( reinterpret_cast<LPWSTR>( lpMsgBuf ) );
LocalFree( lpMsgBuf );
}
else
{
WCHAR sBuffer[32];
PrintMessage( IDS_ERROR_UNKNOWN );
wsprintf( sBuffer, L"%x", iRC );
PrintMessage( sBuffer );
}
}
void CP2EWorker::PrintMessage( LPWSTR psMessage, bool bCRLF /*= true*/ )
{
wprintf( psMessage );
if ( bCRLF )
wprintf( L"\r\n" );
}
void CP2EWorker::PrintMessage( int iID, bool bCRLF /*= true*/ )
{
WCHAR sBuffer[512];
if ( LoadString( NULL, iID, sBuffer, sizeof( sBuffer )/sizeof(WCHAR) ))
PrintMessage( sBuffer, bCRLF );
else
assert( false );
}
void CP2EWorker::PrintUsage()
{
WCHAR sBuffer[512];
for ( int i = IDS_POP2EXCH_USAGE1; i < IDS_POP2EXCH_USAGEEND; i++ )
{
if ( LoadString( NULL, i, sBuffer, sizeof( sBuffer )/sizeof(WCHAR) ))
{
if ( IDS_POP2EXCH_USAGE2 == i )
{
WCHAR sBuffer2[512];
if ( 0 > _snwprintf( sBuffer2, sizeof( sBuffer2 )/sizeof(WCHAR), sBuffer, POP3_SERVER_NAME_L ))
sBuffer2[(sizeof( sBuffer2 )/sizeof(WCHAR))-1] = 0;
wcscpy( sBuffer, sBuffer2 );
}
wprintf( sBuffer );
}
}
}
LPWSTR CP2EWorker::FormatLogString( LPWSTR psLogString )
{
if ( NULL == psLogString )
return psLogString;
LPWSTR psCRLF = wcsstr( psLogString, L"\r\n" );
while ( NULL != psCRLF )
{
*psCRLF = L' ';
*(psCRLF+1) = L' ';
psCRLF = wcsstr( psLogString, L"\r\n" );
}
return psLogString;
}
HRESULT CP2EWorker::RecvResp( SOCKET socket, LPCSTR psExpectedResp )
{
if ( NULL == psExpectedResp )
return E_INVALIDARG;
if ( 3 > strlen( psExpectedResp ))
return E_INVALIDARG;
CHAR sRecvBuffer[SMTP_BUFFERSIZE + 1];
USES_CONVERSION;
HRESULT hr = S_OK;
int iErr;
iErr = recv( socket, sRecvBuffer, SMTP_BUFFERSIZE, 0 );
if ( SOCKET_ERROR != iErr )
{
sRecvBuffer[iErr] = 0;
if ( (3 > iErr) || (0 != strncmp( sRecvBuffer, psExpectedResp, 3 )))
hr = E_UNEXPECTED;
LPWSTR ps = FormatLogString( A2W( sRecvBuffer ));
LOG( L"recv [%s].", ps );
}
else
{
iErr = WSAGetLastError();
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: recv failed. %s", CFormatMessage(iErr).c_str() );
}
return hr;
}
HRESULT CP2EWorker::RegisterDependencies()
{
HRESULT hr = S_OK;
HMODULE hDLL;
HRESULT (STDAPICALLTYPE * lpDllEntryPoint)(void);
tstring strPath = _T("") ;
strPath = GetModulePath () ;
if ( strPath != _T("") )
strPath = strPath + _T("P3Admin.dll") ;
else
hr = E_FAIL ;
if ( S_OK == hr )
{
hDLL = LoadLibrary( strPath.c_str() );
if ( hDLL )
{
(FARPROC&)lpDllEntryPoint = GetProcAddress(hDLL,"DllRegisterServer");
if (lpDllEntryPoint != NULL)
hr = (*lpDllEntryPoint)();
else
hr = HRESULT_FROM_WIN32(GetLastError());
FreeLibrary( hDLL );
}
else
{
PrintMessage( IDS_ERROR_MUSTBERUNFROMCURDIR );
hr = HRESULT_FROM_WIN32(GetLastError());
m_bSuppressPrintError = true;
}
}
if ( S_OK == hr )
{
CComPtr<IP3Config> spIConfig;
WCHAR sBuffer[MAX_PATH+1];
hr = CoCreateInstance( __uuidof( P3Config ), NULL, CLSCTX_ALL, __uuidof( IP3Config ),reinterpret_cast<LPVOID *>( &spIConfig ));
if ( S_OK == hr )
{
DWORD dwRC = GetCurrentDirectoryW( sizeof(sBuffer)/sizeof(WCHAR), sBuffer );
if (( 0 != dwRC ) && ( (sizeof(sBuffer)/sizeof(WCHAR)) > dwRC ))
{
HKEY hKey;
long lRC = RegCreateKey( HKEY_LOCAL_MACHINE, POP3SERVER_SOFTWARE_SUBKEY, &hKey );
if ( ERROR_SUCCESS == lRC )
{
lRC = RegSetDWORD( POP3SERVER_SOFTWARE_SUBKEY, VALUENAME_VERSION, 0 );
if ( ERROR_SUCCESS != lRC )
hr = HRESULT_FROM_WIN32( lRC );
RegCloseKey( hKey );
}
else
hr = HRESULT_FROM_WIN32( lRC );
if ( S_OK == hr )
{
_bstr_t _bstrMailRoot = sBuffer;
hr = spIConfig->put_MailRoot( _bstrMailRoot );
}
}
else
hr = E_FAIL;
}
}
if ( S_OK != hr )
{
UnRegisterDependencies () ;
}
return hr;
}
HRESULT CP2EWorker::SendRecv( SOCKET socket, LPCSTR psSendBuffer, const int iSize, LPCSTR psExpectedResp )
{
if ( NULL == psSendBuffer )
return E_INVALIDARG;
USES_CONVERSION;
HRESULT hr = S_OK;
int iErr;
LPWSTR ps = FormatLogString( A2W( psSendBuffer ));
iErr = send( socket, psSendBuffer, iSize, 0 );
if ( SOCKET_ERROR != iErr )
{
LOG( L"send [%s].", ps );
}
else
{
iErr = WSAGetLastError();
hr = HRESULT_FROM_WIN32( iErr );
LOG( L"ERROR!: send [%s] failed. %s", ps, CFormatMessage(iErr).c_str() );
}
if ( S_OK == hr )
{
hr = RecvResp( socket, psExpectedResp );
}
return hr;
}
HRESULT CP2EWorker::UnRegisterDependencies()
{
HRESULT hr = S_OK;
HMODULE hDLL;
HRESULT (STDAPICALLTYPE * lpDllEntryPoint)(void);
tstring strPath = _T("") ;
strPath = GetModulePath () ;
if ( strPath != _T("") )
strPath = strPath + _T("P3Admin.dll") ;
else
hr = E_FAIL ;
if ( S_OK == hr )
{
hDLL = LoadLibrary( strPath.c_str() );
if ( hDLL )
{
(FARPROC&)lpDllEntryPoint = GetProcAddress(hDLL,"DllUnregisterServer");
if (lpDllEntryPoint != NULL)
hr = (*lpDllEntryPoint)();
else
hr = HRESULT_FROM_WIN32(GetLastError());
FreeLibrary( hDLL );
}
else
hr = HRESULT_FROM_WIN32(GetLastError());
}
if ( S_OK == hr )
{
long lRC = RegDeleteKey( HKEY_LOCAL_MACHINE, POP3SERVER_SOFTWARE_SUBKEY );
if ( ERROR_SUCCESS != lRC )
hr = HRESULT_FROM_WIN32( lRC );
}
return hr;
}
tstring CP2EWorker::GetModulePath ()
{
tstring strDir = _T("");
TCHAR szDrive[_MAX_DRIVE], szDir[_MAX_DIR], szFName[_MAX_FNAME], szExt[_MAX_EXT] ;
DWORD dwSize = MAX_PATH;
TCHAR *szPath = new TCHAR[dwSize+1];
if (!szPath)
return strDir;
// Allocate more space if necessary
while( dwSize == GetModuleFileName( NULL, szPath, dwSize ) )
{
delete [] szPath;
dwSize *= 2;
szPath = new TCHAR[dwSize+1];
if (!szPath)
return strDir;
}
_tsplitpath ( szPath, szDrive, szDir, szFName, szExt ) ;
_tmakepath ( szPath, szDrive, szDir, _T(""), _T("") ) ; // Make the path without filename and extension.
if (_tcslen( szPath ))
strDir = szPath;
if ( szPath[_tcslen(szPath)-1] != '\\' )
strDir = strDir + _T("\\") ;
delete [] szPath;
return strDir;
}
Reference in New Issue Copy Permalink