dns_test.sh

#!/bin/bash

clear
echo " ____  _   _ ____    _____         _    "
echo "|  _ \| \ | / ___|  |_   _|__  ___| |_  "
echo "| | | |  \| \___ \    | |/ _ \/ __| __| "
echo "| |_| | |\  |___) |   | |  __/\__ \ |_  "
echo "|____/|_| \_|____/    |_|\___||___/\__| "
echo "                                        "
echo "----------------------------------------------------- "

if command -v python3 &>/dev/null; then
    &>/dev/null
elif command -v python2 &>/dev/null; then
    &>/dev/null
else
    echo "Please install Python before this script."
    exit 1
fi

if command -v pip3 &>/dev/null; then
    &>/dev/null 
elif command -v pip &>/dev/null; then
    &>/dev/null
else
    echo "Please install pip before this script."
    exit 1
fi

if python -c "import prettytable" &> /dev/null; then
    &>/dev/null
elif python3 -c "import prettytable" &> /dev/null; then
    &>/dev/null
else
    echo "Please install prettytable module before this script."
    exit 1   
fi
echo ""
os_system=$(uname)
if [ "$os_system" == "Darwin" ]; then
    ipv6_local_status=$(ifconfig | grep inet6)
elif [ "$os_system" == "Linux" ]; then
    ipv6_local_status=$(ip -6 addr show)
else
    echo "Unsupported operating system"
    exit 1
fi

ipv6_wan_status=$(curl -6 -s -I www.google.com)
if [ -n "$ipv6_local_status" ] && [ -n "$ipv6_wan_status" ]; then
    echo "Both local machine and WAN support IPv6. Testing both IPv4 and IPv6..."
    while true; do
        read -p "Enter the domain (e.g. google.com): " domain

        if [[ $domain =~ ^[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$ ]]; then
            if nslookup $domain >/dev/null 2>&1; then
                echo ""
                break
            else
                echo "Domain $domain is valid but does not exist. Please enter a valid domain."
            fi
        else
            echo "Invalid domain. Please enter a valid domain."
        fi
    done

    doh_server=($(dig SVCB _dns.resolver.arpa +short | awk -F'1' '{print $2}' | awk -F'. ' '{print $1}'))
    if [ -n "$doh_server" ]; then
        echo "DOH test...."
        doh4="curl -s -H 'accept: application/dns-json' 'https://$doh_server/dns-query?name=$domain&type=A'"
        doh4_output=$(eval "$doh4")
        echo "$doh4_output" >doh4_output

        doh6="curl -s -H 'accept: application/dns-json' 'https://$doh_server/dns-query?name=$domain&type=AAAA'"
        doh6_output=$(eval "$doh6")
        echo "$doh6_output" >doh6_output

        echo "DOT test...."
        dig +tls "$domain" +short > dot
        dig +tls "$domain" AAAA +short > dot6
    else
        echo "No DOH server available"
        doh4_output='{"Answer":[{"name":"N/A","type":"N/A","TTL":"N/A","data":"N/A"}]}'
        echo "$doh4_output" >doh4_output

        doh6_output='{"Answer":[{"name":"N/A","type":"N/A","TTL":"N/A","data":"N/A"}]}'
        echo "$doh6_output" >doh6_output

        echo "No DOT server available"
        echo "N/A" > dot
        echo "N/A" > dot6
    fi

    echo "DNS test...."
    dig "$domain" +short > dns
    dig "$domain" AAAA +short > dns6

    echo "Ping test...."
    json_file="ping_results.json"
    if [ "$os_system" == "Darwin" ]; then
        result1=($(ping -c4 $domain | grep from))
        result2=($(ping6 -c4 $domain | grep from))
        formatted_results=()
        for ((i = 0; i < ${#result1[@]}; i += 8)); do
            formatted_results+=("${result1[i+6]} ${result1[i+7]}")
        done
        for ((i = 0; i < ${#result2[@]}; i += 8)); do
            formatted_results+=("${result2[i+6]} ${result2[i+7]}")
        done        
    else
        result1=($(ping -c4 -4 $domain | grep from))
        result2=($(ping -c4 -6 $domain | grep from))
        formatted_results=()
        for ((i = 0; i < ${#result1[@]}; i += 9)); do
            formatted_results+=("${result1[i+7]} ${result1[i+8]}")
        done
        for ((i = 0; i < ${#result2[@]}; i += 9)); do
            formatted_results+=("${result2[i+7]} ${result2[i+8]}")
        done        
    fi

    json_data="{ \"results\": ["
    for line in "${formatted_results[@]}"; do
        json_data="${json_data} \"$line\","
    done
    json_data="${json_data%,} ]}"
    echo $json_data > $json_file

    echo ""
    python3 dns_test.py
    rm doh4_output doh6_output dns dns6 ping_results.json dot dot6
else
    echo "IPv6 checks failed. Testing IPv4 only..."
    while true; do
        read -p "Enter the domain (e.g. google.com): " domain

        if [[ $domain =~ ^[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$ ]]; then
            if nslookup $domain >/dev/null 2>&1; then
                echo ""
                break
            else
                echo "Domain $domain is valid but does not exist. Please enter a valid domain."
            fi
        else
            echo "Invalid domain. Please enter a valid domain."
        fi
    done

    doh_server=($(dig SVCB _dns.resolver.arpa +short | awk -F'1' '{print $2}' | awk -F'. ' '{print $1}'))

    if [ -n "$doh_server" ]; then
        echo "DOH test...."
        doh4="curl -s -H 'accept: application/dns-json' 'https://$doh_server/dns-query?name=$domain&type=A'"
        doh4_output=$(eval "$doh4")
        echo "$doh4_output" >doh4_output

        doh6_output='{"Answer":[{"name":"N/A","type":"N/A","TTL":"N/A","data":"N/A"}]}'
        echo "$doh6_output" >doh6_output

        echo "DOT test...."
        dig +tls "$domain" +short > dot
        echo "N/A" > dot6
    else
        echo "No DOH server available"
        doh4_output='{"Answer":[{"name":"N/A","type":"N/A","TTL":"N/A","data":"N/A"}]}'
        echo "$doh4_output" >doh4_output

        doh6_output='{"Answer":[{"name":"N/A","type":"N/A","TTL":"N/A","data":"N/A"}]}'
        echo "$doh6_output" >doh6_output

        echo "No DOT server available"
        echo "N/A" > dot
        echo "N/A" > dot6
    fi


    echo "DNS test...."
    dig "$domain" +short > dns
    dig "$domain" AAAA +short > dns6

    echo "Ping test...."
    json_file="ping_results.json"
    if [ "$os_system" == "Darwin" ]; then
        result1=($(ping -c4 $domain | grep from))
        formatted_results=()
        for ((i = 0; i < ${#result1[@]}; i += 8)); do
            formatted_results+=("${result1[i+6]} ${result1[i+7]}")
        done       
    else
        result1=($(ping -c4 -4 $domain | grep from))
        formatted_results=()
        for ((i = 0; i < ${#result1[@]}; i += 9)); do
            formatted_results+=("${result1[i+7]} ${result1[i+8]}")
        done  
    fi
    json_data="{ \"results\": ["
    for line in "${formatted_results[@]}"; do
        json_data="${json_data} \"$line\","
    done
    json_data="${json_data%,} ]}"
    echo $json_data > $json_file

    echo ""
    python3 dns_test.py
    rm doh4_output doh6_output dns dns6 ping_results.json dot dot6
fi