windows-server-2003/termsrv/setup/tstst/tstst.cpp

// Copyright (c) 1998 - 1999 Microsoft Corporation



#include "stdafx.h"
#include <iostream.h>

#include <string>
typedef std::basic_string<TCHAR> TString;

#include <fstream.h>
#include <strstrea.h>
#include <dsrole.h>
#include <dsgetdc.h>
#include <wtsapi32.h>
// #include <ostream>
//#include <cstring>
// #include <sstream>


class ConstCRegistry : public CRegistry
{
public:
    DWORD       OpenKey  (HKEY hKey, LPCTSTR lpSubKey, REGSAM access = KEY_ALL_ACCESS, LPCTSTR lpMachineName = NULL);
};

DWORD ConstCRegistry::OpenKey(HKEY hKey, LPCTSTR lpSubKey, REGSAM access /*= KEY_ALL_ACCESS*/, LPCTSTR lpMachineName /*= NULL*/)
{
    ASSERT(access == KEY_READ);
    return CRegistry::OpenKey(hKey, lpSubKey, KEY_READ, lpMachineName);
}

#define ConstCRegistry CRegistry
//
// global utilities and veraibles.
//


char szOutput[2048];
ostrstream szMoreInfo(szOutput, 4096);


TCHAR *ReturnBuffer()
{
    
    static TCHAR szReturnTchar[512];
    return szReturnTchar;
}



const OSVERSIONINFOEX *GetOSVersionInfo()
{
    static OSVERSIONINFOEX gOsVersion;
    static bGotOnce = FALSE;
    if (!bGotOnce)
    {
        ZeroMemory(&gOsVersion, sizeof(OSVERSIONINFOEX));
        gOsVersion.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
        GetVersionEx( (LPOSVERSIONINFO ) &gOsVersion);
        bGotOnce = TRUE;
    }

    return &gOsVersion;
}



// #include <strstream>
#include "winsock2.h"


// ostringstream sz

#ifndef UNREFERENCED_PARAMETER
#define UNREFERENCED_PARAMETER(P)          (P)
#endif

#define OLD_VER_SET_CONDITION(_m_,_t_,_c_)  _m_=(_m_|(_c_<<(1<<_t_)))


BOOL CheckifBinaryisSigned        (TCHAR *szFile);  // from tscert.cpp
BOOL EnumerateLicenseServers      ();               // from timebomb.cpp
TCHAR *IsBetaSystem                 ();
BOOL HasLicenceGracePeriodExpired ();               // from timebomb.cpp
BOOL ExtractAllTSEvents();


BOOL ValidateProductSuite (LPSTR SuiteName);
BOOL IsTerminalServicesEnabled ( VOID );
DWORD IsStringInMultiString(HKEY hkey, LPCTSTR szkey, LPCTSTR szvalue, LPCTSTR szCheckForString, BOOL *pbFound);
BOOL DoesHydraKeysExists ();
TCHAR *IsServer ();
BOOL IsKernelTSEnable ();
BOOL TSEnabled ();
BOOL DoesProductSuiteContainTS ();
BOOL IsTerminalServerRegistryOk ();
TCHAR *IsTerminalServiceStartBitSet ();
BOOL IsTermDDStartBitSet ();
BOOL GetTSOCLogFileName (char *strFileName, UINT uiSize);
BOOL DidTsOCgetCompleteInstallationMessage ();
BOOL FileExists (char *pszFullNameAndPath);
BOOL IsTSOClogPresent ();
BOOL DidOCMInstallTSEnable();
TCHAR *IsClusteringInstalled ();
BOOL IsRemoteAdminMode ();
BOOL CheckModeRegistry (BOOL bAppCompat);
BOOL IsTerminalServiceRunning ();
BOOL CheckVideoKeys ();
BOOL CheckModePermissions(DWORD *pdwSecurtyMode);
BOOL IsFile128Bit(LPTSTR szFile, BOOL *pb128Bit);
ULONG RDPDRINST_DetectInstall();    // defined in drdetect.cpp
BOOL IsAudioOk( VOID );


TCHAR *aszStack[] = {
//  _T("noexport\%SystemRoot%\\system32\\drivers\\rdpwdd.sys"),
    _T("%SystemRoot%\\system32\\drivers\\termdd.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdasync.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdipx.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdnetb.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdpipe.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdspx.sys"),
    _T("%SystemRoot%\\system32\\drivers\\tdtcp.sys"),
    _T("%SystemRoot%\\system32\\drivers\\rdpwd.sys"),
    _T("%SystemRoot%\\system32\\rdpdd.dll"),
    _T("%SystemRoot%\\system32\\rdpwsx.dll")
};

TCHAR Version[256];
TCHAR *GetTSVersion()
{
	CRegistry oRegTermsrv;
	DWORD cbVersion = 0;
	LPTSTR szVersion = NULL;

	if ((ERROR_SUCCESS == oRegTermsrv.OpenKey(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ)) &&
		(ERROR_SUCCESS == oRegTermsrv.ReadRegString(_T("ProductVersion"), &szVersion, &cbVersion)))
	{
		_tcscpy(Version, szVersion);
		return Version;
	}

	return _T("Error finding Version.");
}

BOOL IsIt50TS()
{
	return (0 == _tcsicmp(Version, _T("5.0")));
}

BOOL IsIt51TS()
{
	return (0 == _tcsicmp(Version, _T("5.1"))) || (0 == _tcsicmp(Version, _T("5.2")));
}


TCHAR *IsLicenceGracePeriodOk ()
{
    return  (HasLicenceGracePeriodExpired () ? _T("Yep, Its expired") : _T("Its Not expired"));
}


BOOL Check_termdd()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\termdd.sys"));
}
BOOL Check_tdasync()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdasync.sys"));
}
BOOL Check_tdipx()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdipx.sys"));
}
BOOL Check_tdnetb()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdnetb.sys"));
}
BOOL Check_tdpipe()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdpipe.sys"));
}
BOOL Check_tdspx()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdspx.sys"));
}
BOOL Check_tdtcp()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\tdtcp.sys"));
}
BOOL Check_rdpwd()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\drivers\\rdpwd.sys"));
}
BOOL Check_rdpdd()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\rdpdd.dll"));
}
BOOL Check_rdpwsx()
{
    return CheckifBinaryisSigned(_T("%SystemRoot%\\system32\\rdpwsx.dll"));
}


BOOL IsRdpDrInstalledProperly ()
{
    bool bPersonal = (GetOSVersionInfo()->wSuiteMask & VER_SUITE_PERSONAL) != 0;
    bool bRdpdrInstalled = RDPDRINST_DetectInstall() != 0;

    return bPersonal != bRdpdrInstalled;
}


TCHAR *GetModePermissions()
{
    CRegistry reg;
    DWORD dwSecurityMode;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ))
    {
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("TSUserEnabled"), &dwSecurityMode))
        {
            if (dwSecurityMode == 0)
            {
                return _T("Its W2k Compatible");
            }
            else if (dwSecurityMode == 1)
            {
                return _T("Its TS4 Compatible");

            }
            else
            {
                szMoreInfo << "SYSTEM\\CurrentControlSet\\Control\\Terminal Server/TSUserEnabled has wrong value" << dwSecurityMode << endl;
                return NULL;

            }
        }
    }

    return NULL;

}

BOOL CheckModePermissions (DWORD *pdwSecurtyMode)
{
//    PERM_WIN2K = 0,
//    PERM_TS4 = 1

    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ))
    {
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("TSUserEnabled"), pdwSecurtyMode))
        {
            return (*pdwSecurtyMode== 0) || (*pdwSecurtyMode== 1);
        }
    }

    return FALSE;
}

TCHAR *GetCypherStrenthOnRdpwd ()
{
    BOOL bFile128bit;
    if ( IsFile128Bit(_T("%SystemRoot%\\system32\\drivers\\rdpwd.sys"), &bFile128bit) )
    {
        return bFile128bit ? _T("128 Bit") : _T("56 Bit");
    }
    else
    {
        return NULL;
    }

}


BOOL IsFile128Bit(LPTSTR szFile, BOOL *pb128Bit)
{
    USES_CONVERSION;
    DWORD dwHandle;

    TCHAR szFullFile[MAX_PATH +1];

    BOOL bSuccess = FALSE;

    if (ExpandEnvironmentStrings(szFile, szFullFile, MAX_PATH))
    {
        if (FileExists(T2A(szFullFile)))
        {
            DWORD dwSize = GetFileVersionInfoSize(szFullFile, &dwHandle);
            if (dwSize > 0)
            {
                BYTE *pbData = new BYTE[dwSize];
                if (pbData)
                {
                    if (GetFileVersionInfo(szFullFile, 0, dwSize, pbData))
                    {
                        TCHAR *szFileDescription;
                        UINT uiLen = 0;
                        if (VerQueryValue(pbData, _T("\\StringFileInfo\\040904B0\\FileDescription"), (LPVOID *)&szFileDescription, &uiLen))
                        {
                            if (_tcsstr(szFileDescription, _T("Not for Export")))
                            {
                                *pb128Bit = TRUE;
                                bSuccess = TRUE;

                            }
                            else if (_tcsstr(szFileDescription, _T("Export Version")))
                            {
                                *pb128Bit = FALSE;
                                bSuccess = TRUE;
                            }
                        }
                    }

                    delete [] pbData;

                }

            }
        }
    }

    return bSuccess;
}



BOOL ValidateProductSuite (LPSTR SuiteName)
{
    BOOL rVal = FALSE;
    LONG Rslt;
    HKEY hKey = NULL;
    DWORD Type = 0;
    DWORD Size = 0;
    LPSTR ProductSuite = NULL;
    LPSTR p;

    Rslt = RegOpenKeyA(
        HKEY_LOCAL_MACHINE,
        "System\\CurrentControlSet\\Control\\ProductOptions",
        &hKey
        );

    if (Rslt != ERROR_SUCCESS)
        goto exit;

    Rslt = RegQueryValueExA( hKey, "ProductSuite", NULL, &Type, NULL, &Size );
    if (Rslt != ERROR_SUCCESS || !Size)
        goto exit;

    ProductSuite = (LPSTR) LocalAlloc( LPTR, Size );
    if (!ProductSuite)
        goto exit;

    Rslt = RegQueryValueExA( hKey, "ProductSuite", NULL, &Type,
        (LPBYTE) ProductSuite, &Size );
     if (Rslt != ERROR_SUCCESS || Type != REG_MULTI_SZ)
        goto exit;

    p = ProductSuite;
    while (*p)
    {
        if (lstrcmpA( p, SuiteName ) == 0)
        {
            rVal = TRUE;
            break;
        }
        p += (lstrlenA( p ) + 1);
    }

exit:
    if (ProductSuite)
        LocalFree( ProductSuite );

    if (hKey)
        RegCloseKey( hKey );

    return rVal;
}


BOOL IsTerminalServicesEnabled( VOID )
{
    BOOL    bResult = FALSE;
    DWORD   dwVersion;
    OSVERSIONINFOEXA osVersionInfo;
    DWORDLONG dwlConditionMask = 0;
    HMODULE hmodK32 = NULL;
    typedef ULONGLONG (*PFnVerSetConditionMask) ( ULONGLONG, ULONG, UCHAR );
    typedef BOOL      (*PFnVerifyVersionInfoA) (POSVERSIONINFOEXA, DWORD, DWORDLONG);
    PFnVerSetConditionMask pfnVerSetConditionMask;
    PFnVerifyVersionInfoA pfnVerifyVersionInfoA;


    dwVersion = GetVersion();

    /* are we running NT ? */
    if (!(dwVersion & 0x80000000))
    {
        // Is it NT 50 or greater ?
        if (LOBYTE(LOWORD(dwVersion)) > 4)
        {
            /* In NT5 we need to use the Product Suite APIs
             Don't static link because it won't load on non-NT5 systems */

            hmodK32 = GetModuleHandleA( "KERNEL32.DLL" );
            if (hmodK32)
            {
                pfnVerSetConditionMask = (PFnVerSetConditionMask )GetProcAddress( hmodK32, "VerSetConditionMask");

                if (pfnVerSetConditionMask)
                {
                    /* get the condition mask. */
                    dwlConditionMask = (*pfnVerSetConditionMask)(dwlConditionMask, VER_SUITENAME, VER_AND);

                    pfnVerifyVersionInfoA = (PFnVerifyVersionInfoA)GetProcAddress( hmodK32, "VerifyVersionInfoA") ;

                    if (pfnVerifyVersionInfoA != NULL)
                    {

                        ZeroMemory(&osVersionInfo, sizeof(osVersionInfo));
                        osVersionInfo.dwOSVersionInfoSize = sizeof(osVersionInfo);
                        osVersionInfo.wSuiteMask = VER_SUITE_TERMINAL;
                        bResult = (*pfnVerifyVersionInfoA)(
                                          &osVersionInfo,
                                          VER_SUITENAME,
                                          dwlConditionMask);
                    }
                }
            }
        }
        else
        {
            /* This is NT 40 */
            bResult = ValidateProductSuite( "Terminal Server" );
        }
    }

    return bResult;
}

/*--------------------------------------------------------------------------------------------------------
* DWORD IsStringInMultiString(HKEY hkey, LPCTSTR szkey, LPCTSTR szvalue, LPCTSTR szCheckForString, BOOL *pbFound)
* checks if parameter string exists in given multistring.
* returns error code.
* -------------------------------------------------------------------------------------------------------*/
DWORD IsStringInMultiString(HKEY hkey, LPCTSTR szkey, LPCTSTR szvalue, LPCTSTR szCheckForString, BOOL *pbFound)
{
    ASSERT(szkey && *szkey);
    ASSERT(szvalue && *szvalue);
    ASSERT(szCheckForString&& *szCheckForString);
    ASSERT(*szkey != '\\');
    ASSERT(pbFound);

    // not yet found.
    *pbFound = FALSE;

    CRegistry reg;
    DWORD dwError = reg.OpenKey(hkey, szkey, KEY_READ);  // open up the required key.
    if (dwError == NO_ERROR)
    {
        LPTSTR szSuiteValue;
        DWORD dwSize;
        dwError = reg.ReadRegMultiString(szvalue, &szSuiteValue, &dwSize);
        if (dwError == NO_ERROR)
        {
            LPCTSTR pTemp = szSuiteValue;
            while(_tcslen(pTemp) > 0 )
            {
                if (_tcscmp(pTemp, szCheckForString) == 0)
                {
                    *pbFound = TRUE;
                    break;
                }

                pTemp += _tcslen(pTemp) + 1; // point to the next string within the multistring.
                if ( DWORD(pTemp - szSuiteValue) > (dwSize / sizeof(TCHAR)))
                    break; // temporary pointer passes the size of the szSuiteValue something is wrong with szSuiteValue.
            }
        }
    }

    return dwError;

}


BOOL DoesHydraKeysExists()
{
    BOOL bStringExists = FALSE;
    DWORD dw = IsStringInMultiString(
        HKEY_LOCAL_MACHINE,
        _T("SYSTEM\\CurrentControlSet\\Control\\ProductOptions"),
        _T("ProductSuite"),
        _T("Terminal Server"),
        &bStringExists);

    return (dw == ERROR_SUCCESS) && bStringExists;
}






TCHAR *IsItAppServer ()
{
    return ((GetOSVersionInfo()->wSuiteMask & VER_SUITE_TERMINAL) &&
           !(GetOSVersionInfo()->wSuiteMask & VER_SUITE_SINGLEUSERTS)) ? _T("Yes") : _T("No");
}


BOOL IsItServer ()
{
    return GetOSVersionInfo()->wProductType != VER_NT_WORKSTATION;
}


TCHAR *GetProductType ()
{
    BYTE wProductType = GetOSVersionInfo()->wProductType;
    _tcscpy(ReturnBuffer(), _T(""));

    if (wProductType == VER_NT_WORKSTATION)
    {
        _tcscat(ReturnBuffer(), _T("VER_NT_WORKSTATION "));
    }

    if (wProductType == VER_NT_DOMAIN_CONTROLLER)
    {
        _tcscat(ReturnBuffer(), _T("VER_NT_DOMAIN_CONTROLLER"));
    }

    if (wProductType == VER_NT_SERVER)
    {
        _tcscat(ReturnBuffer(), _T("VER_NT_SERVER"));
    }

    return ReturnBuffer();

}

TCHAR *GetProductSuite ()
{
    WORD wProductSuite = GetOSVersionInfo()->wSuiteMask;
    _tcscpy(ReturnBuffer(), _T(""));

    if (wProductSuite & VER_SERVER_NT)
    {
        _tcscat(ReturnBuffer(), _T("VER_SERVER_NT "));

    }
    if (wProductSuite & VER_WORKSTATION_NT)
    {
        _tcscat(ReturnBuffer(), _T("VER_WORKSTATION_NT "));

    }
    if (wProductSuite & VER_SUITE_SMALLBUSINESS)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_SMALLBUSINESS "));

    }
    if (wProductSuite & VER_SUITE_ENTERPRISE)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_ENTERPRISE "));

    }
    if (wProductSuite & VER_SUITE_BACKOFFICE)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_BACKOFFICE "));

    }
    if (wProductSuite & VER_SUITE_COMMUNICATIONS)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_COMMUNICATIONS "));

    }
    if (wProductSuite & VER_SUITE_TERMINAL)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_TERMINAL "));

    }
    if (wProductSuite & VER_SUITE_SMALLBUSINESS_RESTRICTED)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_SMALLBUSINESS_RESTRICTED "));

    }
    if (wProductSuite & VER_SUITE_EMBEDDEDNT)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_EMBEDDEDNT "));

    }
    if (wProductSuite & VER_SUITE_DATACENTER)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_DATACENTER "));

    }
    if (wProductSuite & VER_SUITE_SINGLEUSERTS)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_SINGLEUSERTS "));

    }
    if (wProductSuite & VER_SUITE_PERSONAL)
    {
        _tcscat(ReturnBuffer(), _T("VER_SUITE_PERSONAL "));

    }
    return ReturnBuffer();
}


TCHAR *IsServer ()
{
    return IsItServer() ? _T("Its a Server") : _T("Its a WorkStation");
}



BOOL IsKernelTSEnable ()
{
    return IsTerminalServicesEnabled();
}

BOOL TSEnabled ()
{

    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ))
    {
        DWORD dwTSEnabled = 0;
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("TSEnabled"), &dwTSEnabled))
        {
            return dwTSEnabled == 1;
        }
    }

    return FALSE;
}

BOOL DoesProductSuiteContainTS ()
{
    return DoesHydraKeysExists ();
}

BOOL IsTerminalServerRegistryOk ()
{
    CRegistry reg1;
    CRegistry reg2;
    CRegistry reg3;

    return
    (ERROR_SUCCESS == reg1.OpenKey(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ)) &&
    (ERROR_SUCCESS == reg2.OpenKey(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\WinStations"), KEY_READ)) &&
    (ERROR_SUCCESS == reg3.OpenKey(HKEY_LOCAL_MACHINE, _T("Software\\Microsoft\\Windows NT\\CurrentVersion\\Terminal Server"), KEY_READ));
}

TCHAR *GetWinstationList ()
{
    using namespace std;

    TString szWinstationList;


    bool bFoundNonConsoleWinstation = false;

    szWinstationList = _T("");

    CRegistry reg2;
    if (ERROR_SUCCESS == reg2.OpenKey(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\WinStations"), KEY_READ))
    {
        LPTSTR szWinstation;
        DWORD dwSize;

        if (ERROR_SUCCESS == reg2.GetFirstSubKey(&szWinstation, &dwSize))
        {
            bool bFirst = true;

            do
            {
                if (!bFirst)
                {
                    szWinstationList += _T(", ");
                }

                bFirst = false;

                if (0 != _tcsicmp(szWinstation, _T("Console")))
                {
                    bFoundNonConsoleWinstation = true;

                    //
                    // we need to read the listner port for this winstation.
                    //
                    CRegistry regSubKey;
                    if ( ERROR_SUCCESS == regSubKey.OpenKey(reg2, szWinstation, KEY_READ) )
                    {
                        DWORD dwPort;
                        if ( ERROR_SUCCESS != regSubKey.ReadRegDWord(_T("PortNumber"), &dwPort))
                        {
                            dwPort = 0;

                        }
                        TCHAR szPort[20];

                        _itot(dwPort, szPort, 10);
                        
                        szWinstationList += szWinstation; // << _T("(") << dwPort << _T(")");
                        szWinstationList += _T("(");
                        szWinstationList += szPort;
                        szWinstationList += _T(")");
                    }

                }
                else
                {
                    szWinstationList += szWinstation;
                }

            }
            while (ERROR_SUCCESS == reg2.GetNextSubKey(&szWinstation, &dwSize));
        }
    }

    if (szWinstationList.length() == 0)
    {
        szWinstationList = _T("Error, No winstations found.");
    }

    if (!bFoundNonConsoleWinstation)
    {
        szMoreInfo << "ERROR, No non Console Winstation not found" << endl;
    }

    _tcscpy(ReturnBuffer(), szWinstationList.c_str());
    return ReturnBuffer();

}

TCHAR *IsTerminalServiceStartBitSet ()
{

    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Services\\TermService"), KEY_READ))
    {
        DWORD dwTermServStartBit = 0;
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("Start"), &dwTermServStartBit))
        {
            switch (dwTermServStartBit)
            {
                case 2:
                return _T("AutoStart");
                break;

                case 3:
                return _T("Manual Start");
                break;

                case 4:
                return _T("Error, Disabled");
                break;

                default:
                return _T("ERROR:Wrong value for startbit");
            }

        }
    }

    return _T("Error, Reading startbig");
}

BOOL IsTermDDStartBitSet ()
{
    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Services\\TermDD"), KEY_READ))
    {
        DWORD dwTermDDStartBit = 0;
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("Start"), &dwTermDDStartBit))
        {
            return dwTermDDStartBit == 2;
        }
    }

    return FALSE;
}

TCHAR *AreEffectiveConnectionAllowed ()
{
    HMODULE hmodRegAPI = LoadLibrary( _T("RegApi.dll") );

    if (hmodRegAPI)
    {

        typedef BOOLEAN (*PFDenyConnectionPolicy) ();
        PFDenyConnectionPolicy pfnDenyConnectionPolicy;

        pfnDenyConnectionPolicy = (PFDenyConnectionPolicy) GetProcAddress( hmodRegAPI, "RegDenyTSConnectionsPolicy");
        if (pfnDenyConnectionPolicy)
        {
              return (*pfnDenyConnectionPolicy)() ? _T("Not Allowed") : _T("Allowed");

        }
        else
        {
            szMoreInfo << "Failed to get proc RegDenyTSConnectionsPolicy" << endl;
            return _T("Failed");
        }
    }
    else
    {
       szMoreInfo << "Failed to Load regapi.dll" << endl;
       return _T("Failed");
    }


}

TCHAR *AreConnectionsAllowed ()
{
	DWORD dwError;
	CRegistry oRegTermsrv;

	dwError = oRegTermsrv.OpenKey(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ);
	if (ERROR_SUCCESS == dwError)
	{
		DWORD dwDenyConnect;
		dwError = oRegTermsrv.ReadRegDWord(_T("fDenyTSConnections"), &dwDenyConnect);
		if (ERROR_SUCCESS == dwError)
		{
			return dwDenyConnect ? _T("No, Not allowed") : _T("Yes");
		}
	}

	//
	// could not read registry, It means - for 51 connection not allowed. For 50 Connections allowed.
	//

	return IsIt51TS() ? _T("Error, Value not found") : _T("Yes");
}

BOOL GetTSOCLogFileName(char *strFileName, UINT uiSize)
{
    if (!GetSystemWindowsDirectoryA(strFileName, uiSize))
        return FALSE;

    strcat(strFileName, "\\tsoc.log");
    ASSERT(strlen(strFileName) < uiSize);
    return TRUE;
}

char *IncompleteMessage = "Error:TSOC Did not get OC_COMPLETE_INSTALLATION.";
BOOL DidTsOCgetCompleteInstallationMessage ()
{

    if (!IsTSOClogPresent())
    {
        szMoreInfo << "tsoc.log does not exist." << endl;
        return FALSE;
    }

    char strTSOCLog[256];
    GetTSOCLogFileName(strTSOCLog, 256);

	ifstream ifSrc(strTSOCLog);
	if(!ifSrc)
    {
        szMoreInfo << "Failed to open tsoc.log file." << endl;
        return FALSE;
    }


    char tempSrc[256];
	while(!ifSrc.eof())
	{
		ifSrc.getline(tempSrc, 256);
        if (strstr(tempSrc, IncompleteMessage))
        {
            return FALSE;
        }
	}

    return TRUE;
}

BOOL FileExists(char *pszFullNameAndPath)
{
    ASSERT(pszFullNameAndPath);

    if (pszFullNameAndPath && pszFullNameAndPath[0])
    {
        HANDLE hFile = CreateFileA(pszFullNameAndPath, 0, FILE_SHARE_READ | FILE_SHARE_WRITE,
					              NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

	    if (hFile != INVALID_HANDLE_VALUE)
        {
            CloseHandle(hFile);
            return TRUE;
        }
    }

    return FALSE;
}

BOOL IsTSOClogPresent ()
{
    char strTSOCLog[256];
    GetTSOCLogFileName(strTSOCLog, 256);
    return FileExists(strTSOCLog);
}

BOOL DidOCMInstallTSEnable()
{
    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Setup\\OC Manager\\SubComponents"), KEY_READ))
    {
        DWORD dwTSEnabled = 0;
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("tsenable"), &dwTSEnabled))
        {
            return dwTSEnabled == 1;
        }
    }

    return FALSE;
}

TCHAR *IsClusteringInstalled ()
{
    DWORD dwClusterState;
    if (ERROR_SUCCESS == GetNodeClusterState(NULL, &dwClusterState))
    {
        if (dwClusterState != ClusterStateNotInstalled)
        {
            return _T("***Failed. Clustering is installed, this is not compatible with Terminal Server.");
       }

    }

    return _T("Passed");
}


TCHAR *GetTSMode (void)
{
    return IsRemoteAdminMode() ? _T("Remote Admin") : _T("Application Server");
}

BOOL IsRemoteAdminMode ()
{
    // HKLM ,"SYSTEM\CurrentControlSet\Control\Terminal Server","TSAppCompat",0x00010001,0x0
    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ))
    {
        DWORD dwAppCompat = 1;
        if ( ERROR_SUCCESS == reg.ReadRegDWord( _T("TSAppCompat"), &dwAppCompat))
        {
            return dwAppCompat == 0;
        }
        else
        {
            // if the registry TSAppCompat does not exist it means we are in app server mode.
            return FALSE;

        }
    }
    else
    {
        szMoreInfo << "ERROR:SYSTEM\\CurrentControlSet\\Control\\Terminal Server not found!" << endl;
    }

    // this return is bogus.
    return TRUE;

}

BOOL VerifyModeRegistry()
{
    return CheckModeRegistry(!IsRemoteAdminMode());
}

BOOL CheckModeRegistry (BOOL bAppCompat)
{
    USES_CONVERSION;
    BOOL bOk = TRUE;

    CRegistry reg;

    if (IsItServer())
    {
        CRegistry reg1;

        // check registry value
        // for appcompat mode
            //HKLM ,"SOFTWARE\Microsoft\Windows NT\CurrentVersion\Winlogon","AppSetup",0x00000000,"UsrLogon.Cmd"
        // and for remote admin mode
            //HKLM ,"SOFTWARE\Microsoft\Windows NT\CurrentVersion\Winlogon","AppSetup",0x00000000,""


        if ( ERROR_SUCCESS == reg1.OpenKey( HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), KEY_READ))
        {
            LPTSTR str;
            DWORD dwSize;
            DWORD dwError;
            if (ERROR_SUCCESS == (dwError = reg1.ReadRegString(_T("AppSetup"), &str, &dwSize)))
            {
                if (bAppCompat)
                {
                    if (_tcsicmp(str, _T("UsrLogon.Cmd")) != 0)
                    {
                        bOk = FALSE;
                        szMoreInfo << "ERROR: Wrong value (" << T2A(str) << ") for AppSetup, contact makarp/breenh" << endl;
                    }

                }
                else
                {
                    if (_tcslen(str) != 0)
                    {
                        bOk = FALSE;
                        szMoreInfo << "ERROR: Wrong value (" << T2A(str) << ") for AppSetup, contact makarp/breenh" << endl;

                    }

                }

            }
            else
            {
                szMoreInfo << "ERROR reading appsetup registry (" << dwError << ")" << endl;
                bOk = FALSE;
            }

        }
        else
        {
            szMoreInfo << "ERROR:reading SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon" << endl;
            bOk = FALSE;
        }

    }



    // check registry value
    // for appcompat mode
        //HKLM ,"SYSTEM\CurrentControlSet\Control\Terminal Server","IdleWinStationPoolCount",0x00010001,0x2
    // and for remote admin mode
        //HKLM ,"SYSTEM\CurrentControlSet\Control\Terminal Server","IdleWinStationPoolCount",0x00010001,0x0


    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server"), KEY_READ))
    {
        DWORD dwIdleWinstations;
        if (ERROR_SUCCESS == reg.ReadRegDWord(_T("IdleWinStationPoolCount"), &dwIdleWinstations))
        {
            if (bAppCompat)
            {
                if (0x2 != dwIdleWinstations)
                {
                    bOk = FALSE;
                    szMoreInfo << "ERROR: Wrong IdleWinStationPoolCount (" << dwIdleWinstations << ")" << endl;
                }

            }
            else
            {
                if (0 != dwIdleWinstations)
                {
                    bOk = FALSE;
                    szMoreInfo << "ERROR: Wrong IdleWinStationPoolCount (" << dwIdleWinstations << ")" << endl;
                }

            }

        }
        else
        {
            bOk = FALSE;
            szMoreInfo << "ERROR:Reading IdleWinStationPoolCount registry" << endl;

        }

    }
    else
    {
        bOk = FALSE;
        szMoreInfo << "SYSTEM\\CurrentControlSet\\Control\\Terminal Server" << endl;
    }

    return bOk;


}

TCHAR * TermSrvCommandLine ()
{
    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Services\\TermService"), KEY_READ))
    {
        TCHAR *Imagepath;
        DWORD dwSize;
        if ( ERROR_SUCCESS == reg.ReadRegString(_T("ImagePath"), &Imagepath, &dwSize))
        {
            _tcscpy(ReturnBuffer(), Imagepath);

        }
        else
        {
            _tcscpy(ReturnBuffer(), _T("failed to read value"));
        }

    }
    else
    {
        _tcscpy(ReturnBuffer(), _T("failed to open key"));

    }

    return  ReturnBuffer();

    // termsrv executable has to have svchost.exe
}

BOOL IsTerminalServiceRunning ()
{

    BOOL bReturn = FALSE;

    SC_HANDLE hServiceController = OpenSCManager(NULL, NULL, GENERIC_READ);
    if (hServiceController)
    {

        SC_HANDLE hTermServ = OpenService(hServiceController, _T("TermService"), SERVICE_QUERY_STATUS);
        if (hTermServ)
        {
            SERVICE_STATUS tTermServStatus;
            if (QueryServiceStatus(hTermServ, &tTermServStatus))
            {
                bReturn = (tTermServStatus.dwCurrentState == SERVICE_RUNNING);
            }
            else
            {
                szMoreInfo << "Failed to get service status, Error = " << GetLastError() << endl;

            }

            VERIFY(CloseServiceHandle(hTermServ));

        }
        else
        {
            szMoreInfo << "Failed to open TermServ service, Error = " << GetLastError() << endl;
        }

        VERIFY(CloseServiceHandle(hServiceController));
    }
    else
    {
        szMoreInfo << "Failed to Open Service Controller, Error = " << GetLastError() << endl;
    }

    return bReturn;
}


BOOL CheckVideoKeys ()
{
    //    HKLM ,"SYSTEM\CurrentControlSet\Control\Terminal Server\VIDEO\rdpdd","VgaCompatible",0x00000000,"\Device\Video0"
    //    HKLM ,"SYSTEM\CurrentControlSet\Control\Terminal Server\VIDEO\rdpdd","\Device\Video0",0x00000000,"\REGISTRY\Machine\System\ControlSet001\Services\RDPDD\Device0"

    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\VIDEO\\rdpdd"), KEY_READ))
    {
        LPTSTR str = 0;
        DWORD dwSize = 0;
        if (ERROR_SUCCESS == reg.ReadRegString(_T("VgaCompatible"), &str, &dwSize))
        {
            if (0 == _tcsicmp(str, _T("\\Device\\Video0")))
            {
                if (ERROR_SUCCESS == reg.ReadRegString(_T("\\Device\\Video0"), &str, &dwSize))
                {
                    if ((0 == _tcsicmp(str, _T("\\REGISTRY\\Machine\\System\\ControlSet001\\Services\\RDPDD\\Device0"))) ||
                        (0 == _tcsicmp(str, _T("\\REGISTRY\\Machine\\System\\CurrentControlSet\\Services\\RDPDD\\Device0"))))
                    {
                        return TRUE;
                    }
                    else
                    {
                    }
                }
                else
                {
                }
            }
            else
            {

            }

        }
        else
        {
        }
    }
    else
    {
    }

    return FALSE;
}

TCHAR szCompName[256];
TCHAR *GetCompName ()
{
    DWORD dwCompName = 256;

    if (GetComputerName(szCompName, &dwCompName))
    {
        return szCompName;

    }

    return NULL;

}


TCHAR szDomNameWorkgroup[] = _T("<Workgroup>");
TCHAR szDomNameUnknown[] = _T("<Unknown>");
TCHAR *GetDomName ()
{
    DSROLE_PRIMARY_DOMAIN_INFO_BASIC *pDomainInfo = NULL;
    DWORD dwErr;

    //
    // Check if we're in a workgroup
    //
    dwErr = DsRoleGetPrimaryDomainInformation(NULL,
                                              DsRolePrimaryDomainInfoBasic,
                                              (PBYTE *) &pDomainInfo);

    if (ERROR_SUCCESS != dwErr)
        return szDomNameUnknown;

    switch (pDomainInfo->MachineRole)
    {
        case DsRole_RoleStandaloneWorkstation:
        case DsRole_RoleStandaloneServer:
            return szDomNameWorkgroup;
            break;      // just in case
    }

    if (pDomainInfo->DomainNameFlat)
        return pDomainInfo->DomainNameFlat;
    else if (pDomainInfo->DomainNameDns)
        return pDomainInfo->DomainNameDns;
    else
        return szDomNameUnknown;
}


TCHAR szSiteNameWorkgroup[] = _T("<Workgroup>");
TCHAR szSiteNameUnknown[] = _T("<Unknown>");
TCHAR *GetSiteName ()
{
    LPTSTR szSiteName = NULL;
    DWORD dwErr;
    DSROLE_PRIMARY_DOMAIN_INFO_BASIC *pDomainInfo = NULL;

    //
    // Check if we're in a workgroup
    //
    dwErr = DsRoleGetPrimaryDomainInformation(NULL,
                                              DsRolePrimaryDomainInfoBasic,
                                              (PBYTE *) &pDomainInfo);

    if (ERROR_SUCCESS != dwErr)
        return szSiteNameUnknown;

    switch (pDomainInfo->MachineRole)
    {
        case DsRole_RoleStandaloneWorkstation:
        case DsRole_RoleStandaloneServer:
            return szSiteNameWorkgroup;
            break;      // just in case
    }

    dwErr = DsGetSiteName(NULL,&szSiteName);

    if (NO_ERROR != dwErr)
    {
        return szSiteNameUnknown;
    }

    return szSiteName;
}


WCHAR wszIPAddress[128] = L"<error>";
TCHAR *GetIPAddress ()
{
	//get host address
	WORD wVersionRequested = MAKEWORD( 1, 1 ); 
	WSADATA wsaData;
	if (0 == WSAStartup(wVersionRequested,&wsaData))
    {
        char szHostName[256];

        if (0 == gethostname ( szHostName , 256 ))
        {
            hostent *h;
            if (NULL != (h=gethostbyname ( szHostName )))
            {
                in_addr *inaddr=(struct in_addr *)*h->h_addr_list;

                MultiByteToWideChar(CP_ACP,0,inet_ntoa(*inaddr),-1,wszIPAddress,128);
            }
        }
    }

    return wszIPAddress;
}




TCHAR *IsRDPNPinNetProviders ()
{
    TCHAR NEWORK_PROVIDER_ORDER_KEY[] = _T("SYSTEM\\CurrentControlSet\\Control\\NetworkProvider\\Order");
    TCHAR PROVIDER_ORDER_VALUE[]      = _T("ProviderOrder");
    TCHAR RDPNP_ENTRY[]               = _T("RDPNP");
	BOOL bRdpNpExists				  = FALSE;

	// read network privider key.
	CRegistry regNetOrder;
	LPTSTR szOldValue;
	DWORD dwSize;
	if ((ERROR_SUCCESS == regNetOrder.OpenKey(HKEY_LOCAL_MACHINE, NEWORK_PROVIDER_ORDER_KEY, KEY_READ)) &&
	   (ERROR_SUCCESS == regNetOrder.ReadRegString(PROVIDER_ORDER_VALUE, &szOldValue, &dwSize)))
	{
		bRdpNpExists = (_tcsstr(szOldValue, RDPNP_ENTRY) != NULL);
	
	}

	if (TSEnabled () == bRdpNpExists)
	{
		return (_T("Passed"));
	}
	else
	{
		if (bRdpNpExists)
		{
			return _T("Error: RDPNP, exists in ProviderOrder, but TS is disabled!");
		}
		else
		{
			if (IsIt50TS())
			{
				// rdp np is only for 51+ so its ok if its missing for 50.
				return _T("Passed");
			}
			else
			{
				return _T("ERROR, RDPNP is missing from ProviderOrder");
			}
		}
	}
}



TCHAR *IsMultiConnectionAllowed ()
{
	// SOFTWARE\Microsoft\Windows NT\CurrentVersion\Winlogon","AllowMultipleTSSessions
	CRegistry regWL;
	DWORD dwAllowMultipal;
	if ((ERROR_SUCCESS == regWL.OpenKey(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), KEY_READ)) &&
		(ERROR_SUCCESS == regWL.ReadRegDWord(_T("AllowMultipleTSSessions"), &dwAllowMultipal)))
	{
		return dwAllowMultipal ? _T("Yes") : _T("No");
	}

	if (IsIt50TS() || IsItServer())
	{
		return _T("Yes");
	}
	else
	{
		return _T("ERROR, registry missing");
	}
}

TCHAR *LogonType ()
{
	if (0 == _tcsicmp(GetDomName(), szDomNameWorkgroup))
	{
		CRegistry regWL;
		DWORD dwLogonType;
		if ((ERROR_SUCCESS == regWL.OpenKey(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), KEY_READ)) &&
			(ERROR_SUCCESS == regWL.ReadRegDWord(_T("LogonType"), &dwLogonType)))
		{
			return dwLogonType == 0 ? _T("Classic Gina") : _T("New Fancy");
		}

		if (IsIt50TS())
		{
			return _T("Classic");
		}
		else
		{
			return _T("ERROR, registry missing");
		}
	}
	else
	{
		return _T("Classic Gina");
	}
}

BOOL IsTermSrvInSystemContext ()
{
    USES_CONVERSION;
    CRegistry reg;
    if ( ERROR_SUCCESS == reg.OpenKey( HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Services\\TermService"), KEY_READ))
    {
        TCHAR *szObjectName;
        DWORD dwSize;
        if ( ERROR_SUCCESS == reg.ReadRegString( _T("ObjectName"), &szObjectName, &dwSize))
        {
            if (0 == _tcsicmp(szObjectName, _T("LocalSystem")))
            {
                return TRUE;
            }
            else
            {
                szMoreInfo << "ERROR:Termsrv is set to run using (" << T2A(szObjectName) << ") context." << endl;
            }
        }
        else
        {
            szMoreInfo << "failed to read Objectname" << endl;
        }
    }
    else
    {
        szMoreInfo << "failed to open termsrv registry" << endl;
    }

    return FALSE;

}


BOOL DontRun ()
{
    szMoreInfo << "test not run because its not required." << endl;
    return FALSE;
}
BOOL DoRun ()
{
    szMoreInfo << "this test must be run." << endl;
    return TRUE;
}

TCHAR *SystemDirectory()
{
    TCHAR sysdir[MAX_PATH];
    sysdir[0] = '\0';
    if (GetSystemDirectory(sysdir, MAX_PATH) >= MAX_PATH) {
        sysdir[MAX_PATH -1] = '\0';
    }
    _tcscpy(ReturnBuffer(), sysdir);
    return  ReturnBuffer();
}

BOOL IsListenerSessionPresent()
{
    PWTS_SESSION_INFO pWtsSessionInfo;
    DWORD dwCount;
    bool bReturn = false;


    if (WTSEnumerateSessions(
        WTS_CURRENT_SERVER_HANDLE,
        0,
        1,
        &pWtsSessionInfo,
        &dwCount
         ))
    {
        for (DWORD i = 0; i < dwCount; i++)
        {
            if (pWtsSessionInfo[i].State == WTSListen)
            {
                bReturn = true;
                break;
            }
        }

        WTSFreeMemory(pWtsSessionInfo);
    }
    else
    {
        return FALSE;
    }

    return bReturn;
}

BOOL AreRemoteLogonEnabled ()
{
	CRegistry oRegTermsrv;
	if (ERROR_SUCCESS == oRegTermsrv.OpenKey(
								HKEY_LOCAL_MACHINE,
								_T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"),
								KEY_READ
								))
	{
		DWORD dwSize = 0;
		LPTSTR szWinstationDisabled = NULL;
		if (ERROR_SUCCESS == oRegTermsrv.ReadRegString(_T("WinStationsDisabled"), &szWinstationDisabled, &dwSize))
		{
			if (_tcscmp(_T("0"), szWinstationDisabled) != 0)
			{
                return FALSE;
			}
		}
	}

    return TRUE;

}

#define  DOMAIN_LENGTH 17
#define  USERNAME_LENGTH 20


int
#if !defined(_MIPS_) && !defined(_ALPHA_) && !defined(_PPC_)
_cdecl
#endif
main( int  argc , char ** /* argv */)
{
    typedef BOOL    (PFN_BOOL)(void);
    typedef TCHAR * (PFN_TestSubjective)(void);


    USES_CONVERSION;

    struct TVerificationTest
    {
        char szTestName[256];                   // descriptive name of the test
        PFN_BOOL           *pfnNeedRunTest;     // pointer to function that will be called to decide if the test need run, test is run if NULL.
        PFN_TestSubjective *pfnSubTest;         // pointer to function if the test returns TCHAR *
        PFN_BOOL           *pfnObjTest;         // pointer to function if the test returns BOOL
    }
    theTests[] =
    {
        {"Machine Name...............................", NULL, GetCompName, NULL},
        {"Domain Name................................", NULL, GetDomName, NULL},
        {"Site Name..................................", NULL, GetSiteName, NULL},
        {"IP Address.................................", NULL, GetIPAddress, NULL},
        {"System Dir.................................", NULL, SystemDirectory, NULL},
		{"ProductType................................", NULL, GetProductType, NULL},
		{"ProductSuite...............................", NULL, GetProductSuite, NULL},
		{"TS Version.................................", NULL, GetTSVersion, NULL},
        {"Is this a server machine?..................", NULL, IsServer, NULL},
        {"Is tsoc.log Present?.......................", NULL, NULL, IsTSOClogPresent},
        {"Did TsOC get CompleteInstallationMessage...", NULL, NULL, DidTsOCgetCompleteInstallationMessage},
        {"Does ProductSuite Contain TS?..............", IsItServer, NULL, DoesProductSuiteContainTS},
        {"Did OCM Install TSEnable...................", IsIt50TS, NULL, DidOCMInstallTSEnable},
        {"Is TSEnabled registry alright?.............", NULL, NULL, TSEnabled},
        {"Is Kernel TSEnabled?.......................", IsItServer, NULL, IsKernelTSEnable},
        {"Is Terminal Server Registry Ok?............", NULL, NULL, IsTerminalServerRegistryOk},
        {"GetWinstationList..........................", NULL, GetWinstationList, NULL},
        {"Is TermServ Start Bit Ok?..................", NULL, IsTerminalServiceStartBitSet, NULL},
        {"Is TermServ service running?...............", NULL, NULL, IsTerminalServiceRunning},
        {"Termsrv Service Command....................", NULL, TermSrvCommandLine, NULL},
        {"Is Listner session Present ?...............", IsTerminalServiceRunning, NULL, IsListenerSessionPresent},
        {"Is TermSrv in System Context?..............", NULL, NULL, IsTermSrvInSystemContext},
        {"AreEffectiveConnectionAllowed..............", IsIt51TS, AreEffectiveConnectionAllowed, NULL },
        {"Are Connections Allowed?...................", IsIt51TS, AreConnectionsAllowed, NULL},
        {"Are Remote Logons allowed by winlogon?.....", NULL, NULL, AreRemoteLogonEnabled},
        {"Is RdpDr Installed Properly?...............", NULL, NULL, IsRdpDrInstalledProperly},
		{"Is RdpNP in NetProviders?..................", IsIt51TS, IsRDPNPinNetProviders, NULL},
		{"Are Multipal Connections Allowed...........", IsIt51TS, IsMultiConnectionAllowed, NULL},
        {"Are Video keys setup right?................", NULL, NULL, CheckVideoKeys},
        {"What mode is Terminal Server set in?.......", NULL, GetTSMode, NULL},
//        {"Is mode specific registry ok?..............", NULL, NULL, VerifyModeRegistry},
        {"What is permission Mode set to?............", NULL, GetModePermissions, NULL},
        {"Check termdd signature.....................", NULL, NULL, Check_termdd},
        {"Check tdpipe signature.....................", NULL, NULL, Check_tdpipe},
        {"Check tdtcp signature......................", NULL, NULL, Check_tdtcp},
        {"Check rdpwd signature......................", NULL, NULL, Check_rdpwd},
        {"Check rdpdd signature......................", NULL, NULL, Check_rdpdd},
        {"Check rdpwsx signature.....................", NULL, NULL, Check_rdpwsx},
        {"Is it 56 bit or 128?.......................", NULL, GetCypherStrenthOnRdpwd, NULL},
        {"Is this a Beta System?.....................", NULL, IsBetaSystem, NULL},
        {"Has Licence GracePeriod Expired............", NULL, IsLicenceGracePeriodOk, NULL},
        {"Audio test.................................", IsIt51TS, NULL, IsAudioOk}
    };

    BOOL bEverythingOk = TRUE;
    for (int i=0; i < sizeof(theTests)/sizeof(theTests[0]); i++)
    {
        ASSERT(theTests[i].pfnObjTest || theTests[i].pfnSubTest);
        ASSERT(!(theTests[i].pfnObjTest && theTests[i].pfnSubTest));

        if (theTests[i].pfnNeedRunTest && !(*(theTests[i].pfnNeedRunTest))())
        {
            // we asre asked to skip the test.
            cout << theTests[i].szTestName << "Skipped Test" << endl;
        }
        else
        {
            if (theTests[i].pfnObjTest)
            {
                BOOL bResult = (*(theTests[i].pfnObjTest))();
                cout << theTests[i].szTestName <<  (bResult ? "Passed" : "Failed") << endl;
                if (!bResult)
                    bEverythingOk = FALSE;

            }
            else
            {
                char *szStr = T2A((*(theTests[i].pfnSubTest))());
                if (szStr)
                {
                    cout << theTests[i].szTestName << szStr  << endl;
                }
            }
        }


        //
        // if previous test had any details to tell us
        //

        if (szMoreInfo.pcount())
        {
            char *pStr = szMoreInfo.str();
            cout << "   Details:" << pStr;
            cout << "------------------------------------------------" << endl;

            bEverythingOk = FALSE;
            ZeroMemory(pStr, 512);
            szMoreInfo.seekp(ios::beg);

        }



    }

    cout << endl;



    if (argc == 1 ) // if some argumnent is provided skip these time consuming test.
    {

        cout << "Enumerating Licensing Servers (May take some time)...";
        cout.flush();

        if(!IsRemoteAdminMode())
        {
            if (EnumerateLicenseServers())
            {
                    cout << ".....Passed, Found Some!" << endl;

            }
            else
            {
                cout << ".....Failed. No License Server Found." << endl;
            }
        }
        else
        {
            cout << "No License servers applicable in Remote Admin mode...";   
        }

        cout << "Extracting Related Events from Event Log...";
        cout.flush();
        if (ExtractAllTSEvents())
        {
                cout << ".....Found Some. See if they give any clue." << endl;

        }
        else
        {
            cout << ".....Found None." << endl;
        }


    }



    cout << endl;
    if (bEverythingOk)
    {
        cout << endl;
        cout << "**************************************************************" << endl;
        cout << "*** Nothing wrong with TS could detected by this utility.  ***" << endl;
        cout << "*** If you think something is wrong contact the developer. ***" << endl;
        cout << "**************************************************************" << endl;
        return 0;

    }

    return 1;

}