Browse Source

port the RBT code to userland by making it part of libc.

src/lib/libc/gen/tree.c is a copy of src/sys/kern/subr_tree.c, but with
annotations for symbol visibility. changes to one should be reflected
in the other.
the malloc debug code that uses RB code is ported to RBT.
because libc provides the RBT code, procmap doesn't have to reach into
the kernel and build subr_tree.c itself now.
mild enthusiasm from many
ok guenther@
OPENBSD_6_2
dlg 7 years ago
parent
commit
b4e0da4e31
1 changed files with 11 additions and 10 deletions
  1. +11
    -10
      src/lib/libc/stdlib/malloc.c

+ 11
- 10
src/lib/libc/stdlib/malloc.c View File

@ -1,4 +1,4 @@
/* $OpenBSD: malloc.c,v 1.225 2017/05/13 07:11:29 otto Exp $ */
/* $OpenBSD: malloc.c,v 1.226 2017/06/19 03:06:26 dlg Exp $ */
/* /*
* Copyright (c) 2008, 2010, 2011, 2016 Otto Moerbeek <otto@drijf.net> * Copyright (c) 2008, 2010, 2011, 2016 Otto Moerbeek <otto@drijf.net>
* Copyright (c) 2012 Matthew Dempsky <matthew@openbsd.org> * Copyright (c) 2012 Matthew Dempsky <matthew@openbsd.org>
@ -2085,18 +2085,19 @@ struct malloc_leak {
}; };
struct leaknode { struct leaknode {
RB_ENTRY(leaknode) entry;
RBT_ENTRY(leaknode) entry;
struct malloc_leak d; struct malloc_leak d;
}; };
static int
leakcmp(struct leaknode *e1, struct leaknode *e2)
static inline int
leakcmp(const struct leaknode *e1, const struct leaknode *e2)
{ {
return e1->d.f < e2->d.f ? -1 : e1->d.f > e2->d.f; return e1->d.f < e2->d.f ? -1 : e1->d.f > e2->d.f;
} }
static RB_HEAD(leaktree, leaknode) leakhead;
RB_GENERATE_STATIC(leaktree, leaknode, entry, leakcmp)
static RBT_HEAD(leaktree, leaknode) leakhead;
RBT_PROTOTYPE(leaktree, leaknode, entry, leakcmp);
RBT_GENERATE(leaktree, leaknode, entry, leakcmp);
static void static void
putleakinfo(void *f, size_t sz, int cnt) putleakinfo(void *f, size_t sz, int cnt)
@ -2109,7 +2110,7 @@ putleakinfo(void *f, size_t sz, int cnt)
return; return;
key.d.f = f; key.d.f = f;
p = RB_FIND(leaktree, &leakhead, &key);
p = RBT_FIND(leaktree, &leakhead, &key);
if (p == NULL) { if (p == NULL) {
if (page == NULL || if (page == NULL ||
used >= MALLOC_PAGESIZE / sizeof(struct leaknode)) { used >= MALLOC_PAGESIZE / sizeof(struct leaknode)) {
@ -2122,7 +2123,7 @@ putleakinfo(void *f, size_t sz, int cnt)
p->d.f = f; p->d.f = f;
p->d.total_size = sz * cnt; p->d.total_size = sz * cnt;
p->d.count = cnt; p->d.count = cnt;
RB_INSERT(leaktree, &leakhead, p);
RBT_INSERT(leaktree, &leakhead, p);
} else { } else {
p->d.total_size += sz * cnt; p->d.total_size += sz * cnt;
p->d.count += cnt; p->d.count += cnt;
@ -2151,7 +2152,7 @@ dump_leaks(int fd)
malloc_leaks = MMAP(MALLOC_PAGESIZE); malloc_leaks = MMAP(MALLOC_PAGESIZE);
if (malloc_leaks != MAP_FAILED) if (malloc_leaks != MAP_FAILED)
memset(malloc_leaks, 0, MALLOC_PAGESIZE); memset(malloc_leaks, 0, MALLOC_PAGESIZE);
RB_FOREACH(p, leaktree, &leakhead) {
RBT_FOREACH(p, leaktree, &leakhead) {
snprintf(buf, sizeof(buf), "%18p %7zu %6u %6zu\n", p->d.f, snprintf(buf, sizeof(buf), "%18p %7zu %6u %6zu\n", p->d.f,
p->d.total_size, p->d.count, p->d.total_size / p->d.count); p->d.total_size, p->d.count, p->d.total_size / p->d.count);
write(fd, buf, strlen(buf)); write(fd, buf, strlen(buf));
@ -2316,7 +2317,7 @@ malloc_dump(int fd, int poolno, struct dir_info *pool)
pool->delayed_chunks[i] = NULL; pool->delayed_chunks[i] = NULL;
} }
/* XXX leak when run multiple times */ /* XXX leak when run multiple times */
RB_INIT(&leakhead);
RBT_INIT(leaktree, &leakhead);
malloc_dump1(fd, poolno, pool); malloc_dump1(fd, poolno, pool);
errno = saved_errno; errno = saved_errno;
} }


Loading…
Cancel
Save