github.com/zmap/zlint@v1.1.0/lints/lint_dsa_unique_correct_representation_test.go (about)

     1  package lints
     2  
     3  /*
     4   * ZLint Copyright 2018 Regents of the University of Michigan
     5   *
     6   * Licensed under the Apache License, Version 2.0 (the "License"); you may not
     7   * use this file except in compliance with the License. You may obtain a copy
     8   * of the License at http://www.apache.org/licenses/LICENSE-2.0
     9   *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
    13   * implied. See the License for the specific language governing
    14   * permissions and limitations under the License.
    15   */
    16  
    17  import (
    18  	"crypto/dsa"
    19  	"math/big"
    20  	"testing"
    21  )
    22  
    23  func TestDSAUniqueCorrectRepresentation(t *testing.T) {
    24  	inputPath := "../testlint/testCerts/dsaUniqueRep.pem"
    25  	expected := Pass
    26  	out := Lints["e_dsa_unique_correct_representation"].Execute(ReadCertificate(inputPath))
    27  	if out.Status != expected {
    28  		t.Errorf("%s: expected %s, got %s", inputPath, expected, out.Status)
    29  	}
    30  }
    31  
    32  func TestDSANotUniqueCorrectRepresentation(t *testing.T) {
    33  	inputPath := "../testlint/testCerts/dsaUniqueRep.pem"
    34  	c := ReadCertificate(inputPath)
    35  
    36  	// Replace Y with P - 1
    37  	dsaKey := c.PublicKey.(*dsa.PublicKey)
    38  	pMinusOne := big.NewInt(0)
    39  	pMinusOne.Sub(dsaKey.P, big.NewInt(1))
    40  	dsaKey.Y = pMinusOne
    41  
    42  	// Expect failure
    43  	expected := Error
    44  	out := Lints["e_dsa_unique_correct_representation"].Execute(c)
    45  	if out.Status != expected {
    46  		t.Errorf("%s: expected %s, got %s", inputPath, expected, out.Status)
    47  	}
    48  }