PostCallAnalysis.cpp

#include "PostCallAnalysis.h"
#include "BaseAnalysis.h"
#include "DynamicAnalysis.h"
#include "../DynamicUtils.h"

#include <vector>

void PostCallAnalysis::SharedInit(void const* _func_supplier, const char* _target_str, CallParam_t *_params, int64_t num_params) {
    func_supplier = _func_supplier;
    target_str = _target_str;
    for (int i = 0; i < num_params; i++) {
        params.push_back(&_params[i]);
    }
}

PostCallAnalysis::PostCallAnalysis(void const* _func_supplier, CallOp_t* callop) {
    SharedInit(_func_supplier, callop->function_name, callop->params, callop->num_params);
    target_funcs = {callop->target_function};
}
PostCallAnalysis::PostCallAnalysis(void const* _func_supplier, CallTagOp_t* callop) {
    SharedInit(_func_supplier, callop->target_tag, callop->params, callop->num_params);
    target_funcs = DynamicUtils::getFunctionsForTag(callop->target_tag);
}

Fulfillment PostCallAnalysis::functionPreCBImpl(void* const& func, CallsiteInfo const& callsite) {
    for (void const* const& target_func : target_funcs) {
        if (target_func == func) {
            // Target function found, maybe analysis success

            // Check params if needed
            if (params.empty()) {
                uncheckedCallsites.clear();
                return Fulfillment::UNKNOWN; // Cannot return fulfilled until program exit, there may be more callsites to come
            }

            // Check which callsites are satisfied, remove from unchecked
            for (auto callsite_iter = uncheckedCallsites.begin(); callsite_iter != uncheckedCallsites.end();) {
                if (DynamicUtils::checkFuncCallMatch(target_func, params, callsite, *callsite_iter, target_str)) {
                    callsite_iter = uncheckedCallsites.erase(callsite_iter);
                } else {
                    callsite_iter++;
                }
            }
            // For the rest: Maybe actual fulfillment comes later
            return Fulfillment::UNKNOWN;
        }
    }

    if (func == func_supplier) {
        for (CallsiteInfo& uncheckedCallsite : uncheckedCallsites) {
            if (uncheckedCallsite.location == callsite.location) {
                uncheckedCallsite = callsite;
                goto exit_postcall_funccb;
            }
        }
        uncheckedCallsites.push_back(callsite);
    }

    exit_postcall_funccb:
    // Irrelevant function
    return Fulfillment::UNKNOWN;
}

Fulfillment PostCallAnalysis::exitCBImpl(CodePtr const& location) {
    for (CallsiteInfo const& callsite : uncheckedCallsites) {
        references.push_back(callsite.location);
    }
    return uncheckedCallsites.empty() ? Fulfillment::FULFILLED : Fulfillment::VIOLATED;
}