dash/weirdAAL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #44 from carnal0wnage/sns

Browse Source 
Sns
This commit is contained in:
Chris Gates
2018-05-20 08:49:12 -04:00
committed by GitHub
parent 9742a6fb99 1ed839cfda
commit cf89388c82
2 changed files with 143 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
libs/sns.py Normal file
View File

@@ -0,0 +1,97 @@
'''
utilities for working with SNS
'''
import boto3
import botocore
regions = ['us-east-1', 'us-east-2', 'us-west-1', 'us-west-2', 'ca-central-1', 'eu-central-1', 'eu-west-1', 'eu-west-2', 'ap-northeast-1', 'ap-northeast-2', 'ap-southeast-1', 'ap-southeast-2']
session = boto3.Session()
credentials = session.get_credentials()
AWS_ACCESS_KEY_ID = credentials.access_key
def list_sns_topics():
title = "SNS Topics"
print(title)
print("-" * len(title))
try:
for region in regions:
client = boto3.client('sns', region_name=region)
topics = client.list_topics()
print(region)
print("=" * len(region))
if topics['Topics']:
for topic in topics['Topics']:
print(topic)
except botocore.exceptions.ClientError as e:
if e.response['Error']['Code'] == 'InvalidClientTokenId':
sys.exit("The AWS KEY IS INVALID. Exiting")
if e.response['Error']['Code'] == 'AccessDenied':
print('{} : Is NOT a root key' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'SubscriptionRequiredException':
print('{} : Has permissions but isnt signed up for service - usually means you have a root account' .format(AWS_ACCESS_KEY_ID))
else:
print("Unexpected error: {}" .format(e))
except KeyboardInterrupt:
print("CTRL-C received, exiting...")
def list_sns_subscribers(topic,region):
try:
client = boto3.client('sns', region_name=region)
result = client.list_subscriptions_by_topic(TopicArn=topic)
subscriptions = result['Subscriptions']
for sub in subscriptions:
print("Subscription Arn: {}".format(sub['SubscriptionArn']))
print("Protocol: {}".format(sub['Protocol']))
print("Endpoint: {}".format(sub['Endpoint']))
except botocore.exceptions.ClientError as e:
if e.response['Error']['Code'] == 'InvalidClientTokenId':
sys.exit("The AWS KEY IS INVALID. Exiting")
if e.response['Error']['Code'] == 'AccessDenied':
print('{} : Is NOT a root key' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'SubscriptionRequiredException':
print('{} : Has permissions but isnt signed up for service - usually means you have a root account' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'InvalidParameter':
print('The region you provided ({}) is invalid for the Topic ARN. Are you sure this topic exists in this region?'.format(region))
else:
print("Unexpected error: {}" .format(e))
except KeyboardInterrupt:
print("CTRL-C received, exiting...")
def delete_sns_topic(topic, region):
try:
client = boto3.client('sns', region_name=region)
action = client.delete_topic(TopicArn=topic)
print("Deleted Topic: {}".format(topic))
except botocore.exceptions.ClientError as e:
if e.response['Error']['Code'] == 'InvalidClientTokenId':
sys.exit("The AWS KEY IS INVALID. Exiting")
if e.response['Error']['Code'] == 'AccessDenied':
print('{} : Is NOT a root key' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'SubscriptionRequiredException':
print('{} : Has permissions but isnt signed up for service - usually means you have a root account' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'InvalidParameter':
print('The region you provided ({}) is invalid for the Topic ARN. Are you sure this topic exists in this region?'.format(region))
else:
print("Unexpected error: {}" .format(e))
except KeyboardInterrupt:
print("CTRL-C received, exiting...")
def delete_sns_subscriber(endpoint, region):
try:
client = boto3.client('sns', region_name=region)
action = client.delete_endpoint(EndpointArn=endpoint)
except botocore.exceptions.ClientError as e:
if e.response['Error']['Code'] == 'InvalidClientTokenId':
sys.exit("The AWS KEY IS INVALID. Exiting")
if e.response['Error']['Code'] == 'AccessDenied':
print('{} : Is NOT a root key' .format(AWS_ACCESS_KEY_ID))
elif e.response['Error']['Code'] == 'SubscriptionRequiredException':
print('{} : Has permissions but isnt signed up for service - usually means you have a root account' .format(AWS_ACCESS_KEY_ID))
#elif e.response['Error']['Code'] == 'InvalidParameter':
# print('The region you provided ({}) is invalid for the Subscriber endpoint. Are you sure this subscriber exists in this region?'.format(region))
else:
print("Unexpected error: {}" .format(e))
except KeyboardInterrupt:
print("CTRL-C received, exiting...")

46
modules/sns.py Normal file
View File

@@ -0,0 +1,46 @@
'''
SNS module
'''
from libs.sns import *
def module_sns_list_topics():
'''
SNS list all topics
python3 weirdAAL.py -m sns_list_topics -t demo
'''
list_sns_topics()
def module_sns_list_subscribers(*args):
'''
SNS list subscribers for a topic. Takes two arguments - the topic arn and then the region.
python3 weirdAAL.py -m sns_list_subscribers -a arn:aws:sns:us-east-1:123456789123:sometopic,us-east-1
'''
try:
if args[0][0] and args[0][1]:
list_sns_subscribers(args[0][0], args[0][1])
except IndexError:
print("Please provide a topic arn *AND* region, ex: -a arn:aws:sns:us-east-1:123456789123:sometopic,us-east-1")
def module_sns_delete_topic(*args):
'''
SNS delete a topic. Takes two arguments - the topic arn and the region.
python3 weirdAAL.py -m sns_delete_topic -a arn:aws:sns:us-east-1:123456789123:sometopic,us-east-1
'''
try:
if args[0][0] and args[0][1]:
delete_sns_topic(args[0][0], args[0][1])
except IndexError:
print("Please provide a topic arn *AND* region, ex: -a arn:aws:sns:us-east-1:123456789123:sometopic,us-east-1")
# Shit is broke atm
#def module_sns_delete_subscriber(*args):
# '''
# SNS delete a subscriber. Takes two arguments - the subscriber arn and the region.
# python3 weirdAAL.py -m sns_delete_subscriber -a arn:aws:sns:us-east-1:123456789123:pwned-topic:05ac3eaa-703a-4bda-83ad-6861893f7542,us-east-1
# '''
# try:
# if args[0][0] and args[0][1]:
# delete_sns_subscriber(args[0][0], args[0][1])
# except IndexError:
# print("Please provide a subscriber arn *AND* region, ex: -a arn:aws:sns:us-east-1:123456789123:pwned-topic:05ac3eaa-703a-4bda-83ad-6861893f7542,us-east-1")