tools/regression/priv/priv_vfs_generation.c

   1 /*-
   2  * Copyright (c) 2006 nCircle Network Security, Inc.
   3  * Copyright (c) 2007 Robert N. M. Watson
   4  * All rights reserved.
   5  *
   6  * This software was developed by Robert N. M. Watson for the TrustedBSD
   7  * Project under contract to nCircle Network Security, Inc.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR, NCIRCLE NETWORK SECURITY,
  22  * INC., OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
  24  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  25  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  26  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  27  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  28  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 /*
  32  * Confirm that a generation number isn't returned by stat() when not running
  33  * with privilege.  In order to differentiate between a generation of 0 and
  34  * a generation not being returned, we have to create a temporary file known
  35  * to have a non-0 generation.  We try up to MAX_TRIES times, and then give
  36  * up, which is non-ideal, but better than not testing for a problem.
  37  */
  38
  39 #include <sys/stat.h>
  40
  41 #include <err.h>
  42 #include <errno.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45
  46 #include "main.h"
  47
  48 static char fpath[1024];
  49 static int fpath_initialized;
  50
  51 #define MAX_TRIES       100
  52
  53 int
  54 priv_vfs_generation_setup(int asroot, int injail, struct test *test)
  55 {
  56         struct stat sb;
  57         int i;
  58
  59         /*
  60          * The kernel zeros the generation number field when an unprivileged
  61          * user stats a file.  In order to distinguish the two cases, we
  62          * therefore require a file that we know has a non-zero generation
  63          * number.  We try up to MAX_TRIES times and otherwise fail.
  64          */
  65         for (i = 0; i < MAX_TRIES; i++) {
  66                 setup_file("priv_vfs_generation_setup: fpath", fpath,
  67                     UID_ROOT, GID_WHEEL, 0644);
  68                 if (stat(fpath, &sb) < 0) {
  69                         warn("priv_vfs_generation_setup: fstat(%s)", fpath);
  70                         (void)unlink(fpath);
  71                         return (-1);
  72                 }
  73                 if (sb.st_gen != 0) {
  74                         fpath_initialized = 1;
  75                         return (0);
  76                 }
  77         }
  78         warnx("priv_vfs_generation_setup: unable to create gen file");
  79         return (-1);
  80 }
  81
  82 void
  83 priv_vfs_generation(int asroot, int injail, struct test *test)
  84 {
  85         struct stat sb;
  86         int error;
  87
  88         error = stat(fpath, &sb);
  89         if (error < 0)
  90                 warn("priv_vfs_generation(asroot, injail) stat");
  91
  92         if (sb.st_gen == 0) {
  93                 error = -1;
  94                 errno = EPERM;
  95         } else
  96                 error = 0;
  97         if (asroot && injail)
  98                 expect("priv_vfs_generation(asroot, injail)", error, -1,
  99                     EPERM);
 100         if (asroot && !injail)
 101                 expect("priv_vfs_generation(asroot, !injail)", error, 0, 0);
 102         if (!asroot && injail)
 103                 expect("priv_vfs_generation(!asroot, injail)", error, -1,
 104                     EPERM);
 105         if (!asroot && !injail)
 106                 expect("priv_vfs_generation(!asroot, !injail)", error, -1,
 107                     EPERM);
 108 }
 109
 110 void
 111 priv_vfs_generation_cleanup(int asroot, int injail, struct test *test)
 112 {
 113
 114         if (fpath_initialized) {
 115                 (void)unlink(fpath);
 116                 fpath_initialized = 0;
 117         }
 118 }