windows-server-2003/net/ias/mmc/nap/timeofday.cpp

/****************************************************************************************
 * NAME:	TimeOfDay.cpp
 *
 * OVERVIEW
 *
 * APIs for getting the Time of Day constraint
 *  
 * Copyright (C) Microsoft Corporation, 1998 - 1999 .  All Rights Reserved.
 *
 * History:	
 *				2/14/98		Created by	Byao	
 *
 *****************************************************************************************/
#include "precompiled.h"


//
// declarations for IAS mapping APIs
//
#include "textmap.h"

//
// declaration for the API
#include "TimeOfDay.h"
#include "iasdebug.h"

static BYTE		bitSetting[8] = { 0x80, 0x40, 0x20, 0x10, 0x8, 0x4, 0x2, 0x1};

//+---------------------------------------------------------------------------
//
// Function:  ReverseHourMap
//
// Synopsis:  reverse each byte in the hour map
//				
// we have to do this because LogonUI changes the way HourMap is stored(they
// reversed all the bit. We need to do this so our conversion code can leave
// intact.
//
// Arguments: [in] BYTE* map - hour map
//			  [in] INT nByte - how many bytes are in this hour map	
//			  
// History:   byao	4/10/98 10:33:57 PM
//
//+---------------------------------------------------------------------------
void ReverseHourMap(BYTE *map, int nByte)
{
	int i, j;
	BYTE temp;

	for (i=0; i<nByte; i++)
	{
		temp = 0;
		for (j=0;j<8;j++)
		{
			// set the value temp
			if ( map[i] & bitSetting[j] )
			{
				temp |= bitSetting[7-j];
			}
		}
		map[i] = temp;
	}
}

//+---------------------------------------------------------------------------
//
// Function:  HourMapToStr
//
// Synopsis:  convert the 21-byte hour map to a text string in the following
//			  format: 
//				0 8:30-9:30 10:30-15:30; 2 10:00-14:00
//				
//
// Arguments: [in]	BYTE* map - hour map
//			  [out] ATL::CString& strHourMap  - hour map in string format
//
// History:   Created Header		byao	2/14/98 10:33:57 PM
//
//+---------------------------------------------------------------------------
void HourMapToStr(BYTE* map, ATL::CString& strHourMap) 
{
	int			sh, eh;	// start hour, (min), end hour (min)
	BYTE*		pHourMap;
	int			i, j;

	//
	// todo: change to a safer allocation method
	//
	WCHAR		wzTmpStr[128] = L"";
	WCHAR		wzStr[2048] = L"";
	WCHAR		wzHourStr[8192] = L"";

	BOOL		bFirstDay=TRUE;

	pHourMap = map;
	
	for( i = 0; i < 7; i++)	// for each day
	{
		// if any value for this day
		if(*pHourMap || *(pHourMap + 1) || *(pHourMap + 2))
		{
			// the string for this day
				if (bFirstDay) 
				{
					wsprintf(wzTmpStr, _T("%1d"), i);	
					bFirstDay = FALSE;
				}
				else
				{
					wsprintf(wzTmpStr, _T("; %1d"), i);	
				}
				wcscpy(wzStr, wzTmpStr);

				sh = -1; eh = -1;	// not start yet
				for(j = 0; j < 24; j++)	// for every hour
				{
					int	k = j / 8;
					int m = j % 8;
					if(*(pHourMap + k) & bitSetting[m])	// this hour is on
					{
						if(sh == -1)	sh = j;			// set start hour is empty
						eh = j;							// extend end hour
					}
					else	// this is not on
					{
						if(sh != -1)		// some hours need to write out
						{
							wsprintf(wzTmpStr,_T(" %02d:00-%02d:00"), sh, eh + 1);
							wcscat(wzStr, wzTmpStr);
							sh = -1; eh = -1;
						}
					}
				}
				if(sh != -1)
				{
					wsprintf(wzTmpStr, _T(" %02d:00-%02d:00"), sh, eh + 1);
					wcscat(wzStr, wzTmpStr);
					sh = -1; eh = -1;
				}
		
				wcscat(wzHourStr, wzStr);
		}
		pHourMap += 3;
	}

    strHourMap = wzHourStr;

	return;
}		


//+---------------------------------------------------------------------------
//
// Function:  GetTODConstaint
//
// Synopsis:  Get the time of day constraint
//			  This is implemented by calling an API in NT team
//			  LogonScheduleDialog(...);
//
// Arguments: [in/out] ATL::CString &strHourMap - TOD constraint in text format
//
// Returns:   HRESULT - 
//
// History:   Created Header    byao 2/14/98 10:36:27 PM
//
//+---------------------------------------------------------------------------
typedef HRESULT (APIENTRY *PFN_LOGONSCHEDULEDIALOGEX)(
						  HWND hwndParent		// parent window
						, BYTE ** pprgbData		// pointer to pointer to array of 21 bytes
						, LPCTSTR pszTitle		// dialog title
						, DWORD dwFlags			// in what format to accept the time
						);

HRESULT GetTODConstaint( ATL::CString &strHourMap )
{
	TRACE_FUNCTION("::GetTODConstraint");

	BYTE	TimeOfDayHoursMap[21];
	BYTE*	pMap = &(TimeOfDayHoursMap[0]);
	ATL::CString strDialogTitle;
	DWORD	dwRet;
	HRESULT hr	= S_OK;
	PFN_LOGONSCHEDULEDIALOGEX  pfnLogonScheduleDialogEx = NULL; 
	HMODULE					 hLogonScheduleDLL		= NULL;

	ZeroMemory(TimeOfDayHoursMap, 21);

    // 
    // convert the TOD constraint string to HourMap
    // 
	dwRet = IASHourMapFromText(strHourMap, pMap);
   
	if (NO_ERROR != dwRet)
		goto win32_error;

	// ReverseHourMap() will reverse each byte of the hour map, basically
 	// reverse every bit in the byte.
	// we have to do this because LogonUI changes the way HourMap is stored(they
	// reversed all the bit. We need to do this so our conversion code can leave
	// intact.
	//
	// We reverse it here so it can be understood by the LogonSchedule api
	//
	ReverseHourMap(pMap,21);

    // 
    // get the new TOD Constaint
    // 
	if (!strDialogTitle.LoadString(IDS_TOD_DIALOG_TITLE))
		goto win32_error;

	hLogonScheduleDLL = LoadLibrary(_T("loghours.dll"));
	if ( NULL == hLogonScheduleDLL )
	{
		hr = HRESULT_FROM_WIN32(GetLastError());
		ErrorTrace(ERROR_NAPMMC_TIMEOFDAY, "LoadLibrary() failed, err = %x", hr);

		ShowErrorDialog(NULL, IDS_ERROR_CANT_FIND_LOGHOURSDLL, NULL, hr);
		goto win32_error;
	}
	
	pfnLogonScheduleDialogEx = (PFN_LOGONSCHEDULEDIALOGEX) GetProcAddress(hLogonScheduleDLL, "DialinHoursDialogEx");
	if ( ! pfnLogonScheduleDialogEx )
	{
		hr = HRESULT_FROM_WIN32(GetLastError());
		ErrorTrace(ERROR_NAPMMC_TIMEOFDAY, "GetProcAddress() failed for DialinHoursDialogEx, err = %x", hr);

		ShowErrorDialog(NULL, IDS_ERROR_CANT_FIND_LOGHOURSAPI, NULL, hr);
		FreeLibrary(hLogonScheduleDLL);
		goto win32_error;
	}
	
	//
	// now we do have this DLL, call the API
	//
	hr = pfnLogonScheduleDialogEx(	  NULL		// We don't have an HWND available to pass, but NULL asks the dialog to display itself modally as desired.
									, (BYTE**)&pMap
									, strDialogTitle
									, 1		// This is defined in loghrapi.h (which we don't have access to) to mean "accept in local time".
									);
	FreeLibrary(hLogonScheduleDLL);
	DebugTrace(DEBUG_NAPMMC_TIMEOFDAY, "LogonScheduleDialogEx() returned %x", hr);

	if ( FAILED(hr) )
	{
		goto win32_error;
	}

    // 
    // convert the hourmap to a text string
    // 
	// We need to reverse it first so our conversion code can understand it.
	//
	ReverseHourMap(pMap,21);
	HourMapToStr(pMap, strHourMap) ;

	return S_OK;

win32_error:
	ShowErrorDialog(NULL,
					IDS_ERROR_TIMEOFDAY, 
					NULL, 
					HRESULT_FROM_WIN32(GetLastError())
				);
	return hr;
}