Avoid hardcoding database driver name.

k-tamura · Jan 25, 2017 · 1efb39d · 1efb39d
1 parent f7a730a
commit 1efb39d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/main/java/org/t246osslab/easybuggy/troubles/DBConnectionLeakServlet.java b/src/main/java/org/t246osslab/easybuggy/troubles/DBConnectionLeakServlet.java
@@ -39,7 +39,7 @@ protected void service(HttpServletRequest req, HttpServletResponse res) throws S
 
             if (dbDriver != null && !dbDriver.equals("")) {
                 try {
-                    Class.forName("com.mysql.jdbc.Driver");
+                    Class.forName(ApplicationUtils.getDatabaseDriver());
                 } catch (ClassNotFoundException e) {
                     Logger.error(e);
                 }

diff --git a/src/main/java/org/t246osslab/easybuggy/troubles/DeadlockServlet2.java b/src/main/java/org/t246osslab/easybuggy/troubles/DeadlockServlet2.java
@@ -80,7 +80,7 @@ class EmbeddedJavaDb2 {
         try {
             if (dbDriver != null && !dbDriver.equals("")) {
                 try {
-                    Class.forName("com.mysql.jdbc.Driver");
+                    Class.forName(ApplicationUtils.getDatabaseDriver());
                 } catch (ClassNotFoundException e) {
                     Logger.error(e);
                 }
@@ -130,7 +130,7 @@ public String update(int[] ids, Locale locale) {
         try {
             if (dbDriver != null && !dbDriver.equals("")) {
                 try {
-                    Class.forName("com.mysql.jdbc.Driver");
+                    Class.forName(ApplicationUtils.getDatabaseDriver());
                 } catch (ClassNotFoundException e) {
                     Logger.error(e);
                 }

diff --git a/src/main/java/org/t246osslab/easybuggy/vulnerabilities/SQLInjectionServlet.java b/src/main/java/org/t246osslab/easybuggy/vulnerabilities/SQLInjectionServlet.java
@@ -78,7 +78,7 @@ class EmbeddedJavaDb {
             String dbDriver = ApplicationUtils.getDatabaseDriver();
             if (dbDriver != null && !dbDriver.equals("")) {
                 try {
-                    Class.forName("com.mysql.jdbc.Driver");
+                    Class.forName(ApplicationUtils.getDatabaseDriver());
                 } catch (ClassNotFoundException e) {
                     Logger.error(e);
                 }