Skip to content

Commit cdac0dd

Browse files
SQL Injection - DB Bug Resolved
1 parent 560c011 commit cdac0dd

5 files changed

Lines changed: 7 additions & 6 deletions

File tree

.idea/dataSources/d0a64771-9f67-47d7-be54-8ad5c1cfc040.xml

Lines changed: 2 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
16 KB
Binary file not shown.

app.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ def settings():
119119

120120
return render_template('settings.html', level=level, msg="Difficulty Set to " + level)
121121

122+
122123
# Reset DB
123124
@app.route('/reset-db')
124125
@is_logged
@@ -128,6 +129,7 @@ def reset_db():
128129

129130
return render_template('settings.html', msg="Database has been reset!")
130131

132+
131133
# SQL Injection
132134
@app.route('/sql-injection', methods=['POST', 'GET'])
133135
@is_logged

helper/db_manager.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,10 @@ def __init__(self):
88
# Initialize Database
99
self.conn = None
1010
self.cur = None
11+
self.create_db_connection()
1112

1213
def create_db_connection(self):
13-
self.conn = sqlite3.connect('TIWAF.db', check_same_thread=False)
14+
self.conn = sqlite3.connect('TIWAP.db', check_same_thread=False)
1415
self.cur = self.conn.cursor()
1516

1617
def get_db_connection(self):
@@ -116,4 +117,4 @@ def reset_db(self):
116117
self.cur.execute(content)
117118

118119
self.close_db_connection()
119-
self.commit_db()
120+
self.commit_db()

uploads/images.png

2.73 KB
Loading

0 commit comments

Comments
 (0)