ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

pdh: Implement PdhMakeCounterPathA/W.

This commit is contained in:
Hans Leidekker 2009-05-08 11:17:15 +02:00 committed by Alexandre Julliard
parent 9990bde993
commit 754b97c72c
4 changed files with 207 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlls/pdh/pdh.spec
View File

@ -79,8 +79,8 @@
@ stdcall PdhLookupPerfIndexByNameW(wstr wstr ptr)
@ stdcall PdhLookupPerfNameByIndexA(str long ptr ptr)
@ stdcall PdhLookupPerfNameByIndexW(wstr long ptr ptr)
@ stub PdhMakeCounterPathA
@ stub PdhMakeCounterPathW
@ stdcall PdhMakeCounterPathA(ptr ptr ptr long)
@ stdcall PdhMakeCounterPathW(ptr ptr ptr long)
@ stub PdhOpenLogA
@ stub PdhOpenLogW
@ stdcall PdhOpenQuery(wstr long ptr) PdhOpenQueryW

104
dlls/pdh/pdh_main.c
View File

@ -1060,6 +1060,110 @@ PDH_STATUS WINAPI PdhValidatePathExW( PDH_HLOG source, LPCWSTR path )
return PdhValidatePathW( path );
}
/***********************************************************************
* PdhMakeCounterPathA (PDH.@)
*/
PDH_STATUS WINAPI PdhMakeCounterPathA( PDH_COUNTER_PATH_ELEMENTS_A *e, LPSTR buffer,
LPDWORD buflen, DWORD flags )
{
PDH_STATUS ret = PDH_MEMORY_ALLOCATION_FAILURE;
PDH_COUNTER_PATH_ELEMENTS_W eW;
WCHAR *bufferW;
DWORD buflenW;
TRACE("%p %p %p 0x%08x\n", e, buffer, buflen, flags);
if (!e || !buflen) return PDH_INVALID_ARGUMENT;
memset( &eW, 0, sizeof(eW) );
if (e->szMachineName && !(eW.szMachineName = pdh_strdup_aw( e->szMachineName ))) goto done;
if (e->szObjectName && !(eW.szObjectName = pdh_strdup_aw( e->szObjectName ))) goto done;
if (e->szInstanceName && !(eW.szInstanceName = pdh_strdup_aw( e->szInstanceName ))) goto done;
if (e->szParentInstance && !(eW.szParentInstance = pdh_strdup_aw( e->szParentInstance ))) goto done;
if (e->szCounterName && !(eW.szCounterName = pdh_strdup_aw( e->szCounterName ))) goto done;
eW.dwInstanceIndex = e->dwInstanceIndex;
buflenW = 0;
ret = PdhMakeCounterPathW( &eW, NULL, &buflenW, flags );
if (ret == PDH_MORE_DATA)
{
if ((bufferW = heap_alloc( buflenW * sizeof(WCHAR) )))
{
if (!(ret = PdhMakeCounterPathW( &eW, bufferW, &buflenW, flags )))
{
int len = WideCharToMultiByte(CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL);
if (*buflen >= len) WideCharToMultiByte(CP_ACP, 0, bufferW, -1, buffer, *buflen, NULL, NULL);
else ret = PDH_MORE_DATA;
*buflen = len;
}
heap_free( bufferW );
}
}
done:
heap_free( eW.szMachineName );
heap_free( eW.szObjectName );
heap_free( eW.szInstanceName );
heap_free( eW.szParentInstance );
heap_free( eW.szCounterName );
return ret;
}
/***********************************************************************
* PdhMakeCounterPathW (PDH.@)
*/
PDH_STATUS WINAPI PdhMakeCounterPathW( PDH_COUNTER_PATH_ELEMENTS_W *e, LPWSTR buffer,
LPDWORD buflen, DWORD flags )
{
static const WCHAR bslash[] = {'\\',0};
static const WCHAR fslash[] = {'/',0};
static const WCHAR lparen[] = {'(',0};
static const WCHAR rparen[] = {')',0};
static const WCHAR fmt[] = {'#','%','u',0};
WCHAR path[PDH_MAX_COUNTER_NAME], instance[12];
PDH_STATUS ret = ERROR_SUCCESS;
DWORD len;
TRACE("%p %p %p 0x%08x\n", e, buffer, buflen, flags);
if (flags) FIXME("unimplemented flags 0x%08x\n", flags);
if (!e || !e->szCounterName || !e->szObjectName || !buflen)
return PDH_INVALID_ARGUMENT;
path[0] = 0;
if (e->szMachineName)
{
strcatW(path, bslash);
strcatW(path, bslash);
strcatW(path, e->szMachineName);
}
strcatW(path, bslash);
strcatW(path, e->szObjectName);
if (e->szInstanceName)
{
strcatW(path, lparen);
if (e->szParentInstance)
{
strcatW(path, e->szParentInstance);
strcatW(path, fslash);
}
strcatW(path, e->szInstanceName);
sprintfW(instance, fmt, e->dwInstanceIndex);
strcatW(path, instance);
strcatW(path, rparen);
}
strcatW(path, bslash);
strcatW(path, e->szCounterName);
len = strlenW(path) + 1;
if (*buflen >= len) strcpyW(buffer, path);
else ret = PDH_MORE_DATA;
*buflen = len;
return ret;
}
/***********************************************************************
* PdhEnumObjectItemsA (PDH.@)
*/

98
dlls/pdh/tests/pdh.c
View File

@ -837,6 +837,103 @@ static void test_PdhCollectQueryDataEx(void)
ok(status == ERROR_SUCCESS, "PdhCloseQuery failed 0x%08x\n", status);
}
static void test_PdhMakeCounterPathA(void)
{
PDH_STATUS ret;
PDH_COUNTER_PATH_ELEMENTS_A e;
char buffer[1024];
DWORD buflen;
ret = PdhMakeCounterPathA(NULL, NULL, NULL, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
buflen = 0;
ret = PdhMakeCounterPathA(NULL, NULL, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
buflen = 0;
ret = PdhMakeCounterPathA(NULL, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
buflen = sizeof(buffer);
memset(&e, 0, sizeof(e));
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szMachineName = (char *)"machine";
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, ""), "expected \"machine\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szObjectName = (char *)"object";
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, ""), "expected \"machine\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szInstanceName = (char *)"instance";
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, ""), "expected \"machine\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szParentInstance = (char *)"parent";
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, ""), "expected \"machine\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.dwInstanceIndex = 1;
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, ""), "expected \"machine\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szCounterName = (char *)"counter";
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == ERROR_SUCCESS, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, "\\\\machine\\object(parent/instance#1)\\counter"),
"expected \"\\\\machine\\object(parent/instance#1)\\counter\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szParentInstance = NULL;
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == ERROR_SUCCESS, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, "\\\\machine\\object(instance#1)\\counter"),
"expected \"\\\\machine\\object(instance#1)\\counter\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szInstanceName = NULL;
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == ERROR_SUCCESS, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, "\\\\machine\\object\\counter"),
"expected \"\\\\machine\\object\\counter\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szMachineName = NULL;
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == ERROR_SUCCESS, "PdhMakeCounterPathA failed 0x%08x\n", ret);
ok(!strcmp(buffer, "\\object\\counter"),
"expected \"\\object\\counter\" got %s\n", buffer);
buffer[0] = 0;
buflen = sizeof(buffer);
e.szObjectName = NULL;
ret = PdhMakeCounterPathA(&e, buffer, &buflen, 0);
ok(ret == PDH_INVALID_ARGUMENT, "PdhMakeCounterPathA failed 0x%08x\n", ret);
}
START_TEST(pdh)
{
if (PRIMARYLANGID(LANGIDFROMLCID(GetThreadLocale())) != LANG_ENGLISH)
@ -877,4 +974,5 @@ START_TEST(pdh)
if (pPdhValidatePathExW) test_PdhValidatePathExW();
test_PdhCollectQueryDataEx();
test_PdhMakeCounterPathA();
}

3
include/pdh.h
View File

@ -190,6 +190,9 @@ PDH_STATUS WINAPI PdhLookupPerfIndexByNameW(LPCWSTR, LPCWSTR, LPDWORD);
PDH_STATUS WINAPI PdhLookupPerfNameByIndexA(LPCSTR, DWORD, LPSTR, LPDWORD);
PDH_STATUS WINAPI PdhLookupPerfNameByIndexW(LPCWSTR, DWORD, LPWSTR, LPDWORD);
#define PdhLookupPerfNameByIndex WINELIB_NAME_AW(PdhLookupPerfNameByIndex)
PDH_STATUS WINAPI PdhMakeCounterPathA(PDH_COUNTER_PATH_ELEMENTS_A *, LPSTR, LPDWORD, DWORD);
PDH_STATUS WINAPI PdhMakeCounterPathW(PDH_COUNTER_PATH_ELEMENTS_W *, LPWSTR, LPDWORD, DWORD);
#define PdhMakeCounterPath WINELIB_NAME_AW(PdhMakeCounterPath)
PDH_STATUS WINAPI PdhOpenQueryA(LPCSTR, DWORD_PTR, PDH_HQUERY *);
PDH_STATUS WINAPI PdhOpenQueryW(LPCWSTR, DWORD_PTR, PDH_HQUERY *);
#define PdhOpenQuery WINELIB_NAME_AW(PdhOpenQuery)