perf tools: Add thread_map__(alloc|realloc) helpers

In order to have 'struct thread_map' allocation on single place and can
change it easily in following patch.

Using alloc|realloc for static helpers, because thread_map__new is
already used in public interface.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
Cc: Adrian Hunter <adrian.hunter@intel.com>
Cc: Andi Kleen <ak@linux.intel.com>
Cc: David Ahern <dsahern@gmail.com>
Cc: Namhyung Kim <namhyung@kernel.org>
Cc: Peter Zijlstra <a.p.zijlstra@chello.nl>
Cc: Stephane Eranian <eranian@google.com>
Link: http://lkml.kernel.org/r/1434269985-521-3-git-send-email-jolsa@kernel.org
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

1 files changed, 16 insertions, 8 deletions

diff --git a/tools/perf/util/thread_map.c b/tools/perf/util/thread_map.c
index f93b9734735b..f4822bd03709 100644
--- a/tools/perf/util/thread_map.c
+++ b/tools/perf/util/thread_map.c
@@ -20,6 +20,15 @@ static int filter(const struct dirent *dir)
                 return 1;
 }
 
+static struct thread_map *thread_map__realloc(struct thread_map *map, int nr)
+{
+        size_t size = sizeof(*map) + sizeof(pid_t) * nr;
+
+        return realloc(map, size);
+}
+
+#define thread_map__alloc(__nr) thread_map__realloc(NULL, __nr)
+
 struct thread_map *thread_map__new_by_pid(pid_t pid)
 {
         struct thread_map *threads;
@@ -33,7 +42,7 @@ struct thread_map *thread_map__new_by_pid(pid_t pid)
         if (items <= 0)
                 return NULL;
 
-        threads = malloc(sizeof(*threads) + sizeof(pid_t) * items);
+        threads = thread_map__alloc(items);
         if (threads != NULL) {
                 for (i = 0; i < items; i++)
                         threads->map[i] = atoi(namelist[i]->d_name);
@@ -49,7 +58,7 @@ struct thread_map *thread_map__new_by_pid(pid_t pid)
 
 struct thread_map *thread_map__new_by_tid(pid_t tid)
 {
-        struct thread_map *threads = malloc(sizeof(*threads) + sizeof(pid_t));
+        struct thread_map *threads = thread_map__alloc(1);
 
         if (threads != NULL) {
                 threads->map[0] = tid;
@@ -65,8 +74,8 @@ struct thread_map *thread_map__new_by_uid(uid_t uid)
         int max_threads = 32, items, i;
         char path[256];
         struct dirent dirent, *next, **namelist = NULL;
-        struct thread_map *threads = malloc(sizeof(*threads) +
-                                            max_threads * sizeof(pid_t));
+        struct thread_map *threads = thread_map__alloc(max_threads);
+
         if (threads == NULL)
                 goto out;
 
@@ -185,8 +194,7 @@ static struct thread_map *thread_map__new_by_pid_str(const char *pid_str)
                         goto out_free_threads;
 
                 total_tasks += items;
-                nt = realloc(threads, (sizeof(*threads) +
-                                       sizeof(pid_t) * total_tasks));
+                nt = thread_map__realloc(threads, total_tasks);
                 if (nt == NULL)
                         goto out_free_namelist;
 
@@ -216,7 +224,7 @@ out_free_threads:
 
 struct thread_map *thread_map__new_dummy(void)
 {
-        struct thread_map *threads = malloc(sizeof(*threads) + sizeof(pid_t));
+        struct thread_map *threads = thread_map__alloc(1);
 
         if (threads != NULL) {
                 threads->map[0] = -1;
@@ -253,7 +261,7 @@ static struct thread_map *thread_map__new_by_tid_str(const char *tid_str)
                         continue;
 
                 ntasks++;
-                nt = realloc(threads, sizeof(*threads) + sizeof(pid_t) * ntasks);
+                nt = thread_map__realloc(threads, ntasks);
 
                 if (nt == NULL)
                         goto out_free_threads;