Check for EBS being nil before assigning it

2015-04-15 11:49:29 -07:00 · 2015-04-15 11:49:29 -07:00 · 33b4f5cc0a
parent e99cd56b6c
commit 33b4f5cc0a
1 changed files with 5 additions and 1 deletions
--- a/builder/amazon/chroot/step_register_ami.go
+++ b/builder/amazon/chroot/step_register_ami.go
@ -25,7 +25,11 @@ func (s *StepRegisterAMI) Run(state multistep.StateBag) multistep.StepAction {
 	for i, device := range image.BlockDeviceMappings {
 		newDevice := device
 		if newDevice.DeviceName == image.RootDeviceName {
-			newDevice.EBS.SnapshotID = &snapshotId
+			if newDevice.EBS != nil {
+				newDevice.EBS.SnapshotID = &snapshotId
+			} else {
+				newDevice.EBS = &ec2.EBSBlockDevice{SnapshotID: &snapshotId}
+			}
 		}

 		blockDevices[i] = newDevice