从XXE漏洞修复引起Not supported: http://javax.xml.XMLConstants/property/accessExternalDTD说到SPI机制

引子 

　　在使用Fortify扫描时代码报XML External Entity Injection,此漏洞为xml实体注入漏洞，XXE攻击可利用在处理时动态构建文档的 XML 功能。修复方案也包含了增加安全配置，使它不允许将外部实体包含在传入的 XML 文档中。

具体在修复过程中，代码在解析drools的transfer.xls时，调用代码中增加内容，包括serFeature和setAttribute

TransformerFactory factory = TransformerFactory.newInstance();		
factory.setFeature(XMLConstants.FEATURE_SECURE_PROCESSING, true);
factory.setAttribute(XMLConstants.ACCESS_EXTERNAL_DTD,"");

但在执行第二行，

factory.setAttribute(XMLConstants.ACCESS_EXTERNAL_DTD, ""); 

时，抛出异常

不支持：http://javax.xml.XMLConstants/property/accessExternalDTD  
at org.apache.xalan.processor.TransformerFactoryImpl.setAttribute(TransformerFactoryImpl.java:571)  
at XlsJava.main(XlsJava.java:10)

跟踪与定位

仔细查看代码,定义为javax.xml.transform.TransformerFactory的factory在setAttribute时却进入到了路径为org\apache\xalan\processor\TransformerFactoryImpl.class的类。

出现这个情况，因为项目中依赖了xalan的包，而在xalan包中指定了META-INF\services

因为这个设置，将完全限定名称为javax.xml.transform.TransformerFactory类的方法映射到了路径为

org.apache.xalan.processor.TransformerFactoryImpl

的类上，而在此类中经过一系列判断最终抛出异常

解决方案

解决这个问题的方法也很简单，只需要在调用此段代码的工程之下，覆盖xalan包的设置即可。具体实现为，在调用

factory.setAttribute(XMLConstants.ACCESS_EXTERNAL_DTD, "");

的工程下配置META-INF/services路径，新建名称为

javax.xml.transform.TransformerFactory

的文件，其文件内容为

com.sun.org.apache.xalan.internal.xsltc.trax.TransformerFactoryImpl

如图

SPI机制

　　SPI是Service Provider Interface 的简称，即服务提供者接口的意思。上文所使用的处理方法就是SPI机制的实例。SPI自jdk1.6开始引入，此后便有了广泛的应用，最常见的就是数据库连接，JDK提供了一个java.sql.Driver接口，根据不同的数据库厂商来引入不同的JDBC驱动包，比如MySQL这些数据库驱动其实都会实现这个驱动类。SPI机制要求拓展内容需存放在resources/META-INF/services目录下，META-INF用于存储服务提供者（service provider）的配置文件，serviceloader从META-INF/services中文件查找service的实现，该文件具有与service接口相同的限定名，其内容包括实现的限定名列表。如此，serviceloader将调用META-INF/services中文件的具体实现。所以，

javax.xml.transform.TransformerFactory

类的实现，却在调用方法的时候定位到了

org.apache.xalan.processor.TransformerFactoryImpl

　　为什么必须是META-INF/services之下？看了源码你可能就会明白

public final class ServiceLoader<S> implements Iterable<S>{
    private static final String PREFIX = "META-INF/services/";
}

　　

SPI机制的优缺点

优点

1. 其核心思想就是解耦，让接口和实现分离开来
2. 提高框架的扩展性，可以使框架根据实际业务情况启用扩展或替换框架组件

缺点

1. serviceloader对实现类的加载使用的是懒加载，在使用循环遍历时，即使是不必要加载的类同样会被实例化，造成浪费
2. serviceloader不是线程安全的

更多的关于SPI的相关内容，参考官方文档https://docs.oracle.com/javase/6/docs/api/java/util/ServiceLoader.html