xv6-cs450/umalloc.c

91 lines
1.6 KiB
C
Raw Normal View History

#include "types.h"
#include "stat.h"
#include "user.h"
#include "param.h"
2006-09-06 19:27:19 +02:00
// Memory allocator by Kernighan and Ritchie, The C programming Language,
// 2nd ed. Section 8.7.
typedef long Align;
union header {
struct {
union header *ptr;
uint size;
} s;
Align x;
};
typedef union header Header;
static Header base;
static Header *freep = 0;
void
free(void *ap)
{
Header *bp, *p;
2006-08-24 21:24:36 +02:00
2006-09-06 19:27:19 +02:00
bp = (Header*) ap - 1;
for(p = freep; !(bp > p && bp < p->s.ptr); p = p->s.ptr)
if(p >= p->s.ptr && (bp > p || bp < p->s.ptr))
break;
2006-09-06 19:27:19 +02:00
if(bp + bp->s.size == p->s.ptr) {
bp->s.size += p->s.ptr->s.size;
bp->s.ptr = p->s.ptr->s.ptr;
} else
bp->s.ptr = p->s.ptr;
2006-09-06 19:27:19 +02:00
if(p + p->s.size == bp) {
p->s.size += bp->s.size;
p->s.ptr = bp->s.ptr;
} else
p->s.ptr = bp;
freep = p;
}
2006-09-06 19:27:19 +02:00
static Header*
morecore(uint nu)
{
2007-08-09 19:32:40 +02:00
char *p;
Header *hp;
2006-09-06 19:27:19 +02:00
if(nu < PAGE)
nu = PAGE;
2007-08-09 19:32:40 +02:00
p = sbrk(nu * sizeof(Header));
if(p == (char*) -1)
return 0;
2007-08-09 19:32:40 +02:00
hp = (Header*)p;
hp->s.size = nu;
free((void*)(hp + 1));
return freep;
}
2006-09-06 19:27:19 +02:00
void*
malloc(uint nbytes)
{
Header *p, *prevp;
uint nunits;
nunits = (nbytes + sizeof(Header) - 1)/sizeof(Header) + 1;
2006-09-06 19:27:19 +02:00
if((prevp = freep) == 0) {
base.s.ptr = freep = prevp = &base;
base.s.size = 0;
}
2006-09-06 19:27:19 +02:00
for(p = prevp->s.ptr; ; prevp = p, p = p->s.ptr) {
if(p->s.size >= nunits) {
if(p->s.size == nunits)
prevp->s.ptr = p->s.ptr;
else {
p->s.size -= nunits;
p += p->s.size;
p->s.size = nunits;
}
freep = prevp;
2006-09-06 19:27:19 +02:00
return (void*) (p + 1);
}
2006-09-06 19:27:19 +02:00
if(p == freep)
if((p = morecore(nunits)) == 0)
return 0;
}
}