关于IMDShift

IMDShift是一款功能强大的AWS工作负载安全增强工具，该工具基于纯Python开发，可以帮助广大研究人员更好地提升AWS工作负载的安全。

很多研究人员和开发人员在实践中会发现，依赖于元数据节点的AWS工作负载很容易会受到服务器端请求伪造（SSRF）攻击的影响，而IMDShift可以通过自动化的形式将所有的工作负载进程迁移到IMDSv2，在除了增加了额外功能的情况下，IMDSv2还实施了增强型的安全保护措施来帮助AWS工作负载抵御相关安全威胁，并大幅度提升安全性。

功能介绍

1、在各种服务中检测依赖元数据节点的AWS工作负载，包括EC2、ECS、EKS、Lightsail、AutoScaling Groups、Sagemaker Notebooks、Beanstalk（正在开发中）；

2、提供了简单直观的命令行界面，易于使用；

3、支持将所有工作负载自动迁移到IMDSv2；

4、为兼容资源设置了独立的跳点限制更新；

5、为兼容资源启用独立元数据节点操作；

6、提供了迁移过程的详细日志记录；

7、使用了MetadataNoToken识别使用了IMDSv1的资源；

8、内置服务控制策略（SCP）建议；

工具安装

由于该工具基于纯Python 3开发，因此我们首先需要在本地设备上安装并配置好Python 3环境。接下来，广大研究人员可以根据实际需求，选择下列两种工具安装方式。

生产环境安装

git clone https://github.com/ayushpriya10/imdshift.git

cd imdshift/

python3 -m pip install .

开发环境安装

git clone https://github.com/ayushpriya10/imdshift.git

cd imdshift/

python3 -m pip install -e .

工具参数选项

--services TEXT：该参数可以从下列环境中扫描出使用了IMDSv1的工作负载实例：EC2, Sagemaker、ASG、Lightsail、ECS、EKS、Beanstalk；命令格式: "--services EC2,Sagemaker,ASG"；

--include-regions TEXT：该案数明确指定要扫描IMDSv1使用情况的区域范围；命令格式: "--include-regions ap-south-1,ap-southeast-1"；

--exclude-regions TEXT：该参数明确指定要从扫描中排除的区域；命令格式: "--exclude-regions ap-south-1,ap-southeast-1"；

--migrate：这个布尔参数可以控制IMDShift是否要执行迁移任务，默认为“False”；命令格式: "--migrate"

--update-hop-limit INTEGER：该参数指定是否应更新跃点限制以及更新值。建议将跃点限制设置为“2”，以使容器能够使用IMDS节点。如果未传递此参数，则迁移期间不会更新跃点限制；命令格式: "--update-hop-limit 3"；

--enable-imds：这个布尔参数可以让IMDShift启用资源元数据节点查看，并执行迁移操作；               命令格式: "--enable-imds"；

--profile TEXT：该参数允许我们使用任意配置文件，路径为 ~/.aws/credentials file；命令格式: "--profile prod-env"；

--role-arn TEXT：这个参数可以通过AWS STS声明一个用户角色；命令格式: "--role-arn arn:aws:sts::111111111:role/John"；

--print-scps：这个布尔参数用于控制打印可以控制IMDS使用的SCP，默认为“False”；命令格式: "--print-scps"；

--check-imds-usage：这个布尔参数可以启动一次扫描以识别过去30天内在指定区域中有多少个实例使用了IMDSv1，默认为“False”；命令格式: "--check-imds-usage"；

--help：显示工具帮助信息和退出；

工具运行截图

许可证协议

本项目的开发与发布遵循GPL-3.0开源许可证协议。

项目地址

IMDShift：【GitHub传送门】

参考资料

https://github.com/salesforce/metabadger