rsaenh: Split CPImportKey into a helper function for each type of key supported.
[wine] / dlls / pdh / pdh_main.c
1 /*
2  * Performance Data Helper (pdh.dll)
3  *
4  * Copyright 2007 Andrey Turkin
5  * Copyright 2007 Hans Leidekker
6  *
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.
11  *
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.
16  *
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
20  */
21
22 #include <stdarg.h>
23 #include <math.h>
24
25 #define NONAMELESSUNION
26 #define NONAMELESSSTRUCT
27 #include "windef.h"
28 #include "winbase.h"
29
30 #include "pdh.h"
31 #include "pdhmsg.h"
32 #include "winperf.h"
33
34 #include "wine/debug.h"
35 #include "wine/list.h"
36 #include "wine/unicode.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(pdh);
39
40 static CRITICAL_SECTION pdh_handle_cs;
41 static CRITICAL_SECTION_DEBUG pdh_handle_cs_debug =
42 {
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") }
47 };
48 static CRITICAL_SECTION pdh_handle_cs = { &pdh_handle_cs_debug, -1, 0, 0, 0, 0 };
49
50 static inline void *heap_alloc( SIZE_T size )
51 {
52     return HeapAlloc( GetProcessHeap(), 0, size );
53 }
54
55 static inline void *heap_alloc_zero( SIZE_T size )
56 {
57     return HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size );
58 }
59
60 static inline void heap_free( LPVOID mem )
61 {
62     HeapFree( GetProcessHeap(), 0, mem );
63 }
64
65 static inline WCHAR *pdh_strdup( const WCHAR *src )
66 {
67     WCHAR *dst;
68
69     if (!src) return NULL;
70     if ((dst = heap_alloc( (strlenW( src ) + 1) * sizeof(WCHAR) ))) strcpyW( dst, src );
71     return dst;
72 }
73
74 static inline WCHAR *pdh_strdup_aw( const char *src )
75 {
76     int len;
77     WCHAR *dst;
78
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;
83 }
84
85 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
86 {
87     TRACE("(0x%p, %d, %p)\n",hinstDLL,fdwReason,lpvReserved);
88
89     if (fdwReason == DLL_WINE_PREATTACH) return FALSE;    /* prefer native version */
90
91     if (fdwReason == DLL_PROCESS_ATTACH)
92     {
93         DisableThreadLibraryCalls( hinstDLL );
94     }
95
96     return TRUE;
97 }
98
99 union value
100 {
101     LONG     longvalue;
102     double   doublevalue;
103     LONGLONG largevalue;
104 };
105
106 struct counter
107 {
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 */
122 };
123
124 #define PDH_MAGIC_COUNTER   0x50444831 /* 'PDH1' */
125
126 static struct counter *create_counter( void )
127 {
128     struct counter *counter;
129
130     if ((counter = heap_alloc_zero( sizeof(struct counter) )))
131     {
132         counter->magic = PDH_MAGIC_COUNTER;
133         return counter;
134     }
135     return NULL;
136 }
137
138 static void destroy_counter( struct counter *counter )
139 {
140     counter->magic = 0;
141     heap_free( counter->path );
142     heap_free( counter );
143 }
144
145 #define PDH_MAGIC_QUERY     0x50444830 /* 'PDH0' */
146
147 struct query
148 {
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 */
156 };
157
158 static struct query *create_query( void )
159 {
160     struct query *query;
161
162     if ((query = heap_alloc_zero( sizeof(struct query) )))
163     {
164         query->magic = PDH_MAGIC_QUERY;
165         list_init( &query->counters );
166         return query;
167     }
168     return NULL;
169 }
170
171 static void destroy_query( struct query *query )
172 {
173     query->magic = 0;
174     heap_free( query );
175 }
176
177 struct source
178 {
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 */
185 };
186
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};
192
193 static void CALLBACK collect_processor_time( struct counter *counter )
194 {
195     counter->two.largevalue = 500000; /* FIXME */
196     counter->status = PDH_CSTATUS_VALID_DATA;
197 }
198
199 static void CALLBACK collect_uptime( struct counter *counter )
200 {
201     counter->two.largevalue = GetTickCount64();
202     counter->status = PDH_CSTATUS_VALID_DATA;
203 }
204
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)
208
209 #define TYPE_UPTIME \
210     (PERF_SIZE_LARGE | PERF_TYPE_COUNTER | PERF_COUNTER_ELAPSED | PERF_OBJECT_TIMER | PERF_DISPLAY_SECONDS)
211
212 /* counter source registry */
213 static const struct source counter_sources[] =
214 {
215     { 6,    path_processor_time,    collect_processor_time,     TYPE_PROCESSOR_TIME,    -5,     10000000 },
216     { 674,  path_uptime,            collect_uptime,             TYPE_UPTIME,            -3,     1000 }
217 };
218
219 static BOOL pdh_match_path( LPCWSTR fullpath, LPCWSTR path )
220 {
221     const WCHAR *p;
222
223     if (strchrW( path, '\\')) p = fullpath;
224     else p = strrchrW( fullpath, '\\' ) + 1;
225     if (strcmpW( p, path )) return FALSE;
226     return TRUE;
227 }
228
229 /***********************************************************************
230  *              PdhAddCounterA   (PDH.@)
231  */
232 PDH_STATUS WINAPI PdhAddCounterA( PDH_HQUERY query, LPCSTR path,
233                                   DWORD_PTR userdata, PDH_HCOUNTER *counter )
234 {
235     PDH_STATUS ret;
236     WCHAR *pathW;
237
238     TRACE("%p %s %lx %p\n", query, debugstr_a(path), userdata, counter);
239
240     if (!path) return PDH_INVALID_ARGUMENT;
241
242     if (!(pathW = pdh_strdup_aw( path )))
243         return PDH_MEMORY_ALLOCATION_FAILURE;
244
245     ret = PdhAddCounterW( query, pathW, userdata, counter );
246
247     heap_free( pathW );
248     return ret;
249 }
250
251 /***********************************************************************
252  *              PdhAddCounterW   (PDH.@)
253  */
254 PDH_STATUS WINAPI PdhAddCounterW( PDH_HQUERY hquery, LPCWSTR path,
255                                   DWORD_PTR userdata, PDH_HCOUNTER *hcounter )
256 {
257     struct query *query = hquery;
258     struct counter *counter;
259     unsigned int i;
260
261     TRACE("%p %s %lx %p\n", hquery, debugstr_w(path), userdata, hcounter);
262
263     if (!path  || !hcounter) return PDH_INVALID_ARGUMENT;
264
265     EnterCriticalSection( &pdh_handle_cs );
266     if (!query || query->magic != PDH_MAGIC_QUERY)
267     {
268         LeaveCriticalSection( &pdh_handle_cs );
269         return PDH_INVALID_HANDLE;
270     }
271
272     *hcounter = NULL;
273     for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
274     {
275         if (pdh_match_path( counter_sources[i].path, path ))
276         {
277             if ((counter = create_counter()))
278             {
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;
286
287                 list_add_tail( &query->counters, &counter->entry );
288                 *hcounter = counter;
289
290                 LeaveCriticalSection( &pdh_handle_cs );
291                 return ERROR_SUCCESS;
292             }
293             LeaveCriticalSection( &pdh_handle_cs );
294             return PDH_MEMORY_ALLOCATION_FAILURE;
295         }
296     }
297     LeaveCriticalSection( &pdh_handle_cs );
298     return PDH_CSTATUS_NO_COUNTER;
299 }
300
301 /***********************************************************************
302  *              PdhAddEnglishCounterA   (PDH.@)
303  */
304 PDH_STATUS WINAPI PdhAddEnglishCounterA( PDH_HQUERY query, LPCSTR path,
305                                          DWORD_PTR userdata, PDH_HCOUNTER *counter )
306 {
307     TRACE("%p %s %lx %p\n", query, debugstr_a(path), userdata, counter);
308
309     if (!query) return PDH_INVALID_ARGUMENT;
310     return PdhAddCounterA( query, path, userdata, counter );
311 }
312
313 /***********************************************************************
314  *              PdhAddEnglishCounterW   (PDH.@)
315  */
316 PDH_STATUS WINAPI PdhAddEnglishCounterW( PDH_HQUERY query, LPCWSTR path,
317                                          DWORD_PTR userdata, PDH_HCOUNTER *counter )
318 {
319     TRACE("%p %s %lx %p\n", query, debugstr_w(path), userdata, counter);
320
321     if (!query) return PDH_INVALID_ARGUMENT;
322     return PdhAddCounterW( query, path, userdata, counter );
323 }
324
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 )
328 {
329     LONG factor;
330
331     factor = counter->scale ? counter->scale : counter->defaultscale;
332     if (format & PDH_FMT_LONG)
333     {
334         if (format & PDH_FMT_1000) value->u.longValue = raw2->longvalue * 1000;
335         else value->u.longValue = raw2->longvalue * pow( 10, factor );
336     }
337     else if (format & PDH_FMT_LARGE)
338     {
339         if (format & PDH_FMT_1000) value->u.largeValue = raw2->largevalue * 1000;
340         else value->u.largeValue = raw2->largevalue * pow( 10, factor );
341     }
342     else if (format & PDH_FMT_DOUBLE)
343     {
344         if (format & PDH_FMT_1000) value->u.doubleValue = raw2->doublevalue * 1000;
345         else value->u.doubleValue = raw2->doublevalue * pow( 10, factor );
346     }
347     else
348     {
349         WARN("unknown format %x\n", format);
350         return PDH_INVALID_ARGUMENT;
351     }
352     return ERROR_SUCCESS;
353 }
354
355 /***********************************************************************
356  *              PdhCalculateCounterFromRawValue   (PDH.@)
357  */
358 PDH_STATUS WINAPI PdhCalculateCounterFromRawValue( PDH_HCOUNTER handle, DWORD format,
359                                                    PPDH_RAW_COUNTER raw1, PPDH_RAW_COUNTER raw2,
360                                                    PPDH_FMT_COUNTERVALUE value )
361 {
362     PDH_STATUS ret;
363     struct counter *counter = handle;
364
365     TRACE("%p 0x%08x %p %p %p\n", handle, format, raw1, raw2, value);
366
367     if (!value) return PDH_INVALID_ARGUMENT;
368
369     EnterCriticalSection( &pdh_handle_cs );
370     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
371     {
372         LeaveCriticalSection( &pdh_handle_cs );
373         return PDH_INVALID_HANDLE;
374     }
375
376     ret = format_value( counter, format, (union value *)&raw1->SecondValue,
377                                          (union value *)&raw2->SecondValue, value );
378
379     LeaveCriticalSection( &pdh_handle_cs );
380     return ret;
381 }
382
383
384 /***********************************************************************
385  *              PdhCloseQuery   (PDH.@)
386  */
387 PDH_STATUS WINAPI PdhCloseQuery( PDH_HQUERY handle )
388 {
389     struct query *query = handle;
390     struct list *item, *next;
391
392     TRACE("%p\n", handle);
393
394     EnterCriticalSection( &pdh_handle_cs );
395     if (!query || query->magic != PDH_MAGIC_QUERY)
396     {
397         LeaveCriticalSection( &pdh_handle_cs );
398         return PDH_INVALID_HANDLE;
399     }
400
401     if (query->thread)
402     {
403         HANDLE thread = query->thread;
404         SetEvent( query->stop );
405         LeaveCriticalSection( &pdh_handle_cs );
406
407         WaitForSingleObject( thread, INFINITE );
408
409         EnterCriticalSection( &pdh_handle_cs );
410         if (query->magic != PDH_MAGIC_QUERY)
411         {
412             LeaveCriticalSection( &pdh_handle_cs );
413             return ERROR_SUCCESS;
414         }
415         CloseHandle( query->stop );
416         CloseHandle( query->thread );
417         query->thread = NULL;
418     }
419
420     LIST_FOR_EACH_SAFE( item, next, &query->counters )
421     {
422         struct counter *counter = LIST_ENTRY( item, struct counter, entry );
423
424         list_remove( &counter->entry );
425         destroy_counter( counter );
426     }
427
428     destroy_query( query );
429
430     LeaveCriticalSection( &pdh_handle_cs );
431     return ERROR_SUCCESS;
432 }
433
434 /* caller must hold query lock */
435 static void collect_query_data( struct query *query )
436 {
437     struct list *item;
438
439     LIST_FOR_EACH( item, &query->counters )
440     {
441         SYSTEMTIME time;
442         struct counter *counter = LIST_ENTRY( item, struct counter, entry );
443
444         counter->collect( counter );
445
446         GetLocalTime( &time );
447         SystemTimeToFileTime( &time, &counter->stamp );
448     }
449 }
450
451 /***********************************************************************
452  *              PdhCollectQueryData   (PDH.@)
453  */
454 PDH_STATUS WINAPI PdhCollectQueryData( PDH_HQUERY handle )
455 {
456     struct query *query = handle;
457
458     TRACE("%p\n", handle);
459
460     EnterCriticalSection( &pdh_handle_cs );
461     if (!query || query->magic != PDH_MAGIC_QUERY)
462     {
463         LeaveCriticalSection( &pdh_handle_cs );
464         return PDH_INVALID_HANDLE;
465     }
466
467     if (list_empty( &query->counters ))
468     {
469         LeaveCriticalSection( &pdh_handle_cs );
470         return PDH_NO_DATA;
471     }
472
473     collect_query_data( query );
474
475     LeaveCriticalSection( &pdh_handle_cs );
476     return ERROR_SUCCESS;
477 }
478
479 static DWORD CALLBACK collect_query_thread( void *arg )
480 {
481     struct query *query = arg;
482     DWORD interval = query->interval;
483     HANDLE stop = query->stop;
484
485     for (;;)
486     {
487         if (WaitForSingleObject( stop, interval ) != WAIT_TIMEOUT) ExitThread( 0 );
488
489         EnterCriticalSection( &pdh_handle_cs );
490         if (query->magic != PDH_MAGIC_QUERY)
491         {
492             LeaveCriticalSection( &pdh_handle_cs );
493             ExitThread( PDH_INVALID_HANDLE );
494         }
495
496         collect_query_data( query );
497
498         if (!SetEvent( query->wait ))
499         {
500             LeaveCriticalSection( &pdh_handle_cs );
501             ExitThread( 0 );
502         }
503         LeaveCriticalSection( &pdh_handle_cs );
504     }
505 }
506
507 /***********************************************************************
508  *              PdhCollectQueryDataEx   (PDH.@)
509  */
510 PDH_STATUS WINAPI PdhCollectQueryDataEx( PDH_HQUERY handle, DWORD interval, HANDLE event )
511 {
512     PDH_STATUS ret;
513     struct query *query = handle;
514
515     TRACE("%p %d %p\n", handle, interval, event);
516
517     EnterCriticalSection( &pdh_handle_cs );
518     if (!query || query->magic != PDH_MAGIC_QUERY)
519     {
520         LeaveCriticalSection( &pdh_handle_cs );
521         return PDH_INVALID_HANDLE;
522     }
523     if (list_empty( &query->counters ))
524     {
525         LeaveCriticalSection( &pdh_handle_cs );
526         return PDH_NO_DATA;
527     }
528     if (query->thread)
529     {
530         HANDLE thread = query->thread;
531         SetEvent( query->stop );
532         LeaveCriticalSection( &pdh_handle_cs );
533
534         WaitForSingleObject( thread, INFINITE );
535
536         EnterCriticalSection( &pdh_handle_cs );
537         if (query->magic != PDH_MAGIC_QUERY)
538         {
539             LeaveCriticalSection( &pdh_handle_cs );
540             return PDH_INVALID_HANDLE;
541         }
542         CloseHandle( query->thread );
543         query->thread = NULL;
544     }
545     else if (!(query->stop = CreateEventW( NULL, FALSE, FALSE, NULL )))
546     {
547         ret = GetLastError();
548         LeaveCriticalSection( &pdh_handle_cs );
549         return ret;
550     }
551     query->wait = event;
552     query->interval = interval * 1000;
553     if (!(query->thread = CreateThread( NULL, 0, collect_query_thread, query, 0, NULL )))
554     {
555         ret = GetLastError();
556         CloseHandle( query->stop );
557
558         LeaveCriticalSection( &pdh_handle_cs );
559         return ret;
560     }
561
562     LeaveCriticalSection( &pdh_handle_cs );
563     return ERROR_SUCCESS;
564 }
565
566 /***********************************************************************
567  *              PdhCollectQueryDataWithTime   (PDH.@)
568  */
569 PDH_STATUS WINAPI PdhCollectQueryDataWithTime( PDH_HQUERY handle, LONGLONG *timestamp )
570 {
571     struct query *query = handle;
572     struct counter *counter;
573     struct list *item;
574
575     TRACE("%p %p\n", handle, timestamp);
576
577     if (!timestamp) return PDH_INVALID_ARGUMENT;
578
579     EnterCriticalSection( &pdh_handle_cs );
580     if (!query || query->magic != PDH_MAGIC_QUERY)
581     {
582         LeaveCriticalSection( &pdh_handle_cs );
583         return PDH_INVALID_HANDLE;
584     }
585     if (list_empty( &query->counters ))
586     {
587         LeaveCriticalSection( &pdh_handle_cs );
588         return PDH_NO_DATA;
589     }
590
591     collect_query_data( query );
592
593     item = list_head( &query->counters );
594     counter = LIST_ENTRY( item, struct counter, entry );
595
596     *timestamp = ((LONGLONG)counter->stamp.dwHighDateTime << 32) | counter->stamp.dwLowDateTime;
597
598     LeaveCriticalSection( &pdh_handle_cs );
599     return ERROR_SUCCESS;
600 }
601
602 /***********************************************************************
603  *              PdhGetCounterInfoA   (PDH.@)
604  */
605 PDH_STATUS WINAPI PdhGetCounterInfoA( PDH_HCOUNTER handle, BOOLEAN text, LPDWORD size, PPDH_COUNTER_INFO_A info )
606 {
607     struct counter *counter = handle;
608
609     TRACE("%p %d %p %p\n", handle, text, size, info);
610
611     EnterCriticalSection( &pdh_handle_cs );
612     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
613     {
614         LeaveCriticalSection( &pdh_handle_cs );
615         return PDH_INVALID_HANDLE;
616     }
617     if (!size)
618     {
619         LeaveCriticalSection( &pdh_handle_cs );
620         return PDH_INVALID_ARGUMENT;
621     }
622     if (*size < sizeof(PDH_COUNTER_INFO_A))
623     {
624         *size = sizeof(PDH_COUNTER_INFO_A);
625         LeaveCriticalSection( &pdh_handle_cs );
626         return PDH_MORE_DATA;
627     }
628
629     memset( info, 0, sizeof(PDH_COUNTER_INFO_A) );
630
631     info->dwType          = counter->type;
632     info->CStatus         = counter->status;
633     info->lScale          = counter->scale;
634     info->lDefaultScale   = counter->defaultscale;
635     info->dwUserData      = counter->user;
636     info->dwQueryUserData = counter->queryuser;
637
638     *size = sizeof(PDH_COUNTER_INFO_A);
639
640     LeaveCriticalSection( &pdh_handle_cs );
641     return ERROR_SUCCESS;
642 }
643
644 /***********************************************************************
645  *              PdhGetCounterInfoW   (PDH.@)
646  */
647 PDH_STATUS WINAPI PdhGetCounterInfoW( PDH_HCOUNTER handle, BOOLEAN text, LPDWORD size, PPDH_COUNTER_INFO_W info )
648 {
649     struct counter *counter = handle;
650
651     TRACE("%p %d %p %p\n", handle, text, size, info);
652
653     EnterCriticalSection( &pdh_handle_cs );
654     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
655     {
656         LeaveCriticalSection( &pdh_handle_cs );
657         return PDH_INVALID_HANDLE;
658     }
659     if (!size)
660     {
661         LeaveCriticalSection( &pdh_handle_cs );
662         return PDH_INVALID_ARGUMENT;
663     }
664     if (*size < sizeof(PDH_COUNTER_INFO_W))
665     {
666         *size = sizeof(PDH_COUNTER_INFO_W);
667         LeaveCriticalSection( &pdh_handle_cs );
668         return PDH_MORE_DATA;
669     }
670
671     memset( info, 0, sizeof(PDH_COUNTER_INFO_W) );
672
673     info->dwType          = counter->type;
674     info->CStatus         = counter->status;
675     info->lScale          = counter->scale;
676     info->lDefaultScale   = counter->defaultscale;
677     info->dwUserData      = counter->user;
678     info->dwQueryUserData = counter->queryuser;
679
680     *size = sizeof(PDH_COUNTER_INFO_W);
681
682     LeaveCriticalSection( &pdh_handle_cs );
683     return ERROR_SUCCESS;
684 }
685
686 /***********************************************************************
687  *              PdhGetCounterTimeBase   (PDH.@)
688  */
689 PDH_STATUS WINAPI PdhGetCounterTimeBase( PDH_HCOUNTER handle, LONGLONG *base )
690 {
691     struct counter *counter = handle;
692
693     TRACE("%p %p\n", handle, base);
694
695     if (!base) return PDH_INVALID_ARGUMENT;
696
697     EnterCriticalSection( &pdh_handle_cs );
698     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
699     {
700         LeaveCriticalSection( &pdh_handle_cs );
701         return PDH_INVALID_HANDLE;
702     }
703
704     *base = counter->base;
705
706     LeaveCriticalSection( &pdh_handle_cs );
707     return ERROR_SUCCESS;
708 }
709
710 /***********************************************************************
711  *              PdhGetFormattedCounterValue   (PDH.@)
712  */
713 PDH_STATUS WINAPI PdhGetFormattedCounterValue( PDH_HCOUNTER handle, DWORD format,
714                                                LPDWORD type, PPDH_FMT_COUNTERVALUE value )
715 {
716     PDH_STATUS ret;
717     struct counter *counter = handle;
718
719     TRACE("%p %x %p %p\n", handle, format, type, value);
720
721     if (!value) return PDH_INVALID_ARGUMENT;
722
723     EnterCriticalSection( &pdh_handle_cs );
724     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
725     {
726         LeaveCriticalSection( &pdh_handle_cs );
727         return PDH_INVALID_HANDLE;
728     }
729     if (counter->status)
730     {
731         LeaveCriticalSection( &pdh_handle_cs );
732         return PDH_INVALID_DATA;
733     }
734     if (!(ret = format_value( counter, format, &counter->one, &counter->two, value )))
735     {
736         value->CStatus = ERROR_SUCCESS;
737         if (type) *type = counter->type;
738     }
739
740     LeaveCriticalSection( &pdh_handle_cs );
741     return ret;
742 }
743
744 /***********************************************************************
745  *              PdhGetRawCounterValue   (PDH.@)
746  */
747 PDH_STATUS WINAPI PdhGetRawCounterValue( PDH_HCOUNTER handle, LPDWORD type,
748                                          PPDH_RAW_COUNTER value )
749 {
750     struct counter *counter = handle;
751
752     TRACE("%p %p %p\n", handle, type, value);
753
754     if (!value) return PDH_INVALID_ARGUMENT;
755
756     EnterCriticalSection( &pdh_handle_cs );
757     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
758     {
759         LeaveCriticalSection( &pdh_handle_cs );
760         return PDH_INVALID_HANDLE;
761     }
762
763     value->CStatus                  = counter->status;
764     value->TimeStamp.dwLowDateTime  = counter->stamp.dwLowDateTime;
765     value->TimeStamp.dwHighDateTime = counter->stamp.dwHighDateTime;
766     value->FirstValue               = counter->one.largevalue;
767     value->SecondValue              = counter->two.largevalue;
768     value->MultiCount               = 1; /* FIXME */
769
770     if (type) *type = counter->type;
771
772     LeaveCriticalSection( &pdh_handle_cs );
773     return ERROR_SUCCESS;
774 }
775
776 /***********************************************************************
777  *              PdhLookupPerfIndexByNameA   (PDH.@)
778  */
779 PDH_STATUS WINAPI PdhLookupPerfIndexByNameA( LPCSTR machine, LPCSTR name, LPDWORD index )
780 {
781     PDH_STATUS ret;
782     WCHAR *machineW = NULL;
783     WCHAR *nameW;
784
785     TRACE("%s %s %p\n", debugstr_a(machine), debugstr_a(name), index);
786
787     if (!name) return PDH_INVALID_ARGUMENT;
788
789     if (machine && !(machineW = pdh_strdup_aw( machine ))) return PDH_MEMORY_ALLOCATION_FAILURE;
790
791     if (!(nameW = pdh_strdup_aw( name )))
792         return PDH_MEMORY_ALLOCATION_FAILURE;
793
794     ret = PdhLookupPerfIndexByNameW( machineW, nameW, index );
795
796     heap_free( nameW );
797     heap_free( machineW );
798     return ret;
799 }
800
801 /***********************************************************************
802  *              PdhLookupPerfIndexByNameW   (PDH.@)
803  */
804 PDH_STATUS WINAPI PdhLookupPerfIndexByNameW( LPCWSTR machine, LPCWSTR name, LPDWORD index )
805 {
806     unsigned int i;
807
808     TRACE("%s %s %p\n", debugstr_w(machine), debugstr_w(name), index);
809
810     if (!name || !index) return PDH_INVALID_ARGUMENT;
811
812     if (machine)
813     {
814         FIXME("remote machine not supported\n");
815         return PDH_CSTATUS_NO_MACHINE;
816     }
817     for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
818     {
819         if (pdh_match_path( counter_sources[i].path, name ))
820         {
821             *index = counter_sources[i].index;
822             return ERROR_SUCCESS;
823         }
824     }
825     return PDH_STRING_NOT_FOUND;
826 }
827
828 /***********************************************************************
829  *              PdhLookupPerfNameByIndexA   (PDH.@)
830  */
831 PDH_STATUS WINAPI PdhLookupPerfNameByIndexA( LPCSTR machine, DWORD index, LPSTR buffer, LPDWORD size )
832 {
833     PDH_STATUS ret;
834     WCHAR *machineW = NULL;
835     WCHAR bufferW[PDH_MAX_COUNTER_NAME];
836     DWORD sizeW = sizeof(bufferW) / sizeof(WCHAR);
837
838     TRACE("%s %d %p %p\n", debugstr_a(machine), index, buffer, size);
839
840     if (!buffer || !size) return PDH_INVALID_ARGUMENT;
841
842     if (machine && !(machineW = pdh_strdup_aw( machine ))) return PDH_MEMORY_ALLOCATION_FAILURE;
843
844     if (!(ret = PdhLookupPerfNameByIndexW( machineW, index, bufferW, &sizeW )))
845     {
846         int required = WideCharToMultiByte( CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL );
847
848         if (size && *size < required) ret = PDH_MORE_DATA;
849         else WideCharToMultiByte( CP_ACP, 0, bufferW, -1, buffer, required, NULL, NULL );
850         if (size) *size = required;
851     }
852     heap_free( machineW );
853     return ret;
854 }
855
856 /***********************************************************************
857  *              PdhLookupPerfNameByIndexW   (PDH.@)
858  */
859 PDH_STATUS WINAPI PdhLookupPerfNameByIndexW( LPCWSTR machine, DWORD index, LPWSTR buffer, LPDWORD size )
860 {
861     PDH_STATUS ret;
862     unsigned int i;
863
864     TRACE("%s %d %p %p\n", debugstr_w(machine), index, buffer, size);
865
866     if (machine)
867     {
868         FIXME("remote machine not supported\n");
869         return PDH_CSTATUS_NO_MACHINE;
870     }
871
872     if (!buffer || !size) return PDH_INVALID_ARGUMENT;
873     if (!index) return ERROR_SUCCESS;
874
875     for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
876     {
877         if (counter_sources[i].index == index)
878         {
879             WCHAR *p = strrchrW( counter_sources[i].path, '\\' ) + 1;
880             unsigned int required = strlenW( p ) + 1;
881
882             if (*size < required) ret = PDH_MORE_DATA;
883             else
884             {
885                 strcpyW( buffer, p );
886                 ret = ERROR_SUCCESS;
887             }
888             *size = required;
889             return ret;
890         }
891     }
892     return PDH_INVALID_ARGUMENT;
893 }
894
895 /***********************************************************************
896  *              PdhOpenQueryA   (PDH.@)
897  */
898 PDH_STATUS WINAPI PdhOpenQueryA( LPCSTR source, DWORD_PTR userdata, PDH_HQUERY *query )
899 {
900     PDH_STATUS ret;
901     WCHAR *sourceW = NULL;
902
903     TRACE("%s %lx %p\n", debugstr_a(source), userdata, query);
904
905     if (source && !(sourceW = pdh_strdup_aw( source ))) return PDH_MEMORY_ALLOCATION_FAILURE;
906
907     ret = PdhOpenQueryW( sourceW, userdata, query );
908     heap_free( sourceW );
909
910     return ret;
911 }
912
913 /***********************************************************************
914  *              PdhOpenQueryW   (PDH.@)
915  */
916 PDH_STATUS WINAPI PdhOpenQueryW( LPCWSTR source, DWORD_PTR userdata, PDH_HQUERY *handle )
917 {
918     struct query *query;
919
920     TRACE("%s %lx %p\n", debugstr_w(source), userdata, handle);
921
922     if (!handle) return PDH_INVALID_ARGUMENT;
923
924     if (source)
925     {
926         FIXME("log file data source not supported\n");
927         return PDH_INVALID_ARGUMENT;
928     }
929     if ((query = create_query()))
930     {
931         query->user = userdata;
932         *handle = query;
933
934         return ERROR_SUCCESS;
935     }
936     return PDH_MEMORY_ALLOCATION_FAILURE;
937 }
938
939 /***********************************************************************
940  *              PdhRemoveCounter   (PDH.@)
941  */
942 PDH_STATUS WINAPI PdhRemoveCounter( PDH_HCOUNTER handle )
943 {
944     struct counter *counter = handle;
945
946     TRACE("%p\n", handle);
947
948     EnterCriticalSection( &pdh_handle_cs );
949     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
950     {
951         LeaveCriticalSection( &pdh_handle_cs );
952         return PDH_INVALID_HANDLE;
953     }
954
955     list_remove( &counter->entry );
956     destroy_counter( counter );
957
958     LeaveCriticalSection( &pdh_handle_cs );
959     return ERROR_SUCCESS;
960 }
961
962 /***********************************************************************
963  *              PdhSetCounterScaleFactor   (PDH.@)
964  */
965 PDH_STATUS WINAPI PdhSetCounterScaleFactor( PDH_HCOUNTER handle, LONG factor )
966 {
967     struct counter *counter = handle;
968
969     TRACE("%p\n", handle);
970
971     EnterCriticalSection( &pdh_handle_cs );
972     if (!counter || counter->magic != PDH_MAGIC_COUNTER)
973     {
974         LeaveCriticalSection( &pdh_handle_cs );
975         return PDH_INVALID_HANDLE;
976     }
977     if (factor < PDH_MIN_SCALE || factor > PDH_MAX_SCALE)
978     {
979         LeaveCriticalSection( &pdh_handle_cs );
980         return PDH_INVALID_ARGUMENT;
981     }
982
983     counter->scale = factor;
984
985     LeaveCriticalSection( &pdh_handle_cs );
986     return ERROR_SUCCESS;
987 }
988
989 /***********************************************************************
990  *              PdhValidatePathA   (PDH.@)
991  */
992 PDH_STATUS WINAPI PdhValidatePathA( LPCSTR path )
993 {
994     PDH_STATUS ret;
995     WCHAR *pathW;
996
997     TRACE("%s\n", debugstr_a(path));
998
999     if (!path) return PDH_INVALID_ARGUMENT;
1000     if (!(pathW = pdh_strdup_aw( path ))) return PDH_MEMORY_ALLOCATION_FAILURE;
1001
1002     ret = PdhValidatePathW( pathW );
1003
1004     heap_free( pathW );
1005     return ret;
1006 }
1007
1008 static PDH_STATUS validate_path( LPCWSTR path )
1009 {
1010     if (!path || !*path) return PDH_INVALID_ARGUMENT;
1011     if (*path++ != '\\' || !strchrW( path, '\\' )) return PDH_CSTATUS_BAD_COUNTERNAME;
1012     return ERROR_SUCCESS;
1013  }
1014
1015 /***********************************************************************
1016  *              PdhValidatePathW   (PDH.@)
1017  */
1018 PDH_STATUS WINAPI PdhValidatePathW( LPCWSTR path )
1019 {
1020     PDH_STATUS ret;
1021     unsigned int i;
1022
1023     TRACE("%s\n", debugstr_w(path));
1024
1025     if ((ret = validate_path( path ))) return ret;
1026
1027     for (i = 0; i < sizeof(counter_sources) / sizeof(counter_sources[0]); i++)
1028         if (pdh_match_path( counter_sources[i].path, path )) return ERROR_SUCCESS;
1029
1030     return PDH_CSTATUS_NO_COUNTER;
1031 }
1032
1033 /***********************************************************************
1034  *              PdhValidatePathExA   (PDH.@)
1035  */
1036 PDH_STATUS WINAPI PdhValidatePathExA( PDH_HLOG source, LPCSTR path )
1037 {
1038     TRACE("%p %s\n", source, debugstr_a(path));
1039
1040     if (source)
1041     {
1042         FIXME("log file data source not supported\n");
1043         return ERROR_SUCCESS;
1044     }
1045     return PdhValidatePathA( path );
1046 }
1047
1048 /***********************************************************************
1049  *              PdhValidatePathExW   (PDH.@)
1050  */
1051 PDH_STATUS WINAPI PdhValidatePathExW( PDH_HLOG source, LPCWSTR path )
1052 {
1053     TRACE("%p %s\n", source, debugstr_w(path));
1054
1055     if (source)
1056     {
1057         FIXME("log file data source not supported\n");
1058         return ERROR_SUCCESS;
1059     }
1060     return PdhValidatePathW( path );
1061 }
1062
1063 /***********************************************************************
1064  *              PdhEnumObjectItemsA   (PDH.@)
1065  */
1066 PDH_STATUS WINAPI PdhEnumObjectItemsA(LPCSTR szDataSource, LPCSTR szMachineName, LPCSTR szObjectName,
1067                                       LPSTR mszCounterList, LPDWORD pcchCounterListLength, LPSTR mszInstanceList,
1068                                       LPDWORD pcchInstanceListLength, DWORD dwDetailLevel, DWORD dwFlags)
1069 {
1070     FIXME("%s, %s, %s, %p, %p, %p, %p, %d, 0x%x: stub\n", debugstr_a(szDataSource), debugstr_a(szMachineName),
1071          debugstr_a(szObjectName), mszCounterList, pcchCounterListLength, mszInstanceList,
1072          pcchInstanceListLength, dwDetailLevel, dwFlags);
1073
1074     return PDH_NOT_IMPLEMENTED;
1075 }
1076
1077 /***********************************************************************
1078  *              PdhEnumObjectItemsW   (PDH.@)
1079  */
1080 PDH_STATUS WINAPI PdhEnumObjectItemsW(LPCWSTR szDataSource, LPCWSTR szMachineName, LPCWSTR szObjectName,
1081                                       LPWSTR mszCounterList, LPDWORD pcchCounterListLength, LPWSTR mszInstanceList,
1082                                       LPDWORD pcchInstanceListLength, DWORD dwDetailLevel, DWORD dwFlags)
1083 {
1084     FIXME("%s, %s, %s, %p, %p, %p, %p, %d, 0x%x: stub\n", debugstr_w(szDataSource), debugstr_w(szMachineName),
1085          debugstr_w(szObjectName), mszCounterList, pcchCounterListLength, mszInstanceList,
1086          pcchInstanceListLength, dwDetailLevel, dwFlags);
1087
1088     return PDH_NOT_IMPLEMENTED;
1089 }