def _update_to_db(conn, host):
sql_string = "UPDATE host_tbl SET ipv4 = %s, ipv6 = %s, dns = %s, net_name = %s, \
country = %s, oraganization = %s, last_updated = %s"
result = _sql._excute_without_return(conn, sql_string,
tuple([*host.values()]))
return result
def _update_password(conn, user_name, password):
user_temp = _get_by_name(conn, user_name)
if not user_temp:
sql_string = "UPDATE user_tbl u SET u.password = %s WHERE u.user_name = %s"
result = _sql._excute_without_return(conn, sql_string,
tuple([password, user_name]))
return result
return None
def _add_verify_code(conn, verify):
sql_string = "INSERT INTO verify_tbl(time_stamp, user_name, verify_code) \
VALUES(%s, %s, %s)"
result = _sql._excute_without_return(
conn, sql_string,
tuple(
[verify["time_stamp"], verify["user_name"],
verify["verify_code"]]))
return result
def add_host_to_db(conn, host):
host_stamp = _get_by_ip(conn, host)
if host_stamp is None:
# sql_string = "INSERT INTO host_tbl(ipv4,ipv6,dns,net_name,country,oraganization,asn,last_updated)"
sql_string = "INSERT INTO host_tbl(ipv4,ipv6,dns,net_name,country,oraganization,last_updated) \
VALUES(%s,%s,%s,%s,%s,%s,%s)"
result = _sql._excute_without_return(conn, sql_string,
tuple([*host.values()]))
return result
def _add(conn, user):
user_temp = _get_by_name(conn, user["userName"])
if user_temp is None:
sql_string = "INSERT INTO user_tbl(user_name, password, full_name, gender, email, \
phone, address, credits, user_type) \
VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s)"
result = _sql._excute_without_return(
conn, sql_string, tuple([*_convert.user_dict_to_list(user)]))
return result
return None
def _update_info(conn, user):
user_temp = _get_by_name(conn, user["userName"])
if user_temp is not None:
#genarate sql string
sql_string = "UPDATE user_tbl u SET u.full_name = %s, u.gender = %s, u.email = %s, \
u.phone = %s, u.address = %s, u.credits = %s, \
u.user_type = %s WHERE u.user_name = %s"
params = tuple([user["fullName"], user["gender"], \
user["emailAddress"], user["phoneNumber"], user["address"], \
user["credits"], user["userType"], user["userName"]])
result = _sql._excute_without_return(conn, sql_string, params)
return result
return None
def _update_user_type(conn, user):
sql_string = "UPDATE user_tbl SET user_type = %s WHERE user_name = %s"
result = _sql._excute_without_return(
conn, sql_string, tuple([user["userType"], user["userName"]]))
return result
def _add_to_db(conn, port):
sql_string = "INSERT INTO port_tbl(port_num, host_ip, service_name, version, status, cpe) \
VALUES(%s,%s,%s,%s,%s,%s)"
result = _sql._excute_without_return(conn, sql_string, tuple([*port.values()]))
return result
def _delete_verify_code(conn, user_name):
sql_string = "DELETE FROM verify_tbl WHERE user_name = %s"
result = _sql._excute_without_return(conn, sql_string, tuple([user_name]))
return result
def expiry_code(conn, time_stamp):
sql_string = "DELETE FROM verify_tbl WHERE time_stamp < %s"
result = _sql._excute_without_return(conn, sql_string, tuple([time_stamp]))
return result
def _update_status(conn, data):
sql_string = "UPDATE task_tbl SET status = %s WHERE user_name = %s AND ipv4 = %s"
result = _sql._excute_without_return(conn, sql_string, tuple([data[2], data[0], data[1]]))
return result
def _add_to_db(conn, data):
sql_string = "INSERT INTO task_tbl(ipv4, user_name, status, time_stamp) \
VALUE(%s,%s,%s,%s)"
result = _sql._excute_without_return(conn, sql_string, tuple([*data.values()]))
return result
def _update_credits(conn, user_name, _credits):
sql_string = """UPDATE user_tbl SET credits = %s WHERE user_name = %s"""
result = _sql._excute_without_return(conn, sql_string,
tuple([_credits, user_name]))
return result
def _update_by_id(conn, vuln):
sql_string = """UPDATE vuln_tbl SET cve_status = %s, cve_desc = %s WHERE cve_num = %s;"""
result = _sql._excute_without_return(conn, sql_string,
tuple(vuln[-1:] + vuln[:-1]))
return result
def _update_by_id(conn, port):
sql_string = "UPDATE port_tbl SET port_num = %s, host_ip = %s, service_name = %s, \
version = %s, status = %s, cpe = %s WHERE port_id = %s"
result = _sql._excute_without_return(conn, sql_string, tuple([*port.values()]))
return result
def _add_to_db(conn, port_vuln):
sql_string = "INSERT INTO port_vuln_tbl(port_id, cve_num, last_updated) \
VALUES(%s,%s,%s)"
result = _sql._excute_without_return(conn, sql_string, tuple([*port_vuln]))
return result
def _update_by_id(conn, port_vuln):
sql_string = "UPDATE port_vuln_tbl SET cve_num = %s, last_updated = %s WHERE port_id = %s"
result = _sql._excute_without_return(conn, sql_string, tuple([*port_vuln]))
return result
def _delete_from_db(conn, host):
sql_string = "DELETE * FROM host_tbl WHERE ipv4 = %s"
result = _sql._excute_without_return(conn, sql_string,
tuple([host["ipv4"]]))
return result
def _add_to_db(conn, vuln):
sql_string = """INSERT INTO vuln_tbl(cve_num, cve_status, cve_desc) VALUES(%s, %s, %s);"""
result = _sql._excute_without_return(conn, sql_string, tuple(vuln))
return result
