src/sys/bin/psutils/killall.c

// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <fnmatch.h>
#include <inttypes.h>
#include <libgen.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <zircon/syscalls.h>

#include <task-utils/walker.h>

const char* kill_name;
int killed = 0;

zx_status_t process_callback(void* unused_ctx, int depth, zx_handle_t process, zx_koid_t koid,
                             zx_koid_t parent_koid) {
  char name[ZX_MAX_NAME_LEN];
  zx_status_t status = zx_object_get_property(process, ZX_PROP_NAME, name, sizeof(name));
  if (status != ZX_OK) {
    return status;
  }
  if (!strcmp(name, kill_name) || !fnmatch(kill_name, name, 0) ||
      !strcmp(basename(name), kill_name)) {
    zx_task_kill(process);
    printf("Killed %" PRIu64 " %s\n", koid, name);
    killed++;
  }
  return ZX_OK;
}

int main(int argc, char** argv) {
  if (argc != 2) {
    fprintf(stderr, "usage: %s <process>\n", argv[0]);
    fprintf(stderr, "  <process> can be the name of a process, the basename of a process\n");
    fprintf(stderr, "  or glob pattern matching a process name.\n");
    return -1;
  }

  kill_name = argv[1];

  walk_root_job_tree(NULL, process_callback, NULL, NULL);
  if (killed == 0) {
    fprintf(stderr, "no tasks found\n");
    return -1;
  }
}