Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

594 lines
16 KiB

//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1994 - 1996.
//
// File: taskwiz.cxx
//
// Contents: Class which creates and invokes the 'create new task' wizard.
//
// Classes: CTaskWizard
//
// History: 5-20-1997 DavidMun Created
//
//---------------------------------------------------------------------------
#include "..\pch\headers.hxx"
#pragma hdrstop
#include "myheaders.hxx"
#include "..\folderui\jobidl.hxx"
//
// Types
//
// SFindWiz - passed to window enumeration function. BUGBUG when wizard
// on remote machine is supported, the 'tszFocus' member should be the
// server name. For now, it is just the path to the tasks folder.
//
struct SFindWiz
{
BOOL fFound;
LPCTSTR tszFocus;
};
//
// Globals
//
// g_msgFindWizard - private window message used to interrogate the wizard
// dialog proc during the find operation.
//
// TEMPLATE_STR - string used to create private message, also used by
// folderui code to identify the template icon.
//
UINT g_msgFindWizard;
extern const TCHAR TEMPLATE_STR[];
//
// External references
//
extern HRESULT
QuietStartContinueService(); // ..\folderui\schstate.cxx
extern HRESULT
JFGetDataObject(
LPCTSTR pszFolderPath,
LPCITEMIDLIST pidlFolder,
UINT cidl,
LPCITEMIDLIST * apidl,
BOOL fCut,
LPVOID * ppvObj);
extern HRESULT
DisplayJobProperties(
LPDATAOBJECT pdtobj);
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::CTaskWizard
//
// Synopsis: ctor
//
// Arguments: [ptszFolderPath] - path to tasks folder
//
// History: 5-12-1997 DavidMun Created
//
//---------------------------------------------------------------------------
CTaskWizard::CTaskWizard(
LPCTSTR ptszFolderPath,
LPITEMIDLIST pidlFolder)
{
TRACE_CONSTRUCTOR(CTaskWizard);
SecureZeroMemory(_apWizPages, sizeof _apWizPages);
_fAdvanced = FALSE;
_tszJobObjectFullPath[0] = TEXT('\0');
_pTask = NULL;
lstrcpyn(_tszFolderPath, ptszFolderPath, ARRAYLEN(_tszFolderPath));
_pidlFolder = pidlFolder;
#ifdef WIZARD97
_fUse256ColorBmp = Is256ColorSupported();
#endif // WIZARD97
}
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::~CTaskWizard
//
// Synopsis: dtor
//
// History: 5-19-1997 DavidMun Created
//
//---------------------------------------------------------------------------
CTaskWizard::~CTaskWizard()
{
TRACE_DESTRUCTOR(CTaskWizard);
if (_pTask)
{
_pTask->Release();
}
ILFree(_pidlFolder);
}
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::Launch, static
//
// Synopsis: Run the task wizard in a separate thread.
//
// Arguments: [ptszFolderPath] - path to tasks folder.
//
// Returns: HRESULT
//
// History: 5-19-1997 DavidMun Created
//
// Notes: If an instance of the wizard is already running for the
// target machine, makes that the foreground window and
// returns.
//
//---------------------------------------------------------------------------
HRESULT
CTaskWizard::Launch(
LPCTSTR ptszFolderPath,
LPCITEMIDLIST pidlFolder)
{
TRACE_FUNCTION(CTaskWizard::Launch);
HRESULT hr = S_OK;
CTaskWizard *pNewWiz = NULL;
do
{
//
// Start the service if it isn't running, or continue it if it is
// paused. Since this is the wizard, do it on behalf of the user
// without asking first.
//
// Continue on failure, since it is better to let the user at least
// create the task, even if the service can't be started (user might
// not have permission).
//
hr = QuietStartContinueService();
CHECK_HRESULT(hr);
//
// Create a path string that CPropPage will store. This is the
// full path to the tasks folder, with a trailing backslash. The
// CPropPage will truncate at the last backslash, since most other
// callers give it a task filename.
//
ULONG cchPath = lstrlen(ptszFolderPath);
if (cchPath >= MAX_PATH - 1) // reserve space for trailing backslash
{
hr = E_INVALIDARG;
DEBUG_OUT_HRESULT(hr);
break;
}
TCHAR tszFolderPath[MAX_PATH + 1];
StringCchCopy(tszFolderPath, MAX_PATH + 1, ptszFolderPath);
tszFolderPath[cchPath] = TEXT('\\');
tszFolderPath[cchPath + 1] = TEXT('\0');
//
// Look for an instance of the wizard running and focused on our
// folder path. If one is found, it will make itself foreground
// window, and we can quit.
//
hr = _FindWizard(tszFolderPath);
if (hr == S_OK)
{
break;
}
//
// No wizard is up for the current focus. Create a wizard object
// and run it in a new thread.
//
LPITEMIDLIST pidlFolderCopy = ILClone(pidlFolder);
if (!pidlFolderCopy)
{
hr = E_OUTOFMEMORY;
DEBUG_OUT_HRESULT(hr);
break;
}
pNewWiz = new CTaskWizard(tszFolderPath, pidlFolderCopy);
if (!pNewWiz)
{
ILFree(pidlFolderCopy);
hr = E_OUTOFMEMORY;
DEBUG_OUT_HRESULT(hr);
break;
}
HANDLE hThread;
DWORD idThread;
hThread = CreateThread(NULL,
0,
_WizardThreadProc,
(LPVOID) pNewWiz,
0,
&idThread);
if (!hThread)
{
delete pNewWiz;
DEBUG_OUT_LASTERROR;
hr = HRESULT_FROM_LASTERROR;
break;
}
VERIFY(CloseHandle(hThread));
} while (0);
return hr;
}
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::_WizardThreadProc, static
//
// Synopsis: Displays the wizard (and optionally task property sheet)
// using a separate thread.
//
// Arguments: [pvThis] - CTaskWizard pointer
//
// Returns: HRESULT
//
// History: 5-20-1997 DavidMun Created
//
// Notes: The wizard runs in a separate thread so the explorer ui
// isn't stalled.
//
//---------------------------------------------------------------------------
DWORD WINAPI
CTaskWizard::_WizardThreadProc(
LPVOID pvThis)
{
HRESULT hr = OleInitialize(NULL);
if (FAILED(hr))
{
DEBUG_OUT_HRESULT(hr);
return hr;
}
CTaskWizard *pThis = (CTaskWizard *)pvThis;
__try
{
hr = pThis->_DoWizard();
//
// Once _DoWizard returns, the wizard property sheet has closed.
//
// If the user elected to see the new task's property sheet, then
// _fAdvanced will be set, and the completion page should have
// set a valid filename and interface pointer for the task.
//
// Open the property sheet while we're still in the thread.
//
if (SUCCEEDED(hr) && pThis->_fAdvanced)
{
DEBUG_ASSERT(pThis->_pTask);
DEBUG_ASSERT(*pThis->_tszJobObjectFullPath);
//
// Since we want to see the security page if the object is on NT
// on an NTFS partition, we'll have to call the version of
// DisplayJobProperties that takes a data object.
//
// To get a data object describing the task object, we need an
// itemid for the job, so create one.
//
CJobID jid;
jid.LoadDummy(pThis->_tszJobObjectFullPath);
LPCITEMIDLIST pidl = (LPCITEMIDLIST) &jid;
LPDATAOBJECT pdo = NULL;
TCHAR tszFolderPath[MAX_PATH + 1];
StringCchCopy(tszFolderPath, MAX_PATH + 1, pThis->_tszFolderPath);
LPTSTR ptszLastSlash = _tcsrchr(tszFolderPath, TEXT('\\'));
if (ptszLastSlash && lstrlen(ptszLastSlash) == 1)
{
*ptszLastSlash = TEXT('\0');
}
hr = JFGetDataObject(tszFolderPath, // path to tasks dir
pThis->_pidlFolder, // itemid of tasks folder
1, // one itemid in array
&pidl, // namely, this one
FALSE, // not doing cut/paste
(VOID **) &pdo);
if (SUCCEEDED(hr))
{
hr = DisplayJobProperties(pdo);
CHECK_HRESULT(hr);
}
else
{
DEBUG_OUT_HRESULT(hr);
}
}
delete pThis;
}
__finally
{
OleUninitialize();
}
return (DWORD) hr;
}
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::_DoWizard
//
// Synopsis: Create the wizard pages and invoke the wizard.
//
// Returns: HRESULT
//
// History: 5-20-1997 DavidMun Created
//
// Notes: If wizard is successfully created, doesn't return until
// user hits Cancel or Finish.
//
//---------------------------------------------------------------------------
HRESULT
CTaskWizard::_DoWizard()
{
TRACE_METHOD(CTaskWizard, _DoWizard);
HRESULT hr = S_OK;
UINT i = 0;
HPROPSHEETPAGE ahpsp[NUM_TASK_WIZARD_PAGES];
SecureZeroMemory(ahpsp, sizeof(ahpsp));
do
{
//
// Create all the wizard pages
//
_apWizPages[TWP_WELCOME ] = new CWelcomePage (this, _tszFolderPath, &ahpsp[TWP_WELCOME ]);
_apWizPages[TWP_SELECT_PROGRAM] = new CSelectProgramPage(this, _tszFolderPath, &ahpsp[TWP_SELECT_PROGRAM]);
_apWizPages[TWP_SELECT_TRIGGER] = new CSelectTriggerPage(this, _tszFolderPath, &ahpsp[TWP_SELECT_TRIGGER]);
_apWizPages[TWP_DAILY ] = new CDailyPage (this, _tszFolderPath, &ahpsp[TWP_DAILY ]);
_apWizPages[TWP_WEEKLY ] = new CWeeklyPage (this, _tszFolderPath, &ahpsp[TWP_WEEKLY ]);
_apWizPages[TWP_MONTHLY ] = new CMonthlyPage (this, _tszFolderPath, &ahpsp[TWP_MONTHLY ]);
_apWizPages[TWP_ONCE ] = new COncePage (this, _tszFolderPath, &ahpsp[TWP_ONCE ]);
_apWizPages[TWP_PASSWORD ] = new CPasswordPage (this, _tszFolderPath, &ahpsp[TWP_PASSWORD ]);
_apWizPages[TWP_COMPLETION ] = new CCompletionPage (this, _tszFolderPath, &ahpsp[TWP_COMPLETION ]);
//
// Check that all objects and pages could be created
//
for (i = 0; i < NUM_TASK_WIZARD_PAGES; i++)
{
if (!_apWizPages[i] || !ahpsp[i])
{
hr = E_OUTOFMEMORY;
break;
}
}
//
// Manually destroy the pages if one could not be created, then exit
//
if (FAILED(hr))
{
DEBUG_OUT((DEB_ERROR, "Creation failed, destroying pages\n"));
for (i = 0; i < NUM_TASK_WIZARD_PAGES; i++)
{
if (ahpsp[i])
{
VERIFY(DestroyPropertySheetPage(ahpsp[i]));
}
else if (_apWizPages[i])
{
delete _apWizPages[i];
}
}
break;
}
//
// All pages created, display the wizard
//
PROPSHEETHEADER psh;
SecureZeroMemory(&psh, sizeof(psh));
#ifdef WIZARD97
psh.dwFlags = PSH_WIZARD | PSH_WIZARD97 | PSH_WATERMARK | PSH_HEADER;
psh.pszbmWatermark = _fUse256ColorBmp ? MAKEINTRESOURCE(IDB_WATERMARK256) : MAKEINTRESOURCE(IDB_WATERMARK16);
psh.pszbmHeader = _fUse256ColorBmp ? MAKEINTRESOURCE(IDB_BANNER256) : MAKEINTRESOURCE(IDB_BANNER16);
#else
psh.dwFlags = PSH_WIZARD;
#endif // WIZARD97
psh.dwSize = sizeof(psh);
psh.hInstance = g_hInstance;
psh.hwndParent = NULL;
psh.pszCaption = NULL; // ignored for wizards; see CWelcome init
psh.phpage = ahpsp;
psh.nStartPage = 0;
psh.nPages = NUM_TASK_WIZARD_PAGES;
if (PropertySheet(&psh) == -1)
{
hr = HRESULT_FROM_WIN32(GetLastError());
DEBUG_OUT_HRESULT(hr);
}
} while (0);
return hr;
}
//+--------------------------------------------------------------------------
//
// Function: FindWizardEnumWndProc
//
// Synopsis: EnumWindows callback used to search for a create new task
// wizard opened on the specified focus.
//
// Arguments: [hwnd] - top level window handle
// [lParam] - pointer to SFindWiz struct
//
// Returns: TRUE - not found, continue enumeration
// FALSE - found wizard, quit enumerating
//
// Modifies: SFindWiz struct pointed to by [lParam]
//
// History: 5-20-1997 DavidMun Created
//
//---------------------------------------------------------------------------
BOOL CALLBACK
FindWizardEnumWndProc(
HWND hwnd,
LPARAM lParam)
{
SFindWiz *pfw = (SFindWiz *) lParam;
ULONG pid;
GetWindowThreadProcessId(hwnd, &pid);
do
{
//
// If the window isn't in this process (explorer.exe) then ignore
// it.
//
if (pid != GetCurrentProcessId())
{
break;
}
//
// If it isn't the dialog class, it can't be a wizard.
//
if (!IsDialogClass(hwnd))
{
break;
}
//
// Found a dialog window that was created by this process. If it's
// a wizard, then it should return a valid window which is also of
// dialog class.
//
HWND hwndPage = PropSheet_GetCurrentPageHwnd(hwnd);
if (!IsWindow(hwndPage) || !IsDialogClass(hwndPage))
{
break;
}
//
// Could be a wizard page. Ask it if it's THE wizard for the
// focus. Note it's only possible to get away with sending a pointer
// in the message because we've guaranteed the window belongs to this
// process.
//
ULONG ulResult = (ULONG)SendMessage(hwndPage,
g_msgFindWizard,
0,
(LPARAM) pfw->tszFocus);
if (ulResult == g_msgFindWizard)
{
pfw->fFound = TRUE;
}
} while (0);
return !pfw->fFound; // continue enumerating if not found
}
//+--------------------------------------------------------------------------
//
// Member: CTaskWizard::_FindWizard
//
// Synopsis: Search through top level windows to find a create new task
// wizard which is focused on [ptszFolderPath].
//
// Arguments: [ptszFolderPath] - wizard focus
//
// Returns: S_OK - found
// S_FALSE - not found
//
// History: 5-20-1997 DavidMun Created
//
// Notes: BUGBUG change ptszFolderPath to server name
//
//---------------------------------------------------------------------------
HRESULT
CTaskWizard::_FindWizard(
LPCTSTR ptszFolderPath)
{
SFindWiz fw = { FALSE, ptszFolderPath };
if (!g_msgFindWizard)
{
g_msgFindWizard = RegisterWindowMessage(TEMPLATE_STR);
}
EnumWindows(FindWizardEnumWndProc, (LPARAM) &fw);
return fw.fFound ? S_OK : S_FALSE;
}