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

#ifndef _TELNTSRV_H_
#define _TELNTSRV_H_

#include <tchar.h>
#include <winsock2.h>
#ifdef WHISTLER_BUILD
#include <ws2tcpip.h>
#else
#include "ws2tcpip.h"
#endif
#include <IpTypes.h>

#include <TlntDynamicArray.h>
#include <TlntSvr.h>
#include <EnumData.h>
#include <EncliSvr.h>
#include <ClientInfo.h>
#include <Queue.h>


#define WAIT_TIME      5000

DWORD WINAPI  DoIPCWithClients( LPVOID );

typedef struct _FAMILY {
    int       iFamily;
    socklen_t iSocklen;

    SOCKET    sListenSocket;       // the socket that listens for ever
    WSAEVENT  SocketAcceptEvent;
} FAMILY;

#define NUM_FAMILIES 2


class CTelnetService 
{
public:
    virtual ~CTelnetService();

    static  CTelnetService* Instance();

    CQueue *CQList;
    bool    Shutdown( void );
    bool    Pause( void );
    bool    Resume( void );
    void    SystemShutdown( void );
    DWORD   NeedAudit( void );
    
    friend  class CEnumTelnetClientsSvr;
    friend  DWORD WINAPI  DoIPCWithClients( LPVOID );
    bool    ListenerThread( );
private:
    CTelnetService();

    enum{ SOCKET_CLOSE_EVENT = WAIT_OBJECT_0, FD_ACCEPT_EVENT_0, 
        REG_CHANGE_EVENT, FD_ACCEPT_EVENT_1 };
    enum{ SERVER_PAUSED, SERVER_RUNNING };

    bool    Init( void );
    void    InitializeOverlappedStruct( LPOVERLAPPED );
    void    GetPathOfTheExecutable( PCHAR * );
    void    GetPathOfTheExecutable( LPTSTR * );
    PCHAR   GetDefaultLoginScriptFullPath( void );
    bool    GetRegistryValues( void );
    bool    GetLicenseForWorkStation( SOCKET );
    bool    CheckLicense( bool *, CClientInfo * );
    bool    IssueLicense( bool, CClientInfo *);
    bool    StartThreads( void );
    bool    CreateClient( SOCKET , DWORD *, HANDLE *, CClientInfo **);
    bool    SendSocketToClient( HANDLE, SOCKET, DWORD );
    bool    HandleInProcMessages( DWORD );
    bool    IPCDataDriver( CClientInfo* );
    bool    HandleFarEastSpecificRegKeys( void );
    bool    CreateSessionProcess( HANDLE, HANDLE, DWORD*, HANDLE *, HWINSTA *, HDESK * );
    

    bool    ExitTheSession( CClientInfo * );
    bool    AskSessionToShutdown( HANDLE, UCHAR );
    bool    SendSessionIdToClient( HANDLE, LONG );
    bool    InformTheClient( SOCKET, LPSTR );
    bool    StopServicingClient( CClientInfo *, BOOL );
    void    HandleSessionDetailsMessage( CClientInfo * );

    bool    WatchRegistryKeys( void );
    bool    HandleChangeInRegKeys( void );
    bool    SetNewRegKeyValues( DWORD, DWORD, DWORD, LPWSTR, LPWSTR, DWORD );
    bool    RegisterForNotification( void );
    
    bool    GetInAddr( INT, SOCKADDR_STORAGE *, socklen_t * );
    bool    CreateSocket( INT );
    bool    InitTCPIP( void );        

    bool    CreateNewIoCompletionPort( DWORD );
    bool    AssociateDeviceWithCompletionPort ( HANDLE, HANDLE, DWORD_PTR );
    bool    IssueReadFromPipe( CClientInfo* );
    bool    OnCompletionPacket( CClientInfo*, LPOVERLAPPED );
    bool    IssueReadAgain( CClientInfo* );
    
    DWORD   m_dwTelnetPort;
    DWORD   m_dwMaxConnections;
    bool    m_fLogToFile;
    LPWSTR  m_pszIpAddrToListenOn;
    

    static  CTelnetService* s_instance;
    DWORD   m_dwNumOfActiveConnections;
    LONG    m_lServerState;
    WCHAR   m_szDomainName[ MAX_DOMAIN_NAME_LEN + 4 ]; //This may be of form \\solar-dc-01
    bool    m_bIsWorkStation;
    SOCKADDR_STORAGE *m_pssWorkstationList;
    DWORD   m_dwNoOfWorkstations;

    HANDLE  m_hRegChangeEvent;
    HKEY    m_hReadConfigKey;
    HANDLE  m_hSyncAllClientObjAccess;
    FAMILY  m_sFamily[ NUM_FAMILIES ];
    HANDLE  m_hCompletionPort;
    HWINSTA m_hOldWinSta;
    HANDLE  m_hIPCThread;
    
    WSAEVENT m_SocketAcceptEvent;
    HANDLE   m_hSocketCloseEvent;
    
    OVERLAPPED m_oReadFromPipe;
    OVERLAPPED m_oWriteToPipe;
    OVERLAPPED m_oPostedMessage;
};

void LogEvent ( WORD wType, DWORD dwEventID, LPCTSTR pFormat, ... );
void WriteAuditedMsgsToFile( LPSTR );

void   CloseLogFile( LPWSTR *, HANDLE * );
bool   InitializeLogFile( LPWSTR, HANDLE * );

#endif