Cách giải Captcha / Thách thức AWS bằng Python

pip install capsolver requests beautifulsoup4

import capsolver
import requests
from bs4 import BeautifulSoup
import re
from urllib.parse import urlparse

# -------------------VUI LÒNG THAY ĐỔI CÁC GIÁ TRỊ SAU-------------------
# Chi tiết proxy của bạn theo định dạng: http://username:password@ip:port
PROXY = "http://username:password@ip:port"
# URL của trang web mục tiêu bạn muốn giải quyết AWS WAF
PAGE_URL = "https://www.trang-web-bảo-vệ-aws-của-bạn.com"
# Mã API từ bảng điều khiển CapSolver
CAPSOLVER_API_KEY = "Mã_API_Của_Bạn"
# ----------------------------------------------------------------

def solve_aws_waf(is_captcha_challenge, **kwargs):
    """
    Giải quyết thách thức AWS WAF hoặc captcha bằng CapSolver.
    :param is_captcha_challenge: True nếu là captcha (mã trạng thái 405), False nếu là thách thức JS (mã trạng thái 202).
    :param kwargs: Các tham số cần thiết cho nhiệm vụ giải quyết.
    :return: Đối tượng giải pháp, hoặc None nếu thất bại.
    """
    task_type = "AntiAwsWafTask" if is_captcha_challenge else "AntiAwsWafTaskProxyLess"
    print(f"Tạo nhiệm vụ '{task_type}'...")
    
    task_payload = {
        "type": task_type,
        "websiteURL": PAGE_URL,
    }
    
    # Thêm các tham số cụ thể dựa trên loại thách thức
    if is_captcha_challenge:
        # Đối với các thách thức captcha, cần có awsKey, awsIv, awsContext
        task_payload.update({
            "awsKey": kwargs.get("awsKey"),
            "awsIv": kwargs.get("awsIv"),
            "awsContext": kwargs.get("awsContext"),
            "awsChallengeJS": kwargs.get("awsChallengeJS"),
            "proxy": PROXY
        })
    else:
        # Đối với các thách thức JS, chỉ cần URL của awsChallengeJS
        task_payload["awsChallengeJS"] = kwargs.get("awsChallengeJS")

    try:
        solution = capsolver.solve(task_payload)
        if solution and solution.get("errorId", 0) == 0:
            print("Giải pháp được lấy thành công!")
            return solution
        else:
            error_desc = solution.get('errorDescription', 'Lỗi không xác định') if solution else 'Lỗi không xác định'
            print(f"Không thể giải quyết nhiệm vụ: {error_desc}")
            return None
    except Exception as e:
        print(f"Đã xảy ra ngoại lệ khi gọi CapSolver: {e}")
        return None

def main():
    """Hàm chính để thực hiện toàn bộ quy trình."""
    capsolver.api_key = CAPSOLVER_API_KEY
    session = requests.Session()
    session.proxies = {"http": PROXY, "https": PROXY}
    
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36",
        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8",
        "Accept-Language": "en-US,en;q=0.9",
    }

    print(f"Thử truy cập: {PAGE_URL}")
    response = session.get(PAGE_URL, headers=headers)
    print(f"Nhận được mã trạng thái: {response.status_code}")

    solution = None
    
    # Trường hợp 1: Xử lý thách thức JavaScript của AWS (Mã trạng thái 202)
    if response.status_code == 202:
        print("Phát hiện thách thức JavaScript của AWS.")
        soup = BeautifulSoup(response.content, 'html.parser')
        script_tag = soup.find('script', {'src': re.compile(r'token\.awswaf\.com')})
        if script_tag:
            challenge_js_url = script_tag['src']
            print(f"Đã tìm thấy URL thách thức JS: {challenge_js_url}")
            solution = solve_aws_waf(is_captcha_challenge=False, awsChallengeJS=challenge_js_url)
        else:
            print("Lỗi: Không thể tìm thấy script thách thức AWS.")

    # Trường hợp 2: Xử lý captcha của AWS (Mã trạng thái 405)
    elif response.status_code == 405:
        print("Phát hiện captcha của AWS.")
        soup = BeautifulSoup(response.content, 'html.parser')
        script_tag = soup.find('script', {'src': re.compile(r'token\.awswaf\.com')})
        script_text_tag = soup.find('script', string=re.compile('.*key.*'))
        
        if script_tag and script_text_tag:
            challenge_js_url = script_tag['src']
            script_text = script_text_tag.string
            
            key = re.search(r'"key":"(.*?)"', script_text).group(1)
            iv = re.search(r'"iv":"(.*?)"', script_text).group(1)
            context = re.search(r'"context":"(.*?)"', script_text).group(1)
            
            print(f"Đã trích xuất các tham số: Key={key[:5]}..., IV={iv[:5]}..., Context={context[:5]}...")
            solution = solve_aws_waf(
                is_captcha_challenge=True,
                awsKey=key,
                awsIv=iv,
                awsContext=context,
                awsChallengeJS=challenge_js_url
            )
        else:
            print("Lỗi: Không thể trích xuất tất cả các tham số captcha cần thiết từ trang.")

    # Nếu giải quyết thành công, thiết lập cookie và thử lại yêu cầu
    if solution and solution.get("cookie"):
        aws_token = solution["cookie"]
        print(f"Đã lấy được aws-waf-token thành công: {aws_token[:30]}...")
        
        domain = f".{urlparse(PAGE_URL).netloc}"
        session.cookies.set("aws-waf-token", aws_token, domain=domain)
        
        print("Cookie đã được thiết lập. Thử lại yêu cầu...")
        final_response = session.get(PAGE_URL, headers=headers)
        print(f"Mã trạng thái yêu cầu cuối cùng: {final_response.status_code}")
        
        if final_response.status_code == 200:
            print("Đã vượt qua AWS WAF thành công!")
            # print(final_response.text) # Bỏ dấu gạch chéo để xem nội dung trang
        else:
            print("Yêu cầu thất bại sau khi thiết lập cookie. Kiểm tra token hoặc proxy.")
    else:
        print("Không thể vượt qua AWS WAF. Thoát chương trình.")

if __name__ == "__main__":
    main()