[파이썬3.x] nslookup 자동화 툴

import os
import socket
from urllib.parse import urlparse


def nslookup(DOMAIN):
    try:

        ip_list = []

        ais = socket.getaddrinfo(DOMAIN, 0, 0, 0, 0)

        for result in ais:
            ip_list.append(result[-1][0])

        ip_list = list(set(ip_list))

        return (',').join(ip_list)

    except:

        return "N/A"


def getDomain(PATH, FNAME):
    # print PATH + '\\' + FNAME

    with open(PATH + '\\' + FNAME, 'r') as f:

        maldomain = f.readlines()

    if len(maldomain) > 0:

        for i in range(0, len(maldomain)):

            # print i

            if maldomain[i][:4] != "http":
                maldomain[i] = "http://" + maldomain[i].rstrip()

            # print urlparse(maldomain[i]).hostname

            #maldomain[i] = "%-15s / %s" % (nslookup(urlparse(maldomain[i]).hostname), "hxxp" + maldomain[i][4:])
            maldomain[i] = nslookup(urlparse(maldomain[i]).hostname)
            print
            maldomain[i]

    else:

        maldomain = "No URL in File"

    ## 결과 출력

    # print maldomain

    return maldomain


print
"Loading...."

result = open('result.txt', 'w')

for item in getDomain(os.getcwd(), "list.txt"):
    result.write("%s\n" % item)

print
"Done. Thank you!"

 

list.txt 파일을 읽어 result.txt 파일에 결과물을 생성한다.

URL을 한줄에 하나씩 읽는다

list.txt

www.naver.com
www.daum.net

 

result.txt

111.111.111.111 / hxxp://www.naver.com
222.111.111.122 / hxxp://www.daum.com