// IoHandler.h : This file contains the
// Created:  Feb '98
// Author : a-rakeba
// History:
// Copyright (C) 1998 Microsoft Corporation
// All rights reserved.
// Microsoft Confidential 

#if !defined( _IOHANDLER_H_ )
#define _IOHANDLER_H_

#include <cmnhdr.h>

#include <windows.h>
#include <winsock2.h>
#define SECURITY_WIN32
#include <sspi.h>

#include <TlntUtils.h>
#include <Ipc.h>

#define AVAILABE_BUFFER_SIZE( p, c ) ( sizeof( p ) - ( c - p ) )
#define FOOTER   L"\r\n"
#define NEED_HEADER 1
#define NO_HEADER   0

#define SIZEOF_ARROWKEY_SEQ     3

#define PVA_SUCCESS				0
#define PVA_NODATA				-1
#define PVA_BADFORMAT			-2
#define PVA_GUEST				-3
#define PVA_NOMEMORY			-4
#define PVA_INVALIDACCOUNT		-5
#define PVA_OTHERERROR			-6

class CSession;

class CIoHandler 
{
    public:
    CIoHandler();
    virtual ~CIoHandler();
    
    typedef DWORD  IO_OPERATIONS;
    friend  struct CRFCProtocol;
    friend  class  CScraper;

    private:
    int		ParseAndValidateAccount(void);
    int     AuthenticateUser( void );
    int     CheckLicense( void );
    bool    WriteToServer ( UCHAR, DWORD, LPVOID );
    bool    StartNTLMAuth( void );
    bool    DoNTLMAuth( PUCHAR, DWORD, PUCHAR* );
    bool    IsTimedOut ( );
    bool    GetAndSetSocket ( );
    bool    HandlePipeData ( );
    bool    SendDetailsAndAskForLicense ();
    bool    HandleOperatorMessage();
    bool    GetHeaderMessage( LPWSTR* );
    
    bool    RemoveArrowKeysFromBuffer( PDWORD ,PDWORD);

    bool          ProcessUserDataReadFromSocket( DWORD );
    IO_OPERATIONS ProcessDataFromSocket( DWORD );
    IO_OPERATIONS ProcessCommandLine( PDWORD, PDWORD, IO_OPERATIONS );
    IO_OPERATIONS ProcessAuthenticationLine( PDWORD, PDWORD, IO_OPERATIONS );
    IO_OPERATIONS OnDataFromSocket ( );

    protected:

    typedef enum {
        STATE_INIT,
        STATE_NTLMAUTH,
        STATE_SESSION,
        STATE_BANNER_FOR_AUTH,
        STATE_WAIT_FOR_ENV_OPTION,
        STATE_LOGIN_PROMPT, 
        STATE_AUTH_NAME,
        STATE_AUTH_PASSWORD,
        STATE_CHECK_LICENSE,
        STATE_LICENSE_AVAILABILITY_KNOWN,
        STATE_VTERM_INIT_NEGO,
        STATE_VTERM,
        STATE_TERMINATE } CONTROL_STATE;

    CONTROL_STATE m_SocketControlState;
    enum {  WRITE_TO_SOCKET = 0x00000001, READ_FROM_SOCKET = 0x0000002, 
            LOGON_COMMAND = 0x00000010, LOGON_DATA_UNFINISHED = 0x00000100,
            IO_FAIL = 0x00001000 };

    typedef enum {
        NOT_MEMBER_OF_TELNETCLIENTS_GROUP = 3,
        DENY_LICENSE = 4, 
        ISSUE_LICENSE = 5, 
        WAIT_FOR_SERVER_REPLY = 6,
#if BETA
        LICENSE_EXPIRED
#endif
    } LICENSE_LIMIT;

    CtxtHandle m_hContext;

    OVERLAPPED m_oWriteToSocket;
    UCHAR      m_WriteToSocketBuff[ MAX_WRITE_SOCKET_BUFFER ];
    DWORD      m_dwWriteToSocketIoLength;
    WSAPROTOCOL_INFO m_protocolInfo;
    bool       m_bOnlyOnce;
    bool       m_bWaitForEnvOptionOver;

    OVERLAPPED m_oWriteToPipe;

    OVERLAPPED m_oReadFromSocket;
    UCHAR      m_ReadFromSocketBuffer[ MAX_READ_SOCKET_BUFFER ];
    PUCHAR     m_pReadFromSocketBufferCursor;
    DWORD      m_dwReadFromSocketIoLength;

    OVERLAPPED m_oReadFromPipe;
    UCHAR      m_ReadFromPipeBuffer[ IPC_HEADER_SIZE ];    
    UCHAR      *m_pucReadBuffer;
    DWORD      m_dwReadFromPipeIoLength;
    DWORD      m_dwRequestedSize;
    bool       m_bIpcHeader;

    SOCKET     m_sSocket;
    HANDLE     m_hReadPipe;
    HANDLE     m_hWritePipe;
  
    bool       m_fFirstReadFromPipe;
    bool       m_fShutDownAfterIO;
    BOOL       m_fLogonUserResult;
    bool       m_bNTLMAuthenticated;
    bool       fDoNTLMAuthFirstTime;
    CredHandle m_hCredential;
    PSecPkgInfo  m_pspi;
    CSession   *m_pSession;
    int         m_iResult;

	bool		m_bInvalidAccount;
	
    bool      GetUserName();
    bool      WriteToSocket(PUCHAR, DWORD );
    bool      Init( CSession * );
    void      Shutdown( void );
    bool      IssueReadOnPipe( );
    bool      IssueFirstReadOnPipe ( );
    bool      WriteToClient ( );
    bool      IssueReadFromSocket ( );
    bool      OnReadFromPipeCompletion ( );
    bool      OnReadFromSocketCompletion ( );
    void      DisplayOnClientNow();
    void      SendMessageToClient( LPWSTR, bool );
    void      WriteMessageToClientDirectly( LPWSTR );
    void      SendTerminateString( char *);
    void      UpdateIdleTime( UCHAR );
};

#endif // _IOHANDLER_H_