Skip to content
Snippets Groups Projects
Unverified Commit c5f90e46 authored by Richard Cordovano's avatar Richard Cordovano Committed by GitHub
Browse files

Merge pull request #1969 from APriestman/6596_escapePostgresURL

6596 Encode database name for postgresql url
parents 10169f5d b0ca3094
No related branches found
No related tags found
No related merge requests found
......@@ -19,11 +19,15 @@
package org.sleuthkit.datamodel;
import java.io.File;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;
import java.util.logging.Logger;
import java.util.logging.Level;
import org.sleuthkit.datamodel.SQLHelper.PostgreSQLHelper;
import org.sleuthkit.datamodel.SQLHelper.SQLiteHelper;
......@@ -32,6 +36,7 @@
*/
class CaseDatabaseFactory {
private static final Logger logger = Logger.getLogger(CaseDatabaseFactory.class.getName());
private final SQLHelper dbQueryHelper;
private final DbCreationHelper dbCreationHelper;
......@@ -511,12 +516,20 @@ Connection getPostgresConnection() throws TskCoreException {
* @return the connection to the database
*/
Connection getConnection(String databaseName) throws TskCoreException {
String encodedDbName;
try {
encodedDbName = URLEncoder.encode(databaseName, "UTF-8");
} catch (UnsupportedEncodingException ex) {
// Print the warning and continue with the unencoded name
logger.log(Level.WARNING, "Error encoding database name " + databaseName, ex);
encodedDbName = databaseName;
}
StringBuilder url = new StringBuilder();
url.append(JDBC_BASE_URI)
.append(info.getHost())
.append('/') // NON-NLS
.append(databaseName);
.append(encodedDbName);
Connection conn;
try {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment