📄 gpttest.cpp
字号:
_tcscpy( lpszCmdLine, g_pShellInfo->szDllCmdLine );
fRtn = ProcessCmdLine(lpszCmdLine);
delete[] lpszCmdLine;
}
if( !fRtn )
{
g_pKato->Log( LOG_FAIL, TEXT("FAIL : Invalid Command Line Arguments\n"));
return SPR_FAIL;
}
//--------------------------------------
// Added To Framework Program
//--------------------------------------
// Initializes the system to validate the results in the registry
InitializeTests();
break;
case SPM_REGISTER:
// This is the only ShellProc() message that a DLL is required to
// handle (except for SPM_LOAD_DLL if you are UNICODE). This message is
// sent once to the DLL immediately after the SPM_SHELL_INFO message to
// query the DLL for its function table. The spParam will contain a
// pointer to a SPS_REGISTER structure. The DLL should store its
// function table in the lpFunctionTable member of the SPS_REGISTER
// structure. The DLL may return SPR_FAIL to prevent the DLL from
// continuing to load.
Debug(TEXT("ShellProc(SPM_REGISTER, ...) called"));
((LPSPS_REGISTER)spParam)->lpFunctionTable = g_lpFTE;
#ifdef UNICODE
return SPR_HANDLED | SPF_UNICODE;
#else // UNICODE
return SPR_HANDLED;
#endif // UNICODE
case SPM_START_SCRIPT:
// Sent to the DLL immediately before a script is started. It is sent
// to all Tux DLLs, including loaded Tux DLLs that are not in the
// script. All DLLs will receive this message before the first
// TestProc() in the script is called.
Debug(TEXT("ShellProc(SPM_START_SCRIPT, ...) called"));
break;
case SPM_STOP_SCRIPT:
// Sent to the DLL when the script has stopped. This message is sent
// when the script reaches its end, or because the user pressed
// stopped prior to the end of the script. This message is sent to
// all Tux DLLs, including loaded Tux DLLs that are not in the script.
Debug(TEXT("ShellProc(SPM_STOP_SCRIPT, ...) called"));
break;
case SPM_BEGIN_GROUP:
// Sent to the DLL before a group of tests from that DLL is about to
// be executed. This gives the DLL a time to initialize or allocate
// data for the tests to follow. Only the DLL that is next to run
// receives this message. The prior DLL, if any, will first receive
// a SPM_END_GROUP message. For global initialization and
// de-initialization, the DLL should probably use SPM_START_SCRIPT
// and SPM_STOP_SCRIPT, or even SPM_LOAD_DLL and SPM_UNLOAD_DLL.
Debug(TEXT("ShellProc(SPM_BEGIN_GROUP, ...) called"));
g_pKato->BeginLevel(0, TEXT("BEGIN GROUP: GPTTest.DLL"));
break;
case SPM_END_GROUP:
// Sent to the DLL after a group of tests from that DLL has completed
// running. This gives the DLL a time to cleanup after it has been
// run. This message does not mean that the DLL will not be called
// again; it just means that the next test to run belongs to a
// different DLL. SPM_BEGIN_GROUP and SPM_END_GROUP allow the DLL
// to track when it is active and when it is not active.
Debug(TEXT("ShellProc(SPM_END_GROUP, ...) called"));
g_pKato->EndLevel(TEXT("END GROUP: GPTTest.DLL"));
break;
case SPM_BEGIN_TEST:
// Sent to the DLL immediately before a test executes. This gives
// the DLL a chance to perform any common action that occurs at the
// beginning of each test, such as entering a new logging level.
// The spParam parameter will contain a pointer to a SPS_BEGIN_TEST
// structure, which contains the function table entry and some other
// useful information for the next test to execute. If the ShellProc
// function returns SPR_SKIP, then the test case will not execute.
Debug(TEXT("ShellProc(SPM_BEGIN_TEST, ...) called"));
// Start our logging level.
pBT = (LPSPS_BEGIN_TEST)spParam;
g_pKato->BeginLevel(
pBT->lpFTE->dwUniqueID,
TEXT("BEGIN TEST: \"%s\", Threads=%u, Seed=%u"),
pBT->lpFTE->lpDescription,
pBT->dwThreadCount,
pBT->dwRandomSeed);
break;
case SPM_END_TEST:
// Sent to the DLL after a single test executes from the DLL.
// This gives the DLL a time perform any common action that occurs at
// the completion of each test, such as exiting the current logging
// level. The spParam parameter will contain a pointer to a
// SPS_END_TEST structure, which contains the function table entry and
// some other useful information for the test that just completed. If
// the ShellProc returned SPR_SKIP for a given test case, then the
// ShellProc() will not receive a SPM_END_TEST for that given test case.
Debug(TEXT("ShellProc(SPM_END_TEST, ...) called"));
// End our logging level.
pET = (LPSPS_END_TEST)spParam;
g_pKato->EndLevel(
TEXT("END TEST: \"GPT%d - %s\", %s, Time=%u.%03u"),
GPTx,
pET->lpFTE->lpDescription,
pET->dwResult == TPR_SKIP ? TEXT("SKIPPED") :
pET->dwResult == TPR_PASS ? TEXT("PASSED") :
pET->dwResult == TPR_FAIL ? TEXT("FAILED") : TEXT("ABORTED"),
pET->dwExecutionTime / 1000, pET->dwExecutionTime % 1000);
break;
case SPM_EXCEPTION:
// Sent to the DLL whenever code execution in the DLL causes and
// exception fault. TUX traps all exceptions that occur while
// executing code inside a test DLL.
Debug(TEXT("ShellProc(SPM_EXCEPTION, ...) called"));
g_pKato->Log(LOG_EXCEPTION, TEXT("Exception occurred!"));
break;
default:
// Any messages that we haven't processed must, by default, cause us
// to return SPR_NOT_HANDLED. This preserves compatibility with future
// versions of the TUX shell protocol, even if new messages are added.
return SPR_NOT_HANDLED;
}
return SPR_HANDLED;
}
////////////////////////////////////////////////////////////////////////////////
//----------------------------------------------------
// These two functions added To Framework Program
//----------------------------------------------------
//------------------------------------------------------------------------------
//
// Function: InitializeTests
//
// This function initializes the set of tests. It maps the memory to the
// registers.
//
// Parameters:
// None.
//
// Returns:
// TRUE if the memory map succeeds; FALSE if the memory map fails.
//
//------------------------------------------------------------------------------
static BOOL
InitializeTests(void)
{
PHYSICAL_ADDRESS phyAddr;
Debug((TEXT("+%s()\r\n"), __WFUNCTION__));
g_pKato->Log(LOG_COMMENT, TEXT("GptTest on GPT%d\r\n"), GPTx);
hGpt = GptOpenHandle(GPTx);
if ((hGpt == NULL) || (hGpt == INVALID_HANDLE_VALUE))
{
g_pKato->Log(GPT_ZONE_ERROR,
TEXT("GptTest1: GptGetHandle failed with err %d!\r\n"), GetLastError());
hGpt = NULL;
return TPR_FAIL;
}
switch (GPTx)
{
//case 1: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT1; break;
case 2: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT2; break;
case 3: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT3; break;
case 4: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT4; break;
case 5: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT5; break;
case 6: phyAddr.QuadPart = CSP_BASE_REG_PA_GPT6; break;
}
Debug(TEXT("GptTest on GPT%d, PhysAddress (0x%%x)\r\n"), GPTx, phyAddr.QuadPart);
// Allocate the Register
g_pGPT = (PCSP_GPT_REGS)MmMapIoSpace(phyAddr, sizeof(CSP_GPT_REGS), FALSE);
if (g_pGPT == NULL)
{
Debug((TEXT("-%s returning FALSE\r\n"), __WFUNCTION__));
return FALSE;
}
Debug((TEXT("-%s returning TRUE\r\n"), __WFUNCTION__));
return TRUE;
}
//------------------------------------------------------------------------------
//
// Function: FinishTests
//
// This function performs the clean up after the tests have finished.
//
// Parameters:
// None.
//
// Returns:
// None.
//
//------------------------------------------------------------------------------
static VOID
FinishTests(void)
{
LPTSTR eventString = L"GptTest1";
Debug((TEXT("+%s()\r\n"), __WFUNCTION__));
GptReleaseTimerEvent(hGpt, eventString);
GptCloseHandle(hGpt);
Debug((TEXT("-%s()\r\n"), __WFUNCTION__));
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -