1 /*
2  * Public Key Encryption
3  *
4  * Copyright (c) 2015, Intel Corporation
5  * Authors: Tadeusz Struk <tadeusz.struk@intel.com>
6  *
7  * This program is free software; you can redistribute it and/or modify it
8  * under the terms of the GNU General Public License as published by the Free
9  * Software Foundation; either version 2 of the License, or (at your option)
10  * any later version.
11  *
12  */
13 #ifndef _CRYPTO_AKCIPHER_H
14 #define _CRYPTO_AKCIPHER_H
15 #include <linux/crypto.h>
16 
17 /**
18  * struct akcipher_request - public key request
19  *
20  * @base:	Common attributes for async crypto requests
21  * @src:	Source data
22  * @dst:	Destination data
23  * @src_len:	Size of the input buffer
24  * @dst_len:	Size of the output buffer. It needs to be at least
25  *		as big as the expected result depending	on the operation
26  *		After operation it will be updated with the actual size of the
27  *		result.
28  *		In case of error where the dst sgl size was insufficient,
29  *		it will be updated to the size required for the operation.
30  * @__ctx:	Start of private context data
31  */
32 struct akcipher_request {
33 	struct crypto_async_request base;
34 	struct scatterlist *src;
35 	struct scatterlist *dst;
36 	unsigned int src_len;
37 	unsigned int dst_len;
38 	void *__ctx[] CRYPTO_MINALIGN_ATTR;
39 };
40 
41 /**
42  * struct crypto_akcipher - user-instantiated objects which encapsulate
43  * algorithms and core processing logic
44  *
45  * @base:	Common crypto API algorithm data structure
46  */
47 struct crypto_akcipher {
48 	struct crypto_tfm base;
49 };
50 
51 /**
52  * struct akcipher_alg - generic public key algorithm
53  *
54  * @sign:	Function performs a sign operation as defined by public key
55  *		algorithm. In case of error, where the dst_len was insufficient,
56  *		the req->dst_len will be updated to the size required for the
57  *		operation
58  * @verify:	Function performs a sign operation as defined by public key
59  *		algorithm. In case of error, where the dst_len was insufficient,
60  *		the req->dst_len will be updated to the size required for the
61  *		operation
62  * @encrypt:	Function performs an encrypt operation as defined by public key
63  *		algorithm. In case of error, where the dst_len was insufficient,
64  *		the req->dst_len will be updated to the size required for the
65  *		operation
66  * @decrypt:	Function performs a decrypt operation as defined by public key
67  *		algorithm. In case of error, where the dst_len was insufficient,
68  *		the req->dst_len will be updated to the size required for the
69  *		operation
70  * @set_pub_key: Function invokes the algorithm specific set public key
71  *		function, which knows how to decode and interpret
72  *		the BER encoded public key
73  * @set_priv_key: Function invokes the algorithm specific set private key
74  *		function, which knows how to decode and interpret
75  *		the BER encoded private key
76  * @max_size:	Function returns dest buffer size required for a given key.
77  * @init:	Initialize the cryptographic transformation object.
78  *		This function is used to initialize the cryptographic
79  *		transformation object. This function is called only once at
80  *		the instantiation time, right after the transformation context
81  *		was allocated. In case the cryptographic hardware has some
82  *		special requirements which need to be handled by software, this
83  *		function shall check for the precise requirement of the
84  *		transformation and put any software fallbacks in place.
85  * @exit:	Deinitialize the cryptographic transformation object. This is a
86  *		counterpart to @init, used to remove various changes set in
87  *		@init.
88  *
89  * @reqsize:	Request context size required by algorithm implementation
90  * @base:	Common crypto API algorithm data structure
91  */
92 struct akcipher_alg {
93 	int (*sign)(struct akcipher_request *req);
94 	int (*verify)(struct akcipher_request *req);
95 	int (*encrypt)(struct akcipher_request *req);
96 	int (*decrypt)(struct akcipher_request *req);
97 	int (*set_pub_key)(struct crypto_akcipher *tfm, const void *key,
98 			   unsigned int keylen);
99 	int (*set_priv_key)(struct crypto_akcipher *tfm, const void *key,
100 			    unsigned int keylen);
101 	unsigned int (*max_size)(struct crypto_akcipher *tfm);
102 	int (*init)(struct crypto_akcipher *tfm);
103 	void (*exit)(struct crypto_akcipher *tfm);
104 
105 	unsigned int reqsize;
106 	struct crypto_alg base;
107 };
108 
109 /**
110  * DOC: Generic Public Key API
111  *
112  * The Public Key API is used with the algorithms of type
113  * CRYPTO_ALG_TYPE_AKCIPHER (listed as type "akcipher" in /proc/crypto)
114  */
115 
116 /**
117  * crypto_alloc_akcipher() - allocate AKCIPHER tfm handle
118  * @alg_name: is the cra_name / name or cra_driver_name / driver name of the
119  *	      public key algorithm e.g. "rsa"
120  * @type: specifies the type of the algorithm
121  * @mask: specifies the mask for the algorithm
122  *
123  * Allocate a handle for public key algorithm. The returned struct
124  * crypto_akcipher is the handle that is required for any subsequent
125  * API invocation for the public key operations.
126  *
127  * Return: allocated handle in case of success; IS_ERR() is true in case
128  *	   of an error, PTR_ERR() returns the error code.
129  */
130 struct crypto_akcipher *crypto_alloc_akcipher(const char *alg_name, u32 type,
131 					      u32 mask);
132 
crypto_akcipher_tfm(struct crypto_akcipher * tfm)133 static inline struct crypto_tfm *crypto_akcipher_tfm(
134 	struct crypto_akcipher *tfm)
135 {
136 	return &tfm->base;
137 }
138 
__crypto_akcipher_alg(struct crypto_alg * alg)139 static inline struct akcipher_alg *__crypto_akcipher_alg(struct crypto_alg *alg)
140 {
141 	return container_of(alg, struct akcipher_alg, base);
142 }
143 
__crypto_akcipher_tfm(struct crypto_tfm * tfm)144 static inline struct crypto_akcipher *__crypto_akcipher_tfm(
145 	struct crypto_tfm *tfm)
146 {
147 	return container_of(tfm, struct crypto_akcipher, base);
148 }
149 
crypto_akcipher_alg(struct crypto_akcipher * tfm)150 static inline struct akcipher_alg *crypto_akcipher_alg(
151 	struct crypto_akcipher *tfm)
152 {
153 	return __crypto_akcipher_alg(crypto_akcipher_tfm(tfm)->__crt_alg);
154 }
155 
crypto_akcipher_reqsize(struct crypto_akcipher * tfm)156 static inline unsigned int crypto_akcipher_reqsize(struct crypto_akcipher *tfm)
157 {
158 	return crypto_akcipher_alg(tfm)->reqsize;
159 }
160 
akcipher_request_set_tfm(struct akcipher_request * req,struct crypto_akcipher * tfm)161 static inline void akcipher_request_set_tfm(struct akcipher_request *req,
162 					    struct crypto_akcipher *tfm)
163 {
164 	req->base.tfm = crypto_akcipher_tfm(tfm);
165 }
166 
crypto_akcipher_reqtfm(struct akcipher_request * req)167 static inline struct crypto_akcipher *crypto_akcipher_reqtfm(
168 	struct akcipher_request *req)
169 {
170 	return __crypto_akcipher_tfm(req->base.tfm);
171 }
172 
173 /**
174  * crypto_free_akcipher() - free AKCIPHER tfm handle
175  *
176  * @tfm: AKCIPHER tfm handle allocated with crypto_alloc_akcipher()
177  *
178  * If @tfm is a NULL or error pointer, this function does nothing.
179  */
crypto_free_akcipher(struct crypto_akcipher * tfm)180 static inline void crypto_free_akcipher(struct crypto_akcipher *tfm)
181 {
182 	crypto_destroy_tfm(tfm, crypto_akcipher_tfm(tfm));
183 }
184 
185 /**
186  * akcipher_request_alloc() - allocates public key request
187  *
188  * @tfm:	AKCIPHER tfm handle allocated with crypto_alloc_akcipher()
189  * @gfp:	allocation flags
190  *
191  * Return: allocated handle in case of success or NULL in case of an error.
192  */
akcipher_request_alloc(struct crypto_akcipher * tfm,gfp_t gfp)193 static inline struct akcipher_request *akcipher_request_alloc(
194 	struct crypto_akcipher *tfm, gfp_t gfp)
195 {
196 	struct akcipher_request *req;
197 
198 	req = kmalloc(sizeof(*req) + crypto_akcipher_reqsize(tfm), gfp);
199 	if (likely(req))
200 		akcipher_request_set_tfm(req, tfm);
201 
202 	return req;
203 }
204 
205 /**
206  * akcipher_request_free() - zeroize and free public key request
207  *
208  * @req:	request to free
209  */
akcipher_request_free(struct akcipher_request * req)210 static inline void akcipher_request_free(struct akcipher_request *req)
211 {
212 	kzfree(req);
213 }
214 
215 /**
216  * akcipher_request_set_callback() - Sets an asynchronous callback.
217  *
218  * Callback will be called when an asynchronous operation on a given
219  * request is finished.
220  *
221  * @req:	request that the callback will be set for
222  * @flgs:	specify for instance if the operation may backlog
223  * @cmpl:	callback which will be called
224  * @data:	private data used by the caller
225  */
akcipher_request_set_callback(struct akcipher_request * req,u32 flgs,crypto_completion_t cmpl,void * data)226 static inline void akcipher_request_set_callback(struct akcipher_request *req,
227 						 u32 flgs,
228 						 crypto_completion_t cmpl,
229 						 void *data)
230 {
231 	req->base.complete = cmpl;
232 	req->base.data = data;
233 	req->base.flags = flgs;
234 }
235 
236 /**
237  * akcipher_request_set_crypt() - Sets request parameters
238  *
239  * Sets parameters required by crypto operation
240  *
241  * @req:	public key request
242  * @src:	ptr to input scatter list
243  * @dst:	ptr to output scatter list
244  * @src_len:	size of the src input scatter list to be processed
245  * @dst_len:	size of the dst output scatter list
246  */
akcipher_request_set_crypt(struct akcipher_request * req,struct scatterlist * src,struct scatterlist * dst,unsigned int src_len,unsigned int dst_len)247 static inline void akcipher_request_set_crypt(struct akcipher_request *req,
248 					      struct scatterlist *src,
249 					      struct scatterlist *dst,
250 					      unsigned int src_len,
251 					      unsigned int dst_len)
252 {
253 	req->src = src;
254 	req->dst = dst;
255 	req->src_len = src_len;
256 	req->dst_len = dst_len;
257 }
258 
259 /**
260  * crypto_akcipher_maxsize() - Get len for output buffer
261  *
262  * Function returns the dest buffer size required for a given key.
263  * Function assumes that the key is already set in the transformation. If this
264  * function is called without a setkey or with a failed setkey, you will end up
265  * in a NULL dereference.
266  *
267  * @tfm:	AKCIPHER tfm handle allocated with crypto_alloc_akcipher()
268  */
crypto_akcipher_maxsize(struct crypto_akcipher * tfm)269 static inline unsigned int crypto_akcipher_maxsize(struct crypto_akcipher *tfm)
270 {
271 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
272 
273 	return alg->max_size(tfm);
274 }
275 
276 /**
277  * crypto_akcipher_encrypt() - Invoke public key encrypt operation
278  *
279  * Function invokes the specific public key encrypt operation for a given
280  * public key algorithm
281  *
282  * @req:	asymmetric key request
283  *
284  * Return: zero on success; error code in case of error
285  */
crypto_akcipher_encrypt(struct akcipher_request * req)286 static inline int crypto_akcipher_encrypt(struct akcipher_request *req)
287 {
288 	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
289 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
290 
291 	return alg->encrypt(req);
292 }
293 
294 /**
295  * crypto_akcipher_decrypt() - Invoke public key decrypt operation
296  *
297  * Function invokes the specific public key decrypt operation for a given
298  * public key algorithm
299  *
300  * @req:	asymmetric key request
301  *
302  * Return: zero on success; error code in case of error
303  */
crypto_akcipher_decrypt(struct akcipher_request * req)304 static inline int crypto_akcipher_decrypt(struct akcipher_request *req)
305 {
306 	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
307 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
308 
309 	return alg->decrypt(req);
310 }
311 
312 /**
313  * crypto_akcipher_sign() - Invoke public key sign operation
314  *
315  * Function invokes the specific public key sign operation for a given
316  * public key algorithm
317  *
318  * @req:	asymmetric key request
319  *
320  * Return: zero on success; error code in case of error
321  */
crypto_akcipher_sign(struct akcipher_request * req)322 static inline int crypto_akcipher_sign(struct akcipher_request *req)
323 {
324 	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
325 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
326 
327 	return alg->sign(req);
328 }
329 
330 /**
331  * crypto_akcipher_verify() - Invoke public key verify operation
332  *
333  * Function invokes the specific public key verify operation for a given
334  * public key algorithm
335  *
336  * @req:	asymmetric key request
337  *
338  * Return: zero on success; error code in case of error
339  */
crypto_akcipher_verify(struct akcipher_request * req)340 static inline int crypto_akcipher_verify(struct akcipher_request *req)
341 {
342 	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
343 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
344 
345 	return alg->verify(req);
346 }
347 
348 /**
349  * crypto_akcipher_set_pub_key() - Invoke set public key operation
350  *
351  * Function invokes the algorithm specific set key function, which knows
352  * how to decode and interpret the encoded key
353  *
354  * @tfm:	tfm handle
355  * @key:	BER encoded public key
356  * @keylen:	length of the key
357  *
358  * Return: zero on success; error code in case of error
359  */
crypto_akcipher_set_pub_key(struct crypto_akcipher * tfm,const void * key,unsigned int keylen)360 static inline int crypto_akcipher_set_pub_key(struct crypto_akcipher *tfm,
361 					      const void *key,
362 					      unsigned int keylen)
363 {
364 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
365 
366 	return alg->set_pub_key(tfm, key, keylen);
367 }
368 
369 /**
370  * crypto_akcipher_set_priv_key() - Invoke set private key operation
371  *
372  * Function invokes the algorithm specific set key function, which knows
373  * how to decode and interpret the encoded key
374  *
375  * @tfm:	tfm handle
376  * @key:	BER encoded private key
377  * @keylen:	length of the key
378  *
379  * Return: zero on success; error code in case of error
380  */
crypto_akcipher_set_priv_key(struct crypto_akcipher * tfm,const void * key,unsigned int keylen)381 static inline int crypto_akcipher_set_priv_key(struct crypto_akcipher *tfm,
382 					       const void *key,
383 					       unsigned int keylen)
384 {
385 	struct akcipher_alg *alg = crypto_akcipher_alg(tfm);
386 
387 	return alg->set_priv_key(tfm, key, keylen);
388 }
389 #endif
390