xref: /linux-6.15/include/linux/rslib.h (revision 21633981)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Generic Reed Solomon encoder / decoder library
4  *
5  * Copyright (C) 2004 Thomas Gleixner ([email protected])
6  *
7  * RS code lifted from reed solomon library written by Phil Karn
8  * Copyright 2002 Phil Karn, KA9Q
9  */
10 #ifndef _RSLIB_H_
11 #define _RSLIB_H_
12 
13 #include <linux/list.h>
14 #include <linux/types.h>	/* for gfp_t */
15 #include <linux/gfp.h>		/* for GFP_KERNEL */
16 
17 /**
18  * struct rs_codec - rs codec data
19  *
20  * @mm:		Bits per symbol
21  * @nn:		Symbols per block (= (1<<mm)-1)
22  * @alpha_to:	log lookup table
23  * @index_of:	Antilog lookup table
24  * @genpoly:	Generator polynomial
25  * @nroots:	Number of generator roots = number of parity symbols
26  * @fcr:	First consecutive root, index form
27  * @prim:	Primitive element, index form
28  * @iprim:	prim-th root of 1, index form
29  * @gfpoly:	The primitive generator polynominal
30  * @gffunc:	Function to generate the field, if non-canonical representation
31  * @users:	Users of this structure
32  * @list:	List entry for the rs codec list
33 */
34 struct rs_codec {
35 	int		mm;
36 	int		nn;
37 	uint16_t	*alpha_to;
38 	uint16_t	*index_of;
39 	uint16_t	*genpoly;
40 	int		nroots;
41 	int		fcr;
42 	int		prim;
43 	int		iprim;
44 	int		gfpoly;
45 	int		(*gffunc)(int);
46 	int		users;
47 	struct list_head list;
48 };
49 
50 /**
51  * struct rs_control - rs control structure per instance
52  * @codec:	The codec used for this instance
53  */
54 struct rs_control {
55 	struct rs_codec	*codec;
56 };
57 
58 /* General purpose RS codec, 8-bit data width, symbol width 1-15 bit  */
59 #ifdef CONFIG_REED_SOLOMON_ENC8
60 int encode_rs8(struct rs_control *rs, uint8_t *data, int len, uint16_t *par,
61 	       uint16_t invmsk);
62 #endif
63 #ifdef CONFIG_REED_SOLOMON_DEC8
64 int decode_rs8(struct rs_control *rs, uint8_t *data, uint16_t *par, int len,
65 		uint16_t *s, int no_eras, int *eras_pos, uint16_t invmsk,
66 	       uint16_t *corr);
67 #endif
68 
69 /* General purpose RS codec, 16-bit data width, symbol width 1-15 bit  */
70 #ifdef CONFIG_REED_SOLOMON_ENC16
71 int encode_rs16(struct rs_control *rs, uint16_t *data, int len, uint16_t *par,
72 		uint16_t invmsk);
73 #endif
74 #ifdef CONFIG_REED_SOLOMON_DEC16
75 int decode_rs16(struct rs_control *rs, uint16_t *data, uint16_t *par, int len,
76 		uint16_t *s, int no_eras, int *eras_pos, uint16_t invmsk,
77 		uint16_t *corr);
78 #endif
79 
80 struct rs_control *init_rs_gfp(int symsize, int gfpoly, int fcr, int prim,
81 			       int nroots, gfp_t gfp);
82 
83 /**
84  * init_rs - Create a RS control struct and initialize it
85  *  @symsize:	the symbol size (number of bits)
86  *  @gfpoly:	the extended Galois field generator polynomial coefficients,
87  *		with the 0th coefficient in the low order bit. The polynomial
88  *		must be primitive;
89  *  @fcr:	the first consecutive root of the rs code generator polynomial
90  *		in index form
91  *  @prim:	primitive element to generate polynomial roots
92  *  @nroots:	RS code generator polynomial degree (number of roots)
93  *
94  * Allocations use GFP_KERNEL.
95  */
96 static inline struct rs_control *init_rs(int symsize, int gfpoly, int fcr,
97 					 int prim, int nroots)
98 {
99 	return init_rs_gfp(symsize, gfpoly, fcr, prim, nroots, GFP_KERNEL);
100 }
101 
102 struct rs_control *init_rs_non_canonical(int symsize, int (*func)(int),
103 					 int fcr, int prim, int nroots);
104 
105 /* Release a rs control structure */
106 void free_rs(struct rs_control *rs);
107 
108 /** modulo replacement for galois field arithmetics
109  *
110  *  @rs:	Pointer to the RS codec
111  *  @x:		the value to reduce
112  *
113  *  where
114  *  rs->mm = number of bits per symbol
115  *  rs->nn = (2^rs->mm) - 1
116  *
117  *  Simple arithmetic modulo would return a wrong result for values
118  *  >= 3 * rs->nn
119 */
120 static inline int rs_modnn(struct rs_codec *rs, int x)
121 {
122 	while (x >= rs->nn) {
123 		x -= rs->nn;
124 		x = (x >> rs->mm) + (x & rs->nn);
125 	}
126 	return x;
127 }
128 
129 #endif
130