4 files changed, 31 insertions, 29 deletions

diff --git a/tools/perf/builtin-annotate.c b/tools/perf/builtin-annotate.c
index 4c7bc4436236..043d85b7e254 100644
--- a/tools/perf/builtin-annotate.c
+++ b/tools/perf/builtin-annotate.c
@@ -472,17 +472,6 @@ static void output__resort(void)
         }
 }
 
-static void register_idle_thread(void)
-{
-        struct thread *thread = threads__findnew(0, &threads, &last_match);
-
-        if (thread == NULL ||
-                        thread__set_comm(thread, "[idle]")) {
-                fprintf(stderr, "problem inserting idle task.\n");
-                exit(-1);
-        }
-}
-
 static unsigned long total = 0,
                      total_mmap = 0,
                      total_comm = 0,
@@ -970,7 +959,7 @@ static int __cmd_annotate(void)
         uint32_t size;
         char *buf;
 
-        register_idle_thread();
+        register_idle_thread(&threads, &last_match);
 
         input = open(input_name, O_RDONLY);
         if (input < 0) {
diff --git a/tools/perf/builtin-report.c b/tools/perf/builtin-report.c
index cdd46ab11bd5..cdf9a8d27bb9 100644
--- a/tools/perf/builtin-report.c
+++ b/tools/perf/builtin-report.c
@@ -666,12 +666,9 @@ static void dso__calc_col_width(struct dso *self)
         self->slen_calculated = 1;
 }
 
-static int thread__set_comm_adjust(struct thread *self, const char *comm)
+static void thread__comm_adjust(struct thread *self)
 {
-        int ret = thread__set_comm(self, comm);
-
-        if (ret)
-                return ret;
+        char *comm = self->comm;
 
         if (!col_width_list_str && !field_sep &&
             (!comm_list || strlist__has_entry(comm_list, comm))) {
@@ -682,6 +679,16 @@ static int thread__set_comm_adjust(struct thread *self, const char *comm)
                         threads__col_width = slen + 6;
                 }
         }
+}
+
+static int thread__set_comm_adjust(struct thread *self, const char *comm)
+{
+        int ret = thread__set_comm(self, comm);
+
+        if (ret)
+                return ret;
+
+        thread__comm_adjust(self);
 
         return 0;
 }
@@ -1073,17 +1080,6 @@ print_entries:
         return ret;
 }
 
-static void register_idle_thread(void)
-{
-        struct thread *thread = threads__findnew(0, &threads, &last_match);
-
-        if (thread == NULL ||
-                        thread__set_comm_adjust(thread, "[idle]")) {
-                fprintf(stderr, "problem inserting idle task.\n");
-                exit(-1);
-        }
-}
-
 static unsigned long total = 0,
                      total_mmap = 0,
                      total_comm = 0,
@@ -1381,11 +1377,13 @@ static int __cmd_report(void)
         unsigned long offset = 0;
         unsigned long head, shift;
         struct stat input_stat;
+        struct thread *idle;
         event_t *event;
         uint32_t size;
         char *buf;
 
-        register_idle_thread();
+        idle = register_idle_thread(&threads, &last_match);
+        thread__comm_adjust(idle);
 
         if (show_threads)
                 perf_read_values_init(&show_threads_values);
diff --git a/tools/perf/util/thread.c b/tools/perf/util/thread.c
index f98032c135c6..3acd37f54888 100644
--- a/tools/perf/util/thread.c
+++ b/tools/perf/util/thread.c
@@ -80,6 +80,19 @@ threads__findnew(pid_t pid, struct rb_root *threads, struct thread **last_match)
         return th;
 }
 
+struct thread *
+register_idle_thread(struct rb_root *threads, struct thread **last_match)
+{
+        struct thread *thread = threads__findnew(0, threads, last_match);
+
+        if (!thread || thread__set_comm(thread, "[idle]")) {
+                fprintf(stderr, "problem inserting idle task.\n");
+                exit(-1);
+        }
+
+        return thread;
+}
+
 void thread__insert_map(struct thread *self, struct map *map)
 {
         struct map *pos, *tmp;
diff --git a/tools/perf/util/thread.h b/tools/perf/util/thread.h
index b1c66719379b..634f2809a342 100644
--- a/tools/perf/util/thread.h
+++ b/tools/perf/util/thread.h
@@ -13,6 +13,8 @@ struct thread {
 int thread__set_comm(struct thread *self, const char *comm);
 struct thread *
 threads__findnew(pid_t pid, struct rb_root *threads, struct thread **last_match);
+struct thread *
+register_idle_thread(struct rb_root *threads, struct thread **last_match);
 void thread__insert_map(struct thread *self, struct map *map);
 int thread__fork(struct thread *self, struct thread *parent);
 struct map *thread__find_map(struct thread *self, u64 ip);