84 lines
2.1 KiB
C
84 lines
2.1 KiB
C
/* Common code for message queue passing tests.
|
|
Copyright (C) 2004 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 ()
|