if (sortby == SORTSTART) { i = STARTTIME(((KINFO *)a)) - STARTTIME(((KINFO *)b)); if (i == 0) i = STARTuTIME(((KINFO *)a)) - STARTuTIME(((KINFO *)b)); return (i); }
static char	*kludge_oldps_options __P((char *));static int	 pscomp __P((const void *, const void *));static void	 saveuser __P((KINFO *));static void	 scanvars __P((void));static void	 usage __P((void));char dfmt[] = "pid tt state time command";char jfmt[] = "user pid ppid pgid sess jobc state tt time command";char lfmt[] = "uid pid ppid cpu pri nice vsz rss wchan state tt time command";char   o1[] = "pid";char   o2[] = "tt state time command";char ufmt[] = "user pid %cpu %mem vsz rss tt state start time command";char vfmt[] = "pid state time sl re pagein vsz rss lim tsiz %cpu %mem command";kvm_t *kd;intmain(argc, argv)	int argc;	char *argv[];{	struct kinfo_proc *kp;	struct varent *vent;	struct winsize ws;	struct passwd *pwd;	dev_t ttydev;	pid_t pid;	uid_t uid;	int all, ch, flag, i, fmt, lineno, nentries;	int prtheader, wflag, kflag, what, xflg;	char *nlistf, *memf, *swapf, errbuf[_POSIX2_LINE_MAX];	if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, (char *)&ws) == -1 &&	     ioctl(STDERR_FILENO, TIOCGWINSZ, (char *)&ws) == -1 &&	     ioctl(STDIN_FILENO,  TIOCGWINSZ, (char *)&ws) == -1) ||	     ws.ws_col == 0)		termwidth = 79;	else		termwidth = ws.ws_col - 1;	if (argc > 1)		argv[1] = kludge_oldps_options(argv[1]);	all = fmt = prtheader = wflag = kflag = xflg = 0;	pid = -1;	uid = (uid_t) -1;	ttydev = NODEV;	memf = nlistf = swapf = NULL;	while ((ch = getopt(argc, argv,	    "acCeghjkLlM:mN:O:o:p:rSTt:U:uvW:wx")) != -1)		switch((char)ch) {		case 'a':			all = 1;			break;		case 'c':			commandonly = 1;			break;		case 'e':			/* XXX set ufmt */			needenv = 1;			break;		case 'C':			rawcpu = 1;			break;		case 'g':			break;			/* no-op */		case 'h':			prtheader = ws.ws_row > 5 ? ws.ws_row : 22;			break;		case 'j':			parsefmt(jfmt);			fmt = 1;			jfmt[0] = '\0';			break;		case 'k':			kflag++;			break;		case 'L':			showkey();			exit(0);		case 'l':			parsefmt(lfmt);			fmt = 1;			lfmt[0] = '\0';			break;		case 'M':			memf = optarg;			break;		case 'm':			sortby = SORTMEM;			break;		case 'N':			nlistf = optarg;			break;		case 'O':			parsefmt(o1);			parsefmt(optarg);			parsefmt(o2);			o1[0] = o2[0] = '\0';			fmt = 1;			break;		case 'o':			parsefmt(optarg);			fmt = 1;			break;		case 'p':			pid = atol(optarg);			xflg = 1;			break;		case 'r':			sortby = SORTCPU;			break;		case 'S':			sumrusage = 1;			break;		case 'T':			if ((optarg = ttyname(STDIN_FILENO)) == NULL)				errx(1, "stdin: not a terminal");			/* FALLTHROUGH */		case 't': {			struct stat sb;			char *ttypath, pathbuf[MAXPATHLEN];			if (strcmp(optarg, "co") == 0)				ttypath = _PATH_CONSOLE;			else if (*optarg != '/')				(void)snprintf(ttypath = pathbuf,				    sizeof(pathbuf), "%s%s", _PATH_TTY, optarg);			else				ttypath = optarg;			if (stat(ttypath, &sb) == -1)				err(1, "%s", ttypath);			if (!S_ISCHR(sb.st_mode))				errx(1, "%s: not a terminal", ttypath);			ttydev = sb.st_rdev;			break;		}		case 'U':			pwd = getpwnam(optarg);			if (pwd == NULL)				errx(1, "%s: no such user", optarg);			uid = pwd->pw_uid;			endpwent();			xflg = 1;			break;		case 'u':			parsefmt(ufmt);			sortby = SORTCPU;			fmt = 1;			ufmt[0] = '\0';			break;		case 'v':			parsefmt(vfmt);			sortby = SORTMEM;			fmt = 1;			vfmt[0] = '\0';			break;		case 'W':			swapf = optarg;			break;		case 'w':			if (wflag)				termwidth = UNLIMITED;			else if (termwidth < 131)				termwidth = 131;			wflag++;			break;		case 'x':			xflg = 1;			break;		default:			usage();		}	argc -= optind;	argv += optind;#define	BACKWARD_COMPATIBILITY#ifdef	BACKWARD_COMPATIBILITY	if (*argv) {		nlistf = *argv;		if (*++argv) {			memf = *argv;			if (*++argv)				swapf = *argv;		}	}#endif	/*	 * Discard setgid privileges if not the running kernel so that bad	 * guys can't print interesting stuff from kernel memory.	 */	if (nlistf != NULL || memf != NULL || swapf != NULL) {		setegid(getgid());		setgid(getgid());	}	kd = kvm_openfiles(nlistf, memf, swapf, O_RDONLY, errbuf);	if (kd == 0)		errx(1, "%s", errbuf);	setegid(getgid());	setgid(getgid());	if (!fmt)		parsefmt(dfmt);	/* XXX - should be cleaner */	if (!all && ttydev == NODEV && pid == -1 && uid == (uid_t)-1)		uid = getuid();	/*	 * scan requested variables, noting what structures are needed,	 * and adjusting header widths as appropiate.	 */	scanvars();	/*	 * get proc list	 */	if (uid != (uid_t) -1) {		what = KERN_PROC_UID;		flag = uid;	} else if (ttydev != NODEV) {		what = KERN_PROC_TTY;		flag = ttydev;	} else if (pid != -1) {		what = KERN_PROC_PID;		flag = pid;	} else if (kflag) {		what = KERN_PROC_KTHREAD;		flag = 0;	} else {		what = KERN_PROC_ALL;		flag = 0;	}	/*	 * select procs	 */	if ((kp = kvm_getprocs(kd, what, flag, &nentries)) == 0)		errx(1, "%s", kvm_geterr(kd));	if ((kinfo = malloc(nentries * sizeof(*kinfo))) == NULL)		err(1, NULL);	for (i = nentries; --i >= 0; ++kp) {		kinfo[i].ki_p = kp;		if (needuser)			saveuser(&kinfo[i]);	}	/*	 * print header	 */	printheader();	if (nentries == 0)		exit(1);	/*	 * sort proc list	 */	qsort(kinfo, nentries, sizeof(KINFO), pscomp);	/*	 * for each proc, call each variable output function.	 */	for (i = lineno = 0; i < nentries; i++) {		KINFO *ki = &kinfo[i];		if (xflg == 0 && (KI_EPROC(ki)->e_tdev == NODEV ||		    (KI_PROC(ki)->p_flag & P_CONTROLT ) == 0))			continue;		for (vent = vhead; vent; vent = vent->next) {			(vent->var->oproc)(ki, vent);			if (vent->next != NULL)				(void)putchar(' ');		}		(void)putchar('\n');		if (prtheader && lineno++ == prtheader - 4) {			(void)putchar('\n');			printheader();			lineno = 0;		}	}	exit(eval);}static voidscanvars(){	struct varent *vent;	VAR *v;	int i;	for (vent = vhead; vent; vent = vent->next) {		v = vent->var;		i = strlen(v->header);		if (v->width < i)			v->width = i;		totwidth += v->width + 1;	/* +1 for space */		if (v->flag & USER)			needuser = 1;		if (v->flag & COMM)			needcomm = 1;	}	totwidth--;}static voidsaveuser(ki)	KINFO *ki;{	struct pstats pstats;	struct usave *usp;	usp = &ki->ki_u;	if (kvm_read(kd, (u_long)&KI_PROC(ki)->p_addr->u_stats,	    &pstats, sizeof(pstats)) == sizeof(pstats)) {		/*		 * The u-area might be swapped out, and we can't get		 * at it because we have a crashdump and no swap.		 * If it's here fill in these fields, otherwise, just		 * leave them 0.		 */		usp->u_start = pstats.p_start;		usp->u_ru = pstats.p_ru;		usp->u_cru = pstats.p_cru;		usp->u_valid = 1;	} else		usp->u_valid = 0;}static intpscomp(a, b)	const void *a, *b;{	int i;#define VSIZE(k) (KI_EPROC(k)->e_vm.vm_dsize + KI_EPROC(k)->e_vm.vm_ssize + \		  KI_EPROC(k)->e_vm.vm_tsize)	if (sortby == SORTCPU)		return (getpcpu((KINFO *)b) - getpcpu((KINFO *)a));	if (sortby == SORTMEM)		return (VSIZE((KINFO *)b) - VSIZE((KINFO *)a));	i =  KI_EPROC((KINFO *)a)->e_tdev - KI_EPROC((KINFO *)b)->e_tdev;	if (i == 0)		i = KI_PROC((KINFO *)a)->p_pid - KI_PROC((KINFO *)b)->p_pid;	return (i);}/* * ICK (all for getopt), would rather hide the ugliness * here than taint the main code. * *  ps foo -> ps -foo *  ps 34 -> ps -p34 * * The old convention that 't' with no trailing tty arg means the users * tty, is only supported if argv[1] doesn't begin with a '-'.  This same * feature is available with the option 'T', which takes no argument. */static char *kludge_oldps_options(s)	char *s;{	size_t len;	char *newopts, *ns, *cp;	len = strlen(s);	if ((newopts = ns = malloc(len + 3)) == NULL)		err(1, NULL);	/*	 * options begin with '-'	 */	if (*s != '-')		*ns++ = '-';	/* add option flag */	/*	 * gaze to end of argv[1]	 */	cp = s + len - 1;	/*	 * if last letter is a 't' flag with no argument (in the context	 * of the oldps options -- option string NOT starting with a '-' --	 * then convert to 'T' (meaning *this* terminal, i.e. ttyname(0)).	 */	if (*cp == 't' && *s != '-')		*cp = 'T';	else {		/*		 * otherwise check for trailing number, which *may* be a		 * pid.		 */		while (cp >= s && isdigit(*cp))			--cp;	}	cp++;	memmove(ns, s, (size_t)(cp - s));	/* copy up to trailing number */	ns += cp - s;	/*	 * if there's a trailing number, and not a preceding 'p' (pid) or	 * 't' (tty) flag, then assume it's a pid and insert a 'p' flag.	 */	if (isdigit(*cp) && (cp == s || (cp[-1] != 't' && cp[-1] != 'p' &&	    (cp - 1 == s || cp[-2] != 't'))))		*ns++ = 'p';	(void)strcpy(ns, cp);		/* and append the number */	return (newopts);}