mysql防注入原理_mybatis-plus sql注入原理（3.0.1）

MP版本为3.0.1

sql注入原理

①，入口类 com.baomidou.mybatisplus.core.injector.AbstractSqlInjector ，重点是这个方法

public void inspectInject(MapperBuilderAssistant builderAssistant, Class> mapperClass) {

String className = mapperClass.toString();

Set mapperRegistryCache = GlobalConfigUtils.getMapperRegistryCache(builderAssistant.getConfiguration());

if (!mapperRegistryCache.contains(className)) {

List methodList = this.getMethodList();

Assert.notEmpty(methodList, "No effective injection method was found.");

// 循环注入自定义方法，这里开始注入sql

methodList.forEach(m -> m.inject(builderAssistant, mapperClass));

mapperRegistryCache.add(className);

/**

* 初始化 SQL 解析

*/

if (GlobalConfigUtils.getGlobalConfig(builderAssistant.getConfiguration()).isSqlParserCache()) {

SqlParserHelper.initSqlParserInfoCache(mapperClass);

}

}

}

②，由AbstractMethod 的injectMappedStatement 方法完成具体的注入，

我们看到实际是有其实现类的一个个injectMappedStatement 来完成注入的

③，这里我以DeleteById 为例

@Override

public MappedStatement injectMappedStatement(Class> mapperClass, Class> modelClass, TableInfo tableInfo) {

//得到待解析sql的模板

SqlMethod sqlMethod = SqlMethod.DELETE_BY_ID;

//利用语言驱动和配置信息，table元数据，和刚才得到的sql方法模板得到sqlSource

SqlSource sqlSource = languageDriver.createSqlSource(configuration, String.format(sqlMethod.getSql(),

tableInfo.getTableName(), tableInfo.getKeyColumn(), tableInfo.getKeyProperty()), modelClass);

//最后添加到mybatis的configuration里的mappedStatements中

return this.addDeleteMappedStatement(mapperClass, sqlMethod.getMethod(), sqlSource);

}

④，DELETE_BY_ID的模板如下

由此我们可以猜出，在tableInfo 中必须要能得出表名和该表主键

执行AbstractSqlInjector #inspectInject 的原理

我们可以在该方法的第一行打一个断点

看我红框圈起来的部门，我们可以大致猜出，是spring容器在实例化组件时，在实例化组件后调用组件的初始化方法，而引起的一连串反应。

通过翻看源码，我大致解释一下

1，AbstractAutowireCapableBeanFactory调用组件的afterPropertiesSet方法，这个组件的beanName为userMapper，这是一个我定义的一个继承自BaseMapper的一个接口

2，不知怎么地，居然跑去调用 DaoSupport的afterPropertiesSet方法(知道的朋友欢迎留言)

3，然后调用到了其子类的MapperFactoryBean 的checkDaoConfig方法

重点是一行，这个configuration是继承自org.apache.ibatis.session.Configuration 的子类，是有mp自定义的一个类

configuration.addMapper(this.mapperInterface);

4，接下来调用com.baomidou.mybatisplus.core.MybatisMapperRegistry的addMapper方法

断点停在了86行

5，我们看这个parse方法

利用GlobalConfigUtils 得到ISqlInjector，然后调用其inspectInject方法

6，我们看看mp是如何得到这个ISqlInjector的

public static ISqlInjector getSqlInjector(Configuration configuration) {

// fix #140

GlobalConfig globalConfiguration = getGlobalConfig(configuration);

//从配置里拿，如果有，则用配置的

ISqlInjector sqlInjector = globalConfiguration.getSqlInjector();

if (sqlInjector == null) {

//没有配置，默认给一个DefaultSqlInjector

sqlInjector = new DefaultSqlInjector();

globalConfiguration.setSqlInjector(sqlInjector);

}

return sqlInjector;

}

7，而 DefaultSqlInjector 是继承自 AbstractSqlInjector 类的

由于DefaultSqlInjector没有重写AbstractSqlInjector的inspectInject方法，

所以DefaultSqlInjector在调用inspectInject方法时，实际上会用父类AbstractSqlInjector的inspectInject方法。