1 /* Copyright (C) 2004-2006, Advanced Micro Devices, Inc.
3 * This program is free software; you can redistribute it and/or modify
4 * it under the terms of the GNU General Public License as published by
5 * the Free Software Foundation; either version 2 of the License, or
6 * (at your option) any later version.
9 #include <linux/module.h>
10 #include <linux/kernel.h>
11 #include <linux/sched.h>
12 #include <linux/pci.h>
13 #include <linux/pci_ids.h>
14 #include <linux/crypto.h>
15 #include <linux/spinlock.h>
16 #include <crypto/algapi.h>
19 #include <asm/delay.h>
21 #include "geode-aes.h"
23 /* Register definitions */
25 #define AES_CTRLA_REG 0x0000
27 #define AES_CTRL_START 0x01
28 #define AES_CTRL_DECRYPT 0x00
29 #define AES_CTRL_ENCRYPT 0x02
30 #define AES_CTRL_WRKEY 0x04
31 #define AES_CTRL_DCA 0x08
32 #define AES_CTRL_SCA 0x10
33 #define AES_CTRL_CBC 0x20
35 #define AES_INTR_REG 0x0008
37 #define AES_INTRA_PENDING (1 << 16)
38 #define AES_INTRB_PENDING (1 << 17)
40 #define AES_INTR_PENDING (AES_INTRA_PENDING | AES_INTRB_PENDING)
41 #define AES_INTR_MASK 0x07
43 #define AES_SOURCEA_REG 0x0010
44 #define AES_DSTA_REG 0x0014
45 #define AES_LENA_REG 0x0018
46 #define AES_WRITEKEY0_REG 0x0030
47 #define AES_WRITEIV0_REG 0x0040
49 /* A very large counter that is used to gracefully bail out of an
50 * operation in case of trouble
53 #define AES_OP_TIMEOUT 0x50000
55 /* Static structures */
57 static void __iomem
* _iobase
;
58 static spinlock_t lock
;
60 /* Write a 128 bit field (either a writable key or IV) */
62 _writefield(u32 offset
, void *value
)
65 for(i
= 0; i
< 4; i
++)
66 iowrite32(((u32
*) value
)[i
], _iobase
+ offset
+ (i
* 4));
69 /* Read a 128 bit field (either a writable key or IV) */
71 _readfield(u32 offset
, void *value
)
74 for(i
= 0; i
< 4; i
++)
75 ((u32
*) value
)[i
] = ioread32(_iobase
+ offset
+ (i
* 4));
79 do_crypt(void *src
, void *dst
, int len
, u32 flags
)
82 u32 counter
= AES_OP_TIMEOUT
;
84 iowrite32(virt_to_phys(src
), _iobase
+ AES_SOURCEA_REG
);
85 iowrite32(virt_to_phys(dst
), _iobase
+ AES_DSTA_REG
);
86 iowrite32(len
, _iobase
+ AES_LENA_REG
);
88 /* Start the operation */
89 iowrite32(AES_CTRL_START
| flags
, _iobase
+ AES_CTRLA_REG
);
92 status
= ioread32(_iobase
+ AES_INTR_REG
);
93 while(!(status
& AES_INTRA_PENDING
) && --counter
);
96 iowrite32((status
& 0xFF) | AES_INTRA_PENDING
, _iobase
+ AES_INTR_REG
);
97 return counter
? 0 : 1;
101 geode_aes_crypt(struct geode_aes_op
*op
)
107 if (op
->len
== 0 || op
->src
== op
->dst
)
110 if (op
->flags
& AES_FLAGS_COHERENT
)
111 flags
|= (AES_CTRL_DCA
| AES_CTRL_SCA
);
113 if (op
->dir
== AES_DIR_ENCRYPT
)
114 flags
|= AES_CTRL_ENCRYPT
;
116 /* Start the critical section */
118 spin_lock_irqsave(&lock
, iflags
);
120 if (op
->mode
== AES_MODE_CBC
) {
121 flags
|= AES_CTRL_CBC
;
122 _writefield(AES_WRITEIV0_REG
, op
->iv
);
125 if (op
->flags
& AES_FLAGS_USRKEY
) {
126 flags
|= AES_CTRL_WRKEY
;
127 _writefield(AES_WRITEKEY0_REG
, op
->key
);
130 do_crypt(op
->src
, op
->dst
, op
->len
, flags
);
132 if (op
->mode
== AES_MODE_CBC
)
133 _readfield(AES_WRITEIV0_REG
, op
->iv
);
135 spin_unlock_irqrestore(&lock
, iflags
);
140 /* CRYPTO-API Functions */
143 geode_setkey(struct crypto_tfm
*tfm
, const u8
*key
, unsigned int len
)
145 struct geode_aes_op
*op
= crypto_tfm_ctx(tfm
);
147 if (len
!= AES_KEY_LENGTH
) {
148 tfm
->crt_flags
|= CRYPTO_TFM_RES_BAD_KEY_LEN
;
152 memcpy(op
->key
, key
, len
);
157 geode_encrypt(struct crypto_tfm
*tfm
, u8
*out
, const u8
*in
)
159 struct geode_aes_op
*op
= crypto_tfm_ctx(tfm
);
161 if ((out
== NULL
) || (in
== NULL
))
164 op
->src
= (void *) in
;
165 op
->dst
= (void *) out
;
166 op
->mode
= AES_MODE_ECB
;
168 op
->len
= AES_MIN_BLOCK_SIZE
;
169 op
->dir
= AES_DIR_ENCRYPT
;
176 geode_decrypt(struct crypto_tfm
*tfm
, u8
*out
, const u8
*in
)
178 struct geode_aes_op
*op
= crypto_tfm_ctx(tfm
);
180 if ((out
== NULL
) || (in
== NULL
))
183 op
->src
= (void *) in
;
184 op
->dst
= (void *) out
;
185 op
->mode
= AES_MODE_ECB
;
187 op
->len
= AES_MIN_BLOCK_SIZE
;
188 op
->dir
= AES_DIR_DECRYPT
;
194 static struct crypto_alg geode_alg
= {
196 .cra_driver_name
= "geode-aes-128",
199 .cra_flags
= CRYPTO_ALG_TYPE_CIPHER
,
200 .cra_blocksize
= AES_MIN_BLOCK_SIZE
,
201 .cra_ctxsize
= sizeof(struct geode_aes_op
),
202 .cra_module
= THIS_MODULE
,
203 .cra_list
= LIST_HEAD_INIT(geode_alg
.cra_list
),
206 .cia_min_keysize
= AES_KEY_LENGTH
,
207 .cia_max_keysize
= AES_KEY_LENGTH
,
208 .cia_setkey
= geode_setkey
,
209 .cia_encrypt
= geode_encrypt
,
210 .cia_decrypt
= geode_decrypt
216 geode_cbc_decrypt(struct blkcipher_desc
*desc
,
217 struct scatterlist
*dst
, struct scatterlist
*src
,
220 struct geode_aes_op
*op
= crypto_blkcipher_ctx(desc
->tfm
);
221 struct blkcipher_walk walk
;
224 blkcipher_walk_init(&walk
, dst
, src
, nbytes
);
225 err
= blkcipher_walk_virt(desc
, &walk
);
227 while((nbytes
= walk
.nbytes
)) {
228 op
->src
= walk
.src
.virt
.addr
,
229 op
->dst
= walk
.dst
.virt
.addr
;
230 op
->mode
= AES_MODE_CBC
;
231 op
->len
= nbytes
- (nbytes
% AES_MIN_BLOCK_SIZE
);
232 op
->dir
= AES_DIR_DECRYPT
;
234 memcpy(op
->iv
, walk
.iv
, AES_IV_LENGTH
);
236 ret
= geode_aes_crypt(op
);
238 memcpy(walk
.iv
, op
->iv
, AES_IV_LENGTH
);
241 err
= blkcipher_walk_done(desc
, &walk
, nbytes
);
248 geode_cbc_encrypt(struct blkcipher_desc
*desc
,
249 struct scatterlist
*dst
, struct scatterlist
*src
,
252 struct geode_aes_op
*op
= crypto_blkcipher_ctx(desc
->tfm
);
253 struct blkcipher_walk walk
;
256 blkcipher_walk_init(&walk
, dst
, src
, nbytes
);
257 err
= blkcipher_walk_virt(desc
, &walk
);
259 while((nbytes
= walk
.nbytes
)) {
260 op
->src
= walk
.src
.virt
.addr
,
261 op
->dst
= walk
.dst
.virt
.addr
;
262 op
->mode
= AES_MODE_CBC
;
263 op
->len
= nbytes
- (nbytes
% AES_MIN_BLOCK_SIZE
);
264 op
->dir
= AES_DIR_ENCRYPT
;
266 memcpy(op
->iv
, walk
.iv
, AES_IV_LENGTH
);
268 ret
= geode_aes_crypt(op
);
270 err
= blkcipher_walk_done(desc
, &walk
, nbytes
);
276 static struct crypto_alg geode_cbc_alg
= {
277 .cra_name
= "cbc(aes)",
278 .cra_driver_name
= "cbc-aes-geode-128",
280 .cra_flags
= CRYPTO_ALG_TYPE_BLKCIPHER
,
281 .cra_blocksize
= AES_MIN_BLOCK_SIZE
,
282 .cra_ctxsize
= sizeof(struct geode_aes_op
),
284 .cra_type
= &crypto_blkcipher_type
,
285 .cra_module
= THIS_MODULE
,
286 .cra_list
= LIST_HEAD_INIT(geode_cbc_alg
.cra_list
),
289 .min_keysize
= AES_KEY_LENGTH
,
290 .max_keysize
= AES_KEY_LENGTH
,
291 .setkey
= geode_setkey
,
292 .encrypt
= geode_cbc_encrypt
,
293 .decrypt
= geode_cbc_decrypt
,
299 geode_ecb_decrypt(struct blkcipher_desc
*desc
,
300 struct scatterlist
*dst
, struct scatterlist
*src
,
303 struct geode_aes_op
*op
= crypto_blkcipher_ctx(desc
->tfm
);
304 struct blkcipher_walk walk
;
307 blkcipher_walk_init(&walk
, dst
, src
, nbytes
);
308 err
= blkcipher_walk_virt(desc
, &walk
);
310 while((nbytes
= walk
.nbytes
)) {
311 op
->src
= walk
.src
.virt
.addr
,
312 op
->dst
= walk
.dst
.virt
.addr
;
313 op
->mode
= AES_MODE_ECB
;
314 op
->len
= nbytes
- (nbytes
% AES_MIN_BLOCK_SIZE
);
315 op
->dir
= AES_DIR_DECRYPT
;
317 ret
= geode_aes_crypt(op
);
319 err
= blkcipher_walk_done(desc
, &walk
, nbytes
);
326 geode_ecb_encrypt(struct blkcipher_desc
*desc
,
327 struct scatterlist
*dst
, struct scatterlist
*src
,
330 struct geode_aes_op
*op
= crypto_blkcipher_ctx(desc
->tfm
);
331 struct blkcipher_walk walk
;
334 blkcipher_walk_init(&walk
, dst
, src
, nbytes
);
335 err
= blkcipher_walk_virt(desc
, &walk
);
337 while((nbytes
= walk
.nbytes
)) {
338 op
->src
= walk
.src
.virt
.addr
,
339 op
->dst
= walk
.dst
.virt
.addr
;
340 op
->mode
= AES_MODE_ECB
;
341 op
->len
= nbytes
- (nbytes
% AES_MIN_BLOCK_SIZE
);
342 op
->dir
= AES_DIR_ENCRYPT
;
344 ret
= geode_aes_crypt(op
);
346 ret
= blkcipher_walk_done(desc
, &walk
, nbytes
);
352 static struct crypto_alg geode_ecb_alg
= {
353 .cra_name
= "ecb(aes)",
354 .cra_driver_name
= "ecb-aes-geode-128",
356 .cra_flags
= CRYPTO_ALG_TYPE_BLKCIPHER
,
357 .cra_blocksize
= AES_MIN_BLOCK_SIZE
,
358 .cra_ctxsize
= sizeof(struct geode_aes_op
),
360 .cra_type
= &crypto_blkcipher_type
,
361 .cra_module
= THIS_MODULE
,
362 .cra_list
= LIST_HEAD_INIT(geode_ecb_alg
.cra_list
),
365 .min_keysize
= AES_KEY_LENGTH
,
366 .max_keysize
= AES_KEY_LENGTH
,
367 .setkey
= geode_setkey
,
368 .encrypt
= geode_ecb_encrypt
,
369 .decrypt
= geode_ecb_decrypt
,
375 geode_aes_remove(struct pci_dev
*dev
)
377 crypto_unregister_alg(&geode_alg
);
378 crypto_unregister_alg(&geode_ecb_alg
);
379 crypto_unregister_alg(&geode_cbc_alg
);
381 pci_iounmap(dev
, _iobase
);
384 pci_release_regions(dev
);
385 pci_disable_device(dev
);
390 geode_aes_probe(struct pci_dev
*dev
, const struct pci_device_id
*id
)
394 if ((ret
= pci_enable_device(dev
)))
397 if ((ret
= pci_request_regions(dev
, "geode-aes-128")))
400 _iobase
= pci_iomap(dev
, 0, 0);
402 if (_iobase
== NULL
) {
407 spin_lock_init(&lock
);
409 /* Clear any pending activity */
410 iowrite32(AES_INTR_PENDING
| AES_INTR_MASK
, _iobase
+ AES_INTR_REG
);
412 if ((ret
= crypto_register_alg(&geode_alg
)))
415 if ((ret
= crypto_register_alg(&geode_ecb_alg
)))
418 if ((ret
= crypto_register_alg(&geode_cbc_alg
)))
421 printk(KERN_NOTICE
"geode-aes: GEODE AES engine enabled.\n");
425 crypto_unregister_alg(&geode_ecb_alg
);
428 crypto_unregister_alg(&geode_alg
);
431 pci_iounmap(dev
, _iobase
);
434 pci_release_regions(dev
);
437 pci_disable_device(dev
);
439 printk(KERN_ERR
"geode-aes: GEODE AES initialization failed.\n");
443 static struct pci_device_id geode_aes_tbl
[] = {
444 { PCI_VENDOR_ID_AMD
, PCI_DEVICE_ID_AMD_LX_AES
, PCI_ANY_ID
, PCI_ANY_ID
} ,
448 MODULE_DEVICE_TABLE(pci
, geode_aes_tbl
);
450 static struct pci_driver geode_aes_driver
= {
451 .name
= "Geode LX AES",
452 .id_table
= geode_aes_tbl
,
453 .probe
= geode_aes_probe
,
454 .remove
= __devexit_p(geode_aes_remove
)
460 return pci_register_driver(&geode_aes_driver
);
466 pci_unregister_driver(&geode_aes_driver
);
469 MODULE_AUTHOR("Advanced Micro Devices, Inc.");
470 MODULE_DESCRIPTION("Geode LX Hardware AES driver");
471 MODULE_LICENSE("GPL");
473 module_init(geode_aes_init
);
474 module_exit(geode_aes_exit
);