Search in sources :

Example 1 with CreatePolicyRequest

use of software.amazon.awssdk.services.iot.model.CreatePolicyRequest in project aws-sdk-java-v2 by aws.

the class IotControlPlaneIntegrationTest method get_policy_returns_created_policy.

@Test
public void get_policy_returns_created_policy() {
    final CreatePolicyRequest createReq = CreatePolicyRequest.builder().policyName(POLICY_NAME).policyDocument(POLICY_DOC).build();
    CreatePolicyResponse createResult = client.createPolicy(createReq);
    Assert.assertNotNull(createResult.policyArn());
    Assert.assertNotNull(createResult.policyVersionId());
    final GetPolicyVersionRequest request = GetPolicyVersionRequest.builder().policyName(POLICY_NAME).policyVersionId(createResult.policyVersionId()).build();
    GetPolicyVersionResponse result = client.getPolicyVersion(request);
    Assert.assertEquals(createResult.policyArn(), result.policyArn());
    Assert.assertEquals(createResult.policyVersionId(), result.policyVersionId());
}
Also used : CreatePolicyResponse(software.amazon.awssdk.services.iot.model.CreatePolicyResponse) GetPolicyVersionRequest(software.amazon.awssdk.services.iot.model.GetPolicyVersionRequest) GetPolicyVersionResponse(software.amazon.awssdk.services.iot.model.GetPolicyVersionResponse) CreatePolicyRequest(software.amazon.awssdk.services.iot.model.CreatePolicyRequest) Test(org.junit.Test)

Example 2 with CreatePolicyRequest

use of software.amazon.awssdk.services.iot.model.CreatePolicyRequest in project aws-greengrass-nucleus by aws-greengrass.

the class DeviceProvisioningHelper method setupIoTRoleForTes.

/**
 * Create IoT role for using TES.
 *
 * @param roleName       rolaName
 * @param roleAliasName  roleAlias name
 * @param certificateArn certificate arn for the IoT thing
 */
public void setupIoTRoleForTes(String roleName, String roleAliasName, String certificateArn) {
    String roleAliasArn;
    try {
        // Get Role Alias arn
        DescribeRoleAliasRequest describeRoleAliasRequest = DescribeRoleAliasRequest.builder().roleAlias(roleAliasName).build();
        roleAliasArn = iotClient.describeRoleAlias(describeRoleAliasRequest).roleAliasDescription().roleAliasArn();
    } catch (ResourceNotFoundException ranfe) {
        outStream.printf("TES role alias \"%s\" does not exist, creating new alias...%n", roleAliasName);
        // Get IAM role arn in order to attach an alias to it
        String roleArn;
        try {
            GetRoleRequest getRoleRequest = GetRoleRequest.builder().roleName(roleName).build();
            roleArn = iamClient.getRole(getRoleRequest).role().arn();
        } catch (NoSuchEntityException | ResourceNotFoundException rnfe) {
            outStream.printf("TES role \"%s\" does not exist, creating role...%n", roleName);
            CreateRoleRequest createRoleRequest = CreateRoleRequest.builder().roleName(roleName).description("Role for Greengrass IoT things to interact with AWS services using token exchange service").assumeRolePolicyDocument("{\n  \"Version\": \"2012-10-17\",\n" + "  \"Statement\": [\n    {\n      \"Effect\": \"Allow\",\n" + "      \"Principal\": {\n       \"Service\": \"" + tesServiceEndpoints.get(envStage) + "\"\n      },\n      \"Action\": \"sts:AssumeRole\"\n    }\n  ]\n}").build();
            roleArn = iamClient.createRole(createRoleRequest).role().arn();
        }
        CreateRoleAliasRequest createRoleAliasRequest = CreateRoleAliasRequest.builder().roleArn(roleArn).roleAlias(roleAliasName).build();
        roleAliasArn = iotClient.createRoleAlias(createRoleAliasRequest).roleAliasArn();
    }
    // Attach policy role alias to cert
    String iotRolePolicyName = IOT_ROLE_POLICY_NAME_PREFIX + roleAliasName;
    try {
        iotClient.getPolicy(GetPolicyRequest.builder().policyName(iotRolePolicyName).build());
    } catch (ResourceNotFoundException e) {
        outStream.printf("IoT role policy \"%s\" for TES Role alias not exist, creating policy...%n", iotRolePolicyName);
        CreatePolicyRequest createPolicyRequest = CreatePolicyRequest.builder().policyName(iotRolePolicyName).policyDocument("{\n\t\"Version\": \"2012-10-17\",\n\t\"Statement\": {\n" + "\t\t\"Effect\": \"Allow\",\n\t\t\"Action\": \"iot:AssumeRoleWithCertificate\",\n" + "\t\t\"Resource\": \"" + roleAliasArn + "\"\n\t}\n}").build();
        iotClient.createPolicy(createPolicyRequest);
    }
    outStream.println("Attaching TES role policy to IoT thing...");
    AttachPolicyRequest attachPolicyRequest = AttachPolicyRequest.builder().policyName(iotRolePolicyName).target(certificateArn).build();
    iotClient.attachPolicy(attachPolicyRequest);
}
Also used : CreateRoleRequest(software.amazon.awssdk.services.iam.model.CreateRoleRequest) CreatePolicyRequest(software.amazon.awssdk.services.iot.model.CreatePolicyRequest) DescribeRoleAliasRequest(software.amazon.awssdk.services.iot.model.DescribeRoleAliasRequest) CreateRoleAliasRequest(software.amazon.awssdk.services.iot.model.CreateRoleAliasRequest) AttachPolicyRequest(software.amazon.awssdk.services.iot.model.AttachPolicyRequest) GetRoleRequest(software.amazon.awssdk.services.iam.model.GetRoleRequest) ResourceNotFoundException(software.amazon.awssdk.services.iot.model.ResourceNotFoundException)

Aggregations

CreatePolicyRequest (software.amazon.awssdk.services.iot.model.CreatePolicyRequest)2 Test (org.junit.Test)1 CreateRoleRequest (software.amazon.awssdk.services.iam.model.CreateRoleRequest)1 GetRoleRequest (software.amazon.awssdk.services.iam.model.GetRoleRequest)1 AttachPolicyRequest (software.amazon.awssdk.services.iot.model.AttachPolicyRequest)1 CreatePolicyResponse (software.amazon.awssdk.services.iot.model.CreatePolicyResponse)1 CreateRoleAliasRequest (software.amazon.awssdk.services.iot.model.CreateRoleAliasRequest)1 DescribeRoleAliasRequest (software.amazon.awssdk.services.iot.model.DescribeRoleAliasRequest)1 GetPolicyVersionRequest (software.amazon.awssdk.services.iot.model.GetPolicyVersionRequest)1 GetPolicyVersionResponse (software.amazon.awssdk.services.iot.model.GetPolicyVersionResponse)1 ResourceNotFoundException (software.amazon.awssdk.services.iot.model.ResourceNotFoundException)1