31 lines
826 B
C
31 lines
826 B
C
|
#pragma once
|
||
|
|
#include <istream>
|
||
|
|
#include "pe_base.h"
|
||
|
|
|
||
|
|
namespace pe_bliss
|
||
|
|
{
|
||
|
|
class entropy_calculator
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
//Calculates entropy for PE image section
|
||
|
|
static double calculate_entropy(const section& s);
|
||
|
|
|
||
|
|
//Calculates entropy for istream (from current position of stream)
|
||
|
|
static double calculate_entropy(std::istream& file);
|
||
|
|
|
||
|
|
//Calculates entropy for data block
|
||
|
|
static double calculate_entropy(const char* data, size_t length);
|
||
|
|
|
||
|
|
//Calculates entropy for this PE file (only section data)
|
||
|
|
static double calculate_entropy(const pe_base& pe);
|
||
|
|
|
||
|
|
private:
|
||
|
|
entropy_calculator();
|
||
|
|
entropy_calculator(const entropy_calculator&);
|
||
|
|
entropy_calculator& operator=(const entropy_calculator&);
|
||
|
|
|
||
|
|
//Calculates entropy from bytes count
|
||
|
|
static double calculate_entropy(const uint32_t byte_count[256], std::streamoff total_length);
|
||
|
|
};
|
||
|
|
}
|