diff options
| -rw-r--r-- | hurd/hurdprio.c | 63 |
1 files changed, 27 insertions, 36 deletions
diff --git a/hurd/hurdprio.c b/hurd/hurdprio.c index b1e0c06..66bd5cc 100644 --- a/hurd/hurdprio.c +++ b/hurd/hurdprio.c @@ -17,8 +17,9 @@ write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ -#include <hurd/resource.h> #include <hurd.h> +#include <hurd/resource.h> +#include <sys/mman.h> #include <unistd.h> error_t @@ -35,64 +36,54 @@ _hurd_priority_which_map (enum __priority_which which, int who, switch (which) { + default: + return EINVAL; + case PRIO_PROCESS: - npids = 1; - pids[0] = who ?: getpid (); /* XXX function could special-case self? */ - err = 0; + err = (*function) (who ?: getpid (), 0); /* XXX special-case self? */ break; case PRIO_PGRP: err = __USEPORT (PROC, __proc_getpgrppids (port, who, &pids, &npids)); + for (i = 0; !err && i < npids; ++i) + err = (*function) (pids[i], 0); break; case PRIO_USER: if (who == 0) who = geteuid (); err = __USEPORT (PROC, __proc_getallpids (port, &pids, &npids)); - break; - - default: - return EINVAL; - } - - for (i = 0; !err && i < npids; ++i) - { - if (which == PRIO_USER) + for (i = 0; !err && i < npids; ++i) { /* Get procinfo to check the owner. */ int *oldpi = pi; mach_msg_type_number_t oldpisize = pisize; char *tw = 0; size_t twsz = 0; - if (err = __USEPORT (PROC, __proc_getprocinfo (port, pids[i], - &pi_flags, - &pi, &pisize, - &tw, &twsz))) - continue; - if (twsz) - /* Gratuitous. */ - __vm_deallocate (__mach_task_self (), (vm_address_t) tw, twsz); - if (pi != oldpi && oldpi != pibuf) - /* Old buffer from last call was not reused; free it. */ - __vm_deallocate (__mach_task_self (), - (vm_address_t) oldpi, oldpisize * sizeof pi[0]); + err = __USEPORT (PROC, __proc_getprocinfo (port, pids[i], + &pi_flags, + &pi, &pisize, + &tw, &twsz)); + if (!err) + { + if (twsz) /* Gratuitous. */ + __munmap (tw, twsz); + if (pi != oldpi && oldpi != pibuf) + /* Old buffer from last call was not reused; free it. */ + __munmap (oldpi, oldpisize * sizeof pi[0]); - pip = (struct procinfo *) pi; - if (pip->owner != (uid_t) who) - continue; + pip = (struct procinfo *) pi; + if (pip->owner == (uid_t) who) + err = (*function) (pids[i], pip); + } } - else - pip = NULL; - - err = (*function) (pids[i], pip); + break; } if (pids != pidbuf) - __vm_deallocate (__mach_task_self (), - (vm_address_t) pids, npids * sizeof pids[0]); + __munmap (pids, npids * sizeof pids[0]); if (pi != pibuf) - __vm_deallocate (__mach_task_self (), - (vm_address_t) pi, pisize * sizeof pi[0]); + __munmap (pi, pisize * sizeof pi[0]); return err; } |