tools/regression/posixsem2/semtest.c

   1 /* $FreeBSD$ */
   2
   3 #include <sys/types.h>
   4 #include <sys/wait.h>
   5 #include <sys/mman.h>
   6 #include <semaphore.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <err.h>
  10 #include <fcntl.h>
  11 #include <unistd.h>
  12
  13 #define SEM_NAME "/semtst"
  14
  15 int test_unnamed(void);
  16 int test_named(void);
  17
  18 int
  19 test_unnamed(void)
  20 {
  21         sem_t *s;
  22         pid_t pid;
  23         int status;
  24
  25         printf("testing unnamed process-shared semaphore\n");
  26         s = (sem_t *)mmap(NULL, sizeof(sem_t), PROT_READ|PROT_WRITE, MAP_ANON|MAP_SHARED,
  27                 -1, 0);
  28         if (s == MAP_FAILED)
  29                 err(1, "mmap failed");
  30         if (sem_init(s, 1, 0))
  31                 err(2, "sem_init failed");
  32         if ((pid = fork()) == 0) {
  33                 printf("child: sem_wait()\n");
  34                 if (sem_wait(s))
  35                         err(3, "sem_wait failed");
  36                 printf("child: sem_wait() returned\n");
  37                 exit(0);
  38         } else {
  39                 sleep(1);
  40                 printf("parent: sem_post()\n");
  41                 if (sem_post(s))
  42                         err(4, "sem_post failed");
  43                 waitpid(pid, &status, 0);
  44                 if (WIFEXITED(status) && WEXITSTATUS(status) == 0)
  45                         printf("OK.\n");
  46                 else
  47                         printf("Failure.");
  48         }
  49         return (0);
  50 }
  51
  52 int
  53 test_named(void)
  54 {
  55         sem_t *s, *s2;
  56         pid_t pid;
  57         int status;
  58
  59         printf("testing named process-shared semaphore\n");
  60         sem_unlink(SEM_NAME);
  61         s = sem_open(SEM_NAME, O_CREAT, 0777, 0);
  62         if (s == SEM_FAILED)
  63                 err(1, "sem_open failed");
  64         s2 = sem_open(SEM_NAME, O_CREAT, 0777, 0);
  65         if (s2 == SEM_FAILED)
  66                 err(2, "second sem_open call failed");
  67         if (s != s2)
  68                 err(3, "two sem_open calls for same semaphore do not returm same address.");
  69         if (sem_close(s2))
  70                 err(4, "sem_close failed");
  71         if ((pid = fork()) == 0) {
  72                 printf("child: sem_wait()\n");
  73                 if (sem_wait(s))
  74                         err(5, "sem_wait failed");
  75                 printf("child: sem_wait() returned\n");
  76                 exit(0);
  77         } else {
  78                 sleep(1);
  79                 printf("parent: sem_post()\n");
  80                 if (sem_post(s))
  81                         err(6, "sem_post failed");
  82                 waitpid(pid, &status, 0);
  83                 if (WIFEXITED(status) && WEXITSTATUS(status) == 0)
  84                         printf("OK.\n");
  85                 else
  86                         printf("Failure.");
  87         }
  88
  89         if (sem_close(s))
  90                 err(7, "sem_close failed");
  91
  92         return (0);
  93 }
  94
  95 int
  96 main(void)
  97 {
  98         test_unnamed();
  99         test_named();
 100         return (0);
 101 }