1 files changed, 9 insertions, 18 deletions

diff --git a/drivers/s390/crypto/zcrypt_pcica.c b/drivers/s390/crypto/zcrypt_pcica.c
index 818ffe05ac00..6e93b4751782 100644
--- a/drivers/s390/crypto/zcrypt_pcica.c
+++ b/drivers/s390/crypto/zcrypt_pcica.c
@@ -70,6 +70,7 @@ static struct ap_driver zcrypt_pcica_driver = {
         .remove = zcrypt_pcica_remove,
         .receive = zcrypt_pcica_receive,
         .ids = zcrypt_pcica_ids,
+        .request_timeout = PCICA_CLEANUP_TIME,
 };
 
 /**
@@ -290,18 +291,13 @@ static long zcrypt_pcica_modexpo(struct zcrypt_device *zdev,
                 goto out_free;
         init_completion(&work);
         ap_queue_message(zdev->ap_dev, &ap_msg);
-        rc = wait_for_completion_interruptible_timeout(
-                                &work, PCICA_CLEANUP_TIME);
-        if (rc > 0)
+        rc = wait_for_completion_interruptible(&work);
+        if (rc == 0)
                 rc = convert_response(zdev, &ap_msg, mex->outputdata,
                                       mex->outputdatalength);
-        else {
-                /* Signal pending or message timed out. */
+        else
+                /* Signal pending. */
                 ap_cancel_message(zdev->ap_dev, &ap_msg);
-                if (rc == 0)
-                        /* Message timed out. */
-                        rc = -ETIME;
-        }
 out_free:
         kfree(ap_msg.message);
         return rc;
@@ -332,18 +328,13 @@ static long zcrypt_pcica_modexpo_crt(struct zcrypt_device *zdev,
                 goto out_free;
         init_completion(&work);
         ap_queue_message(zdev->ap_dev, &ap_msg);
-        rc = wait_for_completion_interruptible_timeout(
-                                &work, PCICA_CLEANUP_TIME);
-        if (rc > 0)
+        rc = wait_for_completion_interruptible(&work);
+        if (rc == 0)
                 rc = convert_response(zdev, &ap_msg, crt->outputdata,
                                       crt->outputdatalength);
-        else {
-                /* Signal pending or message timed out. */
+        else
+                /* Signal pending. */
                 ap_cancel_message(zdev->ap_dev, &ap_msg);
-                if (rc == 0)
-                        /* Message timed out. */
-                        rc = -ETIME;
-        }
 out_free:
         kfree(ap_msg.message);
         return rc;