instrumentation.TranslateExceptions Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mysql-connector-java Show documentation
Show all versions of mysql-connector-java Show documentation
Fixed bugs repo for official MySQL JDBC connector
The newest version!
/*
* Copyright (c) 2015, 2022, Oracle and/or its affiliates.
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License, version 2.0, as published by the
* Free Software Foundation.
*
* This program is also distributed with certain software (including but not
* limited to OpenSSL) that is licensed under separate terms, as designated in a
* particular file or component or in included license documentation. The
* authors of MySQL hereby grant you an additional permission to link the
* program and your derivative works with the separately licensed software that
* they have included with MySQL.
*
* Without limiting anything contained in the foregoing, this file, which is
* part of MySQL Connector/J, is also subject to the Universal FOSS Exception,
* version 1.0, a copy of which can be found at
* http://oss.oracle.com/licenses/universal-foss-exception.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License, version 2.0,
* for more details.
*
* You should have received a copy of the GNU General Public License along with
* this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
package instrumentation;
import java.lang.reflect.Method;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Savepoint;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.TreeMap;
import com.mysql.cj.QueryBindings;
import com.mysql.cj.exceptions.CJException;
import com.mysql.cj.jdbc.Blob;
import com.mysql.cj.jdbc.BlobFromLocator;
import com.mysql.cj.jdbc.CallableStatement;
import com.mysql.cj.jdbc.CallableStatement.CallableStatementParamInfo;
import com.mysql.cj.jdbc.ClientPreparedStatement;
import com.mysql.cj.jdbc.Clob;
import com.mysql.cj.jdbc.ConnectionImpl;
import com.mysql.cj.jdbc.ConnectionWrapper;
import com.mysql.cj.jdbc.DatabaseMetaData;
import com.mysql.cj.jdbc.DatabaseMetaDataUsingInfoSchema;
import com.mysql.cj.jdbc.JdbcConnection;
import com.mysql.cj.jdbc.JdbcStatement;
import com.mysql.cj.jdbc.MysqlConnectionPoolDataSource;
import com.mysql.cj.jdbc.MysqlDataSource;
import com.mysql.cj.jdbc.MysqlParameterMetadata;
import com.mysql.cj.jdbc.MysqlPooledConnection;
import com.mysql.cj.jdbc.MysqlSQLXML;
import com.mysql.cj.jdbc.MysqlSavepoint;
import com.mysql.cj.jdbc.MysqlXAConnection;
import com.mysql.cj.jdbc.MysqlXADataSource;
import com.mysql.cj.jdbc.MysqlXid;
import com.mysql.cj.jdbc.NClob;
import com.mysql.cj.jdbc.NonRegisteringDriver;
import com.mysql.cj.jdbc.ServerPreparedStatement;
import com.mysql.cj.jdbc.StatementImpl;
import com.mysql.cj.jdbc.SuspendableXAConnection;
import com.mysql.cj.jdbc.exceptions.SQLExceptionsMapping;
import com.mysql.cj.jdbc.ha.LoadBalancedConnection;
import com.mysql.cj.jdbc.ha.LoadBalancedMySQLConnection;
import com.mysql.cj.jdbc.ha.MultiHostMySQLConnection;
import com.mysql.cj.jdbc.ha.ReplicationConnection;
import com.mysql.cj.jdbc.ha.ReplicationMySQLConnection;
import com.mysql.cj.jdbc.result.ResultSetImpl;
import com.mysql.cj.jdbc.result.ResultSetInternalMethods;
import com.mysql.cj.jdbc.result.ResultSetMetaData;
import com.mysql.cj.jdbc.result.UpdatableResultSet;
import com.mysql.cj.protocol.ColumnDefinition;
import com.mysql.cj.protocol.Message;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.NotFoundException;
public class TranslateExceptions {
private static CtClass runTimeException = null;
private static ClassPool pool = ClassPool.getDefault();
private static Map> processed = new TreeMap<>();
private static String EXCEPTION_INTERCEPTOR_GETTER = "getExceptionInterceptor()";
private static String EXCEPTION_INTERCEPTOR_MEMBER = "this.exceptionInterceptor";
private static boolean verbose = false;
public static void main(String[] args) throws Exception {
System.out.println("Applying TranslateExceptions.");
verbose = "true".equalsIgnoreCase(args[1]);
pool.insertClassPath(args[0]);
processed.clear();
runTimeException = pool.get(CJException.class.getName());
// params classes
//CtClass ctServerPreparedQueryBindValue = pool.get(ServerPreparedQueryBindValue.class.getName());
CtClass ctQueryBindings = pool.get(QueryBindings.class.getName());
//CtClass ctByteArray = pool.get(byte[].class.getName());
CtClass ctColumnDefinition = pool.get(ColumnDefinition.class.getName());
CtClass ctLongArray = pool.get(long[].class.getName());
//CtClass ctInputStream = pool.get(InputStream.class.getName());
CtClass ctJdbcConnection = pool.get(JdbcConnection.class.getName());
CtClass ctMysqlSavepoint = pool.get(MysqlSavepoint.class.getName());
//CtClass ctPacketPayload = pool.get(PacketPayload.class.getName());
CtClass ctProperties = pool.get(Properties.class.getName());
CtClass ctResultSet = pool.get(ResultSet.class.getName());
CtClass ctResultSetInternalMethods = pool.get(ResultSetInternalMethods.class.getName());
CtClass ctStatement = pool.get(java.sql.Statement.class.getName());
CtClass ctStatementImpl = pool.get(StatementImpl.class.getName());
CtClass ctString = pool.get(String.class.getName());
CtClass ctMessageBody = pool.get(Message.class.getName());
// class we want to instrument
CtClass clazz;
/*
* java.sql.Blob
*/
// com.mysql.cj.jdbc.Blob implements java.sql.Blob, OutputStreamWatcher
clazz = pool.get(Blob.class.getName());
instrumentJdbcMethods(clazz, java.sql.Blob.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.BlobFromLocator implements java.sql.Blob
clazz = pool.get(BlobFromLocator.class.getName());
instrumentJdbcMethods(clazz, java.sql.Blob.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* java.sql.CallableStatement
*/
// com.mysql.cj.jdbc.CallableStatement extends PreparedStatement implements java.sql.CallableStatement
clazz = pool.get(CallableStatement.class.getName());
instrumentJdbcMethods(clazz, java.sql.CallableStatement.class, false, EXCEPTION_INTERCEPTOR_GETTER);
instrumentJdbcMethods(clazz, JdbcStatement.class, true, EXCEPTION_INTERCEPTOR_GETTER);
// non-JDBC
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkIsOutputParam", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkParameterIndexBounds", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkReadOnlyProcedure", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("convertGetProcedureColumnsToInternalDescriptors", new CtClass[] { ctResultSet }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("determineParameterTypes", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("fakeParameterTypes", new CtClass[] { CtClass.booleanType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("fixParameterName", new CtClass[] { ctString }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("generateParameterMap", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getNamedParamIndex", new CtClass[] { ctString, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getOutputParameters", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("mapOutputParameterIndexToRsIndex", new CtClass[] { CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("retrieveOutParams", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setInOutParamsOnServer", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setOutParams", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* com.mysql.cj.jdbc.StatementWrapper extends WrapperBase implements Statement
*/
// TODO: Does it's own typical exception wrapping, could be instrumented with different catch method
/*
* com.mysql.cj.jdbc.PreparedStatementWrapper extends StatementWrapper implements PreparedStatement
*/
// TODO: Does it's own typical exception wrapping, could be instrumented with different catch method
/*
* com.mysql.cj.jdbc.CallableStatementWrapper extends PreparedStatementWrapper implements CallableStatement
*/
// TODO: Does it's own typical exception wrapping, could be instrumented with different catch method
/*
* java.sql.Clob
*/
// com.mysql.cj.jdbc.Clob implements java.sql.Clob, OutputStreamWatcher, WriterWatcher
clazz = pool.get(Clob.class.getName());
instrumentJdbcMethods(clazz, java.sql.Clob.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
*
* java.sql.Connection extends java.sql.Wrapper
* ----> com.mysql.cj.jdbc.JdbcConnection extends java.sql.Connection, MysqlConnection
* ----------> com.mysql.cj.jdbc.ConnectionImpl
* ----------> com.mysql.cj.jdbc.LoadBalancedConnection extends JdbcConnection
* -------------> com.mysql.cj.jdbc.LoadBalancedMySQLConnection extends MultiHostMySQLConnection implements LoadBalancedConnection
* ----------> com.mysql.cj.jdbc.MultiHostMySQLConnection
* -------> com.mysql.cj.jdbc.ReplicationConnection implements JdbcConnection, PingTarget
* -------> com.mysql.cj.jdbc.ConnectionWrapper
*/
// ConnectionImpl extends AbstractJdbcConnection implements JdbcConnection
clazz = pool.get(ConnectionImpl.class.getName());
instrumentJdbcMethods(clazz, JdbcConnection.class, false, EXCEPTION_INTERCEPTOR_GETTER);
// non-JDBC
catchRuntimeException(clazz,
clazz.getDeclaredMethod("clientPrepareStatement", new CtClass[] { ctString, CtClass.intType, CtClass.intType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("createNewIO", new CtClass[] { CtClass.booleanType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getMetaData", new CtClass[] { CtClass.booleanType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("handleAutoCommitDefaults", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setSavepoint", new CtClass[] { ctMysqlSavepoint }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("versionMeetsMinimum", new CtClass[] { CtClass.intType, CtClass.intType, CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("rollbackNoChecks", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setupServerForTruncationChecks", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.LoadBalancedMySQLConnection extends MultiHostMySQLConnection implements LoadBalancedConnection
clazz = pool.get(LoadBalancedMySQLConnection.class.getName());
instrumentJdbcMethods(clazz, LoadBalancedConnection.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// MultiHostMySQLConnection implements JdbcConnection
clazz = pool.get(MultiHostMySQLConnection.class.getName());
instrumentJdbcMethods(clazz, JdbcConnection.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.ReplicationConnection implements JdbcConnection, PingTarget
clazz = pool.get(ReplicationMySQLConnection.class.getName());
instrumentJdbcMethods(clazz, ReplicationConnection.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// ConnectionWrapper extends WrapperBase implements JdbcConnection
clazz = pool.get(ConnectionWrapper.class.getName());
instrumentJdbcMethods(clazz, JdbcConnection.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
// non-JDBC
catchRuntimeException(clazz, clazz.getDeclaredMethod("clientPrepare", new CtClass[] { ctString }), EXCEPTION_INTERCEPTOR_MEMBER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("clientPrepare", new CtClass[] { ctString, CtClass.intType, CtClass.intType }),
EXCEPTION_INTERCEPTOR_MEMBER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setClientInfo", new CtClass[] { ctString, ctString }), EXCEPTION_INTERCEPTOR_MEMBER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setClientInfo", new CtClass[] { ctProperties }), EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* java.sql.DatabaseMetaData extends java.sql.Wrapper
*/
// com.mysql.cj.jdbc.DatabaseMetaData implements java.sql.DatabaseMetaData
clazz = pool.get(DatabaseMetaData.class.getName());
instrumentJdbcMethods(clazz, java.sql.DatabaseMetaData.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.DatabaseMetaDataUsingInfoSchema extends DatabaseMetaData
clazz = pool.get(DatabaseMetaDataUsingInfoSchema.class.getName());
instrumentJdbcMethods(clazz, java.sql.DatabaseMetaData.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* java.sql.Driver
*/
// com.mysql.cj.jdbc.Driver extends NonRegisteringDriver implements java.sql.Driver
clazz = pool.get(NonRegisteringDriver.class.getName());
instrumentJdbcMethods(clazz, java.sql.Driver.class);
clazz.writeFile(args[0]);
/*
* java.sql.NClob
*/
// com.mysql.cj.jdbc.NClob extends Clob implements java.sql.NClob
clazz = pool.get(NClob.class.getName());
instrumentJdbcMethods(clazz, java.sql.NClob.class);
clazz.writeFile(args[0]);
/*
* java.sql.ParameterMetaData extends java.sql.Wrapper
*/
// com.mysql.cj.jdbc.CallableStatement.CallableStatementParamInfo implements ParameterMetaData
clazz = pool.get(CallableStatementParamInfo.class.getName());
instrumentJdbcMethods(clazz, java.sql.ParameterMetaData.class);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.MysqlParameterMetadata implements ParameterMetaData
clazz = pool.get(MysqlParameterMetadata.class.getName());
instrumentJdbcMethods(clazz, java.sql.ParameterMetaData.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* java.sql.PreparedStatement extends java.sql.Statement (java.sql.Statement extends java.sql.Wrapper)
*/
// com.mysql.cj.jdbc.ClientPreparedStatement extends com.mysql.cj.jdbc.StatementImpl implements java.sql.PreparedStatement
clazz = pool.get(ClientPreparedStatement.class.getName());
instrumentJdbcMethods(clazz, java.sql.PreparedStatement.class, false, EXCEPTION_INTERCEPTOR_GETTER);
instrumentJdbcMethods(clazz, JdbcStatement.class, true, EXCEPTION_INTERCEPTOR_GETTER);
// non-JDBC
catchRuntimeException(clazz, clazz.getDeclaredMethod("toString", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkBounds", new CtClass[] { CtClass.intType, CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkReadOnlySafeStatement", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeBatchedInserts", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeBatchSerially", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz,
clazz.getDeclaredMethod("executeInternal",
new CtClass[] { CtClass.intType, ctMessageBody, CtClass.booleanType, CtClass.booleanType, ctColumnDefinition, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executePreparedBatchAsMultiStatement", new CtClass[] { CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeUpdateInternal", new CtClass[] { CtClass.booleanType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeUpdateInternal", new CtClass[] { ctQueryBindings, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("generateMultiStatementForBatch", new CtClass[] { CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getBytesRepresentation", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getParameterBindings", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("initializeFromQueryInfo", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("isNull", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("prepareBatchedInsertSQL", new CtClass[] { ctJdbcConnection, CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
// catchRuntimeException(clazz,
// clazz.getDeclaredMethod("setBytes", new CtClass[] { CtClass.intType, ctByteArray, CtClass.booleanType, CtClass.booleanType }),
// EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setRetrieveGeneratedKeys", new CtClass[] { CtClass.booleanType }), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* com.mysql.cj.jdbc.ServerPreparedStatement extends ClientPreparedStatement
*/
clazz = pool.get(ServerPreparedStatement.class.getName());
instrumentJdbcMethods(clazz, java.sql.PreparedStatement.class, false, EXCEPTION_INTERCEPTOR_GETTER);
instrumentJdbcMethods(clazz, JdbcStatement.class, true, EXCEPTION_INTERCEPTOR_GETTER);
// non-JDBC
catchRuntimeException(clazz, clazz.getDeclaredMethod("toString", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getBinding", new CtClass[] { CtClass.intType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz,
clazz.getDeclaredMethod("executeInternal",
new CtClass[] { CtClass.intType, ctMessageBody, CtClass.booleanType, CtClass.booleanType, ctColumnDefinition, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
//catchRuntimeException(clazz, clazz.getDeclaredMethod("isRewritableWithMultiValueClause", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("realClose", new CtClass[] { CtClass.booleanType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("serverExecute", new CtClass[] { CtClass.intType, CtClass.booleanType, ctColumnDefinition }),
EXCEPTION_INTERCEPTOR_GETTER);
//catchRuntimeException(clazz, clazz.getDeclaredMethod("serverLongData", new CtClass[] { CtClass.intType, ctServerPreparedQueryBindValue }),
// EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("serverPrepare", new CtClass[] { ctString }), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* java.sql.ResultSet extends java.sql.Wrapper
*/
// com.mysql.cj.jdbc.ResultSetImpl implements com.mysql.cj.jdbc.ResultSetInternalMethods (extends java.sql.ResultSet)
clazz = pool.get(ResultSetImpl.class.getName());
instrumentJdbcMethods(clazz, ResultSetInternalMethods.class, false, EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.UpdatableResultSet extends ResultSetImpl
clazz = pool.get(UpdatableResultSet.class.getName());
instrumentJdbcMethods(clazz, ResultSetInternalMethods.class, false, EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("generateStatements", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* java.sql.ResultSetMetaData extends java.sql.Wrapper
*/
// com.mysql.cj.jdbc.ResultSetMetaData implements java.sql.ResultSetMetaData
clazz = pool.get(ResultSetMetaData.class.getName());
instrumentJdbcMethods(clazz, java.sql.ResultSetMetaData.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* java.sql.Savepoint
*/
// com.mysql.cj.jdbc.MysqlSavepoint implements java.sql.Savepoint
clazz = pool.get(MysqlSavepoint.class.getName());
instrumentJdbcMethods(clazz, Savepoint.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* java.sql.Statement extends java.sql.Wrapper
*/
// com.mysql.cj.jdbc.StatementImpl implements com.mysql.cj.jdbc.Statement (extends java.sql.Statement)
clazz = pool.get(StatementImpl.class.getName());
instrumentJdbcMethods(clazz, JdbcStatement.class, false, EXCEPTION_INTERCEPTOR_GETTER);
// non-JDBC
catchRuntimeException(clazz, clazz.getDeclaredMethod("createResultSetUsingServerFetch", new CtClass[] { ctString }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("doPingInstead", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeInternal", new CtClass[] { ctString, CtClass.booleanType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz,
clazz.getDeclaredMethod("executeBatchUsingMultiQueries", new CtClass[] { CtClass.booleanType, CtClass.intType, CtClass.intType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeUpdateInternal", new CtClass[] { ctString, CtClass.booleanType, CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("executeSimpleNonQuery", new CtClass[] { ctJdbcConnection, ctString }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("generatePingResultSet", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getBatchedGeneratedKeys", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getBatchedGeneratedKeys", new CtClass[] { ctStatement }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getGeneratedKeysInternal", new CtClass[] { CtClass.longType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getLastInsertID", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getLongUpdateCount", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getOpenResultSetCount", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getResultSetInternal", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("processMultiCountsAndKeys", new CtClass[] { ctStatementImpl, CtClass.intType, ctLongArray }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("removeOpenResultSet", new CtClass[] { ctResultSetInternalMethods }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("resetCancelledState", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setHoldResultsOpenOverClose", new CtClass[] { CtClass.booleanType }),
EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setResultSetConcurrency", new CtClass[] { CtClass.intType }), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("useServerFetch", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
catchRuntimeException(clazz, clazz.getDeclaredMethod("checkCancelTimeout", new CtClass[] {}), EXCEPTION_INTERCEPTOR_GETTER);
clazz.writeFile(args[0]);
/*
* java.sql.SQLXML
*/
// com.mysql.cj.jdbc.MysqlSQLXML implements SQLXML
clazz = pool.get(MysqlSQLXML.class.getName());
instrumentJdbcMethods(clazz, java.sql.SQLXML.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* javax.sql.ConnectionPoolDataSource
*/
// MysqlConnectionPoolDataSource extends MysqlDataSource implements ConnectionPoolDataSource
clazz = pool.get(MysqlConnectionPoolDataSource.class.getName());
instrumentJdbcMethods(clazz, javax.sql.ConnectionPoolDataSource.class);
clazz.writeFile(args[0]);
/*
* javax.sql.DataSource
*/
// MysqlDataSource extends JdbcPropertySetImpl implements DataSource, Referenceable, Serializable, JdbcPropertySet
clazz = pool.get(MysqlDataSource.class.getName());
instrumentJdbcMethods(clazz, javax.sql.DataSource.class);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getStringRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setStringRuntimeProperty", new CtClass[] { ctString, ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getBooleanRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setBooleanRuntimeProperty", new CtClass[] { ctString, CtClass.booleanType }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getIntegerRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setIntegerRuntimeProperty", new CtClass[] { ctString, CtClass.intType }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getLongRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setLongRuntimeProperty", new CtClass[] { ctString, CtClass.longType }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getMemorySizeRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setMemorySizeRuntimeProperty", new CtClass[] { ctString, CtClass.intType }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("getEnumRuntimeProperty", new CtClass[] { ctString }), null);
catchRuntimeException(clazz, clazz.getDeclaredMethod("setEnumRuntimeProperty", new CtClass[] { ctString, ctString }), null);
clazz.writeFile(args[0]);
/*
* javax.sql.PooledConnection
*/
// com.mysql.cj.jdbc.MysqlPooledConnection
clazz = pool.get(MysqlPooledConnection.class.getName());
instrumentJdbcMethods(clazz, javax.sql.PooledConnection.class, false, EXCEPTION_INTERCEPTOR_MEMBER);
clazz.writeFile(args[0]);
/*
* javax.sql.XAConnection
* javax.transaction.xa.XAResource
*/
// com.mysql.cj.jdbc.MysqlXAConnection extends MysqlPooledConnection implements XAConnection, XAResource
clazz = pool.get(MysqlXAConnection.class.getName());
instrumentJdbcMethods(clazz, javax.sql.XAConnection.class);
clazz.writeFile(args[0]);
// com.mysql.cj.jdbc.SuspendableXAConnection extends MysqlPooledConnection implements XAConnection, XAResource
clazz = pool.get(SuspendableXAConnection.class.getName());
instrumentJdbcMethods(clazz, javax.sql.XAConnection.class);
clazz.writeFile(args[0]);
/*
* javax.sql.XADataSource
*/
// com.mysql.cj.jdbc.MysqlXADataSource extends MysqlDataSource implements javax.sql.XADataSource
clazz = pool.get(MysqlXADataSource.class.getName());
instrumentJdbcMethods(clazz, javax.sql.DataSource.class);
instrumentJdbcMethods(clazz, javax.sql.XADataSource.class);
clazz.writeFile(args[0]);
/*
* javax.transaction.xa.Xid
*/
// com.mysql.cj.jdbc.MysqlXid implements Xid
clazz = pool.get(MysqlXid.class.getName());
instrumentJdbcMethods(clazz, javax.transaction.xa.Xid.class);
clazz.writeFile(args[0]);
/*
* TODO:
* java.sql.DataTruncation
*/
// com.mysql.cj.jdbc.exceptions.MysqlDataTruncation extends DataTruncation
/*
* TODO:
* java.sql.SQLException
*/
// com.mysql.cj.jdbc.exceptions.NotUpdatable extends SQLException
// com.mysql.cj.jdbc.exceptions.OperationNotSupportedException extends SQLException
// com.mysql.cj.jdbc.exceptions.PacketTooBigException extends SQLException
/*
* TODO:
* java.sql.SQLNonTransientException
*/
// com.mysql.cj.jdbc.exceptions.MySQLQueryInterruptedException extends SQLNonTransientException
// com.mysql.cj.jdbc.exceptions.MySQLStatementCancelledException extends SQLNonTransientException
/*
* TODO:
* java.sql.SQLRecoverableException
*/
// com.mysql.cj.jdbc.exceptions.CommunicationsException extends SQLRecoverableException implements StreamingNotifiable
// ---> com.mysql.cj.jdbc.exceptions.ConnectionFeatureNotAvailableException extends CommunicationsException
/*
* TODO:
* java.sql.SQLTransientException
* ---> java.sql.SQLTimeoutException
*/
// com.mysql.cj.jdbc.exceptions.MySQLTimeoutException extends SQLTimeoutException
/*
* TODO:
* java.sql.SQLTransientException
* ---> java.sql.SQLTransactionRollbackException
*/
// com.mysql.cj.jdbc.exceptions.MySQLTransactionRollbackException extends SQLTransactionRollbackException implements DeadlockTimeoutRollbackMarker
/*
* TODO:
* com.mysql.cj.jdbc.MysqlXAException extends javax.transaction.xa.XAException
*/
/*
* These classes have no implementations in c/J:
*
* java.sql.Array
* java.sql.BatchUpdateException
* java.sql.ClientInfoStatus
* java.sql.Date
* java.sql.DriverManager
* java.sql.DriverPropertyInfo
* java.sql.PseudoColumnUsage
* java.sql.Ref
* java.sql.RowId
* java.sql.RowIdLifetime
* java.sql.SQLClientInfoException
* java.sql.SQLData
* java.sql.SQLDataException
* java.sql.SQLFeatureNotSupportedException
* java.sql.SQLInput
* java.sql.SQLIntegrityConstraintViolationException
* java.sql.SQLInvalidAuthorizationSpecException
* java.sql.SQLNonTransientConnectionException
* java.sql.SQLOutput
* java.sql.SQLPermission
* java.sql.SQLSyntaxErrorException
* java.sql.SQLTransientConnectionException
* java.sql.SQLWarning
* java.sql.Struct
* java.sql.Time
* java.sql.Timestamp
* java.sql.Types
*
* javax.sql.CommonDataSource
* javax.sql.ConnectionEvent
* javax.sql.ConnectionEventListener
* javax.sql.RowSet
* javax.sql.RowSetEvent
* javax.sql.RowSetInternal
* javax.sql.RowSetListener
* javax.sql.RowSetMetaData
* javax.sql.RowSetReader
* javax.sql.RowSetWriter
* javax.sql.StatementEvent
* javax.sql.StatementEventListener
*
* javax.sql.rowset.BaseRowSet
* javax.sql.rowset.CachedRowSet
* javax.sql.rowset.FilteredRowSet
* javax.sql.rowset.JdbcRowSet
* javax.sql.rowset.Joinable
* javax.sql.rowset.JoinRowSet
* javax.sql.rowset.Predicate
* javax.sql.rowset.RowSetFactory
* javax.sql.rowset.RowSetMetaDataImpl
* javax.sql.rowset.RowSetProvider
* javax.sql.rowset.RowSetWarning
* javax.sql.rowset.WebRowSet
*
* javax.sql.rowset.serial.SerialArray
* javax.sql.rowset.serial.SerialBlob
* javax.sql.rowset.serial.SerialClob
* javax.sql.rowset.serial.SerialDatalink
* javax.sql.rowset.serial.SerialException
* javax.sql.rowset.serial.SerialJavaObject
* javax.sql.rowset.serial.SerialRef
* javax.sql.rowset.serial.SerialStruct
* javax.sql.rowset.serial.SQLInputImpl
* javax.sql.rowset.serial.SQLOutputImpl
*
* javax.sql.rowset.spi.SyncFactory
* javax.sql.rowset.spi.SyncFactoryException
* javax.sql.rowset.spi.SyncProvider
* javax.sql.rowset.spi.SyncProviderException
* javax.sql.rowset.spi.SyncResolver
* javax.sql.rowset.spi.TransactionalWriter
* javax.sql.rowset.spi.XmlReader
* javax.sql.rowset.spi.XmlWriter
*/
}
private static void instrumentJdbcMethods(CtClass cjClazz, Class jdbcClass) throws Exception {
instrumentJdbcMethods(cjClazz, jdbcClass, false, null);
}
/**
* Instruments methods of cjClazz defined in jdbcClass.
*
* @param cjClazz
* CtClass to be instrumented.
* @param jdbcClass
* Class from JDBC specification where methods descriptors to be get.
* @param declaredMethodsOnly
* true - instrument methods declared in this class, false - also instrument inherited methods
* @param exceptionInterceptorStr
* exception interceptor reference as a string
* @throws Exception
* if an error occurs
*/
private static void instrumentJdbcMethods(CtClass cjClazz, Class jdbcClass, boolean declaredMethodsOnly, String exceptionInterceptorStr)
throws Exception {
sysOutPrintln("---");
sysOutPrintln(cjClazz.getName());
Method[] methods;
if (declaredMethodsOnly) {
// instrument methods declared in this class which throws SQLException
methods = jdbcClass.getDeclaredMethods();
} else {
// instrument all methods, declared in this class and it's superclasses, which throws SQLException
methods = jdbcClass.getMethods();
}
for (Method method : methods) {
CtMethod ctm = null;
String prefix = "SKIPPED: ";
for (Class exc : method.getExceptionTypes()) {
if (exc.equals(SQLException.class)) {
prefix = "INSTRUMENTING... ";
String jdbcClassName = method.getName();
List params = new LinkedList<>();
for (Class param : method.getParameterTypes()) {
params.add(pool.get(param.getName()));
}
try {
ctm = cjClazz.getDeclaredMethod(jdbcClassName, params.toArray(new CtClass[0]));
} catch (NotFoundException ex) {
// Just ignoring because the only reason is that the method is implemented in superclass
prefix = "NOT FOUND: ";
}
break;
}
}
sysOutPrint(prefix);
sysOutPrint(method.toGenericString());
if (ctm != null) {
if (catchRuntimeException(cjClazz, ctm, exceptionInterceptorStr, false)) {
sysOutPrint(" ... DONE.");
} else {
sysOutPrint(" ... ALREADY PROCESSED!!!");
}
}
sysOutPrintln("");
}
}
private static void catchRuntimeException(CtClass clazz, CtMethod m, String exceptionInterceptorStr) throws Exception {
catchRuntimeException(clazz, m, exceptionInterceptorStr, true);
}
private static boolean catchRuntimeException(CtClass clazz, CtMethod m, String exceptionInterceptorStr, boolean log) throws Exception {
if (isProcessed(clazz.getClassFile().getName(), m)) {
if (log) {
sysOutPrintln("ALREADY PROCESSED!!! " + m);
}
return false;
}
if (log) {
sysOutPrintln(m + ", " + exceptionInterceptorStr);
}
if (exceptionInterceptorStr == null) {
m.addCatch("{throw " + SQLExceptionsMapping.class.getName() + ".translateException(ex);}", runTimeException, "ex");
} else {
m.addCatch("{throw " + SQLExceptionsMapping.class.getName() + ".translateException(ex, " + exceptionInterceptorStr + ");}", runTimeException, "ex");
}
processed.get(clazz.getClassFile().getName()).add(m);
return true;
}
private static boolean isProcessed(String fileName, CtMethod m) throws Exception {
List methods = processed.get(fileName);
if (methods != null) {
if (methods.contains(m)) {
return true;
}
} else {
processed.put(fileName, new LinkedList());
}
return false;
}
private static void sysOutPrint(String s) {
if (verbose) {
System.out.print(s);
}
}
private static void sysOutPrintln(String s) {
if (verbose) {
System.out.println(s);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy