sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

modified: client/dnssender.py

Browse Source 
modified:   database/initdb.py
	new file:   database/test.py
	modified:   server/xiaomiandns.py
This commit is contained in:
Smart-SangGe 2023-03-19 19:46:49 +08:00
parent 8131595d02
commit 5acfa132e4
4 changed files with 63 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/dnssender.py
View File

@ -16,7 +16,9 @@ def resolver(domain):
print("can't find IP")
if __name__ = "__main__":
if __name__ == "__main__":
domain = 'mamahaha.work'
ip = resolver(domain)
print(ip)

2
database/initdb.py
View File

@ -1,6 +1,6 @@
import sqlite3
db_file = 'dns.db'
db_file = 'database/dns.db'
if __name__ == '__main__':
conn = sqlite3.connect(db_file)
cursor = conn.cursor()

23
database/test.py Normal file
View File

@ -0,0 +1,23 @@
import sqlite3
db_file = 'database/dns.db'
if __name__ == '__main__':
conn = sqlite3.connect(db_file)
cursor = conn.cursor()
domain = 'mamahaha.wor12'
ip = "1.1.1.11"
pubkey = "asdfasdfadfsdf"
cursor.execute("SELECT * FROM xiaomiandns WHERE domain = ? OR ip = ? OR pubkey = ?",
(domain, ip, pubkey))
existing_data = cursor.fetchall()
if existing_data:
print("qqqqqq")
else:
# Insert the new data
cursor.execute(
"INSERT INTO xiaomiandns (domain, ip, pubkey) VALUES (?, ?, ?)", (domain, ip, pubkey))
print("Data inserted successfully")
conn.commit()
cursor.close()
conn.close()

49
server/xiaomiandns.py
View File

@ -86,15 +86,14 @@ class DNSServer:
class DNSAPI:
# usage: /add?domian=xxxx&ip=xx.xx.xx.xx&key=xxxxx
# /delete?domian=xxxx&ip=xx.xx.xx.xx&key=xxxxx
# usage: /add?domian=xxxx&ip=xx.xx.xx.xx&pubkey=xxxxx
# /delete?domian=xxxx&ip=xx.xx.xx.xx&prikey=xxxxx
def __init__(self, hostname, port, db_file):
self.hostname = hostname
self.port = port
self.db_file = db_file
def run(self):
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定 IP 地址和端口号
@ -109,16 +108,16 @@ class DNSAPI:
t = threading.Thread(target=self.handle_tcp_request, args=(conn,))
t.start()
def handle_tcp_request(self,conn):
def handle_tcp_request(self, conn):
request = conn.recv(1024).decode('utf-8')
response = self.handle_http_request(request)
conn.send(response)
conn.close()
def handle_http_request(self, request):
request_line, headers= request.split('\r\n\r\n', 2)
request_line, headers = request.split('\r\n\r\n', 2)
method, url, version = request_line.split(' ', 2)
print(method,url)
print(method, url)
if method == 'GET':
response = self.handle_get_request(url)
else:
@ -127,16 +126,16 @@ class DNSAPI:
def handle_get_request(self, url):
# check url start with /add
if re.match(r'^/add\?',url):
if re.match(r'^/add\?', url):
status_code = self.add_data(url[5:])
if status_code = 200:
if status_code == 200:
reason_phrase = 'Add data successful'
else:
reason_phrase = 'Add data unsuccessful'
# check url start with /delete
elif re.match(r'^/delete\?',url):
elif re.match(r'^/delete\?', url):
status_code = self.delete_data(url[9:])
if status_code = 200:
if status_code == 200:
reason_phrase = 'Delete data successful'
else:
reason_phrase = 'Delete data unsuccessful'
@ -145,8 +144,8 @@ class DNSAPI:
reason_phrase = 'unsupport api'
headers = {
'Content-Type': 'text/html',
'Connection': 'close',
'Content-Type': 'text/html',
'Connection': 'close',
}
response = 'HTTP/1.1 {} {}\r\n'.format(status_code, reason_phrase)
return response.encode("utf-8")
@ -155,8 +154,8 @@ class DNSAPI:
status_code = 400
reason_phrase = "unsupport method"
headers = {
'Content-Type': 'text/html',
'Connection': 'close',
'Content-Type': 'text/html',
'Connection': 'close',
}
response = 'HTTP/1.1 {} {}\r\n'.format(status_code, reason_phrase)
return response.encode("utf-8")
@ -164,16 +163,29 @@ class DNSAPI:
def add_data(self, url):
domain = re.search(r'domain=([^&]+)', url)
ip = re.search(r'ip=([^&]+)', url)
key = re.search(r'ip=([^&]+)', url)
pubkey = re.search(r'pubkey=([^&]+)', url)
if domain and ip and key:
domain = domain.group(1)
ip = ip.group(1)
key = key.group(1)
pubkey = pubkey.group(1)
else:
return 400
return 200
conn = sqlite3.connect(self.db_file)
c = conn.cursor()
def delete_data(self,url):
# Check if the data already exists
c.execute(
"SELECT * FROM xiaomiandns WHERE domain = ? OR ip = ? OR pubkey = ?", (domain, ip, pubkey))
existing_data = c.fetchall()
if existing_data:
return 400
else:
# Insert the new data
c.execute(
"INSERT INTO xiaomiandns (domain, ip, pubkey) VALUES (?, ?, ?)", (domain, ip, pubkey))
return 200
def delete_data(self, url):
m = re.search(r'domain=([^&]+)', url)
if m:
domain = m.group(1)
@ -183,7 +195,6 @@ class DNSAPI:
return 200
if __name__ == '__main__':
# some config
db_file = '../database/dns.db'