authorgravatar for 124872+jedisct1@users.noreply.github.comFrank Denis <124872+jedisct1@users.noreply.github.com> 2020-10-10 21:12:24+02:00
committergravatar for 124872+jedisct1@users.noreply.github.comFrank Denis <124872+jedisct1@users.noreply.github.com> 2020-10-10 22:45:41+02:00
log9f109ba0ebb4167a18a80d8ff0c173d574fc4577
tree6c5773c966f0ca5de89a1d0edc9cc79c2b9fc3d5
parent459128e05902008937205ea154e6c556e9264a4d

Simpler ChaCha20 vector code

1 files changed, 18 insertions(+), 33 deletions(-)

lib/std/crypto/chacha20.zig+18-33
......@@ -34,67 +34,52 @@ const ChaCha20VecImpl = struct {
3434 };
3535 }
3636
37 inline fn chacha20Core(x: *BlockVec, input: BlockVec) void {
38 const rot8 = [_]i32{ 3, 0, 1, 2, 7, 4, 5, 6, 11, 8, 9, 10, 15, 12, 13, 14 };
39 const rot16 = [_]i32{ 2, 3, 0, 1, 6, 7, 4, 5, 10, 11, 8, 9, 14, 15, 12, 13 };
37 inline fn rot(x: Lane, comptime n: comptime_int) Lane {
38 return (x << @splat(4, @as(u5, n))) | (x >> @splat(4, @as(u5, 32 - n)));
39 }
4040
41 inline fn chacha20Core(x: *BlockVec, input: BlockVec) void {
4142 x.* = input;
4243
4344 var r: usize = 0;
4445 while (r < 20) : (r += 2) {
4546 x[0] +%= x[1];
4647 x[3] ^= x[0];
47 x[3] = @bitCast(Vector(4, u32), @shuffle(u8, @bitCast(Vector(16, u8), x[3]), undefined, rot16));
48 x[3] = rot(x[3], 16);
4849
4950 x[2] +%= x[3];
5051 x[1] ^= x[2];
51
52 var t1 = x[1];
53 x[1] <<= @splat(4, @as(u5, 12));
54 t1 >>= @splat(4, @as(u5, 20));
55 x[1] ^= t1;
52 x[1] = rot(x[1], 12);
5653
5754 x[0] +%= x[1];
5855 x[3] ^= x[0];
59 x[0] = @shuffle(u32, x[0], undefined, Vector(4, i32){ 3, 0, 1, 2 });
60 x[3] = @bitCast(Vector(4, u32), @shuffle(u8, @bitCast(Vector(16, u8), x[3]), undefined, rot8));
56 x[0] = @shuffle(u32, x[0], undefined, [_]i32{ 3, 0, 1, 2 });
57 x[3] = rot(x[3], 8);
6158
6259 x[2] +%= x[3];
63 x[3] = @shuffle(u32, x[3], undefined, Vector(4, i32){ 2, 3, 0, 1 });
60 x[3] = @shuffle(u32, x[3], undefined, [_]i32{ 2, 3, 0, 1 });
6461 x[1] ^= x[2];
65 x[2] = @shuffle(u32, x[2], undefined, Vector(4, i32){ 1, 2, 3, 0 });
66
67 t1 = x[1];
68 x[1] <<= @splat(4, @as(u5, 7));
69 t1 >>= @splat(4, @as(u5, 25));
70 x[1] ^= t1;
62 x[2] = @shuffle(u32, x[2], undefined, [_]i32{ 1, 2, 3, 0 });
63 x[1] = rot(x[1], 7);
7164
7265 x[0] +%= x[1];
7366 x[3] ^= x[0];
74 x[3] = @bitCast(Vector(4, u32), @shuffle(u8, @bitCast(Vector(16, u8), x[3]), undefined, rot16));
67 x[3] = rot(x[3], 16);
7568
7669 x[2] +%= x[3];
7770 x[1] ^= x[2];
78
79 t1 = x[1];
80 x[1] <<= @splat(4, @as(u5, 12));
81 t1 >>= @splat(4, @as(u5, 20));
82 x[1] ^= t1;
71 x[1] = rot(x[1], 12);
8372
8473 x[0] +%= x[1];
8574 x[3] ^= x[0];
86 x[0] = @shuffle(u32, x[0], undefined, Vector(4, i32){ 1, 2, 3, 0 });
87 x[3] = @bitCast(Vector(4, u32), @shuffle(u8, @bitCast(Vector(16, u8), x[3]), undefined, rot8));
75 x[0] = @shuffle(u32, x[0], undefined, [_]i32{ 1, 2, 3, 0 });
76 x[3] = rot(x[3], 8);
8877
8978 x[2] +%= x[3];
90 x[3] = @shuffle(u32, x[3], undefined, Vector(4, i32){ 2, 3, 0, 1 });
79 x[3] = @shuffle(u32, x[3], undefined, [_]i32{ 2, 3, 0, 1 });
9180 x[1] ^= x[2];
92 x[2] = @shuffle(u32, x[2], undefined, Vector(4, i32){ 3, 0, 1, 2 });
93
94 t1 = x[1];
95 x[1] <<= @splat(4, @as(u5, 7));
96 t1 >>= @splat(4, @as(u5, 25));
97 x[1] ^= t1;
81 x[2] = @shuffle(u32, x[2], undefined, [_]i32{ 3, 0, 1, 2 });
82 x[1] = rot(x[1], 7);
9883 }
9984 }
10085