1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*___INFO__MARK_BEGIN__*/
/*************************************************************************
*
* The Contents of this file are made available subject to the terms of
* the Sun Industry Standards Source License Version 1.2
*
* Sun Microsystems Inc., March, 2001
*
*
* Sun Industry Standards Source License Version 1.2
* =================================================
* The contents of this file are subject to the Sun Industry Standards
* Source License Version 1.2 (the "License"); You may not use this file
* except in compliance with the License. You may obtain a copy of the
* License at http://gridengine.sunsource.net/Gridengine_SISSL_license.html
*
* Software provided under this License is provided on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING,
* WITHOUT LIMITATION, WARRANTIES THAT THE SOFTWARE IS FREE OF DEFECTS,
* MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE, OR NON-INFRINGING.
* See the License for the specific provisions governing your rights and
* obligations concerning the Software.
*
* The Initial Developer of the Original Code is: Sun Microsystems, Inc.
*
* Copyright: 2001 by Sun Microsystems, Inc.
*
* All Rights Reserved.
*
************************************************************************/
/*___INFO__MARK_END__*/
#include "stdafx.h"
#include <windows.h>
#include <iostream>
#include <stdlib.h>
#include <stdio.h>
#include <Userenv.h>
#include <pdh.h>
#include <pdhmsg.h>
#include "CSehTranslator.h"
#include "CException.h"
#include "CError.h"
#include "CTimer.h"
#include "CTrace.h"
#include "CProcess.h"
#include "CRegistryKey.h"
#include "CConfiguration.h"
#include "CShepherd.h"
#include "CWorker.h"
using namespace std;
using namespace GridEngine;
int wmain(int argc, LPTSTR *argv[], LPTSTR *envp[]) {
// enable the SEH to CEH translator
_se_translator_function pOldTranslator;
pOldTranslator = _set_se_translator(TranslateSeToCe);
// enable tracing
CTrace TheTrace(L"Test");
CTrace::CEnter Enter(CTrace::Layer::KERNEL, L"wmain(int, LPTSTR**, LPTSTR**)");
try {
HQUERY HostQuery;
PDH_STATUS Ret;
DWORD dwUserValue;
dwUserValue = 1;
Ret = PdhOpenQuery(0, dwUserValue, &HostQuery);
if (Ret != ERROR_SUCCESS) {
WCHAR szMessage[4096];
wsprintf(szMessage, L"%s (0x%lx)", L"PdhOpenQuery failed", Ret);
throw CCodineException(szMessage, __FILE__, __LINE__);
} else {
DWORD dwSize = 1024;
WCHAR *szBuffer;
do {
dwSize *= 2;
szBuffer = new TCHAR[dwSize];
Ret = PdhExpandWildCardPath(NULL, L"\\Processor(*)\\% Processor Time", szBuffer, &dwSize, 0);
} while (Ret == PDH_MORE_DATA);
if (Ret != PDH_CSTATUS_VALID_DATA) {
WCHAR szMessage[4096];
wsprintf(szMessage, L"%s (0x%lx)", L"PdhExpandWildCardPath failed", Ret);
throw CCodineException(szMessage, __FILE__, __LINE__);
} else {
int nCounter;
while(*szBuffer != L'\0') {
wcout << szBuffer << endl;
szBuffer += wcslen(szBuffer) + 1;
}
}
Ret = PdhCloseQuery(HostQuery);
if (Ret != ERROR_SUCCESS) {
WCHAR szMessage[4096];
wsprintf(szMessage, L"%s (0x%lx)", L"PdhCloseQuery failed", Ret);
throw CCodineException(szMessage, __FILE__, __LINE__);
}
}
} catch (CCodineException& ex) {
wcout << ex;
} catch (CException& ex) {
wcout << ex;
} catch (...) {
wcout << L"Caught unknown exception in main()" << endl;
}
// disable trabslator
_set_se_translator(pOldTranslator);
return 0;
}
|
