123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137 |
- /* $OpenBSD$ */
- /*
- * Copyright (c) 2009 Nicholas Marriott <nicholas.marriott@gmail.com>
- *
- * Permission to use, copy, modify, and distribute this software for any
- * purpose with or without fee is hereby granted, provided that the above
- * copyright notice and this permission notice appear in all copies.
- *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
- * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
- * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
- */
- #include <sys/param.h>
- #include <sys/proc.h>
- #include <sys/stat.h>
- #include <sys/sysctl.h>
- #include <errno.h>
- #include <event.h>
- #include <stdlib.h>
- #include <string.h>
- #include <unistd.h>
- #define is_runnable(p) \
- ((p)->p_stat == LSRUN || (p)->p_stat == SIDL)
- #define is_stopped(p) \
- ((p)->p_stat == SSTOP || (p)->p_stat == SZOMB)
- struct kinfo_proc2 *cmp_procs(struct kinfo_proc2 *, struct kinfo_proc2 *);
- char *osdep_get_name(int, char *);
- char *osdep_get_cwd(int);
- struct event_base *osdep_event_init(void);
- struct kinfo_proc2 *
- cmp_procs(struct kinfo_proc2 *p1, struct kinfo_proc2 *p2)
- {
- if (is_runnable(p1) && !is_runnable(p2))
- return (p1);
- if (!is_runnable(p1) && is_runnable(p2))
- return (p2);
- if (is_stopped(p1) && !is_stopped(p2))
- return (p1);
- if (!is_stopped(p1) && is_stopped(p2))
- return (p2);
- if (p1->p_estcpu > p2->p_estcpu)
- return (p1);
- if (p1->p_estcpu < p2->p_estcpu)
- return (p2);
- if (p1->p_slptime < p2->p_slptime)
- return (p1);
- if (p1->p_slptime > p2->p_slptime)
- return (p2);
- if (p1->p_pid > p2->p_pid)
- return (p1);
- return (p2);
- }
- char *
- osdep_get_name(int fd, __unused char *tty)
- {
- int mib[6];
- struct stat sb;
- size_t len, i;
- struct kinfo_proc2 *buf, *newbuf, *bestp;
- char *name;
- if (stat(tty, &sb) == -1)
- return (NULL);
- if ((mib[3] = tcgetpgrp(fd)) == -1)
- return (NULL);
- buf = NULL;
- len = sizeof(bestp);
- mib[0] = CTL_KERN;
- mib[1] = KERN_PROC2;
- mib[2] = KERN_PROC_PGRP;
- mib[4] = sizeof (*buf);
- mib[5] = 0;
- retry:
- if (sysctl(mib, __arraycount(mib), NULL, &len, NULL, 0) == -1)
- return (NULL);
- if ((newbuf = realloc(buf, len * sizeof (*buf))) == NULL)
- goto error;
- buf = newbuf;
- mib[5] = len / sizeof(*buf);
- if (sysctl(mib, __arraycount(mib), buf, &len, NULL, 0) == -1) {
- if (errno == ENOMEM)
- goto retry; /* possible infinite loop? */
- goto error;
- }
- bestp = NULL;
- for (i = 0; i < len / sizeof (*buf); i++) {
- if (buf[i].p_tdev != sb.st_rdev)
- continue;
- if (bestp == NULL)
- bestp = &buf[i];
- else
- bestp = cmp_procs(&buf[i], bestp);
- }
- name = NULL;
- if (bestp != NULL)
- name = strdup(bestp->p_comm);
- free(buf);
- return (name);
- error:
- free(buf);
- return (NULL);
- }
- char *
- osdep_get_cwd(int fd)
- {
- return (NULL);
- }
- struct event_base *
- osdep_event_init(void)
- {
- return (event_init());
- }
|