1*52576792Salex /* $OpenBSD: ps.c,v 1.11 1999/04/28 20:55:14 alex Exp $ */ 2df930be7Sderaadt /* $NetBSD: ps.c,v 1.15 1995/05/18 20:33:25 mycroft Exp $ */ 3df930be7Sderaadt 4df930be7Sderaadt /*- 5df930be7Sderaadt * Copyright (c) 1990, 1993, 1994 6df930be7Sderaadt * The Regents of the University of California. All rights reserved. 7df930be7Sderaadt * 8df930be7Sderaadt * Redistribution and use in source and binary forms, with or without 9df930be7Sderaadt * modification, are permitted provided that the following conditions 10df930be7Sderaadt * are met: 11df930be7Sderaadt * 1. Redistributions of source code must retain the above copyright 12df930be7Sderaadt * notice, this list of conditions and the following disclaimer. 13df930be7Sderaadt * 2. Redistributions in binary form must reproduce the above copyright 14df930be7Sderaadt * notice, this list of conditions and the following disclaimer in the 15df930be7Sderaadt * documentation and/or other materials provided with the distribution. 16df930be7Sderaadt * 3. All advertising materials mentioning features or use of this software 17df930be7Sderaadt * must display the following acknowledgement: 18df930be7Sderaadt * This product includes software developed by the University of 19df930be7Sderaadt * California, Berkeley and its contributors. 20df930be7Sderaadt * 4. Neither the name of the University nor the names of its contributors 21df930be7Sderaadt * may be used to endorse or promote products derived from this software 22df930be7Sderaadt * without specific prior written permission. 23df930be7Sderaadt * 24df930be7Sderaadt * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 25df930be7Sderaadt * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 26df930be7Sderaadt * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 27df930be7Sderaadt * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 28df930be7Sderaadt * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 29df930be7Sderaadt * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 30df930be7Sderaadt * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 31df930be7Sderaadt * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 32df930be7Sderaadt * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 33df930be7Sderaadt * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 34df930be7Sderaadt * SUCH DAMAGE. 35df930be7Sderaadt */ 36df930be7Sderaadt 37df930be7Sderaadt #ifndef lint 38df930be7Sderaadt static char copyright[] = 39df930be7Sderaadt "@(#) Copyright (c) 1990, 1993, 1994\n\ 40df930be7Sderaadt The Regents of the University of California. All rights reserved.\n"; 41df930be7Sderaadt #endif /* not lint */ 42df930be7Sderaadt 43df930be7Sderaadt #ifndef lint 44df930be7Sderaadt #if 0 45df930be7Sderaadt static char sccsid[] = "@(#)ps.c 8.4 (Berkeley) 4/2/94"; 46df930be7Sderaadt #else 47*52576792Salex static char rcsid[] = "$OpenBSD: ps.c,v 1.11 1999/04/28 20:55:14 alex Exp $"; 48df930be7Sderaadt #endif 49df930be7Sderaadt #endif /* not lint */ 50df930be7Sderaadt 51df930be7Sderaadt #include <sys/param.h> 52df930be7Sderaadt #include <sys/user.h> 53df930be7Sderaadt #include <sys/time.h> 54df930be7Sderaadt #include <sys/resource.h> 55df930be7Sderaadt #include <sys/proc.h> 56df930be7Sderaadt #include <sys/stat.h> 57df930be7Sderaadt #include <sys/ioctl.h> 58df930be7Sderaadt #include <sys/sysctl.h> 59*52576792Salex #include <sys/types.h> 60df930be7Sderaadt 61df930be7Sderaadt #include <ctype.h> 62df930be7Sderaadt #include <err.h> 63df930be7Sderaadt #include <errno.h> 64df930be7Sderaadt #include <fcntl.h> 65df930be7Sderaadt #include <kvm.h> 66df930be7Sderaadt #include <nlist.h> 67df930be7Sderaadt #include <paths.h> 68*52576792Salex #include <pwd.h> 69df930be7Sderaadt #include <stdio.h> 70df930be7Sderaadt #include <stdlib.h> 71df930be7Sderaadt #include <string.h> 72df930be7Sderaadt #include <unistd.h> 736763c4d8Sderaadt #include <limits.h> 74df930be7Sderaadt 75df930be7Sderaadt #include "ps.h" 76df930be7Sderaadt 77df930be7Sderaadt KINFO *kinfo; 78df930be7Sderaadt struct varent *vhead, *vtail; 79df930be7Sderaadt 80df930be7Sderaadt int eval; /* exit value */ 81df930be7Sderaadt int rawcpu; /* -C */ 82df930be7Sderaadt int sumrusage; /* -S */ 83df930be7Sderaadt int termwidth; /* width of screen (0 == infinity) */ 84df930be7Sderaadt int totwidth; /* calculated width of requested variables */ 85df930be7Sderaadt 86df930be7Sderaadt int needuser, needcomm, needenv, commandonly; 87df930be7Sderaadt 88df930be7Sderaadt enum sort { DEFAULT, SORTMEM, SORTCPU } sortby = DEFAULT; 89df930be7Sderaadt 90df930be7Sderaadt static char *kludge_oldps_options __P((char *)); 91df930be7Sderaadt static int pscomp __P((const void *, const void *)); 92df930be7Sderaadt static void saveuser __P((KINFO *)); 93df930be7Sderaadt static void scanvars __P((void)); 94df930be7Sderaadt static void usage __P((void)); 95df930be7Sderaadt 96df930be7Sderaadt char dfmt[] = "pid tt state time command"; 97df930be7Sderaadt char jfmt[] = "user pid ppid pgid sess jobc state tt time command"; 98df930be7Sderaadt char lfmt[] = "uid pid ppid cpu pri nice vsz rss wchan state tt time command"; 99df930be7Sderaadt char o1[] = "pid"; 100df930be7Sderaadt char o2[] = "tt state time command"; 101df930be7Sderaadt char ufmt[] = "user pid %cpu %mem vsz rss tt state start time command"; 102df930be7Sderaadt char vfmt[] = "pid state time sl re pagein vsz rss lim tsiz %cpu %mem command"; 103df930be7Sderaadt 104df930be7Sderaadt kvm_t *kd; 105df930be7Sderaadt 106df930be7Sderaadt int 107df930be7Sderaadt main(argc, argv) 108df930be7Sderaadt int argc; 109df930be7Sderaadt char *argv[]; 110df930be7Sderaadt { 111df930be7Sderaadt struct kinfo_proc *kp; 112df930be7Sderaadt struct varent *vent; 113df930be7Sderaadt struct winsize ws; 114*52576792Salex struct passwd *pwd; 115df930be7Sderaadt dev_t ttydev; 116df930be7Sderaadt pid_t pid; 117df930be7Sderaadt uid_t uid; 118df930be7Sderaadt int all, ch, flag, i, fmt, lineno, nentries; 119df930be7Sderaadt int prtheader, wflag, what, xflg; 1206763c4d8Sderaadt char *nlistf, *memf, *swapf, errbuf[_POSIX2_LINE_MAX]; 121df930be7Sderaadt 122df930be7Sderaadt if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, (char *)&ws) == -1 && 123df930be7Sderaadt ioctl(STDERR_FILENO, TIOCGWINSZ, (char *)&ws) == -1 && 124df930be7Sderaadt ioctl(STDIN_FILENO, TIOCGWINSZ, (char *)&ws) == -1) || 125df930be7Sderaadt ws.ws_col == 0) 126df930be7Sderaadt termwidth = 79; 127df930be7Sderaadt else 128df930be7Sderaadt termwidth = ws.ws_col - 1; 129df930be7Sderaadt 130df930be7Sderaadt if (argc > 1) 131df930be7Sderaadt argv[1] = kludge_oldps_options(argv[1]); 132df930be7Sderaadt 133df930be7Sderaadt all = fmt = prtheader = wflag = xflg = 0; 134df930be7Sderaadt pid = -1; 135df930be7Sderaadt uid = (uid_t) -1; 136df930be7Sderaadt ttydev = NODEV; 137df930be7Sderaadt memf = nlistf = swapf = NULL; 138df930be7Sderaadt while ((ch = getopt(argc, argv, 139*52576792Salex "acCeghjLlM:mN:O:o:p:rSTt:U:uvW:wx")) != EOF) 140df930be7Sderaadt switch((char)ch) { 141df930be7Sderaadt case 'a': 142df930be7Sderaadt all = 1; 143df930be7Sderaadt break; 144df930be7Sderaadt case 'c': 145df930be7Sderaadt commandonly = 1; 146df930be7Sderaadt break; 147df930be7Sderaadt case 'e': /* XXX set ufmt */ 148df930be7Sderaadt needenv = 1; 149df930be7Sderaadt break; 150df930be7Sderaadt case 'C': 151df930be7Sderaadt rawcpu = 1; 152df930be7Sderaadt break; 153df930be7Sderaadt case 'g': 154df930be7Sderaadt break; /* no-op */ 155df930be7Sderaadt case 'h': 156df930be7Sderaadt prtheader = ws.ws_row > 5 ? ws.ws_row : 22; 157df930be7Sderaadt break; 158df930be7Sderaadt case 'j': 159df930be7Sderaadt parsefmt(jfmt); 160df930be7Sderaadt fmt = 1; 161df930be7Sderaadt jfmt[0] = '\0'; 162df930be7Sderaadt break; 163df930be7Sderaadt case 'L': 164df930be7Sderaadt showkey(); 165df930be7Sderaadt exit(0); 166df930be7Sderaadt case 'l': 167df930be7Sderaadt parsefmt(lfmt); 168df930be7Sderaadt fmt = 1; 169df930be7Sderaadt lfmt[0] = '\0'; 170df930be7Sderaadt break; 171df930be7Sderaadt case 'M': 172df930be7Sderaadt memf = optarg; 173df930be7Sderaadt break; 174df930be7Sderaadt case 'm': 175df930be7Sderaadt sortby = SORTMEM; 176df930be7Sderaadt break; 177df930be7Sderaadt case 'N': 178df930be7Sderaadt nlistf = optarg; 179df930be7Sderaadt break; 180df930be7Sderaadt case 'O': 181df930be7Sderaadt parsefmt(o1); 182df930be7Sderaadt parsefmt(optarg); 183df930be7Sderaadt parsefmt(o2); 184df930be7Sderaadt o1[0] = o2[0] = '\0'; 185df930be7Sderaadt fmt = 1; 186df930be7Sderaadt break; 187df930be7Sderaadt case 'o': 188df930be7Sderaadt parsefmt(optarg); 189df930be7Sderaadt fmt = 1; 190df930be7Sderaadt break; 191df930be7Sderaadt case 'p': 192df930be7Sderaadt pid = atol(optarg); 193df930be7Sderaadt xflg = 1; 194df930be7Sderaadt break; 195df930be7Sderaadt case 'r': 196df930be7Sderaadt sortby = SORTCPU; 197df930be7Sderaadt break; 198df930be7Sderaadt case 'S': 199df930be7Sderaadt sumrusage = 1; 200df930be7Sderaadt break; 201df930be7Sderaadt case 'T': 202df930be7Sderaadt if ((optarg = ttyname(STDIN_FILENO)) == NULL) 203df930be7Sderaadt errx(1, "stdin: not a terminal"); 204df930be7Sderaadt /* FALLTHROUGH */ 205df930be7Sderaadt case 't': { 206df930be7Sderaadt struct stat sb; 207df930be7Sderaadt char *ttypath, pathbuf[MAXPATHLEN]; 208df930be7Sderaadt 209df930be7Sderaadt if (strcmp(optarg, "co") == 0) 210df930be7Sderaadt ttypath = _PATH_CONSOLE; 211df930be7Sderaadt else if (*optarg != '/') 212df930be7Sderaadt (void)snprintf(ttypath = pathbuf, 213df930be7Sderaadt sizeof(pathbuf), "%s%s", _PATH_TTY, optarg); 214df930be7Sderaadt else 215df930be7Sderaadt ttypath = optarg; 216df930be7Sderaadt if (stat(ttypath, &sb) == -1) 217df930be7Sderaadt err(1, "%s", ttypath); 218df930be7Sderaadt if (!S_ISCHR(sb.st_mode)) 219df930be7Sderaadt errx(1, "%s: not a terminal", ttypath); 220df930be7Sderaadt ttydev = sb.st_rdev; 221df930be7Sderaadt break; 222df930be7Sderaadt } 223*52576792Salex case 'U': 224*52576792Salex pwd = getpwnam(optarg); 225*52576792Salex if (pwd == NULL) 226*52576792Salex errx(1, "%s: no such user", optarg); 227*52576792Salex uid = pwd->pw_uid; 228*52576792Salex endpwent(); 229*52576792Salex xflg = 1; 230*52576792Salex break; 231df930be7Sderaadt case 'u': 232df930be7Sderaadt parsefmt(ufmt); 233df930be7Sderaadt sortby = SORTCPU; 234df930be7Sderaadt fmt = 1; 235df930be7Sderaadt ufmt[0] = '\0'; 236df930be7Sderaadt break; 237df930be7Sderaadt case 'v': 238df930be7Sderaadt parsefmt(vfmt); 239df930be7Sderaadt sortby = SORTMEM; 240df930be7Sderaadt fmt = 1; 241df930be7Sderaadt vfmt[0] = '\0'; 242df930be7Sderaadt break; 243df930be7Sderaadt case 'W': 244df930be7Sderaadt swapf = optarg; 245df930be7Sderaadt break; 246df930be7Sderaadt case 'w': 247df930be7Sderaadt if (wflag) 248df930be7Sderaadt termwidth = UNLIMITED; 249df930be7Sderaadt else if (termwidth < 131) 250df930be7Sderaadt termwidth = 131; 251df930be7Sderaadt wflag++; 252df930be7Sderaadt break; 253df930be7Sderaadt case 'x': 254df930be7Sderaadt xflg = 1; 255df930be7Sderaadt break; 256df930be7Sderaadt default: 257df930be7Sderaadt usage(); 258df930be7Sderaadt } 259df930be7Sderaadt argc -= optind; 260df930be7Sderaadt argv += optind; 261df930be7Sderaadt 262df930be7Sderaadt #define BACKWARD_COMPATIBILITY 263df930be7Sderaadt #ifdef BACKWARD_COMPATIBILITY 264df930be7Sderaadt if (*argv) { 265df930be7Sderaadt nlistf = *argv; 266df930be7Sderaadt if (*++argv) { 267df930be7Sderaadt memf = *argv; 268df930be7Sderaadt if (*++argv) 269df930be7Sderaadt swapf = *argv; 270df930be7Sderaadt } 271df930be7Sderaadt } 272df930be7Sderaadt #endif 273df930be7Sderaadt /* 274df930be7Sderaadt * Discard setgid privileges if not the running kernel so that bad 275df930be7Sderaadt * guys can't print interesting stuff from kernel memory. 276df930be7Sderaadt */ 2774bd36bfbStholo if (nlistf != NULL || memf != NULL || swapf != NULL) { 2784bd36bfbStholo setegid(getgid()); 279df930be7Sderaadt setgid(getgid()); 2804bd36bfbStholo } 281df930be7Sderaadt 282df930be7Sderaadt kd = kvm_openfiles(nlistf, memf, swapf, O_RDONLY, errbuf); 283df930be7Sderaadt if (kd == 0) 284df930be7Sderaadt errx(1, "%s", errbuf); 285df930be7Sderaadt 286e46873ebSderaadt setegid(getgid()); 287e46873ebSderaadt setgid(getgid()); 288e46873ebSderaadt 289df930be7Sderaadt if (!fmt) 290df930be7Sderaadt parsefmt(dfmt); 291df930be7Sderaadt 292*52576792Salex /* XXX - should be cleaner */ 293*52576792Salex if (!all && ttydev == NODEV && pid == -1 && uid == (uid_t)-1) 294df930be7Sderaadt uid = getuid(); 295df930be7Sderaadt 296df930be7Sderaadt /* 297df930be7Sderaadt * scan requested variables, noting what structures are needed, 298df930be7Sderaadt * and adjusting header widths as appropiate. 299df930be7Sderaadt */ 300df930be7Sderaadt scanvars(); 301df930be7Sderaadt /* 302df930be7Sderaadt * get proc list 303df930be7Sderaadt */ 304df930be7Sderaadt if (uid != (uid_t) -1) { 305df930be7Sderaadt what = KERN_PROC_UID; 306df930be7Sderaadt flag = uid; 307df930be7Sderaadt } else if (ttydev != NODEV) { 308df930be7Sderaadt what = KERN_PROC_TTY; 309df930be7Sderaadt flag = ttydev; 310df930be7Sderaadt } else if (pid != -1) { 311df930be7Sderaadt what = KERN_PROC_PID; 312df930be7Sderaadt flag = pid; 313df930be7Sderaadt } else { 314df930be7Sderaadt what = KERN_PROC_ALL; 315df930be7Sderaadt flag = 0; 316df930be7Sderaadt } 317df930be7Sderaadt /* 318df930be7Sderaadt * select procs 319df930be7Sderaadt */ 320df930be7Sderaadt if ((kp = kvm_getprocs(kd, what, flag, &nentries)) == 0) 321df930be7Sderaadt errx(1, "%s", kvm_geterr(kd)); 322df930be7Sderaadt if ((kinfo = malloc(nentries * sizeof(*kinfo))) == NULL) 323df930be7Sderaadt err(1, NULL); 324df930be7Sderaadt for (i = nentries; --i >= 0; ++kp) { 325df930be7Sderaadt kinfo[i].ki_p = kp; 326df930be7Sderaadt if (needuser) 327df930be7Sderaadt saveuser(&kinfo[i]); 328df930be7Sderaadt } 329df930be7Sderaadt /* 330df930be7Sderaadt * print header 331df930be7Sderaadt */ 332df930be7Sderaadt printheader(); 333df930be7Sderaadt if (nentries == 0) 334a30d4cafSmichaels exit(1); 335df930be7Sderaadt /* 336df930be7Sderaadt * sort proc list 337df930be7Sderaadt */ 338df930be7Sderaadt qsort(kinfo, nentries, sizeof(KINFO), pscomp); 339df930be7Sderaadt /* 340df930be7Sderaadt * for each proc, call each variable output function. 341df930be7Sderaadt */ 342df930be7Sderaadt for (i = lineno = 0; i < nentries; i++) { 343df930be7Sderaadt KINFO *ki = &kinfo[i]; 344df930be7Sderaadt 345df930be7Sderaadt if (xflg == 0 && (KI_EPROC(ki)->e_tdev == NODEV || 346df930be7Sderaadt (KI_PROC(ki)->p_flag & P_CONTROLT ) == 0)) 347df930be7Sderaadt continue; 348df930be7Sderaadt for (vent = vhead; vent; vent = vent->next) { 349df930be7Sderaadt (vent->var->oproc)(ki, vent); 350df930be7Sderaadt if (vent->next != NULL) 351df930be7Sderaadt (void)putchar(' '); 352df930be7Sderaadt } 353df930be7Sderaadt (void)putchar('\n'); 354df930be7Sderaadt if (prtheader && lineno++ == prtheader - 4) { 355df930be7Sderaadt (void)putchar('\n'); 356df930be7Sderaadt printheader(); 357df930be7Sderaadt lineno = 0; 358df930be7Sderaadt } 359df930be7Sderaadt } 360df930be7Sderaadt exit(eval); 361df930be7Sderaadt } 362df930be7Sderaadt 363df930be7Sderaadt static void 364df930be7Sderaadt scanvars() 365df930be7Sderaadt { 366df930be7Sderaadt struct varent *vent; 367df930be7Sderaadt VAR *v; 368df930be7Sderaadt int i; 369df930be7Sderaadt 370df930be7Sderaadt for (vent = vhead; vent; vent = vent->next) { 371df930be7Sderaadt v = vent->var; 372df930be7Sderaadt i = strlen(v->header); 373df930be7Sderaadt if (v->width < i) 374df930be7Sderaadt v->width = i; 375df930be7Sderaadt totwidth += v->width + 1; /* +1 for space */ 376df930be7Sderaadt if (v->flag & USER) 377df930be7Sderaadt needuser = 1; 378df930be7Sderaadt if (v->flag & COMM) 379df930be7Sderaadt needcomm = 1; 380df930be7Sderaadt } 381df930be7Sderaadt totwidth--; 382df930be7Sderaadt } 383df930be7Sderaadt 384df930be7Sderaadt static void 385df930be7Sderaadt saveuser(ki) 386df930be7Sderaadt KINFO *ki; 387df930be7Sderaadt { 388df930be7Sderaadt struct pstats pstats; 389df930be7Sderaadt struct usave *usp; 390df930be7Sderaadt 391df930be7Sderaadt usp = &ki->ki_u; 392df930be7Sderaadt if (kvm_read(kd, (u_long)&KI_PROC(ki)->p_addr->u_stats, 393df930be7Sderaadt (char *)&pstats, sizeof(pstats)) == sizeof(pstats)) { 394df930be7Sderaadt /* 395df930be7Sderaadt * The u-area might be swapped out, and we can't get 396df930be7Sderaadt * at it because we have a crashdump and no swap. 397df930be7Sderaadt * If it's here fill in these fields, otherwise, just 398df930be7Sderaadt * leave them 0. 399df930be7Sderaadt */ 400df930be7Sderaadt usp->u_start = pstats.p_start; 401df930be7Sderaadt usp->u_ru = pstats.p_ru; 402df930be7Sderaadt usp->u_cru = pstats.p_cru; 403df930be7Sderaadt usp->u_valid = 1; 404df930be7Sderaadt } else 405df930be7Sderaadt usp->u_valid = 0; 406df930be7Sderaadt } 407df930be7Sderaadt 408df930be7Sderaadt static int 409df930be7Sderaadt pscomp(a, b) 410df930be7Sderaadt const void *a, *b; 411df930be7Sderaadt { 412df930be7Sderaadt int i; 413df930be7Sderaadt #define VSIZE(k) (KI_EPROC(k)->e_vm.vm_dsize + KI_EPROC(k)->e_vm.vm_ssize + \ 414df930be7Sderaadt KI_EPROC(k)->e_vm.vm_tsize) 415df930be7Sderaadt 416df930be7Sderaadt if (sortby == SORTCPU) 417df930be7Sderaadt return (getpcpu((KINFO *)b) - getpcpu((KINFO *)a)); 418df930be7Sderaadt if (sortby == SORTMEM) 419df930be7Sderaadt return (VSIZE((KINFO *)b) - VSIZE((KINFO *)a)); 420df930be7Sderaadt i = KI_EPROC((KINFO *)a)->e_tdev - KI_EPROC((KINFO *)b)->e_tdev; 421df930be7Sderaadt if (i == 0) 422df930be7Sderaadt i = KI_PROC((KINFO *)a)->p_pid - KI_PROC((KINFO *)b)->p_pid; 423df930be7Sderaadt return (i); 424df930be7Sderaadt } 425df930be7Sderaadt 426df930be7Sderaadt /* 427df930be7Sderaadt * ICK (all for getopt), would rather hide the ugliness 428df930be7Sderaadt * here than taint the main code. 429df930be7Sderaadt * 430df930be7Sderaadt * ps foo -> ps -foo 431df930be7Sderaadt * ps 34 -> ps -p34 432df930be7Sderaadt * 433df930be7Sderaadt * The old convention that 't' with no trailing tty arg means the users 434df930be7Sderaadt * tty, is only supported if argv[1] doesn't begin with a '-'. This same 435df930be7Sderaadt * feature is available with the option 'T', which takes no argument. 436df930be7Sderaadt */ 437df930be7Sderaadt static char * 438df930be7Sderaadt kludge_oldps_options(s) 439df930be7Sderaadt char *s; 440df930be7Sderaadt { 441df930be7Sderaadt size_t len; 442df930be7Sderaadt char *newopts, *ns, *cp; 443df930be7Sderaadt 444df930be7Sderaadt len = strlen(s); 4454c163512Sderaadt if ((newopts = ns = malloc(len + 3)) == NULL) 446df930be7Sderaadt err(1, NULL); 447df930be7Sderaadt /* 448df930be7Sderaadt * options begin with '-' 449df930be7Sderaadt */ 450df930be7Sderaadt if (*s != '-') 451df930be7Sderaadt *ns++ = '-'; /* add option flag */ 452df930be7Sderaadt /* 453df930be7Sderaadt * gaze to end of argv[1] 454df930be7Sderaadt */ 455df930be7Sderaadt cp = s + len - 1; 456df930be7Sderaadt /* 457df930be7Sderaadt * if last letter is a 't' flag with no argument (in the context 458df930be7Sderaadt * of the oldps options -- option string NOT starting with a '-' -- 459df930be7Sderaadt * then convert to 'T' (meaning *this* terminal, i.e. ttyname(0)). 460df930be7Sderaadt */ 461df930be7Sderaadt if (*cp == 't' && *s != '-') 462df930be7Sderaadt *cp = 'T'; 463df930be7Sderaadt else { 464df930be7Sderaadt /* 465df930be7Sderaadt * otherwise check for trailing number, which *may* be a 466df930be7Sderaadt * pid. 467df930be7Sderaadt */ 468df930be7Sderaadt while (cp >= s && isdigit(*cp)) 469df930be7Sderaadt --cp; 470df930be7Sderaadt } 471df930be7Sderaadt cp++; 472df930be7Sderaadt memmove(ns, s, (size_t)(cp - s)); /* copy up to trailing number */ 473df930be7Sderaadt ns += cp - s; 474df930be7Sderaadt /* 475df930be7Sderaadt * if there's a trailing number, and not a preceding 'p' (pid) or 476df930be7Sderaadt * 't' (tty) flag, then assume it's a pid and insert a 'p' flag. 477df930be7Sderaadt */ 478764064c4Smickey if (isdigit(*cp) && (cp == s || (cp[-1] != 't' && cp[-1] != 'p' && 479764064c4Smickey (cp - 1 == s || cp[-2] != 't')))) 480df930be7Sderaadt *ns++ = 'p'; 481df930be7Sderaadt (void)strcpy(ns, cp); /* and append the number */ 482df930be7Sderaadt 483df930be7Sderaadt return (newopts); 484df930be7Sderaadt } 485df930be7Sderaadt 486df930be7Sderaadt static void 487df930be7Sderaadt usage() 488df930be7Sderaadt { 489df930be7Sderaadt 490df930be7Sderaadt (void)fprintf(stderr, 491df930be7Sderaadt "usage:\t%s\n\t %s\n\t%s\n", 492*52576792Salex "ps [-aChjlmrSTuvwx] [-O|o fmt] [-p pid] [-t tty] [-U user]", 493df930be7Sderaadt "[-M core] [-N system] [-W swap]", 494df930be7Sderaadt "ps [-L]"); 495df930be7Sderaadt exit(1); 496df930be7Sderaadt } 497