org.mariadb.jdbc.util.NativeSql Maven / Gradle / Ivy
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (c) 2012-2014 Monty Program Ab
// Copyright (c) 2015-2021 MariaDB Corporation Ab
package org.mariadb.jdbc.util;
import java.sql.SQLException;
import java.util.Locale;
import org.mariadb.jdbc.client.Context;
public final class NativeSql {
public static String parse(String sql, Context context) throws SQLException {
if (!sql.contains("{")) {
return sql;
}
ClientParser.LexState state = ClientParser.LexState.Normal;
char lastChar = '\0';
boolean singleQuotes = false;
int lastEscapePart = 0;
StringBuilder sb = new StringBuilder();
char[] query = sql.toCharArray();
int queryLength = query.length;
int escapeIdx = 0;
boolean inEscape = false;
for (int idx = 0; idx < queryLength; idx++) {
char car = query[idx];
if (state == ClientParser.LexState.Escape
&& !((car == '\'' && singleQuotes) || (car == '"' && !singleQuotes))) {
state = ClientParser.LexState.String;
if (!inEscape) sb.append(car);
lastChar = car;
continue;
}
switch (car) {
case '*':
if (state == ClientParser.LexState.Normal && lastChar == '/') {
state = ClientParser.LexState.SlashStarComment;
}
break;
case '/':
if (state == ClientParser.LexState.SlashStarComment && lastChar == '*') {
state = ClientParser.LexState.Normal;
} else if (state == ClientParser.LexState.Normal && lastChar == '/') {
state = ClientParser.LexState.EOLComment;
}
break;
case '#':
if (state == ClientParser.LexState.Normal) {
state = ClientParser.LexState.EOLComment;
}
break;
case '-':
if (state == ClientParser.LexState.Normal && lastChar == '-') {
state = ClientParser.LexState.EOLComment;
}
break;
case '\n':
if (state == ClientParser.LexState.EOLComment) {
state = ClientParser.LexState.Normal;
}
break;
case '"':
if (state == ClientParser.LexState.Normal) {
state = ClientParser.LexState.String;
singleQuotes = false;
} else if (state == ClientParser.LexState.String && !singleQuotes) {
state = ClientParser.LexState.Normal;
} else if (state == ClientParser.LexState.Escape) {
state = ClientParser.LexState.String;
}
break;
case '\'':
if (state == ClientParser.LexState.Normal) {
state = ClientParser.LexState.String;
singleQuotes = true;
} else if (state == ClientParser.LexState.String && singleQuotes) {
state = ClientParser.LexState.Normal;
} else if (state == ClientParser.LexState.Escape) {
state = ClientParser.LexState.String;
}
break;
case '\\':
if (state == ClientParser.LexState.String) {
state = ClientParser.LexState.Escape;
}
break;
case '`':
if (state == ClientParser.LexState.Backtick) {
state = ClientParser.LexState.Normal;
} else if (state == ClientParser.LexState.Normal) {
state = ClientParser.LexState.Backtick;
}
break;
case '{':
if (state == ClientParser.LexState.Normal) {
if (!inEscape) {
inEscape = true;
lastEscapePart = idx;
}
escapeIdx++;
}
break;
case '}':
if (state == ClientParser.LexState.Normal) {
if (inEscape) {
escapeIdx--;
if (escapeIdx == 0) {
String str = sql.substring(lastEscapePart, idx + 1);
String escapedSeq = resolveEscapes(str, context);
sb.append(escapedSeq);
inEscape = false;
continue;
}
}
}
break;
}
if (!inEscape) sb.append(car);
lastChar = car;
}
if (inEscape) {
throw new SQLException(
"Invalid escape sequence , missing closing '}' character in '" + sql + "'");
}
return sb.toString();
}
private static String resolveEscapes(String escaped, Context context) throws SQLException {
int endIndex = escaped.length() - 1;
String escapedLower = escaped.toLowerCase(Locale.ROOT);
if (escaped.startsWith("{fn ")) {
String resolvedParams = replaceFunctionParameter(escaped.substring(4, endIndex), context);
return parse(resolvedParams, context);
} else if (escapedLower.startsWith("{oj ")) {
// Outer join
// the server supports "oj" in any case, even "oJ"
return parse(escaped.substring(4, endIndex), context);
} else if (escaped.startsWith("{d ")) {
// date literal
return escaped.substring(3, endIndex);
} else if (escaped.startsWith("{t ")) {
// time literal
return escaped.substring(3, endIndex);
} else if (escaped.startsWith("{ts ")) {
// timestamp literal
return escaped.substring(4, endIndex);
} else if (escaped.startsWith("{d'")) {
// date literal, no space
return escaped.substring(2, endIndex);
} else if (escaped.startsWith("{t'")) {
// time literal
return escaped.substring(2, endIndex);
} else if (escaped.startsWith("{ts'")) {
// timestamp literal
return escaped.substring(3, endIndex);
} else if (escaped.startsWith("{call ") || escaped.startsWith("{CALL ")) {
// We support uppercase "{CALL" only because Connector/J supports it. It is not in the JDBC
// spec.
return parse(escaped.substring(1, endIndex), context);
} else if (escaped.startsWith("{escape ")) {
return escaped.substring(1, endIndex);
} else if (escaped.startsWith("{?")) {
// likely ?=call(...)
return parse(escaped.substring(1, endIndex), context);
} else if (escaped.startsWith("{ ") || escaped.startsWith("{\n")) {
// Spaces and newlines before keyword, this is not JDBC compliant, however some it works in
// some drivers,
// so we support it, too
for (int i = 2; i < escaped.length(); i++) {
if (!Character.isWhitespace(escaped.charAt(i))) {
return resolveEscapes("{" + escaped.substring(i), context);
}
}
} else if (escaped.startsWith("{\r\n")) {
// Spaces and newlines before keyword, this is not JDBC compliant, however some it works in
// some drivers,
// so we support it, too
for (int i = 3; i < escaped.length(); i++) {
if (!Character.isWhitespace(escaped.charAt(i))) {
return resolveEscapes("{" + escaped.substring(i), context);
}
}
}
throw new SQLException("unknown escape sequence " + escaped);
}
/**
* Helper function to replace function parameters in escaped string. 3 functions are handles :
*
*
* - CONVERT(value, type): replacing SQL_XXX types to convertible type, i.e SQL_BIGINT to
* INTEGER
*
- TIMESTAMPDIFF(type, ...): replacing type SQL_TSI_XXX in type with XXX, i.e SQL_TSI_HOUR
* with HOUR
*
- TIMESTAMPADD(type, ...): replacing type SQL_TSI_XXX in type with XXX, i.e SQL_TSI_HOUR
* with HOUR
*
*
* caution: this use MariaDB server conversion: 'SELECT CONVERT('2147483648', INTEGER)' will
* return a BIGINT. MySQL will throw a syntax error.
*
* @param functionString input string
* @return unescaped string
*/
private static String replaceFunctionParameter(String functionString, Context context) {
char[] input = functionString.toCharArray();
StringBuilder sb = new StringBuilder();
int index;
for (index = 0; index < input.length; index++) {
if (input[index] != ' ') {
break;
}
}
for (;
index < input.length
&& ((input[index] >= 'a' && input[index] <= 'z')
|| (input[index] >= 'A' && input[index] <= 'Z'));
index++) {
sb.append(input[index]);
}
String func = sb.toString().toLowerCase(Locale.ROOT);
switch (func) {
case "convert":
// Handle "convert(value, type)" case
// extract last parameter, after the last ','
int lastCommaIndex = functionString.lastIndexOf(',');
int firstParentheses = functionString.indexOf('(');
String value = functionString.substring(firstParentheses + 1, lastCommaIndex);
for (index = lastCommaIndex + 1; index < input.length; index++) {
if (!Character.isWhitespace(input[index])) {
break;
}
}
int endParam = index + 1;
for (; endParam < input.length; endParam++) {
if ((input[endParam] < 'a' || input[endParam] > 'z')
&& (input[endParam] < 'A' || input[endParam] > 'Z')
&& input[endParam] != '_') {
break;
}
}
String typeParam = new String(input, index, endParam - index).toUpperCase(Locale.ROOT);
if (typeParam.startsWith("SQL_")) {
typeParam = typeParam.substring(4);
}
switch (typeParam) {
case "BOOLEAN":
return "1=" + value;
case "BIGINT":
case "SMALLINT":
case "TINYINT":
typeParam = "SIGNED INTEGER";
break;
case "BIT":
typeParam = "UNSIGNED INTEGER";
break;
case "BLOB":
case "VARBINARY":
case "LONGVARBINARY":
case "ROWID":
typeParam = "BINARY";
break;
case "NCHAR":
case "CLOB":
case "NCLOB":
case "DATALINK":
case "VARCHAR":
case "NVARCHAR":
case "LONGVARCHAR":
case "LONGNVARCHAR":
case "SQLXML":
case "LONGNCHAR":
typeParam = "CHAR";
break;
case "DOUBLE":
case "FLOAT":
if (context.getVersion().isMariaDBServer()
|| context.getVersion().versionGreaterOrEqual(8, 0, 17)) {
typeParam = "DOUBLE";
break;
}
return "0.0+" + value;
case "REAL":
case "NUMERIC":
typeParam = "DECIMAL";
break;
case "TIMESTAMP":
typeParam = "DATETIME";
break;
default:
break;
}
return new String(input, 0, index)
+ typeParam
+ new String(input, endParam, input.length - endParam);
case "timestampdiff":
case "timestampadd":
// Skip to first parameter
for (; index < input.length; index++) {
if (!Character.isWhitespace(input[index]) && input[index] != '(') {
break;
}
}
if (index < input.length - 8) {
String paramPrefix = new String(input, index, 8);
if ("SQL_TSI_".equals(paramPrefix)) {
return new String(input, 0, index)
+ new String(input, index + 8, input.length - (index + 8));
}
}
return functionString;
default:
return functionString;
}
}
}