Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-14 20:48:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add bitshift operators

Browse source
This commit is contained in:
AmazingPP 2022-07-21 10:59:19 +08:00
parent 1d2c12e980
commit f2bfa61fc6
10 changed files with 57 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/Expr.cc
View file

@ -51,6 +51,8 @@ const char* expr_name(ExprTag t)
"&",
"|",
"^",
"<<",
">>",
"&&",
"||",
"<",
@ -957,6 +959,12 @@ ValPtr BinaryExpr::Fold(Val* v1, Val* v2) const
case EXPR_XOR:
DO_UINT_FOLD(^);
break;
case EXPR_LSHIFT:
DO_INT_FOLD(<<);
break;
case EXPR_RSHIFT:
DO_INT_FOLD(>>);
break;
case EXPR_AND_AND:
DO_INT_FOLD(&&);
@ -2156,7 +2164,22 @@ BitExpr::BitExpr(ExprTag arg_tag, ExprPtr arg_op1, ExprPtr arg_op2)
if ( IsVector(bt2) )
bt2 = t2->AsVectorType()->Yield()->Tag();
if ( (bt1 == TYPE_COUNT) && (bt2 == TYPE_COUNT) )
if ( tag == EXPR_LSHIFT || tag == EXPR_RSHIFT )
{
if ( IsIntegral(bt1) && bt2 == TYPE_COUNT )
{
if ( is_vector(op1) || is_vector(op2) )
SetType(make_intrusive<VectorType>(base_type(bt1)));
else
SetType(base_type(bt1));
}
else if ( IsIntegral(bt1) && bt2 == TYPE_INT )
ExprError("requires \"count\" right operand");
else
ExprError("requires integral operands");
}
else if ( (bt1 == TYPE_COUNT) && (bt2 == TYPE_COUNT) )
{
if ( is_vector(op1) || is_vector(op2) )
SetType(make_intrusive<VectorType>(base_type(TYPE_COUNT)));