Add basic RAID[56] support
[platform/upstream/btrfs-progs.git] / raid6.c
1 /* -*- linux-c -*- ------------------------------------------------------- *
2  *
3  *   Copyright 2002-2004 H. Peter Anvin - All Rights Reserved
4  *
5  *   This program is free software; you can redistribute it and/or modify
6  *   it under the terms of the GNU General Public License as published by
7  *   the Free Software Foundation, Inc., 53 Temple Place Ste 330,
8  *   Boston MA 02111-1307, USA; either version 2 of the License, or
9  *   (at your option) any later version; incorporated herein by reference.
10  *
11  * ----------------------------------------------------------------------- */
12
13 /*
14  * raid6int1.c
15  *
16  * 1-way unrolled portable integer math RAID-6 instruction set
17  *
18  * This file was postprocessed using unroll.pl and then ported to userspace
19  */
20 #include <stdint.h>
21 #include <unistd.h>
22 /*
23  * This is the C data type to use
24  */
25
26 /* Change this from BITS_PER_LONG if there is something better... */
27 #if BITS_PER_LONG == 64
28 # define NBYTES(x) ((x) * 0x0101010101010101UL)
29 # define NSIZE  8
30 # define NSHIFT 3
31 typedef uint64_t unative_t;
32 #else
33 # define NBYTES(x) ((x) * 0x01010101U)
34 # define NSIZE  4
35 # define NSHIFT 2
36 typedef uint32_t unative_t;
37 #endif
38
39 /*
40  * These sub-operations are separate inlines since they can sometimes be
41  * specially optimized using architecture-specific hacks.
42  */
43
44 /*
45  * The SHLBYTE() operation shifts each byte left by 1, *not*
46  * rolling over into the next byte
47  */
48 static inline __attribute_const__ unative_t SHLBYTE(unative_t v)
49 {
50         unative_t vv;
51
52         vv = (v << 1) & NBYTES(0xfe);
53         return vv;
54 }
55
56 /*
57  * The MASK() operation returns 0xFF in any byte for which the high
58  * bit is 1, 0x00 for any byte for which the high bit is 0.
59  */
60 static inline __attribute_const__ unative_t MASK(unative_t v)
61 {
62         unative_t vv;
63
64         vv = v & NBYTES(0x80);
65         vv = (vv << 1) - (vv >> 7); /* Overflow on the top bit is OK */
66         return vv;
67 }
68
69
70 void raid6_gen_syndrome(int disks, size_t bytes, void **ptrs)
71 {
72         uint8_t **dptr = (uint8_t **)ptrs;
73         uint8_t *p, *q;
74         int d, z, z0;
75
76         unative_t wd0, wq0, wp0, w10, w20;
77
78         z0 = disks - 3;         /* Highest data disk */
79         p = dptr[z0+1];         /* XOR parity */
80         q = dptr[z0+2];         /* RS syndrome */
81
82         for ( d = 0 ; d < bytes ; d += NSIZE*1 ) {
83                 wq0 = wp0 = *(unative_t *)&dptr[z0][d+0*NSIZE];
84                 for ( z = z0-1 ; z >= 0 ; z-- ) {
85                         wd0 = *(unative_t *)&dptr[z][d+0*NSIZE];
86                         wp0 ^= wd0;
87                         w20 = MASK(wq0);
88                         w10 = SHLBYTE(wq0);
89                         w20 &= NBYTES(0x1d);
90                         w10 ^= w20;
91                         wq0 = w10 ^ wd0;
92                 }
93                 *(unative_t *)&p[d+NSIZE*0] = wp0;
94                 *(unative_t *)&q[d+NSIZE*0] = wq0;
95         }
96 }
97