Python – Ataque de Deserialización Yaml (DES-Yaml)

# Importa las librerías necesarias
from flask import Flask, render_template, redirect
import yaml
import base64

# Crea una instancia de la aplicación Flask. La aplicación buscará plantillas HTML en una carpeta llamada 'templates'
# y archivos estáticos (como CSS y JavaScript) en una carpeta llamada 'static'.
app = Flask(__name__, static_url_path='/static', static_folder='static')

# Habilita el modo de depuración para la aplicación Flask. Esto proporcionará más información en caso de que ocurra un error.
app.config['DEBUG'] = True

# Define la ruta principal ("/") de la aplicación. Cuando un cliente solicite esta ruta, la aplicación redirigirá
# al cliente a otra ruta con un código 302, que indica una redirección temporal.
@app.route("/")
def start():
    return redirect("/information/eWFtbDogVGhlIGluZm9ybWF0aW9uIHBhZ2UgaXMgc3RpbGwgdW5kZXIgY29uc3RydWN0aW9uLCB1cGRhdGVzIGNvbWluZyBzb29uIQ==", code=302)

# Define una ruta que acepta una entrada como parte de la URL.
# Este bloque intentará decodificar la entrada como un objeto serializado en base64, y luego deserializarlo usando YAML.
@app.route("/information/<input>", methods=['GET'])
def deserialization(input):
    try:
        # Si no hay ninguna entrada, se renderea la plantilla por defecto
        if not input:
            return render_template("information/index.html")

        # Decodifica la entrada de base64
        yaml_file = base64.b64decode(input)

        # Intenta cargar el contenido YAML
        content = yaml.load(yaml_file, Loader=yaml.FullLoader)
    except:
        # En caso de un error durante la decodificación o deserialización, se muestra un mensaje de error.
        content = "The application was unable to deserialize the object!"

    # Renderiza la plantilla con el contenido deserializado
    return render_template("index.html", content=content['yaml'])

# Define una página de error personalizada para el error 404 (No encontrado).
@app.errorhandler(404)
def page_not_found(e):
    return render_template("404.html")

# Si este script se está ejecutando como el archivo principal, arranca la aplicación Flask.
if __name__ == "__main__":
    app.run(host='0.0.0.0')