plan9fox/sys/src/cmd/5e/seg.c
aiju 0b22dfd1f6 5e: added note support
added wstat / fwstat / alarm
2011-06-25 16:17:20 +02:00

137 lines
2.2 KiB
C

#include <u.h>
#include <libc.h>
#include <thread.h>
#include "dat.h"
#include "fns.h"
Segment *
newseg(u32int start, u32int size, int idx)
{
Segment *s;
s = emallocz(sizeof *s);
incref(s);
s->start = start;
s->size = size;
s->dref = emalloc(size + sizeof(Ref));
memset(s->dref, 0, sizeof(Ref));
incref(s->dref);
s->data = s->dref + 1;
if(idx == SEGBSS)
s->flags = SEGFLLOCK;
P->S[idx] = s;
return s;
}
void
freesegs(void)
{
Segment **s;
for(s = P->S; s < P->S + SEGNUM; s++) {
if(*s == nil)
continue;
if(decref((*s)->dref) == 0)
free((*s)->dref);
if(decref(*s) == 0)
free(*s);
*s = nil;
}
}
void *
vaddr(u32int addr, u32int len, Segment **seg)
{
Segment **ss, *s;
for(ss = P->S; ss < P->S + SEGNUM; ss++) {
if(*ss == nil)
continue;
s = *ss;
if(addr >= s->start && addr < s->start + s->size) {
if(addr + len > s->start + s->size)
break;
if(s->flags & SEGFLLOCK)
rlock(&s->rw);
*seg = s;
return (char *)s->data + (addr - s->start);
}
}
suicide("fault %.8ux (%d) @ %.8ux", addr, len, P->R[15]);
return nil;
}
void *
vaddrnol(u32int addr, u32int len)
{
Segment *seg;
void *ret;
ret = vaddr(addr, len, &seg);
segunlock(seg);
return ret;
}
/* might be made a macro for hurr durr performance */
void
segunlock(Segment *s)
{
if(s->flags & SEGFLLOCK)
runlock(&s->rw);
}
void *
copyifnec(u32int addr, int len, int *copied)
{
void *targ, *ret;
Segment *seg;
targ = vaddr(addr, len > 0 ? len : 0, &seg);
if((seg->flags & SEGFLLOCK) == 0) {
*copied = 0;
return targ;
}
if(len < 0)
len = strlen(targ) + 1;
ret = emalloc(len);
setmalloctag(ret, getcallerpc(&addr));
memcpy(ret, targ, len);
segunlock(seg);
*copied = 1;
return ret;
}
void *
bufifnec(u32int addr, int len, int *buffered)
{
void *targ, *v;
Segment *seg;
targ = vaddr(addr, len, &seg);
if((seg->flags & SEGFLLOCK) == 0) {
*buffered = 0;
return targ;
}
segunlock(seg);
*buffered = 1;
v = emalloc(len);
setmalloctag(v, getcallerpc(&addr));
return v;
}
void
copyback(u32int addr, int len, void *data)
{
void *targ;
Segment *seg;
if(len <= 0) {
free(data);
return;
}
targ = vaddr(addr, len, &seg);
memmove(targ, data, len);
segunlock(seg);
free(data);
}