1*fc2512cfSRobert Mustacchi /*
2*fc2512cfSRobert Mustacchi  * This file and its contents are supplied under the terms of the
3*fc2512cfSRobert Mustacchi  * Common Development and Distribution License ("CDDL"), version 1.0.
4*fc2512cfSRobert Mustacchi  * You may only use this file in accordance with the terms of version
5*fc2512cfSRobert Mustacchi  * 1.0 of the CDDL.
6*fc2512cfSRobert Mustacchi  *
7*fc2512cfSRobert Mustacchi  * A full copy of the text of the CDDL should have accompanied this
8*fc2512cfSRobert Mustacchi  * source.  A copy of the CDDL is also available via the Internet at
9*fc2512cfSRobert Mustacchi  * http://www.illumos.org/license/CDDL.
10*fc2512cfSRobert Mustacchi  */
11*fc2512cfSRobert Mustacchi 
12*fc2512cfSRobert Mustacchi /*
13*fc2512cfSRobert Mustacchi  * Copyright 2016 Joyent, Inc.
14*fc2512cfSRobert Mustacchi  */
15*fc2512cfSRobert Mustacchi 
16*fc2512cfSRobert Mustacchi /*
17*fc2512cfSRobert Mustacchi  * Basic tests for aligned_alloc(3C). Note that we test ENOMEM failure by
18*fc2512cfSRobert Mustacchi  * relying on the implementation of the current libc malloc. Specifically we go
19*fc2512cfSRobert Mustacchi  * through and add a mapping so we can't expand the heap and then use it up. If
20*fc2512cfSRobert Mustacchi  * the memory allocator is ever changed, this test will start failing, at which
21*fc2512cfSRobert Mustacchi  * point, it may not be worth the cost of keeping it around.
22*fc2512cfSRobert Mustacchi  */
23*fc2512cfSRobert Mustacchi 
24*fc2512cfSRobert Mustacchi #include <stdlib.h>
25*fc2512cfSRobert Mustacchi #include <errno.h>
26*fc2512cfSRobert Mustacchi #include <libproc.h>
27*fc2512cfSRobert Mustacchi #include <sys/sysmacros.h>
28*fc2512cfSRobert Mustacchi #include <sys/mman.h>
29*fc2512cfSRobert Mustacchi #include <sys/debug.h>
30*fc2512cfSRobert Mustacchi 
31*fc2512cfSRobert Mustacchi int
main(void)32*fc2512cfSRobert Mustacchi main(void)
33*fc2512cfSRobert Mustacchi {
34*fc2512cfSRobert Mustacchi 	pstatus_t status;
35*fc2512cfSRobert Mustacchi 	void *buf;
36*fc2512cfSRobert Mustacchi 
37*fc2512cfSRobert Mustacchi 	/*
38*fc2512cfSRobert Mustacchi 	 * Alignment must be sizeof (void *) (word) aligned.
39*fc2512cfSRobert Mustacchi 	 */
40*fc2512cfSRobert Mustacchi 	VERIFY3P(aligned_alloc(sizeof (void *) - 1, 16), ==, NULL);
41*fc2512cfSRobert Mustacchi 	VERIFY3S(errno, ==, EINVAL);
42*fc2512cfSRobert Mustacchi 
43*fc2512cfSRobert Mustacchi 	VERIFY3P(aligned_alloc(sizeof (void *) + 1, 16), ==, NULL);
44*fc2512cfSRobert Mustacchi 	VERIFY3S(errno, ==, EINVAL);
45*fc2512cfSRobert Mustacchi 
46*fc2512cfSRobert Mustacchi 
47*fc2512cfSRobert Mustacchi 	VERIFY3P(aligned_alloc(23, 16), ==, NULL);
48*fc2512cfSRobert Mustacchi 	VERIFY3S(errno, ==, EINVAL);
49*fc2512cfSRobert Mustacchi 
50*fc2512cfSRobert Mustacchi 	buf = aligned_alloc(sizeof (void *), 16);
51*fc2512cfSRobert Mustacchi 	VERIFY3P(buf, !=, NULL);
52*fc2512cfSRobert Mustacchi 	free(buf);
53*fc2512cfSRobert Mustacchi 
54*fc2512cfSRobert Mustacchi 	/*
55*fc2512cfSRobert Mustacchi 	 * Cause ENOMEM
56*fc2512cfSRobert Mustacchi 	 */
57*fc2512cfSRobert Mustacchi 	VERIFY0(proc_get_status(getpid(), &status));
58*fc2512cfSRobert Mustacchi 	VERIFY3P(mmap((caddr_t)P2ROUNDUP(status.pr_brkbase +
59*fc2512cfSRobert Mustacchi 	    status.pr_brksize, 0x1000), 0x1000,
60*fc2512cfSRobert Mustacchi 	    PROT_READ, MAP_ANON | MAP_FIXED | MAP_PRIVATE, -1, 0),
61*fc2512cfSRobert Mustacchi 	    !=, (void *)-1);
62*fc2512cfSRobert Mustacchi 
63*fc2512cfSRobert Mustacchi 	for (;;) {
64*fc2512cfSRobert Mustacchi 		if (malloc(16) == NULL)
65*fc2512cfSRobert Mustacchi 			break;
66*fc2512cfSRobert Mustacchi 	}
67*fc2512cfSRobert Mustacchi 
68*fc2512cfSRobert Mustacchi 	for (;;) {
69*fc2512cfSRobert Mustacchi 		if (aligned_alloc(sizeof (void *), 16) == NULL)
70*fc2512cfSRobert Mustacchi 			break;
71*fc2512cfSRobert Mustacchi 	}
72*fc2512cfSRobert Mustacchi 
73*fc2512cfSRobert Mustacchi 	VERIFY3P(aligned_alloc(sizeof (void *), 16), ==, NULL);
74*fc2512cfSRobert Mustacchi 	VERIFY3S(errno, ==, ENOMEM);
75*fc2512cfSRobert Mustacchi 
76*fc2512cfSRobert Mustacchi 	return (0);
77*fc2512cfSRobert Mustacchi }
78