Merge pull request #17 from nspcc-dev/fix/refactoring

Remove `unsafe` from code
This commit is contained in:
fyrchik 2019-10-10 12:48:58 +03:00 committed by GitHub
commit b27c17ce19
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 53 additions and 40 deletions

View file

@ -3,28 +3,25 @@ package gf127
import ( import (
"encoding/binary" "encoding/binary"
"encoding/hex" "encoding/hex"
"unsafe"
) )
// GF127x2 represents a pair of elements of GF(2^127) stored together. // GF127x2 represents a pair of elements of GF(2^127) stored together.
type GF127x2 [4]uint64 type GF127x2 [2]GF127
// Split returns 2 components of pair without additional allocations. // Split returns 2 components of pair without additional allocations.
func Split(a *GF127x2) (*GF127, *GF127) { func Split(a *GF127x2) (*GF127, *GF127) {
return (*GF127)(unsafe.Pointer(a)), (*GF127)(unsafe.Pointer(&(*a)[2])) return &a[0], &a[1]
} }
// CombineTo 2 elements of GF(2^127) to the respective components of pair. // CombineTo 2 elements of GF(2^127) to the respective components of pair.
func CombineTo(a *GF127, b *GF127, c *GF127x2) { func CombineTo(a *GF127, b *GF127, c *GF127x2) {
c[0] = a[0] c[0] = *a
c[1] = a[1] c[1] = *b
c[2] = b[0]
c[3] = b[1]
} }
// Equal checks if both elements of GF(2^127) pair are equal. // Equal checks if both elements of GF(2^127) pair are equal.
func (a *GF127x2) Equal(b *GF127x2) bool { func (a *GF127x2) Equal(b *GF127x2) bool {
return a[0] == b[0] && a[1] == b[1] && a[2] == b[2] && a[3] == b[3] return a[0] == b[0] && a[1] == b[1]
} }
// String returns hex-encoded representation, starting with MSB. // String returns hex-encoded representation, starting with MSB.
@ -37,10 +34,10 @@ func (a *GF127x2) String() string {
// ByteArray represents element of GF(2^127) as byte array of length 32. // ByteArray represents element of GF(2^127) as byte array of length 32.
func (a *GF127x2) ByteArray() (buf []byte) { func (a *GF127x2) ByteArray() (buf []byte) {
buf = make([]byte, 32) buf = make([]byte, 32)
binary.BigEndian.PutUint64(buf, a[1]) binary.BigEndian.PutUint64(buf, a[0][1])
binary.BigEndian.PutUint64(buf[8:], a[0]) binary.BigEndian.PutUint64(buf[8:], a[0][0])
binary.BigEndian.PutUint64(buf[16:], a[3]) binary.BigEndian.PutUint64(buf[16:], a[1][1])
binary.BigEndian.PutUint64(buf[24:], a[2]) binary.BigEndian.PutUint64(buf[24:], a[1][0])
return return
} }

View file

@ -11,8 +11,8 @@ var testCasesSplit = []struct {
h1 *GF127 h1 *GF127
h2 *GF127 h2 *GF127
}{ }{
{&GF127x2{123, 31, 141, 9}, &GF127{123, 31}, &GF127{141, 9}}, {&GF127x2{GF127{123, 31}, GF127{141, 9}}, &GF127{123, 31}, &GF127{141, 9}},
{&GF127x2{maxUint64, 0, 0, maxUint64}, &GF127{maxUint64, 0}, &GF127{0, maxUint64}}, {&GF127x2{GF127{maxUint64, 0}, GF127{0, maxUint64}}, &GF127{maxUint64, 0}, &GF127{0, maxUint64}},
} }
func TestSplit(t *testing.T) { func TestSplit(t *testing.T) {
@ -32,9 +32,18 @@ func TestCombineTo(t *testing.T) {
} }
var testCasesMul10x2 = [][2]*GF127x2{ var testCasesMul10x2 = [][2]*GF127x2{
{&GF127x2{123, 0, 123, 0}, &GF127x2{246, 0, 246, 0}}, {
{&GF127x2{maxUint64, 2, 0, 1}, &GF127x2{maxUint64 - 1, 5, 0, 2}}, &GF127x2{GF127{123, 0}, GF127{123, 0}},
{&GF127x2{0, maxUint64 >> 1, maxUint64, 2}, &GF127x2{1 + 1<<63, maxUint64>>1 - 1, maxUint64 - 1, 5}}, &GF127x2{GF127{246, 0}, GF127{246, 0}},
},
{
&GF127x2{GF127{maxUint64, 2}, GF127{0, 1}},
&GF127x2{GF127{maxUint64 - 1, 5}, GF127{0, 2}},
},
{
&GF127x2{GF127{0, maxUint64 >> 1}, GF127{maxUint64, 2}},
&GF127x2{GF127{1 + 1<<63, maxUint64>>1 - 1}, GF127{maxUint64 - 1, 5}},
},
} }
func TestMul10x2(t *testing.T) { func TestMul10x2(t *testing.T) {
@ -46,9 +55,18 @@ func TestMul10x2(t *testing.T) {
} }
var testCasesMul11x2 = [][2]*GF127x2{ var testCasesMul11x2 = [][2]*GF127x2{
{&GF127x2{123, 0, 123, 0}, &GF127x2{141, 0, 141, 0}}, {
{&GF127x2{maxUint64, 2, 0, 1}, &GF127x2{1, 7, 0, 3}}, &GF127x2{GF127{123, 0}, GF127{123, 0}},
{&GF127x2{0, maxUint64 >> 1, maxUint64, 2}, &GF127x2{1 + 1<<63, 1, 1, 7}}, &GF127x2{GF127{141, 0}, GF127{141, 0}},
},
{
&GF127x2{GF127{maxUint64, 2}, GF127{0, 1}},
&GF127x2{GF127{1, 7}, GF127{0, 3}},
},
{
&GF127x2{GF127{0, maxUint64 >> 1}, GF127{maxUint64, 2}},
&GF127x2{GF127{1 + 1<<63, 1}, GF127{1, 7}},
},
} }
func TestMul11x2(t *testing.T) { func TestMul11x2(t *testing.T) {

View file

@ -44,8 +44,8 @@ func (d *digest2) Sum(in []byte) []byte {
return append(in, h[:]...) return append(in, h[:]...)
} }
func (d *digest2) Reset() { func (d *digest2) Reset() {
d.x[0] = gf127.GF127x2{1, 0, 0, 0} d.x[0] = gf127.GF127x2{gf127.GF127{1, 0}, gf127.GF127{0, 0}}
d.x[1] = gf127.GF127x2{0, 0, 1, 0} d.x[1] = gf127.GF127x2{gf127.GF127{0, 0}, gf127.GF127{1, 0}}
} }
func (d *digest2) Size() int { return hashSize } func (d *digest2) Size() int { return hashSize }
func (d *digest2) BlockSize() int { return hashBlockSize } func (d *digest2) BlockSize() int { return hashBlockSize }

View file

@ -38,8 +38,8 @@ func (d *digest3) Sum(in []byte) []byte {
return append(in, h[:]...) return append(in, h[:]...)
} }
func (d *digest3) Reset() { func (d *digest3) Reset() {
d.x[0] = gf127.GF127x2{1, 0, 0, 0} d.x[0] = gf127.GF127x2{gf127.GF127{1, 0}, gf127.GF127{0, 0}}
d.x[1] = gf127.GF127x2{0, 0, 1, 0} d.x[1] = gf127.GF127x2{gf127.GF127{0, 0}, gf127.GF127{1, 0}}
} }
func (d *digest3) Size() int { return hashSize } func (d *digest3) Size() int { return hashSize }
func (d *digest3) BlockSize() int { return hashBlockSize } func (d *digest3) BlockSize() int { return hashBlockSize }

View file

@ -1,5 +1,17 @@
#include "textflag.h" #include "textflag.h"
// mul2 multiplicates FROM by 2, stores result in R1
// and uses R1, R2 and R3 for internal computations.
#define mul2(FROM, TO, R2, R3) \
VPSLLQ $1, FROM, TO \
VPALIGNR $8, TO, FROM, R2 \
PSRLQ $63, R2 \
MOVUPD ·x127x63(SB), R3 \
ANDPD TO, R3 \
VPUNPCKHQDQ R3, R3, R3 \
XORPD R2, TO \
XORPD R3, TO
// func mulBitRight(c00, c01, c10, c11, e *[2]uint64) // func mulBitRight(c00, c01, c10, c11, e *[2]uint64)
TEXT ·mulBitRight(SB),NOSPLIT,$0 TEXT ·mulBitRight(SB),NOSPLIT,$0
MOVQ c00+0(FP), AX MOVQ c00+0(FP), AX
@ -14,14 +26,7 @@ TEXT ·mulBitRight(SB),NOSPLIT,$0
MOVUPD (DX), X3 MOVUPD (DX), X3
// c00 *= 2 // c00 *= 2
VPSLLQ $1, X0, X5 mul2(X0, X5, X6, X7)
VPALIGNR $8, X5, X0, X6
PSRLQ $63, X6
MOVUPD ·x127x63(SB), X7
ANDPD X5, X7
VPUNPCKHQDQ X7, X7, X7
XORPD X6, X5
XORPD X7, X5
MOVUPD X5, X0 MOVUPD X5, X0
// c00 += c01 // c00 += c01
@ -29,14 +34,7 @@ TEXT ·mulBitRight(SB),NOSPLIT,$0
MOVUPD X0, (AX) MOVUPD X0, (AX)
// c10 *= 2 // c10 *= 2
VPSLLQ $1, X2, X5 mul2(X2, X5, X6, X7)
VPALIGNR $8, X5, X2, X6
PSRLQ $63, X6
MOVUPD ·x127x63(SB), X7
ANDPD X5, X7
VPUNPCKHQDQ X7, X7, X7
XORPD X6, X5
XORPD X7, X5
MOVUPD X5, X2 MOVUPD X5, X2
// c10 += c11 // c10 += c11