commit ec52ddfc437b438f65099088b40c1cc6c994ba0e
author Andrew Scull <ascull@google.com> Tue Aug 20 10:41:01 2019 +0100
committer Andrew Scull <ascull@google.com> Tue Aug 20 16:02:30 2019 +0000
tree 6e11791fe7e4484b28db9a68320c13125cf7648d
parent 4b5d036586e5601a14df575d67dbdcfe4b1b1b80

Lock vCPU when accessing enabled_and_pending_count.

Change-Id: Ic71d89c3b7a8acec71ac593228219e4044e58664

src/api.c

diff --git a/src/api.c b/src/api.c
index b3cab1a..081865e 100644
--- a/src/api.c
+++ b/src/api.c
@@ -1016,6 +1016,27 @@
 }
 
 /**
+ * Checks whether the vCPU's attempt to block for a message has already been
+ * interrupted or whether it is allowed to block.
+ */
+bool api_spci_msg_recv_block_interrupted(struct vcpu *current)
+{
+	bool interrupted;
+
+	sl_lock(&current->lock);
+
+	/*
+	 * Don't block if there are enabled and pending interrupts, to match
+	 * behaviour of wait_for_interrupt.
+	 */
+	interrupted = (current->interrupts.enabled_and_pending_count > 0);
+
+	sl_unlock(&current->lock);
+
+	return interrupted;
+}
+
+/**
  * Receives a message from the mailbox. If one isn't available, this function
  * can optionally block the caller until one becomes available.
  *
@@ -1058,12 +1079,7 @@
 	 * that time to SPCI_SUCCESS.
 	 */
 	return_code = SPCI_INTERRUPTED;
-
-	/*
-	 * Don't block if there are enabled and pending interrupts, to match
-	 * behaviour of wait_for_interrupt.
-	 */
-	if (current->interrupts.enabled_and_pending_count > 0) {
+	if (api_spci_msg_recv_block_interrupted(current)) {
 		goto out;
 	}