Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
bbe86a216e | ||
|
effc4cec21 |
@@ -15,14 +15,13 @@ import (
|
|||||||
"github.com/coreos/coreos-cloudinit/datasource/metadata/ec2"
|
"github.com/coreos/coreos-cloudinit/datasource/metadata/ec2"
|
||||||
"github.com/coreos/coreos-cloudinit/datasource/proc_cmdline"
|
"github.com/coreos/coreos-cloudinit/datasource/proc_cmdline"
|
||||||
"github.com/coreos/coreos-cloudinit/datasource/url"
|
"github.com/coreos/coreos-cloudinit/datasource/url"
|
||||||
"github.com/coreos/coreos-cloudinit/datasource/waagent"
|
|
||||||
"github.com/coreos/coreos-cloudinit/initialize"
|
"github.com/coreos/coreos-cloudinit/initialize"
|
||||||
"github.com/coreos/coreos-cloudinit/pkg"
|
"github.com/coreos/coreos-cloudinit/pkg"
|
||||||
"github.com/coreos/coreos-cloudinit/system"
|
"github.com/coreos/coreos-cloudinit/system"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
version = "0.10.6"
|
version = "0.10.5"
|
||||||
datasourceInterval = 100 * time.Millisecond
|
datasourceInterval = 100 * time.Millisecond
|
||||||
datasourceMaxInterval = 30 * time.Second
|
datasourceMaxInterval = 30 * time.Second
|
||||||
datasourceTimeout = 5 * time.Minute
|
datasourceTimeout = 5 * time.Minute
|
||||||
@@ -35,7 +34,6 @@ var (
|
|||||||
sources struct {
|
sources struct {
|
||||||
file string
|
file string
|
||||||
configDrive string
|
configDrive string
|
||||||
waagent string
|
|
||||||
metadataService bool
|
metadataService bool
|
||||||
ec2MetadataService string
|
ec2MetadataService string
|
||||||
cloudSigmaMetadataService bool
|
cloudSigmaMetadataService bool
|
||||||
@@ -55,7 +53,6 @@ func init() {
|
|||||||
flag.BoolVar(&flags.ignoreFailure, "ignore-failure", false, "Exits with 0 status in the event of malformed input from user-data")
|
flag.BoolVar(&flags.ignoreFailure, "ignore-failure", false, "Exits with 0 status in the event of malformed input from user-data")
|
||||||
flag.StringVar(&flags.sources.file, "from-file", "", "Read user-data from provided file")
|
flag.StringVar(&flags.sources.file, "from-file", "", "Read user-data from provided file")
|
||||||
flag.StringVar(&flags.sources.configDrive, "from-configdrive", "", "Read data from provided cloud-drive directory")
|
flag.StringVar(&flags.sources.configDrive, "from-configdrive", "", "Read data from provided cloud-drive directory")
|
||||||
flag.StringVar(&flags.sources.waagent, "from-waagent", "", "Read data from provided waagent directory")
|
|
||||||
flag.BoolVar(&flags.sources.metadataService, "from-metadata-service", false, "[DEPRECATED - Use -from-ec2-metadata] Download data from metadata service")
|
flag.BoolVar(&flags.sources.metadataService, "from-metadata-service", false, "[DEPRECATED - Use -from-ec2-metadata] Download data from metadata service")
|
||||||
flag.StringVar(&flags.sources.ec2MetadataService, "from-ec2-metadata", "", "Download EC2 data from the provided url")
|
flag.StringVar(&flags.sources.ec2MetadataService, "from-ec2-metadata", "", "Download EC2 data from the provided url")
|
||||||
flag.BoolVar(&flags.sources.cloudSigmaMetadataService, "from-cloudsigma-metadata", false, "Download data from CloudSigma server context")
|
flag.BoolVar(&flags.sources.cloudSigmaMetadataService, "from-cloudsigma-metadata", false, "Download data from CloudSigma server context")
|
||||||
@@ -84,9 +81,6 @@ var (
|
|||||||
"from-configdrive": "/media/configdrive",
|
"from-configdrive": "/media/configdrive",
|
||||||
"convert-netconf": "debian",
|
"convert-netconf": "debian",
|
||||||
},
|
},
|
||||||
"azure": oemConfig{
|
|
||||||
"from-waagent": "/var/lib/waagent",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -284,9 +278,6 @@ func getDatasources() []datasource.Datasource {
|
|||||||
if flags.sources.digitalOceanMetadataService != "" {
|
if flags.sources.digitalOceanMetadataService != "" {
|
||||||
dss = append(dss, digitalocean.NewDatasource(flags.sources.digitalOceanMetadataService))
|
dss = append(dss, digitalocean.NewDatasource(flags.sources.digitalOceanMetadataService))
|
||||||
}
|
}
|
||||||
if flags.sources.waagent != "" {
|
|
||||||
dss = append(dss, waagent.NewDatasource(flags.sources.waagent))
|
|
||||||
}
|
|
||||||
if flags.sources.procCmdLine {
|
if flags.sources.procCmdLine {
|
||||||
dss = append(dss, proc_cmdline.NewDatasource())
|
dss = append(dss, proc_cmdline.NewDatasource())
|
||||||
}
|
}
|
||||||
|
@@ -1,108 +0,0 @@
|
|||||||
package waagent
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"encoding/xml"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"net"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
)
|
|
||||||
|
|
||||||
type waagent struct {
|
|
||||||
root string
|
|
||||||
readFile func(filename string) ([]byte, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewDatasource(root string) *waagent {
|
|
||||||
return &waagent{root, ioutil.ReadFile}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) IsAvailable() bool {
|
|
||||||
_, err := os.Stat(path.Join(a.root, "provisioned"))
|
|
||||||
return !os.IsNotExist(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) AvailabilityChanges() bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) ConfigRoot() string {
|
|
||||||
return a.root
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) FetchMetadata() ([]byte, error) {
|
|
||||||
metadataBytes, err := a.tryReadFile(path.Join(a.root, "SharedConfig.xml"))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if len(metadataBytes) == 0 {
|
|
||||||
return metadataBytes, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type Instance struct {
|
|
||||||
Id string `xml:"id,attr"`
|
|
||||||
Address string `xml:"address,attr"`
|
|
||||||
InputEndpoints struct {
|
|
||||||
Endpoints []struct {
|
|
||||||
LoadBalancedPublicAddress string `xml:"loadBalancedPublicAddress,attr"`
|
|
||||||
} `xml:"Endpoint"`
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type SharedConfig struct {
|
|
||||||
Incarnation struct {
|
|
||||||
Instance string `xml:"instance,attr"`
|
|
||||||
}
|
|
||||||
Instances struct {
|
|
||||||
Instances []Instance `xml:"Instance"`
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var metadata SharedConfig
|
|
||||||
if err := xml.Unmarshal(metadataBytes, &metadata); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
var instance Instance
|
|
||||||
for _, i := range metadata.Instances.Instances {
|
|
||||||
if i.Id == metadata.Incarnation.Instance {
|
|
||||||
instance = i
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
attrs := map[string]string{
|
|
||||||
"local-ipv4": instance.Address,
|
|
||||||
}
|
|
||||||
for _, e := range instance.InputEndpoints.Endpoints {
|
|
||||||
host, _, err := net.SplitHostPort(e.LoadBalancedPublicAddress)
|
|
||||||
if err == nil {
|
|
||||||
attrs["public-ipv4"] = host
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return json.Marshal(attrs)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) FetchUserdata() ([]byte, error) {
|
|
||||||
return a.tryReadFile(path.Join(a.root, "CustomData"))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) FetchNetworkConfig(filename string) ([]byte, error) {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) Type() string {
|
|
||||||
return "waagent"
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *waagent) tryReadFile(filename string) ([]byte, error) {
|
|
||||||
fmt.Printf("Attempting to read from %q\n", filename)
|
|
||||||
data, err := a.readFile(filename)
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
err = nil
|
|
||||||
}
|
|
||||||
return data, err
|
|
||||||
}
|
|
@@ -1,168 +0,0 @@
|
|||||||
package waagent
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"os"
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
type mockFilesystem map[string][]byte
|
|
||||||
|
|
||||||
func (m mockFilesystem) readFile(filename string) ([]byte, error) {
|
|
||||||
if contents := m[filename]; contents != nil {
|
|
||||||
return contents, nil
|
|
||||||
}
|
|
||||||
return nil, os.ErrNotExist
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFetchMetadata(t *testing.T) {
|
|
||||||
for _, tt := range []struct {
|
|
||||||
root string
|
|
||||||
files mockFilesystem
|
|
||||||
metadata map[string]string
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"/",
|
|
||||||
mockFilesystem{},
|
|
||||||
nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/",
|
|
||||||
mockFilesystem{"/SharedConfig.xml": []byte("")},
|
|
||||||
nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/var/lib/waagent",
|
|
||||||
mockFilesystem{"/var/lib/waagent/SharedConfig.xml": []byte("")},
|
|
||||||
nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/var/lib/waagent",
|
|
||||||
mockFilesystem{"/var/lib/waagent/SharedConfig.xml": []byte(`<?xml version="1.0" encoding="utf-8"?>
|
|
||||||
<SharedConfig version="1.0.0.0" goalStateIncarnation="1">
|
|
||||||
<Deployment name="c8f9e4c9c18948e1bebf57c5685da756" guid="{1d10394f-c741-4a1a-a6bb-278f213c5a5e}" incarnation="0" isNonCancellableTopologyChangeEnabled="false">
|
|
||||||
<Service name="core-test-1" guid="{00000000-0000-0000-0000-000000000000}" />
|
|
||||||
<ServiceInstance name="c8f9e4c9c18948e1bebf57c5685da756.0" guid="{1e202e9a-8ffe-4915-b6ef-4118c9628fda}" />
|
|
||||||
</Deployment>
|
|
||||||
<Incarnation number="1" instance="core-test-1" guid="{8767eb4b-b445-4783-b1f5-6c0beaf41ea0}" />
|
|
||||||
<Role guid="{53ecc81e-257f-fbc9-a53a-8cf1a0a122b4}" name="core-test-1" settleTimeSeconds="0" />
|
|
||||||
<LoadBalancerSettings timeoutSeconds="0" waitLoadBalancerProbeCount="8">
|
|
||||||
<Probes>
|
|
||||||
<Probe name="D41D8CD98F00B204E9800998ECF8427E" />
|
|
||||||
<Probe name="C9DEC1518E1158748FA4B6081A8266DD" />
|
|
||||||
</Probes>
|
|
||||||
</LoadBalancerSettings>
|
|
||||||
<OutputEndpoints>
|
|
||||||
<Endpoint name="core-test-1:openInternalEndpoint" type="SFS">
|
|
||||||
<Target instance="core-test-1" endpoint="openInternalEndpoint" />
|
|
||||||
</Endpoint>
|
|
||||||
</OutputEndpoints>
|
|
||||||
<Instances>
|
|
||||||
<Instance id="core-test-1" address="100.73.202.64">
|
|
||||||
<FaultDomains randomId="0" updateId="0" updateCount="0" />
|
|
||||||
<InputEndpoints>
|
|
||||||
<Endpoint name="openInternalEndpoint" address="100.73.202.64" protocol="any" isPublic="false" enableDirectServerReturn="false" isDirectAddress="false" disableStealthMode="false">
|
|
||||||
<LocalPorts>
|
|
||||||
<LocalPortSelfManaged />
|
|
||||||
</LocalPorts>
|
|
||||||
</Endpoint>
|
|
||||||
<Endpoint name="ssh" address="100.73.202.64:22" protocol="tcp" hostName="core-test-1ContractContract" isPublic="true" loadBalancedPublicAddress="191.239.39.77:22" enableDirectServerReturn="false" isDirectAddress="false" disableStealthMode="false">
|
|
||||||
<LocalPorts>
|
|
||||||
<LocalPortRange from="22" to="22" />
|
|
||||||
</LocalPorts>
|
|
||||||
</Endpoint>
|
|
||||||
</InputEndpoints>
|
|
||||||
</Instance>
|
|
||||||
</Instances>
|
|
||||||
</SharedConfig>`)},
|
|
||||||
map[string]string{
|
|
||||||
"local-ipv4": "100.73.202.64",
|
|
||||||
"public-ipv4": "191.239.39.77",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
} {
|
|
||||||
a := waagent{tt.root, tt.files.readFile}
|
|
||||||
metadataBytes, err := a.FetchMetadata()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("bad error for %q: want %q, got %q", tt, nil, err)
|
|
||||||
}
|
|
||||||
var metadata map[string]string
|
|
||||||
if len(metadataBytes) > 0 {
|
|
||||||
if err := json.Unmarshal(metadataBytes, &metadata); err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(tt.metadata, metadata) {
|
|
||||||
t.Fatalf("bad metadata for %q: want %q, got %q", tt, tt.metadata, metadata)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFetchUserdata(t *testing.T) {
|
|
||||||
for _, tt := range []struct {
|
|
||||||
root string
|
|
||||||
files mockFilesystem
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"/",
|
|
||||||
mockFilesystem{},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/",
|
|
||||||
mockFilesystem{"/CustomData": []byte{}},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/var/lib/waagent/",
|
|
||||||
mockFilesystem{"/var/lib/waagent/CustomData": []byte{}},
|
|
||||||
},
|
|
||||||
} {
|
|
||||||
a := waagent{tt.root, tt.files.readFile}
|
|
||||||
_, err := a.FetchUserdata()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("bad error for %q: want %q, got %q", tt, nil, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestConfigRoot(t *testing.T) {
|
|
||||||
for _, tt := range []struct {
|
|
||||||
root string
|
|
||||||
configRoot string
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"/",
|
|
||||||
"/",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"/var/lib/waagent",
|
|
||||||
"/var/lib/waagent",
|
|
||||||
},
|
|
||||||
} {
|
|
||||||
a := waagent{tt.root, nil}
|
|
||||||
if configRoot := a.ConfigRoot(); configRoot != tt.configRoot {
|
|
||||||
t.Fatalf("bad config root for %q: want %q, got %q", tt, tt.configRoot, configRoot)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNewDatasource(t *testing.T) {
|
|
||||||
for _, tt := range []struct {
|
|
||||||
root string
|
|
||||||
expectRoot string
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
root: "",
|
|
||||||
expectRoot: "",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
root: "/var/lib/waagent",
|
|
||||||
expectRoot: "/var/lib/waagent",
|
|
||||||
},
|
|
||||||
} {
|
|
||||||
service := NewDatasource(tt.root)
|
|
||||||
if service.root != tt.expectRoot {
|
|
||||||
t.Fatalf("bad root (%q): want %q, got %q", tt.root, tt.expectRoot, service.root)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
29
test
29
test
@@ -13,22 +13,19 @@ COVER=${COVER:-"-cover"}
|
|||||||
|
|
||||||
source ./build
|
source ./build
|
||||||
|
|
||||||
declare -a TESTPKGS=(
|
declare -a TESTPKGS=(initialize
|
||||||
datasource
|
system
|
||||||
datasource/configdrive
|
datasource
|
||||||
datasource/file
|
datasource/configdrive
|
||||||
datasource/metadata
|
datasource/file
|
||||||
datasource/metadata/cloudsigma
|
datasource/metadata
|
||||||
datasource/metadata/digitalocean
|
datasource/metadata/cloudsigma
|
||||||
datasource/metadata/ec2
|
datasource/metadata/digitalocean
|
||||||
datasource/proc_cmdline
|
datasource/metadata/ec2
|
||||||
datasource/url
|
datasource/proc_cmdline
|
||||||
datasource/waagent
|
datasource/url
|
||||||
initialize
|
pkg
|
||||||
network
|
network)
|
||||||
pkg
|
|
||||||
system
|
|
||||||
)
|
|
||||||
|
|
||||||
if [ -z "$PKG" ]; then
|
if [ -z "$PKG" ]; then
|
||||||
GOFMTPATH="${TESTPKGS[*]} coreos-cloudinit.go"
|
GOFMTPATH="${TESTPKGS[*]} coreos-cloudinit.go"
|
||||||
|
Reference in New Issue
Block a user