X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=libust%2Flttng-ust-abi.c;h=be9d1d31d82133e6e2277aeaaa520c359e34753c;hb=4cfec15c93af7e0cfe3ce769ee90486bb8ab7c37;hp=31306fb2c34e5997ac53ff40f317dc20922d66fd;hpb=959ad2f13fd45eb91c8f69f34618895f44e3bc18;p=lttng-ust.git

diff --git a/libust/lttng-ust-abi.c b/libust/lttng-ust-abi.c
index 31306fb2..be9d1d31 100644
--- a/libust/lttng-ust-abi.c
+++ b/libust/lttng-ust-abi.c
@@ -272,8 +272,14 @@ long lttng_abi_add_context(int objd,
 		return -EPERM;
 
 	switch (context_param->ctx) {
+	case LTTNG_UST_CONTEXT_PTHREAD_ID:
+		return lttng_add_pthread_id_to_ctx(ctx);
 	case LTTNG_UST_CONTEXT_VTID:
-		//TODO return lttng_add_vtid_to_ctx(ctx);
+		return lttng_add_vtid_to_ctx(ctx);
+	case LTTNG_UST_CONTEXT_VPID:
+		return lttng_add_vpid_to_ctx(ctx);
+	case LTTNG_UST_CONTEXT_PROCNAME:
+		return lttng_add_procname_to_ctx(ctx);
 	default:
 		return -EINVAL;
 	}
@@ -501,7 +507,7 @@ static const struct objd_ops lttng_session_ops = {
 };
 
 struct stream_priv_data {
-	struct lib_ring_buffer *buf;
+	struct lttng_ust_lib_ring_buffer *buf;
 	struct ltt_channel *ltt_chan;
 };
 
@@ -509,7 +515,7 @@ static
 int lttng_abi_open_stream(int channel_objd, struct lttng_ust_stream *info)
 {
 	struct ltt_channel *channel = objd_private(channel_objd);
-	struct lib_ring_buffer *buf;
+	struct lttng_ust_lib_ring_buffer *buf;
 	struct stream_priv_data *priv;
 	int stream_objd, ret;
 
@@ -626,6 +632,8 @@ long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
 		return ltt_channel_enable(channel);
 	case LTTNG_UST_DISABLE:
 		return ltt_channel_disable(channel);
+	case LTTNG_UST_FLUSH_BUFFER:
+		return channel->ops->flush_buffer(channel->chan, channel->handle);
 	default:
 		return -EINVAL;
 	}
@@ -647,6 +655,8 @@ long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
 static
 long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg)
 {
+	struct ltt_channel *channel = objd_private(objd);
+
 	switch (cmd) {
 	case LTTNG_UST_STREAM:
 	{
@@ -656,6 +666,8 @@ long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg)
 		/* stream used as output */
 		return lttng_abi_open_stream(objd, stream);
 	}
+	case LTTNG_UST_FLUSH_BUFFER:
+		return channel->ops->flush_buffer(channel->chan, channel->handle);
 	default:
 		return -EINVAL;
 	}
@@ -721,13 +733,10 @@ static const struct objd_ops lttng_metadata_ops = {
  *
  *	This object descriptor implements lttng commands:
  *		(None for now. Access is done directly though shm.)
- *		TODO: Add buffer flush.
  */
 static
 long lttng_rb_cmd(int objd, unsigned int cmd, unsigned long arg)
 {
-	//struct stream_priv_data *priv = objd_private(objd);
-
 	switch (cmd) {
 	default:
 		return -EINVAL;
@@ -738,7 +747,7 @@ static
 int lttng_rb_release(int objd)
 {
 	struct stream_priv_data *priv = objd_private(objd);
-	struct lib_ring_buffer *buf;
+	struct lttng_ust_lib_ring_buffer *buf;
 	struct ltt_channel *channel;
 
 	if (priv) {