shithub: riscv

ref: f88a55e79b5bf656e7f9578d1318a955b9a4963a
dir: /sys/src/libthread/kill.c/

View raw version
#include <u.h>
#include <libc.h>
#include <thread.h>
#include "threadimpl.h"

static void tinterrupt(Proc*, Thread*);

static void
threadxxxgrp(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 void
threadxxx(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;
}

void
threadkillgrp(int grp)
{
	threadxxxgrp(grp, 1);
}

void
threadkill(int id)
{
	threadxxx(id, 1);
}

void
threadintgrp(int grp)
{
	threadxxxgrp(grp, 0);
}

void
threadint(int id)
{
	threadxxx(id, 0);
}

static int
writeprocctl(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 void
tinterrupt(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;
	}
}