Method getEC2WorkloadsPerRegion

aws/workloads.go:313–368

(r string, wg *sync.WaitGroup, semaphore chan struct{}, dataReceiver chan Workload)

}

func (m *WorkloadsModule) getEC2WorkloadsPerRegion(r string, wg *sync.WaitGroup, semaphore chan struct{}, dataReceiver chan Workload) {
	defer func() {
		m.CommandCounter.DecrExecuting()
		m.CommandCounter.IncrComplete()
		wg.Done()

	}()
	semaphore <- struct{}{}
	defer func() {
		<-semaphore
	}()

	// Get EC2 instances
	ec2Instances, err := sdk.CachedEC2DescribeInstances(m.EC2Client, aws.ToString(m.Caller.Account), r)
	if err != nil {
		m.modLog.Error(err)
	}
	for _, instance := range ec2Instances {
		var profileArn, profileName, name string

		// The name is in a tag so we have to do this to grab the value from the right tag
		for _, tag := range instance.Tags {
			if *tag.Key == "Name" {
				name = *tag.Value
			}
		}
		if name == "" {
			name = aws.ToString(instance.InstanceId)
		}

		if instance.IamInstanceProfile != nil {
			profileArn = aws.ToString(instance.IamInstanceProfile.Arn)

			// Extracting instance profile name from ARN
			profileName = strings.Split(profileArn, "/")[len(strings.Split(profileArn, "/"))-1]

			// Describe the IAM instance profile
			profileOutput, err := sdk.CachedIamGetInstanceProfile(m.IAMClient, aws.ToString(m.Caller.Account), profileName)
			if err != nil {
				m.modLog.Errorf("failed to get instance profile for %s, %v", profileArn, err)
				continue
			}

			for _, role := range profileOutput.Roles {
				dataReceiver <- Workload{
					AWSService: "EC2",
					Region:     r,
					Type:       "instance",
					Name:       name,
					Arn:        fmt.Sprintf("arn:aws:ec2:%s:%s:instance/%s", r, aws.ToString(m.Caller.Account), aws.ToString(instance.InstanceId)),
					Role:       aws.ToString(role.Arn),
				}
			}
		}
	}
}

func (m *WorkloadsModule) getECSWorkloadsPerRegion(r string, wg *sync.WaitGroup, semaphore chan struct{}, dataReceiver chan Workload) {