Date: Mon, 19 Apr 2010 00:46:48 +0000 From: Alexander Krizhanovsky <ak@natsys-lab.com> To: freebsd-hackers@freebsd.org Cc: ur4ina@gmail.com Subject: [PATCH] RUSAGE_THREAD Message-ID: <4BCBA7F8.3060109@natsys-lab.com>
next in thread | raw e-mail | index | archive | help
This is a multi-part message in MIME format. --------------030802090104090509010408 Content-Type: text/plain; charset=KOI8-R; format=flowed Content-Transfer-Encoding: 7bit Hi all! This patch implements per-thread rusage statistic (like RUSAGE_THREAD in Linux and RUSAGE_LWP in OpenSolaris). Unfortunately, we have to acquire a number of locks to read/update system and user times for current thread rusage information because it's also used for whole process statistic and needs to be zeroed. Any comments are very appreciated. It's tested against 8.0-RELEASE. Appropriate PR is submitted. -- Alexander Krizhanovsky NatSys Lab. (http://natsys-lab.com) tel: +7 (916) 717-3899, +7 (499) 747-6304 e-mail: ak@natsys-lab.com --------------030802090104090509010408 Content-Type: text/plain; name="patch-rusage-thread.txt" Content-Transfer-Encoding: 7bit Content-Disposition: inline; filename="patch-rusage-thread.txt" --- sys/sys/resource.h.orig 2009-10-25 01:10:29.000000000 +0000 +++ sys/sys/resource.h 2010-04-11 23:31:14.000000000 +0000 @@ -56,6 +56,7 @@ #define RUSAGE_SELF 0 #define RUSAGE_CHILDREN -1 +#define RUSAGE_THREAD 1 struct rusage { struct timeval ru_utime; /* user time used */ --- sys/kern/kern_resource.c.orig 2009-10-25 01:10:29.000000000 +0000 +++ sys/kern/kern_resource.c 2010-04-18 23:49:04.000000000 +0000 @@ -76,6 +76,7 @@ static void calcru1(struct proc *p, stru struct timeval *up, struct timeval *sp); static int donice(struct thread *td, struct proc *chgp, int n); static struct uidinfo *uilookup(uid_t uid); +static void ruxagg_tlock(struct proc *p, struct thread *td); /* * Resource controls and accounting. @@ -623,9 +624,7 @@ lim_cb(void *arg) return; PROC_SLOCK(p); FOREACH_THREAD_IN_PROC(p, td) { - thread_lock(td); - ruxagg(&p->p_rux, td); - thread_unlock(td); + ruxagg_tlock(p, td); } PROC_SUNLOCK(p); if (p->p_rux.rux_runtime > p->p_cpulimit * cpu_tickrate()) { @@ -836,9 +835,7 @@ calcru(struct proc *p, struct timeval *u FOREACH_THREAD_IN_PROC(p, td) { if (td->td_incruntime == 0) continue; - thread_lock(td); - ruxagg(&p->p_rux, td); - thread_unlock(td); + ruxagg_tlock(p, td); } calcru1(p, &p->p_rux, up, sp); } @@ -918,6 +915,29 @@ calcru1(struct proc *p, struct rusage_ex sp->tv_usec = su % 1000000; } +static void +calctru(struct thread *td) +{ + u_int64_t tu = cputick2usec(td->td_incruntime); + u_int64_t ut = td->td_uticks; + u_int64_t it = td->td_iticks; + u_int64_t st = td->td_sticks; + u_int64_t tt, uu, su; + + tt = ut + st + it; + if (!tt) { + /* Avoid divide by zero */ + st = 1; + tt = 1; + } + uu = td->td_ru.ru_utime.tv_usec + (ut * tu) / tt; + su = td->td_ru.ru_stime.tv_usec + (st * tu) / tt; + td->td_ru.ru_utime.tv_sec += uu / 1000000; + td->td_ru.ru_utime.tv_usec = uu % 1000000; + td->td_ru.ru_stime.tv_sec += su / 1000000; + td->td_ru.ru_stime.tv_usec = su % 1000000; +} + #ifndef _SYS_SYSPROTO_H_ struct getrusage_args { int who; @@ -939,10 +959,7 @@ getrusage(td, uap) } int -kern_getrusage(td, who, rup) - struct thread *td; - int who; - struct rusage *rup; +kern_getrusage(struct thread *td, int who, struct rusage *rup) { struct proc *p; int error; @@ -961,6 +978,13 @@ kern_getrusage(td, who, rup) calccru(p, &rup->ru_utime, &rup->ru_stime); break; + case RUSAGE_THREAD: + PROC_SLOCK(p); + ruxagg_tlock(p, td); + PROC_SUNLOCK(p); + *rup = td->td_ru; + break; + default: error = EINVAL; } @@ -1010,12 +1034,24 @@ ruxagg(struct rusage_ext *rux, struct th rux->rux_uticks += td->td_uticks; rux->rux_sticks += td->td_sticks; rux->rux_iticks += td->td_iticks; + + /* update thread rusage before ticks counters cleaning */ + calctru(td); + td->td_incruntime = 0; td->td_uticks = 0; td->td_iticks = 0; td->td_sticks = 0; } +static void +ruxagg_tlock(struct proc *p, struct thread *td) +{ + thread_lock(td); + ruxagg(&p->p_rux, td); + thread_unlock(td); +} + /* * Update the rusage_ext structure and fetch a valid aggregate rusage * for proc p if storage for one is supplied. @@ -1030,9 +1066,7 @@ rufetch(struct proc *p, struct rusage *r *ru = p->p_ru; if (p->p_numthreads > 0) { FOREACH_THREAD_IN_PROC(p, td) { - thread_lock(td); - ruxagg(&p->p_rux, td); - thread_unlock(td); + ruxagg_tlock(p, td); rucollect(ru, &td->td_ru); } } --------------030802090104090509010408--
Want to link to this message? Use this URL: <https://mail-archive.FreeBSD.org/cgi/mid.cgi?4BCBA7F8.3060109>