def __init__(self):
load_env_config()
self.r = redis.StrictRedis(
host=os.getenv("REDIS_HOST"),
port=os.getenv("REDIS_PORT"),
db=os.getenv("REDIS_DB"))
def test_default_env_settings(self):
load_env_config()
self.assertEqual("root", os.getenv("MYSQL_USERNAME"))
self.assertEqual("password", os.getenv("MYSQL_PASSWORD"))
self.assertEqual("localhost", os.getenv("MYSQL_HOST"))
self.assertEqual("headers", os.getenv("MYSQL_DATABASE"))
def test_custom_env_settings(self):
os.environ['MYSQL_USERNAME'] = "******"
os.environ['MYSQL_PASSWORD'] = "******"
os.environ['MYSQL_HOST'] = "custom_host"
os.environ['MYSQL_DATABASE'] = "custom_db"
load_env_config()
self.assertEqual("custom_user", os.getenv("MYSQL_USERNAME"))
self.assertEqual("custom_password", os.getenv("MYSQL_PASSWORD"))
self.assertEqual("custom_host", os.getenv("MYSQL_HOST"))
self.assertEqual("custom_db", os.getenv("MYSQL_DATABASE"))
from lib.utils.util import load_env_config
from lib.scanner.headers import Headers
import os
load_env_config()
def main():
print(
"-------------------------------------------------------------------")
print(
"Bienvenido a la herramienta de escaneo de encabezados de seguridad.")
print(
"Basado en el proyecto 'OWASP security headers' (https://wiki.owasp.org/index.php/OWASP_Secure_Headers_Project)"
)
print(
"-------------------------------------------------------------------")
print(
"Por favor ingrese la URL del sitio al cuál desea realizar el escaneo")
print(
"Para un sitio web desplegado en un dominio, el formato debe ser como el siguiente: facebook.com "
)
print(
"Para un sitio web desplegado localmente, el formato debe ser como el siguiente: 127.0.0.1:8000 "
)
url = str(input())
threads_number = os.getenv('THREAD_NUMBER')
Headers().run(url, int(threads_number))
if __name__ == '__main__':
def __init__(self):
load_env_config()
requests.packages.urllib3.disable_warnings()
def __init__(self):
load_env_config()
self.headers_filter = os.getenv('HEADERS').lower().split(',')