git.cworth.org Git - apitrace/blob - os_win32.cpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2010 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26 #include <windows.h>
  27 #include <string.h>
  28 #include <stdio.h>
  29
  30 #include "os.hpp"
  31
  32
  33 namespace OS {
  34
  35
  36 /*
  37  * Trick from http://locklessinc.com/articles/pthreads_on_windows/
  38  */
  39 static CRITICAL_SECTION
  40 CriticalSection = {
  41     (PCRITICAL_SECTION_DEBUG)-1, -1, 0, 0, 0, 0
  42 };
  43
  44
  45 void
  46 AcquireMutex(void)
  47 {
  48     EnterCriticalSection(&CriticalSection);
  49 }
  50
  51
  52 void
  53 ReleaseMutex(void)
  54 {
  55     LeaveCriticalSection(&CriticalSection);
  56 }
  57
  58
  59 bool
  60 GetProcessName(char *str, size_t size)
  61 {
  62     char szProcessPath[PATH_MAX];
  63     char *lpProcessName;
  64     char *lpProcessExt;
  65
  66     GetModuleFileNameA(NULL, szProcessPath, sizeof(szProcessPath)/sizeof(szProcessPath[0]));
  67
  68     lpProcessName = strrchr(szProcessPath, '\\');
  69     lpProcessName = lpProcessName ? lpProcessName + 1 : szProcessPath;
  70
  71     lpProcessExt = strrchr(lpProcessName, '.');
  72     if (lpProcessExt) {
  73         *lpProcessExt = '\0';
  74     }
  75
  76     strncpy(str, lpProcessName, size);
  77
  78     return true;
  79 }
  80
  81 bool
  82 GetCurrentDir(char *str, size_t size)
  83 {
  84     DWORD ret;
  85     ret = GetCurrentDirectoryA(size, str);
  86     str[size - 1] = 0;
  87     return ret == 0 ? false : true;
  88 }
  89
  90 void
  91 DebugMessage(const char *format, ...)
  92 {
  93     char buf[4096];
  94
  95     va_list ap;
  96     va_start(ap, format);
  97     fflush(stdout);
  98     vsnprintf(buf, sizeof buf, format, ap);
  99     va_end(ap);
 100
 101     OutputDebugStringA(buf);
 102
 103     /*
 104      * Also write the message to stderr, when a debugger is not present (to
 105      * avoid duplicate messages in command line debuggers).
 106      */
 107 #if _WIN32_WINNT > 0x0400
 108     if (!IsDebuggerPresent()) {
 109         fflush(stdout);
 110         fputs(buf, stderr);
 111         fflush(stderr);
 112     }
 113 #endif
 114 }
 115
 116 long long GetTime(void)
 117 {
 118     static LARGE_INTEGER frequency;
 119     LARGE_INTEGER counter;
 120     if (!frequency.QuadPart)
 121         QueryPerformanceFrequency(&frequency);
 122     QueryPerformanceCounter(&counter);
 123     return counter.QuadPart*1000000LL/frequency.QuadPart;
 124 }
 125
 126 void
 127 Abort(void)
 128 {
 129 #ifndef NDEBUG
 130     DebugBreak();
 131 #else
 132     ExitProcess(0);
 133 #endif
 134 }
 135
 136 } /* namespace OS */