140 lines
1.9 KiB
C
140 lines
1.9 KiB
C
#include <u.h>
|
|
#include <libc.h>
|
|
#include <thread.h>
|
|
#include "threadimpl.h"
|
|
#include <tos.h>
|
|
|
|
int
|
|
threadid(void)
|
|
{
|
|
return _threadgetproc()->thread->id;
|
|
}
|
|
|
|
int
|
|
threadpid(int id)
|
|
{
|
|
int pid;
|
|
Proc *p;
|
|
Thread *t;
|
|
|
|
if (id < 0)
|
|
return -1;
|
|
if (id == 0)
|
|
return _threadgetproc()->pid;
|
|
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){
|
|
pid = p->pid;
|
|
unlock(&p->lock);
|
|
unlock(&_threadpq.lock);
|
|
return pid;
|
|
}
|
|
unlock(&p->lock);
|
|
}
|
|
unlock(&_threadpq.lock);
|
|
return -1;
|
|
}
|
|
|
|
int
|
|
threadsetgrp(int ng)
|
|
{
|
|
int og;
|
|
Thread *t;
|
|
|
|
t = _threadgetproc()->thread;
|
|
og = t->grp;
|
|
t->grp = ng;
|
|
return og;
|
|
}
|
|
|
|
int
|
|
threadgetgrp(void)
|
|
{
|
|
return _threadgetproc()->thread->grp;
|
|
}
|
|
|
|
void
|
|
threadsetname(char *fmt, ...)
|
|
{
|
|
int fd;
|
|
char buf[128];
|
|
va_list arg;
|
|
Proc *p;
|
|
Thread *t;
|
|
|
|
p = _threadgetproc();
|
|
t = p->thread;
|
|
if (t->cmdname)
|
|
free(t->cmdname);
|
|
va_start(arg, fmt);
|
|
t->cmdname = vsmprint(fmt, arg);
|
|
va_end(arg);
|
|
if(t->cmdname && p->nthreads == 1){
|
|
snprint(buf, sizeof buf, "#p/%lud/args", _tos->pid); //getpid());
|
|
if((fd = open(buf, OWRITE)) >= 0){
|
|
write(fd, t->cmdname, strlen(t->cmdname)+1);
|
|
close(fd);
|
|
}
|
|
}
|
|
}
|
|
|
|
char*
|
|
threadgetname(void)
|
|
{
|
|
return _threadgetproc()->thread->cmdname;
|
|
}
|
|
|
|
void**
|
|
threaddata(void)
|
|
{
|
|
return &_threadgetproc()->thread->udata[0];
|
|
}
|
|
|
|
void**
|
|
_workerdata(void)
|
|
{
|
|
return &_threadgetproc()->wdata;
|
|
}
|
|
|
|
void**
|
|
procdata(void)
|
|
{
|
|
return &_threadgetproc()->udata;
|
|
}
|
|
|
|
static Lock privlock;
|
|
static int privmask = 1;
|
|
|
|
int
|
|
tprivalloc(void)
|
|
{
|
|
int i;
|
|
|
|
lock(&privlock);
|
|
for(i=0; i<NPRIV; i++)
|
|
if(!(privmask&(1<<i))){
|
|
privmask |= 1<<i;
|
|
unlock(&privlock);
|
|
return i;
|
|
}
|
|
unlock(&privlock);
|
|
return -1;
|
|
}
|
|
|
|
void
|
|
tprivfree(int i)
|
|
{
|
|
if(i < 0 || i >= NPRIV)
|
|
abort();
|
|
lock(&privlock);
|
|
privmask &= ~(1<<i);
|
|
}
|
|
|
|
void**
|
|
tprivaddr(int i)
|
|
{
|
|
return &_threadgetproc()->thread->udata[i];
|
|
}
|