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 	ret = run_descriptor_jr(desc);
69 
70 	if (ret) {
71 		printf("Error in blob decapsulation: %d\n", ret);
72 	} else {
73 		size = ALIGN(len, ARCH_DMA_MINALIGN);
74 		invalidate_dcache_range((unsigned long)dst,
75 					(unsigned long)dst + size);
76 
77 		puts("Blob decapsulation successful.\n");
78 	}
79 
80 	free(desc);
81 	return ret;
82 }
83 
84 /**
85  * blob_encap() - Encapsulate the data as a blob
86  * @key_mod:    - Key modifier address
87  * @src:        - Source address (data)
88  * @dst:        - Destination address (blob)
89  * @len:        - Size of data to be encapsulated
90  *
91  * Note: Start and end of the key_mod, src and dst buffers have to be aligned to
92  * the cache line size (ARCH_DMA_MINALIGN) for the CAAM operation to succeed.
93  *
94  * Returns zero on success, negative on error.
95  */
blob_encap(u8 * key_mod,u8 * src,u8 * dst,u32 len)96 int blob_encap(u8 *key_mod, u8 *src, u8 *dst, u32 len)
97 {
98 	int ret, size, i = 0;
99 	u32 *desc;
100 
101 	if (!IS_ALIGNED((uintptr_t)key_mod, ARCH_DMA_MINALIGN) ||
102 	    !IS_ALIGNED((uintptr_t)src, ARCH_DMA_MINALIGN) ||
103 	    !IS_ALIGNED((uintptr_t)dst, ARCH_DMA_MINALIGN)) {
104 		puts("Error: blob_encap: Address arguments are not aligned!\n");
105 		return -EINVAL;
106 	}
107 
108 	printf("\nEncapsulating data to form blob\n");
109 	desc = malloc_cache_aligned(sizeof(int) * MAX_CAAM_DESCSIZE);
110 	if (!desc) {
111 		debug("Not enough memory for descriptor allocation\n");
112 		return -ENOMEM;
113 	}
114 
115 	size = ALIGN(16, ARCH_DMA_MINALIGN);
116 	flush_dcache_range((unsigned long)key_mod,
117 			   (unsigned long)key_mod + size);
118 
119 	size = ALIGN(len, ARCH_DMA_MINALIGN);
120 	flush_dcache_range((unsigned long)src,
121 			   (unsigned long)src + size);
122 
123 	inline_cnstr_jobdesc_blob_encap(desc, key_mod, src, dst, len);
124 
125 	debug("Descriptor dump:\n");
126 	for (i = 0; i < 14; i++)
127 		debug("Word[%d]: %08x\n", i, *(desc + i));
128 
129 	size = ALIGN(sizeof(int) * MAX_CAAM_DESCSIZE, ARCH_DMA_MINALIGN);
130 	flush_dcache_range((unsigned long)desc,
131 			   (unsigned long)desc + size);
132 
133 	ret = run_descriptor_jr(desc);
134 
135 	if (ret) {
136 		printf("Error in blob encapsulation: %d\n", ret);
137 	} else {
138 		size = ALIGN(BLOB_SIZE(len), ARCH_DMA_MINALIGN);
139 		invalidate_dcache_range((unsigned long)dst,
140 					(unsigned long)dst + size);
141 
142 		puts("Blob encapsulation successful.\n");
143 	}
144 
145 	free(desc);
146 	return ret;
147 }
148 
149 #ifdef CONFIG_CMD_DEKBLOB
blob_dek(const u8 * src,u8 * dst,u8 len)150 int blob_dek(const u8 *src, u8 *dst, u8 len)
151 {
152 	int ret, size, i = 0;
153 	u32 *desc;
154 
155 	int out_sz =  WRP_HDR_SIZE + len + KEY_BLOB_SIZE + MAC_SIZE;
156 
157 	puts("\nEncapsulating provided DEK to form blob\n");
158 	desc = memalign(ARCH_DMA_MINALIGN,
159 			sizeof(uint32_t) * DEK_BLOB_DESCSIZE);
160 	if (!desc) {
161 		debug("Not enough memory for descriptor allocation\n");
162 		return -ENOMEM;
163 	}
164 
165 	ret = inline_cnstr_jobdesc_blob_dek(desc, src, dst, len);
166 	if (ret) {
167 		debug("Error in Job Descriptor Construction:  %d\n", ret);
168 	} else {
169 		size = roundup(sizeof(uint32_t) * DEK_BLOB_DESCSIZE,
170 			      ARCH_DMA_MINALIGN);
171 		flush_dcache_range((unsigned long)desc,
172 				   (unsigned long)desc + size);
173 		size = roundup(sizeof(uint8_t) * out_sz, ARCH_DMA_MINALIGN);
174 		flush_dcache_range((unsigned long)dst,
175 				   (unsigned long)dst + size);
176 
177 		ret = run_descriptor_jr(desc);
178 	}
179 
180 	if (ret) {
181 		debug("Error in Encapsulation %d\n", ret);
182 	   goto err;
183 	}
184 
185 	size = roundup(out_sz, ARCH_DMA_MINALIGN);
186 	invalidate_dcache_range((unsigned long)dst, (unsigned long)dst+size);
187 
188 	puts("DEK Blob\n");
189 	for (i = 0; i < out_sz; i++)
190 		printf("%02X", ((uint8_t *)dst)[i]);
191 	printf("\n");
192 
193 err:
194 	free(desc);
195 	return ret;
196 }
197 #endif
198