1 /***********************license start************************************ 2 * Copyright (c) 2003-2017 Cavium, Inc. 3 * All rights reserved. 4 * 5 * License: one of 'Cavium License' or 'GNU General Public License Version 2' 6 * 7 * This file is provided under the terms of the Cavium License (see below) 8 * or under the terms of GNU General Public License, Version 2, as 9 * published by the Free Software Foundation. When using or redistributing 10 * this file, you may do so under either license. 11 * 12 * Cavium License: Redistribution and use in source and binary forms, with 13 * or without modification, are permitted provided that the following 14 * conditions are met: 15 * 16 * * Redistributions of source code must retain the above copyright 17 * notice, this list of conditions and the following disclaimer. 18 * 19 * * Redistributions in binary form must reproduce the above 20 * copyright notice, this list of conditions and the following 21 * disclaimer in the documentation and/or other materials provided 22 * with the distribution. 23 * 24 * * Neither the name of Cavium Inc. nor the names of its contributors may be 25 * used to endorse or promote products derived from this software without 26 * specific prior written permission. 27 * 28 * This Software, including technical data, may be subject to U.S. export 29 * control laws, including the U.S. Export Administration Act and its 30 * associated regulations, and may be subject to export or import 31 * regulations in other countries. 32 * 33 * TO THE MAXIMUM EXTENT PERMITTED BY LAW, THE SOFTWARE IS PROVIDED "AS IS" 34 * AND WITH ALL FAULTS AND CAVIUM INC. MAKES NO PROMISES, REPRESENTATIONS 35 * OR WARRANTIES, EITHER EXPRESS, IMPLIED, STATUTORY, OR OTHERWISE, WITH 36 * RESPECT TO THE SOFTWARE, INCLUDING ITS CONDITION, ITS CONFORMITY TO ANY 37 * REPRESENTATION OR DESCRIPTION, OR THE EXISTENCE OF ANY LATENT OR PATENT 38 * DEFECTS, AND CAVIUM SPECIFICALLY DISCLAIMS ALL IMPLIED (IF ANY) 39 * WARRANTIES OF TITLE, MERCHANTABILITY, NONINFRINGEMENT, FITNESS FOR A 40 * PARTICULAR PURPOSE, LACK OF VIRUSES, ACCURACY OR COMPLETENESS, QUIET 41 * ENJOYMENT, QUIET POSSESSION OR CORRESPONDENCE TO DESCRIPTION. THE 42 * ENTIRE RISK ARISING OUT OF USE OR PERFORMANCE OF THE SOFTWARE LIES 43 * WITH YOU. 44 ***********************license end**************************************/ 45 46 #include "zip_crypto.h" 47 48 static void zip_static_init_zip_ops(struct zip_operation *zip_ops, 49 int lzs_flag) 50 { 51 zip_ops->flush = ZIP_FLUSH_FINISH; 52 53 /* equivalent to level 6 of opensource zlib */ 54 zip_ops->speed = 1; 55 56 if (!lzs_flag) { 57 zip_ops->ccode = 0; /* Auto Huffman */ 58 zip_ops->lzs_flag = 0; 59 zip_ops->format = ZLIB_FORMAT; 60 } else { 61 zip_ops->ccode = 3; /* LZS Encoding */ 62 zip_ops->lzs_flag = 1; 63 zip_ops->format = LZS_FORMAT; 64 } 65 zip_ops->begin_file = 1; 66 zip_ops->history_len = 0; 67 zip_ops->end_file = 1; 68 zip_ops->compcode = 0; 69 zip_ops->csum = 1; /* Adler checksum desired */ 70 } 71 72 static int zip_ctx_init(struct zip_kernel_ctx *zip_ctx, int lzs_flag) 73 { 74 struct zip_operation *comp_ctx = &zip_ctx->zip_comp; 75 struct zip_operation *decomp_ctx = &zip_ctx->zip_decomp; 76 77 zip_static_init_zip_ops(comp_ctx, lzs_flag); 78 zip_static_init_zip_ops(decomp_ctx, lzs_flag); 79 80 comp_ctx->input = zip_data_buf_alloc(MAX_INPUT_BUFFER_SIZE); 81 if (!comp_ctx->input) 82 return -ENOMEM; 83 84 comp_ctx->output = zip_data_buf_alloc(MAX_OUTPUT_BUFFER_SIZE); 85 if (!comp_ctx->output) 86 goto err_comp_input; 87 88 decomp_ctx->input = zip_data_buf_alloc(MAX_INPUT_BUFFER_SIZE); 89 if (!decomp_ctx->input) 90 goto err_comp_output; 91 92 decomp_ctx->output = zip_data_buf_alloc(MAX_OUTPUT_BUFFER_SIZE); 93 if (!decomp_ctx->output) 94 goto err_decomp_input; 95 96 return 0; 97 98 err_decomp_input: 99 zip_data_buf_free(decomp_ctx->input, MAX_INPUT_BUFFER_SIZE); 100 101 err_comp_output: 102 zip_data_buf_free(comp_ctx->output, MAX_OUTPUT_BUFFER_SIZE); 103 104 err_comp_input: 105 zip_data_buf_free(comp_ctx->input, MAX_INPUT_BUFFER_SIZE); 106 107 return -ENOMEM; 108 } 109 110 static void zip_ctx_exit(struct zip_kernel_ctx *zip_ctx) 111 { 112 struct zip_operation *comp_ctx = &zip_ctx->zip_comp; 113 struct zip_operation *dec_ctx = &zip_ctx->zip_decomp; 114 115 zip_data_buf_free(comp_ctx->input, MAX_INPUT_BUFFER_SIZE); 116 zip_data_buf_free(comp_ctx->output, MAX_OUTPUT_BUFFER_SIZE); 117 118 zip_data_buf_free(dec_ctx->input, MAX_INPUT_BUFFER_SIZE); 119 zip_data_buf_free(dec_ctx->output, MAX_OUTPUT_BUFFER_SIZE); 120 } 121 122 static int zip_compress(const u8 *src, unsigned int slen, 123 u8 *dst, unsigned int *dlen, 124 struct zip_kernel_ctx *zip_ctx) 125 { 126 struct zip_operation *zip_ops = NULL; 127 struct zip_state *zip_state; 128 struct zip_device *zip = NULL; 129 int ret; 130 131 if (!zip_ctx || !src || !dst || !dlen) 132 return -ENOMEM; 133 134 zip = zip_get_device(zip_get_node_id()); 135 if (!zip) 136 return -ENODEV; 137 138 zip_state = kzalloc(sizeof(*zip_state), GFP_ATOMIC); 139 if (!zip_state) 140 return -ENOMEM; 141 142 zip_ops = &zip_ctx->zip_comp; 143 144 zip_ops->input_len = slen; 145 zip_ops->output_len = *dlen; 146 memcpy(zip_ops->input, src, slen); 147 148 ret = zip_deflate(zip_ops, zip_state, zip); 149 150 if (!ret) { 151 *dlen = zip_ops->output_len; 152 memcpy(dst, zip_ops->output, *dlen); 153 } 154 kfree(zip_state); 155 return ret; 156 } 157 158 static int zip_decompress(const u8 *src, unsigned int slen, 159 u8 *dst, unsigned int *dlen, 160 struct zip_kernel_ctx *zip_ctx) 161 { 162 struct zip_operation *zip_ops = NULL; 163 struct zip_state *zip_state; 164 struct zip_device *zip = NULL; 165 int ret; 166 167 if (!zip_ctx || !src || !dst || !dlen) 168 return -ENOMEM; 169 170 zip = zip_get_device(zip_get_node_id()); 171 if (!zip) 172 return -ENODEV; 173 174 zip_state = kzalloc(sizeof(*zip_state), GFP_ATOMIC); 175 if (!zip_state) 176 return -ENOMEM; 177 178 zip_ops = &zip_ctx->zip_decomp; 179 memcpy(zip_ops->input, src, slen); 180 181 /* Work around for a bug in zlib which needs an extra bytes sometimes */ 182 if (zip_ops->ccode != 3) /* Not LZS Encoding */ 183 zip_ops->input[slen++] = 0; 184 185 zip_ops->input_len = slen; 186 zip_ops->output_len = *dlen; 187 188 ret = zip_inflate(zip_ops, zip_state, zip); 189 190 if (!ret) { 191 *dlen = zip_ops->output_len; 192 memcpy(dst, zip_ops->output, *dlen); 193 } 194 kfree(zip_state); 195 return ret; 196 } 197 198 /* Legacy Compress framework start */ 199 int zip_alloc_comp_ctx_deflate(struct crypto_tfm *tfm) 200 { 201 int ret; 202 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm); 203 204 ret = zip_ctx_init(zip_ctx, 0); 205 206 return ret; 207 } 208 209 int zip_alloc_comp_ctx_lzs(struct crypto_tfm *tfm) 210 { 211 int ret; 212 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm); 213 214 ret = zip_ctx_init(zip_ctx, 1); 215 216 return ret; 217 } 218 219 void zip_free_comp_ctx(struct crypto_tfm *tfm) 220 { 221 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm); 222 223 zip_ctx_exit(zip_ctx); 224 } 225 226 int zip_comp_compress(struct crypto_tfm *tfm, 227 const u8 *src, unsigned int slen, 228 u8 *dst, unsigned int *dlen) 229 { 230 int ret; 231 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm); 232 233 ret = zip_compress(src, slen, dst, dlen, zip_ctx); 234 235 return ret; 236 } 237 238 int zip_comp_decompress(struct crypto_tfm *tfm, 239 const u8 *src, unsigned int slen, 240 u8 *dst, unsigned int *dlen) 241 { 242 int ret; 243 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm); 244 245 ret = zip_decompress(src, slen, dst, dlen, zip_ctx); 246 247 return ret; 248 } /* Legacy compress framework end */ 249 250 /* SCOMP framework start */ 251 void *zip_alloc_scomp_ctx_deflate(struct crypto_scomp *tfm) 252 { 253 int ret; 254 struct zip_kernel_ctx *zip_ctx; 255 256 zip_ctx = kzalloc(sizeof(*zip_ctx), GFP_KERNEL); 257 if (!zip_ctx) 258 return ERR_PTR(-ENOMEM); 259 260 ret = zip_ctx_init(zip_ctx, 0); 261 262 if (ret) { 263 kfree_sensitive(zip_ctx); 264 return ERR_PTR(ret); 265 } 266 267 return zip_ctx; 268 } 269 270 void *zip_alloc_scomp_ctx_lzs(struct crypto_scomp *tfm) 271 { 272 int ret; 273 struct zip_kernel_ctx *zip_ctx; 274 275 zip_ctx = kzalloc(sizeof(*zip_ctx), GFP_KERNEL); 276 if (!zip_ctx) 277 return ERR_PTR(-ENOMEM); 278 279 ret = zip_ctx_init(zip_ctx, 1); 280 281 if (ret) { 282 kfree_sensitive(zip_ctx); 283 return ERR_PTR(ret); 284 } 285 286 return zip_ctx; 287 } 288 289 void zip_free_scomp_ctx(struct crypto_scomp *tfm, void *ctx) 290 { 291 struct zip_kernel_ctx *zip_ctx = ctx; 292 293 zip_ctx_exit(zip_ctx); 294 kfree_sensitive(zip_ctx); 295 } 296 297 int zip_scomp_compress(struct crypto_scomp *tfm, 298 const u8 *src, unsigned int slen, 299 u8 *dst, unsigned int *dlen, void *ctx) 300 { 301 int ret; 302 struct zip_kernel_ctx *zip_ctx = ctx; 303 304 ret = zip_compress(src, slen, dst, dlen, zip_ctx); 305 306 return ret; 307 } 308 309 int zip_scomp_decompress(struct crypto_scomp *tfm, 310 const u8 *src, unsigned int slen, 311 u8 *dst, unsigned int *dlen, void *ctx) 312 { 313 int ret; 314 struct zip_kernel_ctx *zip_ctx = ctx; 315 316 ret = zip_decompress(src, slen, dst, dlen, zip_ctx); 317 318 return ret; 319 } /* SCOMP framework end */ 320