mirror of
https://github.com/hashcat/hashcat.git
synced 2024-11-15 20:39:17 +00:00
126 lines
2.2 KiB
C
126 lines
2.2 KiB
C
/**
|
|
* Author......: See docs/credits.txt
|
|
* License.....: MIT
|
|
*/
|
|
|
|
#if defined (__APPLE__)
|
|
#include <stdio.h>
|
|
#endif
|
|
|
|
#include "common.h"
|
|
#include "types.h"
|
|
#include "memory.h"
|
|
#include "event.h"
|
|
#include "affinity.h"
|
|
|
|
#if defined (__APPLE__)
|
|
static void CPU_ZERO (cpu_set_t *cs)
|
|
{
|
|
cs->count = 0;
|
|
}
|
|
|
|
static void CPU_SET (int num, cpu_set_t *cs)
|
|
{
|
|
cs->count |= (1 << num);
|
|
}
|
|
|
|
static int CPU_ISSET (int num, cpu_set_t *cs)
|
|
{
|
|
return (cs->count & (1 << num));
|
|
}
|
|
|
|
static int pthread_setaffinity_np (pthread_t thread, size_t cpu_size, cpu_set_t *cpu_set)
|
|
{
|
|
int core;
|
|
|
|
for (core = 0; core < (8 * (int) cpu_size); core++)
|
|
{
|
|
if (CPU_ISSET (core, cpu_set)) break;
|
|
}
|
|
|
|
thread_affinity_policy_data_t policy = { core };
|
|
|
|
return thread_policy_set (pthread_mach_thread_np (thread), THREAD_AFFINITY_POLICY, (thread_policy_t) &policy, 1);
|
|
}
|
|
#endif
|
|
|
|
#if defined (__FreeBSD__)
|
|
typedef cpuset_t cpu_set_t;
|
|
#endif
|
|
|
|
int set_cpu_affinity (hashcat_ctx_t *hashcat_ctx)
|
|
{
|
|
const user_options_t *user_options = hashcat_ctx->user_options;
|
|
|
|
if (user_options->cpu_affinity == NULL) return 0;
|
|
|
|
#if defined (_WIN)
|
|
DWORD_PTR aff_mask = 0;
|
|
#else
|
|
cpu_set_t cpuset;
|
|
CPU_ZERO (&cpuset);
|
|
#endif
|
|
|
|
char *devices = mystrdup (user_options->cpu_affinity);
|
|
|
|
char *next = strtok (devices, ",");
|
|
|
|
do
|
|
{
|
|
int cpu_id = atoi (next);
|
|
|
|
if (cpu_id == 0)
|
|
{
|
|
#if defined (_WIN)
|
|
aff_mask = 0;
|
|
#else
|
|
CPU_ZERO (&cpuset);
|
|
#endif
|
|
|
|
break;
|
|
}
|
|
|
|
if (cpu_id > 32)
|
|
{
|
|
event_log_error (hashcat_ctx, "ERROR: Invalid cpu_id %u specified", cpu_id);
|
|
|
|
return (-1);
|
|
}
|
|
|
|
#if defined (_WIN)
|
|
aff_mask |= 1u << (cpu_id - 1);
|
|
#else
|
|
CPU_SET ((cpu_id - 1), &cpuset);
|
|
#endif
|
|
|
|
} while ((next = strtok (NULL, ",")) != NULL);
|
|
|
|
myfree (devices);
|
|
|
|
#if defined (_WIN)
|
|
|
|
SetProcessAffinityMask (GetCurrentProcess (), aff_mask);
|
|
|
|
if (SetThreadAffinityMask (GetCurrentThread (), aff_mask) == 0)
|
|
{
|
|
event_log_error (hashcat_ctx, "ERROR: %s", "SetThreadAffinityMask()");
|
|
|
|
return -1;
|
|
}
|
|
|
|
#else
|
|
|
|
pthread_t thread = pthread_self ();
|
|
|
|
if (pthread_setaffinity_np (thread, sizeof (cpu_set_t), &cpuset) == -1)
|
|
{
|
|
event_log_error (hashcat_ctx, "ERROR: %s", "pthread_setaffinity_np()");
|
|
|
|
return -1;
|
|
}
|
|
|
|
#endif
|
|
|
|
return 0;
|
|
}
|