Mercurial > hg > toybox

view toys/lsb/pidof.c @ 745:0faab963ea92

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Meddle. The <1 has to come first in the option string, normalize whitespace, sprintf of %d maxes out at -2 billion ala 12 bytes with null terminator so we don't need a length check in a 4k buffer, use the "%*s" feature of printf to prepend whitespace for us, take advantage of c99 defining ! to return 0 or 1.
author Rob Landley <rob@landley.net>
date Sun, 09 Dec 2012 13:57:04 -0600
parents 43e6ec52aa29
children f169d9708518
line wrap: on
line source

/* pidof.c - Print the Process IDs of all processes with the given names.
 *
 * Copyright 2012 Andreas Heck <aheck@gmx.de>
 * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
 *
 * http://refspecs.linuxfoundation.org/LSB_4.1.0/LSB-Core-generic/LSB-Core-generic/pidof.html

USE_PIDOF(NEWTOY(pidof, "<1so:", TOYFLAG_USR|TOYFLAG_BIN))

config PIDOF
  bool "pidof"
  default y
  help
    usage: pidof [-s] [-o omitpid[,omitpid..]] [NAME]...

    Print the PIDs of all processes with the given names.
    -s	single shot, only return one pid.
    -o	omits processes with specified PID
*/

#define FOR_pidof
#include "toys.h"

GLOBALS(
  char *omit;
)

static int print_pid(pid_t pid)
{
  char * res;
  int len;

  sprintf(toybuf, "%d", pid);
  len = strlen(toybuf);

  // Check omit string
  if (toys.optflags & FLAG_o)
  {
    res = strstr(TT.omit, toybuf);
    if (res && (res == TT.omit || res[-1] == ',') &&
      (res[len] == ',' || res[len] == 0)) return 1;
  }
  xprintf("%*s", len+(!toys.exitval), toybuf);
  toys.exitval = 0;

  return !(toys.optflags & FLAG_s);
}

void pidof_main(void)
{
  toys.exitval = 1;
  for_each_pid_with_name_in(toys.optargs, print_pid);
  if (!toys.exitval) xputc('\n');
}