mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 14:48:21 +00:00
7b2ee2514e
Idea being for everything to be automatically released when process exits just so there's less potential noise from leak profilers.
65 lines
1.4 KiB
C++
65 lines
1.4 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/Options.h"
|
|
#include "zeek/script_opt/ScriptOpt.h"
|
|
#include "zeek/script_opt/ProfileFunc.h"
|
|
#include "zeek/script_opt/Inline.h"
|
|
|
|
|
|
namespace zeek::detail {
|
|
|
|
|
|
std::unordered_set<const Func*> non_recursive_funcs;
|
|
|
|
// Tracks all of the loaded functions (including event handlers and hooks).
|
|
static std::vector<FuncInfo> funcs;
|
|
|
|
|
|
FuncInfo::FuncInfo(ScriptFuncPtr _func, ScopePtr _scope, StmtPtr _body)
|
|
: func(std::move(_func)), scope(std::move(_scope)), body(std::move(_body))
|
|
{}
|
|
|
|
void FuncInfo::SetProfile(std::unique_ptr<ProfileFunc> _pf)
|
|
{ pf = std::move(_pf); }
|
|
|
|
void analyze_func(ScriptFuncPtr f)
|
|
{
|
|
funcs.emplace_back(f, ScopePtr{NewRef{}, f->GetScope()}, f->CurrentBody());
|
|
}
|
|
|
|
static void check_env_opt(const char* opt, bool& opt_flag)
|
|
{
|
|
if ( getenv(opt) )
|
|
opt_flag = true;
|
|
}
|
|
|
|
void analyze_scripts(Options& opts)
|
|
{
|
|
auto& analysis_options = opts.analysis_options;
|
|
|
|
static bool did_init = false;
|
|
|
|
if ( ! did_init )
|
|
{
|
|
check_env_opt("ZEEK_INLINE", analysis_options.inliner);
|
|
did_init = true;
|
|
}
|
|
|
|
if ( ! analysis_options.inliner )
|
|
return;
|
|
|
|
for ( auto& f : funcs )
|
|
{
|
|
f.SetProfile(std::make_unique<ProfileFunc>(true));
|
|
f.Body()->Traverse(f.Profile());
|
|
}
|
|
|
|
Inliner* inl = nullptr;
|
|
if ( analysis_options.inliner )
|
|
inl = new Inliner(funcs, analysis_options.report_recursive);
|
|
|
|
delete inl;
|
|
}
|
|
|
|
|
|
} // namespace zeek::detail
|