(_hurd_priority_which_map): New arg PI_FLAGS, passed through to
[kopensolaris-gnu/glibc.git] / hurd / hurdprio.c
1 /* Support code for dealing with priorities in the Hurd.
2 Copyright (C) 1994, 1995 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Library General Public License as
7 published by the Free Software Foundation; either version 2 of the
8 License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 Library General Public License for more details.
14
15 You should have received a copy of the GNU Library General Public
16 License along with the GNU C Library; see the file COPYING.LIB.  If
17 not, write to the Free Software Foundation, Inc., 675 Mass Ave,
18 Cambridge, MA 02139, USA.  */
19
20 #include <hurd/resource.h>
21 #include <hurd.h>
22
23 error_t
24 _hurd_priority_which_map (enum __priority_which which, int who,
25                           error_t (*function) (pid_t, struct procinfo *),
26                           int pi_flags)
27 {
28   mach_msg_type_number_t npids = 64, i;
29   pid_t pidbuf[npids], *pids;
30   error_t err;
31   struct procinfo *pip;
32   int pibuf[sizeof *pip + 5 * sizeof (pip->threadinfos[0])], *pi = pibuf;
33   mach_msg_type_number_t pisize = sizeof (pibuf) / sizeof (int);
34
35   switch (which)
36     {
37     case PRIO_PROCESS:
38       npids = 1;
39       pids[0] = who;
40       err = 0;
41       break;
42
43     case PRIO_PGRP:
44       err = __USEPORT (PROC, __proc_getpgrppids (port, who, &pids, &npids));
45       break;
46
47     case PRIO_USER:
48       err = __USEPORT (PROC, __proc_getallpids (port, &pids, &npids));
49       break;
50
51     default:
52       return EINVAL;
53     }
54
55   for (i = 0; !err && i < npids; ++i)
56     {
57       if (which == PRIO_USER)
58         {
59           /* Get procinfo to check the owner.  */
60           int *oldpi = pi;
61           mach_msg_type_number_t oldpisize = pisize;
62           if (err = __USEPORT (PROC, __proc_getprocinfo (port, pids[i],
63                                                          pi_flags,
64                                                          &pi, &pisize)))
65             continue;
66           if (pi != oldpi && oldpi != pibuf)
67             /* Old buffer from last call was not reused; free it.  */
68             __vm_deallocate (__mach_task_self (),
69                              (vm_address_t) oldpi, oldpisize * sizeof pi[0]);
70
71           pip = (struct procinfo *) pi;
72           if (pip->owner != (uid_t) who)
73             continue;
74         }
75       else
76         pip = NULL;
77
78       err = (*function) (pids[i], pip);
79     }
80
81   if (pids != pidbuf)
82     __vm_deallocate (__mach_task_self (),
83                      (vm_address_t) pids, npids * sizeof pids[0]);
84   if (pi != pibuf)
85     __vm_deallocate (__mach_task_self (),
86                      (vm_address_t) pi, pisize * sizeof pi[0]);
87
88   return err;
89 }