summaryrefslogtreecommitdiff
path: root/usr/src/cmd/vndadm/test/tst/lib/create.enomem.c
blob: 9203e369ae784fc36b725a0d06713f277837c176 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright (c) 2014 Joyent, Inc.  All rights reserved.
 */

/*
 * Verify that we can't allocate a handle when in an ENOMEM situation.
 */

#include <procfs.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/mman.h>
#include <sys/sysmacros.h>
#include <assert.h>
#include <strings.h>

#include <libvnd.h>

int
main(int argc, const char *argv[])
{
	int fd;
	int syserr;
	vnd_errno_t vnderr;
	vnd_handle_t *vhp;
	pstatus_t status;
	void *addr;

	if (argc < 2) {
		(void) fprintf(stderr, "missing arguments...\n");
		return (1);
	}

	if (strlen(argv[1]) >= LIBVND_NAMELEN) {
		(void) fprintf(stderr, "vnic name too long...\n");
		return (1);
	}

	fd = open("/proc/self/status", O_RDONLY);
	if (fd < 0)
		exit(1);
	if (read(fd, &status, sizeof (status)) != sizeof (status))
		exit(1);

	addr = mmap((caddr_t)P2ROUNDUP(status.pr_brkbase +
	    status.pr_brksize, 0x1000), 0x1000,
	    PROT_READ, MAP_ANON | MAP_FIXED | MAP_PRIVATE, -1, 0);
	if (addr == (void *)-1) {
		perror("mmap");
		exit(1);
	}

	/* malloc an approximate size of the vnd_handle_t */
	for (;;) {
		void *buf;

		buf = malloc(8);
		if (buf == NULL)
			break;
	}

	for (;;) {
		void *buf;

		buf = malloc(4);
		if (buf == NULL)
			break;
	}

	vhp = vnd_create(NULL, argv[1], argv[1], &vnderr, &syserr);
	assert(vhp == NULL);
	assert(vnderr == VND_E_NOMEM);
	assert(syserr == 0);

	return (0);
}