xref: /src/usr.bin/xlint/common/mem.c

/*	$NetBSD: mem.c,v 1.11 2020/12/29 11:35:11 rillig Exp $	*/

/*
 * Copyright (c) 1994, 1995 Jochen Pohl
 * All Rights Reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by Jochen Pohl for
 *	The NetBSD Project.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
#if defined(__RCSID) && !defined(lint)
__RCSID("$NetBSD: mem.c,v 1.11 2020/12/29 11:35:11 rillig Exp $");
#endif

#include <sys/param.h>
#include <sys/types.h>
#include <sys/mman.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>

#include "lint.h"

void *
xmalloc(size_t s)
{
	void	*p;

	if ((p = malloc(s)) == NULL)
		nomem();
	return p;
}

void *
xcalloc(size_t n, size_t s)
{
	void	*p;

	if ((p = calloc(n, s)) == NULL)
		nomem();
	return p;
}

void *
xrealloc(void *p, size_t s)
{
	void *n;

	if ((n = realloc(p, s)) == NULL) {
		free(p);
		nomem();
	}
	p = n;
	return p;
}

char *
xstrdup(const char *s)
{
	char	*s2;

	if ((s2 = strdup(s)) == NULL)
		nomem();
	return s2;
}

void
nomem(void)
{

	errx(1, "virtual memory exhausted");
}

void
xasprintf(char **buf, const char *fmt, ...)
{
	int e;
	va_list ap;

	va_start(ap, fmt);
	e = vasprintf(buf, fmt, ap);
	va_end(ap);
	if (e < 0)
		nomem();
}

#if defined(MAP_ANONYMOUS) && !defined(MAP_ANON)
#define	MAP_ANON	MAP_ANONYMOUS
#endif

void *
xmapalloc(size_t len)
{
	static const int prot = PROT_READ | PROT_WRITE;
	static int fd = -1;
	void *p;
#ifdef MAP_ANON
	static const int flags = MAP_ANON | MAP_PRIVATE;
#else
	static const int flags = MAP_PRIVATE;

	if (fd == -1) {
		if ((fd = open("/dev/zero", O_RDWR)) == -1)
			err(1, "Cannot open `/dev/zero'");
	}
#endif
	p = mmap(NULL, len, prot, flags, fd, (off_t)0);
	if (p == (void *)-1)
		err(1, "Cannot map memory for %lu bytes", (unsigned long)len);
	return p;
}