linux-imx

/*

 * Cryptographic API.

 *

 * Copyright (c) 2017-present, Facebook, Inc.

 *

 * This program is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 as published by

 * the Free Software Foundation.

 *

 * This program is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for

 * more details.

 */

#include <linux/crypto.h>

#include <linux/init.h>

#include <linux/interrupt.h>

#include <linux/mm.h>

#include <linux/module.h>

#include <linux/net.h>

#include <linux/vmalloc.h>

#include <linux/zstd.h>

#include <crypto/internal/scompress.h>

​

​

#define ZSTD_DEF_LEVEL  3

​

struct zstd_ctx {

    ZSTD_CCtx *cctx;

    ZSTD_DCtx *dctx;

    void *cwksp;

    void *dwksp;

};

​

static ZSTD_parameters zstd_params(void)

{

    return ZSTD_getParams(ZSTD_DEF_LEVEL, 0, 0);

}

​

static int zstd_comp_init(struct zstd_ctx *ctx)

{

    int ret = 0;

    const ZSTD_parameters params = zstd_params();

    const size_t wksp_size = ZSTD_CCtxWorkspaceBound(params.cParams);

​

    ctx->cwksp = vzalloc(wksp_size);

    if (!ctx->cwksp) {

        ret = -ENOMEM;

        goto out;

    }

​

    ctx->cctx = ZSTD_initCCtx(ctx->cwksp, wksp_size);

    if (!ctx->cctx) {

        ret = -EINVAL;

        goto out_free;

    }

out:

    return ret;

out_free:

    vfree(ctx->cwksp);

    goto out;

}

​