From 86eafc4d77f0bbc0eaa7fc58e5076a085257f259 Mon Sep 17 00:00:00 2001 From: taojinlong Date: Tue, 8 Oct 2024 18:47:57 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=20pg=20jdbc=20=E6=A0=A1=E9=AA=8C?= =?UTF-8?q?=E9=9D=9E=E6=B3=95=E5=AD=97=E7=AC=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../io/dataease/dto/datasource/PgConfiguration.java | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/core/backend/src/main/java/io/dataease/dto/datasource/PgConfiguration.java b/core/backend/src/main/java/io/dataease/dto/datasource/PgConfiguration.java index f1365071dd..400af7852f 100644 --- a/core/backend/src/main/java/io/dataease/dto/datasource/PgConfiguration.java +++ b/core/backend/src/main/java/io/dataease/dto/datasource/PgConfiguration.java @@ -5,12 +5,18 @@ import lombok.Getter; import lombok.Setter; import org.apache.commons.lang3.StringUtils; +import java.net.URLDecoder; +import java.util.Arrays; +import java.util.List; + @Getter @Setter public class PgConfiguration extends JdbcConfiguration { private String driver = "org.postgresql.Driver"; private String extraParams = ""; + private List illegalParameters = Arrays.asList("socketFactory", "socketFactoryArg", "sslfactory", "sslfactoryarg", "loggerLevel", "loggerFile", "allowUrlInLocalInfile", "allowLoadLocalInfileInPath"); + public String getJdbc() { if (StringUtils.isEmpty(extraParams.trim())) { @@ -27,6 +33,11 @@ public class PgConfiguration extends JdbcConfiguration { .replace("SCHEMA", getSchema().trim()); } } else { + for (String illegalParameter : illegalParameters) { + if (getExtraParams().toLowerCase().contains(illegalParameter.toLowerCase()) || URLDecoder.decode(getExtraParams()).contains(illegalParameter.toLowerCase())) { + throw new RuntimeException("Illegal parameter: " + illegalParameter); + } + } return "jdbc:postgresql://HOSTNAME:PORT/DATABASE?EXTRA_PARAMS" .replace("HOSTNAME", getHost().trim()) .replace("PORT", getPort().toString().trim())