mirror of
https://github.com/hashcat/hashcat.git
synced 2024-11-22 16:18:09 +00:00
254 lines
12 KiB
C
254 lines
12 KiB
C
/**
|
|
* Authors.....: Jens Steube <jens.steube@gmail.com>
|
|
* Gabriele Gristina <matrix@hashcat.net>
|
|
* magnum <john.magnum@hushmail.com>
|
|
*
|
|
* License.....: MIT
|
|
*/
|
|
|
|
#ifndef EXT_NVML_H
|
|
#define EXT_NVML_H
|
|
|
|
#if defined(HAVE_HWMON)
|
|
|
|
#include <common.h>
|
|
|
|
/**
|
|
* Declarations from nvml.h
|
|
*/
|
|
|
|
typedef struct nvmlDevice_st* nvmlDevice_t;
|
|
|
|
typedef struct nvmlPciInfo_st {
|
|
char busId[16];
|
|
unsigned int domain;
|
|
unsigned int bus;
|
|
unsigned int device;
|
|
unsigned int pciDeviceId;
|
|
unsigned int pciSubSystemId;
|
|
} nvmlPciInfo_t;
|
|
|
|
typedef struct nvmlUtilization_st {
|
|
unsigned int gpu; // GPU kernel execution last second, percent
|
|
unsigned int memory; // GPU memory read/write last second, percent
|
|
} nvmlUtilization_t;
|
|
|
|
typedef enum nvmlTemperatureSensors_enum {
|
|
NVML_TEMPERATURE_GPU = 0 // Temperature sensor for the GPU die
|
|
} nvmlTemperatureSensors_t;
|
|
|
|
typedef enum nvmlReturn_enum {
|
|
NVML_SUCCESS = 0, // The operation was successful
|
|
NVML_ERROR_UNINITIALIZED = 1, // NVML was not first initialized with nvmlInit()
|
|
NVML_ERROR_INVALID_ARGUMENT = 2, // A supplied argument is invalid
|
|
NVML_ERROR_NOT_SUPPORTED = 3, // The requested operation is not available on target device
|
|
NVML_ERROR_NO_PERMISSION = 4, // The current user does not have permission for operation
|
|
NVML_ERROR_ALREADY_INITIALIZED = 5, // Deprecated: Multiple initializations are now allowed through ref counting
|
|
NVML_ERROR_NOT_FOUND = 6, // A query to find an object was unsuccessful
|
|
NVML_ERROR_INSUFFICIENT_SIZE = 7, // An input argument is not large enough
|
|
NVML_ERROR_INSUFFICIENT_POWER = 8, // A device's external power cables are not properly attached
|
|
NVML_ERROR_DRIVER_NOT_LOADED = 9, // NVIDIA driver is not loaded
|
|
NVML_ERROR_TIMEOUT = 10, // User provided timeout passed
|
|
NVML_ERROR_UNKNOWN = 999 // An internal driver error occurred
|
|
} nvmlReturn_t;
|
|
|
|
typedef enum nvmlClockType_enum {
|
|
NVML_CLOCK_GRAPHICS = 0,
|
|
NVML_CLOCK_SM = 1,
|
|
NVML_CLOCK_MEM = 2
|
|
} nvmlClockType_t;
|
|
|
|
typedef enum nvmlTemperatureThresholds_enum
|
|
{
|
|
NVML_TEMPERATURE_THRESHOLD_SHUTDOWN = 0, // Temperature at which the GPU will shut down
|
|
// for HW protection
|
|
NVML_TEMPERATURE_THRESHOLD_SLOWDOWN = 1, // Temperature at which the GPU will begin slowdown
|
|
// Keep this last
|
|
NVML_TEMPERATURE_THRESHOLD_COUNT
|
|
} nvmlTemperatureThresholds_t;
|
|
|
|
/**
|
|
* Compute mode.
|
|
*
|
|
* NVML_COMPUTEMODE_EXCLUSIVE_PROCESS was added in CUDA 4.0.
|
|
* Earlier CUDA versions supported a single exclusive mode,
|
|
* which is equivalent to NVML_COMPUTEMODE_EXCLUSIVE_THREAD in CUDA 4.0 and beyond.
|
|
*/
|
|
typedef enum nvmlComputeMode_enum
|
|
{
|
|
NVML_COMPUTEMODE_DEFAULT = 0, //!< Default compute mode -- multiple contexts per device
|
|
NVML_COMPUTEMODE_EXCLUSIVE_THREAD = 1, //!< Compute-exclusive-thread mode -- only one context per device, usable from one thread at a time
|
|
NVML_COMPUTEMODE_PROHIBITED = 2, //!< Compute-prohibited mode -- no contexts per device
|
|
NVML_COMPUTEMODE_EXCLUSIVE_PROCESS = 3, //!< Compute-exclusive-process mode -- only one context per device, usable from multiple threads at a time
|
|
|
|
// Keep this last
|
|
NVML_COMPUTEMODE_COUNT
|
|
} nvmlComputeMode_t;
|
|
|
|
/**
|
|
* GPU Operation Mode
|
|
*
|
|
* GOM allows to reduce power usage and optimize GPU throughput by disabling GPU features.
|
|
*
|
|
* Each GOM is designed to meet specific user needs.
|
|
*/
|
|
typedef enum nvmlGom_enum
|
|
{
|
|
NVML_GOM_ALL_ON = 0, //!< Everything is enabled and running at full speed
|
|
|
|
NVML_GOM_COMPUTE = 1, //!< Designed for running only compute tasks. Graphics operations
|
|
//!< are not allowed
|
|
|
|
NVML_GOM_LOW_DP = 2 //!< Designed for running graphics applications that don't require
|
|
//!< high bandwidth double precision
|
|
} nvmlGpuOperationMode_t;
|
|
|
|
/***************************************************************************************************/
|
|
/** @addtogroup nvmlClocksThrottleReasons
|
|
* @{
|
|
*/
|
|
/***************************************************************************************************/
|
|
|
|
/** Nothing is running on the GPU and the clocks are dropping to Idle state
|
|
* \note This limiter may be removed in a later release
|
|
*/
|
|
#define nvmlClocksThrottleReasonGpuIdle 0x0000000000000001LL
|
|
|
|
/** GPU clocks are limited by current setting of applications clocks
|
|
*
|
|
* @see nvmlDeviceSetApplicationsClocks
|
|
* @see nvmlDeviceGetApplicationsClock
|
|
*/
|
|
#define nvmlClocksThrottleReasonApplicationsClocksSetting 0x0000000000000002LL
|
|
|
|
/**
|
|
* @deprecated Renamed to \ref nvmlClocksThrottleReasonApplicationsClocksSetting
|
|
* as the name describes the situation more accurately.
|
|
*/
|
|
#define nvmlClocksThrottleReasonUserDefinedClocks nvmlClocksThrottleReasonApplicationsClocksSetting
|
|
|
|
/** SW Power Scaling algorithm is reducing the clocks below requested clocks
|
|
*
|
|
* @see nvmlDeviceGetPowerUsage
|
|
* @see nvmlDeviceSetPowerManagementLimit
|
|
* @see nvmlDeviceGetPowerManagementLimit
|
|
*/
|
|
#define nvmlClocksThrottleReasonSwPowerCap 0x0000000000000004LL
|
|
|
|
/** HW Slowdown (reducing the core clocks by a factor of 2 or more) is engaged
|
|
*
|
|
* This is an indicator of:
|
|
* - temperature being too high
|
|
* - External Power Brake Assertion is triggered (e.g. by the system power supply)
|
|
* - Power draw is too high and Fast Trigger protection is reducing the clocks
|
|
* - May be also reported during PState or clock change
|
|
* - This behavior may be removed in a later release.
|
|
*
|
|
* @see nvmlDeviceGetTemperature
|
|
* @see nvmlDeviceGetTemperatureThreshold
|
|
* @see nvmlDeviceGetPowerUsage
|
|
*/
|
|
#define nvmlClocksThrottleReasonHwSlowdown 0x0000000000000008LL
|
|
|
|
/** Some other unspecified factor is reducing the clocks */
|
|
#define nvmlClocksThrottleReasonUnknown 0x8000000000000000LL
|
|
|
|
/** Bit mask representing no clocks throttling
|
|
*
|
|
* Clocks are as high as possible.
|
|
* */
|
|
#define nvmlClocksThrottleReasonNone 0x0000000000000000LL
|
|
|
|
|
|
/*
|
|
* End of declarations from nvml.h
|
|
**/
|
|
|
|
typedef nvmlDevice_t HM_ADAPTER_NVML;
|
|
|
|
#include <shared.h>
|
|
|
|
#if defined(_WIN32) || defined(__WIN32__) || defined(__CYGWIN__)
|
|
#define NVML_API_CALL __stdcall
|
|
#else
|
|
#define NVML_API_CALL
|
|
#endif
|
|
|
|
typedef const char * (*NVML_API_CALL NVML_ERROR_STRING) (nvmlReturn_t);
|
|
typedef int (*NVML_API_CALL NVML_INIT) (void);
|
|
typedef int (*NVML_API_CALL NVML_SHUTDOWN) (void);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_NAME) (nvmlDevice_t, char *, unsigned int);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_HANDLE_BY_INDEX) (unsigned int, nvmlDevice_t *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_TEMPERATURE) (nvmlDevice_t, nvmlTemperatureSensors_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_FAN_SPEED) (nvmlDevice_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_POWER_USAGE) (nvmlDevice_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_UTILIZATION_RATES) (nvmlDevice_t, nvmlUtilization_t *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_CLOCKINFO) (nvmlDevice_t, nvmlClockType_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_THRESHOLD) (nvmlDevice_t, nvmlTemperatureThresholds_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_CURRPCIELINKGENERATION) (nvmlDevice_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_CURRPCIELINKWIDTH) (nvmlDevice_t, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_CURRENTCLOCKSTHROTTLEREASONS) (nvmlDevice_t, unsigned long long *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_SUPPORTEDCLOCKSTHROTTLEREASONS) (nvmlDevice_t, unsigned long long *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_SET_COMPUTEMODE) (nvmlDevice_t, nvmlComputeMode_t);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_SET_OPERATIONMODE) (nvmlDevice_t, nvmlGpuOperationMode_t);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_POWERMANAGEMENTLIMITCONSTRAINTS) (nvmlDevice_t, unsigned int *, unsigned int *);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_SET_POWERMANAGEMENTLIMIT) (nvmlDevice_t, unsigned int);
|
|
typedef nvmlReturn_t (*NVML_API_CALL NVML_DEVICE_GET_POWERMANAGEMENTLIMIT) (nvmlDevice_t, unsigned int *);
|
|
|
|
typedef struct
|
|
{
|
|
NVML_LIB lib;
|
|
|
|
NVML_ERROR_STRING nvmlErrorString;
|
|
NVML_INIT nvmlInit;
|
|
NVML_SHUTDOWN nvmlShutdown;
|
|
NVML_DEVICE_GET_NAME nvmlDeviceGetName;
|
|
NVML_DEVICE_GET_HANDLE_BY_INDEX nvmlDeviceGetHandleByIndex;
|
|
NVML_DEVICE_GET_TEMPERATURE nvmlDeviceGetTemperature;
|
|
NVML_DEVICE_GET_FAN_SPEED nvmlDeviceGetFanSpeed;
|
|
NVML_DEVICE_GET_POWER_USAGE nvmlDeviceGetPowerUsage;
|
|
NVML_DEVICE_GET_UTILIZATION_RATES nvmlDeviceGetUtilizationRates;
|
|
NVML_DEVICE_GET_CLOCKINFO nvmlDeviceGetClockInfo;
|
|
NVML_DEVICE_GET_THRESHOLD nvmlDeviceGetTemperatureThreshold;
|
|
NVML_DEVICE_GET_CURRPCIELINKGENERATION nvmlDeviceGetCurrPcieLinkGeneration;
|
|
NVML_DEVICE_GET_CURRPCIELINKWIDTH nvmlDeviceGetCurrPcieLinkWidth;
|
|
NVML_DEVICE_GET_CURRENTCLOCKSTHROTTLEREASONS nvmlDeviceGetCurrentClocksThrottleReasons;
|
|
NVML_DEVICE_GET_SUPPORTEDCLOCKSTHROTTLEREASONS nvmlDeviceGetSupportedClocksThrottleReasons;
|
|
NVML_DEVICE_SET_COMPUTEMODE nvmlDeviceSetComputeMode;
|
|
NVML_DEVICE_SET_OPERATIONMODE nvmlDeviceSetGpuOperationMode;
|
|
NVML_DEVICE_GET_POWERMANAGEMENTLIMITCONSTRAINTS nvmlDeviceGetPowerManagementLimitConstraints;
|
|
NVML_DEVICE_SET_POWERMANAGEMENTLIMIT nvmlDeviceSetPowerManagementLimit;
|
|
NVML_DEVICE_GET_POWERMANAGEMENTLIMIT nvmlDeviceGetPowerManagementLimit;
|
|
|
|
} hm_nvml_lib_t;
|
|
|
|
#define NVML_PTR hm_nvml_lib_t
|
|
|
|
int nvml_init (NVML_PTR *lib);
|
|
void nvml_close (NVML_PTR *lib);
|
|
|
|
const char * hm_NVML_nvmlErrorString (NVML_PTR *nvml, nvmlReturn_t nvml_rc);
|
|
nvmlReturn_t hm_NVML_nvmlInit (NVML_PTR *nvml);
|
|
nvmlReturn_t hm_NVML_nvmlShutdown (NVML_PTR *nvml);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetName (NVML_PTR *nvml, int, nvmlDevice_t device, char *name, unsigned int length);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetHandleByIndex (NVML_PTR *nvml, int, unsigned int index, nvmlDevice_t *device);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetTemperature (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlTemperatureSensors_t sensorType, unsigned int *temp);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetFanSpeed (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *speed);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetPowerUsage (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *power);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetUtilizationRates (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlUtilization_t *utilization);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetClockInfo (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlClockType_t type, unsigned int *clock);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetTemperatureThreshold (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlTemperatureThresholds_t thresholdType, unsigned int *temp);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetCurrPcieLinkGeneration (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *currLinkGen);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetCurrPcieLinkWidth (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *currLinkWidth);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetCurrentClocksThrottleReasons (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned long long *clocksThrottleReasons);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetSupportedClocksThrottleReasons (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned long long *supportedClocksThrottleReasons);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceSetComputeMode (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlComputeMode_t mode);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceSetGpuOperationMode (NVML_PTR *nvml, int, nvmlDevice_t device, nvmlGpuOperationMode_t mode);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetPowerManagementLimitConstraints (NVML_PTR *nvml, int, nvmlDevice_t device, unsigned int *minLimit, unsigned int *maxLimit);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceSetPowerManagementLimit (NVML_PTR *nvml, int skip_warnings, nvmlDevice_t device, unsigned int limit);
|
|
nvmlReturn_t hm_NVML_nvmlDeviceGetPowerManagementLimit (NVML_PTR *nvml, int skip_warnings, nvmlDevice_t device, unsigned int *limit);
|
|
|
|
#endif // HAVE_HWMON
|
|
|
|
#endif // EXT_NVML_H
|