mirror/dble
1
0
Fork 0
mirror of https://github.com/actiontech/dble.git synced 2026-01-08 05:40:15 -06:00
Code Issues Packages Projects Releases Wiki Activity

cherry pick from inner-692

Browse Source
This commit is contained in:
wd2365151147
2020-11-19 15:20:00 +08:00
parent a9f11fb3e7
commit 8c35827cec
3 changed files with 21 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/actiontech/dble/config/model/user/ManagerUserConfig.java
View File

@@ -34,8 +34,8 @@ public class ManagerUserConfig extends UserConfig {
schemaInfo.setSchema((schemaInfo.getSchema().toLowerCase())); schemaInfo.setSchema((schemaInfo.getSchema().toLowerCase()));
schemaInfo.setTable(schemaInfo.getTable().toLowerCase()); schemaInfo.setTable(schemaInfo.getTable().toLowerCase());
if (!ManagerSchemaInfo.SCHEMA_NAME.equals(schemaInfo.getSchema())) { if (!ManagerSchemaInfo.SCHEMA_NAME.equals(schemaInfo.getSchema())) {
String msg = "Access denied for user '" + user + "' to database '" + schemaInfo.getSchema() + "'"; String msg = "Unknown database '" + schemaInfo.getSchema() + "'";
throw new SQLException(msg, "HY000", ErrorCode.ER_DBACCESS_DENIED_ERROR); throw new SQLException(msg, "42000", ErrorCode.ER_BAD_DB_ERROR);
} }
if (!ManagerSchemaInfo.getInstance().getTables().containsKey(schemaInfo.getTable())) { if (!ManagerSchemaInfo.getInstance().getTables().containsKey(schemaInfo.getTable())) {
String msg = "Table " + StringUtil.getFullName(schemaInfo.getSchema(), schemaInfo.getTable()) + " doesn't exist"; String msg = "Table " + StringUtil.getFullName(schemaInfo.getSchema(), schemaInfo.getTable()) + " doesn't exist";

15
src/main/java/com/actiontech/dble/services/manager/response/Describe.java
View File

@@ -5,7 +5,6 @@
package com.actiontech.dble.services.manager.response; package com.actiontech.dble.services.manager.response;
import com.actiontech.dble.backend.mysql.PacketUtil; import com.actiontech.dble.backend.mysql.PacketUtil;
import com.actiontech.dble.config.ErrorCode;
import com.actiontech.dble.config.Fields; import com.actiontech.dble.config.Fields;
import com.actiontech.dble.meta.ColumnMeta; import com.actiontech.dble.meta.ColumnMeta;
import com.actiontech.dble.net.mysql.*; import com.actiontech.dble.net.mysql.*;
@@ -64,21 +63,13 @@ public final class Describe {
SQLStatement statement = RouteStrategyFactory.getRouteStrategy().parserSQL(stmt); SQLStatement statement = RouteStrategyFactory.getRouteStrategy().parserSQL(stmt);
MySqlExplainStatement describeStatement = (MySqlExplainStatement) statement; MySqlExplainStatement describeStatement = (MySqlExplainStatement) statement;
SchemaUtil.SchemaInfo schemaInfo = SchemaUtil.getSchemaInfo(service.getUser(), service.getSchema(), describeStatement.getTableName(), null); SchemaUtil.SchemaInfo schemaInfo = SchemaUtil.getSchemaInfo(service.getUser(), service.getSchema(), describeStatement.getTableName(), null);
schemaName = schemaInfo.getSchema().toLowerCase(); // schemaName = schemaInfo.getSchema().toLowerCase();
tableName = schemaInfo.getTable().toLowerCase(); tableName = schemaInfo.getTable().toLowerCase();
} catch (SQLException e) { } catch (SQLException e) {
service.writeErrMessage(e.getSQLState(), e.getMessage(), e.getErrorCode()); service.writeErrMessage(e.getSQLState(), e.getMessage(), e.getErrorCode());
return; return;
} }
if (!ManagerSchemaInfo.SCHEMA_NAME.equals(schemaName)) {
service.writeErrMessage("42000", "Unknown database '" + schemaName + "'", ErrorCode.ER_BAD_DB_ERROR);
return;
}
ManagerBaseTable table = ManagerSchemaInfo.getInstance().getTables().get(tableName);
if (table == null) {
service.writeErrMessage("42S02", " Table '" + ManagerSchemaInfo.SCHEMA_NAME + "." + tableName + "' doesn't exist", ErrorCode.ER_NO_SUCH_TABLE);
return;
}
ByteBuffer buffer = service.allocate(); ByteBuffer buffer = service.allocate();
// write header // write header
@@ -94,6 +85,8 @@ public final class Describe {
// write rows // write rows
byte packetId = EOF.getPacketId(); byte packetId = EOF.getPacketId();
ManagerBaseTable table = ManagerSchemaInfo.getInstance().getTables().get(tableName);
for (ColumnMeta column : table.getColumnsMeta()) { for (ColumnMeta column : table.getColumnsMeta()) {
RowDataPacket row = new RowDataPacket(FIELD_COUNT); RowDataPacket row = new RowDataPacket(FIELD_COUNT);
row.add(StringUtil.encode(column.getName(), service.getCharset().getResults())); row.add(StringUtil.encode(column.getName(), service.getCharset().getResults()));

15
src/main/java/com/actiontech/dble/services/mysqlauthenticate/util/AuthUtil.java
View File

@@ -14,6 +14,7 @@ import com.actiontech.dble.config.model.user.UserConfig;
import com.actiontech.dble.config.model.user.UserName; import com.actiontech.dble.config.model.user.UserName;
import com.actiontech.dble.net.connection.AbstractConnection; import com.actiontech.dble.net.connection.AbstractConnection;
import com.actiontech.dble.net.connection.FrontendConnection; import com.actiontech.dble.net.connection.FrontendConnection;
import com.actiontech.dble.services.manager.information.ManagerSchemaInfo;
import com.actiontech.dble.services.mysqlauthenticate.PluginName; import com.actiontech.dble.services.mysqlauthenticate.PluginName;
import com.actiontech.dble.services.mysqlauthenticate.SecurityUtil; import com.actiontech.dble.services.mysqlauthenticate.SecurityUtil;
import com.actiontech.dble.singleton.CapClientFoundRows; import com.actiontech.dble.singleton.CapClientFoundRows;
@@ -74,6 +75,13 @@ public final class AuthUtil {
if ((Capabilities.CLIENT_FOUND_ROWS == (Capabilities.CLIENT_FOUND_ROWS & clientFlags)) != CapClientFoundRows.getInstance().isEnableCapClientFoundRows()) { if ((Capabilities.CLIENT_FOUND_ROWS == (Capabilities.CLIENT_FOUND_ROWS & clientFlags)) != CapClientFoundRows.getInstance().isEnableCapClientFoundRows()) {
return "The client requested CLIENT_FOUND_ROWS capabilities does not match, in the manager use show @@cap_client_found_rows check latest status."; return "The client requested CLIENT_FOUND_ROWS capabilities does not match, in the manager use show @@cap_client_found_rows check latest status.";
} }
} else if (userConfig instanceof ManagerUserConfig) {
switch (checkManagerSchema(schema)) {
case ErrorCode.ER_BAD_DB_ERROR:
return "Unknown database '" + schema + "'";
default:
break;
}
} }
//check maxconnection //check maxconnection
@@ -167,4 +175,11 @@ public final class AuthUtil {
} }
} }
private static int checkManagerSchema(String schema) {
if (schema != null && !ManagerSchemaInfo.SCHEMA_NAME.equals(schema.toLowerCase())) {
return ErrorCode.ER_BAD_DB_ERROR;
}
return 0;
}
} }