Annotation of researchv9/cmd/ps/doselect.c, revision 1.1.1.1

1.1       root        1: #include <stdio.h>
                      2: #include "ps.h"
                      3: #include <sys/text.h>
                      4: #include <sys/inode.h>
                      5: #include <sys/file.h>
                      6: 
                      7: #define Error(msg)     return (fprintf(stderr, "%s %s: %s\n", progname, s, msg), \
                      8:                                close(fd), -1)
                      9: #define Reject         return (close(fd), -1)
                     10: #define Accept         return fd
                     11: 
                     12: doselect(s, pp, up)
                     13: char *s; register struct proc *pp; register struct user *up;
                     14: {
                     15:        char sstr[32], *error, *getuarea(); register Select *sp;
                     16:        register fd;
                     17: 
                     18:        up->u_procp = 0;
                     19: 
                     20:        if ((fd = open(strcat(strcpy(sstr, "/proc/"), s), 0)) < 0)
                     21:                return -1;
                     22:        if (!Ioctl(fd, PIOCGETPR, pp))
                     23:                Error("/proc ioctl error");
                     24: 
                     25:        if (!Nflag && !tflag && !Fflag) {
                     26:                if (!aflag && !rflag && pp->p_uid != myuid && pp->p_pid != mypid)
                     27:                        Reject;
                     28:                if (error = getuarea(fd, pp, up))
                     29:                        Error(error);
                     30:                if (!aflag && rflag && up->u_uid != myuid && up->u_ruid != myuid)
                     31:                        Reject;
                     32:                if (!xflag && xreject(pp, up))
                     33:                        Reject;
                     34:                Accept;
                     35:        }
                     36:        for (sp=selbeg; sp<selend; sp++) {
                     37:                if (sp->flag & SELTTY) {
                     38:                        if (error = getuarea(fd, pp, up))
                     39:                                Error(error);
                     40:                        if (strncmp(sp->id, gettty(up->u_ttyino), 5))
                     41:                                continue;
                     42:                        if (!(sp->flag & SELXFL) && xreject(pp, up))
                     43:                                continue;
                     44:                        Accept;
                     45:                } else if (sp->flag & SELFIL) {
                     46:                        if (error = getuarea(fd, pp, up))
                     47:                                Error(error);
                     48:                        if (fdselect(sp->dev, sp->ino, pp, up))
                     49:                                Accept;
                     50:                } else if (atoi(sp->id) == pp->p_pid) {
                     51:                        if (error = getuarea(fd, pp, up))
                     52:                                Error(error);
                     53:                        Accept;
                     54:                }
                     55:        }
                     56:        Reject;
                     57: }
                     58: 
                     59: char *
                     60: getuarea(fd, pp, up)
                     61: register struct proc *pp; register struct user *up;
                     62: {
                     63:        extern errno;
                     64:        struct { char block[ctob(btoc(sizeof(struct user)))];} block, *bp;
                     65: 
                     66:        if (up->u_procp)
                     67:                return 0;
                     68:        if (pp->p_flag & SLOAD) {
                     69:                register ntry = 4;
                     70:                do
                     71:                        if (Sread(fd, UBASE, up))
                     72:                                return 0;
                     73:                while (errno == EBUSY && --ntry > 0);
                     74:                return "/proc read error";
                     75:        }
                     76:        bp = &block;
                     77:        if (Sread(drum, pp->p_swaddr*NBPG/ctod(1), bp)) {
                     78:                *up = *(struct user *)&block;
                     79:                return 0;
                     80:        }
                     81:        return "drum read error";
                     82: }
                     83: 
                     84: xreject(pp, up)
                     85: register struct proc *pp; register struct user *up;
                     86: {
                     87:        return (pp->p_stat != SSTOP &&
                     88:                (strcmp(up->u_comm, "sh") == 0 ||
                     89:                 strcmp(up->u_comm, "csh") == 0 ||
                     90:                 strcmp(up->u_comm, "init") == 0 ||
                     91:                 up->u_comm[0] == 0));
                     92: }
                     93: 
                     94: fdselect(dev, ino, pp, up)
                     95: register struct proc *pp; register struct user *up;
                     96: {
                     97:        register k;
                     98:        struct text x; struct file f; struct inode i;
                     99: 
                    100:        if (Kread(pp->p_textp, &x) && Kread(x.x_iptr, &i) &&
                    101:            dev == i.i_dev && ino == i.i_number)
                    102:                return 1;
                    103:        if (Kread(up->u_cdir, &i) && dev == i.i_dev && ino == i.i_number)
                    104:                return 1;
                    105:        if (Kread(up->u_rdir, &i) && dev == i.i_dev && ino == i.i_number)
                    106:                return 1;
                    107: 
                    108:        for (k = 0; k < NOFILE; k++) {
                    109:                if (!Kread(up->u_ofile[k], &f) || !Kread(f.f_inode, &i))
                    110:                        continue;
                    111:                if (dev == i.i_dev && ino == i.i_number)
                    112:                        return 1;
                    113:        }
                    114:        return 0;
                    115: }

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.