cve_2021_45046_apache_log4j

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Filename: cve_2021_45046_apache_log4j.py
# Version: 1.0.0
# Author: Jeoi Reqi
# Vulnerability Source: https://nvd.nist.gov/vuln/detail/CVE-2021-45046

"""
Description:
This script checks for the presence of vulnerable software configurations on the user's machine
in relation to the CVE-2021-45046 vulnerability. It compares the list of installed software on
the system with a predefined list of vulnerable software configurations associated with the CVE.
If any vulnerable software configurations are found, it displays a warning message to prompt
the user to take immediate action to mitigate the vulnerability.

Requirements:
- Python 3.x: The script is written in Python 3 and requires a Python interpreter of version 3 or higher to run.
- WMI Module: The script uses the WMI module to interact with the Windows Management Instrumentation (WMI) API
  to retrieve a list of installed software on a Windows system. Ensure that the WMI module is installed.
  You can install it using pip: `pip install WMI`.

Usage:
1. Ensure Python 3.x is installed on your system.
2. Install the WMI module by running `pip install WMI`.
3. Run the script using the command `python cve_2021_45046.py`.
4. The script will verify if any vulnerable software configurations are installed on your machine and
   provide instructions for mitigation if necessary.

Functions:
- get_installed_software(): Retrieves a list of installed software on the user's machine using the WMI module.
- check_for_vulnerabilities(): Compares the list of installed software with a predefined list of vulnerable
  software configurations and displays a warning message if any vulnerable software configurations are found.

Important Notes:
- The predefined list of vulnerable software configurations in this script corresponds to the CVE-2021-45046 vulnerability.
- The script only supports Windows systems due to its dependency on the WMI module for retrieving installed software information.
"""

import wmi

# List of vulnerable software configurations
vulnerable_software = [
    "Apache Log4j 2.0 up to (excluding) 2.12.2",
    "Apache Log4j 2.13.0 up to (excluding) 2.16.0"
]

def get_installed_software():
    """
    Retrieves a list of installed software on the user's machine using the WMI module.

    Returns:
        list: A list containing the names of installed software.
    """
    c = wmi.WMI()
    installed_software = []
    for item in c.Win32_Product():
        installed_software.append(item.Caption)
    return installed_software

def check_for_vulnerabilities():
    """
    Compares the list of installed software with a predefined list of vulnerable software configurations
    and displays a warning message if any vulnerable software configurations are found.
    """
    installed_software = get_installed_software()
    vulnerable_installed = [software for software in installed_software if software in vulnerable_software]
    if vulnerable_installed:
        print("\nWarning:\nThe following vulnerable software configurations are installed on your machine:")
        for software in vulnerable_installed:
            print("- " + software)
        print("\nPlease take immediate action to mitigate the vulnerability by applying updates per vendor instructions.\n")
    else:
        print("\nAll clear!\nNone of the vulnerable software configurations are installed on your machine.\n")

if __name__ == "__main__":
    print("Verifying vulnerable software configurations...")
    check_for_vulnerabilities()