pe_sieve_api.cpp

Go to the documentation of this file.

#include <windows.h>
#include <string>
#include <iostream>
 
#include "pe_sieve.h"
 
#define PESIEVE_EXPORTS
#include <pe_sieve_api.h>
 
#define LIB_NAME "PE-sieve"
 
using namespace pesieve;
 
size_t print_report(const pesieve::ReportEx& report, const pesieve::t_params args, const t_report_type rtype, char* json_buf, size_t json_buf_size)
{
    if (rtype == REPORT_NONE) return 0;
 
    size_t level = 1;
    std::string report_str = report_to_json(report, rtype, args.results_filter, args.json_lvl, level);
    const size_t report_len = report_str.length();
    if (!report_len) return 0;
 
    const size_t report_size = report_len + 1;// including the '\0' terminator
    if (json_buf && json_buf_size) {
        ::memset(json_buf, 0, json_buf_size);
        size_t max_len = report_len <= (json_buf_size - 1) ? report_len : (json_buf_size - 1);
        ::memcpy(json_buf, report_str.c_str(), max_len);
    }
    return report_size;
}
 
PEsieve_report PESIEVE_API_FUNC PESieve_scan_ex(IN const PEsieve_params *args, IN const PEsieve_rtype rtype, OUT char* json_buf, IN size_t json_buf_size, OUT size_t* needed_size)
{
    if (!args || IsBadReadPtr((LPVOID)args, sizeof(PEsieve_params))) {
        pesieve::t_report empty = { 0 };
        empty.errors = 1;
        return empty;
    }
    const PEsieve_params _args = *args;
    const pesieve::ReportEx* report = pesieve::scan_and_dump(_args);
    pesieve::t_report summary = { 0 };
    summary.pid = _args.pid;
    summary.errors = pesieve::ERROR_SCAN_FAILURE;
    if (!report) {
        return summary;
    }
    if (report->scan_report) {
        summary = report->scan_report->generateSummary();
    }
    //check the pointers:
    if (json_buf) {
        if (!json_buf_size || IsBadWritePtr(json_buf, json_buf_size)) {
            json_buf = nullptr;
            json_buf_size = 0;
        }
    }
    if (needed_size && IsBadWritePtr(needed_size, sizeof(size_t))) {
        needed_size = nullptr;
    }
 
    //print the report (only if any valid output buffer was passed)
    if (json_buf || needed_size) {
        const size_t report_size = print_report(*report, _args, rtype, json_buf, json_buf_size);
        if (needed_size) {
            *needed_size = report_size;
        }
    }
    delete report;
    return summary;
}
 
PEsieve_report PESIEVE_API_FUNC PESieve_scan(IN const PEsieve_params *args)
{
    return PESieve_scan_ex(args, REPORT_NONE, nullptr, 0, nullptr);
}
 
void PESIEVE_API_FUNC PESieve_help(void)
{
    std::string my_info = pesieve::info();
 
    std::cout << my_info;
    MessageBox(NULL, my_info.c_str(), LIB_NAME, MB_ICONINFORMATION);
}
 
extern const DWORD PESIEVE_API PESieve_version = MAKELONG(MAKEWORD(PESIEVE_PATCH_VERSION, PESIEVE_MICRO_VERSION), MAKEWORD(PESIEVE_MINOR_VERSION, PESIEVE_MAJOR_VERSION));