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 }