github.com/hxx258456/fabric-ca-gm@v0.0.3-0.20221111064038-a268ad7e3a37/lib/client/credential/x509/credential_test.go

/*
Copyright IBM Corp. All Rights Reserved.

SPDX-License-Identifier: Apache-2.0
*/

package x509_test

import (
	"bytes"
	"encoding/hex"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"

	http "github.com/hxx258456/ccgo/gmhttp"

	"github.com/hxx258456/ccgo/x509"

	"github.com/hxx258456/fabric-ca-gm/internal/pkg/api"
	"github.com/hxx258456/fabric-ca-gm/internal/pkg/util"
	"github.com/hxx258456/fabric-ca-gm/lib"
	"github.com/hxx258456/fabric-ca-gm/lib/client/credential"
	. "github.com/hxx258456/fabric-ca-gm/lib/client/credential/x509"
	"github.com/hxx258456/fabric-ca-gm/lib/client/credential/x509/mocks"
	"github.com/hxx258456/fabric-gm/bccsp/factory"

	// "github.com/hxx258456/ccgo/x509"
	"github.com/stretchr/testify/assert"
)

const (
	testDataDir = "../../../../testdata"
)

func TestX509Credential(t *testing.T) {
	clientHome, err := ioutil.TempDir(testDataDir, "x509credtest")
	if err != nil {
		t.Fatalf("Failed to create temp directory: %s", err.Error())
	}
	defer os.RemoveAll(clientHome)

	err = lib.CopyFile(filepath.Join(testDataDir, "ec256-1-cert.pem"), filepath.Join(clientHome, "ec256-1-cert.pem"))
	if err != nil {
		t.Fatalf("Failed to copy ec256-1-cert.pem to %s: %s", clientHome, err.Error())
	}
	err = os.MkdirAll(filepath.Join(clientHome, "msp/keystore"), 0777)
	if err != nil {
		t.Fatalf("Failed to create msp/keystore directory: %s", err.Error())
	}

	client := &lib.Client{
		Config: &lib.ClientConfig{
			URL: "http://localhost:7054",
			CSP: &factory.FactoryOpts{
				SwOpts: &factory.SwOpts{
					HashFamily: "SHA2",
					SecLevel:   256,
					FileKeystore: &factory.FileKeystoreOpts{
						KeyStorePath: "msp/keystore",
					},
				},
			},
		},
		HomeDir: clientHome,
	}
	certFile := filepath.Join(client.HomeDir, "fake-cert.pem")
	keyFile := filepath.Join(client.HomeDir, "fake-key.pem")
	x509Cred := NewCredential(certFile, keyFile, client)

	assert.Equal(t, x509Cred.Type(), CredType, "Type for a X509Credential instance must be X509")
	_, err = x509Cred.Val()
	assert.Error(t, err, "Val should return error as credential has not been loaded from disk or set")
	if err != nil {
		assert.Equal(t, err.Error(), "X509 Credential value is not set")
	}
	_, err = x509Cred.EnrollmentID()
	assert.Error(t, err, "EnrollmentID should return an error as credential has not been loaded from disk or set")
	if err != nil {
		assert.Equal(t, err.Error(), "X509 Credential value is not set")
	}

	err = x509Cred.Store()
	assert.Error(t, err, "Store should return an error as credential has not been loaded from disk or set")

	err = x509Cred.SetVal("hello")
	assert.Error(t, err, "SetVal should fail as it expects an object of type *Signer")

	_, err = x509Cred.RevokeSelf()
	assert.Error(t, err, "RevokeSelf should return an error as credential has not been loaded from disk or set")

	// Set valid value
	certBytes, err := util.ReadFile(filepath.Join(testDataDir, "ec256-1-cert.pem"))
	if err != nil {
		t.Fatalf("Failed to read file: %s", err.Error())
	}
	signer, err := NewSigner(nil, certBytes)
	if err != nil {
		t.Fatalf("Failed to create signer: %s", err.Error())
	}
	err = x509Cred.SetVal(signer)
	assert.NoError(t, err, "SetVal should not fail")

	// Load the credential from client msp, which should fail as enrollment cert/key pair is not present
	err = x509Cred.Load()
	assert.Error(t, err, "Load should have failed to load non-existent certificate file")

	certFile = filepath.Join(client.HomeDir, "ec256-1-cert.pem")
	keyFile = filepath.Join(client.HomeDir, "ec256-1-key.pem")

	err = client.Init()
	if err != nil {
		t.Fatalf("Failed to initialize client: %s", err.Error())
	}
	x509Cred = NewCredential(certFile, keyFile, client)
	err = x509Cred.Load()
	assert.Error(t, err, "Load should have failed to load non-existent key file")
	assert.Contains(t, err.Error(), "Could not find the private key in the BCCSP keystore nor in the keyfile")

	err = lib.CopyFile(filepath.Join(testDataDir, "ec256-1-key.pem"), filepath.Join(client.HomeDir, "ec256-1-key.pem"))
	if err != nil {
		t.Fatalf("Failed to copy ec256-1-key.pem to %s: %s", clientHome, err.Error())
	}
	err = x509Cred.Load()
	assert.NoError(t, err, "Load should not fail as both cert and key files exist and are valid")

	// Should error if it fails to write cert to the specified file
	if err = os.Chmod(certFile, 0000); err != nil {
		t.Fatalf("Failed to chmod certificate file %s: %s", certFile, err.Error())
	}
	err = x509Cred.Store()
	assert.Errorf(t, err, "Store should fail as %s is not writable", certFile)
	if err = os.Chmod(certFile, 0644); err != nil {
		t.Fatalf("Failed to chmod certificate file %s: %s", certFile, err.Error())
	}

	// Success cases
	err = x509Cred.Load()
	assert.NoError(t, err, "Load should not fail as cert exists and key is in bccsp keystore")

	_, err = x509Cred.Val()
	assert.NoError(t, err, "Val should not return error as x509 credential has been loaded")

	_, err = x509Cred.EnrollmentID()
	assert.NoError(t, err, "EnrollmentID should not return error as credential has been loaded")

	err = x509Cred.Store()
	assert.NoError(t, err, "Store should not fail as x509 credential is set and cert file path is valid")

	body := []byte("hello")
	req, err := http.NewRequest("GET", "localhost:7054/enroll", bytes.NewReader(body))
	if err != nil {
		t.Fatalf("Failed to create HTTP request: %s", err.Error())
	}
	_, err = x509Cred.CreateToken(req, body)
	assert.NoError(t, err, "CreateToken should not return error")
}

func TestRevokeSelf(t *testing.T) {
	clientHome, err := ioutil.TempDir(testDataDir, "revokeselftest")
	if err != nil {
		t.Fatalf("Failed to create temp directory: %s", err.Error())
	}
	defer os.RemoveAll(clientHome)

	err = lib.CopyFile(filepath.Join(testDataDir, "ec256-1-cert.pem"), filepath.Join(clientHome, "ec256-1-cert.pem"))
	if err != nil {
		t.Fatalf("Failed to copy ec256-1-cert.pem to %s: %s", clientHome, err.Error())
	}
	err = os.MkdirAll(filepath.Join(clientHome, "msp/keystore"), 0777)
	if err != nil {
		t.Fatalf("Failed to create msp/keystore directory: %s", err.Error())
	}
	keystore := filepath.Join(clientHome, "msp/keystore/ec256-1-key.pem")
	err = lib.CopyFile(filepath.Join(testDataDir, "ec256-1-key.pem"), keystore)
	if err != nil {
		t.Fatalf("Failed to copy ec256-1-key.pem to %s: %s", keystore, err.Error())
	}

	id := new(mocks.Identity)
	client := new(mocks.Client)
	opts := &factory.FactoryOpts{
		SwOpts: &factory.SwOpts{
			HashFamily: "SHA2",
			SecLevel:   256,
		},
	}
	bccsp, err := util.InitBCCSP(&opts, "msp", clientHome)
	if err != nil {
		t.Fatalf("Failed initialize BCCSP: %s", err.Error())
	}
	client.On("GetCSP").Return(bccsp)
	client.On("GetCSP").Return(nil)
	certFile := filepath.Join(clientHome, "ec256-1-cert.pem")
	cert, err := readCert(certFile)
	if err != nil {
		t.Fatalf("Failed to read the cert: %s", err.Error())
	}
	x509Cred := NewCredential(certFile, keystore, client)
	err = x509Cred.Load()
	if err != nil {
		t.Fatalf("Should not fail to load x509 credential as cert exists and key is in bccsp keystore: %s", err.Error())
	}
	name, err := x509Cred.EnrollmentID()
	assert.NoError(t, err, "EnrollmentID() should not return an error")
	client.On("NewX509Identity", name, []credential.Credential{x509Cred}).Return(id)

	serial := util.GetSerialAsHex(cert.SerialNumber)
	aki := hex.EncodeToString(cert.AuthorityKeyId)
	req := &api.RevocationRequest{
		Serial: serial,
		AKI:    aki,
	}
	id.On("Revoke", req).Return(&api.RevocationResponse{}, nil)

	_, err = x509Cred.RevokeSelf()
	assert.NoError(t, err)

	body := []byte{}
	httpReq, err := http.NewRequest("GET", "localhost:7054/enroll", bytes.NewReader(body))
	if err != nil {
		t.Fatalf("Failed to create HTTP request: %s", err.Error())
	}
	_, err = x509Cred.CreateToken(httpReq, body)
	assert.NoError(t, err)
}

func readCert(certFile string) (*x509.Certificate, error) {
	certBytes, err := util.ReadFile(certFile)
	if err != nil {
		return nil, err
	}
	cert, err := util.GetX509CertificateFromPEM(certBytes)
	if err != nil {
		return nil, err
	}
	return cert, nil
}