mirror of
https://github.com/zeek/zeek.git
synced 2025-10-09 18:18:19 +00:00
Implement from_json bif
This commit is contained in:
Fupeng Zhao
mAsk°
parent
b739bdbbfb
commit
584e68434d
27 changed files with 531 additions and 1 deletions
301
src/Val.cc
301
src/Val.cc
|
|
@ -6,6 +6,8 @@
|
|||
|
||||
#include <netdb.h>
|
||||
#include <netinet/in.h>
|
||||
#include <rapidjson/document.h>
|
||||
#include <rapidjson/error/en.h>
|
||||
#include <sys/param.h>
|
||||
#include <sys/types.h>
|
||||
#include <unistd.h>
|
||||
|
|
@ -1061,6 +1063,305 @@ StringValPtr StringVal::Replace(RE_Matcher* re, const String& repl, bool do_all)
|
|||
return make_intrusive<StringVal>(new String(true, result, r - result));
|
||||
}
|
||||
|
||||
static std::variant<ValPtr, std::string> BuildVal(const rapidjson::Value& j, const TypePtr& t)
|
||||
{
|
||||
if ( j.IsNull() )
|
||||
return Val::nil;
|
||||
|
||||
switch ( t->Tag() )
|
||||
{
|
||||
case TYPE_BOOL:
|
||||
{
|
||||
if ( ! j.IsBool() )
|
||||
goto mismatch_err;
|
||||
|
||||
return val_mgr->Bool(j.GetBool());
|
||||
}
|
||||
|
||||
case TYPE_INT:
|
||||
{
|
||||
if ( ! j.IsInt64() )
|
||||
goto mismatch_err;
|
||||
|
||||
return val_mgr->Int(j.GetInt64());
|
||||
}
|
||||
|
||||
case TYPE_COUNT:
|
||||
{
|
||||
if ( ! j.IsUint64() )
|
||||
goto mismatch_err;
|
||||
|
||||
return val_mgr->Count(j.GetUint64());
|
||||
}
|
||||
|
||||
case TYPE_TIME:
|
||||
{
|
||||
if ( ! j.IsNumber() )
|
||||
goto mismatch_err;
|
||||
|
||||
return make_intrusive<TimeVal>(j.GetDouble());
|
||||
}
|
||||
|
||||
case TYPE_DOUBLE:
|
||||
{
|
||||
if ( ! j.IsNumber() )
|
||||
goto mismatch_err;
|
||||
|
||||
return make_intrusive<DoubleVal>(j.GetDouble());
|
||||
}
|
||||
|
||||
case TYPE_INTERVAL:
|
||||
{
|
||||
if ( ! j.IsNumber() )
|
||||
goto mismatch_err;
|
||||
|
||||
return make_intrusive<IntervalVal>(j.GetDouble());
|
||||
}
|
||||
|
||||
case TYPE_PORT:
|
||||
{
|
||||
if ( ! j.IsString() )
|
||||
goto mismatch_err;
|
||||
|
||||
int port = 0;
|
||||
if ( j.GetStringLength() > 0 && j.GetStringLength() < 10 )
|
||||
{
|
||||
char* slash;
|
||||
errno = 0;
|
||||
port = strtol(j.GetString(), &slash, 10);
|
||||
if ( ! errno )
|
||||
{
|
||||
++slash;
|
||||
if ( util::streq(slash, "tcp") )
|
||||
return val_mgr->Port(port, TRANSPORT_TCP);
|
||||
else if ( util::streq(slash, "udp") )
|
||||
return val_mgr->Port(port, TRANSPORT_UDP);
|
||||
else if ( util::streq(slash, "icmp") )
|
||||
return val_mgr->Port(port, TRANSPORT_ICMP);
|
||||
else if ( util::streq(slash, "unknown") )
|
||||
return val_mgr->Port(port, TRANSPORT_UNKNOWN);
|
||||
}
|
||||
}
|
||||
|
||||
return "wrong port format, must be /[0-9]{1,5}\\/(tcp|udp|icmp|unknown)/";
|
||||
}
|
||||
|
||||
case TYPE_PATTERN:
|
||||
{
|
||||
if ( ! j.IsString() )
|
||||
goto mismatch_err;
|
||||
|
||||
std::string candidate(j.GetString(), j.GetStringLength());
|
||||
if ( candidate.size() > 2 && candidate.front() == candidate.back() &&
|
||||
candidate.back() == '/' )
|
||||
{
|
||||
// Remove the '/'s
|
||||
candidate.erase(0, 1);
|
||||
candidate.erase(candidate.size() - 1);
|
||||
}
|
||||
|
||||
auto re = std::make_unique<RE_Matcher>(candidate.c_str());
|
||||
if ( ! re->Compile() )
|
||||
return "error compiling pattern";
|
||||
|
||||
return make_intrusive<PatternVal>(re.release());
|
||||
}
|
||||
|
||||
case TYPE_ADDR:
|
||||
case TYPE_SUBNET:
|
||||
{
|
||||
if ( ! j.IsString() )
|
||||
goto mismatch_err;
|
||||
|
||||
int width = 0;
|
||||
std::string candidate;
|
||||
|
||||
if ( t->Tag() == TYPE_ADDR )
|
||||
candidate = std::string(j.GetString(), j.GetStringLength());
|
||||
else
|
||||
{
|
||||
std::string_view subnet_sv(j.GetString(), j.GetStringLength());
|
||||
auto pos = subnet_sv.find('/');
|
||||
if ( pos == subnet_sv.npos )
|
||||
return util::fmt("Invalid value for subnet: %s", j.GetString());
|
||||
|
||||
candidate = std::string(j.GetString(), pos);
|
||||
|
||||
errno = 0;
|
||||
char* end;
|
||||
width = strtol(subnet_sv.data() + pos + 1, &end, 10);
|
||||
if ( subnet_sv.data() + pos + 1 == end || errno )
|
||||
return util::fmt("Invalid value for subnet: %s", j.GetString());
|
||||
}
|
||||
|
||||
if ( candidate.front() == '[' )
|
||||
candidate.erase(0, 1);
|
||||
if ( candidate.back() == ']' )
|
||||
candidate.erase(candidate.size() - 1);
|
||||
|
||||
if ( t->Tag() == TYPE_ADDR )
|
||||
return make_intrusive<AddrVal>(candidate);
|
||||
else
|
||||
return make_intrusive<SubNetVal>(candidate.c_str(), width);
|
||||
}
|
||||
|
||||
case TYPE_ENUM:
|
||||
{
|
||||
if ( ! j.IsString() )
|
||||
goto mismatch_err;
|
||||
|
||||
auto et = t->AsEnumType();
|
||||
auto intval = et->Lookup({j.GetString(), j.GetStringLength()});
|
||||
|
||||
if ( intval < 0 )
|
||||
return util::fmt("'%s' is not a valid enum for '%s'.", j.GetString(),
|
||||
et->GetName().c_str());
|
||||
|
||||
return et->GetEnumVal(intval);
|
||||
}
|
||||
|
||||
case TYPE_STRING:
|
||||
{
|
||||
if ( ! j.IsString() )
|
||||
goto mismatch_err;
|
||||
|
||||
return make_intrusive<StringVal>(j.GetStringLength(), j.GetString());
|
||||
}
|
||||
|
||||
case TYPE_TABLE:
|
||||
{
|
||||
if ( ! j.IsArray() )
|
||||
goto mismatch_err;
|
||||
|
||||
if ( ! t->IsSet() )
|
||||
goto unsupport_err;
|
||||
|
||||
auto tt = t->AsSetType();
|
||||
auto tl = tt->GetIndices();
|
||||
auto tv = make_intrusive<TableVal>(IntrusivePtr{NewRef{}, tt});
|
||||
|
||||
for ( const auto& item : j.GetArray() )
|
||||
{
|
||||
std::variant<ValPtr, std::string> v;
|
||||
|
||||
if ( tl->GetTypes().size() == 1 )
|
||||
v = BuildVal(item, tl->GetPureType());
|
||||
else
|
||||
v = BuildVal(item, tl);
|
||||
|
||||
if ( ! get_if<ValPtr>(&v) )
|
||||
return v;
|
||||
|
||||
if ( ! std::get<ValPtr>(v) )
|
||||
continue;
|
||||
|
||||
tv->Assign(std::move(std::get<ValPtr>(v)), nullptr);
|
||||
}
|
||||
|
||||
return tv;
|
||||
}
|
||||
|
||||
case TYPE_RECORD:
|
||||
{
|
||||
if ( ! j.IsObject() )
|
||||
goto mismatch_err;
|
||||
|
||||
auto rt = t->AsRecordType();
|
||||
auto rv = make_intrusive<RecordVal>(IntrusivePtr{NewRef{}, rt});
|
||||
for ( int i = 0; i < rt->NumFields(); ++i )
|
||||
{
|
||||
auto td_i = rt->FieldDecl(i);
|
||||
auto m_it = j.FindMember(td_i->id);
|
||||
bool has_member = m_it != j.MemberEnd();
|
||||
bool member_is_null = has_member ? m_it->value.IsNull() : true;
|
||||
|
||||
if ( ! has_member || member_is_null )
|
||||
{
|
||||
if ( ! td_i->GetAttr(detail::ATTR_OPTIONAL) &&
|
||||
! td_i->GetAttr(detail::ATTR_DEFAULT) )
|
||||
return util::fmt("Record '%s' field '%s' is null or missing",
|
||||
t->GetName().c_str(), td_i->id);
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
auto v = BuildVal(m_it->value, td_i->type);
|
||||
if ( ! get_if<ValPtr>(&v) )
|
||||
return v;
|
||||
|
||||
rv->Assign(i, std::move(std::get<ValPtr>(v)));
|
||||
}
|
||||
|
||||
return rv;
|
||||
}
|
||||
|
||||
case TYPE_LIST:
|
||||
{
|
||||
if ( ! j.IsArray() )
|
||||
goto mismatch_err;
|
||||
|
||||
auto lt = t->AsTypeList();
|
||||
|
||||
if ( j.GetArray().Size() < lt->GetTypes().size() )
|
||||
return "index type doesn't match";
|
||||
|
||||
auto lv = make_intrusive<ListVal>(TYPE_ANY);
|
||||
|
||||
for ( size_t i = 0; i < lt->GetTypes().size(); i++ )
|
||||
{
|
||||
auto v = BuildVal(j.GetArray()[i], lt->GetTypes()[i]);
|
||||
if ( ! get_if<ValPtr>(&v) )
|
||||
return v;
|
||||
|
||||
lv->Append(std::move(std::get<ValPtr>(v)));
|
||||
}
|
||||
|
||||
return lv;
|
||||
}
|
||||
|
||||
case TYPE_VECTOR:
|
||||
{
|
||||
if ( ! j.IsArray() )
|
||||
goto mismatch_err;
|
||||
|
||||
auto vt = t->AsVectorType();
|
||||
auto vv = make_intrusive<VectorVal>(IntrusivePtr{NewRef{}, vt});
|
||||
for ( const auto& item : j.GetArray() )
|
||||
{
|
||||
auto v = BuildVal(item, vt->Yield());
|
||||
if ( ! get_if<ValPtr>(&v) )
|
||||
return v;
|
||||
|
||||
if ( ! std::get<ValPtr>(v) )
|
||||
continue;
|
||||
|
||||
vv->Assign(vv->Size(), std::move(std::get<ValPtr>(v)));
|
||||
}
|
||||
|
||||
return vv;
|
||||
}
|
||||
|
||||
default:
|
||||
unsupport_err:
|
||||
return util::fmt("type '%s' unsupport", type_name(t->Tag()));
|
||||
}
|
||||
|
||||
mismatch_err:
|
||||
return util::fmt("type '%s' mismatch", type_name(t->Tag()));
|
||||
}
|
||||
|
||||
std::variant<ValPtr, std::string> ValFromJSON(std::string_view json_str, const TypePtr& t)
|
||||
{
|
||||
rapidjson::Document doc;
|
||||
rapidjson::ParseResult ok = doc.Parse(json_str.data(), json_str.length());
|
||||
|
||||
if ( ! ok )
|
||||
return util::fmt("JSON parse error: %s Offset: %lu", rapidjson::GetParseError_En(ok.Code()),
|
||||
ok.Offset());
|
||||
|
||||
return BuildVal(doc, t);
|
||||
}
|
||||
|
||||
ValPtr StringVal::DoClone(CloneState* state)
|
||||
{
|
||||
// We could likely treat this type as immutable and return a reference
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue