summary refs log tree commit diff
path: root/crypto/src/pqc/crypto/saber/SaberUtilities.cs
blob: d25eb8d2dfc0cd1d6429e58299e7fa4a27ad1fe1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
namespace Org.BouncyCastle.Pqc.Crypto.Saber
{
    internal class SaberUtilities
    {
        private readonly int SABER_N;
        private readonly int SABER_L;
        private readonly int SABER_ET;
        private readonly int SABER_POLYBYTES;
        private readonly int SABER_EP;
        private readonly int SABER_KEYBYTES;

        internal SaberUtilities(SaberEngine engine)
        {
            this.SABER_N = engine.N;
            this.SABER_L = engine.L;
            this.SABER_ET = engine.ET;
            this.SABER_POLYBYTES = engine.PolyBytes;
            this.SABER_EP = engine.EP;
            this.SABER_KEYBYTES = engine.KeyBytes;
        }

        public void POLT2BS(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            if (SABER_ET == 3)
            {
                for (j = 0; j < SABER_N / 8; j++)
                {
                    offset_byte = (short) (3 * j);
                    offset_data = (short) (8 * j);
                    bytes[byteIndex + offset_byte + 0] = (byte) ((data[offset_data + 0] & 0x7) |
                                                                 ((data[offset_data + 1] & 0x7) << 3) |
                                                                 ((data[offset_data + 2] & 0x3) << 6));
                    bytes[byteIndex + offset_byte + 1] = (byte) (((data[offset_data + 2] >> 2) & 0x01) |
                                                                 ((data[offset_data + 3] & 0x7) << 1) |
                                                                 ((data[offset_data + 4] & 0x7) << 4) |
                                                                 (((data[offset_data + 5]) & 0x01) << 7));
                    bytes[byteIndex + offset_byte + 2] = (byte) (((data[offset_data + 5] >> 1) & 0x03) |
                                                                 ((data[offset_data + 6] & 0x7) << 2) |
                                                                 ((data[offset_data + 7] & 0x7) << 5));
                }
            }
            else if (SABER_ET == 4)
            {
                for (j = 0; j < SABER_N / 2; j++)
                {
                    offset_byte = j;
                    offset_data = (short) (2 * j);
                    bytes[byteIndex + offset_byte] =
                        (byte) ((data[offset_data] & 0x0f) | ((data[offset_data + 1] & 0x0f) << 4));
                }
            }
            else if (SABER_ET == 6)
            {
                for (j = 0; j < SABER_N / 4; j++)
                {
                    offset_byte = (short) (3 * j);
                    offset_data = (short) (4 * j);
                    bytes[byteIndex + offset_byte + 0] =
                        (byte) ((data[offset_data + 0] & 0x3f) | ((data[offset_data + 1] & 0x03) << 6));
                    bytes[byteIndex + offset_byte + 1] = (byte) (((data[offset_data + 1] >> 2) & 0x0f) |
                                                                 ((data[offset_data + 2] & 0x0f) << 4));
                    bytes[byteIndex + offset_byte + 2] = (byte) (((data[offset_data + 2] >> 4) & 0x03) |
                                                                 ((data[offset_data + 3] & 0x3f) << 2));
                }
            }
        }

        public void BS2POLT(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            if (SABER_ET == 3)
            {
                for (j = 0; j < SABER_N / 8; j++)
                {
                    offset_byte = (short) (3 * j);
                    offset_data = (short) (8 * j);
                    data[offset_data + 0] = (short) ((bytes[byteIndex + offset_byte + 0]) & 0x07);
                    data[offset_data + 1] = (short) (((bytes[byteIndex + offset_byte + 0]) >> 3) & 0x07);
                    data[offset_data + 2] = (short) ((((bytes[byteIndex + offset_byte + 0]) >> 6) & 0x03) |
                                                     (((bytes[byteIndex + offset_byte + 1]) & 0x01) << 2));
                    data[offset_data + 3] = (short) (((bytes[byteIndex + offset_byte + 1]) >> 1) & 0x07);
                    data[offset_data + 4] = (short) (((bytes[byteIndex + offset_byte + 1]) >> 4) & 0x07);
                    data[offset_data + 5] = (short) ((((bytes[byteIndex + offset_byte + 1]) >> 7) & 0x01) |
                                                     (((bytes[byteIndex + offset_byte + 2]) & 0x03) << 1));
                    data[offset_data + 6] = (short) ((bytes[byteIndex + offset_byte + 2] >> 2) & 0x07);
                    data[offset_data + 7] = (short) ((bytes[byteIndex + offset_byte + 2] >> 5) & 0x07);
                }
            }
            else if (SABER_ET == 4)
            {
                for (j = 0; j < SABER_N / 2; j++)
                {
                    offset_byte = j;
                    offset_data = (short) (2 * j);
                    data[offset_data] = (short) (bytes[byteIndex + offset_byte] & 0x0f);
                    data[offset_data + 1] = (short) ((bytes[byteIndex + offset_byte] >> 4) & 0x0f);
                }
            }
            else if (SABER_ET == 6)
            {
                for (j = 0; j < SABER_N / 4; j++)
                {
                    offset_byte = (short) (3 * j);
                    offset_data = (short) (4 * j);
                    data[offset_data + 0] = (short) (bytes[byteIndex + offset_byte + 0] & 0x3f);
                    data[offset_data + 1] = (short) (((bytes[byteIndex + offset_byte + 0] >> 6) & 0x03) |
                                                     ((bytes[byteIndex + offset_byte + 1] & 0x0f) << 2));
                    data[offset_data + 2] = (short) (((bytes[byteIndex + offset_byte + 1] & 0xff) >> 4) |
                                                     ((bytes[byteIndex + offset_byte + 2] & 0x03) << 4));
                    data[offset_data + 3] = (short) ((bytes[byteIndex + offset_byte + 2] & 0xff) >> 2);
                }
            }

        }

        private void POLq2BS(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            for (j = 0; j < SABER_N / 8; j++)
            {
                offset_byte = (short) (13 * j);
                offset_data = (short) (8 * j);
                bytes[byteIndex + offset_byte + 0] = (byte) (data[offset_data + 0] & (0xff));
                bytes[byteIndex + offset_byte + 1] =
                    (byte) (((data[offset_data + 0] >> 8) & 0x1f) | ((data[offset_data + 1] & 0x07) << 5));
                bytes[byteIndex + offset_byte + 2] = (byte) ((data[offset_data + 1] >> 3) & 0xff);
                bytes[byteIndex + offset_byte + 3] =
                    (byte) (((data[offset_data + 1] >> 11) & 0x03) | ((data[offset_data + 2] & 0x3f) << 2));
                bytes[byteIndex + offset_byte + 4] =
                    (byte) (((data[offset_data + 2] >> 6) & 0x7f) | ((data[offset_data + 3] & 0x01) << 7));
                bytes[byteIndex + offset_byte + 5] = (byte) ((data[offset_data + 3] >> 1) & 0xff);
                bytes[byteIndex + offset_byte + 6] =
                    (byte) (((data[offset_data + 3] >> 9) & 0x0f) | ((data[offset_data + 4] & 0x0f) << 4));
                bytes[byteIndex + offset_byte + 7] = (byte) ((data[offset_data + 4] >> 4) & 0xff);
                bytes[byteIndex + offset_byte + 8] =
                    (byte) (((data[offset_data + 4] >> 12) & 0x01) | ((data[offset_data + 5] & 0x7f) << 1));
                bytes[byteIndex + offset_byte + 9] =
                    (byte) (((data[offset_data + 5] >> 7) & 0x3f) | ((data[offset_data + 6] & 0x03) << 6));
                bytes[byteIndex + offset_byte + 10] = (byte) ((data[offset_data + 6] >> 2) & 0xff);
                bytes[byteIndex + offset_byte + 11] =
                    (byte) (((data[offset_data + 6] >> 10) & 0x07) | ((data[offset_data + 7] & 0x1f) << 3));
                bytes[byteIndex + offset_byte + 12] = (byte) ((data[offset_data + 7] >> 5) & 0xff);
            }
        }

        private void BS2POLq(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            for (j = 0; j < SABER_N / 8; j++)
            {
                offset_byte = (short) (13 * j);
                offset_data = (short) (8 * j);
                data[offset_data + 0] = (short) ((bytes[byteIndex + offset_byte + 0] & (0xff)) |
                                                 ((bytes[byteIndex + offset_byte + 1] & 0x1f) << 8));
                data[offset_data + 1] = (short) ((bytes[byteIndex + offset_byte + 1] >> 5 & (0x07)) |
                                                 ((bytes[byteIndex + offset_byte + 2] & 0xff) << 3) |
                                                 ((bytes[byteIndex + offset_byte + 3] & 0x03) << 11));
                data[offset_data + 2] = (short) ((bytes[byteIndex + offset_byte + 3] >> 2 & (0x3f)) |
                                                 ((bytes[byteIndex + offset_byte + 4] & 0x7f) << 6));
                data[offset_data + 3] = (short) ((bytes[byteIndex + offset_byte + 4] >> 7 & (0x01)) |
                                                 ((bytes[byteIndex + offset_byte + 5] & 0xff) << 1) |
                                                 ((bytes[byteIndex + offset_byte + 6] & 0x0f) << 9));
                data[offset_data + 4] = (short) ((bytes[byteIndex + offset_byte + 6] >> 4 & (0x0f)) |
                                                 ((bytes[byteIndex + offset_byte + 7] & 0xff) << 4) |
                                                 ((bytes[byteIndex + offset_byte + 8] & 0x01) << 12));
                data[offset_data + 5] = (short) ((bytes[byteIndex + offset_byte + 8] >> 1 & (0x7f)) |
                                                 ((bytes[byteIndex + offset_byte + 9] & 0x3f) << 7));
                data[offset_data + 6] = (short) ((bytes[byteIndex + offset_byte + 9] >> 6 & (0x03)) |
                                                 ((bytes[byteIndex + offset_byte + 10] & 0xff) << 2) |
                                                 ((bytes[byteIndex + offset_byte + 11] & 0x07) << 10));
                data[offset_data + 7] = (short) ((bytes[byteIndex + offset_byte + 11] >> 3 & (0x1f)) |
                                                 ((bytes[byteIndex + offset_byte + 12] & 0xff) << 5));
            }
        }

        private void POLp2BS(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            for (j = 0; j < SABER_N / 4; j++)
            {
                offset_byte = (short) (5 * j);
                offset_data = (short) (4 * j);
                bytes[byteIndex + offset_byte + 0] = (byte) (data[offset_data + 0] & (0xff));
                bytes[byteIndex + offset_byte + 1] =
                    (byte) (((data[offset_data + 0] >> 8) & 0x03) | ((data[offset_data + 1] & 0x3f) << 2));
                bytes[byteIndex + offset_byte + 2] =
                    (byte) (((data[offset_data + 1] >> 6) & 0x0f) | ((data[offset_data + 2] & 0x0f) << 4));
                bytes[byteIndex + offset_byte + 3] =
                    (byte) (((data[offset_data + 2] >> 4) & 0x3f) | ((data[offset_data + 3] & 0x03) << 6));
                bytes[byteIndex + offset_byte + 4] = (byte) ((data[offset_data + 3] >> 2) & 0xff);
            }
        }

        public void BS2POLp(byte[] bytes, int byteIndex, short[] data)
        {
            short j, offset_byte, offset_data;
            for (j = 0; j < SABER_N / 4; j++)
            {
                offset_byte = (short) (5 * j);
                offset_data = (short) (4 * j);
                data[offset_data + 0] = (short) ((bytes[byteIndex + offset_byte + 0] & (0xff)) |
                                                 ((bytes[byteIndex + offset_byte + 1] & 0x03) << 8));
                data[offset_data + 1] = (short) (((bytes[byteIndex + offset_byte + 1] >> 2) & (0x3f)) |
                                                 ((bytes[byteIndex + offset_byte + 2] & 0x0f) << 6));
                data[offset_data + 2] = (short) (((bytes[byteIndex + offset_byte + 2] >> 4) & (0x0f)) |
                                                 ((bytes[byteIndex + offset_byte + 3] & 0x3f) << 4));
                data[offset_data + 3] = (short) (((bytes[byteIndex + offset_byte + 3] >> 6) & (0x03)) |
                                                 ((bytes[byteIndex + offset_byte + 4] & 0xff) << 2));
            }
        }

        public void POLVECq2BS(byte[] bytes, short[][] data)
        {
            byte i;
            for (i = 0; i < SABER_L; i++)
            {
                POLq2BS(bytes, i * SABER_POLYBYTES, data[i]);
            }
        }

        public void BS2POLVECq(byte[] bytes, int byteIndex, short[][] data)
        {
            byte i;
            for (i = 0; i < SABER_L; i++)
            {
                BS2POLq(bytes, byteIndex + i * SABER_POLYBYTES, data[i]);
            }
        }

        public void POLVECp2BS(byte[] bytes, short[][] data)
        {
            byte i;
            for (i = 0; i < SABER_L; i++)
            {
                POLp2BS(bytes, i * (SABER_EP * SABER_N / 8), data[i]);
            }
        }

        public void BS2POLVECp(byte[] bytes, short[][] data)
        {
            byte i;
            for (i = 0; i < SABER_L; i++)
            {
                BS2POLp(bytes, i * (SABER_EP * SABER_N / 8), data[i]);
            }
        }

        public void BS2POLmsg(byte[] bytes, short[] data)
        {
            byte i, j;
            for (j = 0; j < SABER_KEYBYTES; j++)
            {
                for (i = 0; i < 8; i++)
                {
                    data[j * 8 + i] = (short) ((bytes[j] >> i) & 0x01);
                }
            }
        }

        public void POLmsg2BS(byte[] bytes, short[] data)
        {
            byte i, j;
            for (j = 0; j < SABER_KEYBYTES; j++)
            {
                for (i = 0; i < 8; i++)
                {
                    bytes[j] = (byte) (bytes[j] | ((data[j * 8 + i] & 0x01) << i));
                }
            }
        }
    }
}