Function getProcessTree

src/targets/node/nodeAttacherCluster.ts:76–109

* Returns a process tree rooting at the give process ID.

(
  rootPid: number,
  logger: ILogger,
)

 * Returns a process tree rooting at the give process ID.
 */
async function getProcessTree(
  rootPid: number,
  logger: ILogger,
): Promise<IProcessTreeNode | undefined> {
  const map = new Map<number, IProcessTreeNode>();

  map.set(0, { children: [], pid: 0, ppid: 0, command: '', args: '' });

  try {
    await processTree.lookup(({ pid, ppid, command, args }) => {
      if (pid !== ppid) {
        map.set(pid, { pid, ppid, command, args, children: [] });
      }
      return map;
    }, null);
  } catch (err) {
    logger.warn(LogTag.Internal, 'Error getting child process tree', err);
    return undefined;
  }

  const values = map.values();
  for (const p of values) {
    const parent = map.get(p.ppid);
    if (parent && parent !== p) {
      parent.children.push(p);
    }
  }

  if (!isNaN(rootPid) && rootPid > 0) {
    return map.get(rootPid);
  }

  return map.get(0);
}