5452c32f50
This adds the `agent/connect/ca/plugin` library for consuming/serving Connect CA providers as [go-plugin](https://github.com/hashicorp/go-plugin) plugins. This **does not** wire this up in any way to Consul itself, so this will not enable using these plugins yet. ## Why? We want to enable CA providers to be pluggable without modifying Consul so that any CA or PKI system can potentially back the Connect certificates. This CA system may also be used in the future for easier bootstrapping and internal cluster security. ### go-plugin The benefit of `go-plugin` is that for the plugin consumer, the fact that the interface implementation is communicating over multi-process RPC is invisible. Internals of Consul will continue to just use `ca.Provider` interface implementations as if they're local. For plugin _authors_, they simply have to implement the interface. The network/transport/process management issues are handled by go-plugin itself. The CA provider plugins support both `net/rpc` and gRPC transports. This enables easy authoring in any language. go-plugin handles the actual protocol handshake and connection. This is just a feature of go-plugin. `go-plugin` is already in production use for years by Packer, Terraform, Nomad, Vault, and Sentinel. We've shown stability for both desktop and server-side software. It is very mature. ## Implementation Details ### `map[string]interface{}` The `Configure` method passes a `map[string]interface{}`. This map contains only Go primitives and containers of primitives (no funcs, chans, etc.). For `net/rpc` we encode as-is using Gob. For gRPC we marshal to JSON and transmit as a `bytes` type. This is the same approach we take with Vault and other software. Note that this is just the transport protocol, the end software views it fully decoded. ### `x509.Certificate` and `CertificateRequest` We transmit the raw ASN.1 bytes and decode on the other side. Unit tests are verifying we get the same cert/csrs across the wire. ### Testing `go-plugin` exposes test helpers that enable testing the full plugin RPC over real loopback network connections. We test all endpoints for success and error for both `net/rpc` and gRPC. ### Vendoring This PR doesn't introduce vendoring for two reasons: 1. @banks's `f-envoy` branch introduces a lot of these and I didn't want conflict. 2. The library isn't actually used yet so it doesn't introduce compile-time errors (it does introduce test errors). ## Next Steps With this in place, we need to figure out the proper way to actually hook these up to Consul, load them, etc. This discussion can happen elsewhere, since regardless of approach this plugin library implementation is the exact same.
312 lines
8.7 KiB
Go
312 lines
8.7 KiB
Go
package plugin
|
|
|
|
import (
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/agent/connect"
|
|
"github.com/hashicorp/consul/agent/connect/ca"
|
|
"github.com/hashicorp/go-plugin"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProvider_Configure(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Basic configure
|
|
m.On("Configure", "foo", false, map[string]interface{}{
|
|
"string": "bar",
|
|
"number": float64(42), // because json
|
|
}).Once().Return(nil)
|
|
require.NoError(p.Configure("foo", false, map[string]interface{}{
|
|
"string": "bar",
|
|
"number": float64(42),
|
|
}))
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("Configure", "foo", false, map[string]interface{}{}).Once().Return(errors.New("hello world"))
|
|
err := p.Configure("foo", false, map[string]interface{}{})
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_GenerateRoot(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("GenerateRoot").Once().Return(nil)
|
|
require.NoError(p.GenerateRoot())
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("GenerateRoot").Once().Return(errors.New("hello world"))
|
|
err := p.GenerateRoot()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_ActiveRoot(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("ActiveRoot").Once().Return("foo", nil)
|
|
actual, err := p.ActiveRoot()
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("ActiveRoot").Once().Return("", errors.New("hello world"))
|
|
actual, err = p.ActiveRoot()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_GenerateIntermediateCSR(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("GenerateIntermediateCSR").Once().Return("foo", nil)
|
|
actual, err := p.GenerateIntermediateCSR()
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("GenerateIntermediateCSR").Once().Return("", errors.New("hello world"))
|
|
actual, err = p.GenerateIntermediateCSR()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_SetIntermediate(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("SetIntermediate", "foo", "bar").Once().Return(nil)
|
|
err := p.SetIntermediate("foo", "bar")
|
|
require.NoError(err)
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("SetIntermediate", "foo", "bar").Once().Return(errors.New("hello world"))
|
|
err = p.SetIntermediate("foo", "bar")
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_ActiveIntermediate(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("ActiveIntermediate").Once().Return("foo", nil)
|
|
actual, err := p.ActiveIntermediate()
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("ActiveIntermediate").Once().Return("", errors.New("hello world"))
|
|
actual, err = p.ActiveIntermediate()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_GenerateIntermediate(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("GenerateIntermediate").Once().Return("foo", nil)
|
|
actual, err := p.GenerateIntermediate()
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("GenerateIntermediate").Once().Return("", errors.New("hello world"))
|
|
actual, err = p.GenerateIntermediate()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_Sign(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Create a CSR
|
|
csrPEM, _ := connect.TestCSR(t, connect.TestSpiffeIDService(t, "web"))
|
|
block, _ := pem.Decode([]byte(csrPEM))
|
|
csr, err := x509.ParseCertificateRequest(block.Bytes)
|
|
require.NoError(err)
|
|
require.NoError(csr.CheckSignature())
|
|
|
|
// No error
|
|
m.On("Sign", mock.Anything).Once().Return("foo", nil).Run(func(args mock.Arguments) {
|
|
csr := args.Get(0).(*x509.CertificateRequest)
|
|
require.NoError(csr.CheckSignature())
|
|
})
|
|
actual, err := p.Sign(csr)
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("Sign", mock.Anything).Once().Return("", errors.New("hello world"))
|
|
actual, err = p.Sign(csr)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_SignIntermediate(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Create a CSR
|
|
csrPEM, _ := connect.TestCSR(t, connect.TestSpiffeIDService(t, "web"))
|
|
block, _ := pem.Decode([]byte(csrPEM))
|
|
csr, err := x509.ParseCertificateRequest(block.Bytes)
|
|
require.NoError(err)
|
|
require.NoError(csr.CheckSignature())
|
|
|
|
// No error
|
|
m.On("SignIntermediate", mock.Anything).Once().Return("foo", nil).Run(func(args mock.Arguments) {
|
|
csr := args.Get(0).(*x509.CertificateRequest)
|
|
require.NoError(csr.CheckSignature())
|
|
})
|
|
actual, err := p.SignIntermediate(csr)
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("SignIntermediate", mock.Anything).Once().Return("", errors.New("hello world"))
|
|
actual, err = p.SignIntermediate(csr)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_CrossSignCA(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Create a CSR
|
|
root := connect.TestCA(t, nil)
|
|
block, _ := pem.Decode([]byte(root.RootCert))
|
|
crt, err := x509.ParseCertificate(block.Bytes)
|
|
require.NoError(err)
|
|
|
|
// No error
|
|
m.On("CrossSignCA", mock.Anything).Once().Return("foo", nil).Run(func(args mock.Arguments) {
|
|
actual := args.Get(0).(*x509.Certificate)
|
|
require.True(crt.Equal(actual))
|
|
})
|
|
actual, err := p.CrossSignCA(crt)
|
|
require.NoError(err)
|
|
require.Equal(actual, "foo")
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("CrossSignCA", mock.Anything).Once().Return("", errors.New("hello world"))
|
|
actual, err = p.CrossSignCA(crt)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
func TestProvider_Cleanup(t *testing.T) {
|
|
testPlugin(t, func(t *testing.T, m *ca.MockProvider, p ca.Provider) {
|
|
require := require.New(t)
|
|
|
|
// Try cleanup with no error
|
|
m.On("Cleanup").Once().Return(nil)
|
|
require.NoError(p.Cleanup())
|
|
m.AssertExpectations(t)
|
|
|
|
// Try with an error
|
|
m.Mock = mock.Mock{}
|
|
m.On("Cleanup").Once().Return(errors.New("hello world"))
|
|
err := p.Cleanup()
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "hello")
|
|
m.AssertExpectations(t)
|
|
})
|
|
}
|
|
|
|
// testPlugin runs the given test function callback for all supported
|
|
// transports of the plugin RPC layer.
|
|
func testPlugin(t *testing.T, f func(t *testing.T, m *ca.MockProvider, actual ca.Provider)) {
|
|
t.Run("net/rpc", func(t *testing.T) {
|
|
// Create a mock provider
|
|
mockP := new(ca.MockProvider)
|
|
client, _ := plugin.TestPluginRPCConn(t, map[string]plugin.Plugin{
|
|
Name: &ProviderPlugin{Impl: mockP},
|
|
}, nil)
|
|
defer client.Close()
|
|
|
|
// Request the provider
|
|
raw, err := client.Dispense(Name)
|
|
require.NoError(t, err)
|
|
provider := raw.(ca.Provider)
|
|
|
|
// Call the test function
|
|
f(t, mockP, provider)
|
|
})
|
|
|
|
t.Run("gRPC", func(t *testing.T) {
|
|
// Create a mock provider
|
|
mockP := new(ca.MockProvider)
|
|
client, _ := plugin.TestPluginGRPCConn(t, map[string]plugin.Plugin{
|
|
Name: &ProviderPlugin{Impl: mockP},
|
|
})
|
|
defer client.Close()
|
|
|
|
// Request the provider
|
|
raw, err := client.Dispense(Name)
|
|
require.NoError(t, err)
|
|
provider := raw.(ca.Provider)
|
|
|
|
// Call the test function
|
|
f(t, mockP, provider)
|
|
})
|
|
}
|