#!/bin/bash
#
# Copyright (C) - 2012 Christian Babeux <christian.babeux@efficios.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License, version 2 only, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

TEST_DESC="Filtering - Invalid filters"

CURDIR=$(dirname $0)/
TESTDIR=$CURDIR/../..
LTTNG_BIN="lttng"
SESSION_NAME="filter-invalid"
EVENT_NAME="bogus"
ENABLE_EVENT_STDERR="/tmp/invalid-filters-stderr"
TRACE_PATH=$(mktemp -d)

source $TESTDIR/utils.sh

print_test_banner "$TEST_DESC"

function enable_ust_lttng_event_filter
{
	sess_name="$1"
	event_name="$2"
	filter="$3"
	echo -n "Enabling lttng event with filtering and invalid filter "

	$TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event $event_name -s $sess_name -u --filter "$filter" 2> $ENABLE_EVENT_STDERR 1> /dev/null

	# Enable must fail
	if [ $? -eq 0 ]; then
		print_fail
		return 1
	else
		print_ok
		return 0
	fi
}

function test_invalid_filter
{
	test_invalid_filter="$1"

	echo ""
	echo -e "=== Testing filter expression with invalid filter"
	echo -e "Filter: $test_invalid_filter"

	# Create session
	create_lttng_session $SESSION_NAME $TRACE_PATH

	# Apply filter
	enable_ust_lttng_event_filter $SESSION_NAME $EVENT_NAME "$test_invalid_filter"

	# Destroy session
	destroy_lttng_session $SESSION_NAME
}

function test_bytecode_limit
{
	# Current bytecode limitation is 65536 bytes long.
	# Generate a huge bytecode with some perl-fu
	BYTECODE_LIMIT=`perl -e 'print "intfield" . " && 1" x5460'`

	echo ""
	echo -e "=== Testing filter bytecode limits (64KiB)"

	# Create session
	create_lttng_session $SESSION_NAME $TRACE_PATH

	# Apply filter
	enable_ust_lttng_event_filter $SESSION_NAME $EVENT_NAME "$BYTECODE_LIMIT"

	# Destroy session
	destroy_lttng_session $SESSION_NAME
}

IFS=$'\n'
INVALID_FILTERS=(
		# Unsupported ops
		"intfield*1"
		"intfield/1"
		"intfield+1"
		"intfield-1"
		"intfield>>1"
		"intfield<<1"
		"intfield&1"
		"intfield|1"
		"intfield^1"
		"~intfield"
		"1+11111-3333+1"
		"(1+2)*(55*666)"
		"1+2*55*666"
		"asdf + 1 > 1"
		"asdfas < 2332 || asdf + 1 > 1"
		"!+-+++-------+++++++++++-----!!--!44+1"
		"aaa||(gg)+(333----1)"
		"1+1"
		# Unmatched parenthesis
		"((((((((((((((intfield)))))))))))))"
		'0 || ("abc" != "def")) && (3 < 4)'
		# Field dereference
		"a.a.a.a.a.a.a.a.a.a.a.a.a.a.a.a.a.a.a.a"
		"a->"
		"a-->a"
		"a->a"
		"a.b.c->d.e.f+1"
		"!a.f.d"
		"asdf.asdfsd.sadf < 4"
		"asdfasdf->asdfasdf < 2"
		# String can't be root node
		"\"somestring\""
		# Unary op on string not allowed
		"!\"somestring\""
		# Comparison with string type not allowed
		"\"somestring\" > 42"
		"\"somestring\" > 42.0"
		"42 > \"somestring\""
		"42.0 > \"somestring\""
		# Logical operator with string type not allowed
		"\"somestring\" || 1"
		"1 || \"somestring\""
		# Nesting of binary operator not allowed
		"1 | (1 | (1 | 1))"
		"1 > (1 > (1 > 1))"
		)

start_lttng_sessiond
for FILTER in ${INVALID_FILTERS[@]};
do
	test_invalid_filter "$FILTER"
done

test_bytecode_limit

unset IFS
stop_lttng_sessiond

rm -f $ENABLE_EVENT_STDERR
rm -rf $TRACE_PATH
