Commit 9fec9f8e authored by Mikhail Zaslonko's avatar Mikhail Zaslonko Committed by Andrew Morton
Browse files

lib/zlib: Split deflate and inflate states for DFLTCC 

Currently deflate and inflate both use a common state struct. There are
several variables in this struct that we don't need for inflate, and
more may be coming in the future. Therefore split them in two separate
structs.
Apart from that, introduce separate headers for dfltcc_deflate and
dfltcc_inflate.

This commit is based on:
  https://github.com/zlib-ng/zlib-ng/commit/c592b1b

Link: https://lkml.kernel.org/r/20230126131428.1222214-7-zaslonko@linux.ibm.com


Signed-off-by: default avatarMikhail Zaslonko <zaslonko@linux.ibm.com>
Acked-by: default avatarIlya Leoshkevich <iii@linux.ibm.com>
Cc: Heiko Carstens <hca@linux.ibm.com>
Cc: Vasily Gorbik <gor@linux.ibm.com>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
parent cbf12540

lib/zlib_deflate/deflate.c

+2 −2
Original line number Diff line number Diff line
@@ -54,7 +54,7 @@ 


/* architecture-specific bits */

#ifdef CONFIG_ZLIB_DFLTCC

#  include "../zlib_dfltcc/dfltcc.h"

#  include "../zlib_dfltcc/dfltcc_deflate.h"

#else

#define DEFLATE_RESET_HOOK(strm) do {} while (0)

#define DEFLATE_HOOK(strm, flush, bstate) 0
@@ -106,7 +106,7 @@ typedef struct deflate_workspace { 
    deflate_state deflate_memory;

#ifdef CONFIG_ZLIB_DFLTCC

    /* State memory for s390 hardware deflate */

    struct dfltcc_state dfltcc_memory;

    struct dfltcc_deflate_state dfltcc_memory;

#endif

    Byte *window_memory;

    Pos *prev_memory;

lib/zlib_dfltcc/dfltcc.c

+3 −22
Original line number Diff line number Diff line
@@ -23,37 +23,18 @@ char *oesc_msg( 
    }

}



void dfltcc_reset(

    z_streamp strm,

    uInt size

)

{

    struct dfltcc_state *dfltcc_state =

        (struct dfltcc_state *)((char *)strm->state + size);

    struct dfltcc_qaf_param *param =

        (struct dfltcc_qaf_param *)&dfltcc_state->param;



void dfltcc_reset_state(struct dfltcc_state *dfltcc_state) {

    /* Initialize available functions */

    if (is_dfltcc_enabled()) {

        dfltcc(DFLTCC_QAF, param, NULL, NULL, NULL, NULL, NULL);

        memmove(&dfltcc_state->af, param, sizeof(dfltcc_state->af));

        dfltcc(DFLTCC_QAF, &dfltcc_state->param, NULL, NULL, NULL, NULL, NULL);

        memmove(&dfltcc_state->af, &dfltcc_state->param, sizeof(dfltcc_state->af));

    } else

        memset(&dfltcc_state->af, 0, sizeof(dfltcc_state->af));



    /* Initialize parameter block */

    memset(&dfltcc_state->param, 0, sizeof(dfltcc_state->param));

    dfltcc_state->param.nt = 1;



    /* Initialize tuning parameters */

    if (zlib_dfltcc_support == ZLIB_DFLTCC_FULL_DEBUG)

        dfltcc_state->level_mask = DFLTCC_LEVEL_MASK_DEBUG;

    else

        dfltcc_state->level_mask = DFLTCC_LEVEL_MASK;

    dfltcc_state->block_size = DFLTCC_BLOCK_SIZE;

    dfltcc_state->block_threshold = DFLTCC_FIRST_FHT_BLOCK_SIZE;

    dfltcc_state->dht_threshold = DFLTCC_DHT_MIN_SAMPLE_SIZE;

    dfltcc_state->param.ribm = DFLTCC_RIBM;

}

EXPORT_SYMBOL(dfltcc_reset);



MODULE_LICENSE("GPL");

lib/zlib_dfltcc/dfltcc.h

+10 −42
Original line number Diff line number Diff line
@@ -93,64 +93,32 @@ static_assert(sizeof(struct dfltcc_param_v0) == 1536); 
struct dfltcc_state {

    struct dfltcc_param_v0 param;      /* Parameter block */

    struct dfltcc_qaf_param af;        /* Available functions */

    char msg[64];                      /* Buffer for strm->msg */

};



/*

 *  Extension of inflate_state and deflate_state for DFLTCC.

 */

struct dfltcc_deflate_state {

    struct dfltcc_state common;        /* Parameter block */

    uLong level_mask;                  /* Levels on which to use DFLTCC */

    uLong block_size;                  /* New block each X bytes */

    uLong block_threshold;             /* New block after total_in > X */

    uLong dht_threshold;               /* New block only if avail_in >= X */

    char msg[64];                      /* Buffer for strm->msg */

};



#define ALIGN_UP(p, size) (__typeof__(p))(((uintptr_t)(p) + ((size) - 1)) & ~((size) - 1))

/* Resides right after inflate_state or deflate_state */

#define GET_DFLTCC_STATE(state) ((struct dfltcc_state *)((char *)(state) + ALIGN_UP(sizeof(*state), 8)))



/* External functions */

int dfltcc_can_deflate(z_streamp strm);

int dfltcc_deflate(z_streamp strm,

                   int flush,

                   block_state *result);

void dfltcc_reset(z_streamp strm, uInt size);

int dfltcc_can_inflate(z_streamp strm);

typedef enum {

    DFLTCC_INFLATE_CONTINUE,

    DFLTCC_INFLATE_BREAK,

    DFLTCC_INFLATE_SOFTWARE,

} dfltcc_inflate_action;

dfltcc_inflate_action dfltcc_inflate(z_streamp strm,

                                     int flush, int *ret);

void dfltcc_reset_state(struct dfltcc_state *dfltcc_state);



static inline int is_dfltcc_enabled(void)

{

return (zlib_dfltcc_support != ZLIB_DFLTCC_DISABLED &&

        test_facility(DFLTCC_FACILITY));

}



#define DEFLATE_RESET_HOOK(strm) \

    dfltcc_reset((strm), sizeof(deflate_state))



#define DEFLATE_HOOK dfltcc_deflate



#define DEFLATE_NEED_CHECKSUM(strm) (!dfltcc_can_deflate((strm)))



#define DEFLATE_DFLTCC_ENABLED() is_dfltcc_enabled()



#define INFLATE_RESET_HOOK(strm) \

    dfltcc_reset((strm), sizeof(struct inflate_state))



#define INFLATE_TYPEDO_HOOK(strm, flush) \

    if (dfltcc_can_inflate((strm))) { \

        dfltcc_inflate_action action; \

\

        RESTORE(); \

        action = dfltcc_inflate((strm), (flush), &ret); \

        LOAD(); \

        if (action == DFLTCC_INFLATE_CONTINUE) \

            break; \

        else if (action == DFLTCC_INFLATE_BREAK) \

            goto inf_leave; \

    }



#define INFLATE_NEED_CHECKSUM(strm) (!dfltcc_can_inflate((strm)))



#define INFLATE_NEED_UPDATEWINDOW(strm) (!dfltcc_can_inflate((strm)))



#endif /* DFLTCC_H */

lib/zlib_dfltcc/dfltcc_deflate.c

+27 −8
Original line number Diff line number Diff line
@@ -2,11 +2,13 @@ 


#include "../zlib_deflate/defutil.h"

#include "dfltcc_util.h"

#include "dfltcc.h"

#include "dfltcc_deflate.h"

#include <asm/setup.h>

#include <linux/export.h>

#include <linux/zutil.h>



#define GET_DFLTCC_DEFLATE_STATE(state) ((struct dfltcc_deflate_state *)GET_DFLTCC_STATE(state))



/*

 * Compress.

 */
@@ -15,7 +17,7 @@ int dfltcc_can_deflate( 
)

{

    deflate_state *state = (deflate_state *)strm->state;

    struct dfltcc_state *dfltcc_state = GET_DFLTCC_STATE(state);

    struct dfltcc_deflate_state *dfltcc_state = GET_DFLTCC_DEFLATE_STATE(state);



    /* Check for kernel dfltcc command line parameter */

    if (zlib_dfltcc_support == ZLIB_DFLTCC_DISABLED ||
@@ -28,15 +30,32 @@ int dfltcc_can_deflate( 
        return 0;



    /* Unsupported hardware */

    if (!is_bit_set(dfltcc_state->af.fns, DFLTCC_GDHT) ||

            !is_bit_set(dfltcc_state->af.fns, DFLTCC_CMPR) ||

            !is_bit_set(dfltcc_state->af.fmts, DFLTCC_FMT0))

    if (!is_bit_set(dfltcc_state->common.af.fns, DFLTCC_GDHT) ||

            !is_bit_set(dfltcc_state->common.af.fns, DFLTCC_CMPR) ||

            !is_bit_set(dfltcc_state->common.af.fmts, DFLTCC_FMT0))

        return 0;



    return 1;

}

EXPORT_SYMBOL(dfltcc_can_deflate);



void dfltcc_reset_deflate_state(z_streamp strm) {

    deflate_state *state = (deflate_state *)strm->state;

    struct dfltcc_deflate_state *dfltcc_state = GET_DFLTCC_DEFLATE_STATE(state);



    dfltcc_reset_state(&dfltcc_state->common);



    /* Initialize tuning parameters */

    if (zlib_dfltcc_support == ZLIB_DFLTCC_FULL_DEBUG)

        dfltcc_state->level_mask = DFLTCC_LEVEL_MASK_DEBUG;

    else

        dfltcc_state->level_mask = DFLTCC_LEVEL_MASK;

    dfltcc_state->block_size = DFLTCC_BLOCK_SIZE;

    dfltcc_state->block_threshold = DFLTCC_FIRST_FHT_BLOCK_SIZE;

    dfltcc_state->dht_threshold = DFLTCC_DHT_MIN_SAMPLE_SIZE;

}

EXPORT_SYMBOL(dfltcc_reset_deflate_state);



static void dfltcc_gdht(

    z_streamp strm

)
@@ -104,8 +123,8 @@ int dfltcc_deflate( 
)

{

    deflate_state *state = (deflate_state *)strm->state;

    struct dfltcc_state *dfltcc_state = GET_DFLTCC_STATE(state);

    struct dfltcc_param_v0 *param = &dfltcc_state->param;

    struct dfltcc_deflate_state *dfltcc_state = GET_DFLTCC_DEFLATE_STATE(state);

    struct dfltcc_param_v0 *param = &dfltcc_state->common.param;

    uInt masked_avail_in;

    dfltcc_cc cc;

    int need_empty_block;
@@ -244,7 +263,7 @@ int dfltcc_deflate( 
    } while (cc == DFLTCC_CC_AGAIN);



    /* Translate parameter block to stream */

    strm->msg = oesc_msg(dfltcc_state->msg, param->oesc);

    strm->msg = oesc_msg(dfltcc_state->common.msg, param->oesc);

    state->bi_valid = param->sbb;

    if (state->bi_valid == 0)

        state->bi_buf = 0; /* Avoid accessing next_out */

lib/zlib_dfltcc/dfltcc_deflate.h

0 → 100644
+21 −0
Original line number Diff line number Diff line 
// SPDX-License-Identifier: Zlib

#ifndef DFLTCC_DEFLATE_H

#define DFLTCC_DEFLATE_H



#include "dfltcc.h"



/* External functions */

int dfltcc_can_deflate(z_streamp strm);

int dfltcc_deflate(z_streamp strm,

                   int flush,

                   block_state *result);

void dfltcc_reset_deflate_state(z_streamp strm);



#define DEFLATE_RESET_HOOK(strm) \

    dfltcc_reset_deflate_state((strm))



#define DEFLATE_HOOK dfltcc_deflate



#define DEFLATE_NEED_CHECKSUM(strm) (!dfltcc_can_deflate((strm)))



#endif /* DFLTCC_DEFLATE_H */