mac80211: Switch to new AEAD interface

This patch makes use of the new AEAD interface which uses a single
SG list instead of separate lists for the AD and plain text.

Tested-by: Johannes Berg <johannes@sipsolutions.net>
Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

1 files changed, 14 insertions, 16 deletions

diff --git a/net/mac80211/aes_gcm.c b/net/mac80211/aes_gcm.c
index b91c9d7bf665..3afe361fd27c 100644
--- a/net/mac80211/aes_gcm.c
+++ b/net/mac80211/aes_gcm.c
@@ -18,7 +18,7 @@
 void ieee80211_aes_gcm_encrypt(struct crypto_aead *tfm, u8 *j_0, u8 *aad,
                                u8 *data, size_t data_len, u8 *mic)
 {
-        struct scatterlist assoc, pt, ct[2];
+        struct scatterlist sg[3];
 
         char aead_req_data[sizeof(struct aead_request) +
                            crypto_aead_reqsize(tfm)]
@@ -27,15 +27,14 @@ void ieee80211_aes_gcm_encrypt(struct crypto_aead *tfm, u8 *j_0, u8 *aad,
 
         memset(aead_req, 0, sizeof(aead_req_data));
 
-        sg_init_one(&pt, data, data_len);
-        sg_init_one(&assoc, &aad[2], be16_to_cpup((__be16 *)aad));
-        sg_init_table(ct, 2);
-        sg_set_buf(&ct[0], data, data_len);
-        sg_set_buf(&ct[1], mic, IEEE80211_GCMP_MIC_LEN);
+        sg_init_table(sg, 3);
+        sg_set_buf(&sg[0], &aad[2], be16_to_cpup((__be16 *)aad));
+        sg_set_buf(&sg[1], data, data_len);
+        sg_set_buf(&sg[2], mic, IEEE80211_GCMP_MIC_LEN);
 
         aead_request_set_tfm(aead_req, tfm);
-        aead_request_set_assoc(aead_req, &assoc, assoc.length);
-        aead_request_set_crypt(aead_req, &pt, ct, data_len, j_0);
+        aead_request_set_crypt(aead_req, sg, sg, data_len, j_0);
+        aead_request_set_ad(aead_req, sg[0].length);
 
         crypto_aead_encrypt(aead_req);
 }
@@ -43,7 +42,7 @@ void ieee80211_aes_gcm_encrypt(struct crypto_aead *tfm, u8 *j_0, u8 *aad,
 int ieee80211_aes_gcm_decrypt(struct crypto_aead *tfm, u8 *j_0, u8 *aad,
                               u8 *data, size_t data_len, u8 *mic)
 {
-        struct scatterlist assoc, pt, ct[2];
+        struct scatterlist sg[3];
         char aead_req_data[sizeof(struct aead_request) +
                            crypto_aead_reqsize(tfm)]
                 __aligned(__alignof__(struct aead_request));
@@ -54,16 +53,15 @@ int ieee80211_aes_gcm_decrypt(struct crypto_aead *tfm, u8 *j_0, u8 *aad,
 
         memset(aead_req, 0, sizeof(aead_req_data));
 
-        sg_init_one(&pt, data, data_len);
-        sg_init_one(&assoc, &aad[2], be16_to_cpup((__be16 *)aad));
-        sg_init_table(ct, 2);
-        sg_set_buf(&ct[0], data, data_len);
-        sg_set_buf(&ct[1], mic, IEEE80211_GCMP_MIC_LEN);
+        sg_init_table(sg, 3);
+        sg_set_buf(&sg[0], &aad[2], be16_to_cpup((__be16 *)aad));
+        sg_set_buf(&sg[1], data, data_len);
+        sg_set_buf(&sg[2], mic, IEEE80211_GCMP_MIC_LEN);
 
         aead_request_set_tfm(aead_req, tfm);
-        aead_request_set_assoc(aead_req, &assoc, assoc.length);
-        aead_request_set_crypt(aead_req, ct, &pt,
+        aead_request_set_crypt(aead_req, sg, sg,
                                data_len + IEEE80211_GCMP_MIC_LEN, j_0);
+        aead_request_set_ad(aead_req, sg[0].length);
 
         return crypto_aead_decrypt(aead_req);
 }