116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/goamz/ec2"
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/mitchellh/packer/common/uuid"
|
|
"github.com/mitchellh/packer/packer"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
type StepSecurityGroup struct {
|
|
SecurityGroupIds []string
|
|
SSHPort int
|
|
VpcId string
|
|
|
|
createdGroupId string
|
|
}
|
|
|
|
func (s *StepSecurityGroup) Run(state multistep.StateBag) multistep.StepAction {
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
if len(s.SecurityGroupIds) > 0 {
|
|
log.Printf("Using specified security groups: %v", s.SecurityGroupIds)
|
|
state.Put("securityGroupIds", s.SecurityGroupIds)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
if s.SSHPort == 0 {
|
|
panic("SSHPort must be set to a non-zero value.")
|
|
}
|
|
|
|
// Create the group
|
|
ui.Say("Creating temporary security group for this instance...")
|
|
groupName := fmt.Sprintf("packer %s", uuid.TimeOrderedUUID())
|
|
log.Printf("Temporary group name: %s", groupName)
|
|
group := ec2.SecurityGroup{
|
|
Name: groupName,
|
|
Description: "Temporary group for Packer",
|
|
VpcId: s.VpcId,
|
|
}
|
|
groupResp, err := ec2conn.CreateSecurityGroup(group)
|
|
if err != nil {
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Set the group ID so we can delete it later
|
|
s.createdGroupId = groupResp.Id
|
|
|
|
// Authorize the SSH access
|
|
perms := []ec2.IPPerm{
|
|
ec2.IPPerm{
|
|
Protocol: "tcp",
|
|
FromPort: s.SSHPort,
|
|
ToPort: s.SSHPort,
|
|
SourceIPs: []string{"0.0.0.0/0"},
|
|
},
|
|
}
|
|
|
|
// We loop and retry this a few times because sometimes the security
|
|
// group isn't available immediately because AWS resources are eventaully
|
|
// consistent.
|
|
ui.Say("Authorizing SSH access on the temporary security group...")
|
|
for i := 0; i < 5; i++ {
|
|
_, err = ec2conn.AuthorizeSecurityGroup(groupResp.SecurityGroup, perms)
|
|
if err == nil {
|
|
break
|
|
}
|
|
|
|
log.Printf("Error authorizing. Will sleep and retry. %s", err)
|
|
time.Sleep((time.Duration(i) * time.Second) + 1)
|
|
}
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf("Error creating temporary security group: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Set some state data for use in future steps
|
|
state.Put("securityGroupIds", []string{s.createdGroupId})
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepSecurityGroup) Cleanup(state multistep.StateBag) {
|
|
if s.createdGroupId == "" {
|
|
return
|
|
}
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Deleting temporary security group...")
|
|
|
|
var err error
|
|
for i := 0; i < 5; i++ {
|
|
_, err = ec2conn.DeleteSecurityGroup(ec2.SecurityGroup{Id: s.createdGroupId})
|
|
if err == nil {
|
|
break
|
|
}
|
|
|
|
log.Printf("Error deleting security group: %s", err)
|
|
time.Sleep(5 * time.Second)
|
|
}
|
|
|
|
if err != nil {
|
|
ui.Error(fmt.Sprintf(
|
|
"Error cleaning up security group. Please delete the group manually: %s", s.createdGroupId))
|
|
}
|
|
}
|