1/*-
2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3 *
4 * Copyright (c) 2018 Mark Johnston <markj@FreeBSD.org>
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28#include <sys/cdefs.h>
29__FBSDID("$FreeBSD$");
30
31#include <sys/param.h>
32#include <sys/mman.h>
33
34#include <errno.h>
35#include <stdint.h>
36#include <string.h>
37#include <unistd.h>
38
39#include <atf-c.h>
40
41ATF_TC(brk_basic);
42ATF_TC_HEAD(brk_basic, tc)
43{
44	atf_tc_set_md_var(tc, "descr", "Verify basic brk() functionality");
45}
46ATF_TC_BODY(brk_basic, tc)
47{
48	void *oldbrk, *newbrk;
49	int error;
50
51	/* Reset the break. */
52	error = brk(0);
53	ATF_REQUIRE_MSG(error == 0, "brk: %s", strerror(errno));
54
55	oldbrk = sbrk(0);
56	ATF_REQUIRE(oldbrk != (void *)-1);
57
58	/* Try to allocate a page. */
59	error = brk((void *)((intptr_t)oldbrk + PAGE_SIZE * 2));
60	ATF_REQUIRE_MSG(error == 0, "brk: %s", strerror(errno));
61
62	/*
63	 * Attempt to set the break below minbrk.  This should have no effect.
64	 */
65	error = brk((void *)((intptr_t)oldbrk - 1));
66	ATF_REQUIRE_MSG(error == 0, "brk: %s", strerror(errno));
67	newbrk = sbrk(0);
68	ATF_REQUIRE_MSG(newbrk != (void *)-1, "sbrk: %s", strerror(errno));
69	ATF_REQUIRE(newbrk == oldbrk);
70}
71
72ATF_TC(sbrk_basic);
73ATF_TC_HEAD(sbrk_basic, tc)
74{
75	atf_tc_set_md_var(tc, "descr", "Verify basic sbrk() functionality");
76}
77ATF_TC_BODY(sbrk_basic, tc)
78{
79	void *newbrk, *oldbrk;
80	int *p;
81
82	oldbrk = sbrk(0);
83	ATF_REQUIRE_MSG(oldbrk != (void *)-1, "sbrk: %s", strerror(errno));
84	p = sbrk(sizeof(*p));
85	*p = 0;
86	ATF_REQUIRE(oldbrk == p);
87
88	newbrk = sbrk(-sizeof(*p));
89	ATF_REQUIRE_MSG(newbrk != (void *)-1, "sbrk: %s", strerror(errno));
90	ATF_REQUIRE(oldbrk == sbrk(0));
91
92	oldbrk = sbrk(PAGE_SIZE * 2 + 1);
93	ATF_REQUIRE_MSG(oldbrk != (void *)-1, "sbrk: %s", strerror(errno));
94	memset(oldbrk, 0, PAGE_SIZE * 2 + 1);
95	newbrk = sbrk(-(PAGE_SIZE * 2 + 1));
96	ATF_REQUIRE_MSG(newbrk != (void *)-1, "sbrk: %s", strerror(errno));
97	ATF_REQUIRE(sbrk(0) == oldbrk);
98}
99
100ATF_TC(mlockfuture);
101ATF_TC_HEAD(mlockfuture, tc)
102{
103	atf_tc_set_md_var(tc, "descr",
104	    "Verify that mlockall(MCL_FUTURE) applies to the data segment");
105}
106ATF_TC_BODY(mlockfuture, tc)
107{
108	void *oldbrk, *n, *newbrk;
109	int error;
110	char v;
111
112	error = mlockall(MCL_FUTURE);
113	ATF_REQUIRE_MSG(error == 0,
114	    "mlockall: %s", strerror(errno));
115
116	/*
117	 * Advance the break so that at least one page is added to the data
118	 * segment.  This page should be automatically faulted in to the address
119	 * space.
120	 */
121	oldbrk = sbrk(0);
122	ATF_REQUIRE(oldbrk != (void *)-1);
123	newbrk = sbrk(PAGE_SIZE * 2);
124	ATF_REQUIRE(newbrk != (void *)-1);
125
126	n = (void *)(((uintptr_t)oldbrk + PAGE_SIZE) & ~PAGE_SIZE);
127	v = 0;
128	error = mincore(n, PAGE_SIZE, &v);
129	ATF_REQUIRE_MSG(error == 0,
130	    "mincore: %s", strerror(errno));
131	ATF_REQUIRE_MSG((v & MINCORE_INCORE) != 0,
132	    "unexpected page flags %#x", v);
133
134	error = brk(oldbrk);
135	ATF_REQUIRE(error == 0);
136
137	error = munlockall();
138	ATF_REQUIRE_MSG(error == 0,
139	    "munlockall: %s", strerror(errno));
140}
141
142ATF_TP_ADD_TCS(tp)
143{
144	ATF_TP_ADD_TC(tp, brk_basic);
145	ATF_TP_ADD_TC(tp, sbrk_basic);
146	ATF_TP_ADD_TC(tp, mlockfuture);
147
148	return (atf_no_error());
149}
150