64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# imports ######################################################################
|
||
|
|
import sqlite3
|
||
|
|
import sys
|
||
|
|
import urllib.request
|
||
|
|
import re
|
||
|
|
|
||
|
|
# functions ####################################################################
|
||
|
|
def extract_cookies_to_jar(cookie_db):
|
||
|
|
try:
|
||
|
|
|
||
|
|
# Connection
|
||
|
|
conn = sqlite3.connect(cookie_db)
|
||
|
|
|
||
|
|
# Execute SELECT query
|
||
|
|
cur = conn.cursor()
|
||
|
|
r = cur.execute("SELECT name, value FROM moz_cookies WHERE host LIKE '%newbiecontest.org%' AND name LIKE '%SMFCookie%'")
|
||
|
|
|
||
|
|
cookie = r.fetchone()
|
||
|
|
|
||
|
|
return cookie
|
||
|
|
|
||
|
|
except sqlite3.Error as e:
|
||
|
|
print("An error occurred:", e.args[0])
|
||
|
|
|
||
|
|
def send_resquest(url, cookie):
|
||
|
|
opener = urllib.request.build_opener()
|
||
|
|
opener.addheaders.append(('Cookie', cookie[0] + "=" + cookie[1]))
|
||
|
|
|
||
|
|
f = opener.open(url)
|
||
|
|
|
||
|
|
number = re.findall(r'\d+', f.read().decode('utf-8'))
|
||
|
|
|
||
|
|
return number
|
||
|
|
|
||
|
|
def send_number(url, number):
|
||
|
|
opener = urllib.request.build_opener()
|
||
|
|
opener.addheaders.append(('Cookie', cookie[0] + "=" + cookie[1]))
|
||
|
|
|
||
|
|
f = opener.open(url + "?solution=" + number[0])
|
||
|
|
|
||
|
|
print(f.read().decode('utf-8'))
|
||
|
|
|
||
|
|
# main #########################################################################
|
||
|
|
if __name__ == '__main__':
|
||
|
|
|
||
|
|
# if arguments are missing
|
||
|
|
if len(sys.argv) != 4:
|
||
|
|
print("[i] Usage : cookie.db url")
|
||
|
|
print("Get. The. Fuck. Out")
|
||
|
|
|
||
|
|
# extract related cookies from db
|
||
|
|
cookie = extract_cookies_to_jar(sys.argv[1])
|
||
|
|
|
||
|
|
# send request, get the number
|
||
|
|
number = send_resquest(sys.argv[2], cookie)
|
||
|
|
|
||
|
|
# send the number back
|
||
|
|
send_number(sys.argv[3], number)
|
||
|
|
|
||
|
|
print("Done.")
|
||
|
|
|