X-Git-Url: http://git.lttng.org/?p=lttng-modules.git;a=blobdiff_plain;f=lttng-context.c;h=bed3d1c24602fde4207c4764d1cf8d6bff168121;hp=406f479de92994217b2c1e5c480086752b173e27;hb=ea15538d675ad5685c1e4ce1c393f25f682eb06b;hpb=d0e59d6f099f4663cc62bffb59d960cd4235064f

diff --git a/lttng-context.c b/lttng-context.c
index 406f479d..bed3d1c2 100644
--- a/lttng-context.c
+++ b/lttng-context.c
@@ -95,12 +95,12 @@ struct lttng_ctx_field *lttng_append_context(struct lttng_ctx **ctx_p)
 		struct lttng_ctx_field *new_fields;
 
 		ctx->allocated_fields = max_t(size_t, 1, 2 * ctx->allocated_fields);
-		new_fields = kzalloc(ctx->allocated_fields * sizeof(struct lttng_ctx_field), GFP_KERNEL);
+		new_fields = lttng_kvzalloc(ctx->allocated_fields * sizeof(struct lttng_ctx_field), GFP_KERNEL);
 		if (!new_fields)
 			return NULL;
 		if (ctx->fields)
 			memcpy(new_fields, ctx->fields, sizeof(*ctx->fields) * ctx->nr_fields);
-		kfree(ctx->fields);
+		lttng_kvfree(ctx->fields);
 		ctx->fields = new_fields;
 	}
 	field = &ctx->fields[ctx->nr_fields];
@@ -128,6 +128,7 @@ void lttng_context_update(struct lttng_ctx *ctx)
 			field_align = type->u.basic.integer.alignment;
 			break;
 		case atype_array:
+		case atype_array_bitfield:
 		{
 			struct lttng_basic_type *btype;
 
@@ -141,6 +142,8 @@ void lttng_context_update(struct lttng_ctx *ctx)
 
 			case atype_array:
 			case atype_sequence:
+			case atype_array_bitfield:
+			case atype_sequence_bitfield:
 			case atype_struct:
 			case atype_array_compound:
 			case atype_sequence_compound:
@@ -152,6 +155,7 @@ void lttng_context_update(struct lttng_ctx *ctx)
 			break;
 		}
 		case atype_sequence:
+		case atype_sequence_bitfield:
 		{
 			struct lttng_basic_type *btype;
 
@@ -164,6 +168,8 @@ void lttng_context_update(struct lttng_ctx *ctx)
 			case atype_string:
 			case atype_array:
 			case atype_sequence:
+			case atype_array_bitfield:
+			case atype_sequence_bitfield:
 			case atype_struct:
 			case atype_array_compound:
 			case atype_sequence_compound:
@@ -186,6 +192,8 @@ void lttng_context_update(struct lttng_ctx *ctx)
 
 			case atype_array:
 			case atype_sequence:
+			case atype_array_bitfield:
+			case atype_sequence_bitfield:
 			case atype_struct:
 			case atype_array_compound:
 			case atype_sequence_compound:
@@ -240,7 +248,7 @@ void lttng_destroy_context(struct lttng_ctx *ctx)
 		if (ctx->fields[i].destroy)
 			ctx->fields[i].destroy(&ctx->fields[i]);
 	}
-	kfree(ctx->fields);
+	lttng_kvfree(ctx->fields);
 	kfree(ctx);
 }