X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=liblttng-ust%2Flttng-filter-specialize.c;h=d115585d8c9a1cc85b482a72bf981e5cad031ffa;hb=6548fca464f68be5ded73be6b9a479daf08e62d6;hp=ef0d811bf5c342418a146ed52799694047857183;hpb=0305960f8200d1db7002b75d9a5c3ea37541e2c5;p=lttng-ust.git

diff --git a/liblttng-ust/lttng-filter-specialize.c b/liblttng-ust/lttng-filter-specialize.c
index ef0d811b..d115585d 100644
--- a/liblttng-ust/lttng-filter-specialize.c
+++ b/liblttng-ust/lttng-filter-specialize.c
@@ -20,6 +20,7 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
+#define _LGPL_SOURCE
 #include "lttng-filter.h"
 
 int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
@@ -73,17 +74,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_EQ_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_EQ_S64;
 				else
-					insn->op = FILTER_OP_EQ_DOUBLE;
+					insn->op = FILTER_OP_EQ_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_EQ_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_EQ_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_EQ_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -106,17 +118,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_NE_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_NE_S64;
 				else
-					insn->op = FILTER_OP_NE_DOUBLE;
+					insn->op = FILTER_OP_NE_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_NE_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_NE_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_NE_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -139,17 +162,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_GT_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_GT_S64;
 				else
-					insn->op = FILTER_OP_GT_DOUBLE;
+					insn->op = FILTER_OP_GT_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_GT_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_GT_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_GT_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -172,17 +206,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_LT_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_LT_S64;
 				else
-					insn->op = FILTER_OP_LT_DOUBLE;
+					insn->op = FILTER_OP_LT_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_LT_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_LT_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_LT_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -205,17 +250,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_GE_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_GE_S64;
 				else
-					insn->op = FILTER_OP_GE_DOUBLE;
+					insn->op = FILTER_OP_GE_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_GE_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_GE_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_GE_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -237,17 +293,28 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 				goto end;
 
 			case REG_STRING:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				insn->op = FILTER_OP_LE_STRING;
 				break;
 			case REG_S64:
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
 				if (vstack_bx(stack)->type == REG_S64)
 					insn->op = FILTER_OP_LE_S64;
 				else
-					insn->op = FILTER_OP_LE_DOUBLE;
+					insn->op = FILTER_OP_LE_DOUBLE_S64;
 				break;
 			case REG_DOUBLE:
-				insn->op = FILTER_OP_LE_DOUBLE;
+				if (vstack_bx(stack)->type == REG_UNKNOWN)
+					break;
+				if (vstack_bx(stack)->type == REG_S64)
+					insn->op = FILTER_OP_LE_S64_DOUBLE;
+				else
+					insn->op = FILTER_OP_LE_DOUBLE;
 				break;
+			case REG_UNKNOWN:
+				break;	/* Dynamic typing. */
 			}
 			vstack_ax(stack)->type = REG_S64;
 			next_pc += sizeof(struct binary_op);
@@ -272,6 +339,18 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 		case FILTER_OP_LT_DOUBLE:
 		case FILTER_OP_GE_DOUBLE:
 		case FILTER_OP_LE_DOUBLE:
+		case FILTER_OP_EQ_DOUBLE_S64:
+		case FILTER_OP_NE_DOUBLE_S64:
+		case FILTER_OP_GT_DOUBLE_S64:
+		case FILTER_OP_LT_DOUBLE_S64:
+		case FILTER_OP_GE_DOUBLE_S64:
+		case FILTER_OP_LE_DOUBLE_S64:
+		case FILTER_OP_EQ_S64_DOUBLE:
+		case FILTER_OP_NE_S64_DOUBLE:
+		case FILTER_OP_GT_S64_DOUBLE:
+		case FILTER_OP_LT_S64_DOUBLE:
+		case FILTER_OP_GE_S64_DOUBLE:
+		case FILTER_OP_LE_S64_DOUBLE:
 		{
 			/* Pop 2, push 1 */
 			if (vstack_pop(stack)) {
@@ -300,6 +379,8 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			case REG_DOUBLE:
 				insn->op = FILTER_OP_UNARY_PLUS_DOUBLE;
 				break;
+			case REG_UNKNOWN:	/* Dynamic typing. */
+				break;
 			}
 			/* Pop 1, push 1 */
 			next_pc += sizeof(struct unary_op);
@@ -322,6 +403,8 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			case REG_DOUBLE:
 				insn->op = FILTER_OP_UNARY_MINUS_DOUBLE;
 				break;
+			case REG_UNKNOWN:	/* Dynamic typing. */
+				break;
 			}
 			/* Pop 1, push 1 */
 			next_pc += sizeof(struct unary_op);
@@ -344,6 +427,8 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			case REG_DOUBLE:
 				insn->op = FILTER_OP_UNARY_NOT_DOUBLE;
 				break;
+			case REG_UNKNOWN:	/* Dynamic typing. */
+				break;
 			}
 			/* Pop 1, push 1 */
 			next_pc += sizeof(struct unary_op);
@@ -366,19 +451,37 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 		case FILTER_OP_AND:
 		case FILTER_OP_OR:
 		{
+			/* Continue to next instruction */
+			/* Pop 1 when jump not taken */
+			if (vstack_pop(stack)) {
+				ret = -EINVAL;
+				goto end;
+			}
 			next_pc += sizeof(struct logical_op);
 			break;
 		}
 
-		/* load */
+		/* load field ref */
 		case FILTER_OP_LOAD_FIELD_REF:
 		{
 			ERR("Unknown field ref type\n");
 			ret = -EINVAL;
 			goto end;
 		}
+		/* get context ref */
+		case FILTER_OP_GET_CONTEXT_REF:
+		{
+			if (vstack_push(stack)) {
+				ret = -EINVAL;
+				goto end;
+			}
+			vstack_ax(stack)->type = REG_UNKNOWN;
+			next_pc += sizeof(struct load_op) + sizeof(struct field_ref);
+			break;
+		}
 		case FILTER_OP_LOAD_FIELD_REF_STRING:
 		case FILTER_OP_LOAD_FIELD_REF_SEQUENCE:
+		case FILTER_OP_GET_CONTEXT_REF_STRING:
 		{
 			if (vstack_push(stack)) {
 				ret = -EINVAL;
@@ -389,6 +492,7 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			break;
 		}
 		case FILTER_OP_LOAD_FIELD_REF_S64:
+		case FILTER_OP_GET_CONTEXT_REF_S64:
 		{
 			if (vstack_push(stack)) {
 				ret = -EINVAL;
@@ -399,6 +503,7 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			break;
 		}
 		case FILTER_OP_LOAD_FIELD_REF_DOUBLE:
+		case FILTER_OP_GET_CONTEXT_REF_DOUBLE:
 		{
 			if (vstack_push(stack)) {
 				ret = -EINVAL;
@@ -409,6 +514,7 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			break;
 		}
 
+		/* load from immediate operand */
 		case FILTER_OP_LOAD_STRING:
 		{
 			struct load_op *insn = (struct load_op *) pc;
@@ -467,6 +573,8 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			case REG_DOUBLE:
 				insn->op = FILTER_OP_CAST_DOUBLE_TO_S64;
 				break;
+			case REG_UNKNOWN:
+				break;
 			}
 			/* Pop 1, push 1 */
 			vstack_ax(stack)->type = REG_S64;
@@ -486,7 +594,6 @@ int lttng_filter_specialize_bytecode(struct bytecode_runtime *bytecode)
 			break;
 		}
 
-
 		}
 	}
 end: