使用 Python 和 Boto3 查找并验证 AWS 中未使用的安全组-Python教程-PHP中文网

使用 Python 和 Boto3 查找并验证 AWS 中未使用的安全组

DDD

发布： 2024-12-13 22:27:17

转载

930人浏览过

使用 python 和 boto3 查找并验证 aws 中未使用的安全组

有效管理 aws 安全组对于维护安全且经济高效的云环境至关重要。安全组是 aws 网络安全的重要组成部分，但随着时间的推移，未使用的安全组会不断累积。这些未使用的组不仅会使您的环境变得混乱，还可能带来安全风险或不必要地增加成本。

在本文中，我们将探讨如何使用 python 和 boto3 识别 aws 环境中未使用的安全组、验证它们并确保它们不被任何其他资源引用。我们还将研究如何安全地确定是否可以删除这些组。

先决条件

要学习本教程，您需要以下内容：

aws 账户：确保您有权访问要搜索未使用的安全组的 aws 环境。
boto3 已安装：您可以通过运行以下命令来安装 boto3 python sdk：

   pip install boto3

登录后复制

已配置 aws 凭证：确保您使用 aws cli 或使用 iam 角色或环境变量直接在代码中配置了 aws 凭证。

立即学习“Python免费学习笔记（深入）”；

KAIZAN.ai

使用AI来改善客户服体验，提高忠诚度

查看详情

代码分解

让我们看一下代码，用于识别给定 aws 区域中未使用的安全组、验证它们并检查它们是否被任何其他组引用。

第 1 步：获取所有安全组和 eni

import boto3
from botocore.exceptions import clienterror

def get_unused_security_groups(region='us-east-1'):
    """
    find security groups that are not being used by any resources.
    """
    ec2_client = boto3.client('ec2', region_name=region)

    try:
        # get all security groups
        security_groups = ec2_client.describe_security_groups()['securitygroups']

        # get all network interfaces
        enis = ec2_client.describe_network_interfaces()['networkinterfaces']

        # create set of security groups in use
        used_sg_ids = set()

        # check security groups attached to enis
        for eni in enis:
            for group in eni['groups']:
                used_sg_ids.add(group['groupid'])

        # find unused security groups
        unused_groups = []
        for sg in security_groups:
            if sg['groupid'] not in used_sg_ids:
                # skip default security groups as they cannot be deleted
                if sg['groupname'] != 'default':
                    unused_groups.append({
                        'groupid': sg['groupid'],
                        'groupname': sg['groupname'],
                        'description': sg['description'],
                        'vpcid': sg.get('vpcid', 'ec2-classic')
                    })

        # print results
        if unused_groups:
            print(f"\nfound {len(unused_groups)} unused security groups in {region}:")
            print("-" * 80)
            for group in unused_groups:
                print(f"security group id: {group['groupid']}")
                print(f"name: {group['groupname']}")
                print(f"description: {group['description']}")
                print(f"vpc id: {group['vpcid']}")
                print("-" * 80)
        else:
            print(f"\nno unused security groups found in {region}")

        return unused_groups

    except clienterror as e:
        print(f"error retrieving security groups: {str(e)}")
        return none

登录后复制

获取安全组：我们首先调用describe_security_groups方法获取指定区域内的所有安全组。
检索网络接口：接下来，我们使用describe_network_interfaces检索所有网络接口。每个网络接口都可以有一个或多个与其关联的安全组。
识别已使用的安全组：对于每个网络接口，我们将关联的安全组 id 添加到名为used_sg_ids 的集合中。
查找未使用的组：然后我们将安全组 id 与正在使用的组 id 进行比较。如果一个组没有被使用（即它的id不在used_sg_ids集合中），我们认为它没有被使用，除了默认的安全组，它不能被删除。

步骤 2：检查安全组引用

def check_sg_references(ec2_client, group_id):
    """
    check if a security group is referenced in other security groups' rules
    """
    try:
        # check if the security group is referenced in other groups
        response = ec2_client.describe_security_groups(
            filters=[
                {
                    'name': 'ip-permission.group-id',
                    'values': [group_id]
                }
            ]
        )

        referencing_groups = response['securitygroups']

        # check for egress rules
        response = ec2_client.describe_security_groups(
            filters=[
                {
                    'name': 'egress.ip-permission.group-id',
                    'values': [group_id]
                }
            ]
        )

        referencing_groups.extend(response['securitygroups'])

        return referencing_groups

    except clienterror as e:
        print(f"error checking security group references: {str(e)}")
        return none

登录后复制

检查引用：此函数检查特定安全组是否被任何其他安全组引用。它通过根据入站 (ip-permission.group-id) 和出站 (egress.ip-permission.group-id) 规则过滤安全组来实现此目的。
返回引用组：如果引用组，则该函数返回引用安全组的列表。如果没有，则返回 none。

步骤 3：验证未使用的安全组

def validate_unused_groups(region='us-east-1'):
    """
    validate and provide detailed information about unused security groups
    """
    ec2_client = boto3.client('ec2', region_name=region)
    unused_groups = get_unused_security_groups(region)

    if not unused_groups:
        return

    print("\nvalidating security group references...")
    print("-" * 80)

    for group in unused_groups:
        group_id = group['groupid']
        referencing_groups = check_sg_references(ec2_client, group_id)

        if referencing_groups:
            print(f"\nsecurity group {group_id} ({group['groupname']}) is referenced by:")
            for ref_group in referencing_groups:
                print(f"- {ref_group['groupid']} ({ref_group['groupname']})")
        else:
            print(f"\nsecurity group {group_id} ({group['groupname']}) is not referenced by any other groups")
            print("this security group can be safely deleted if not needed")

登录后复制

验证未使用的安全组：在最后一步中，脚本首先检索未使用的安全组。然后，对于每个未使用的组，它会检查是否有任何其他安全组在其规则中引用它。
输出：脚本输出该组是否被引用，如果没有，则可以安全删除。

运行脚本

要运行脚本，只需执行 validate_unused_groups 函数即可。例如，当区域设置为 us-east-1 时，脚本将：

检索 us-east-1 中的所有安全组和网络接口。
识别未使用的安全组。
验证并报告这些未使用的组是否被其他安全组引用。

示例输出

Found 5 unused security groups in us-east-1:
--------------------------------------------------------------------------------
Security Group ID: sg-12345678
Name: unused-sg-1
Description: Unused security group
VPC ID: vpc-abcde123
--------------------------------------------------------------------------------
Security Group sg-12345678 (unused-sg-1) is not referenced by any other groups
This security group can be safely deleted if not needed
--------------------------------------------------------------------------------
Security Group ID: sg-23456789
Name: unused-sg-2
Description: Another unused group
VPC ID: vpc-abcde123
--------------------------------------------------------------------------------
Security Group sg-23456789 (unused-sg-2) is referenced by:
- sg-34567890 (some-other-sg)

登录后复制