tee: optee: Fix RPC call break system sleep operation.

Utilize freezer help functions to tell freezer to ignoring tasks
which wait on RPC result, because tee-supplicant may already be
frozen.

Signed-off-by: Shown Han <[email protected]>

Author: Shown Han

drivers/tee/optee/call.c

@@ -48,8 +48,14 @@ static void optee_cq_wait_init(struct optee_call_queue *cq,
 static void optee_cq_wait_for_completion(struct optee_call_queue *cq,
 					 struct optee_call_waiter *w)
 {
+	/* Tell freezers to ignore the current task. */
+	freezer_do_not_count();
+
 	wait_for_completion(&w->c);
 
+	/* Tell freezer to stop ignoring current task. */
+	freezer_count();
+
 	mutex_lock(&cq->mutex);
 
 	/* Move to end of list to get out of the way for other waiters */

drivers/tee/optee/optee_private.h

@@ -10,6 +10,7 @@
 #include <linux/semaphore.h>
 #include <linux/tee_drv.h>
 #include <linux/types.h>
+#include <linux/freezer.h>
 #include "optee_msg.h"
 
 #define OPTEE_MAX_ARG_SIZE	1024

drivers/tee/optee/rpc.c

@@ -77,7 +77,13 @@ static void wq_sleep(struct optee_wait_queue *wq, u32 key)
 	struct wq_entry *w = wq_entry_get(wq, key);
 
 	if (w) {
+		/* Tell freezers to ignore the current task. */
+		freezer_do_not_count();
+
 		wait_for_completion(&w->c);
+
+		/* Tell freezer to stop ignoring current task. */
+		freezer_count();
 		mutex_lock(&wq->mu);
 		list_del(&w->link);
 		mutex_unlock(&wq->mu);

drivers/tee/optee/supp.c

@@ -108,6 +108,10 @@ u32 optee_supp_thrd_req(struct tee_context *ctx, u32 func, size_t num_params,
 	/* Tell an eventual waiter there's a new request */
 	complete(&supp->reqs_c);
 
+
+	/* Tell freezers to ignore the current task. */
+	freezer_do_not_count();
+
 	/*
 	 * Wait for supplicant to process and return result, once we've
 	 * returned from wait_for_completion(&req->c) successfully we have
@@ -143,6 +147,9 @@ u32 optee_supp_thrd_req(struct tee_context *ctx, u32 func, size_t num_params,
 		}
 	}
 
+	/* Tell freezer to stop ignoring current task. */
+	freezer_count();
+
 	ret = req->ret;
 	kfree(req);