cis-kuldeep
/
PyBitmessage-2021-04-27
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added additional checks against SQL-injection

This commit is contained in:
Dmitri Bogomolov 2018-03-12 18:28:32 +02:00
parent cc2fa62b90
commit 9bca4faf9f
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/helper_db.py
View File

@ -9,6 +9,9 @@ from helper_sql import sqlExecute, sqlQuery
__all__ = ["search_sql", "check_match"] __all__ = ["search_sql", "check_match"]
_groups = ("blacklist", "whitelist", "subscriptions", "addressbook")
_groups_enable = ("blacklist", "whitelist", "subscriptions")
# + genAckPayload # + genAckPayload
def put_sent( def put_sent(
@ -93,8 +96,18 @@ def put_pubkey(address, address_version, data, used_personally=None):
def _in_group_already(address, group="addressbook"): def _in_group_already(address, group="addressbook"):
return sqlQuery( if group not in _groups:
"SELECT enabled FROM %s WHERE address=?" % group, address) return True
# elif group in _groups_enable:
# try:
# return sqlQuery(
# "SELECT enabled FROM %s WHERE address=?" % group, address
# )[-1][0]
# except IndexError:
# return
else:
return sqlQuery(
"SELECT * FROM %s WHERE address=?" % group, address)
def put_addresslist(label, address, group="blacklist", enabled=True): def put_addresslist(label, address, group="blacklist", enabled=True):
@ -148,6 +161,8 @@ def get_addressbook():
def get_addresslist(group="blacklist"): def get_addresslist(group="blacklist"):
"""Generator for address list given by group arg""" """Generator for address list given by group arg"""
if group not in _groups:
return
queryreturn = sqlQuery("SELECT * FROM %s" % group) queryreturn = sqlQuery("SELECT * FROM %s" % group)
for row in queryreturn: for row in queryreturn:
yield row yield row
@ -158,6 +173,8 @@ def get_label(address, group="addressbook"):
Get address label from address list given by group arg Get address label from address list given by group arg
(default is addressbook) (default is addressbook)
""" """
if group not in _groups:
return
queryreturn = sqlQuery( queryreturn = sqlQuery(
"SELECT label FROM %s WHERE address=?" % group, address) "SELECT label FROM %s WHERE address=?" % group, address)
try: try:
@ -168,6 +185,8 @@ def get_label(address, group="addressbook"):
def set_label(address, label, group="addressbook"): def set_label(address, label, group="addressbook"):
"""Set address label in the address list given by group arg""" """Set address label in the address list given by group arg"""
if group not in _groups:
return
sqlExecute("UPDATE %s set label=? WHERE address=?" % group, label, address) sqlExecute("UPDATE %s set label=? WHERE address=?" % group, label, address)