From: Christian Babeux <christian.babeux@efficios.com>
Date: Thu, 6 Sep 2012 21:05:01 +0000 (-0400)
Subject: Fix: Filter ABI changes to support FILTER_BYTECODE_MAX_LEN (65536)
X-Git-Tag: v2.1.0-rc1~4
X-Git-Url: https://git.lttng.org./?a=commitdiff_plain;h=2734ca65d2fb2e63c11d61b9ee8821dd9292aa3c;p=lttng-ust.git

Fix: Filter ABI changes to support FILTER_BYTECODE_MAX_LEN (65536)

In order to support the filter bytecode maximum length (65536 bytes),
the lttng_ust_filter_bytecode len field type must be able to
hold more than a uint16_t. Change the field type to a uint32_t.

Also, since the relocation table is located at the end of the actual
bytecode, the reloc_table_offset (reloc_offset in ust-abi) field must
support offset values larger than 65535. Change the field type to a
uint32_t. This change will allow support of relocation table appended
to larger bytecode without breaking the ABI if the need arise in the
future.

Both changes currently breaks the filter ABI, but this should be a
reasonable compromise since the filtering feature has not been
released yet.

Signed-off-by: Christian Babeux <christian.babeux@efficios.com>
Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
---

diff --git a/include/lttng/ust-abi.h b/include/lttng/ust-abi.h
index 0489e89b..91639a7c 100644
--- a/include/lttng/ust-abi.h
+++ b/include/lttng/ust-abi.h
@@ -174,8 +174,8 @@ struct lttng_ust_calibrate {
 
 #define FILTER_BYTECODE_MAX_LEN		65536
 struct lttng_ust_filter_bytecode {
-	uint16_t len;
-	uint16_t reloc_offset;
+	uint32_t len;
+	uint32_t reloc_offset;
 	char data[0];
 };
 
diff --git a/include/ust-comm.h b/include/ust-comm.h
index b09fcca4..4a3e4cea 100644
--- a/include/ust-comm.h
+++ b/include/ust-comm.h
@@ -132,8 +132,8 @@ struct ustcomm_ust_msg {
 		struct lttng_ust_tracer_version version;
 		struct lttng_ust_tracepoint_iter tracepoint;
 		struct {
-			uint16_t data_size;	/* following filter data */
-			uint16_t reloc_offset;
+			uint32_t data_size;	/* following filter data */
+			uint32_t reloc_offset;
 		} filter;
 	} u;
 };
diff --git a/liblttng-ust/lttng-ust-comm.c b/liblttng-ust/lttng-ust-comm.c
index 842876fb..be64acd0 100644
--- a/liblttng-ust/lttng-ust-comm.c
+++ b/liblttng-ust/lttng-ust-comm.c
@@ -293,6 +293,14 @@ int handle_message(struct sock_info *sock_info,
 			ret = -EINVAL;
 			goto error;
 		}
+
+		if (lum->u.filter.reloc_offset > lum->u.filter.data_size - 1) {
+			ERR("Filter reloc offset %u is not within data\n",
+				lum->u.filter.reloc_offset);
+			ret = -EINVAL;
+			goto error;
+		}
+
 		bytecode = zmalloc(sizeof(*bytecode) + lum->u.filter.data_size);
 		if (!bytecode) {
 			ret = -ENOMEM;