]> CyberLeo.Net >> Repos - FreeBSD/stable/10.git/blob - lib/libkvm/tests/kvm_open2_test.c
MFC r316099:
[FreeBSD/stable/10.git] / lib / libkvm / tests / kvm_open2_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_open2_negative_test_nonexistent_corefile);
46 ATF_TC_BODY(kvm_open2_negative_test_nonexistent_corefile, tc)
47 {
48
49         errbuf_clear();
50         ATF_CHECK(kvm_open2(NULL, "/nonexistent", O_RDONLY, NULL, NULL) == NULL);
51         ATF_CHECK(!errbuf_has_error(errbuf));
52         errbuf_clear();
53         ATF_CHECK(kvm_open2(NULL, "/nonexistent", O_RDONLY,
54             errbuf, NULL) == NULL);
55         ATF_CHECK(errbuf_has_error(errbuf));
56 }
57
58 ATF_TC_WITHOUT_HEAD(kvm_open2_negative_test_nonexistent_execfile);
59 ATF_TC_BODY(kvm_open2_negative_test_nonexistent_execfile, tc)
60 {
61
62         errbuf_clear();
63         ATF_CHECK(kvm_open2("/nonexistent", _PATH_DEVZERO, O_RDONLY,
64             NULL, NULL) == NULL);
65         ATF_CHECK(strlen(errbuf) == 0);
66         errbuf_clear();
67         ATF_CHECK(kvm_open2("/nonexistent", _PATH_DEVZERO, O_RDONLY,
68             errbuf, NULL) == NULL);
69         ATF_CHECK(errbuf_has_error(errbuf));
70 }
71
72 ATF_TC(kvm_open2_negative_test_invalid_corefile);
73 ATF_TC_HEAD(kvm_open2_negative_test_invalid_corefile, tc)
74 {
75
76         atf_tc_set_md_var(tc, "require.user", "root");
77 }
78
79 ATF_TC_BODY(kvm_open2_negative_test_invalid_corefile, tc)
80 {
81         kvm_t *kd;
82
83         errbuf_clear();
84         atf_utils_create_file("some-file", "this is a text file");
85         kd = kvm_open2(NULL, "some-file", O_RDONLY, errbuf, NULL);
86         ATF_CHECK(errbuf_has_error(errbuf));
87         ATF_REQUIRE_MSG(kd == NULL, "kvm_open2 succeeded");
88 }
89
90 ATF_TC(kvm_open2_negative_test_invalid_execfile);
91 ATF_TC_HEAD(kvm_open2_negative_test_invalid_execfile, tc)
92 {
93
94         atf_tc_set_md_var(tc, "require.user", "root");
95 }
96
97 ATF_TC_BODY(kvm_open2_negative_test_invalid_execfile, tc)
98 {
99         kvm_t *kd;
100
101         errbuf_clear();
102         atf_utils_create_file("some-file", "this is a text file");
103         kd = kvm_open2("some-file", "/bin/sh", O_RDONLY, errbuf, NULL);
104         ATF_CHECK(errbuf_has_error(errbuf));
105         ATF_REQUIRE_MSG(kd == NULL, "kvm_open2 succeeded unexpectedly");
106 }
107
108 ATF_TP_ADD_TCS(tp)
109 {
110
111         ATF_TP_ADD_TC(tp, kvm_open2_negative_test_invalid_corefile);
112         ATF_TP_ADD_TC(tp, kvm_open2_negative_test_invalid_execfile);
113         ATF_TP_ADD_TC(tp, kvm_open2_negative_test_nonexistent_corefile);
114         ATF_TP_ADD_TC(tp, kvm_open2_negative_test_nonexistent_execfile);
115
116         return (atf_no_error());
117 }