-
Notifications
You must be signed in to change notification settings - Fork 50
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
e4cdd05
commit 4c8a858
Showing
15 changed files
with
1,952 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| package efs | ||
|
|
||
| import ( | ||
| "github.com/akutz/gofig" | ||
| ) | ||
|
|
||
| const ( | ||
| // Name is the provider's name. | ||
| Name = "efs" | ||
| ) | ||
|
|
||
| func init() { | ||
| registerConfig() | ||
| } | ||
|
|
||
| func registerConfig() { | ||
| r := gofig.NewRegistration("EFS") | ||
| r.Key(gofig.String, "", "", "", "efs.accessKey") | ||
| r.Key(gofig.String, "", "", "", "efs.secretKey") | ||
| r.Key(gofig.String, "", "", "Comma separated security group ids", "efs.securityGroups") | ||
| r.Key(gofig.String, "", "", "AWS region", "efs.region") | ||
| r.Key(gofig.String, "", "", "Tag prefix for EFS naming", "efs.tag") | ||
| gofig.Register(r) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,196 @@ | ||
| package executor | ||
|
|
||
| import ( | ||
| "bufio" | ||
| "fmt" | ||
| "io" | ||
| "io/ioutil" | ||
| "net/http" | ||
| "os" | ||
| "strings" | ||
|
|
||
| "github.com/akutz/gofig" | ||
| "github.com/akutz/goof" | ||
|
|
||
| "github.com/emccode/libstorage/api/registry" | ||
| "github.com/emccode/libstorage/api/types" | ||
| "github.com/emccode/libstorage/drivers/storage/efs" | ||
| ) | ||
|
|
||
| // driver is the storage executor for the efs storage driver. | ||
| type driver struct { | ||
| config gofig.Config | ||
| subnetResolver SubnetResolver | ||
| } | ||
|
|
||
| const ( | ||
| idDelimiter = "/" | ||
| mountinfoFormat = "%d %d %d:%d %s %s %s %s" | ||
| ) | ||
|
|
||
| func init() { | ||
| registry.RegisterStorageExecutor(efs.Name, newDriver) | ||
| } | ||
|
|
||
| func newDriver() types.StorageExecutor { | ||
| return &driver{ | ||
| subnetResolver: NewAwsVpcSubnetResolver(), | ||
| } | ||
| } | ||
|
|
||
| func (d *driver) Init(ctx types.Context, config gofig.Config) error { | ||
| d.config = config | ||
| return nil | ||
| } | ||
|
|
||
| func (d *driver) Name() string { | ||
| return efs.Name | ||
| } | ||
|
|
||
| // InstanceID returns the local instance ID for the test | ||
| func InstanceID() (*types.InstanceID, error) { | ||
| return newDriver().InstanceID(nil, nil) | ||
| } | ||
|
|
||
| // InstanceID returns the aws instance configuration | ||
| func (d *driver) InstanceID( | ||
| ctx types.Context, | ||
| opts types.Store) (*types.InstanceID, error) { | ||
|
|
||
| subnetID, err := d.subnetResolver.ResolveSubnet() | ||
| if err != nil { | ||
| return nil, goof.WithError("no ec2metadata subnet id", err) | ||
| } | ||
|
|
||
| iid := &types.InstanceID{Driver: efs.Name} | ||
| if err := iid.MarshalMetadata(subnetID); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return iid, nil | ||
| } | ||
|
|
||
| func (d *driver) NextDevice( | ||
| ctx types.Context, | ||
| opts types.Store) (string, error) { | ||
| return "", types.ErrNotImplemented | ||
| } | ||
|
|
||
| func (d *driver) LocalDevices( | ||
| ctx types.Context, | ||
| opts *types.LocalDevicesOpts) (*types.LocalDevices, error) { | ||
|
|
||
| mtt, err := parseMountTable() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| idmnt := make(map[string]string) | ||
| for _, mt := range mtt { | ||
| idmnt[mt.Source] = mt.MountPoint | ||
| } | ||
|
|
||
| return &types.LocalDevices{ | ||
| Driver: efs.Name, | ||
| DeviceMap: idmnt, | ||
| }, nil | ||
| } | ||
|
|
||
| func parseMountTable() ([]*types.MountInfo, error) { | ||
| f, err := os.Open("/proc/self/mountinfo") | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| defer f.Close() | ||
|
|
||
| return parseInfoFile(f) | ||
| } | ||
|
|
||
| func parseInfoFile(r io.Reader) ([]*types.MountInfo, error) { | ||
| var ( | ||
| s = bufio.NewScanner(r) | ||
| out = []*types.MountInfo{} | ||
| ) | ||
|
|
||
| for s.Scan() { | ||
| if err := s.Err(); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| var ( | ||
| p = &types.MountInfo{} | ||
| text = s.Text() | ||
| optionalFields string | ||
| ) | ||
|
|
||
| if _, err := fmt.Sscanf(text, mountinfoFormat, | ||
| &p.ID, &p.Parent, &p.Major, &p.Minor, | ||
| &p.Root, &p.MountPoint, &p.Opts, &optionalFields); err != nil { | ||
| return nil, fmt.Errorf("Scanning '%s' failed: %s", text, err) | ||
| } | ||
| // Safe as mountinfo encodes mountpoints with spaces as \040. | ||
| index := strings.Index(text, " - ") | ||
| postSeparatorFields := strings.Fields(text[index+3:]) | ||
| if len(postSeparatorFields) < 3 { | ||
| return nil, fmt.Errorf("Error found less than 3 fields post '-' in %q", text) | ||
| } | ||
|
|
||
| if optionalFields != "-" { | ||
| p.Optional = optionalFields | ||
| } | ||
|
|
||
| p.FSType = postSeparatorFields[0] | ||
| p.Source = postSeparatorFields[1] | ||
| p.VFSOpts = strings.Join(postSeparatorFields[2:], " ") | ||
| out = append(out, p) | ||
| } | ||
| return out, nil | ||
| } | ||
|
|
||
| // SubnetResolver defines interface that can resolve subnet from environment | ||
| type SubnetResolver interface { | ||
| ResolveSubnet() (string, error) | ||
| } | ||
|
|
||
| // AwsVpcSubnetResolver is thin interface that resolves instance subnet from | ||
| // ec2metadata service. This helper is used instead of bringing AWS SDK to | ||
| // executor on purpose to keep executor dependencies minimal. | ||
| type AwsVpcSubnetResolver struct { | ||
| ec2MetadataIPAddress string | ||
| } | ||
|
|
||
| // ResolveSubnet determines VPC subnet id on running AWS instance | ||
| func (r *AwsVpcSubnetResolver) ResolveSubnet() (string, error) { | ||
| resp, err := http.Get(r.getURL("mac")) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| mac, err := ioutil.ReadAll(resp.Body) | ||
| resp.Body.Close() | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| resp, err = http.Get(r.getURL(fmt.Sprintf("network/interfaces/macs/%s/subnet-id", mac))) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| subnetID, err := ioutil.ReadAll(resp.Body) | ||
| resp.Body.Close() | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| return string(subnetID), nil | ||
| } | ||
|
|
||
| func (r *AwsVpcSubnetResolver) getURL(path string) string { | ||
| return fmt.Sprintf("http://%s/latest/meta-data/%s", r.ec2MetadataIPAddress, path) | ||
| } | ||
|
|
||
| // NewAwsVpcSubnetResolver creates AwsVpcSubnetResolver for default AWS endpoint | ||
| func NewAwsVpcSubnetResolver() *AwsVpcSubnetResolver { | ||
| return &AwsVpcSubnetResolver{ | ||
| ec2MetadataIPAddress: "169.254.169.254", | ||
| } | ||
| } |
Oops, something went wrong.