src/ext_nvapi.c

   1 /**
   2  * Authors.....: Jens Steube <jens.steube@gmail.com>
   3  *               Gabriele Gristina <matrix@hashcat.net>
   4  *
   5  * License.....: MIT
   6  */
   7
   8 #include <ext_nvapi.h>
   9
  10 int nvapi_init (NVAPI_PTR *nvapi)
  11 {
  12   if (!nvapi) return (-1);
  13
  14   memset (nvapi, 0, sizeof (NVAPI_PTR));
  15
  16   #if __x86_64__
  17   nvapi->lib = hc_dlopen ("nvapi64.dll");
  18   #elif __x86__
  19   nvapi->lib = hc_dlopen ("nvapi.dll");
  20   #endif
  21
  22   if (!nvapi->lib)
  23   {
  24     //if (data.quiet == 0)
  25     //  log_info ("WARNING: load NVAPI library failed, proceed without NVAPI HWMon enabled.");
  26
  27     return (-1);
  28   }
  29
  30   HC_LOAD_FUNC(nvapi, nvapi_QueryInterface,                     NVAPI_QUERYINTERFACE,                     NVAPI,                0)
  31   HC_LOAD_ADDR(nvapi, NvAPI_Initialize,                         NVAPI_INITIALIZE,                         nvapi_QueryInterface, 0x0150E828, NVAPI, 0)
  32   HC_LOAD_ADDR(nvapi, NvAPI_Unload,                             NVAPI_UNLOAD,                             nvapi_QueryInterface, 0xD22BDD7E, NVAPI, 0)
  33   HC_LOAD_ADDR(nvapi, NvAPI_GetErrorMessage,                    NVAPI_GETERRORMESSAGE,                    nvapi_QueryInterface, 0x6C2D048C, NVAPI, 0)
  34   HC_LOAD_ADDR(nvapi, NvAPI_EnumPhysicalGPUs,                   NVAPI_ENUMPHYSICALGPUS,                   nvapi_QueryInterface, 0xE5AC921F, NVAPI, 0)
  35   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetDynamicPstatesInfoEx,        NVAPI_GPU_GETDYNAMICPSTATESINFOEX,        nvapi_QueryInterface, 0x60DED2ED, NVAPI, 0)
  36   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetThermalSettings,             NVAPI_GPU_GETTHERMALSETTINGS,             nvapi_QueryInterface, 0xE3640A56, NVAPI, 0)
  37   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetTachReading,                 NVAPI_GPU_GETTACHREADING,                 nvapi_QueryInterface, 0x5F608315, NVAPI, 0)
  38   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetCoolerSettings,              NVAPI_GPU_GETCOOLERSETTINGS,              nvapi_QueryInterface, 0xDA141340, NVAPI, 0)
  39   HC_LOAD_ADDR(nvapi, NvAPI_GPU_SetCoolerLevels,                NVAPI_GPU_SETCOOLERLEVELS,                nvapi_QueryInterface, 0x891FA0AE, NVAPI, 0)
  40   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetAllClockFrequencies,         NVAPI_GPU_GETALLCLOCKFREQUENCIES,         nvapi_QueryInterface, 0xDCB616C3, NVAPI, 0)
  41   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetCurrentPCIEDownstreamWidth,  NVAPI_GPU_GETCURRENTPCIEDOWNSTREAMWIDTH,  nvapi_QueryInterface, 0xD048C3B1, NVAPI, 0)
  42   HC_LOAD_ADDR(nvapi, NvAPI_GPU_GetPerfDecreaseInfo,            NVAPI_GPU_GETPERFDECREASEINFO,            nvapi_QueryInterface, 0x7F7F4600, NVAPI, 0)
  43
  44   return 0;
  45 }
  46
  47 void nvapi_close (NVAPI_PTR *nvapi)
  48 {
  49   if (nvapi)
  50   {
  51     if (nvapi->lib)
  52       hc_dlclose (nvapi->lib);
  53
  54     myfree (nvapi);
  55   }
  56 }
  57
  58 int hm_NvAPI_Initialize (NVAPI_PTR *nvapi)
  59 {
  60   if (!nvapi) return (-1);
  61
  62   NvAPI_Status NvAPI_rc = nvapi->NvAPI_Initialize ();
  63
  64   if (NvAPI_rc == NVAPI_LIBRARY_NOT_FOUND) NvAPI_rc = NVAPI_OK; // not a bug
  65
  66   if (NvAPI_rc != NVAPI_OK)
  67   {
  68     NvAPI_ShortString string = { 0 };
  69
  70     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
  71
  72     log_info ("WARN: %s %d %s\n", "NvAPI_Initialize()", NvAPI_rc, string);
  73   }
  74
  75   return NvAPI_rc;
  76 }
  77
  78 int hm_NvAPI_Unload (NVAPI_PTR *nvapi)
  79 {
  80   if (!nvapi) return (-1);
  81
  82   NvAPI_Status NvAPI_rc = nvapi->NvAPI_Unload ();
  83
  84   if (NvAPI_rc != NVAPI_OK)
  85   {
  86     NvAPI_ShortString string = { 0 };
  87
  88     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
  89
  90     log_info ("WARN: %s %d %s\n", "NvAPI_Unload()", NvAPI_rc, string);
  91   }
  92
  93   return NvAPI_rc;
  94 }
  95
  96 int hm_NvAPI_GetErrorMessage (NVAPI_PTR *nvapi, NvAPI_Status NvAPI_rc, NvAPI_ShortString string)
  97 {
  98   if (!nvapi) return (-1);
  99
 100   return nvapi->NvAPI_GetErrorMessage (NvAPI_rc, string);
 101 }
 102
 103 int hm_NvAPI_EnumPhysicalGPUs (NVAPI_PTR *nvapi, NvPhysicalGpuHandle nvGPUHandle[NVAPI_MAX_PHYSICAL_GPUS], NvU32 *pGpuCount)
 104 {
 105   if (!nvapi) return (-1);
 106
 107   NvAPI_Status NvAPI_rc = nvapi->NvAPI_EnumPhysicalGPUs (nvGPUHandle, pGpuCount);
 108
 109   if (NvAPI_rc != NVAPI_OK)
 110   {
 111     NvAPI_ShortString string = { 0 };
 112
 113     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 114
 115     log_info ("WARN: %s %d %s\n", "NvAPI_EnumPhysicalGPUs()", NvAPI_rc, string);
 116   }
 117
 118   return NvAPI_rc;
 119 }
 120
 121 int hm_NvAPI_GPU_GetThermalSettings (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NvU32 sensorIndex, NV_GPU_THERMAL_SETTINGS *pThermalSettings)
 122 {
 123   if (!nvapi) return (-1);
 124
 125   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetThermalSettings (hPhysicalGpu, sensorIndex, pThermalSettings);
 126
 127   if (NvAPI_rc != NVAPI_OK)
 128   {
 129     NvAPI_ShortString string = { 0 };
 130
 131     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 132
 133     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetThermalSettings()", NvAPI_rc, string);
 134   }
 135
 136   return NvAPI_rc;
 137 }
 138
 139 int hm_NvAPI_GPU_GetTachReading (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGPU, NvU32 *pValue)
 140 {
 141   if (!nvapi) return (-1);
 142
 143   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetTachReading (hPhysicalGPU, pValue);
 144
 145   if (NvAPI_rc != NVAPI_OK)
 146   {
 147     NvAPI_ShortString string = { 0 };
 148
 149     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 150
 151     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetTachReading()", NvAPI_rc, string);
 152   }
 153
 154   return NvAPI_rc;
 155 }
 156
 157 int hm_NvAPI_GPU_GetCoolerSettings (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NvU32 coolerIndex, NV_GPU_COOLER_SETTINGS *pCoolerSettings)
 158 {
 159   if (!nvapi) return (-1);
 160
 161   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetCoolerSettings (hPhysicalGpu, coolerIndex, pCoolerSettings);
 162
 163   if (NvAPI_rc != NVAPI_OK)
 164   {
 165     NvAPI_ShortString string = { 0 };
 166
 167     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 168
 169     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetCoolerSettings()", NvAPI_rc, string);
 170   }
 171
 172   return NvAPI_rc;
 173 }
 174
 175 int hm_NvAPI_GPU_SetCoolerLevels (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NvU32 coolerIndex, NV_GPU_COOLER_LEVELS *pCoolerLevels)
 176 {
 177   if (!nvapi) return (-1);
 178
 179   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_SetCoolerLevels (hPhysicalGpu, coolerIndex, pCoolerLevels);
 180
 181   if (NvAPI_rc != NVAPI_OK)
 182   {
 183     NvAPI_ShortString string = { 0 };
 184
 185     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 186
 187     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_SetCoolerLevels()", NvAPI_rc, string);
 188   }
 189
 190   return NvAPI_rc;
 191 }
 192
 193 int hm_NvAPI_GPU_GetDynamicPstatesInfoEx (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NV_GPU_DYNAMIC_PSTATES_INFO_EX *pDynamicPstatesInfoEx)
 194 {
 195   if (!nvapi) return (-1);
 196
 197   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetDynamicPstatesInfoEx (hPhysicalGpu, pDynamicPstatesInfoEx);
 198
 199   if (NvAPI_rc != NVAPI_OK)
 200   {
 201     NvAPI_ShortString string = { 0 };
 202
 203     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 204
 205     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetDynamicPstatesInfoEx()", NvAPI_rc, string);
 206   }
 207
 208   return NvAPI_rc;
 209 }
 210
 211 int hm_NvAPI_GPU_GetAllClockFrequencies (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NV_GPU_CLOCK_FREQUENCIES *pClkFreqs)
 212 {
 213   if (!nvapi) return (-1);
 214
 215   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetAllClockFrequencies (hPhysicalGpu, pClkFreqs);
 216
 217   if (NvAPI_rc != NVAPI_OK)
 218   {
 219     NvAPI_ShortString string = { 0 };
 220
 221     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 222
 223     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetAllClockFrequencies()", NvAPI_rc, string);
 224   }
 225
 226   return NvAPI_rc;
 227 }
 228
 229 int hm_NvAPI_GPU_GetCurrentPCIEDownstreamWidth (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NvU32 *pWidth)
 230 {
 231   if (!nvapi) return (-1);
 232
 233   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetCurrentPCIEDownstreamWidth (hPhysicalGpu, pWidth);
 234
 235   if (NvAPI_rc != NVAPI_OK)
 236   {
 237     NvAPI_ShortString string = { 0 };
 238
 239     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 240
 241     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetCurrentPCIEDownstreamWidth()", NvAPI_rc, string);
 242   }
 243
 244   return NvAPI_rc;
 245 }
 246
 247 int hm_NvAPI_GPU_GetPerfDecreaseInfo (NVAPI_PTR *nvapi, NvPhysicalGpuHandle hPhysicalGpu, NvU32 *pPerfDecrInfo)
 248 {
 249   if (!nvapi) return (-1);
 250
 251   NvAPI_Status NvAPI_rc = nvapi->NvAPI_GPU_GetPerfDecreaseInfo (hPhysicalGpu, pPerfDecrInfo);
 252
 253   if (NvAPI_rc != NVAPI_OK)
 254   {
 255     NvAPI_ShortString string = { 0 };
 256
 257     hm_NvAPI_GetErrorMessage (nvapi, NvAPI_rc, string);
 258
 259     log_info ("WARN: %s %d %s\n", "NvAPI_GPU_GetPerfDecreaseInfo()", NvAPI_rc, string);
 260   }
 261
 262   return NvAPI_rc;
 263 }
 264
 265 #ifdef __MINGW64__
 266
 267 void __security_check_cookie (uintptr_t _StackCookie)
 268 {
 269   (void) _StackCookie;
 270 }
 271
 272 void __GSHandlerCheck ()
 273 {
 274 }
 275
 276 #endif