X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=src%2Fcommon%2Fuserspace-probe.c;h=c96d0e3b2d186fff22f216c4f745bc2f68bac795;hb=d2dc232d580ae18400445b6e34fabe1023208f20;hp=67b0ee69820c708e28bbd297674f8c3d055b7539;hpb=0f7c296359bf90005e1dadf2d7a02a4b223a8031;p=lttng-tools.git

diff --git a/src/common/userspace-probe.c b/src/common/userspace-probe.c
index 67b0ee698..c96d0e3b2 100644
--- a/src/common/userspace-probe.c
+++ b/src/common/userspace-probe.c
@@ -20,7 +20,7 @@
 #include <lttng/userspace-probe-internal.h>
 #include <sys/stat.h>
 #include <sys/types.h>
-#include <sys/unistd.h>
+#include <unistd.h>
 
 static
 int lttng_userspace_probe_location_function_set_binary_fd_handle(
@@ -1382,18 +1382,21 @@ int lttng_userspace_probe_location_tracepoint_create_from_payload(
 
 	probe_name = lttng_strndup(probe_name_src, LTTNG_SYMBOL_NAME_LEN);
 	if (!probe_name) {
-		PERROR("lttng_strndup");
+		PERROR("Failed to allocate probe name");
+		ret = -LTTNG_ERR_INVALID;
 		goto end;
 	}
 	provider_name = lttng_strndup(provider_name_src, LTTNG_SYMBOL_NAME_LEN);
 	if (!provider_name) {
-		PERROR("lttng_strndup");
+		PERROR("Failed to allocate provider name");
+		ret = -LTTNG_ERR_INVALID;
 		goto end;
 	}
 
-	binary_path = lttng_strndup(binary_path_src, LTTNG_SYMBOL_NAME_LEN);
+	binary_path = lttng_strndup(binary_path_src, LTTNG_PATH_MAX);
 	if (!binary_path) {
-		PERROR("lttng_strndup");
+		PERROR("Failed to allocate binary path");
+		ret = -LTTNG_ERR_INVALID;
 		goto end;
 	}