Commit ccf7e72b authored by Hong zhi guo's avatar Hong zhi guo Committed by David S. Miller

macvtap: use prepare_to_wait/finish_wait to ensure mb

instead of raw assignment to current->state
Signed-off-by: default avatarHong Zhiguo <honkiko@gmail.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent e9939c80
...@@ -847,13 +847,12 @@ static ssize_t macvtap_do_read(struct macvtap_queue *q, struct kiocb *iocb, ...@@ -847,13 +847,12 @@ static ssize_t macvtap_do_read(struct macvtap_queue *q, struct kiocb *iocb,
const struct iovec *iv, unsigned long len, const struct iovec *iv, unsigned long len,
int noblock) int noblock)
{ {
DECLARE_WAITQUEUE(wait, current); DEFINE_WAIT(wait);
struct sk_buff *skb; struct sk_buff *skb;
ssize_t ret = 0; ssize_t ret = 0;
add_wait_queue(sk_sleep(&q->sk), &wait);
while (len) { while (len) {
current->state = TASK_INTERRUPTIBLE; prepare_to_wait(sk_sleep(&q->sk), &wait, TASK_INTERRUPTIBLE);
/* Read frames from the queue */ /* Read frames from the queue */
skb = skb_dequeue(&q->sk.sk_receive_queue); skb = skb_dequeue(&q->sk.sk_receive_queue);
...@@ -875,8 +874,7 @@ static ssize_t macvtap_do_read(struct macvtap_queue *q, struct kiocb *iocb, ...@@ -875,8 +874,7 @@ static ssize_t macvtap_do_read(struct macvtap_queue *q, struct kiocb *iocb,
break; break;
} }
current->state = TASK_RUNNING; finish_wait(sk_sleep(&q->sk), &wait);
remove_wait_queue(sk_sleep(&q->sk), &wait);
return ret; return ret;
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment