made API more injection-resistant (lmao)
This commit is contained in:
parent
abc64e2dff
commit
462399de43
12
api/api.py
12
api/api.py
@ -71,16 +71,16 @@ def remove_message():
|
||||
cur = db.cursor()
|
||||
token = request.form['token']
|
||||
message_id = request.form['message_id']
|
||||
res = cur.execute(f"SELECT user_id FROM user WHERE token='{token}'")
|
||||
res = cur.execute("SELECT user_id FROM user WHERE token= ?", (token,))
|
||||
res = res.fetchone()
|
||||
if res is not None:
|
||||
(user_id,) = res
|
||||
res = cur.execute(f"SELECT message_id, user_id FROM message WHERE message_id='{message_id}'")
|
||||
res = cur.execute("SELECT message_id, user_id FROM message WHERE message_id= ?", (message_id,))
|
||||
res = res.fetchone()
|
||||
if res is not None:
|
||||
(message_id, message_user_id) = res
|
||||
if message_user_id == user_id:
|
||||
cur.execute(f"DELETE FROM message WHERE message_id='{message_id}'")
|
||||
cur.execute("DELETE FROM message WHERE message_id= ?", (message_id,))
|
||||
db.commit()
|
||||
return Response(status=HTTPStatus.NO_CONTENT)
|
||||
else:
|
||||
@ -97,16 +97,16 @@ def edit_message():
|
||||
token = request.form['token']
|
||||
message_id = request.form['message_id']
|
||||
new_message = request.form['message']
|
||||
res = cur.execute(f"SELECT user_id FROM user WHERE token='{token}'")
|
||||
res = cur.execute("SELECT user_id FROM user WHERE token = ?", (token,))
|
||||
res = res.fetchone()
|
||||
if res is not None and new_message is not None:
|
||||
(user_id,) = res
|
||||
res = cur.execute(f"SELECT message_id, user_id FROM message WHERE message_id='{message_id}'")
|
||||
res = cur.execute("SELECT message_id, user_id FROM message WHERE message_id= ?", (message_id,))
|
||||
res = res.fetchone()
|
||||
if res is not None:
|
||||
(message_id, message_user_id) = res
|
||||
if message_user_id == user_id:
|
||||
cur.execute(f"UPDATE message SET message = '{new_message}' WHERE message_id='{message_id}'");
|
||||
cur.execute("UPDATE message SET message = ? WHERE message_id= ?", (new_message, message_id));
|
||||
db.commit()
|
||||
return Response(status=HTTPStatus.NO_CONTENT)
|
||||
else:
|
||||
|
BIN
api/forum.db
BIN
api/forum.db
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user