X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=libust%2Flttng-ust-comm.c;h=c9cb777709dab0d84ffffebe26352fcfdfbde563;hb=b35d179d7be2ab0cb4a9c698b86d78928089fc39;hp=e73d324d4815d1439cc756defe762a3897d2f270;hpb=204d45df751cfc84133317c88bc6c46e0de12b52;p=lttng-ust.git

diff --git a/libust/lttng-ust-comm.c b/libust/lttng-ust-comm.c
index e73d324d..c9cb7777 100644
--- a/libust/lttng-ust-comm.c
+++ b/libust/lttng-ust-comm.c
@@ -23,7 +23,8 @@
 #include <sys/socket.h>
 #include <unistd.h>
 #include <errno.h>
-#include <lttng-sessiond-comm.h>
+#include <ust/lttng-ust-abi.h>
+#include <lttng-ust-comm.h>
 #include <ust/usterr-signal-safe.h>
 #include <pthread.h>
 
@@ -79,13 +80,13 @@ int handle_message(int sock, struct lttcomm_ust_msg *lum)
 	int ret;
 
 	switch (lum->cmd_type) {
-	case LTTNG_UST_CREATE_SESSION:
+	case UST_CREATE_SESSION:
 	{
 		struct lttcomm_ust_reply lur;
 
 		DBG("Handling create session message");
 		memset(&lur, 0, sizeof(lur));
-		lur.cmd_type = LTTNG_UST_CREATE_SESSION;
+		lur.cmd_type = UST_CREATE_SESSION;
 
 		/* ... */
 		ret = 0;
@@ -224,4 +225,5 @@ void __attribute__((destructor)) lttng_ust_comm_exit(void)
 	if (ret) {
 		ERR("Error cancelling local ust listener thread");
 	}
+	lttng_ust_abi_exit();
 }