mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 14:48:21 +00:00
c0dd2b4e81
better (than nothing) run-time errors for compiled scripts fixes for dealing with recursive types in compiled scripts fix for values in compiled scripts containing embedded comment markers
89 lines
2.3 KiB
C++
89 lines
2.3 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/script_opt/CPP/Func.h"
|
|
|
|
#include <broker/error.hh>
|
|
|
|
#include "zeek/Desc.h"
|
|
#include "zeek/broker/Data.h"
|
|
|
|
namespace zeek::detail
|
|
{
|
|
|
|
using namespace std;
|
|
|
|
unordered_map<p_hash_type, CompiledScript> compiled_scripts;
|
|
unordered_map<string, unordered_set<p_hash_type>> added_bodies;
|
|
unordered_map<p_hash_type, void (*)()> standalone_callbacks;
|
|
vector<void (*)()> standalone_finalizations;
|
|
|
|
void CPPFunc::Describe(ODesc* d) const
|
|
{
|
|
d->AddSP("compiled function");
|
|
d->Add(name);
|
|
}
|
|
|
|
CPPStmt::CPPStmt(const char* _name, const char* filename, int line_num)
|
|
: Stmt(STMT_CPP), name(_name)
|
|
{
|
|
// We build a fake CallExpr node to be used for error-reporting.
|
|
// It doesn't matter that it matches the actual function/event/hook
|
|
// type-checking-wise, but it *does* need to type-check.
|
|
auto no_args = make_intrusive<RecordType>(nullptr);
|
|
auto no_yield = base_type(TYPE_VOID);
|
|
auto ft = make_intrusive<FuncType>(no_args, no_yield, FUNC_FLAVOR_FUNCTION);
|
|
|
|
vector<StmtPtr> no_bodies;
|
|
vector<int> no_priorities;
|
|
|
|
auto sf = make_intrusive<ScriptFunc>(name, ft, no_bodies, no_priorities);
|
|
auto fv = make_intrusive<FuncVal>(sf);
|
|
auto empty_args = make_intrusive<ListExpr>();
|
|
|
|
ce = make_intrusive<CallExpr>(make_intrusive<ConstExpr>(fv), empty_args);
|
|
Location loc(filename, line_num, line_num, 1, 1);
|
|
ce->SetLocationInfo(&loc);
|
|
}
|
|
|
|
CPPLambdaFunc::CPPLambdaFunc(string _name, FuncTypePtr ft, CPPStmtPtr _l_body)
|
|
: ScriptFunc(move(_name), move(ft), {_l_body}, {0})
|
|
{
|
|
l_body = move(_l_body);
|
|
}
|
|
|
|
broker::expected<broker::data> CPPLambdaFunc::SerializeCaptures() const
|
|
{
|
|
auto vals = l_body->SerializeLambdaCaptures();
|
|
|
|
broker::vector rval;
|
|
rval.emplace_back(string("CopyFrame"));
|
|
|
|
broker::vector body;
|
|
|
|
for ( const auto& val : vals )
|
|
{
|
|
auto expected = Broker::detail::val_to_data(val.get());
|
|
if ( ! expected )
|
|
return broker::ec::invalid_data;
|
|
|
|
TypeTag tag = val->GetType()->Tag();
|
|
broker::vector val_tuple{move(*expected), static_cast<broker::integer>(tag)};
|
|
body.emplace_back(move(val_tuple));
|
|
}
|
|
|
|
rval.emplace_back(move(body));
|
|
|
|
return {move(rval)};
|
|
}
|
|
|
|
void CPPLambdaFunc::SetCaptures(Frame* f)
|
|
{
|
|
l_body->SetLambdaCaptures(f);
|
|
}
|
|
|
|
FuncPtr CPPLambdaFunc::DoClone()
|
|
{
|
|
return make_intrusive<CPPLambdaFunc>(name, type, l_body->Clone());
|
|
}
|
|
|
|
} // zeek::detail
|