drivers/crypto/cavium/zip/zip_crypto.c

   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 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 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 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 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                 kzfree(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                 kzfree(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         kzfree(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 */