VMware vSphere Integrated Containers provider (#206)
* Add Virtual Kubelet provider for VIC Initial virtual kubelet provider for VMware VIC. This provider currently handles creating and starting of a pod VM via the VIC portlayer and persona server. Image store handling via the VIC persona server. This provider currently requires the feature/wolfpack branch of VIC. * Added pod stop and delete. Also added node capacity. Added the ability to stop and delete pod VMs via VIC. Also retrieve node capacity information from the VCH. * Cleanup and readme file Some file clean up and added a Readme.md markdown file for the VIC provider. * Cleaned up errors, added function comments, moved operation code 1. Cleaned up error handling. Set standard for creating errors. 2. Added method prototype comments for all interface functions. 3. Moved PodCreator, PodStarter, PodStopper, and PodDeleter to a new folder. * Add mocking code and unit tests for podcache, podcreator, and podstarter Used the unit test framework used in VIC to handle assertions in the provider's unit test. Mocking code generated using OSS project mockery, which is compatible with the testify assertion framework. * Vendored packages for the VIC provider Requires feature/wolfpack branch of VIC and a few specific commit sha of projects used within VIC. * Implementation of POD Stopper and Deleter unit tests (#4) * Updated files for initial PR
This commit is contained in:
491
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/source.go
generated
vendored
Normal file
491
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/source.go
generated
vendored
Normal file
@@ -0,0 +1,491 @@
|
||||
// Copyright 2016-2017 VMware, Inc. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package admiral
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
log "github.com/Sirupsen/logrus"
|
||||
"github.com/go-openapi/runtime"
|
||||
rtclient "github.com/go-openapi/runtime/client"
|
||||
strfmt "github.com/go-openapi/strfmt"
|
||||
"github.com/mitchellh/mapstructure"
|
||||
"github.com/pkg/errors"
|
||||
|
||||
"github.com/vmware/govmomi/vim25/types"
|
||||
vchcfg "github.com/vmware/vic/lib/config"
|
||||
"github.com/vmware/vic/lib/config/dynamic"
|
||||
"github.com/vmware/vic/lib/config/dynamic/admiral/client"
|
||||
"github.com/vmware/vic/lib/config/dynamic/admiral/client/config_registries"
|
||||
"github.com/vmware/vic/lib/config/dynamic/admiral/client/projects"
|
||||
"github.com/vmware/vic/lib/config/dynamic/admiral/client/resources_compute"
|
||||
"github.com/vmware/vic/lib/config/dynamic/admiral/models"
|
||||
"github.com/vmware/vic/pkg/vsphere/session"
|
||||
"github.com/vmware/vic/pkg/vsphere/tags"
|
||||
"github.com/vmware/vic/pkg/vsphere/vm"
|
||||
)
|
||||
|
||||
const (
|
||||
VicProductCategory = "VsphereIntegratedContainers"
|
||||
ProductVMTag = "ProductVM"
|
||||
admiralEndpointKey = "guestinfo.vicova.admiral.endpoint"
|
||||
|
||||
clusterFilter = "(address eq '%s' and customProperties.__containerHostType eq 'VCH')"
|
||||
|
||||
// defaultVicRegistry is the query string to fetch the default VIC registry from Admiral
|
||||
defaultVicRegistry = "default-vic-registry"
|
||||
)
|
||||
|
||||
// #nosec
|
||||
const admiralTokenKey = "guestinfo.vicova.engine.token"
|
||||
|
||||
var (
|
||||
trueStr = "true"
|
||||
projectsFilter = "customProperties.__enableContentTrust eq 'true'"
|
||||
|
||||
// #nosec
|
||||
admClient = &http.Client{
|
||||
Transport: // copied from http.DefaultTransport
|
||||
&http.Transport{
|
||||
Proxy: http.ProxyFromEnvironment,
|
||||
DialContext: (&net.Dialer{
|
||||
Timeout: 30 * time.Second,
|
||||
KeepAlive: 30 * time.Second,
|
||||
DualStack: true,
|
||||
}).DialContext,
|
||||
MaxIdleConns: 100,
|
||||
IdleConnTimeout: 90 * time.Second,
|
||||
TLSHandshakeTimeout: 10 * time.Second,
|
||||
ExpectContinueTimeout: 1 * time.Second,
|
||||
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
// NewSource creates a new Admiral dynamic config source. sess
|
||||
// is a valid vsphere session object. vchID is a unique identifier
|
||||
// that will be used to lookup the VCH in the Admiral instance; currently
|
||||
// this is a URI to the docker endpoint in the VCH.
|
||||
func NewSource(sess *session.Session, vchID string) dynamic.Source {
|
||||
return &source{
|
||||
d: DefaultDiscovery,
|
||||
sess: sess,
|
||||
vchID: vchID,
|
||||
}
|
||||
}
|
||||
|
||||
var DefaultDiscovery = &productDiscovery{
|
||||
clients: make(map[string]*tags.RestClient),
|
||||
}
|
||||
|
||||
type source struct {
|
||||
mu sync.Mutex
|
||||
d discovery
|
||||
sess *session.Session
|
||||
vchID string
|
||||
lastCfg *vchcfg.VirtualContainerHostConfigSpec
|
||||
|
||||
// the discovered product VM
|
||||
v *vm.VirtualMachine
|
||||
// cached values to minimize running discovery
|
||||
projs []string
|
||||
c *client.Admiral
|
||||
}
|
||||
|
||||
// Get returns the dynamic config portion from an Admiral instance. For now,
|
||||
// this is empty pending details from the Admiral team.
|
||||
func (a *source) Get(ctx context.Context) (*vchcfg.VirtualContainerHostConfigSpec, error) {
|
||||
a.mu.Lock()
|
||||
lastCfg := a.lastCfg
|
||||
a.mu.Unlock()
|
||||
|
||||
c, projs, err := a.discover(ctx)
|
||||
if err != nil {
|
||||
log.Warnf("could not locate management portal, returning last known config: %s", err)
|
||||
return lastCfg, nil
|
||||
}
|
||||
|
||||
if len(projs) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
wl, err := a.whitelist(ctx, c, projs)
|
||||
if err != nil {
|
||||
log.Warnf("could not get whitelist from management portal, returning last known config: %s")
|
||||
return lastCfg, nil
|
||||
}
|
||||
|
||||
log.Debugf("got whitelist: %+v", wl)
|
||||
|
||||
newCfg := &vchcfg.VirtualContainerHostConfigSpec{
|
||||
Registry: vchcfg.Registry{RegistryWhitelist: wl},
|
||||
}
|
||||
|
||||
a.mu.Lock()
|
||||
defer a.mu.Unlock()
|
||||
|
||||
err = nil
|
||||
if !a.diff(newCfg) {
|
||||
err = dynamic.ErrConfigNotModified
|
||||
}
|
||||
|
||||
a.lastCfg = newCfg
|
||||
return newCfg, err
|
||||
}
|
||||
|
||||
func (a *source) diff(newCfg *vchcfg.VirtualContainerHostConfigSpec) bool {
|
||||
if a.lastCfg == nil {
|
||||
if newCfg == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
if newCfg == nil {
|
||||
return true
|
||||
}
|
||||
|
||||
// compare whitelists
|
||||
if len(a.lastCfg.RegistryWhitelist) != len(newCfg.RegistryWhitelist) {
|
||||
return true
|
||||
}
|
||||
|
||||
for _, w1 := range a.lastCfg.RegistryWhitelist {
|
||||
found := false
|
||||
for _, w2 := range newCfg.RegistryWhitelist {
|
||||
if w2 == w1 {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !found {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func (a *source) discover(ctx context.Context) (*client.Admiral, []string, error) {
|
||||
a.mu.Lock()
|
||||
c := a.c
|
||||
v := a.v
|
||||
a.mu.Unlock()
|
||||
|
||||
var vms []*vm.VirtualMachine
|
||||
removed := true
|
||||
if c != nil {
|
||||
projs, err := a.projects(ctx, c)
|
||||
if err != nil {
|
||||
log.Debugf("could not get projects: %s", err)
|
||||
vms = append(vms, v)
|
||||
removed = false
|
||||
} else if len(projs) > 0 {
|
||||
return c, projs, nil
|
||||
}
|
||||
}
|
||||
|
||||
a.mu.Lock()
|
||||
defer a.mu.Unlock()
|
||||
|
||||
if c != a.c && a.c != nil {
|
||||
return a.c, a.projs, nil
|
||||
}
|
||||
|
||||
var err error
|
||||
// if there isn't a list of potential product VMs
|
||||
// already, then run discovery
|
||||
if removed {
|
||||
log.Infof("running product VM discovery")
|
||||
vms, err = a.d.Discover(ctx, a.sess)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
}
|
||||
|
||||
for _, v := range vms {
|
||||
u, token, err := admiralEndpoint(ctx, v)
|
||||
if err != nil {
|
||||
log.Warnf("ignoring potential product VM %s: %s", v, err)
|
||||
continue
|
||||
}
|
||||
|
||||
rt := rtclient.NewWithClient(u.Host, u.Path, []string{u.Scheme}, admClient)
|
||||
rt.DefaultAuthentication = &admiralAuth{token: token}
|
||||
c := client.New(rt, strfmt.Default)
|
||||
projs, err := a.projects(ctx, c)
|
||||
if err == nil && len(projs) > 0 {
|
||||
log.Infof("using Admiral endpoint at %s", u)
|
||||
a.c = c
|
||||
a.projs = projs
|
||||
a.v = v
|
||||
return c, projs, nil
|
||||
}
|
||||
|
||||
log.Warnf("ignoring product VM Admiral endpoint %s since it does not have the VCH added to any project", u)
|
||||
}
|
||||
|
||||
if removed {
|
||||
a.c = nil
|
||||
a.v = nil
|
||||
a.projs = nil
|
||||
err = nil
|
||||
} else {
|
||||
// only return source unavailable if
|
||||
// the VCH was not removed from the
|
||||
// Admiral instance
|
||||
err = dynamic.ErrSourceUnavailable
|
||||
}
|
||||
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
func admiralEndpoint(ctx context.Context, v *vm.VirtualMachine) (*url.URL, string, error) {
|
||||
values, err := keys(ctx, v, []string{admiralEndpointKey, admiralTokenKey})
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
token := values[admiralTokenKey]
|
||||
if token == "" {
|
||||
// not a useable product installation
|
||||
return nil, "", fmt.Errorf("empty token set in product VM %s", v)
|
||||
}
|
||||
|
||||
u, err := url.Parse(values[admiralEndpointKey])
|
||||
if err != nil {
|
||||
return nil, "", fmt.Errorf("bad admiral endpoint %s in product VM %s: %s", values[admiralEndpointKey], v, err)
|
||||
}
|
||||
|
||||
return u, token, nil
|
||||
}
|
||||
|
||||
type admiralAuth struct {
|
||||
token string
|
||||
}
|
||||
|
||||
func (a *admiralAuth) AuthenticateRequest(req runtime.ClientRequest, _ strfmt.Registry) error {
|
||||
return req.SetHeaderParam("x-xenon-auth-token", a.token)
|
||||
}
|
||||
|
||||
func (a *source) projects(ctx context.Context, c *client.Admiral) ([]string, error) {
|
||||
ids := []string{a.vchID}
|
||||
if u, err := url.Parse(a.vchID); err == nil {
|
||||
if u.Scheme == "" {
|
||||
ids = append(ids, "https://"+a.vchID)
|
||||
} else {
|
||||
ids = append(ids, strings.TrimPrefix(a.vchID, u.Scheme+"://"))
|
||||
}
|
||||
}
|
||||
|
||||
var err error
|
||||
var comps *resources_compute.GetResourcesComputeOK
|
||||
for _, vchID := range ids {
|
||||
filter := fmt.Sprintf(clusterFilter, vchID)
|
||||
log.Debugf("getting compute resources with filter %s", filter)
|
||||
comps, err = c.ResourcesCompute.GetResourcesCompute(resources_compute.NewGetResourcesComputeParamsWithContext(ctx).WithDollarFilter(&filter))
|
||||
if err == nil && comps.Payload.DocumentCount > 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if comps.Payload.DocumentCount == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// more than one project has the VCH added, just pick the first one
|
||||
comp := &models.ComVmwarePhotonControllerModelResourcesComputeServiceComputeState{}
|
||||
if err := mapstructure.Decode(comps.Payload.Documents[comps.Payload.DocumentLinks[0]], comp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return comp.TenantLinks, nil
|
||||
}
|
||||
|
||||
func (a *source) whitelist(ctx context.Context, c *client.Admiral, hostProjs []string) ([]string, error) {
|
||||
// find at least one project with enable content trust
|
||||
// that also contains the vch
|
||||
projs, err := c.Projects.GetProjects(projects.NewGetProjectsParamsWithContext(ctx).WithDollarFilter(&projectsFilter))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
trust := false
|
||||
for _, t := range hostProjs {
|
||||
for _, p := range projs.Payload.DocumentLinks {
|
||||
if t == p {
|
||||
trust = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if trust {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !trust {
|
||||
// no project with enable content trust and vch
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
regs, err := c.ConfigRegistries.GetConfigRegistriesID(config_registries.NewGetConfigRegistriesIDParamsWithContext(ctx).WithID(defaultVicRegistry))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if regs.Payload == nil {
|
||||
// No default VIC registry configured.
|
||||
log.Warnf("no default VIC registry found despite content trust being enabled")
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
m := regs.Payload
|
||||
wl := []string{m.Address}
|
||||
|
||||
return wl, nil
|
||||
}
|
||||
|
||||
type discovery interface {
|
||||
Discover(ctx context.Context, sess *session.Session) ([]*vm.VirtualMachine, error)
|
||||
}
|
||||
|
||||
type productDiscovery struct {
|
||||
mu sync.Mutex
|
||||
clients map[string]*tags.RestClient
|
||||
}
|
||||
|
||||
func (o *productDiscovery) Discover(ctx context.Context, sess *session.Session) ([]*vm.VirtualMachine, error) {
|
||||
service, err := url.Parse(sess.Service)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
service.User = sess.User
|
||||
|
||||
o.mu.Lock()
|
||||
t := o.clients[service.String()]
|
||||
if t == nil {
|
||||
t = tags.NewClient(service, sess.Insecure, sess.Thumbprint)
|
||||
o.clients[service.String()] = t
|
||||
}
|
||||
o.mu.Unlock()
|
||||
|
||||
var tag string
|
||||
tag, err = findOVATag(ctx, t)
|
||||
if err != nil {
|
||||
err = errors.Errorf("could not find ova tag: %s", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
objs, err := t.ListAttachedObjects(ctx, tag)
|
||||
if err != nil || len(objs) == 0 {
|
||||
err = errors.Errorf("could not find ova vm: %s", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var vms []*vm.VirtualMachine
|
||||
for _, o := range objs {
|
||||
if o.Type == nil || o.ID == nil {
|
||||
log.Warnf("skipping invalid object reference %+v", o)
|
||||
continue
|
||||
}
|
||||
|
||||
if *o.Type != "VirtualMachine" {
|
||||
// not a virtual machine
|
||||
continue
|
||||
}
|
||||
|
||||
v := vm.NewVirtualMachine(ctx, sess, types.ManagedObjectReference{Type: *o.Type, Value: *o.ID})
|
||||
st, err := v.PowerState(ctx)
|
||||
if err != nil {
|
||||
log.Warnf("ignoring potential product VM: %s", err)
|
||||
continue
|
||||
}
|
||||
|
||||
if st != types.VirtualMachinePowerStatePoweredOn {
|
||||
log.Warnf("ignoring potential product VM %s: not powered on", *o.ID)
|
||||
continue
|
||||
}
|
||||
|
||||
vms = append(vms, v)
|
||||
}
|
||||
|
||||
return vms, nil
|
||||
}
|
||||
|
||||
func findOVATag(ctx context.Context, t *tags.RestClient) (string, error) {
|
||||
cats, err := t.GetCategoriesByName(ctx, VicProductCategory)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// just use the first one
|
||||
if len(cats) == 0 {
|
||||
return "", errors.New("could not find tag")
|
||||
}
|
||||
|
||||
cat := cats[0]
|
||||
tags, err := t.GetTagByNameForCategory(ctx, ProductVMTag, cat.ID)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
if len(tags) == 0 {
|
||||
return "", errors.New("could not find tag")
|
||||
}
|
||||
|
||||
return tags[0].ID, nil
|
||||
}
|
||||
|
||||
func keys(ctx context.Context, v *vm.VirtualMachine, keys []string) (map[string]string, error) {
|
||||
ovs, err := v.FetchExtraConfigBaseOptions(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
res := make(map[string]string)
|
||||
for _, k := range keys {
|
||||
found := false
|
||||
for _, ov := range ovs {
|
||||
log.Debugf("key: %s", ov.GetOptionValue().Key)
|
||||
if k == ov.GetOptionValue().Key {
|
||||
log.Debugf("found %s", ov.GetOptionValue().Key)
|
||||
res[k] = ov.GetOptionValue().Value.(string)
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !found {
|
||||
return nil, errors.Errorf("key not found: %s", k)
|
||||
}
|
||||
}
|
||||
|
||||
return res, nil
|
||||
}
|
||||
72
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/source_test.go
generated
vendored
Normal file
72
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/source_test.go
generated
vendored
Normal file
@@ -0,0 +1,72 @@
|
||||
// Copyright 2017 VMware, Inc. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
package admiral
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"net/url"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
log "github.com/Sirupsen/logrus"
|
||||
|
||||
"github.com/vmware/vic/pkg/vsphere/session"
|
||||
)
|
||||
|
||||
var (
|
||||
target *string
|
||||
user *string
|
||||
password *string
|
||||
thumbprint *string
|
||||
vchID *string
|
||||
)
|
||||
|
||||
func init() {
|
||||
thumbprint = flag.String("thumbprint", "", "ESX/VC thumbprint")
|
||||
target = flag.String("target", "", "ESX/VC target")
|
||||
user = flag.String("user", "root", "ESX/VC username")
|
||||
password = flag.String("password", "", "ESX/VC password")
|
||||
vchID = flag.String("vch", "", "VCH docker endpoint")
|
||||
}
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
os.Exit(m.Run())
|
||||
}
|
||||
|
||||
func TestGetConfig(t *testing.T) {
|
||||
if target == nil || *target == "" {
|
||||
t.SkipNow()
|
||||
}
|
||||
|
||||
log.SetLevel(log.DebugLevel)
|
||||
sess := session.NewSession(&session.Config{
|
||||
Service: *target,
|
||||
User: url.UserPassword(*user, *password),
|
||||
Insecure: true,
|
||||
Thumbprint: *thumbprint,
|
||||
})
|
||||
|
||||
ctx := context.Background()
|
||||
sess.Connect(ctx)
|
||||
|
||||
src := NewSource(sess, *vchID)
|
||||
cfg, err := src.Get(ctx)
|
||||
if err != nil {
|
||||
log.Errorf("error: %s", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
log.Infof("%+v", cfg)
|
||||
}
|
||||
33976
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/swagger.json
generated
vendored
Normal file
33976
vendor/github.com/vmware/vic/lib/config/dynamic/admiral/swagger.json
generated
vendored
Normal file
File diff suppressed because it is too large
Load Diff
43
vendor/github.com/vmware/vic/lib/config/dynamic/dynamic.go
generated
vendored
Normal file
43
vendor/github.com/vmware/vic/lib/config/dynamic/dynamic.go
generated
vendored
Normal file
@@ -0,0 +1,43 @@
|
||||
// Copyright 2016-2017 VMware, Inc. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package dynamic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/vmware/vic/lib/config"
|
||||
)
|
||||
|
||||
var ErrConfigNotModified = errors.New("config not modified")
|
||||
var ErrAccessDenied = errors.New("access denied")
|
||||
var ErrSourceUnavailable = errors.New("source not available")
|
||||
|
||||
// Source is configuration source, remote or otherwise
|
||||
type Source interface {
|
||||
// Get returns a config object. If the remote/local source
|
||||
// is not available, Get returns nil, with an error indicating
|
||||
// what went wrong. If the configuration has not changed since
|
||||
// the last time Get was called, it returns
|
||||
// (nil, ErrConfigNotModified). If the remote/local source
|
||||
// denies access, it returns (nil, ErrAccessDenied).
|
||||
Get(ctx context.Context) (*config.VirtualContainerHostConfigSpec, error)
|
||||
}
|
||||
|
||||
// Merger is used to merge two config objects
|
||||
type Merger interface {
|
||||
// Merge merges two config objects returning the resulting config.
|
||||
Merge(orig, other *config.VirtualContainerHostConfigSpec) (*config.VirtualContainerHostConfigSpec, error)
|
||||
}
|
||||
134
vendor/github.com/vmware/vic/lib/config/dynamic/merger.go
generated
vendored
Normal file
134
vendor/github.com/vmware/vic/lib/config/dynamic/merger.go
generated
vendored
Normal file
@@ -0,0 +1,134 @@
|
||||
// Copyright 2016-2017 VMware, Inc. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package dynamic
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/vmware/vic/lib/config"
|
||||
"github.com/vmware/vic/pkg/registry"
|
||||
)
|
||||
|
||||
type merger struct {
|
||||
}
|
||||
|
||||
func NewMerger() Merger {
|
||||
return &merger{}
|
||||
}
|
||||
|
||||
// Merge merges two config objects together. For now only
|
||||
// whitelist registries are merged.
|
||||
func (m *merger) Merge(orig, other *config.VirtualContainerHostConfigSpec) (*config.VirtualContainerHostConfigSpec, error) {
|
||||
// merge strategy:
|
||||
//
|
||||
// origWl empty, otherWl empty => empty
|
||||
//
|
||||
// origWl empty, otherWl not empty => otherWl
|
||||
//
|
||||
// origWl not empty, otherWl empty => origWl
|
||||
//
|
||||
// origWl not empty, otherWl not empty => merge result
|
||||
// in this case, each entry in the resulting
|
||||
// whitelist must be a more restrictive
|
||||
// version of at least one entry in origWl
|
||||
//
|
||||
// The whitelist that is used is always otherWl
|
||||
// in this case given that the above rule is not
|
||||
// violated.
|
||||
otherWl, err := ParseRegistries(other.RegistryWhitelist)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
origWl, err := ParseRegistries(orig.RegistryWhitelist)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var wl registry.Set
|
||||
if wl, err = origWl.Merge(otherWl, &whitelistMerger{}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(origWl) > 0 {
|
||||
// check if every entry in wl is a subset of an
|
||||
// entry in origWl
|
||||
for _, e := range wl {
|
||||
found := false
|
||||
for _, o := range origWl {
|
||||
if o.Contains(e) {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !found {
|
||||
return nil, fmt.Errorf("whitelist merge allows entries that are not in the original whitelist")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// only use otherWl if its non-empty
|
||||
//
|
||||
// if otherWl is empty and origWl is
|
||||
// not empty, we use origWl, which
|
||||
// should be the same as wl after the
|
||||
// merge
|
||||
if len(otherWl) > 0 {
|
||||
wl = otherWl
|
||||
}
|
||||
|
||||
res := *orig
|
||||
res.RegistryWhitelist = wl.Strings()
|
||||
return &res, nil
|
||||
}
|
||||
|
||||
func ParseRegistries(regs []string) (registry.Set, error) {
|
||||
var s registry.Set
|
||||
for _, r := range regs {
|
||||
e := registry.ParseEntry(r)
|
||||
if e != nil {
|
||||
s = append(s, e)
|
||||
continue
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("could not parse entry %s", r)
|
||||
}
|
||||
|
||||
return s, nil
|
||||
}
|
||||
|
||||
type whitelistMerger struct{}
|
||||
|
||||
// Merge merges two registry entries. The merge fails if merging orig and other would
|
||||
// broaden orig's scope. The result of the merge is other if that is more restrictive.
|
||||
// if orig equals other, the result is orig.
|
||||
func (w *whitelistMerger) Merge(orig, other registry.Entry) (registry.Entry, error) {
|
||||
if orig.Equal(other) {
|
||||
return orig, nil
|
||||
}
|
||||
|
||||
if other.Contains(orig) {
|
||||
return nil, fmt.Errorf("merge of %s and %s would broaden %s", orig, other, orig)
|
||||
}
|
||||
|
||||
// more restrictive result is OK
|
||||
if orig.Contains(other) {
|
||||
return other, nil
|
||||
}
|
||||
|
||||
// no merge
|
||||
return nil, nil
|
||||
}
|
||||
302
vendor/github.com/vmware/vic/lib/config/dynamic/merger_test.go
generated
vendored
Normal file
302
vendor/github.com/vmware/vic/lib/config/dynamic/merger_test.go
generated
vendored
Normal file
@@ -0,0 +1,302 @@
|
||||
// Copyright 2017 VMware, Inc. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
package dynamic
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
|
||||
"github.com/vmware/vic/lib/config"
|
||||
"github.com/vmware/vic/pkg/registry"
|
||||
)
|
||||
|
||||
func TestWhitelistMerger(t *testing.T) {
|
||||
var tests = []struct {
|
||||
orig, other registry.Entry
|
||||
res registry.Entry
|
||||
err error
|
||||
}{
|
||||
{
|
||||
orig: registry.ParseEntry("10.10.10.10"),
|
||||
other: registry.ParseEntry("10.10.10.10"),
|
||||
res: registry.ParseEntry("10.10.10.10"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("10.10.10.10"),
|
||||
other: registry.ParseEntry("10.10.10.20"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("10.10.10.10/24"),
|
||||
other: registry.ParseEntry("10.10.10.10/24"),
|
||||
res: registry.ParseEntry("10.10.10.10/24"),
|
||||
},
|
||||
{
|
||||
other: registry.ParseEntry("10.10.10.10/24"),
|
||||
orig: registry.ParseEntry("192.168.1.0/24"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("10.10.10.10/24"),
|
||||
other: registry.ParseEntry("10.10.10.10/16"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("10.10.10.10/16"),
|
||||
other: registry.ParseEntry("10.10.10.10/24"),
|
||||
res: registry.ParseEntry("10.10.10.10/24"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("*.google.com"),
|
||||
other: registry.ParseEntry("*.google.com"),
|
||||
res: registry.ParseEntry("*.google.com"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("*.yahoo.com"),
|
||||
other: registry.ParseEntry("*.google.com"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("*.google.com"),
|
||||
other: registry.ParseEntry("mail.google.com"),
|
||||
res: registry.ParseEntry("mail.google.com"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("mail.google.com"),
|
||||
other: registry.ParseEntry("*.google.com"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("192.168.1.1:123"),
|
||||
other: registry.ParseEntry("192.168.1.1"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("192.168.1.1"),
|
||||
other: registry.ParseEntry("192.168.1.1:123"),
|
||||
res: registry.ParseEntry("192.168.1.1"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("foo:123"),
|
||||
other: registry.ParseEntry("foo"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("foo"),
|
||||
other: registry.ParseEntry("foo:123"),
|
||||
res: registry.ParseEntry("foo"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("http://foo"),
|
||||
other: registry.ParseEntry("foo:123"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("http://foo"),
|
||||
other: registry.ParseEntry("http://foo:123"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("http://foo:123"),
|
||||
other: registry.ParseEntry("http://foo"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("http://foo/bar"),
|
||||
other: registry.ParseEntry("http://foo"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("https://foo/bar"),
|
||||
other: registry.ParseEntry("http://foo/bar"),
|
||||
},
|
||||
{
|
||||
orig: registry.ParseEntry("https://foo"),
|
||||
other: registry.ParseEntry("foo"),
|
||||
err: assert.AnError,
|
||||
},
|
||||
}
|
||||
|
||||
m := &whitelistMerger{}
|
||||
|
||||
for _, te := range tests {
|
||||
res, err := m.Merge(te.orig, te.other)
|
||||
if te.err == nil {
|
||||
assert.Nil(t, err, "case: orig: %s, other: %s, err: %v, res: %s", te.orig, te.other, te.err, te.res)
|
||||
} else {
|
||||
assert.NotNil(t, err, "case: orig: %s, other: %s, err: %v, res: %s", te.orig, te.other, te.err, te.res)
|
||||
}
|
||||
|
||||
if te.res == nil {
|
||||
assert.Nil(t, res, "case: orig: %s, other: %s, err: %v, res: %s", te.orig, te.other, te.err, te.res)
|
||||
} else {
|
||||
assert.True(t, te.res.Equal(res), "%s merge %s, %s (expected) == %s (actual)", te.orig, te.other, te.res, res)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMergerMergeWhitelist(t *testing.T) {
|
||||
var tests = []struct {
|
||||
orig, other, res *config.VirtualContainerHostConfigSpec
|
||||
err error
|
||||
}{
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: nil,
|
||||
},
|
||||
},
|
||||
res: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
},
|
||||
// disallow the merge if the other whitelist
|
||||
// has an entry that is not allowed on the
|
||||
// original whitelist
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"foo.docker.io", "malicious.io"},
|
||||
},
|
||||
},
|
||||
err: assert.AnError,
|
||||
},
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"*.docker.io"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"bar.docker.io", "foo.docker.io"},
|
||||
},
|
||||
},
|
||||
res: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"bar.docker.io", "foo.docker.io"},
|
||||
},
|
||||
},
|
||||
},
|
||||
// result of a merge is always the other if
|
||||
// its a subset of the original
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io", "harbor.ci.local"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
res: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
},
|
||||
// empty original whitelist and non-empty other
|
||||
// whitelist results in other whitelist
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: nil,
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
res: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"docker.io"},
|
||||
},
|
||||
},
|
||||
},
|
||||
// more permissive other whitelist results in error
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"foo.docker.io"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"*.docker.io"},
|
||||
},
|
||||
},
|
||||
err: assert.AnError,
|
||||
},
|
||||
// less permissive other whitelist results in other whitelist
|
||||
{
|
||||
orig: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"*.docker.io"},
|
||||
},
|
||||
},
|
||||
other: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"foo.docker.io"},
|
||||
},
|
||||
},
|
||||
res: &config.VirtualContainerHostConfigSpec{
|
||||
Registry: config.Registry{
|
||||
RegistryWhitelist: []string{"foo.docker.io"},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
m := NewMerger()
|
||||
for _, te := range tests {
|
||||
t.Logf("orig: %+v, other: %+v, err: %+v", te.orig.RegistryWhitelist, te.other.RegistryWhitelist, te.err)
|
||||
if te.res != nil {
|
||||
t.Logf("res: %+v", te.res.RegistryWhitelist)
|
||||
}
|
||||
res, err := m.Merge(te.orig, te.other)
|
||||
if te.err != nil {
|
||||
assert.NotNil(t, err, "expected error, got nil")
|
||||
assert.Nil(t, res)
|
||||
continue
|
||||
}
|
||||
|
||||
assert.Nil(t, err, "expected no error, got \"%+v\"", err)
|
||||
assert.NotNil(t, res)
|
||||
assert.Len(t, res.RegistryWhitelist, len(te.res.RegistryWhitelist), "expected %v, got %v", te.res.RegistryWhitelist, res.RegistryWhitelist)
|
||||
for i := range res.RegistryWhitelist {
|
||||
found := false
|
||||
for j := range te.res.RegistryWhitelist {
|
||||
if res.RegistryWhitelist[i] == te.res.RegistryWhitelist[j] {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
assert.True(t, found, "expected whitelist %v, got %v", te.res.RegistryWhitelist, res.RegistryWhitelist)
|
||||
if !found {
|
||||
t.FailNow()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user