github.com/AESNooper/go/src@v0.0.0-20220218095104-b56a4ab1bbbb/crypto/ed25519/internal/edwards25519/scalar_alias_test.go (about)

     1  // Copyright (c) 2019 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package edwards25519
     6  
     7  import (
     8  	"testing"
     9  	"testing/quick"
    10  )
    11  
    12  func TestScalarAliasing(t *testing.T) {
    13  	checkAliasingOneArg := func(f func(v, x *Scalar) *Scalar, v, x Scalar) bool {
    14  		x1, v1 := x, x
    15  
    16  		// Calculate a reference f(x) without aliasing.
    17  		if out := f(&v, &x); out != &v || !isReduced(out) {
    18  			return false
    19  		}
    20  
    21  		// Test aliasing the argument and the receiver.
    22  		if out := f(&v1, &v1); out != &v1 || v1 != v || !isReduced(out) {
    23  			return false
    24  		}
    25  
    26  		// Ensure the arguments was not modified.
    27  		return x == x1
    28  	}
    29  
    30  	checkAliasingTwoArgs := func(f func(v, x, y *Scalar) *Scalar, v, x, y Scalar) bool {
    31  		x1, y1, v1 := x, y, Scalar{}
    32  
    33  		// Calculate a reference f(x, y) without aliasing.
    34  		if out := f(&v, &x, &y); out != &v || !isReduced(out) {
    35  			return false
    36  		}
    37  
    38  		// Test aliasing the first argument and the receiver.
    39  		v1 = x
    40  		if out := f(&v1, &v1, &y); out != &v1 || v1 != v || !isReduced(out) {
    41  			return false
    42  		}
    43  		// Test aliasing the second argument and the receiver.
    44  		v1 = y
    45  		if out := f(&v1, &x, &v1); out != &v1 || v1 != v || !isReduced(out) {
    46  			return false
    47  		}
    48  
    49  		// Calculate a reference f(x, x) without aliasing.
    50  		if out := f(&v, &x, &x); out != &v || !isReduced(out) {
    51  			return false
    52  		}
    53  
    54  		// Test aliasing the first argument and the receiver.
    55  		v1 = x
    56  		if out := f(&v1, &v1, &x); out != &v1 || v1 != v || !isReduced(out) {
    57  			return false
    58  		}
    59  		// Test aliasing the second argument and the receiver.
    60  		v1 = x
    61  		if out := f(&v1, &x, &v1); out != &v1 || v1 != v || !isReduced(out) {
    62  			return false
    63  		}
    64  		// Test aliasing both arguments and the receiver.
    65  		v1 = x
    66  		if out := f(&v1, &v1, &v1); out != &v1 || v1 != v || !isReduced(out) {
    67  			return false
    68  		}
    69  
    70  		// Ensure the arguments were not modified.
    71  		return x == x1 && y == y1
    72  	}
    73  
    74  	for name, f := range map[string]interface{}{
    75  		"Negate": func(v, x Scalar) bool {
    76  			return checkAliasingOneArg((*Scalar).Negate, v, x)
    77  		},
    78  		"Multiply": func(v, x, y Scalar) bool {
    79  			return checkAliasingTwoArgs((*Scalar).Multiply, v, x, y)
    80  		},
    81  		"Add": func(v, x, y Scalar) bool {
    82  			return checkAliasingTwoArgs((*Scalar).Add, v, x, y)
    83  		},
    84  		"Subtract": func(v, x, y Scalar) bool {
    85  			return checkAliasingTwoArgs((*Scalar).Subtract, v, x, y)
    86  		},
    87  	} {
    88  		err := quick.Check(f, &quick.Config{MaxCountScale: 1 << 5})
    89  		if err != nil {
    90  			t.Errorf("%v: %v", name, err)
    91  		}
    92  	}
    93  }