Fix Bug [Athena Datasource] datasource other params illegal (#13032)

This commit is contained in:
Jrebel-i 2023-03-30 16:15:29 +08:00 committed by GitHub
parent d64004abea
commit 3630b00c9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -43,7 +43,7 @@ public abstract class AbstractDataSourceProcessor implements DataSourceProcessor
private static final Pattern DATABASE_PATTER = Pattern.compile("^[a-zA-Z0-9\\_\\-\\.]+$");
private static final Pattern PARAMS_PATTER = Pattern.compile("^[a-zA-Z0-9\\-\\_\\/\\@\\.]+$");
private static final Pattern PARAMS_PATTER = Pattern.compile("^[a-zA-Z0-9\\-\\_\\/\\@\\.\\:]+$");
private static final Set<String> POSSIBLE_MALICIOUS_KEYS = Sets.newHashSet("allowLoadLocalInfile");

View File

@ -54,6 +54,7 @@ public class AbstractDataSourceProcessorTest {
AbstractDataSourceProcessor mockDataSourceProcessor = mock(AbstractDataSourceProcessor.class);
Map<String, String> other = new HashMap<>();
other.put("allowLoadLocalInfile", "whatever");
other.put("OutputLocation", "s3://test-data-xini/athena-query");
doThrow(new IllegalArgumentException()).when(mockDataSourceProcessor).checkOther(other);
}