1 files changed, 93 insertions, 0 deletions
diff --git a/emulator/qtools/profile_pid.cpp b/emulator/qtools/profile_pid.cpp
new file mode 100644
index 0000000..aa37847
--- /dev/null
+++ b/emulator/qtools/profile_pid.cpp
@@ -0,0 +1,93 @@
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <inttypes.h>
+#include <string.h>
+#include "trace_reader.h"
+#include "parse_options.h"
+
+typedef TraceReader<> TraceReaderType;
+
+#include "parse_options-inl.h"
+
+// This function is called from quicksort to compare the cpu time
+// of processes and sort into decreasing order.
+int cmp_dec_cpu_time(const void *a, const void *b) {
+  ProcessState *proc1, *proc2;
+
+  proc1 = (ProcessState*)a;
+  proc2 = (ProcessState*)b;
+  if (proc1 == NULL) {
+    if (proc2 == NULL)
+      return 0;
+    return 1;
+  }
+  if (proc2 == NULL)
+    return -1;
+  if (proc1->cpu_time < proc2->cpu_time)
+    return 1;
+  if (proc1->cpu_time > proc2->cpu_time)
+    return -1;
+  // If the cpu_time times are the same, then sort into increasing
+  // order of pid.
+  return proc1->pid - proc2->pid;
+}
+
+void Usage(const char *program)
+{
+  fprintf(stderr, "Usage: %s [options] trace_file\n", program);
+  OptionsUsage();
+}
+
+int main(int argc, char **argv) {
+  // Parse the options
+  ParseOptions(argc, argv);
+  if (argc - optind != 1) {
+    Usage(argv[0]);
+    exit(1);
+  }
+
+  char *trace_filename = argv[optind];
+  TraceReader<> *trace = new TraceReader<>;
+  trace->Open(trace_filename);
+  trace->SetRoot(root);
+
+  while (1) {
+    BBEvent event, ignored;
+    symbol_type *dummy_sym;
+
+    if (GetNextValidEvent(trace, &event, &ignored, &dummy_sym))
+      break;
+  }
+
+  int num_procs;
+  ProcessState *processes = trace->GetProcesses(&num_procs);
+  qsort(processes, num_procs, sizeof(ProcessState), cmp_dec_cpu_time);
+
+  uint64_t total_time = 0;
+  for (int ii = 0; ii < num_procs; ++ii) {
+    total_time += processes[ii].cpu_time;
+  }
+
+  uint64_t sum_time = 0;
+  printf("  pid parent   cpu_time      %%      %% flags argv\n"); 
+  ProcessState *pstate = &processes[0];
+  for (int ii = 0; ii < num_procs; ++ii, ++pstate) {
+    sum_time += pstate->cpu_time;
+    double per = 100.0 * pstate->cpu_time / total_time;
+    double sum_per = 100.0 * sum_time / total_time;
+    char *print_flags = "";
+    if ((pstate->flags & ProcessState::kCalledExec) == 0)
+      print_flags = "T";
+    if (pstate->name == NULL)
+      pstate->name = "";
+    printf("%5d  %5d %10llu %6.2f %6.2f %5s %s",
+           pstate->pid, pstate->parent_pid, pstate->cpu_time,
+           per, sum_per, print_flags, pstate->name);
+    for (int ii = 1; ii < pstate->argc; ++ii) {
+      printf(" %s", pstate->argv[ii]);
+    }
+    printf("\n");
+  }
+  return 0;
+}