|
|
/************************************************************************************************
Copyright (c) 2001 Microsoft Corporation
File Name: IOContext.hxxAbstract: Declare the POP3_CONTEXT ClassNotes: History: 08/01/2001 Created by Hao Yu (haoyu)
************************************************************************************************/
#ifndef __POP3_CONTEXT_H__
#define __POP3_CONTEXT_H__
#include <IOContext.h>
#include "Mailbox.h"
#include <NTAuth.h>
#define INIT_STATE 0
#define AUTH_STATE 1
#define TRANS_STATE 2
#define UPDATE_STATE 3
#define MD5_HASH_SIZE 32
#define MAX_INT_LEN 9
#define SZ_NTLM "NTLM"
#define RESP_INVALID_MAIL_NUMBER "-ERR No such message\r\n"
#define RESP_SERVER_NOT_AVAILABLE "-ERR Service is not available\r\n"
#define RESP_UNKNOWN_COMMAND "-ERR Unacceptable command\r\n"
#define RESP_OK "+OK\r\n"
#define RESP_SERVER_ERROR "-ERR server error\r\n"
#define RESP_SERVER_READY "+OK %S %S ready.\r\n"
#define RESP_SERVER_QUIT "+OK %S %S signing off.\r\n"
#define RESP_ACCOUNT_ERROR "-ERR Logon Failure\r\n"
#define RESP_END_OF_MULTILINE "\r\n.\r\n"
#define RESP_END_OF_LIST ".\r\n"
#define RESP_MSG_MARKED_DELETED "+OK Message marked as deleted\r\n"
#define RESP_AUTH_DONE "+OK User successfully logged on\r\n"
#define RESP_AUTH_METHODS "+OK\r\n" SZ_NTLM RESP_END_OF_MULTILINE
#define RESP_CMD_NOT_VALID "-ERR Command not valid\r\n"
#define RESP_CMD_NOT_SUPPORTED "-ERR Command is not valid in this state\r\n"
#define RESP_INVALID_LENGTH "-ERR Invalid message Length\r\n"
#define RESP_RESET "+OK Messages unmarked as deleted\r\n"
#define RESP_NO_USER_NAME "-ERR No username\r\n"
#define RESP_SPA_REQUIRED "-ERR SPA Required, use AUTH or APOP\r\n"
#define RESP_SERVER_GREETING L"Microsoft Windows POP3 Service Version 1.0"
#define ERR_NO_SUCH_MSG 0xf0000001
#define ERR_MSG_ALREADY_DELETED 0xf0000002
#define ERR_CAN_NOT_OPEN_FILE 0xf0000003
#define ERR_CAN_NOT_SET_FILE_CURSOR 0xf0000004
#define AUTH_OTHER 0
#define AUTH_AD 1
#define AUTH_LOCAL_SAM 2
#define MAX_FAILED_AUTH 3
#define COMMAND_SIZE 4
const int ciCommandSize[]={4,4,4,4,4,4,4,4,4,4,4,3,4};
const char cszCommands[][5]={"STAT", "LIST", "RETR", "DELE", "NOOP", "RSET", "QUIT", "USER", "PASS", "UIDL", "APOP", "TOP", "AUTH"};
typedef enum enumCommand{ CMD_STAT=0, CMD_LIST=1, CMD_RETR=2, CMD_DELE=3, CMD_NOOP=4, CMD_RSET=5, CMD_QUIT=6, CMD_USER=7, CMD_PASS=8, CMD_UIDL=9, CMD_APOP=10, CMD_TOP=11, CMD_AUTH=12, CMD_UNKNOWN}POP3_CMD;
class POP3_CONTEXT{ DWORD m_dwCurrentState; BOOL m_bFileTransmitPending; BOOL m_bCommandComplete; DWORD m_dwCommandSize; char m_szCommandBuffer[POP3_REQUEST_BUF_SIZE]; WCHAR m_wszUserName[POP3_MAX_ADDRESS_LENGTH]; char m_szDomainName[POP3_MAX_DOMAIN_LENGTH]; char m_szPassword[MAX_PATH]; WCHAR m_wszGreeting[MAX_PATH + 64]; int m_cPswdSize; CMailBox m_MailBox; IO_CONTEXT *m_pIoContext; CAuthServer m_AuthServer; DWORD m_dwAuthStatus; DWORD m_dwFailedAuthCount;public:
POP3_CONTEXT(); ~POP3_CONTEXT(); void TimeOut(IO_CONTEXT *pIoContext); void ProcessRequest(IO_CONTEXT *pIOContext, OVERLAPPED *pOverlapped, DWORD dwBytesRcvd); void WaitForCommand(); void SendResponse(char *szBuf); void SendResponse(DWORD dwResult, char *szBuf); void Reset(); BOOL Unauthenticated();private: POP3_CMD ParseCommand(); BOOL ProcessAuthStateCommands(POP3_CMD CurrentCmd, DWORD dwBytesRcvd); BOOL ProcessTransStateCommands(POP3_CMD CurrentCmd, DWORD dwBytesRcvd); void TerminateConnection(PIO_CONTEXT pIoContext); BOOL GetNextStringParameter(char *szInput, char *szOutput, DWORD dwOutputSize); BOOL GetNextNumParameter(char **pszInput, int *piOutout); BOOL IsEndOfCommand(char *szInput);
};
typedef POP3_CONTEXT * PPOP3_CONTEXT;
#endif //__POP3_CONTEXT_H__
|
