2013-07-25 01:19:04 -04:00
|
|
|
package instance
|
|
|
|
|
|
|
|
import (
|
2018-01-22 18:32:33 -05:00
|
|
|
"context"
|
2013-07-25 01:19:04 -04:00
|
|
|
"fmt"
|
2014-06-04 17:58:11 -04:00
|
|
|
|
2015-06-26 11:43:13 -04:00
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
2015-06-03 17:13:52 -04:00
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
2017-04-04 16:39:01 -04:00
|
|
|
awscommon "github.com/hashicorp/packer/builder/amazon/common"
|
2019-10-09 19:59:51 -04:00
|
|
|
"github.com/hashicorp/packer/common/random"
|
2019-08-22 16:17:35 -04:00
|
|
|
confighelper "github.com/hashicorp/packer/helper/config"
|
2018-01-19 19:18:44 -05:00
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
2017-04-04 16:39:01 -04:00
|
|
|
"github.com/hashicorp/packer/packer"
|
2013-07-25 01:19:04 -04:00
|
|
|
)
|
|
|
|
|
2017-08-28 12:18:23 -04:00
|
|
|
type StepRegisterAMI struct {
|
2019-08-22 16:17:35 -04:00
|
|
|
EnableAMIENASupport confighelper.Trilean
|
2017-08-28 12:18:23 -04:00
|
|
|
EnableAMISriovNetSupport bool
|
2019-10-09 19:59:51 -04:00
|
|
|
AMISkipBuildRegion bool
|
2017-08-28 12:18:23 -04:00
|
|
|
}
|
2013-07-25 01:19:04 -04:00
|
|
|
|
2018-06-01 19:17:30 -04:00
|
|
|
func (s *StepRegisterAMI) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
2013-08-31 16:03:13 -04:00
|
|
|
config := state.Get("config").(*Config)
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
|
|
manifestPath := state.Get("remote_manifest_path").(string)
|
|
|
|
ui := state.Get("ui").(packer.Ui)
|
2013-07-25 01:19:04 -04:00
|
|
|
|
|
|
|
ui.Say("Registering the AMI...")
|
2019-10-09 19:59:51 -04:00
|
|
|
|
|
|
|
// Create the image
|
|
|
|
amiName := config.AMIName
|
|
|
|
state.Put("intermediary_image", false)
|
|
|
|
if config.AMIEncryptBootVolume.True() || s.AMISkipBuildRegion {
|
|
|
|
state.Put("intermediary_image", true)
|
|
|
|
|
|
|
|
// From AWS SDK docs: You can encrypt a copy of an unencrypted snapshot,
|
|
|
|
// but you cannot use it to create an unencrypted copy of an encrypted
|
|
|
|
// snapshot. Your default CMK for EBS is used unless you specify a
|
|
|
|
// non-default key using KmsKeyId.
|
|
|
|
|
|
|
|
// If encrypt_boot is nil or true, we need to create a temporary image
|
|
|
|
// so that in step_region_copy, we can copy it with the correct
|
|
|
|
// encryption
|
|
|
|
amiName = random.AlphaNum(7)
|
|
|
|
}
|
|
|
|
|
2015-04-05 17:58:48 -04:00
|
|
|
registerOpts := &ec2.RegisterImageInput{
|
|
|
|
ImageLocation: &manifestPath,
|
2019-10-09 19:59:51 -04:00
|
|
|
Name: aws.String(amiName),
|
2019-06-18 06:44:24 -04:00
|
|
|
BlockDeviceMappings: config.AMIMappings.BuildEC2BlockDeviceMappings(),
|
2015-06-26 11:43:13 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
if config.AMIVirtType != "" {
|
|
|
|
registerOpts.VirtualizationType = aws.String(config.AMIVirtType)
|
2013-07-25 01:19:04 -04:00
|
|
|
}
|
2013-07-25 01:45:55 -04:00
|
|
|
|
2017-08-28 12:18:23 -04:00
|
|
|
if s.EnableAMISriovNetSupport {
|
Always set both SRIOV and ENA when Enhanced Networking is enabled
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.
2017-02-21 20:46:16 -05:00
|
|
|
// Set SriovNetSupport to "simple". See http://goo.gl/icuXh5
|
|
|
|
// As of February 2017, this applies to C3, C4, D2, I2, R3, and M4 (excluding m4.16xlarge)
|
2015-08-17 20:44:01 -04:00
|
|
|
registerOpts.SriovNetSupport = aws.String("simple")
|
2017-08-28 12:18:23 -04:00
|
|
|
}
|
2019-08-22 16:17:35 -04:00
|
|
|
if s.EnableAMIENASupport.True() {
|
2017-08-28 12:18:23 -04:00
|
|
|
// Set EnaSupport to true
|
Always set both SRIOV and ENA when Enhanced Networking is enabled
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.
2017-02-21 20:46:16 -05:00
|
|
|
// As of February 2017, this applies to C5, I3, P2, R4, X1, and m4.16xlarge
|
|
|
|
registerOpts.EnaSupport = aws.Bool(true)
|
2014-06-04 17:58:11 -04:00
|
|
|
}
|
|
|
|
|
2013-07-25 01:45:55 -04:00
|
|
|
registerResp, err := ec2conn.RegisterImage(registerOpts)
|
|
|
|
if err != nil {
|
2013-08-31 16:03:13 -04:00
|
|
|
state.Put("error", fmt.Errorf("Error registering AMI: %s", err))
|
|
|
|
ui.Error(state.Get("error").(error).Error())
|
2013-07-25 01:19:04 -04:00
|
|
|
return multistep.ActionHalt
|
|
|
|
}
|
|
|
|
|
2013-07-25 01:45:55 -04:00
|
|
|
// Set the AMI ID in the state
|
2015-08-17 20:44:01 -04:00
|
|
|
ui.Say(fmt.Sprintf("AMI: %s", *registerResp.ImageId))
|
2013-07-25 01:45:55 -04:00
|
|
|
amis := make(map[string]string)
|
2015-08-17 20:44:01 -04:00
|
|
|
amis[*ec2conn.Config.Region] = *registerResp.ImageId
|
2013-08-31 16:03:13 -04:00
|
|
|
state.Put("amis", amis)
|
2013-07-25 01:19:04 -04:00
|
|
|
|
2013-07-25 01:56:37 -04:00
|
|
|
// Wait for the image to become ready
|
|
|
|
ui.Say("Waiting for AMI to become ready...")
|
2018-06-01 19:17:30 -04:00
|
|
|
if err := awscommon.WaitUntilAMIAvailable(ctx, ec2conn, *registerResp.ImageId); err != nil {
|
2013-07-25 01:56:37 -04:00
|
|
|
err := fmt.Errorf("Error waiting for AMI: %s", err)
|
2013-08-31 16:03:13 -04:00
|
|
|
state.Put("error", err)
|
2013-07-25 01:56:37 -04:00
|
|
|
ui.Error(err.Error())
|
|
|
|
return multistep.ActionHalt
|
|
|
|
}
|
|
|
|
|
2017-01-06 09:17:02 -05:00
|
|
|
state.Put("snapshots", map[string][]string{})
|
|
|
|
|
2013-07-25 01:19:04 -04:00
|
|
|
return multistep.ActionContinue
|
|
|
|
}
|
|
|
|
|
2013-08-31 16:03:13 -04:00
|
|
|
func (s *StepRegisterAMI) Cleanup(multistep.StateBag) {}
|