mirror of
https://github.com/hashcat/hashcat.git
synced 2024-12-16 19:58:25 +00:00
163 lines
4.1 KiB
C
163 lines
4.1 KiB
C
/**
|
|
* Author......: See docs/credits.txt
|
|
* License.....: MIT
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "types.h"
|
|
#include "memory.h"
|
|
#include "event.h"
|
|
#include "dictstat.h"
|
|
|
|
int sort_by_dictstat (const void *s1, const void *s2)
|
|
{
|
|
dictstat_t *d1 = (dictstat_t *) s1;
|
|
dictstat_t *d2 = (dictstat_t *) s2;
|
|
|
|
#if defined (__linux__)
|
|
d2->stat.st_atim = d1->stat.st_atim;
|
|
#else
|
|
d2->stat.st_atime = d1->stat.st_atime;
|
|
#endif
|
|
|
|
return memcmp (&d1->stat, &d2->stat, sizeof (struct stat));
|
|
}
|
|
|
|
int dictstat_init (hashcat_ctx_t *hashcat_ctx)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
folder_config_t *folder_config = hashcat_ctx->folder_config;
|
|
user_options_t *user_options = hashcat_ctx->user_options;
|
|
|
|
dictstat_ctx->enabled = false;
|
|
|
|
if (user_options->benchmark == true) return 0;
|
|
if (user_options->keyspace == true) return 0;
|
|
if (user_options->left == true) return 0;
|
|
if (user_options->opencl_info == true) return 0;
|
|
if (user_options->show == true) return 0;
|
|
if (user_options->usage == true) return 0;
|
|
if (user_options->version == true) return 0;
|
|
|
|
if (user_options->attack_mode == ATTACK_MODE_BF) return 0;
|
|
|
|
dictstat_ctx->enabled = true;
|
|
dictstat_ctx->filename = (char *) hcmalloc (hashcat_ctx, HCBUFSIZ_TINY); VERIFY_PTR (dictstat_ctx->filename);
|
|
dictstat_ctx->base = (dictstat_t *) hccalloc (hashcat_ctx, MAX_DICTSTAT, sizeof (dictstat_t)); VERIFY_PTR (dictstat_ctx->base);
|
|
dictstat_ctx->cnt = 0;
|
|
|
|
snprintf (dictstat_ctx->filename, HCBUFSIZ_TINY - 1, "%s/hashcat.dictstat", folder_config->profile_dir);
|
|
|
|
FILE *fp = fopen (dictstat_ctx->filename, "ab");
|
|
|
|
if (fp == NULL)
|
|
{
|
|
event_log_error (hashcat_ctx, "%s: %s", dictstat_ctx->filename, strerror (errno));
|
|
|
|
return -1;
|
|
}
|
|
|
|
fclose (fp);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void dictstat_destroy (hashcat_ctx_t *hashcat_ctx)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
|
|
if (dictstat_ctx->enabled == false) return;
|
|
|
|
hcfree (dictstat_ctx->filename);
|
|
hcfree (dictstat_ctx->base);
|
|
|
|
memset (dictstat_ctx, 0, sizeof (dictstat_ctx_t));
|
|
}
|
|
|
|
void dictstat_read (hashcat_ctx_t *hashcat_ctx)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
|
|
if (dictstat_ctx->enabled == false) return;
|
|
|
|
FILE *fp = fopen (dictstat_ctx->filename, "rb");
|
|
|
|
if (fp == NULL)
|
|
{
|
|
// first run, file does not exist, do not error out
|
|
|
|
return;
|
|
}
|
|
|
|
while (!feof (fp))
|
|
{
|
|
dictstat_t d;
|
|
|
|
const int nread = fread (&d, sizeof (dictstat_t), 1, fp);
|
|
|
|
if (nread == 0) continue;
|
|
|
|
lsearch (&d, dictstat_ctx->base, &dictstat_ctx->cnt, sizeof (dictstat_t), sort_by_dictstat);
|
|
|
|
if (dictstat_ctx->cnt == MAX_DICTSTAT)
|
|
{
|
|
event_log_error (hashcat_ctx, "There are too many entries in the %s database. You have to remove/rename it.", dictstat_ctx->filename);
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
fclose (fp);
|
|
}
|
|
|
|
int dictstat_write (hashcat_ctx_t *hashcat_ctx)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
|
|
if (dictstat_ctx->enabled == false) return 0;
|
|
|
|
FILE *fp = fopen (dictstat_ctx->filename, "wb");
|
|
|
|
if (fp == NULL)
|
|
{
|
|
event_log_error (hashcat_ctx, "%s: %s", dictstat_ctx->filename, strerror (errno));
|
|
|
|
return -1;
|
|
}
|
|
|
|
fwrite (dictstat_ctx->base, sizeof (dictstat_t), dictstat_ctx->cnt, fp);
|
|
|
|
fclose (fp);
|
|
|
|
return 0;
|
|
}
|
|
|
|
u64 dictstat_find (hashcat_ctx_t *hashcat_ctx, dictstat_t *d)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
|
|
if (dictstat_ctx->enabled == false) return 0;
|
|
|
|
dictstat_t *d_cache = (dictstat_t *) lfind (d, dictstat_ctx->base, &dictstat_ctx->cnt, sizeof (dictstat_t), sort_by_dictstat);
|
|
|
|
if (d_cache == NULL) return 0;
|
|
|
|
return d_cache->cnt;
|
|
}
|
|
|
|
void dictstat_append (hashcat_ctx_t *hashcat_ctx, dictstat_t *d)
|
|
{
|
|
dictstat_ctx_t *dictstat_ctx = hashcat_ctx->dictstat_ctx;
|
|
|
|
if (dictstat_ctx->enabled == false) return;
|
|
|
|
if (dictstat_ctx->cnt == MAX_DICTSTAT)
|
|
{
|
|
event_log_error (hashcat_ctx, "There are too many entries in the %s database. You have to remove/rename it.", dictstat_ctx->filename);
|
|
|
|
return;
|
|
}
|
|
|
|
lsearch (d, dictstat_ctx->base, &dictstat_ctx->cnt, sizeof (dictstat_t), sort_by_dictstat);
|
|
}
|