ddraw: Register / unregister the ddraw window class from DllMain().
[wine/testsucceed.git] / dlls / pdh / pdh_main.c
blobbdf83a94792006b72a309ce9c14d7ad4306f9842
1 /*
2 * Performance Data Helper (pdh.dll)
4 * Copyright 2007 Andrey Turkin
5 * Copyright 2007 Hans Leidekker
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22 #include <stdarg.h>
23 #include <math.h>
25 #define NONAMELESSUNION
26 #define NONAMELESSSTRUCT
27 #include "windef.h"
28 #include "winbase.h"
30 #include "pdh.h"
31 #include "pdhmsg.h"
32 #include "winperf.h"
34 #include "wine/debug.h"
35 #include "wine/list.h"
36 #include "wine/unicode.h"
38 WINE_DEFAULT_DEBUG_CHANNEL(pdh);
40 static CRITICAL_SECTION pdh_handle_cs;
41 static CRITICAL_SECTION_DEBUG pdh_handle_cs_debug =
43 0, 0, &pdh_handle_cs,
44 { &pdh_handle_cs_debug.ProcessLocksList,
45 &pdh_handle_cs_debug.ProcessLocksList },
46 0, 0, { (DWORD_PTR)(__FILE__ ": pdh_handle_cs") }
48 static CRITICAL_SECTION pdh_handle_cs = { &pdh_handle_cs_debug, -1, 0, 0, 0, 0 };
50 static inline void *heap_alloc( SIZE_T size )
52 return HeapAlloc( GetProcessHeap(), 0, size );
55 static inline void *heap_alloc_zero( SIZE_T size )
57 return HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size );
60 static inline void heap_free( LPVOID mem )
62 HeapFree( GetProcessHeap(), 0, mem );
65 static inline WCHAR *pdh_strdup( const WCHAR *src )
67 WCHAR *dst;
69 if (!src) return NULL;
70 if ((dst = heap_alloc( (strlenW( src ) + 1) * sizeof(WCHAR) ))) strcpyW( dst, src );
71 return dst;
74 static inline WCHAR *pdh_strdup_aw( const char *src )
76 int len;
77 WCHAR *dst;
79 if (!src) return NULL;
80 len = MultiByteToWideChar( CP_ACP, 0, src, -1, NULL, 0 );
81 if ((dst = heap_alloc( len * sizeof(WCHAR) ))) MultiByteToWideChar( CP_ACP, 0, src, -1, dst, len );
82 return dst;
85 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
87 TRACE("(0x%p, %d, %p)\n",hinstDLL,fdwReason,lpvReserved);
89 if (fdwReason == DLL_WINE_PREATTACH) return FALSE; /* prefer native version */
91 if (fdwReason == DLL_PROCESS_ATTACH)
93 DisableThreadLibraryCalls( hinstDLL );
96 return TRUE;
99 union value
101 LONG longvalue;
102 double doublevalue;
103 LONGLONG largevalue;
106 struct counter
108 DWORD magic; /* signature */
109 struct list entry; /* list entry */
110 WCHAR *path; /* identifier */
111 DWORD type; /* counter type */
112 DWORD status; /* update status */
113 LONG scale; /* scale factor */
114 LONG defaultscale; /* default scale factor */
115 DWORD_PTR user; /* user data */
116 DWORD_PTR queryuser; /* query user data */
117 LONGLONG base; /* samples per second */
118 FILETIME stamp; /* time stamp */
119 void (CALLBACK *collect)( struct counter * ); /* collect callback */
120 union value one; /* first value */
121 union value two; /* second value */
124 #define PDH_MAGIC_COUNTER 0x50444831 /* 'PDH1' */
126 static struct counter *create_counter( void )
128 struct counter *counter;
130 if ((counter = heap_alloc_zero( sizeof(struct counter) )))
132 counter->magic = PDH_MAGIC_COUNTER;
133 return counter;
135 return NULL;
138 static void destroy_counter( struct counter *counter )
140 counter->magic = 0;
141 heap_free( counter->path );
142 heap_free( counter );
145 #define PDH_MAGIC_QUERY 0x50444830 /* 'PDH0' */
147 struct query
149 DWORD magic; /* signature */
150 DWORD_PTR user; /* user data */
151 HANDLE thread; /* collect thread */
152 DWORD interval; /* collect interval */
153 HANDLE wait; /* wait event */
154 HANDLE stop; /* stop event */
155 struct list counters; /* counter list */
158 static struct query *create_query( void )
160 struct query *query;
162 if ((query = heap_alloc_zero( sizeof(struct query) )))
164 query->magic = PDH_MAGIC_QUERY;
165 list_init( &query->counters );
166 return query;
168 return NULL;
171 static void destroy_query( struct query *query )
173 query->magic = 0;
174 heap_free( query );
177 struct source
179 DWORD index; /* name index */
180 const WCHAR *path; /* identifier */
181 void (CALLBACK *collect)( struct counter * ); /* collect callback */
182 DWORD type; /* counter type */
183 LONG scale; /* default scale factor */
184 LONGLONG base; /* samples per second */
187 static const WCHAR path_processor_time[] =
188 {'\\','P','r','o','c','e','s','s','o','r','(','_','T','o','t','a','l',')',
189 '\\','%',' ','P','r','o','c','e','s','s','o','r',' ','T','i','m','e',0};
190 static const WCHAR path_uptime[] =
191 {'\\','S','y','s','t','e','m', '\\', 'S','y','s','t','e','m',' ','U','p',' ','T','i','m','e',0};
193 static void CALLBACK collect_processor_time( struct counter *counter )
195 counter->two.largevalue = 500000; /* FIXME */
196 counter->status = PDH_CSTATUS_VALID_DATA;
199 static void CALLBACK collect_uptime( struct counter *counter )
201 counter->two.largevalue = GetTickCount64();
202 counter->status = PDH_CSTATUS_VALID_DATA;
205 #define TYPE_PROCESSOR_TIME \
206 (PERF_SIZE_LARGE | PERF_TYPE_COUNTER | PERF_COUNTER_RATE | PERF_TIMER_100NS | PERF_DELTA_COUNTER | \
207 PERF_INVERSE_COUNTER | PERF_DISPLAY_PERCENT)
209 #define TYPE_UPTIME \
210 (PERF_SIZE_LARGE | PERF_TYPE_COUNTER | PERF_COUNTER_ELAPSED | PERF_OBJECT_TIMER | PERF_DISPLAY_SECONDS)
212 /* counter source registry */
213 static const struct source counter_sources[] =
215 { 6, path_processor_time, collect_processor_time, TYPE_PROCESSOR_TIME, -5, 10000000 },
216 { 674, path_uptime, collect_uptime, TYPE_UPTIME, -3, 1000 }
219 static BOOL pdh_match_path( LPCWSTR fullpath, LPCWSTR path )
221 const WCHAR *p;
223 if (strchrW( path, '\\')) p = fullpath;
224 else p = strrchrW( fullpath, '\\' ) + 1;
225 if (strcmpW( p, path )) return FALSE;
226 return TRUE;
229 /***********************************************************************
230 * PdhAddCounterA (PDH.@)
232 PDH_STATUS WINAPI PdhAddCounterA( PDH_HQUERY query, LPCSTR path,
233 DWORD_PTR userdata, PDH_HCOUNTER *counter )
235 PDH_STATUS ret;
236 WCHAR *pathW;
238 TRACE("%p %s %lx %p\n", query, debugstr_a(path), userdata, counter);
240 if (!path) return PDH_INVALID_ARGUMENT;
242 if (!(pathW = pdh_strdup_aw( path )))
243 return PDH_MEMORY_ALLOCATION_FAILURE;
245 ret = PdhAddCounterW( query, pathW, userdata, counter );
247 heap_free( pathW );
248 return ret;
251 /***********************************************************************
252 * PdhAddCounterW (PDH.@)
254 PDH_STATUS WINAPI PdhAddCounterW( PDH_HQUERY hquery, LPCWSTR path,
255 DWORD_PTR userdata, PDH_HCOUNTER *hcounter )
257 struct query *query = hquery;
258 struct counter *counter;
259 unsigned int i;
261 TRACE("%p %s %lx %p\n", hquery, debugstr_w(path), userdata, hcounter);
263 if (!path || !hcounter) return PDH_INVALID_ARGUMENT;
265 EnterCriticalSection( &pdh_handle_cs );
266 if (!query || query->magic != PDH_MAGIC_QUERY)
268 LeaveCriticalSection( &pdh_handle_cs );
269 return PDH_INVALID_HANDLE;
272 *hcounter = NULL;
273 for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
275 if (pdh_match_path( counter_sources[i].path, path ))
277 if ((counter = create_counter()))
279 counter->path = pdh_strdup( counter_sources[i].path );
280 counter->collect = counter_sources[i].collect;
281 counter->type = counter_sources[i].type;
282 counter->defaultscale = counter_sources[i].scale;
283 counter->base = counter_sources[i].base;
284 counter->queryuser = query->user;
285 counter->user = userdata;
287 list_add_tail( &query->counters, &counter->entry );
288 *hcounter = counter;
290 LeaveCriticalSection( &pdh_handle_cs );
291 return ERROR_SUCCESS;
293 LeaveCriticalSection( &pdh_handle_cs );
294 return PDH_MEMORY_ALLOCATION_FAILURE;
297 LeaveCriticalSection( &pdh_handle_cs );
298 return PDH_CSTATUS_NO_COUNTER;
301 /***********************************************************************
302 * PdhAddEnglishCounterA (PDH.@)
304 PDH_STATUS WINAPI PdhAddEnglishCounterA( PDH_HQUERY query, LPCSTR path,
305 DWORD_PTR userdata, PDH_HCOUNTER *counter )
307 TRACE("%p %s %lx %p\n", query, debugstr_a(path), userdata, counter);
309 if (!query) return PDH_INVALID_ARGUMENT;
310 return PdhAddCounterA( query, path, userdata, counter );
313 /***********************************************************************
314 * PdhAddEnglishCounterW (PDH.@)
316 PDH_STATUS WINAPI PdhAddEnglishCounterW( PDH_HQUERY query, LPCWSTR path,
317 DWORD_PTR userdata, PDH_HCOUNTER *counter )
319 TRACE("%p %s %lx %p\n", query, debugstr_w(path), userdata, counter);
321 if (!query) return PDH_INVALID_ARGUMENT;
322 return PdhAddCounterW( query, path, userdata, counter );
325 /* caller must hold counter lock */
326 static PDH_STATUS format_value( struct counter *counter, DWORD format, union value *raw1,
327 union value *raw2, PDH_FMT_COUNTERVALUE *value )
329 LONG factor;
331 factor = counter->scale ? counter->scale : counter->defaultscale;
332 if (format & PDH_FMT_LONG)
334 if (format & PDH_FMT_1000) value->u.longValue = raw2->longvalue * 1000;
335 else value->u.longValue = raw2->longvalue * pow( 10, factor );
337 else if (format & PDH_FMT_LARGE)
339 if (format & PDH_FMT_1000) value->u.largeValue = raw2->largevalue * 1000;
340 else value->u.largeValue = raw2->largevalue * pow( 10, factor );
342 else if (format & PDH_FMT_DOUBLE)
344 if (format & PDH_FMT_1000) value->u.doubleValue = raw2->doublevalue * 1000;
345 else value->u.doubleValue = raw2->doublevalue * pow( 10, factor );
347 else
349 WARN("unknown format %x\n", format);
350 return PDH_INVALID_ARGUMENT;
352 return ERROR_SUCCESS;
355 /***********************************************************************
356 * PdhCalculateCounterFromRawValue (PDH.@)
358 PDH_STATUS WINAPI PdhCalculateCounterFromRawValue( PDH_HCOUNTER handle, DWORD format,
359 PPDH_RAW_COUNTER raw1, PPDH_RAW_COUNTER raw2,
360 PPDH_FMT_COUNTERVALUE value )
362 PDH_STATUS ret;
363 struct counter *counter = handle;
365 TRACE("%p 0x%08x %p %p %p\n", handle, format, raw1, raw2, value);
367 if (!value) return PDH_INVALID_ARGUMENT;
369 EnterCriticalSection( &pdh_handle_cs );
370 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
372 LeaveCriticalSection( &pdh_handle_cs );
373 return PDH_INVALID_HANDLE;
376 ret = format_value( counter, format, (union value *)&raw1->SecondValue,
377 (union value *)&raw2->SecondValue, value );
379 LeaveCriticalSection( &pdh_handle_cs );
380 return ret;
384 /***********************************************************************
385 * PdhCloseQuery (PDH.@)
387 PDH_STATUS WINAPI PdhCloseQuery( PDH_HQUERY handle )
389 struct query *query = handle;
390 struct list *item, *next;
392 TRACE("%p\n", handle);
394 EnterCriticalSection( &pdh_handle_cs );
395 if (!query || query->magic != PDH_MAGIC_QUERY)
397 LeaveCriticalSection( &pdh_handle_cs );
398 return PDH_INVALID_HANDLE;
401 if (query->thread)
403 HANDLE thread = query->thread;
404 SetEvent( query->stop );
405 LeaveCriticalSection( &pdh_handle_cs );
407 WaitForSingleObject( thread, INFINITE );
409 EnterCriticalSection( &pdh_handle_cs );
410 if (query->magic != PDH_MAGIC_QUERY)
412 LeaveCriticalSection( &pdh_handle_cs );
413 return ERROR_SUCCESS;
415 CloseHandle( query->stop );
416 CloseHandle( query->thread );
417 query->thread = NULL;
420 LIST_FOR_EACH_SAFE( item, next, &query->counters )
422 struct counter *counter = LIST_ENTRY( item, struct counter, entry );
424 list_remove( &counter->entry );
425 destroy_counter( counter );
428 destroy_query( query );
430 LeaveCriticalSection( &pdh_handle_cs );
431 return ERROR_SUCCESS;
434 /* caller must hold query lock */
435 static void collect_query_data( struct query *query )
437 struct list *item;
439 LIST_FOR_EACH( item, &query->counters )
441 SYSTEMTIME time;
442 struct counter *counter = LIST_ENTRY( item, struct counter, entry );
444 counter->collect( counter );
446 GetLocalTime( &time );
447 SystemTimeToFileTime( &time, &counter->stamp );
451 /***********************************************************************
452 * PdhCollectQueryData (PDH.@)
454 PDH_STATUS WINAPI PdhCollectQueryData( PDH_HQUERY handle )
456 struct query *query = handle;
458 TRACE("%p\n", handle);
460 EnterCriticalSection( &pdh_handle_cs );
461 if (!query || query->magic != PDH_MAGIC_QUERY)
463 LeaveCriticalSection( &pdh_handle_cs );
464 return PDH_INVALID_HANDLE;
467 if (list_empty( &query->counters ))
469 LeaveCriticalSection( &pdh_handle_cs );
470 return PDH_NO_DATA;
473 collect_query_data( query );
475 LeaveCriticalSection( &pdh_handle_cs );
476 return ERROR_SUCCESS;
479 static DWORD CALLBACK collect_query_thread( void *arg )
481 struct query *query = arg;
482 DWORD interval = query->interval;
483 HANDLE stop = query->stop;
485 for (;;)
487 if (WaitForSingleObject( stop, interval ) != WAIT_TIMEOUT) ExitThread( 0 );
489 EnterCriticalSection( &pdh_handle_cs );
490 if (query->magic != PDH_MAGIC_QUERY)
492 LeaveCriticalSection( &pdh_handle_cs );
493 ExitThread( PDH_INVALID_HANDLE );
496 collect_query_data( query );
498 if (!SetEvent( query->wait ))
500 LeaveCriticalSection( &pdh_handle_cs );
501 ExitThread( 0 );
503 LeaveCriticalSection( &pdh_handle_cs );
507 /***********************************************************************
508 * PdhCollectQueryDataEx (PDH.@)
510 PDH_STATUS WINAPI PdhCollectQueryDataEx( PDH_HQUERY handle, DWORD interval, HANDLE event )
512 PDH_STATUS ret;
513 struct query *query = handle;
515 TRACE("%p %d %p\n", handle, interval, event);
517 EnterCriticalSection( &pdh_handle_cs );
518 if (!query || query->magic != PDH_MAGIC_QUERY)
520 LeaveCriticalSection( &pdh_handle_cs );
521 return PDH_INVALID_HANDLE;
523 if (list_empty( &query->counters ))
525 LeaveCriticalSection( &pdh_handle_cs );
526 return PDH_NO_DATA;
528 if (query->thread)
530 HANDLE thread = query->thread;
531 SetEvent( query->stop );
532 LeaveCriticalSection( &pdh_handle_cs );
534 WaitForSingleObject( thread, INFINITE );
536 EnterCriticalSection( &pdh_handle_cs );
537 if (query->magic != PDH_MAGIC_QUERY)
539 LeaveCriticalSection( &pdh_handle_cs );
540 return PDH_INVALID_HANDLE;
542 CloseHandle( query->thread );
543 query->thread = NULL;
545 else if (!(query->stop = CreateEventW( NULL, FALSE, FALSE, NULL )))
547 ret = GetLastError();
548 LeaveCriticalSection( &pdh_handle_cs );
549 return ret;
551 query->wait = event;
552 query->interval = interval * 1000;
553 if (!(query->thread = CreateThread( NULL, 0, collect_query_thread, query, 0, NULL )))
555 ret = GetLastError();
556 CloseHandle( query->stop );
558 LeaveCriticalSection( &pdh_handle_cs );
559 return ret;
562 LeaveCriticalSection( &pdh_handle_cs );
563 return ERROR_SUCCESS;
566 /***********************************************************************
567 * PdhCollectQueryDataWithTime (PDH.@)
569 PDH_STATUS WINAPI PdhCollectQueryDataWithTime( PDH_HQUERY handle, LONGLONG *timestamp )
571 struct query *query = handle;
572 struct counter *counter;
573 struct list *item;
575 TRACE("%p %p\n", handle, timestamp);
577 if (!timestamp) return PDH_INVALID_ARGUMENT;
579 EnterCriticalSection( &pdh_handle_cs );
580 if (!query || query->magic != PDH_MAGIC_QUERY)
582 LeaveCriticalSection( &pdh_handle_cs );
583 return PDH_INVALID_HANDLE;
585 if (list_empty( &query->counters ))
587 LeaveCriticalSection( &pdh_handle_cs );
588 return PDH_NO_DATA;
591 collect_query_data( query );
593 item = list_head( &query->counters );
594 counter = LIST_ENTRY( item, struct counter, entry );
596 *timestamp = ((LONGLONG)counter->stamp.dwHighDateTime << 32) | counter->stamp.dwLowDateTime;
598 LeaveCriticalSection( &pdh_handle_cs );
599 return ERROR_SUCCESS;
602 /***********************************************************************
603 * PdhExpandWildCardPathA (PDH.@)
605 PDH_STATUS WINAPI PdhExpandWildCardPathA( LPCSTR szDataSource, LPCSTR szWildCardPath, LPSTR mszExpandedPathList, LPDWORD pcchPathListLength, DWORD dwFlags )
607 FIXME("%s, %s, %p, %p, 0x%x: stub\n", debugstr_a(szDataSource), debugstr_a(szWildCardPath), mszExpandedPathList, pcchPathListLength, dwFlags);
608 return PDH_NOT_IMPLEMENTED;
611 /***********************************************************************
612 * PdhExpandWildCardPathW (PDH.@)
614 PDH_STATUS WINAPI PdhExpandWildCardPathW( LPCWSTR szDataSource, LPCWSTR szWildCardPath, LPWSTR mszExpandedPathList, LPDWORD pcchPathListLength, DWORD dwFlags )
616 FIXME("%s, %s, %p, %p, 0x%x: stub\n", debugstr_w(szDataSource), debugstr_w(szWildCardPath), mszExpandedPathList, pcchPathListLength, dwFlags);
617 return PDH_NOT_IMPLEMENTED;
620 /***********************************************************************
621 * PdhGetCounterInfoA (PDH.@)
623 PDH_STATUS WINAPI PdhGetCounterInfoA( PDH_HCOUNTER handle, BOOLEAN text, LPDWORD size, PPDH_COUNTER_INFO_A info )
625 struct counter *counter = handle;
627 TRACE("%p %d %p %p\n", handle, text, size, info);
629 EnterCriticalSection( &pdh_handle_cs );
630 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
632 LeaveCriticalSection( &pdh_handle_cs );
633 return PDH_INVALID_HANDLE;
635 if (!size)
637 LeaveCriticalSection( &pdh_handle_cs );
638 return PDH_INVALID_ARGUMENT;
640 if (*size < sizeof(PDH_COUNTER_INFO_A))
642 *size = sizeof(PDH_COUNTER_INFO_A);
643 LeaveCriticalSection( &pdh_handle_cs );
644 return PDH_MORE_DATA;
647 memset( info, 0, sizeof(PDH_COUNTER_INFO_A) );
649 info->dwType = counter->type;
650 info->CStatus = counter->status;
651 info->lScale = counter->scale;
652 info->lDefaultScale = counter->defaultscale;
653 info->dwUserData = counter->user;
654 info->dwQueryUserData = counter->queryuser;
656 *size = sizeof(PDH_COUNTER_INFO_A);
658 LeaveCriticalSection( &pdh_handle_cs );
659 return ERROR_SUCCESS;
662 /***********************************************************************
663 * PdhGetCounterInfoW (PDH.@)
665 PDH_STATUS WINAPI PdhGetCounterInfoW( PDH_HCOUNTER handle, BOOLEAN text, LPDWORD size, PPDH_COUNTER_INFO_W info )
667 struct counter *counter = handle;
669 TRACE("%p %d %p %p\n", handle, text, size, info);
671 EnterCriticalSection( &pdh_handle_cs );
672 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
674 LeaveCriticalSection( &pdh_handle_cs );
675 return PDH_INVALID_HANDLE;
677 if (!size)
679 LeaveCriticalSection( &pdh_handle_cs );
680 return PDH_INVALID_ARGUMENT;
682 if (*size < sizeof(PDH_COUNTER_INFO_W))
684 *size = sizeof(PDH_COUNTER_INFO_W);
685 LeaveCriticalSection( &pdh_handle_cs );
686 return PDH_MORE_DATA;
689 memset( info, 0, sizeof(PDH_COUNTER_INFO_W) );
691 info->dwType = counter->type;
692 info->CStatus = counter->status;
693 info->lScale = counter->scale;
694 info->lDefaultScale = counter->defaultscale;
695 info->dwUserData = counter->user;
696 info->dwQueryUserData = counter->queryuser;
698 *size = sizeof(PDH_COUNTER_INFO_W);
700 LeaveCriticalSection( &pdh_handle_cs );
701 return ERROR_SUCCESS;
704 /***********************************************************************
705 * PdhGetCounterTimeBase (PDH.@)
707 PDH_STATUS WINAPI PdhGetCounterTimeBase( PDH_HCOUNTER handle, LONGLONG *base )
709 struct counter *counter = handle;
711 TRACE("%p %p\n", handle, base);
713 if (!base) return PDH_INVALID_ARGUMENT;
715 EnterCriticalSection( &pdh_handle_cs );
716 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
718 LeaveCriticalSection( &pdh_handle_cs );
719 return PDH_INVALID_HANDLE;
722 *base = counter->base;
724 LeaveCriticalSection( &pdh_handle_cs );
725 return ERROR_SUCCESS;
728 /***********************************************************************
729 * PdhGetDllVersion (PDH.@)
731 PDH_STATUS WINAPI PdhGetDllVersion( LPDWORD version )
733 if (!version)
734 return PDH_INVALID_ARGUMENT;
736 *version = PDH_VERSION;
738 return ERROR_SUCCESS;
741 /***********************************************************************
742 * PdhGetFormattedCounterValue (PDH.@)
744 PDH_STATUS WINAPI PdhGetFormattedCounterValue( PDH_HCOUNTER handle, DWORD format,
745 LPDWORD type, PPDH_FMT_COUNTERVALUE value )
747 PDH_STATUS ret;
748 struct counter *counter = handle;
750 TRACE("%p %x %p %p\n", handle, format, type, value);
752 if (!value) return PDH_INVALID_ARGUMENT;
754 EnterCriticalSection( &pdh_handle_cs );
755 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
757 LeaveCriticalSection( &pdh_handle_cs );
758 return PDH_INVALID_HANDLE;
760 if (counter->status)
762 LeaveCriticalSection( &pdh_handle_cs );
763 return PDH_INVALID_DATA;
765 if (!(ret = format_value( counter, format, &counter->one, &counter->two, value )))
767 value->CStatus = ERROR_SUCCESS;
768 if (type) *type = counter->type;
771 LeaveCriticalSection( &pdh_handle_cs );
772 return ret;
775 /***********************************************************************
776 * PdhGetRawCounterValue (PDH.@)
778 PDH_STATUS WINAPI PdhGetRawCounterValue( PDH_HCOUNTER handle, LPDWORD type,
779 PPDH_RAW_COUNTER value )
781 struct counter *counter = handle;
783 TRACE("%p %p %p\n", handle, type, value);
785 if (!value) return PDH_INVALID_ARGUMENT;
787 EnterCriticalSection( &pdh_handle_cs );
788 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
790 LeaveCriticalSection( &pdh_handle_cs );
791 return PDH_INVALID_HANDLE;
794 value->CStatus = counter->status;
795 value->TimeStamp.dwLowDateTime = counter->stamp.dwLowDateTime;
796 value->TimeStamp.dwHighDateTime = counter->stamp.dwHighDateTime;
797 value->FirstValue = counter->one.largevalue;
798 value->SecondValue = counter->two.largevalue;
799 value->MultiCount = 1; /* FIXME */
801 if (type) *type = counter->type;
803 LeaveCriticalSection( &pdh_handle_cs );
804 return ERROR_SUCCESS;
807 /***********************************************************************
808 * PdhLookupPerfIndexByNameA (PDH.@)
810 PDH_STATUS WINAPI PdhLookupPerfIndexByNameA( LPCSTR machine, LPCSTR name, LPDWORD index )
812 PDH_STATUS ret;
813 WCHAR *machineW = NULL;
814 WCHAR *nameW;
816 TRACE("%s %s %p\n", debugstr_a(machine), debugstr_a(name), index);
818 if (!name) return PDH_INVALID_ARGUMENT;
820 if (machine && !(machineW = pdh_strdup_aw( machine ))) return PDH_MEMORY_ALLOCATION_FAILURE;
822 if (!(nameW = pdh_strdup_aw( name )))
823 return PDH_MEMORY_ALLOCATION_FAILURE;
825 ret = PdhLookupPerfIndexByNameW( machineW, nameW, index );
827 heap_free( nameW );
828 heap_free( machineW );
829 return ret;
832 /***********************************************************************
833 * PdhLookupPerfIndexByNameW (PDH.@)
835 PDH_STATUS WINAPI PdhLookupPerfIndexByNameW( LPCWSTR machine, LPCWSTR name, LPDWORD index )
837 unsigned int i;
839 TRACE("%s %s %p\n", debugstr_w(machine), debugstr_w(name), index);
841 if (!name || !index) return PDH_INVALID_ARGUMENT;
843 if (machine)
845 FIXME("remote machine not supported\n");
846 return PDH_CSTATUS_NO_MACHINE;
848 for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
850 if (pdh_match_path( counter_sources[i].path, name ))
852 *index = counter_sources[i].index;
853 return ERROR_SUCCESS;
856 return PDH_STRING_NOT_FOUND;
859 /***********************************************************************
860 * PdhLookupPerfNameByIndexA (PDH.@)
862 PDH_STATUS WINAPI PdhLookupPerfNameByIndexA( LPCSTR machine, DWORD index, LPSTR buffer, LPDWORD size )
864 PDH_STATUS ret;
865 WCHAR *machineW = NULL;
866 WCHAR bufferW[PDH_MAX_COUNTER_NAME];
867 DWORD sizeW = sizeof(bufferW) / sizeof(WCHAR);
869 TRACE("%s %d %p %p\n", debugstr_a(machine), index, buffer, size);
871 if (!buffer || !size) return PDH_INVALID_ARGUMENT;
873 if (machine && !(machineW = pdh_strdup_aw( machine ))) return PDH_MEMORY_ALLOCATION_FAILURE;
875 if (!(ret = PdhLookupPerfNameByIndexW( machineW, index, bufferW, &sizeW )))
877 int required = WideCharToMultiByte( CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL );
879 if (size && *size < required) ret = PDH_MORE_DATA;
880 else WideCharToMultiByte( CP_ACP, 0, bufferW, -1, buffer, required, NULL, NULL );
881 if (size) *size = required;
883 heap_free( machineW );
884 return ret;
887 /***********************************************************************
888 * PdhLookupPerfNameByIndexW (PDH.@)
890 PDH_STATUS WINAPI PdhLookupPerfNameByIndexW( LPCWSTR machine, DWORD index, LPWSTR buffer, LPDWORD size )
892 PDH_STATUS ret;
893 unsigned int i;
895 TRACE("%s %d %p %p\n", debugstr_w(machine), index, buffer, size);
897 if (machine)
899 FIXME("remote machine not supported\n");
900 return PDH_CSTATUS_NO_MACHINE;
903 if (!buffer || !size) return PDH_INVALID_ARGUMENT;
904 if (!index) return ERROR_SUCCESS;
906 for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
908 if (counter_sources[i].index == index)
910 WCHAR *p = strrchrW( counter_sources[i].path, '\\' ) + 1;
911 unsigned int required = strlenW( p ) + 1;
913 if (*size < required) ret = PDH_MORE_DATA;
914 else
916 strcpyW( buffer, p );
917 ret = ERROR_SUCCESS;
919 *size = required;
920 return ret;
923 return PDH_INVALID_ARGUMENT;
926 /***********************************************************************
927 * PdhOpenQueryA (PDH.@)
929 PDH_STATUS WINAPI PdhOpenQueryA( LPCSTR source, DWORD_PTR userdata, PDH_HQUERY *query )
931 PDH_STATUS ret;
932 WCHAR *sourceW = NULL;
934 TRACE("%s %lx %p\n", debugstr_a(source), userdata, query);
936 if (source && !(sourceW = pdh_strdup_aw( source ))) return PDH_MEMORY_ALLOCATION_FAILURE;
938 ret = PdhOpenQueryW( sourceW, userdata, query );
939 heap_free( sourceW );
941 return ret;
944 /***********************************************************************
945 * PdhOpenQueryW (PDH.@)
947 PDH_STATUS WINAPI PdhOpenQueryW( LPCWSTR source, DWORD_PTR userdata, PDH_HQUERY *handle )
949 struct query *query;
951 TRACE("%s %lx %p\n", debugstr_w(source), userdata, handle);
953 if (!handle) return PDH_INVALID_ARGUMENT;
955 if (source)
957 FIXME("log file data source not supported\n");
958 return PDH_INVALID_ARGUMENT;
960 if ((query = create_query()))
962 query->user = userdata;
963 *handle = query;
965 return ERROR_SUCCESS;
967 return PDH_MEMORY_ALLOCATION_FAILURE;
970 /***********************************************************************
971 * PdhRemoveCounter (PDH.@)
973 PDH_STATUS WINAPI PdhRemoveCounter( PDH_HCOUNTER handle )
975 struct counter *counter = handle;
977 TRACE("%p\n", handle);
979 EnterCriticalSection( &pdh_handle_cs );
980 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
982 LeaveCriticalSection( &pdh_handle_cs );
983 return PDH_INVALID_HANDLE;
986 list_remove( &counter->entry );
987 destroy_counter( counter );
989 LeaveCriticalSection( &pdh_handle_cs );
990 return ERROR_SUCCESS;
993 /***********************************************************************
994 * PdhSetCounterScaleFactor (PDH.@)
996 PDH_STATUS WINAPI PdhSetCounterScaleFactor( PDH_HCOUNTER handle, LONG factor )
998 struct counter *counter = handle;
1000 TRACE("%p\n", handle);
1002 EnterCriticalSection( &pdh_handle_cs );
1003 if (!counter || counter->magic != PDH_MAGIC_COUNTER)
1005 LeaveCriticalSection( &pdh_handle_cs );
1006 return PDH_INVALID_HANDLE;
1008 if (factor < PDH_MIN_SCALE || factor > PDH_MAX_SCALE)
1010 LeaveCriticalSection( &pdh_handle_cs );
1011 return PDH_INVALID_ARGUMENT;
1014 counter->scale = factor;
1016 LeaveCriticalSection( &pdh_handle_cs );
1017 return ERROR_SUCCESS;
1020 /***********************************************************************
1021 * PdhValidatePathA (PDH.@)
1023 PDH_STATUS WINAPI PdhValidatePathA( LPCSTR path )
1025 PDH_STATUS ret;
1026 WCHAR *pathW;
1028 TRACE("%s\n", debugstr_a(path));
1030 if (!path) return PDH_INVALID_ARGUMENT;
1031 if (!(pathW = pdh_strdup_aw( path ))) return PDH_MEMORY_ALLOCATION_FAILURE;
1033 ret = PdhValidatePathW( pathW );
1035 heap_free( pathW );
1036 return ret;
1039 static PDH_STATUS validate_path( LPCWSTR path )
1041 if (!path || !*path) return PDH_INVALID_ARGUMENT;
1042 if (*path++ != '\\' || !strchrW( path, '\\' )) return PDH_CSTATUS_BAD_COUNTERNAME;
1043 return ERROR_SUCCESS;
1046 /***********************************************************************
1047 * PdhValidatePathW (PDH.@)
1049 PDH_STATUS WINAPI PdhValidatePathW( LPCWSTR path )
1051 PDH_STATUS ret;
1052 unsigned int i;
1054 TRACE("%s\n", debugstr_w(path));
1056 if ((ret = validate_path( path ))) return ret;
1058 for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
1059 if (pdh_match_path( counter_sources[i].path, path )) return ERROR_SUCCESS;
1061 return PDH_CSTATUS_NO_COUNTER;
1064 /***********************************************************************
1065 * PdhValidatePathExA (PDH.@)
1067 PDH_STATUS WINAPI PdhValidatePathExA( PDH_HLOG source, LPCSTR path )
1069 TRACE("%p %s\n", source, debugstr_a(path));
1071 if (source)
1073 FIXME("log file data source not supported\n");
1074 return ERROR_SUCCESS;
1076 return PdhValidatePathA( path );
1079 /***********************************************************************
1080 * PdhValidatePathExW (PDH.@)
1082 PDH_STATUS WINAPI PdhValidatePathExW( PDH_HLOG source, LPCWSTR path )
1084 TRACE("%p %s\n", source, debugstr_w(path));
1086 if (source)
1088 FIXME("log file data source not supported\n");
1089 return ERROR_SUCCESS;
1091 return PdhValidatePathW( path );
1094 /***********************************************************************
1095 * PdhMakeCounterPathA (PDH.@)
1097 PDH_STATUS WINAPI PdhMakeCounterPathA( PDH_COUNTER_PATH_ELEMENTS_A *e, LPSTR buffer,
1098 LPDWORD buflen, DWORD flags )
1100 PDH_STATUS ret = PDH_MEMORY_ALLOCATION_FAILURE;
1101 PDH_COUNTER_PATH_ELEMENTS_W eW;
1102 WCHAR *bufferW;
1103 DWORD buflenW;
1105 TRACE("%p %p %p 0x%08x\n", e, buffer, buflen, flags);
1107 if (!e || !buflen) return PDH_INVALID_ARGUMENT;
1109 memset( &eW, 0, sizeof(eW) );
1110 if (e->szMachineName && !(eW.szMachineName = pdh_strdup_aw( e->szMachineName ))) goto done;
1111 if (e->szObjectName && !(eW.szObjectName = pdh_strdup_aw( e->szObjectName ))) goto done;
1112 if (e->szInstanceName && !(eW.szInstanceName = pdh_strdup_aw( e->szInstanceName ))) goto done;
1113 if (e->szParentInstance && !(eW.szParentInstance = pdh_strdup_aw( e->szParentInstance ))) goto done;
1114 if (e->szCounterName && !(eW.szCounterName = pdh_strdup_aw( e->szCounterName ))) goto done;
1115 eW.dwInstanceIndex = e->dwInstanceIndex;
1117 buflenW = 0;
1118 ret = PdhMakeCounterPathW( &eW, NULL, &buflenW, flags );
1119 if (ret == PDH_MORE_DATA)
1121 if ((bufferW = heap_alloc( buflenW * sizeof(WCHAR) )))
1123 if (!(ret = PdhMakeCounterPathW( &eW, bufferW, &buflenW, flags )))
1125 int len = WideCharToMultiByte(CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL);
1126 if (*buflen >= len) WideCharToMultiByte(CP_ACP, 0, bufferW, -1, buffer, *buflen, NULL, NULL);
1127 else ret = PDH_MORE_DATA;
1128 *buflen = len;
1130 heap_free( bufferW );
1132 else
1133 ret = PDH_MEMORY_ALLOCATION_FAILURE;
1136 done:
1137 heap_free( eW.szMachineName );
1138 heap_free( eW.szObjectName );
1139 heap_free( eW.szInstanceName );
1140 heap_free( eW.szParentInstance );
1141 heap_free( eW.szCounterName );
1142 return ret;
1145 /***********************************************************************
1146 * PdhMakeCounterPathW (PDH.@)
1148 PDH_STATUS WINAPI PdhMakeCounterPathW( PDH_COUNTER_PATH_ELEMENTS_W *e, LPWSTR buffer,
1149 LPDWORD buflen, DWORD flags )
1151 static const WCHAR bslash[] = {'\\',0};
1152 static const WCHAR fslash[] = {'/',0};
1153 static const WCHAR lparen[] = {'(',0};
1154 static const WCHAR rparen[] = {')',0};
1155 static const WCHAR fmt[] = {'#','%','u',0};
1157 WCHAR path[PDH_MAX_COUNTER_NAME], instance[12];
1158 PDH_STATUS ret = ERROR_SUCCESS;
1159 DWORD len;
1161 TRACE("%p %p %p 0x%08x\n", e, buffer, buflen, flags);
1163 if (flags) FIXME("unimplemented flags 0x%08x\n", flags);
1165 if (!e || !e->szCounterName || !e->szObjectName || !buflen)
1166 return PDH_INVALID_ARGUMENT;
1168 path[0] = 0;
1169 if (e->szMachineName)
1171 strcatW(path, bslash);
1172 strcatW(path, bslash);
1173 strcatW(path, e->szMachineName);
1175 strcatW(path, bslash);
1176 strcatW(path, e->szObjectName);
1177 if (e->szInstanceName)
1179 strcatW(path, lparen);
1180 if (e->szParentInstance)
1182 strcatW(path, e->szParentInstance);
1183 strcatW(path, fslash);
1185 strcatW(path, e->szInstanceName);
1186 sprintfW(instance, fmt, e->dwInstanceIndex);
1187 strcatW(path, instance);
1188 strcatW(path, rparen);
1190 strcatW(path, bslash);
1191 strcatW(path, e->szCounterName);
1193 len = strlenW(path) + 1;
1194 if (*buflen >= len) strcpyW(buffer, path);
1195 else ret = PDH_MORE_DATA;
1196 *buflen = len;
1197 return ret;
1200 /***********************************************************************
1201 * PdhEnumObjectItemsA (PDH.@)
1203 PDH_STATUS WINAPI PdhEnumObjectItemsA(LPCSTR szDataSource, LPCSTR szMachineName, LPCSTR szObjectName,
1204 LPSTR mszCounterList, LPDWORD pcchCounterListLength, LPSTR mszInstanceList,
1205 LPDWORD pcchInstanceListLength, DWORD dwDetailLevel, DWORD dwFlags)
1207 FIXME("%s, %s, %s, %p, %p, %p, %p, %d, 0x%x: stub\n", debugstr_a(szDataSource), debugstr_a(szMachineName),
1208 debugstr_a(szObjectName), mszCounterList, pcchCounterListLength, mszInstanceList,
1209 pcchInstanceListLength, dwDetailLevel, dwFlags);
1211 return PDH_NOT_IMPLEMENTED;
1214 /***********************************************************************
1215 * PdhEnumObjectItemsW (PDH.@)
1217 PDH_STATUS WINAPI PdhEnumObjectItemsW(LPCWSTR szDataSource, LPCWSTR szMachineName, LPCWSTR szObjectName,
1218 LPWSTR mszCounterList, LPDWORD pcchCounterListLength, LPWSTR mszInstanceList,
1219 LPDWORD pcchInstanceListLength, DWORD dwDetailLevel, DWORD dwFlags)
1221 FIXME("%s, %s, %s, %p, %p, %p, %p, %d, 0x%x: stub\n", debugstr_w(szDataSource), debugstr_w(szMachineName),
1222 debugstr_w(szObjectName), mszCounterList, pcchCounterListLength, mszInstanceList,
1223 pcchInstanceListLength, dwDetailLevel, dwFlags);
1225 return PDH_NOT_IMPLEMENTED;
1228 /***********************************************************************
1229 * PdhSetDefaultRealTimeDataSource (PDH.@)
1231 PDH_STATUS WINAPI PdhSetDefaultRealTimeDataSource( DWORD source )
1233 FIXME("%u\n", source);
1234 return ERROR_SUCCESS;