korg/linux
0
0
Fork 0
mirror of https://mirrors.bfsu.edu.cn/git/linux.git synced 2024-09-21 20:22:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf comm: Fix comm_str__put() for reference count checking

Browse Source 
Searching for the entry in the array needs to avoid the intermediate
pointer with reference count checking.

Refactor the array removal to binary search for the entry.

Change the array to hold an entry with a reference count (so the
intermediate pointer can work) and remove from the array when the
reference count on a comm_str falls to 1.

Fixes: 13ca628716 ("perf comm: Add reference count checking to 'struct comm_str'")
Signed-off-by: Ian Rogers <irogers@google.com>
Cc: Adrian Hunter <adrian.hunter@intel.com>
Cc: Alexander Shishkin <alexander.shishkin@linux.intel.com>
Cc: Ingo Molnar <mingo@redhat.com>
Cc: James Clark <james.clark@arm.com>
Cc: Jiri Olsa <jolsa@kernel.org>
Cc: Kan Liang <kan.liang@linux.intel.com>
Cc: Leo Yan <leo.yan@linux.dev>
Cc: Mark Rutland <mark.rutland@arm.com>
Cc: Namhyung Kim <namhyung@kernel.org>
Cc: Peter Zijlstra <peterz@infradead.org>
Link: https://lore.kernel.org/r/20240508035301.1554434-3-irogers@google.com
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>
This commit is contained in:
Ian Rogers 2024-05-07 20:52:59 -07:00 committed by Arnaldo Carvalho de Melo
parent 90f01afb0d
commit de6a908384
1 changed files with 31 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
tools/perf/util/comm.c
View File

@ -19,6 +19,8 @@ static struct comm_strs {
int capacity;
} _comm_strs;
static void comm_strs__remove_if_last(struct comm_str *cs);
static void comm_strs__init(void)
{
init_rwsem(&_comm_strs.lock);
@ -58,22 +60,15 @@ static struct comm_str *comm_str__get(struct comm_str *cs)
static void comm_str__put(struct comm_str *cs)
{
if (cs && refcount_dec_and_test(comm_str__refcnt(cs))) {
struct comm_strs *comm_strs = comm_strs__get();
int i;
if (!cs)
return;
down_write(&comm_strs->lock);
for (i = 0; i < comm_strs->num_strs; i++) {
if (comm_strs->strs[i] == cs)
break;
}
for (; i < comm_strs->num_strs - 1; i++)
comm_strs->strs[i] = comm_strs->strs[i + 1];
comm_strs->num_strs--;
up_write(&comm_strs->lock);
if (refcount_dec_and_test(comm_str__refcnt(cs))) {
RC_CHK_FREE(cs);
} else {
if (refcount_read(comm_str__refcnt(cs)) == 1)
comm_strs__remove_if_last(cs);
RC_CHK_PUT(cs);
}
}
@ -107,6 +102,28 @@ static int comm_str__search(const void *_key, const void *_member)
return strcmp(key, comm_str__str(member));
}
static void comm_strs__remove_if_last(struct comm_str *cs)
{
struct comm_strs *comm_strs = comm_strs__get();
down_write(&comm_strs->lock);
/*
* Are there only references from the array, if so remove the array
* reference under the write lock so that we don't race with findnew.
*/
if (refcount_read(comm_str__refcnt(cs)) == 1) {
struct comm_str **entry;
entry = bsearch(comm_str__str(cs), comm_strs->strs, comm_strs->num_strs,
sizeof(struct comm_str *), comm_str__search);
comm_str__put(*entry);
for (int i = entry - comm_strs->strs; i < comm_strs->num_strs - 1; i++)
comm_strs->strs[i] = comm_strs->strs[i + 1];
comm_strs->num_strs--;
}
up_write(&comm_strs->lock);
}
static struct comm_str *__comm_strs__find(struct comm_strs *comm_strs, const char *str)
{
struct comm_str **result;
@ -158,7 +175,7 @@ static struct comm_str *comm_strs__findnew(const char *str)
}
}
up_write(&comm_strs->lock);
return result;
return comm_str__get(result);
}
struct comm *comm__new(const char *str, u64 timestamp, bool exec)