tools/test/stress2/misc/mlockall3.sh

   1 #!/bin/sh
   2
   3 #
   4 # Copyright (c) 2013 Peter Holm <pho@FreeBSD.org>
   5 # All rights reserved.
   6 #
   7 # Redistribution and use in source and binary forms, with or without
   8 # modification, are permitted provided that the following conditions
   9 # are met:
  10 # 1. Redistributions of source code must retain the above copyright
  11 #    notice, this list of conditions and the following disclaimer.
  12 # 2. Redistributions in binary form must reproduce the above copyright
  13 #    notice, this list of conditions and the following disclaimer in the
  14 #    documentation and/or other materials provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26 # SUCH DAMAGE.
  27 #
  28
  29 # mlockall(2) test scenario.
  30 # "panic: vm_page_unwire: page xxx's wire count is zero" seen.
  31 # http://people.freebsd.org/~pho/stress/log/mlockall.txt
  32
  33 [ `id -u ` -ne 0 ] && echo "Must be root!" && exit 1
  34
  35 . ../default.cfg
  36
  37 dir=/tmp
  38 odir=`pwd`
  39 cd $dir
  40 sed '1,/^EOF/d' < $odir/$0 > $dir/mlockall3.c
  41 mycc -o mlockall3 -Wall -Wextra mlockall3.c -lpthread || exit 1
  42 rm -f mlockall3.c
  43 cd $odir
  44
  45 /tmp/mlockall3
  46
  47 killall mlockall3 > /dev/null 2>&1
  48 rm -f /tmp/mlockall3
  49 exit
  50
  51 EOF
  52 #include <sys/types.h>
  53 #include <err.h>
  54 #include <errno.h>
  55 #include <fcntl.h>
  56 #include <pthread.h>
  57 #include <pwd.h>
  58 #include <signal.h>
  59 #include <sys/mman.h>
  60 #include <stdint.h>
  61 #include <stdio.h>
  62 #include <stdlib.h>
  63 #include <string.h>
  64 #include <sys/param.h>
  65 #include <sys/stat.h>
  66 #include <sys/syscall.h>
  67 #include <sys/wait.h>
  68 #include <unistd.h>
  69
  70 #define N (128 * 1024 / (int)sizeof(u_int32_t))
  71 u_int32_t r[N];
  72
  73 static void
  74 hand(int i __unused) {  /* handler */
  75         _exit(1);
  76 }
  77
  78 unsigned long
  79 makearg(void)
  80 {
  81         unsigned int i;
  82         unsigned long val;
  83
  84         val = arc4random();
  85         i   = arc4random() % 100;
  86         if (i < 20)
  87                 val = val & 0xff;
  88         if (i >= 20 && i < 40)
  89                 val = val & 0xffff;
  90         if (i >= 40 && i < 60)
  91                 val = (unsigned long)(r) | (val & 0xffff);
  92 #if defined(__LP64__)
  93         if (i >= 60) {
  94                 val = (val << 32) | arc4random();
  95                 if (i > 80)
  96                         val = val & 0x00007fffffffffffUL;
  97         }
  98 #endif
  99
 100         return(val);
 101 }
 102
 103 void *
 104 calls(void *arg __unused)
 105 {
 106         int i;
 107         unsigned long arg1, arg2, arg3;
 108
 109         usleep(1000);
 110         for (i = 0; i < 500; i++) {
 111                 arg1 = makearg();
 112                 arg2 = makearg();
 113                 arg3 = makearg();
 114
 115                 alarm(1);
 116                 syscall(SYS_mlockall, arg1, arg2, arg3);
 117         }
 118
 119         return (0);
 120 }
 121
 122 int
 123 main(void)
 124 {
 125         struct passwd *pw;
 126         pid_t pid;
 127         pthread_t cp[50];
 128         int e, i, j;
 129
 130         if ((pw = getpwnam("nobody")) == NULL)
 131                 err(1, "no such user: nobody");
 132
 133         if (setgroups(1, &pw->pw_gid) ||
 134             setegid(pw->pw_gid) || setgid(pw->pw_gid) ||
 135             seteuid(pw->pw_uid) || setuid(pw->pw_uid))
 136                 err(1, "Can't drop privileges to \"nobody\"");
 137         endpwent();
 138
 139         signal(SIGALRM, hand);
 140         signal(SIGILL,  hand);
 141         signal(SIGFPE,  hand);
 142         signal(SIGSEGV, hand);
 143         signal(SIGBUS,  hand);
 144         signal(SIGURG,  hand);
 145         signal(SIGSYS,  hand);
 146         signal(SIGTRAP, hand);
 147
 148         alarm(180);
 149         for (i = 0; i < 8000; i++) {
 150                 if ((pid = fork()) == 0) {
 151                         for (j = 0; j < N; j++)
 152                                 r[j] = arc4random();
 153                         for (j = 0; j < 50; j++)
 154                                 if ((e = pthread_create(&cp[j], NULL, calls, NULL)) != 0)
 155                                         errc(1, e, "pthread_create");
 156
 157                         for (j = 0; j < 50; j++)
 158                                 pthread_join(cp[j], NULL);
 159                         _exit(0);
 160                 }
 161                 if (pid == -1)
 162                         err(1, "fork()");
 163                 wait(NULL);
 164         }
 165
 166         return (0);
 167 }