rizin/librz/hash/algorithms/entropy/entropy.c
Giovanni 82e7917294
Refactoring Hash (#1082)
* Added authors and licenses and plugin access to crypto related files
* removed `rz_hash` and related files
* introduced `rz_msg_digest_*` with plugins
* updated all old methods with newer ones.
* added HMAC support to `rz-hash` (moved key option under `-K`)
* removed wrong algorithms on `woD`/`woE`
* fixed parity digest size
* fixed the behaviour on all hash functions
* added unit test for hash
* optimized code for HMAC key calculation
* removed hash legacy tests
2021-05-02 20:47:17 +08:00

40 lines
971 B
C

// SPDX-FileCopyrightText: 2021 deroad <wargio@libero.it>
// SPDX-License-Identifier: LGPL-3.0-only
#include "entropy.h"
#include <stdlib.h>
#include <math.h>
#include <rz_util/rz_assert.h>
bool rz_entropy_init(RzEntropy *ctx) {
rz_return_val_if_fail(ctx, false);
memset(ctx, 0, sizeof(RzEntropy));
return true;
}
bool rz_entropy_update(RzEntropy *ctx, const ut8 *data, size_t len) {
rz_return_val_if_fail(ctx && data, false);
for (size_t i = 0; i < len; i++) {
ctx->count[data[i]]++;
}
ctx->size += len;
return true;
}
bool rz_entropy_final(ut8 *digest, RzEntropy *ctx, bool fraction) {
rz_return_val_if_fail(ctx && digest, false);
double p, entropy = 0.0;
ut64 count;
for (size_t i = 0; i < 256; i++) {
count = ctx->count[i];
if (count) {
p = ((double)count) / ctx->size;
entropy -= p * log2(p);
}
}
if (fraction && ctx->size) {
entropy /= log2((double)RZ_MIN(ctx->size, 256));
}
rz_write_be_double(digest, entropy);
return true;
}