为了检测内网的端口扫描写了个脚本。。。

事情起源

之前一直在做域控搭建，结果域控机（DC）老是莫名其妙重启，排查事件管理器后发现了点东西

1559234162802

一堆的电源报警事件（该事件只会在系统非正常关机、重启时出现），后面进ESXi管理面板准备重装时发现蓝屏了，报错显示srv.sys驱动出现问题，经过查询后发现系统有MS17-010漏洞（wannacry干的，能把2016打蓝屏也是没谁了）

处理过程

打补丁，用脚本检测内网的扫描情况，汇报上面让他们处理去吧。。

脚本

'''
@Description: 
@Author: Sp4ce
@Github: https://github.com/NS-Sp4ce
@Date: 2019-05-29 23:18:36
@LastEditors: Sp4ce
@LastEditTime: 2019-05-29 23:18:36
'''
# 服务器
import socket
import threading
import time
import os
# 处理客户端请求
# 监听任意地址
IP_ADDRESS = '0.0.0.0'
# 记录的文件名
FILENAME = 'log.txt'
# 监听端口
PORT = 445


def printalert(string):
    print('\033[1;31m ' + string + ' \033[0m')


def printsuccess(string):
    print('\033[1;32m ' + string + ' \033[0m')


def printwarning(string):
    print('\033[1;33m ' + string + ' \033[0m')


def printinfo(string):
    print('\033[0;34m ' + string + ' \033[0m')


class tcp_server():
    def __init__(self):
        self.tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.tcp_socket.bind((IP_ADDRESS, PORT))
        self.tcp_socket.listen(250)
        self.proceses = []
        printsuccess('Server Start Success,Waiting For Connect....\n')

    def run_server(self):
        while True:
            sock, addr = self.tcp_socket.accept()
            # 创建新线程来处理每个客户端连接
            t = threading.Thread(target=self.tcp_server, args=(sock, addr))
            t.start()
            t.join()

    def tcp_server(self, sock, addr):
        # 事件产生时间
        TIME = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
        printinfo('[i] Accept new connection from %s:%s' % addr + ' ...at [' + str(TIME) + ']\n')
        printinfo('当前线程：' + threading.current_thread().name + '\n')
        self.writelog(addr)
        sock.close()
        printinfo('[-] Connection from %s:%s closed \n' % addr)

    # 写日志
    def writelog(self, addr):
        # 事件产生时间
        TIME = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
        with open(FILENAME, 'r') as file:
            line = file.read().splitlines()
            if addr[0] in line:
                printwarning('[!] Detect ' + str(addr[0]) + ' Connect ' + str(PORT) + ' Port At [' + str(TIME) + '] And This IP Has Been Recorded!\n')
            else:
                with open(FILENAME, 'a+') as file1:
                    file1.write(str(addr[0]) + '\n')
                    printalert('[+] Detect ' + str(addr[0]) + ' Connect  ' + str(PORT) + ' Port At [' + str(TIME) + '] .\n')


def main():
    s = tcp_server()
    s.run_server()


if __name__ == '__main__':
    if not os.path.exists(FILENAME):
        printwarning('[!] Log file is not exists, will create it.\n')
        os.system(r"touch {}".format(FILENAME))
        printsuccess('[+] Create log file success !\n')
    main()