X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=ustctl%2Fustctl.c;h=d29097567aed1b93143c5f22c66a391f0992833b;hb=d99cad3d410921d1778a9e12dd746a862a9401d3;hp=b9f2dce83b43913bc6aaedd766d160d9a81accdb;hpb=b9318b35947bbd6d3b2998b28d974c5e5bfabcf6;p=ust.git

diff --git a/ustctl/ustctl.c b/ustctl/ustctl.c
index b9f2dce..d290975 100644
--- a/ustctl/ustctl.c
+++ b/ustctl/ustctl.c
@@ -22,8 +22,7 @@
 #include <stdlib.h>
 #include <fcntl.h>
 
-#include "ustcomm.h"
-#include "ustcmd.h"
+#include "ust/ustcmd.h"
 #include "usterr.h"
 
 enum command {
@@ -148,7 +147,7 @@ int parse_opts_long(int argc, char **argv, struct ust_opts *opts)
 	if (argc - optind > 0 && opts->cmd != GET_ONLINE_PIDS) {
 		int i;
 		int pididx=0;
-		opts->pids = malloc((argc-optind+1) * sizeof(pid_t));
+		opts->pids = zmalloc((argc-optind+1) * sizeof(pid_t));
 
 		for(i=optind; i<argc; i++) {
 			/* don't take any chances, use a long long */