109 lines · 1.6 KB · C
RawDownload
#include <u.h>#include <libc.h>#include <thread.h>#include "threadimpl.h" static void tinterrupt(Proc*, Thread*); static voidthreadxxxgrp(int grp, int dokill){	Proc *p;	Thread *t; 	lock(&_threadpq.lock);	for(p=_threadpq.head; p; p=p->next){		lock(&p->lock);		for(t=p->threads.head; t; t=t->nextt)			if(t->grp == grp){				if(dokill)					t->moribund = 1;				tinterrupt(p, t);			}		unlock(&p->lock);	}	unlock(&_threadpq.lock);	_threadbreakrendez();} static voidthreadxxx(int id, int dokill){	Proc *p;	Thread *t; 	lock(&_threadpq.lock);	for(p=_threadpq.head; p; p=p->next){		lock(&p->lock);		for(t=p->threads.head; t; t=t->nextt)			if(t->id == id){				if(dokill)					t->moribund = 1;				tinterrupt(p, t);				unlock(&p->lock);				unlock(&_threadpq.lock);				_threadbreakrendez();				return;			}		unlock(&p->lock);	}	unlock(&_threadpq.lock);	_threaddebug(DBGNOTE, "Can't find thread to kill");	return;} voidthreadkillgrp(int grp){	threadxxxgrp(grp, 1);} voidthreadkill(int id){	threadxxx(id, 1);} voidthreadintgrp(int grp){	threadxxxgrp(grp, 0);} voidthreadint(int id){	threadxxx(id, 0);} static intwriteprocctl(int pid, char *ctl){	char buf[32];	int fd; 	snprint(buf, sizeof(buf), "/proc/%lud/ctl", (ulong)pid);	fd = open(buf, OWRITE|OCEXEC);	if(fd < 0)		return -1;	if(write(fd, ctl, strlen(ctl)) < 0){		close(fd);		return -1;	}	close(fd);	return 0;} static voidtinterrupt(Proc *p, Thread *t){	switch(t->state){	case Running:		if(writeprocctl(p->pid, "interrupt") < 0)			postnote(PNPROC, p->pid, "threadint");		break;	case Rendezvous:		_threadflagrendez(t);		break;	}}