luck/tmp_suning_uos_patched
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

virtio: Fix scheduling while atomic in virtio_balloon stats

Browse Source 
This is a fix for my earlier patch: "virtio: Add memory statistics reporting to
the balloon driver (V4)".

I discovered that all_vm_events() can sleep and therefore stats collection
cannot be done in interrupt context.  One solution is to handle the interrupt
by noting that stats need to be collected and waking the existing vballoon
kthread which will complete the work via stats_handle_request().  Rusty, is
this a saner way of doing business?

There is one issue that I would like a broader opinion on.  In stats_request, I
update vb->need_stats_update and then wake up the kthread.  The kthread uses
vb->need_stats_update as a condition variable.  Do I need a memory barrier
between the update and wake_up to ensure that my kthread sees the correct
value?  My testing suggests that it is not needed but I would like some
confirmation from the experts.

Signed-off-by: Adam Litke <agl@us.ibm.com>
To: Rusty Russell <rusty@rustcorp.com.au>
Cc: Anthony Liguori <aliguori@linux.vnet.ibm.com>
Cc: linux-kernel@vger.kernel.org
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Adam Litke 2009-12-10 16:35:15 -06:00 committed by Rusty Russell
parent 9564e138b1
commit 1f34c71afe
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
drivers/virtio/virtio_balloon.c
View File

@ -51,6 +51,7 @@ struct virtio_balloon
u32 pfns[256]; u32 pfns[256];
/* Memory statistics */ /* Memory statistics */
int need_stats_update;
struct virtio_balloon_stat stats[VIRTIO_BALLOON_S_NR]; struct virtio_balloon_stat stats[VIRTIO_BALLOON_S_NR];
}; };
@ -193,20 +194,30 @@ static void update_balloon_stats(struct virtio_balloon *vb)
* the stats queue operates in reverse. The driver initializes the virtqueue * the stats queue operates in reverse. The driver initializes the virtqueue
* with a single buffer. From that point forward, all conversations consist of * with a single buffer. From that point forward, all conversations consist of
* a hypervisor request (a call to this function) which directs us to refill * a hypervisor request (a call to this function) which directs us to refill
* the virtqueue with a fresh stats buffer. * the virtqueue with a fresh stats buffer. Since stats collection can sleep,
* we notify our kthread which does the actual work via stats_handle_request().
*/ */
static void stats_ack(struct virtqueue *vq) static void stats_request(struct virtqueue *vq)
{ {
struct virtio_balloon *vb; struct virtio_balloon *vb;
unsigned int len; unsigned int len;
struct scatterlist sg;
vb = vq->vq_ops->get_buf(vq, &len); vb = vq->vq_ops->get_buf(vq, &len);
if (!vb) if (!vb)
return; return;
vb->need_stats_update = 1;
wake_up(&vb->config_change);
}
static void stats_handle_request(struct virtio_balloon *vb)
{
struct virtqueue *vq;
struct scatterlist sg;
vb->need_stats_update = 0;
update_balloon_stats(vb); update_balloon_stats(vb);
vq = vb->stats_vq;
sg_init_one(&sg, vb->stats, sizeof(vb->stats)); sg_init_one(&sg, vb->stats, sizeof(vb->stats));
if (vq->vq_ops->add_buf(vq, &sg, 1, 0, vb) < 0) if (vq->vq_ops->add_buf(vq, &sg, 1, 0, vb) < 0)
BUG(); BUG();
@ -249,8 +260,11 @@ static int balloon(void *_vballoon)
try_to_freeze(); try_to_freeze();
wait_event_interruptible(vb->config_change, wait_event_interruptible(vb->config_change,
(diff = towards_target(vb)) != 0 (diff = towards_target(vb)) != 0
|| vb->need_stats_update
|| kthread_should_stop() || kthread_should_stop()
|| freezing(current)); || freezing(current));
if (vb->need_stats_update)
stats_handle_request(vb);
if (diff > 0) if (diff > 0)
fill_balloon(vb, diff); fill_balloon(vb, diff);
else if (diff < 0) else if (diff < 0)
@ -264,7 +278,7 @@ static int virtballoon_probe(struct virtio_device *vdev)
{ {
struct virtio_balloon *vb; struct virtio_balloon *vb;
struct virtqueue *vqs[3]; struct virtqueue *vqs[3];
vq_callback_t *callbacks[] = { balloon_ack, balloon_ack, stats_ack }; vq_callback_t *callbacks[] = { balloon_ack, balloon_ack, stats_request };
const char *names[] = { "inflate", "deflate", "stats" }; const char *names[] = { "inflate", "deflate", "stats" };
int err, nvqs; int err, nvqs;