X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=liblttngctl%2Fliblttngctl.c;h=2464bce3b2d1eb30ab0252a2fc4f224a5a85b0c0;hp=0cc3268f78e4520bed4d9f5b70177772737c46be;hb=7d8234d9e6f162ee642cdbec911f46c29b012c3d;hpb=9f19cc17a942a7089fc209a2527d8b2960c83a00

diff --git a/liblttngctl/liblttngctl.c b/liblttngctl/liblttngctl.c
index 0cc3268f7..2464bce3b 100644
--- a/liblttngctl/liblttngctl.c
+++ b/liblttngctl/liblttngctl.c
@@ -279,22 +279,36 @@ end:
 }
 
 /*
- * Copy domain to lttcomm_session_msg domain.
- *
- * Return -1 if the domain is unkown.
+ * Copy domain to lttcomm_session_msg domain. If unknown domain, default domain
+ * will be the kernel.
  */
-static int copy_lttng_domain(struct lttng_domain *dom)
+static void copy_lttng_domain(struct lttng_domain *dom)
 {
-	switch (dom->type) {
-	case LTTNG_DOMAIN_KERNEL:
-	case LTTNG_DOMAIN_UST:
-	case LTTNG_DOMAIN_UST_EXEC_NAME:
-	case LTTNG_DOMAIN_UST_PID:
-	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
-		memcpy(&lsm.domain, dom, sizeof(struct lttng_domain));
-		return 0;
-	default:
-		return -1;
+	if (dom) {
+		switch (dom->type) {
+		case LTTNG_DOMAIN_KERNEL:
+		case LTTNG_DOMAIN_UST:
+		case LTTNG_DOMAIN_UST_EXEC_NAME:
+		case LTTNG_DOMAIN_UST_PID:
+		case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
+			memcpy(&lsm.domain, dom, sizeof(struct lttng_domain));
+			break;
+		default:
+			lsm.domain.type = LTTNG_DOMAIN_KERNEL;
+			break;
+		}
+	}
+}
+
+/*
+ * Copy string from src to dst and enforce null terminated byte.
+ */
+static void copy_string(char *dst, const char *src, size_t len)
+{
+	if (src && dst) {
+		strncpy(dst, src, len);
+		/* Enforce the NULL terminated byte */
+		dst[len - 1] = '\0';
 	}
 }
 
@@ -303,7 +317,7 @@ static int copy_lttng_domain(struct lttng_domain *dom)
  */
 int lttng_start_tracing(const char *session_name)
 {
-	strncpy(lsm.session_name, session_name, NAME_MAX);
+	copy_string(lsm.session.name, session_name, NAME_MAX);
 	return ask_sessiond(LTTNG_START_TRACE, NULL);
 }
 
@@ -312,7 +326,7 @@ int lttng_start_tracing(const char *session_name)
  */
 int lttng_stop_tracing(const char *session_name)
 {
-	strncpy(lsm.session_name, session_name, NAME_MAX);
+	copy_string(lsm.session.name, session_name, NAME_MAX);
 	return ask_sessiond(LTTNG_STOP_TRACE, NULL);
 }
 
@@ -323,31 +337,15 @@ int lttng_add_context(struct lttng_domain *domain,
 		struct lttng_event_context *ctx, const char *event_name,
 		const char *channel_name)
 {
-	int ret;
-
-	if (channel_name != NULL) {
-		strncpy(lsm.u.context.channel_name, channel_name, NAME_MAX);
-	}
+	copy_string(lsm.u.context.channel_name, channel_name, NAME_MAX);
+	copy_string(lsm.u.context.event_name, event_name, NAME_MAX);
+	copy_lttng_domain(domain);
 
-	if (event_name != NULL) {
-		strncpy(lsm.u.context.event_name, event_name, NAME_MAX);
+	if (ctx) {
+		memcpy(&lsm.u.context.ctx, ctx, sizeof(struct lttng_event_context));
 	}
 
-	memcpy(&lsm.u.context.ctx, ctx, sizeof(struct lttng_event_context));
-
-	switch (domain->type) {
-	case LTTNG_DOMAIN_KERNEL:
-		ret = ask_sessiond(LTTNG_KERNEL_ADD_CONTEXT, NULL);
-		break;
-	case LTTNG_DOMAIN_UST:
-		ret = LTTCOMM_NOT_IMPLEMENTED;
-		break;
-	default:
-		ret = LTTCOMM_UNKNOWN_DOMAIN;
-		break;
-	};
-
-	return ret;
+	return ask_sessiond(LTTNG_ADD_CONTEXT, NULL);
 }
 
 /*
@@ -359,125 +357,92 @@ int lttng_enable_event(struct lttng_domain *domain,
 	int ret;
 
 	if (channel_name == NULL) {
-		strncpy(lsm.u.enable.channel_name, DEFAULT_CHANNEL_NAME, NAME_MAX);
+		copy_string(lsm.u.enable.channel_name, DEFAULT_CHANNEL_NAME, NAME_MAX);
 	} else {
-		strncpy(lsm.u.enable.channel_name, channel_name, NAME_MAX);
+		copy_string(lsm.u.enable.channel_name, channel_name, NAME_MAX);
 	}
 
-	switch (domain->type) {
-		case LTTNG_DOMAIN_KERNEL:
-			if (ev == NULL) {
-				ret = ask_sessiond(LTTNG_KERNEL_ENABLE_ALL_EVENT, NULL);
-			} else {
-				memcpy(&lsm.u.enable.event, ev, sizeof(struct lttng_event));
-				ret = ask_sessiond(LTTNG_KERNEL_ENABLE_EVENT, NULL);
-			}
-			break;
-		case LTTNG_DOMAIN_UST:
-			ret = LTTCOMM_NOT_IMPLEMENTED;
-			break;
-		default:
-			ret = LTTCOMM_UNKNOWN_DOMAIN;
-			break;
-	};
+	copy_lttng_domain(domain);
+
+	if (ev == NULL) {
+		ret = ask_sessiond(LTTNG_ENABLE_ALL_EVENT, NULL);
+	} else {
+		memcpy(&lsm.u.enable.event, ev, sizeof(struct lttng_event));
+		ret = ask_sessiond(LTTNG_ENABLE_EVENT, NULL);
+	}
 
 	return ret;
 }
 
 /*
- * Disable an event in the kernel tracer.
+ * Disable event of a channel and domain.
  */
 int lttng_disable_event(struct lttng_domain *domain, const char *name,
 		const char *channel_name)
 {
-	int ret;
+	int ret = -1;
 
 	if (channel_name == NULL) {
-		strncpy(lsm.u.disable.channel_name, DEFAULT_CHANNEL_NAME, NAME_MAX);
+		copy_string(lsm.u.disable.channel_name, DEFAULT_CHANNEL_NAME, NAME_MAX);
 	} else {
-		strncpy(lsm.u.disable.channel_name, channel_name, NAME_MAX);
+		copy_string(lsm.u.disable.channel_name, channel_name, NAME_MAX);
 	}
 
-	switch (domain->type) {
-		case LTTNG_DOMAIN_KERNEL:
-			if (name == NULL) {
-				ret = ask_sessiond(LTTNG_KERNEL_DISABLE_ALL_EVENT, NULL);
-			} else {
-				strncpy(lsm.u.disable.name, name, NAME_MAX);
-				ret = ask_sessiond(LTTNG_KERNEL_DISABLE_EVENT, NULL);
-			}
-			break;
-		case LTTNG_DOMAIN_UST:
-			ret = LTTCOMM_NOT_IMPLEMENTED;
-			break;
-		default:
-			ret = LTTCOMM_UNKNOWN_DOMAIN;
-			break;
-	};
+	copy_lttng_domain(domain);
+
+	if (name == NULL) {
+		ret = ask_sessiond(LTTNG_DISABLE_ALL_EVENT, NULL);
+	} else {
+		copy_string(lsm.u.disable.name, name, NAME_MAX);
+		ret = ask_sessiond(LTTNG_DISABLE_EVENT, NULL);
+	}
 
 	return ret;
 }
 
 /*
- * Enable recording for a channel for the kernel tracer.
+ * Enable channel per domain
  */
 int lttng_enable_channel(struct lttng_domain *domain,
 		struct lttng_channel *chan)
 {
-	int ret;
+	if (chan) {
+		memcpy(&lsm.u.channel.chan, chan, sizeof(struct lttng_channel));
+	}
 
-	memcpy(&lsm.u.channel.chan, chan, sizeof(struct lttng_channel));
+	copy_lttng_domain(domain);
 
-	switch (domain->type) {
-		case LTTNG_DOMAIN_KERNEL:
-			ret = ask_sessiond(LTTNG_KERNEL_ENABLE_CHANNEL, NULL);
-			break;
-		case LTTNG_DOMAIN_UST:
-			ret = LTTCOMM_NOT_IMPLEMENTED;
-			break;
-		default:
-			ret = LTTCOMM_UNKNOWN_DOMAIN;
-			break;
-	};
-
-	return ret;
+	return ask_sessiond(LTTNG_ENABLE_CHANNEL, NULL);
 }
 
 /*
- * Disable recording for the channel for the kernel tracer.
+ * All tracing will be stopped for registered events of the channel.
  */
 int lttng_disable_channel(struct lttng_domain *domain, const char *name)
 {
-	int ret;
-
-	strncpy(lsm.u.disable.channel_name, name, NAME_MAX);
+	copy_string(lsm.u.disable.channel_name, name, NAME_MAX);
+	copy_lttng_domain(domain);
 
-	switch (domain->type) {
-		case LTTNG_DOMAIN_KERNEL:
-			ret = ask_sessiond(LTTNG_KERNEL_DISABLE_CHANNEL, NULL);
-			break;
-		case LTTNG_DOMAIN_UST:
-			ret = LTTCOMM_NOT_IMPLEMENTED;
-			break;
-		default:
-			ret = LTTCOMM_UNKNOWN_DOMAIN;
-			break;
-	};
-
-	return ret;
+	return ask_sessiond(LTTNG_DISABLE_CHANNEL, NULL);
 }
 
 /*
- * List all available kernel events.
+ * List all available tracepoints of domain.
  *
- * Return the size (bytes) of the list and set the event_list array.
+ * Return the size (bytes) of the list and set the events array.
  * On error, return negative value.
  */
-int lttng_list_kernel_events(struct lttng_event **events)
+int lttng_list_tracepoints(struct lttng_domain *domain,
+		struct lttng_event **events)
 {
 	int ret;
 
-	ret = ask_sessiond(LTTNG_KERNEL_LIST_EVENTS, (void **) events);
+	copy_lttng_domain(domain);
+
+	ret = ask_sessiond(LTTNG_LIST_TRACEPOINTS, (void **) events);
+	if (ret < 0) {
+		return ret;
+	}
 
 	return ret / sizeof(struct lttng_event);
 }
@@ -499,8 +464,8 @@ const char *lttng_get_readable_code(int code)
  */
 int lttng_create_session(const char *name, const char *path)
 {
-	strncpy(lsm.session_name, name, NAME_MAX);
-	strncpy(lsm.path, path, PATH_MAX);
+	copy_string(lsm.session.name, name, NAME_MAX);
+	copy_string(lsm.session.path, path, PATH_MAX);
 	return ask_sessiond(LTTNG_CREATE_SESSION, NULL);
 }
 
@@ -509,7 +474,7 @@ int lttng_create_session(const char *name, const char *path)
  */
 int lttng_destroy_session(const char *name)
 {
-	strncpy(lsm.session_name, name, NAME_MAX);
+	copy_string(lsm.session.name, name, NAME_MAX);
 	return ask_sessiond(LTTNG_DESTROY_SESSION, NULL);
 }
 
@@ -538,7 +503,7 @@ int lttng_list_domains(const char *session_name, struct lttng_domain **domains)
 {
 	int ret;
 
-	strncpy(lsm.session_name, session_name, NAME_MAX);
+	copy_string(lsm.session.name, session_name, NAME_MAX);
 	ret = ask_sessiond(LTTNG_LIST_DOMAINS, (void**) domains);
 	if (ret < 0) {
 		return ret;
@@ -555,11 +520,8 @@ int lttng_list_channels(struct lttng_domain *domain,
 {
 	int ret;
 
-	strncpy(lsm.session_name, session_name, NAME_MAX);
-	ret = copy_lttng_domain(domain);
-	if (ret < 0) {
-		return -LTTCOMM_UNKNOWN_DOMAIN;
-	}
+	copy_string(lsm.session.name, session_name, NAME_MAX);
+	copy_lttng_domain(domain);
 
 	ret = ask_sessiond(LTTNG_LIST_CHANNELS, (void**) channels);
 	if (ret < 0) {
@@ -578,12 +540,9 @@ int lttng_list_events(struct lttng_domain *domain,
 {
 	int ret;
 
-	strncpy(lsm.session_name, session_name, NAME_MAX);
-	strncpy(lsm.u.list.channel_name, channel_name, NAME_MAX);
-	ret = copy_lttng_domain(domain);
-	if (ret < 0) {
-		return -LTTCOMM_UNKNOWN_DOMAIN;
-	}
+	copy_string(lsm.session.name, session_name, NAME_MAX);
+	copy_string(lsm.u.list.channel_name, channel_name, NAME_MAX);
+	copy_lttng_domain(domain);
 
 	ret = ask_sessiond(LTTNG_LIST_EVENTS, (void**) events);
 	if (ret < 0) {
@@ -598,7 +557,7 @@ int lttng_list_events(struct lttng_domain *domain,
  */
 void lttng_set_session_name(const char *name)
 {
-	strncpy(lsm.session_name, name, NAME_MAX);
+	copy_string(lsm.session.name, name, NAME_MAX);
 }
 
 /*
@@ -616,6 +575,22 @@ int lttng_set_tracing_group(const char *name)
 	return 0;
 }
 
+/*
+ *  lttng_calibrate
+ */
+int lttng_calibrate(struct lttng_domain *domain,
+		struct lttng_calibrate *calibrate)
+{
+	int ret;
+
+	copy_lttng_domain(domain);
+
+	memcpy(&lsm.u.calibrate, calibrate, sizeof(struct lttng_calibrate));
+	ret = ask_sessiond(LTTNG_CALIBRATE, NULL);
+
+	return ret;
+}
+
 /*
  *  lttng_check_session_daemon
  *
@@ -633,13 +608,22 @@ int lttng_session_daemon_alive(void)
 		return ret;
 	}
 
-	/* If socket exist, we consider the daemon started */
+	/* If socket exist, we check if the daemon listens to connect. */
 	ret = access(sessiond_sock_path, F_OK);
 	if (ret < 0) {
 		/* Not alive */
 		return 0;
 	}
 
+	ret = lttcomm_connect_unix_sock(sessiond_sock_path);
+	if (ret < 0) {
+		/* Not alive */
+		return 0;
+	}
+	ret = lttcomm_close_unix_sock(ret);
+	if (ret < 0)
+		perror("lttcomm_close_unix_sock");
+
 	/* Is alive */
 	return 1;
 }