37 lines
918 B
Python
37 lines
918 B
Python
import requests
|
|
import sys
|
|
|
|
|
|
|
|
|
|
|
|
def post_submit(team_token, flag, url):
|
|
if team_token is None or flag is None:
|
|
raise Exception('team token or flag wrong')
|
|
form_data = {
|
|
"token":team_token,
|
|
"flag":flag
|
|
}
|
|
response = requests.post(url, data=form_data)
|
|
return response.text
|
|
|
|
|
|
def get_submit(team_token, flag, url):
|
|
if not team_token or not flag:
|
|
raise Exception('team token or flag wrong')
|
|
url = url + token + "&flag=" + flag
|
|
response = requests.get(url)
|
|
return response.text
|
|
|
|
|
|
if __name__ == '__main__':
|
|
token = ""
|
|
flag = ""
|
|
base_url = ' ' #提交flag的服务器地址
|
|
# url = "https://ctf.bugku.com/pvp/submit.html?token=" + token + "&flag=" + flag
|
|
method = "" #get或post
|
|
if method == "get":
|
|
response = get_submit(token, flag, base_url)
|
|
elif method == "post":
|
|
response = post_submit(token, flag, base_url)
|