/* Common code for message queue passing tests. Copyright (C) 2004-2018 Free Software Foundation, Inc. This file is part of the GNU C Library. Contributed by Jakub Jelinek <jakub@redhat.com>, 2004. The GNU C Library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. The GNU C Library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with the GNU C Library; if not, see <http://www.gnu.org/licenses/>. */ #include <mqueue.h> #include <search.h> #include <stdlib.h> #include <stdio.h> #include <string.h> #include <sys/uio.h> #include <unistd.h> static int temp_mq_fd; /* Add temporary files in list. */ static void __attribute__ ((unused)) add_temp_mq (const char *name) { struct iovec iov[2]; iov[0].iov_base = (char *) name; iov[0].iov_len = strlen (name); iov[1].iov_base = (char *) "\n"; iov[1].iov_len = 1; if (writev (temp_mq_fd, iov, 2) != iov[0].iov_len + 1) printf ("Could not record temp mq filename %s\n", name); } /* Delete all temporary message queues. */ static void do_cleanup (void) { if (lseek (temp_mq_fd, 0, SEEK_SET) != 0) return; FILE *f = fdopen (temp_mq_fd, "r"); if (f == NULL) return; char *line = NULL; size_t n = 0; ssize_t rets; while ((rets = getline (&line, &n, f)) > 0) { if (line[rets - 1] != '\n') continue; line[rets - 1] = '\0'; mq_unlink (line); } fclose (f); } static void do_prepare (void) { char name [] = "/tmp/tst-mqueueN.XXXXXX"; temp_mq_fd = mkstemp (name); if (temp_mq_fd == -1) { printf ("Could not create temporary file %s: %m\n", name); exit (1); } unlink (name); } #define PREPARE(argc, argv) do_prepare () #define CLEANUP_HANDLER do_cleanup ()