1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright 2014 Freescale Semiconductor, Inc.
4  *
5  */
6 
7 #include <common.h>
8 #include <cpu_func.h>
9 #include <log.h>
10 #include <malloc.h>
11 #include <memalign.h>
12 #include <fsl_sec.h>
13 #include <asm/cache.h>
14 #include <linux/errno.h>
15 #include "jobdesc.h"
16 #include "desc.h"
17 #include "jr.h"
18 
19 /**
20  * blob_decap() - Decapsulate the data from a blob
21  * @key_mod:    - Key modifier address
22  * @src:        - Source address (blob)
23  * @dst:        - Destination address (data)
24  * @len:        - Size of decapsulated data
25  *
26  * Note: Start and end of the key_mod, src and dst buffers have to be aligned to
27  * the cache line size (ARCH_DMA_MINALIGN) for the CAAM operation to succeed.
28  *
29  * Returns zero on success, negative on error.
30  */
blob_decap(u8 * key_mod,u8 * src,u8 * dst,u32 len)31 int blob_decap(u8 *key_mod, u8 *src, u8 *dst, u32 len)
32 {
33 	int ret, size, i = 0;
34 	u32 *desc;
35 
36 	if (!IS_ALIGNED((uintptr_t)key_mod, ARCH_DMA_MINALIGN) ||
37 	    !IS_ALIGNED((uintptr_t)src, ARCH_DMA_MINALIGN) ||
38 	    !IS_ALIGNED((uintptr_t)dst, ARCH_DMA_MINALIGN)) {
39 		puts("Error: blob_decap: Address arguments are not aligned!\n");
40 		return -EINVAL;
41 	}
42 
43 	printf("\nDecapsulating blob to get data\n");
44 	desc = malloc_cache_aligned(sizeof(int) * MAX_CAAM_DESCSIZE);
45 	if (!desc) {
46 		debug("Not enough memory for descriptor allocation\n");
47 		return -ENOMEM;
48 	}
49 
50 	size = ALIGN(16, ARCH_DMA_MINALIGN);
51 	flush_dcache_range((unsigned long)key_mod,
52 			   (unsigned long)key_mod + size);
53 
54 	size = ALIGN(BLOB_SIZE(len), ARCH_DMA_MINALIGN);
55 	flush_dcache_range((unsigned long)src,
56 			   (unsigned long)src + size);
57 
58 	inline_cnstr_jobdesc_blob_decap(desc, key_mod, src, dst, len);
59 
60 	debug("Descriptor dump:\n");
61 	for (i = 0; i < 14; i++)
62 		debug("Word[%d]: %08x\n", i, *(desc + i));
63 
64 	size = ALIGN(sizeof(int) * MAX_CAAM_DESCSIZE, ARCH_DMA_MINALIGN);
65 	flush_dcache_range((unsigned long)desc,
66 			   (unsigned long)desc + size);
67 
68 	flush_dcache_range((unsigned long)dst,
69 			   (unsigned long)dst + size);
70 
71 	ret = run_descriptor_jr(desc);
72 
73 	if (ret) {
74 		printf("Error in blob decapsulation: %d\n", ret);
75 	} else {
76 		size = ALIGN(len, ARCH_DMA_MINALIGN);
77 		invalidate_dcache_range((unsigned long)dst,
78 					(unsigned long)dst + size);
79 
80 		puts("Blob decapsulation successful.\n");
81 	}
82 
83 	free(desc);
84 	return ret;
85 }
86 
87 /**
88  * blob_encap() - Encapsulate the data as a blob
89  * @key_mod:    - Key modifier address
90  * @src:        - Source address (data)
91  * @dst:        - Destination address (blob)
92  * @len:        - Size of data to be encapsulated
93  *
94  * Note: Start and end of the key_mod, src and dst buffers have to be aligned to
95  * the cache line size (ARCH_DMA_MINALIGN) for the CAAM operation to succeed.
96  *
97  * Returns zero on success, negative on error.
98  */
blob_encap(u8 * key_mod,u8 * src,u8 * dst,u32 len)99 int blob_encap(u8 *key_mod, u8 *src, u8 *dst, u32 len)
100 {
101 	int ret, size, i = 0;
102 	u32 *desc;
103 
104 	if (!IS_ALIGNED((uintptr_t)key_mod, ARCH_DMA_MINALIGN) ||
105 	    !IS_ALIGNED((uintptr_t)src, ARCH_DMA_MINALIGN) ||
106 	    !IS_ALIGNED((uintptr_t)dst, ARCH_DMA_MINALIGN)) {
107 		puts("Error: blob_encap: Address arguments are not aligned!\n");
108 		return -EINVAL;
109 	}
110 
111 	printf("\nEncapsulating data to form blob\n");
112 	desc = malloc_cache_aligned(sizeof(int) * MAX_CAAM_DESCSIZE);
113 	if (!desc) {
114 		debug("Not enough memory for descriptor allocation\n");
115 		return -ENOMEM;
116 	}
117 
118 	size = ALIGN(16, ARCH_DMA_MINALIGN);
119 	flush_dcache_range((unsigned long)key_mod,
120 			   (unsigned long)key_mod + size);
121 
122 	size = ALIGN(len, ARCH_DMA_MINALIGN);
123 	flush_dcache_range((unsigned long)src,
124 			   (unsigned long)src + size);
125 
126 	inline_cnstr_jobdesc_blob_encap(desc, key_mod, src, dst, len);
127 
128 	debug("Descriptor dump:\n");
129 	for (i = 0; i < 14; i++)
130 		debug("Word[%d]: %08x\n", i, *(desc + i));
131 
132 	size = ALIGN(sizeof(int) * MAX_CAAM_DESCSIZE, ARCH_DMA_MINALIGN);
133 	flush_dcache_range((unsigned long)desc,
134 			   (unsigned long)desc + size);
135 
136 	flush_dcache_range((unsigned long)dst,
137 			   (unsigned long)dst + size);
138 
139 	ret = run_descriptor_jr(desc);
140 
141 	if (ret) {
142 		printf("Error in blob encapsulation: %d\n", ret);
143 	} else {
144 		size = ALIGN(BLOB_SIZE(len), ARCH_DMA_MINALIGN);
145 		invalidate_dcache_range((unsigned long)dst,
146 					(unsigned long)dst + size);
147 
148 		puts("Blob encapsulation successful.\n");
149 	}
150 
151 	free(desc);
152 	return ret;
153 }
154 
155 #ifdef CONFIG_CMD_DEKBLOB
blob_dek(const u8 * src,u8 * dst,u8 len)156 int blob_dek(const u8 *src, u8 *dst, u8 len)
157 {
158 	int ret, size, i = 0;
159 	u32 *desc;
160 
161 	int out_sz =  WRP_HDR_SIZE + len + KEY_BLOB_SIZE + MAC_SIZE;
162 
163 	puts("\nEncapsulating provided DEK to form blob\n");
164 	desc = memalign(ARCH_DMA_MINALIGN,
165 			sizeof(uint32_t) * DEK_BLOB_DESCSIZE);
166 	if (!desc) {
167 		debug("Not enough memory for descriptor allocation\n");
168 		return -ENOMEM;
169 	}
170 
171 	ret = inline_cnstr_jobdesc_blob_dek(desc, src, dst, len);
172 	if (ret) {
173 		debug("Error in Job Descriptor Construction:  %d\n", ret);
174 	} else {
175 		size = roundup(sizeof(uint32_t) * DEK_BLOB_DESCSIZE,
176 			      ARCH_DMA_MINALIGN);
177 		flush_dcache_range((unsigned long)desc,
178 				   (unsigned long)desc + size);
179 		size = roundup(sizeof(uint8_t) * out_sz, ARCH_DMA_MINALIGN);
180 		flush_dcache_range((unsigned long)dst,
181 				   (unsigned long)dst + size);
182 
183 		ret = run_descriptor_jr(desc);
184 	}
185 
186 	if (ret) {
187 		debug("Error in Encapsulation %d\n", ret);
188 	   goto err;
189 	}
190 
191 	size = roundup(out_sz, ARCH_DMA_MINALIGN);
192 	invalidate_dcache_range((unsigned long)dst, (unsigned long)dst+size);
193 
194 	puts("DEK Blob\n");
195 	for (i = 0; i < out_sz; i++)
196 		printf("%02X", ((uint8_t *)dst)[i]);
197 	printf("\n");
198 
199 err:
200 	free(desc);
201 	return ret;
202 }
203 #endif
204