ref: c1b943384c1cf8bddbad7df63319d597710f943d
dir: /proc.c/
#include <u.h>
#include <libc.h>
#include <thread.h>
#include <bio.h>
#include <mach.h>
#include <ctype.h>
#include "dat.h"
#include "fns.h"
#pragma pack on
typedef struct Tos Tos;
struct Tos {
struct /* Per process profiling */
{
ulong pp; /* known to be 0(ptr) */
ulong next; /* known to be 4(ptr) */
ulong last;
ulong first;
ulong pid;
ulong what;
} prof;
uvlong cyclefreq; /* cycle clock frequency if there is one, 0 otherwise */
vlong kcycles; /* cycles spent in kernel */
vlong pcycles; /* cycles spent in process (kernel + user) */
ulong pid; /* might as well put the pid here */
ulong clock;
/* top of stack is here */
};
#pragma pack off
Process plist;
Lock plistlock;
void
initproc(void)
{
P = emallocz(sizeof(Process));
P->pid = getpid();
P->fd = newfd();
incref(&nproc);
plist.prev = P;
plist.next = P;
P->prev = &plist;
P->next = &plist;
if(vfp)
resetvfp();
else
resetfpa();
}
void
addproc(Process *p)
{
lock(&plistlock);
p->prev = plist.prev;
p->next = &plist;
plist.prev->next = p;
plist.prev = p;
unlock(&plistlock);
}
void
remproc(Process *p)
{
lock(&plistlock);
p->prev->next = p->next;
p->next->prev = p->prev;
unlock(&plistlock);
}
Process *
findproc(int pid)
{
Process *p;
lock(&plistlock);
for(p = plist.next; p != &plist; p = p->next)
if(p->pid == pid)
break;
unlock(&plistlock);
if(p != &plist)
return p;
return nil;
}
static void
copyname(char *file)
{
char *p;
p = strrchr(file, '/');
if(p == nil)
p = file;
else
p++;
strncpy(P->name, p, NAMEMAX);
if(P->path != nil && decref(P->path) == 0)
free(P->path);
P->path = emallocz(sizeof(Ref) + strlen(file)+1);
incref(P->path);
strcpy((char*)(P->path + 1), file);
}
static void
initstack(int argc, char **argv)
{
ulong tos, sp, ap, size, i, len;
tos = (mach->utop & ~7) - sizeof(Tos) * 2;
sp = tos;
size = 8;
for(i = 0; i < argc; i++)
size += strlen(argv[i]) + 5;
sp -= size;
sp &= ~7;
P->R[0] = tos;
P->R[1] = mach->utop - 4;
P->R[13] = sp;
*(ulong *) vaddrnol(sp, 4, AWR) = argc;
sp += 4;
ap = sp + (argc + 1) * 4;
for(i = 0; i < argc; i++) {
*(ulong *) vaddrnol(sp, 4, AWR) = ap;
sp += 4;
len = strlen(argv[i]) + 1;
memcpy(vaddrnol(ap, len, AWR), argv[i], len);
ap += len;
}
*(ulong *) vaddrnol(sp, 4, AWR) = 0;
inittos();
}
void
inittos(void)
{
ulong tos;
tos = (mach->utop & ~7) - sizeof(Tos) * 2;
((Tos *) vaddrnol(tos, sizeof(Tos), AWR))->pid = P->pid;
}
static int
loadscript(int fd, char *file, int argc, char **argv)
{
char buf[513], *p, **q, **nargv;
int rc, nargc, i;
seek(fd, 0, 0);
rc = readn(fd, buf, 512);
if(rc <= 0)
goto invalid;
close(fd);
buf[rc] = 0;
p = strchr(buf, '\n');
if(p == nil)
goto invalid;
*p = 0;
while(isspace(*--p))
*p = 0;
nargc = 0;
p = buf + 2;
while(*p) {
while(*p && isspace(*p))
p++;
nargc++;
while(*p && !isspace(*p))
p++;
}
if(nargc == 0)
goto invalid;
nargv = emallocz(sizeof(char *) * (nargc + argc));
q = nargv;
p = buf + 2;
while(*p) {
while(*p && isspace(*p))
p++;
*(p-1) = 0;
*q++ = p;
while(*p && !isspace(*p))
p++;
}
*q++ = file;
for(i = 1; i < argc; i++)
*q++ = argv[i];
rc = loadtext(*nargv, argc + nargc, nargv);
free(nargv);
return rc;
invalid:
werrstr("exec header invalid");
return -1;
}
u32int
hookaddr(char *name)
{
Symbol s;
if(lookup(name, nil, &s) == -1)
return -1;
return s.value;
}
static void
inithooks(void)
{
mallocaddr = hookaddr("poolalloc");
reallocaddr = hookaddr("poolrealloc");
freeaddr = hookaddr("poolfree");
setmalloctagaddr = hookaddr("setmalloctag");
setrealloctagaddr = hookaddr("setrealloctag");
vtraceaddr = hookaddr("vtrace");
}
int
loadtext(char *file, int argc, char **argv)
{
int fd;
Fhdr fp;
Segment *text, *data, *bss, *stk;
char buf[2];
fd = open(file, OREAD);
if(fd < 0) return -1;
if(pread(fd, buf, 2, 0) == 2 && buf[0] == '#' && buf[1] == '!')
return loadscript(fd, file, argc, argv);
seek(fd, 0, 0);
if(crackhdr(fd, &fp) == 0 || fp.magic != E_MAGIC) {
werrstr("exec header invalid");
return -1;
}
copyname(file);
P->notehandler = P->innote = P->notein = P->noteout = 0;
freesegs();
memset(P->R, 0, sizeof(P->R));
P->CPSR = 0;
text = newseg(fp.txtaddr - fp.hdrsz, fp.txtsz + fp.hdrsz, SEGTEXT);
data = newseg(fp.dataddr, fp.datsz, SEGDATA);
bss = newseg(fp.dataddr + fp.datsz, fp.bsssz, SEGBSS);
stk = newseg((mach->utop & ~7) - STACKSIZE, STACKSIZE, SEGSTACK);
if(check){
markvalid(text, 0, fp.txtsz+fp.hdrsz, MARKALLOC|MARKINIT);
markvalid(data, 0, fp.datsz, MARKALLOC|MARKINIT);
markvalid(bss, 0, fp.bsssz, MARKALLOC|MARKINIT);
markvalid(stk, 0, stk->size, MARKALLOC|MARKINIT);
}
seek(fd, fp.txtoff - fp.hdrsz, 0);
if(readn(fd, text->data, fp.txtsz + fp.hdrsz) < fp.txtsz + fp.hdrsz)
sysfatal("%r");
seek(fd, fp.datoff, 0);
if(readn(fd, data->data, fp.datsz) < fp.datsz)
sysfatal("%r");
memset(bss->data, 0, bss->size);
P->R[15] = fp.entry;
if(syminit(fd, &fp) < 0)
fprint(2, "initializing symbol table: %r\n");
if(check)
inithooks();
close(fd);
fdclear(P->fd);
initstack(argc, argv);
if(vfp)
resetvfp();
else
resetfpa();
return 0;
}
void
cherrstr(char *str, ...)
{
va_list va;
va_start(va, str);
vsnprint(P->errbuf, ERRMAX, str, va);
va_end(va);
}
u32int
noteerr(u32int x, u32int y)
{
if(((int)x) >= ((int)y))
return x;
rerrstr(P->errbuf, ERRMAX);
return x;
}
Fd *
newfd(void)
{
Fd *fd;
fd = emallocz(sizeof(*fd));
incref(fd);
return fd;
}
Fd *
copyfd(Fd *old)
{
Fd *new;
rlock(old);
new = newfd();
if(old->nfds > 0) {
new->nfds = old->nfds;
new->fds = emalloc(old->nfds);
memcpy(new->fds, old->fds, old->nfds);
}
runlock(old);
return new;
}
void
fddecref(Fd *fd)
{
if(decref(fd) == 0) {
free(fd->fds);
free(fd);
}
}
int
iscexec(Fd *fd, int n)
{
int r;
r = 0;
rlock(fd);
if(n / 8 < fd->nfds)
r = (fd->fds[n / 8] & (1 << (n % 8))) != 0;
runlock(fd);
return r;
}
void
setcexec(Fd *fd, int n, int status)
{
int old;
wlock(fd);
if(n / 8 >= fd->nfds) {
if(status == 0) {
wunlock(fd);
return;
}
old = fd->nfds;
fd->nfds = (n / 8) + 1;
fd->fds = erealloc(fd->fds, fd->nfds);
memset(fd->fds + old, 0, fd->nfds - old);
}
if(status == 0)
fd->fds[n / 8] &= ~(1 << (n % 8));
else
fd->fds[n / 8] |= (1 << (n % 8));
wunlock(fd);
}
void
fdclear(Fd *fd)
{
int i, j, k;
wlock(fd);
if(fd->nfds == 0) {
wunlock(fd);
return;
}
for(i = 0; i < fd->nfds; i++) {
j = fd->fds[i];
for(k = 0; k < 8; k++)
if(j & (1<<k))
close(8 * i + k);
}
free(fd->fds);
fd->nfds = 0;
fd->fds = nil;
wunlock(fd);
}
/* call this from a notehandler if you don't want the front to fall off */
void
addnote(char *msg)
{
int new;
new = P->notein + 1;
if((new - P->noteout) % NNOTE == 0)
return;
strncpy(P->notes[P->notein % NNOTE], msg, ERRMAX - 1);
P->notein = new;
}
/* the following code is not for the weak of heart */
void
donote(char *msg, ulong type)
{
extern int vtrace;
int rc;
u32int *ureg, *sp, uregp, msgp;
char *msgb;
if(P->notehandler == 0)
exits(msg);
clrex();
uregp = P->R[13] - 18 * 4;
ureg = vaddrnol(uregp, 18 * 4, 0);
memcpy(ureg, P->R, 15 * 4);
ureg[15] = type;
ureg[16] = P->CPSR;
ureg[17] = P->R[15];
P->R[13] = uregp;
msgp = P->R[13] -= ERRMAX;
msgb = vaddrnol(msgp, ERRMAX, 0);
strncpy(msgb, msg, ERRMAX);
P->R[13] -= 3 * 4;
sp = vaddrnol(P->R[13], 3 * 4, 0);
sp[0] = 0;
sp[2] = msgp;
P->R[0] = uregp;
P->R[15] = P->notehandler;
P->innote = 1;
switch(rc = setjmp(P->notejmp) - 1) {
case -1:
for(;;) {
if(ultraverbose || vtrace)
dump();
step();
}
case NDFLT:
exits(msg);
case NCONT:
break;
default:
sysfatal("unhandled noted argument %d", rc);
}
P->innote = 0;
ureg = vaddrnol(uregp, 18 * 4, 0); /* just to be sure */
memcpy(P->R, ureg, 15 * 4);
P->CPSR = ureg[16];
P->R[15] = ureg[17];
}