[patch 05/18] x86, bts: fix race between per-task and per-cpu branch tracing

markus.t.metzger
2009-04-02T14:59:07Z

[-- Attachment #1: tip.master.ds.get_tracer.patch --]
[-- Type: text/plain, Size: 5233 bytes --]

Per-task branch tracing installs a debug store context with the traced
task. This immediately results in the branch trace control bits to be
cleared for the next context switch of that task, if not set before.

Either per-cpu or per-task tracing are allowed at the same time.

An active per-cpu tracing would be disabled even if the per-task tracing
request is rejected and the task debug store context removed.

Check the tracing type (per-cpu or per-task) before installing a task
debug store context.

Signed-off-by: Markus Metzger <markus.t.metzger@intel.com>
---
arch/x86/kernel/ds.c | 78 46 + 32 - 0 !
1 file changed, 46 insertions(+), 32 deletions(-)

Index: b/arch/x86/kernel/ds.c
===================================================================
--- a/arch/x86/kernel/ds.c
+++ b/arch/x86/kernel/ds.c
@@ -193,12 +193,28 @@ static DEFINE_SPINLOCK(ds_lock);
*/
static atomic_t tracers = ATOMIC_INIT(0);

-static inline void get_tracer(struct task_struct *task)
+static inline int get_tracer(struct task_struct *task)
{
- if (task)
+ int error;
+
+ spin_lock_irq(&ds_lock);
+
+ if (task) {
+ error = -EPERM;
+ if (atomic_read(&tracers) < 0)
+ goto out;
atomic_inc(&tracers);
- else
+ } else {
+ error = -EPERM;
+ if (atomic_read(&tracers) > 0)
+ goto out;
atomic_dec(&tracers);
+ }
+
+ error = 0;
+out:
+ spin_unlock_irq(&ds_lock);
+ return error;
}

static inline void put_tracer(struct task_struct *task)
@@ -209,14 +225,6 @@ static inline void put_tracer(struct tas
atomic_inc(&tracers);
}

-static inline int check_tracer(struct task_struct *task)
-{
- return task ?
- (atomic_read(&tracers) >= 0) :
- (atomic_read(&tracers) <= 0);
-}
-
-
/*
* The DS context is either attached to a thread or to a cpu:
* - in the former case, the thread_struct contains a pointer to the
@@ -707,6 +715,10 @@ struct bts_tracer *ds_request_bts(struct
if (ovfl)
goto out;

+ error = get_tracer(task);
+ if (error < 0)
+ goto out;
+
/*
* Per-cpu tracing is typically requested using smp_call_function().
* We must not sleep.
@@ -714,7 +726,7 @@ struct bts_tracer *ds_request_bts(struct
error = -ENOMEM;
tracer = kzalloc(sizeof(*tracer), GFP_ATOMIC);
if (!tracer)
- goto out;
+ goto out_put_tracer;
tracer->ovfl = ovfl;

error = ds_request(&tracer->ds, &tracer->trace.ds,
@@ -726,13 +738,8 @@ struct bts_tracer *ds_request_bts(struct
spin_lock_irqsave(&ds_lock, irq);

error = -EPERM;
- if (!check_tracer(task))
- goto out_unlock;
- get_tracer(task);
-
- error = -EPERM;
if (tracer->ds.context->bts_master)
- goto out_put_tracer;
+ goto out_unlock;
tracer->ds.context->bts_master = tracer;

spin_unlock_irqrestore(&ds_lock, irq);
@@ -746,13 +753,13 @@ struct bts_tracer *ds_request_bts(struct

return tracer;

- out_put_tracer:
- put_tracer(task);
out_unlock:
spin_unlock_irqrestore(&ds_lock, irq);
ds_put_context(tracer->ds.context);
out_tracer:
kfree(tracer);
+ out_put_tracer:
+ put_tracer(task);
out:
return ERR_PTR(error);
}
@@ -771,6 +778,10 @@ struct pebs_tracer *ds_request_pebs(stru
if (ovfl)
goto out;

+ error = get_tracer(task);
+ if (error < 0)
+ goto out;
+
/*
* Per-cpu tracing is typically requested using smp_call_function().
* We must not sleep.
@@ -778,7 +789,7 @@ struct pebs_tracer *ds_request_pebs(stru
error = -ENOMEM;
tracer = kzalloc(sizeof(*tracer), GFP_ATOMIC);
if (!tracer)
- goto out;
+ goto out_put_tracer;
tracer->ovfl = ovfl;

error = ds_request(&tracer->ds, &tracer->trace.ds,
@@ -789,13 +800,8 @@ struct pebs_tracer *ds_request_pebs(stru
spin_lock_irqsave(&ds_lock, irq);

error = -EPERM;
- if (!check_tracer(task))
- goto out_unlock;
- get_tracer(task);
-
- error = -EPERM;
if (tracer->ds.context->pebs_master)
- goto out_put_tracer;
+ goto out_unlock;
tracer->ds.context->pebs_master = tracer;

spin_unlock_irqrestore(&ds_lock, irq);
@@ -805,32 +811,36 @@ struct pebs_tracer *ds_request_pebs(stru

return tracer;

- out_put_tracer:
- put_tracer(task);
out_unlock:
spin_unlock_irqrestore(&ds_lock, irq);
ds_put_context(tracer->ds.context);
out_tracer:
kfree(tracer);
+ out_put_tracer:
+ put_tracer(task);
out:
return ERR_PTR(error);
}

void ds_release_bts(struct bts_tracer *tracer)
{
+ struct task_struct *task;
+
if (!tracer)
return;

+ task = tracer->ds.context->task;
+
ds_suspend_bts(tracer);

WARN_ON_ONCE(tracer->ds.context->bts_master != tracer);
tracer->ds.context->bts_master = NULL;

/* Make sure tracing stopped and the tracer is not in use. */
- wait_to_unschedule(tracer->ds.context->task);
+ wait_to_unschedule(task);

- put_tracer(tracer->ds.context->task);
ds_put_context(tracer->ds.context);
+ put_tracer(task);

kfree(tracer);
}
@@ -886,16 +896,20 @@ void ds_resume_bts(struct bts_tracer *tr

void ds_release_pebs(struct pebs_tracer *tracer)
{
+ struct task_struct *task;
+
if (!tracer)
return;

+ task = tracer->ds.context->task;
+
ds_suspend_pebs(tracer);

WARN_ON_ONCE(tracer->ds.context->pebs_master != tracer);
tracer->ds.context->pebs_master = NULL;

- put_tracer(tracer->ds.context->task);
ds_put_context(tracer->ds.context);
+ put_tracer(task);

kfree(tracer);
}