lib/libkvm/tests/kvm_open_test.c

   1 /*-
   2  * Copyright (c) 2017 Ngie Cooper <ngie@freebsd.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include <sys/param.h>
  31 #include <errno.h>
  32 #include <fcntl.h>
  33 #include <kvm.h>
  34 #include <limits.h>
  35 #include <paths.h>
  36 #include <signal.h>
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <string.h>
  40
  41 #include <atf-c.h>
  42
  43 #include "kvm_test_common.h"
  44
  45 ATF_TC_WITHOUT_HEAD(kvm_open_negative_test_nonexistent_corefile);
  46 ATF_TC_BODY(kvm_open_negative_test_nonexistent_corefile, tc)
  47 {
  48
  49         ATF_CHECK(kvm_open(NULL, "/nonexistent", NULL, O_RDONLY, NULL) == NULL);
  50         ATF_CHECK(kvm_open(NULL, "/nonexistent", NULL, O_RDONLY,
  51             getprogname()) == NULL);
  52 }
  53
  54 ATF_TC_WITHOUT_HEAD(kvm_open_negative_test_nonexistent_execfile);
  55 ATF_TC_BODY(kvm_open_negative_test_nonexistent_execfile, tc)
  56 {
  57
  58         ATF_CHECK(kvm_open("/nonexistent", _PATH_DEVZERO, NULL, O_RDONLY,
  59             NULL) == NULL);
  60         ATF_CHECK(kvm_open("/nonexistent", _PATH_DEVZERO, NULL, O_RDONLY,
  61             getprogname()) == NULL);
  62 }
  63
  64 ATF_TC(kvm_open_negative_test_invalid_corefile);
  65 ATF_TC_HEAD(kvm_open_negative_test_invalid_corefile, tc)
  66 {
  67
  68         atf_tc_set_md_var(tc, "require.user", "root");
  69 }
  70
  71 ATF_TC_BODY(kvm_open_negative_test_invalid_corefile, tc)
  72 {
  73         kvm_t *kd;
  74
  75         atf_utils_create_file("some-file", "this is a text file");
  76         kd = kvm_open(NULL, "some-file", NULL, O_RDONLY, getprogname());
  77         ATF_REQUIRE_MSG(kd == NULL, "kvm_open didn't return NULL on failure");
  78 }
  79
  80 ATF_TC(kvm_open_negative_test_invalid_execfile);
  81 ATF_TC_HEAD(kvm_open_negative_test_invalid_execfile, tc)
  82 {
  83
  84         atf_tc_set_md_var(tc, "require.user", "root");
  85 }
  86
  87 ATF_TC_BODY(kvm_open_negative_test_invalid_execfile, tc)
  88 {
  89         kvm_t *kd;
  90
  91         atf_utils_create_file("some-file", "this is a text file");
  92         kd = kvm_open("some-file", "/bin/sh", NULL, O_RDONLY, getprogname());
  93         ATF_REQUIRE_MSG(kd == NULL, "kvm_open succeeded unexpectedly");
  94 }
  95
  96 ATF_TP_ADD_TCS(tp)
  97 {
  98
  99         ATF_TP_ADD_TC(tp, kvm_open_negative_test_invalid_corefile);
 100         ATF_TP_ADD_TC(tp, kvm_open_negative_test_invalid_execfile);
 101         ATF_TP_ADD_TC(tp, kvm_open_negative_test_nonexistent_corefile);
 102         ATF_TP_ADD_TC(tp, kvm_open_negative_test_nonexistent_execfile);
 103
 104         return (atf_no_error());
 105 }