*/
#define _GNU_SOURCE
-#include <errno.h>
-#include <fcntl.h>
-#include <limits.h>
-#include <stdio.h>
#include <stdlib.h>
-#include <string.h>
#include <unistd.h>
#include <common/error.h>
#include "utils.h"
+#include "lttng-sessiond.h"
+
+int ht_cleanup_pipe[2] = { -1, -1 };
/*
* Write to writable pipe used to notify a thread.
*/
int notify_thread_pipe(int wpipe)
{
- int ret;
-
- ret = write(wpipe, "!", 1);
- if (ret < 0) {
- PERROR("write poll pipe");
- }
-
- return ret;
-}
-
-/*
- * Return pointer to home directory path using the env variable HOME.
- *
- * No home, NULL is returned.
- */
-const char *get_home_dir(void)
-{
- return ((const char *) getenv("HOME"));
-}
-
-/*
- * Create a pipe in dst.
- */
-int utils_create_pipe(int *dst)
-{
- int ret;
+ ssize_t ret;
- if (dst == NULL) {
- return -1;
+ /* Ignore if the pipe is invalid. */
+ if (wpipe < 0) {
+ return 0;
}
- ret = pipe(dst);
- if (ret < 0) {
- PERROR("create pipe");
+ ret = lttng_write(wpipe, "!", 1);
+ if (ret < 1) {
+ PERROR("write poll pipe");
}
- return ret;
+ return (int) ret;
}
-/*
- * Create pipe and set CLOEXEC flag to both fd.
- *
- * Make sure the pipe opened by this function are closed at some point. Use
- * utils_close_pipe().
- */
-int utils_create_pipe_cloexec(int *dst)
+void ht_cleanup_push(struct lttng_ht *ht)
{
- int ret, i;
+ ssize_t ret;
+ int fd = ht_cleanup_pipe[1];
- if (dst == NULL) {
- return -1;
- }
-
- ret = utils_create_pipe(dst);
- if (ret < 0) {
- goto error;
+ if (!ht) {
+ return;
}
-
- for (i = 0; i < 2; i++) {
- ret = fcntl(dst[i], F_SETFD, FD_CLOEXEC);
+ if (fd < 0)
+ return;
+ ret = lttng_write(fd, &ht, sizeof(ht));
+ if (ret < sizeof(ht)) {
+ PERROR("write ht cleanup pipe %d", fd);
if (ret < 0) {
- PERROR("fcntl pipe cloexec");
- goto error;
+ ret = -errno;
}
+ goto error;
}
+ /* All good. Don't send back the write positive ret value. */
+ ret = 0;
error:
- return ret;
-}
-
-/*
- * Close both read and write side of the pipe.
- */
-void utils_close_pipe(int *src)
-{
- int i, ret;
-
- if (src == NULL) {
- return;
- }
-
- for (i = 0; i < 2; i++) {
- /* Safety check */
- if (src[i] < 0) {
- continue;
- }
-
- ret = close(src[i]);
- if (ret) {
- PERROR("close pipe");
- }
- }
+ assert(!ret);
}