Add support for sourceOwner policy (#113)

kibertoad · web-flow · commit a47c67c30b5f · 2024-03-31T10:24:21.000+03:00
diff --git a/packages/sns/lib/sns/AbstractSnsService.ts b/packages/sns/lib/sns/AbstractSnsService.ts
@@ -71,6 +71,7 @@ export type ExistingSNSOptionsMultiSchema<
 
 export type ExtraSNSCreationParams = {
   queueUrlsWithSubscribePermissionsPrefix?: string
+  allowedSourceOwner?: string
 }
 
 export type SNSCreationConfig = {
diff --git a/packages/sns/lib/utils/snsAttributeUtils.spec.ts b/packages/sns/lib/utils/snsAttributeUtils.spec.ts
@@ -9,14 +9,36 @@ import { generateFilterAttributes, generateTopicSubscriptionPolicy } from './sns
 
 describe('snsAttributeUtils', () => {
   describe('generateTopicSubscriptionPolicy', () => {
-    it('resolves policy', () => {
-      const resolvedPolicy = generateTopicSubscriptionPolicy(
-        'arn:aws:sns:eu-central-1:632374391739:test-sns-some-service',
-        'arn:aws:sqs:eu-central-1:632374391739:test-sqs-*',
+    it('resolves policy for both params', () => {
+      const resolvedPolicy = generateTopicSubscriptionPolicy({
+        topicArn: 'arn:aws:sns:eu-central-1:632374391739:test-sns-some-service',
+        allowedSqsQueueUrlPrefix: 'arn:aws:sqs:eu-central-1:632374391739:test-sqs-*',
+        allowedSourceOwner: '111111111111',
+      })
+
+      expect(resolvedPolicy).toBe(
+        `{"Version":"2012-10-17","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"arn:aws:sns:eu-central-1:632374391739:test-sns-some-service","Condition":{"StringEquals":{"AWS:SourceOwner": "111111111111"},"StringLike":{"sns:Endpoint":"arn:aws:sqs:eu-central-1:632374391739:test-sqs-*"}}}]}`,
+      )
+    })
+
+    it('resolves policy for one param', () => {
+      const resolvedPolicy = generateTopicSubscriptionPolicy({
+        topicArn: 'arn:aws:sns:eu-central-1:632374391739:test-sns-some-service',
+        allowedSourceOwner: '111111111111',
+      })
+
+      expect(resolvedPolicy).toBe(
+        `{"Version":"2012-10-17","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"arn:aws:sns:eu-central-1:632374391739:test-sns-some-service","Condition":{"StringEquals":{"AWS:SourceOwner": "111111111111"}}}]}`,
       )
+    })
+
+    it('resolves policy for zero params', () => {
+      const resolvedPolicy = generateTopicSubscriptionPolicy({
+        topicArn: 'arn:aws:sns:eu-central-1:632374391739:test-sns-some-service',
+      })
 
       expect(resolvedPolicy).toBe(
-        `{"Version":"2012-10-17","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"arn:aws:sns:eu-central-1:632374391739:test-sns-some-service","Condition":{"StringLike":{"sns:Endpoint":"arn:aws:sqs:eu-central-1:632374391739:test-sqs-*"}}}]}`,
+        `{"Version":"2012-10-17","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"arn:aws:sns:eu-central-1:632374391739:test-sns-some-service","Condition":{}}]}`,
       )
     })
   })
diff --git a/packages/sns/lib/utils/snsAttributeUtils.ts b/packages/sns/lib/utils/snsAttributeUtils.ts
@@ -3,11 +3,23 @@ import type { ZodSchema } from 'zod'
 // See https://docs.aws.amazon.com/IAM/latest/UserGuide/reference_policies_elements_version.html
 const POLICY_VERSION = '2012-10-17'
 
-export function generateTopicSubscriptionPolicy(
-  topicArn: string,
-  supportedSqsQueueUrlPrefix: string,
-) {
-  return `{"Version":"${POLICY_VERSION}","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"${topicArn}","Condition":{"StringLike":{"sns:Endpoint":"${supportedSqsQueueUrlPrefix}"}}}]}`
+export type TopicSubscriptionPolicyParams = {
+  topicArn: string
+  allowedSqsQueueUrlPrefix?: string
+  allowedSourceOwner?: string
+}
+
+export function generateTopicSubscriptionPolicy(params: TopicSubscriptionPolicyParams) {
+  const sourceOwnerFragment = params.allowedSourceOwner
+    ? `"StringEquals":{"AWS:SourceOwner": "${params.allowedSourceOwner}"}`
+    : ''
+  const supportedSqsQueueUrlPrefixFragment = params.allowedSqsQueueUrlPrefix
+    ? `"StringLike":{"sns:Endpoint":"${params.allowedSqsQueueUrlPrefix}"}`
+    : ''
+  const commaFragment =
+    sourceOwnerFragment.length > 0 && supportedSqsQueueUrlPrefixFragment.length > 0 ? ',' : ''
+
+  return `{"Version":"${POLICY_VERSION}","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"${params.topicArn}","Condition":{${sourceOwnerFragment}${commaFragment}${supportedSqsQueueUrlPrefixFragment}}}]}`
 }
 
 export function generateFilterAttributes(
diff --git a/packages/sns/lib/utils/snsInitter.ts b/packages/sns/lib/utils/snsInitter.ts
@@ -47,6 +47,7 @@ export async function initSnsSqs(
         updateAttributesIfExists: creationConfig.updateAttributesIfExists,
         queueUrlsWithSubscribePermissionsPrefix:
           creationConfig.queueUrlsWithSubscribePermissionsPrefix,
+        allowedSourceOwner: creationConfig.allowedSourceOwner,
         topicArnsWithPublishPermissionsPrefix: creationConfig.topicArnsWithPublishPermissionsPrefix,
         logger: extraParams?.logger,
       },
@@ -171,6 +172,7 @@ export async function initSns(
   }
   const topicArn = await assertTopic(snsClient, creationConfig.topic, {
     queueUrlsWithSubscribePermissionsPrefix: creationConfig.queueUrlsWithSubscribePermissionsPrefix,
+    allowedSourceOwner: creationConfig.allowedSourceOwner,
   })
   return {
     topicArn,
diff --git a/packages/sns/lib/utils/snsSubscriber.ts b/packages/sns/lib/utils/snsSubscriber.ts
@@ -25,6 +25,7 @@ export async function subscribeToTopic(
 ) {
   const topicArn = await assertTopic(snsClient, topicConfiguration, {
     queueUrlsWithSubscribePermissionsPrefix: extraParams?.queueUrlsWithSubscribePermissionsPrefix,
+    allowedSourceOwner: extraParams?.allowedSourceOwner,
   })
   const { queueUrl, queueArn } = await assertQueue(sqsClient, queueConfiguration, {
     topicArnsWithPublishPermissionsPrefix: extraParams?.topicArnsWithPublishPermissionsPrefix,
diff --git a/packages/sns/lib/utils/snsUtils.ts b/packages/sns/lib/utils/snsUtils.ts
@@ -85,14 +85,15 @@ export async function assertTopic(
   }
   const topicArn = response.TopicArn
 
-  if (extraParams?.queueUrlsWithSubscribePermissionsPrefix) {
+  if (extraParams?.queueUrlsWithSubscribePermissionsPrefix || extraParams?.allowedSourceOwner) {
     const setTopicAttributesCommand = new SetTopicAttributesCommand({
       TopicArn: topicArn,
       AttributeName: 'Policy',
-      AttributeValue: generateTopicSubscriptionPolicy(
+      AttributeValue: generateTopicSubscriptionPolicy({
         topicArn,
-        extraParams.queueUrlsWithSubscribePermissionsPrefix,
-      ),
+        allowedSqsQueueUrlPrefix: extraParams.queueUrlsWithSubscribePermissionsPrefix,
+        allowedSourceOwner: extraParams.allowedSourceOwner,
+      }),
     })
     await snsClient.send(setTopicAttributesCommand)
   }
diff --git a/packages/sns/test/publishers/SnsPermissionPublisher.spec.ts b/packages/sns/test/publishers/SnsPermissionPublisher.spec.ts
@@ -50,6 +50,26 @@ describe('SNSPermissionPublisher', () => {
       )
     })
 
+    it('sets correct policy when two policy fields are set', async () => {
+      const newPublisher = new SnsPermissionPublisherMonoSchema(diContainer.cradle, {
+        creationConfig: {
+          topic: {
+            Name: 'policy-topic',
+          },
+          queueUrlsWithSubscribePermissionsPrefix: 'dummy*',
+          allowedSourceOwner: '111111111111',
+        },
+      })
+
+      await newPublisher.init()
+
+      const topic = await getTopicAttributes(snsClient, newPublisher.topicArn)
+
+      expect(topic.result?.attributes?.Policy).toBe(
+        `{"Version":"2012-10-17","Id":"__default_policy_ID","Statement":[{"Sid":"AllowSQSSubscription","Effect":"Allow","Principal":{"AWS":"*"},"Action":["sns:Subscribe"],"Resource":"arn:aws:sns:eu-west-1:000000000000:policy-topic","Condition":{"StringEquals":{"AWS:SourceOwner":"111111111111"},"StringLike":{"sns:Endpoint":"dummy*"}}}]}`,
+      )
+    })
+
     // FixMe https://github.com/localstack/localstack/issues/9306
     it.skip('throws an error when invalid queue locator is passed', async () => {
       const newPublisher = new SnsPermissionPublisherMonoSchema(diContainer.cradle, {

Original file line number	Diff line number	Diff line change
`@@ -71,6 +71,7 @@ export type ExistingSNSOptionsMultiSchema<`
`71`	`71`
`72`	`72`	`export type ExtraSNSCreationParams = {`
`73`	`73`	`queueUrlsWithSubscribePermissionsPrefix?: string`
	`74`	`+ allowedSourceOwner?: string`
`74`	`75`	`}`
`75`	`76`
`76`	`77`	`export type SNSCreationConfig = {`