zeek/src/ZVal.cc

// See the file "COPYING" in the main distribution directory for copyright.

#include "zeek/File.h"
#include "zeek/Func.h"
#include "zeek/OpaqueVal.h"
#include "zeek/Reporter.h"
#include "zeek/ZeekString.h"

using namespace zeek;

bool* ZVal::zval_was_nil_addr = nullptr;

ZVal::ZVal(ValPtr v, const TypePtr& t)
	{
	if ( ! v )
		{
		// This can happen for some forms of error propagation.
		// We can deal with it iff the type is managed, and thus
		// we can employ a "nil" placeholder.
		ASSERT(IsManagedType(t));
		managed_val = nullptr;
		return;
		}

	const auto& vt = v->GetType();

	if ( vt->Tag() != t->Tag() && t->Tag() != TYPE_ANY )
		{
		if ( t->InternalType() == TYPE_INTERNAL_OTHER || t->InternalType() != vt->InternalType() )
			reporter->InternalError("type inconsistency in ZVal constructor");
		}

	switch ( t->Tag() )
		{
		case TYPE_BOOL:
		case TYPE_INT:
		case TYPE_ENUM:
			int_val = v->AsInt();
			break;

		case TYPE_COUNT:
		case TYPE_PORT:
			uint_val = v->AsCount();
			break;

		case TYPE_DOUBLE:
		case TYPE_INTERVAL:
		case TYPE_TIME:
			double_val = v->AsDouble();
			break;

		case TYPE_FUNC:
			func_val = v->AsFunc();
			Ref(func_val);
			break;

		case TYPE_FILE:
			file_val = v->AsFile();
			Ref(file_val);
			break;

		case TYPE_LIST:
			list_val = v.release()->AsListVal();
			break;

		case TYPE_OPAQUE:
			opaque_val = v.release()->AsOpaqueVal();
			break;

		case TYPE_PATTERN:
			re_val = v.release()->AsPatternVal();
			break;

		case TYPE_TABLE:
			table_val = v.release()->AsTableVal();
			break;

		case TYPE_VECTOR:
			vector_val = v.release()->AsVectorVal();
			break;

		case TYPE_RECORD:
			record_val = v.release()->AsRecordVal();
			break;

		case TYPE_STRING:
			string_val = v.release()->AsStringVal();
			break;

		case TYPE_ADDR:
			addr_val = v.release()->AsAddrVal();
			break;

		case TYPE_SUBNET:
			subnet_val = v.release()->AsSubNetVal();
			break;

		case TYPE_ANY:
			any_val = v.release();
			break;

		case TYPE_TYPE:
			type_val = v.release()->AsTypeVal();
			break;

		case TYPE_ERROR:
		case TYPE_TIMER:
		case TYPE_UNION:
		case TYPE_VOID:
			reporter->InternalError("bad type in ZVal constructor");
		}
	}

ZVal::ZVal(const TypePtr& t)
	{
	switch ( t->Tag() )
		{
		case TYPE_BOOL:
		case TYPE_INT:
		case TYPE_ENUM:
			int_val = 0;
			break;

		case TYPE_COUNT:
		case TYPE_PORT:
			uint_val = 0;
			break;

		case TYPE_DOUBLE:
		case TYPE_INTERVAL:
		case TYPE_TIME:
			double_val = 0.0;
			break;

		case TYPE_FUNC:
			func_val = nullptr;
			break;

		case TYPE_FILE:
			file_val = nullptr;
			break;

		case TYPE_LIST:
			list_val = nullptr;
			break;

		case TYPE_OPAQUE:
			opaque_val = nullptr;
			break;

		case TYPE_PATTERN:
			re_val = nullptr;
			break;

		case TYPE_TABLE:
			table_val = nullptr;
			break;

		case TYPE_VECTOR:
			vector_val = nullptr;
			break;

		case TYPE_RECORD:
			record_val = nullptr;
			break;

		case TYPE_STRING:
			string_val = nullptr;
			break;

		case TYPE_ADDR:
			addr_val = nullptr;
			break;

		case TYPE_SUBNET:
			subnet_val = nullptr;
			break;

		case TYPE_ANY:
			any_val = nullptr;
			break;

		case TYPE_TYPE:
			type_val = nullptr;
			break;

		case TYPE_ERROR:
		case TYPE_TIMER:
		case TYPE_UNION:
		case TYPE_VOID:
			reporter->InternalError("bad type in ZVal constructor");
		}
	}

ValPtr ZVal::ToVal(const TypePtr& t) const
	{
	Val* v;

	switch ( t->Tag() )
		{
		case TYPE_INT:
			return val_mgr->Int(int_val);

		case TYPE_BOOL:
			return val_mgr->Bool(int_val ? true : false);

		case TYPE_PORT:
			return val_mgr->Port(uint_val);

		case TYPE_COUNT:
			return val_mgr->Count(uint_val);

		case TYPE_DOUBLE:
			return make_intrusive<DoubleVal>(double_val);

		case TYPE_INTERVAL:
			return make_intrusive<IntervalVal>(double_val, Seconds);

		case TYPE_TIME:
			return make_intrusive<TimeVal>(double_val);

		case TYPE_ENUM:
			return t->AsEnumType()->GetEnumVal(int_val);

		case TYPE_FUNC:
			if ( func_val )
				{
				FuncPtr fv_ptr = {NewRef{}, func_val};
				return make_intrusive<FuncVal>(fv_ptr);
				}

			return nullptr;

		case TYPE_FILE:
			if ( file_val )
				{
				FilePtr fv_ptr = {NewRef{}, file_val};
				return make_intrusive<FileVal>(fv_ptr);
				}

			return nullptr;

		case TYPE_ADDR:
			v = addr_val;
			break;
		case TYPE_SUBNET:
			v = subnet_val;
			break;
		case TYPE_STRING:
			v = string_val;
			break;
		case TYPE_LIST:
			v = list_val;
			break;
		case TYPE_OPAQUE:
			v = opaque_val;
			break;
		case TYPE_TABLE:
			v = table_val;
			break;
		case TYPE_RECORD:
			v = record_val;
			break;
		case TYPE_VECTOR:
			v = vector_val;
			break;
		case TYPE_PATTERN:
			v = re_val;
			break;
		case TYPE_ANY:
			v = any_val;
			break;
		case TYPE_TYPE:
			v = type_val;
			break;

		case TYPE_ERROR:
		case TYPE_TIMER:
		case TYPE_UNION:
		case TYPE_VOID:
		default:
			v = nullptr;
			reporter->InternalError("bad type in ZVal::ToVal: %s", type_name(t->Tag()));
		}

	if ( v )
		return {NewRef{}, v};

	if ( zval_was_nil_addr )
		*zval_was_nil_addr = true;

	return nullptr;
	}

bool ZVal::IsManagedType(const TypePtr& t)
	{
	switch ( t->Tag() )
		{
		case zeek::TYPE_ADDR:
		case zeek::TYPE_ANY:
		case zeek::TYPE_FILE:
		case zeek::TYPE_FUNC:
		case zeek::TYPE_LIST:
		case zeek::TYPE_OPAQUE:
		case zeek::TYPE_PATTERN:
		case zeek::TYPE_RECORD:
		case zeek::TYPE_STRING:
		case zeek::TYPE_SUBNET:
		case zeek::TYPE_TABLE:
		case zeek::TYPE_TYPE:
		case zeek::TYPE_VECTOR:
			return true;

		default:
			return false;
		}
	}