HIVE-20355: Clean up parameter of HiveConnection.setSchema (Daniel Da…

…i, reviewed by Sankar Hariappan)
sekingme · Sep 15, 2018 · 2c8e679 · 2c8e679
1 parent 01083a6
commit 2c8e679
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jdbc/src/java/org/apache/hive/jdbc/HiveConnection.java b/jdbc/src/java/org/apache/hive/jdbc/HiveConnection.java
@@ -1492,6 +1492,9 @@ public void setSchema(String schema) throws SQLException {
     if (schema == null || schema.isEmpty()) {
       throw new SQLException("Schema name is null or empty");
     }
+    if (schema.contains(";")) {
+      throw new SQLException("invalid schema name");
+    }
     Statement stmt = createStatement();
     stmt.execute("use " + schema);
     stmt.close();