test parse_time
git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@14323 ec53bebd-3082-4978-b11e-865c3cabbd6b
This commit is contained in:
Love Hörnquist Åstrand
195
lib/roken/test-mem.c
Normal file
195
lib/roken/test-mem.c
Normal file
@@ -0,0 +1,195 @@
|
||||
/*
|
||||
* Copyright (c) 1999 - 2004 Kungliga Tekniska H<>gskolan
|
||||
* (Royal Institute of Technology, Stockholm, Sweden).
|
||||
* 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. Neither the name of the Institute nor the names of its contributors
|
||||
* may be used to endorse or promote products derived from this software
|
||||
* without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``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 INSTITUTE OR CONTRIBUTORS 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.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include <config.h>
|
||||
#endif
|
||||
#ifdef HAVE_SYS_MMAN_H
|
||||
#include <sys/mman.h>
|
||||
#endif
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <err.h>
|
||||
#include <roken.h>
|
||||
|
||||
#include "test-mem.h"
|
||||
|
||||
RCSID("$Id$");
|
||||
|
||||
/* #undef HAVE_MMAP */
|
||||
|
||||
struct {
|
||||
void *start;
|
||||
size_t size;
|
||||
void *data_start;
|
||||
size_t data_size;
|
||||
enum rk_test_mem_type type;
|
||||
} map;
|
||||
|
||||
struct sigaction sa, osa;
|
||||
|
||||
char *testname;
|
||||
|
||||
static RETSIGTYPE
|
||||
segv_handler(int sig)
|
||||
{
|
||||
int fd;
|
||||
char msg[] = "SIGSEGV i current test: ";
|
||||
|
||||
fd = open("/dev/stdout", O_WRONLY, 0600);
|
||||
if (fd >= 0) {
|
||||
write(fd, msg, sizeof(msg) - 1);
|
||||
write(fd, testname, strlen(testname));
|
||||
write(fd, "\n", 1);
|
||||
close(fd);
|
||||
}
|
||||
_exit(1);
|
||||
}
|
||||
|
||||
#define TESTREC() \
|
||||
if (testname) \
|
||||
errx(1, "test %s run recursively on %s", name, testname); \
|
||||
testname = strdup(name); \
|
||||
if (testname == NULL) \
|
||||
errx(1, "malloc");
|
||||
|
||||
|
||||
void *
|
||||
rk_test_mem_alloc(enum rk_test_mem_type type, const char *name,
|
||||
void *buf, size_t size)
|
||||
{
|
||||
#ifndef HAVE_MMAP
|
||||
unsigned char *p;
|
||||
|
||||
TESTREC();
|
||||
|
||||
p = malloc(size + 2);
|
||||
if (p == NULL)
|
||||
errx(1, "malloc");
|
||||
map.type = type;
|
||||
map.start = p;
|
||||
map.size = size + 2;
|
||||
p[0] = 0xff;
|
||||
p[map.size] = 0xff;
|
||||
map.data_start = p + 1;
|
||||
#else
|
||||
unsigned char *p;
|
||||
int flags, ret, fd;
|
||||
size_t pagesize = getpagesize();
|
||||
|
||||
TESTREC();
|
||||
|
||||
map.type = type;
|
||||
|
||||
#ifdef MAP_ANON
|
||||
flags = MAP_ANON;
|
||||
fd = -1;
|
||||
#else
|
||||
flags = 0;
|
||||
fd = open ("/dev/zero", O_RDONLY);
|
||||
if(fd < 0)
|
||||
err (1, "open /dev/zero");
|
||||
#endif
|
||||
flags |= MAP_PRIVATE;
|
||||
|
||||
map.size = size + pagesize - (size % pagesize) + pagesize * 2;
|
||||
|
||||
p = (char *)mmap(0, map.size, PROT_READ | PROT_WRITE,
|
||||
flags, fd, 0);
|
||||
if (p == (unsigned char *)MAP_FAILED)
|
||||
err (1, "mmap");
|
||||
|
||||
map.start = p;
|
||||
|
||||
ret = mprotect (p, pagesize, 0);
|
||||
if (ret < 0)
|
||||
err (1, "mprotect");
|
||||
|
||||
ret = mprotect (p + map.size - pagesize, pagesize, 0);
|
||||
if (ret < 0)
|
||||
err (1, "mprotect");
|
||||
|
||||
switch (type) {
|
||||
case RK_TM_OVERRUN:
|
||||
map.data_start = p + map.size - pagesize - size;
|
||||
break;
|
||||
case RK_TM_UNDERRUN:
|
||||
map.data_start = p + pagesize;
|
||||
break;
|
||||
default:
|
||||
abort();
|
||||
}
|
||||
#endif
|
||||
sigemptyset (&sa.sa_mask);
|
||||
sa.sa_flags = 0;
|
||||
#ifdef SA_RESETHAND
|
||||
sa.sa_flags |= SA_RESETHAND;
|
||||
#endif
|
||||
sa.sa_handler = segv_handler;
|
||||
sigaction (SIGSEGV, &sa, &osa);
|
||||
|
||||
map.data_size = size;
|
||||
if (buf)
|
||||
memcpy(map.data_start, buf, size);
|
||||
return map.data_start;
|
||||
}
|
||||
|
||||
void
|
||||
rk_test_mem_free(const char *map_name)
|
||||
{
|
||||
#ifndef HAVE_MMAP
|
||||
unsigned char *p = map.start;
|
||||
|
||||
if (testname == NULL)
|
||||
errx(1, "test_mem_free call on no free");
|
||||
|
||||
if (p[0] != 0xff)
|
||||
errx(1, "%s: %s underrun %x\n", testname, map_name, p[0]);
|
||||
if (p[map.size] != 0xff)
|
||||
errx(1, "%s: %s overrun %x\n", testname, map_name, p[map.size - 1]);
|
||||
free(map.start);
|
||||
#else
|
||||
int ret;
|
||||
|
||||
if (testname == NULL)
|
||||
errx(1, "test_mem_free call on no free");
|
||||
|
||||
ret = munmap (map.start, map.size);
|
||||
if (ret < 0)
|
||||
err (1, "munmap");
|
||||
#endif
|
||||
free(testname);
|
||||
testname = NULL;
|
||||
|
||||
sigaction (SIGSEGV, &osa, NULL);
|
||||
}
|
||||
Reference in New Issue
Block a user