1cc3cdf7b5
Set SriovNetSupport to "simple". As of February 2017, this applies to C3, C4, D2, I2, R3, and M4 (excluding m4.16xlarge). Set EnaSupport to true. As of February 2017, this applies to C5, I3, P2, R4, X1, and m4.16xlarge.
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
// StepSourceAMIInfo extracts critical information from the source AMI
|
|
// that is used throughout the AMI creation process.
|
|
//
|
|
// Produces:
|
|
// source_image *ec2.Image - the source AMI info
|
|
type StepSourceAMIInfo struct {
|
|
SourceAmi string
|
|
EnhancedNetworking bool
|
|
AmiFilters AmiFilterOptions
|
|
}
|
|
|
|
// Build a slice of AMI filter options from the filters provided.
|
|
func buildAmiFilters(input map[*string]*string) []*ec2.Filter {
|
|
var filters []*ec2.Filter
|
|
for k, v := range input {
|
|
filters = append(filters, &ec2.Filter{
|
|
Name: k,
|
|
Values: []*string{v},
|
|
})
|
|
}
|
|
return filters
|
|
}
|
|
|
|
type imageSort []*ec2.Image
|
|
|
|
func (a imageSort) Len() int { return len(a) }
|
|
func (a imageSort) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
func (a imageSort) Less(i, j int) bool {
|
|
itime, _ := time.Parse(time.RFC3339, *a[i].CreationDate)
|
|
jtime, _ := time.Parse(time.RFC3339, *a[j].CreationDate)
|
|
return itime.Unix() < jtime.Unix()
|
|
}
|
|
|
|
// Returns the most recent AMI out of a slice of images.
|
|
func mostRecentAmi(images []*ec2.Image) *ec2.Image {
|
|
sortedImages := images
|
|
sort.Sort(imageSort(sortedImages))
|
|
return sortedImages[len(sortedImages)-1]
|
|
}
|
|
|
|
func (s *StepSourceAMIInfo) Run(state multistep.StateBag) multistep.StepAction {
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
params := &ec2.DescribeImagesInput{}
|
|
|
|
if s.SourceAmi != "" {
|
|
params.ImageIds = []*string{&s.SourceAmi}
|
|
}
|
|
|
|
// We have filters to apply
|
|
if len(s.AmiFilters.Filters) > 0 {
|
|
params.Filters = buildAmiFilters(s.AmiFilters.Filters)
|
|
}
|
|
if len(s.AmiFilters.Owners) > 0 {
|
|
params.Owners = s.AmiFilters.Owners
|
|
}
|
|
|
|
log.Printf("Using AMI Filters %v", params)
|
|
imageResp, err := ec2conn.DescribeImages(params)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error querying AMI: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if len(imageResp.Images) == 0 {
|
|
err := fmt.Errorf("No AMI was found matching filters: %v", params)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if len(imageResp.Images) > 1 && s.AmiFilters.MostRecent == false {
|
|
err := fmt.Errorf("Your query returned more than one result. Please try a more specific search, or set most_recent to true.")
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
var image *ec2.Image
|
|
if s.AmiFilters.MostRecent {
|
|
image = mostRecentAmi(imageResp.Images)
|
|
} else {
|
|
image = imageResp.Images[0]
|
|
}
|
|
|
|
ui.Message(fmt.Sprintf("Found Image ID: %s", *image.ImageId))
|
|
|
|
// Enhanced Networking can only be enabled on HVM AMIs.
|
|
// See http://goo.gl/icuXh5
|
|
if s.EnhancedNetworking && *image.VirtualizationType != "hvm" {
|
|
err := fmt.Errorf("Cannot enable enhanced networking, source AMI '%s' is not HVM", s.SourceAmi)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
state.Put("source_image", image)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepSourceAMIInfo) Cleanup(multistep.StateBag) {}
|