pioneer
2 years ago
commit
d20bd24a52
19 changed files with 1305 additions and 0 deletions
@ -0,0 +1,6 @@ |
|||||||
|
# open-JSD-10202 |
||||||
|
|
||||||
|
JSD-10202 客户的文件都是加密的，所以数据填报时上传Excel动作需要先跑一下他们的解密应用，再读取到填报页面；同样在使用报表的导出功能时，需要先跑一下他们的加密应用，保障导出的文件是经过加密的\ |
||||||
|
免责说明：该源码为第三方爱好者提供，不保证源码和方案的可靠性，也不提供任何形式的源码教学指导和协助！\ |
||||||
|
仅作为开发者学习参考使用！禁止用于任何商业用途！\ |
||||||
|
为保护开发者隐私，开发者信息已隐去！若原开发者希望公开自己的信息，可联系【pioneer】处理。 |
||||||
Binary file not shown.
@ -0,0 +1,22 @@ |
|||||||
|
<?xml version="1.0" encoding="UTF-8" standalone="no"?> |
||||||
|
<plugin> |
||||||
|
<id>com.fr.plugin.xx.zyjn.decode</id> |
||||||
|
<name><![CDATA[中冶焦耐加密解密_EK]]></name> |
||||||
|
<active>yes</active> |
||||||
|
<version>1.0</version> |
||||||
|
<env-version>10.0</env-version> |
||||||
|
<jartime>2018-07-31</jartime> |
||||||
|
<vendor>fr.open</vendor> |
||||||
|
<description><![CDATA[中冶焦耐加密解密_jsd-10202]]></description> |
||||||
|
<change-notes><![CDATA[ |
||||||
|
[2022-06-28]【1.0】初始化插件。<br/> |
||||||
|
]]></change-notes> |
||||||
|
<extra-decision> |
||||||
|
<GlobalRequestFilterProvider class="com.fr.plugin.xx.zyjn.decode.GlobalFilter"/> |
||||||
|
</extra-decision> |
||||||
|
<!--<extra-report> |
||||||
|
<ExportExtensionProcessor class="com.fr.plugin.xx.zyjn.decode.encode.EncodeExportExtensionProvider"/> |
||||||
|
</extra-report>--> |
||||||
|
<function-recorder class="com.fr.plugin.xx.zyjn.decode.decode.DecodeFilter"/> |
||||||
|
<lifecycle-monitor class="com.fr.plugin.xx.zyjn.decode.LRGT"/> |
||||||
|
</plugin> |
||||||
@ -0,0 +1,42 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode; |
||||||
|
|
||||||
|
import com.fr.config.*; |
||||||
|
import com.fr.config.holder.Conf; |
||||||
|
import com.fr.config.holder.factory.Holders; |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @since 2021/12/04 |
||||||
|
*/ |
||||||
|
@Visualization(category = "加解密配置") |
||||||
|
public class DecodeConfig extends DefaultConfiguration { |
||||||
|
|
||||||
|
private static volatile DecodeConfig config = null; |
||||||
|
|
||||||
|
public static DecodeConfig getInstance() { |
||||||
|
if (config == null) { |
||||||
|
config = ConfigContext.getConfigInstance(DecodeConfig.class); |
||||||
|
} |
||||||
|
return config; |
||||||
|
} |
||||||
|
|
||||||
|
@Identifier(value = "debugSwitch", name = "插件调试开关", description = "日志调试模式", status = Status.SHOW) |
||||||
|
private Conf<Boolean> debugSwitch = Holders.simple(true); |
||||||
|
|
||||||
|
public Boolean getDebugSwitch() { |
||||||
|
return this.debugSwitch.get(); |
||||||
|
} |
||||||
|
|
||||||
|
public void setDebugSwitch(Boolean debugSwitch) { |
||||||
|
this.debugSwitch.set(debugSwitch); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
@Override |
||||||
|
public Object clone() throws CloneNotSupportedException { |
||||||
|
DecodeConfig cloned = (DecodeConfig) super.clone(); |
||||||
|
cloned.debugSwitch = (Conf<Boolean>) debugSwitch.clone(); |
||||||
|
return cloned; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,97 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode; |
||||||
|
|
||||||
|
import com.fr.decision.fun.impl.AbstractGlobalRequestFilterProvider; |
||||||
|
import com.fr.intelli.record.Focus; |
||||||
|
import com.fr.intelli.record.Original; |
||||||
|
import com.fr.log.FineLoggerFactory; |
||||||
|
import com.fr.plugin.context.PluginContexts; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.decode.DecodeFilter; |
||||||
|
import com.fr.plugin.transform.FunctionRecorder; |
||||||
|
import com.fr.record.analyzer.EnableMetrics; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
import com.fr.stable.fun.Authorize; |
||||||
|
import com.fr.web.utils.WebUtils; |
||||||
|
|
||||||
|
import javax.servlet.FilterChain; |
||||||
|
import javax.servlet.ServletException; |
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2020/11/20 |
||||||
|
*/ |
||||||
|
@FunctionRecorder |
||||||
|
@Authorize(callSignKey = PluginConstants.PLUGIN_ID) |
||||||
|
@EnableMetrics |
||||||
|
public class GlobalFilter extends AbstractGlobalRequestFilterProvider { |
||||||
|
@Override |
||||||
|
public String filterName() { |
||||||
|
return "zdgl"; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
@Focus(id = PluginConstants.PLUGIN_ID, text = PluginConstants.PLUGIN_NAME, source = Original.PLUGIN) |
||||||
|
public String[] urlPatterns() { |
||||||
|
if (PluginContexts.currentContext().isAvailable()) { |
||||||
|
return new String[]{ |
||||||
|
"/*" |
||||||
|
}; |
||||||
|
} else { |
||||||
|
return new String[]{"/neverbeused"}; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void doFilter(HttpServletRequest req, HttpServletResponse res, FilterChain filterChain) { |
||||||
|
|
||||||
|
if (isDecode(req)) { |
||||||
|
new DecodeFilter().doFilter(req, res, filterChain); |
||||||
|
return; |
||||||
|
} |
||||||
|
/*if (isEncode(req)) { |
||||||
|
new EncodeFilter().doFilter(req, res, filterChain); |
||||||
|
return; |
||||||
|
}*/ |
||||||
|
|
||||||
|
try { |
||||||
|
filterChain.doFilter(req, res); |
||||||
|
} catch (IOException e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage()); |
||||||
|
} catch (ServletException e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private boolean isDecode(HttpServletRequest req) { |
||||||
|
if (req.getRequestURI().endsWith("attach/upload")) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
if (req.getRequestURI().endsWith("/dataset/upload")) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
String op = WebUtils.getHTTPRequestParameter(req, "op"); |
||||||
|
String cmd = WebUtils.getHTTPRequestParameter(req, "cmd"); |
||||||
|
if (StringUtils.equals("fr_attach", op) && StringUtils.equals("ah_upload", cmd)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
if (StringUtils.equals("fr_write", op) && StringUtils.equals("imp_w_excel_data", cmd)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
private boolean isEncode(HttpServletRequest req) { |
||||||
|
String op = WebUtils.getHTTPRequestParameter(req, "op"); |
||||||
|
String cmd = WebUtils.getHTTPRequestParameter(req, "cmd"); |
||||||
|
if (StringUtils.equals("fr_attach", op) && StringUtils.equals("ah_download", cmd)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
if (StringUtils.equals("customcommit", op) && StringUtils.equals("downloadfile", cmd)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,27 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode; |
||||||
|
|
||||||
|
import com.fr.plugin.context.PluginContext; |
||||||
|
import com.fr.plugin.observer.inner.AbstractPluginLifecycleMonitor; |
||||||
|
|
||||||
|
/** |
||||||
|
* 配置信息初始化 |
||||||
|
*/ |
||||||
|
|
||||||
|
public class LRGT extends AbstractPluginLifecycleMonitor { |
||||||
|
@Override |
||||||
|
public void afterRun(PluginContext pluginContext) { |
||||||
|
DecodeConfig.getInstance(); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void beforeStop(PluginContext pluginContext) { |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void beforeUninstall(PluginContext pluginContext) { |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void afterInstall(PluginContext var1) { |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,120 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode; |
||||||
|
|
||||||
|
import com.fr.log.FineLoggerFactory; |
||||||
|
import com.fr.log.FineLoggerProvider; |
||||||
|
import com.fr.plugin.context.PluginContexts; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @since 2021/12/04 |
||||||
|
*/ |
||||||
|
public final class LogUtils { |
||||||
|
private static final String DEBUG_PREFIX = "[插件调试] "; |
||||||
|
private static String LOG_PREFIX = PluginConstants.PLUGIN_NAME; |
||||||
|
private static final String PLUGIN_VERSION; |
||||||
|
|
||||||
|
private static final FineLoggerProvider LOGGER = FineLoggerFactory.getLogger(); |
||||||
|
|
||||||
|
static { |
||||||
|
String version = PluginContexts.currentContext().getMarker().getVersion(); |
||||||
|
if (StringUtils.isNotBlank(version)) { |
||||||
|
PLUGIN_VERSION = "[v" + version + "] "; |
||||||
|
} else { |
||||||
|
PLUGIN_VERSION = "[unknown version] "; |
||||||
|
} |
||||||
|
|
||||||
|
LOG_PREFIX = LOG_PREFIX + PLUGIN_VERSION; |
||||||
|
} |
||||||
|
|
||||||
|
public static void setPrefix(String prefix) { |
||||||
|
if (prefix != null) { |
||||||
|
LOG_PREFIX = prefix; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public static boolean isDebugEnabled() { |
||||||
|
return LOGGER.isDebugEnabled(); |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug(String s) { |
||||||
|
LOGGER.debug(LOG_PREFIX + s); |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug(String s, Object... objects) { |
||||||
|
LOGGER.debug(LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug(String s, Throwable throwable) { |
||||||
|
LOGGER.debug(LOG_PREFIX + s, throwable); |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug4plugin(String s) { |
||||||
|
if (DecodeConfig.getInstance().getDebugSwitch()) { |
||||||
|
LOGGER.error(DEBUG_PREFIX + LOG_PREFIX + s); |
||||||
|
} else { |
||||||
|
LOGGER.debug(LOG_PREFIX + s); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug4plugin(String s, Object... objects) { |
||||||
|
if (DecodeConfig.getInstance().getDebugSwitch()) { |
||||||
|
LOGGER.error(DEBUG_PREFIX + LOG_PREFIX + s, objects); |
||||||
|
} else { |
||||||
|
LOGGER.debug(LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public static void debug4plugin(String s, Throwable throwable) { |
||||||
|
if (DecodeConfig.getInstance().getDebugSwitch()) { |
||||||
|
LOGGER.error(DEBUG_PREFIX + LOG_PREFIX + s, throwable); |
||||||
|
} else { |
||||||
|
LOGGER.debug(LOG_PREFIX + s, throwable); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
public static boolean isInfoEnabled() { |
||||||
|
return LOGGER.isInfoEnabled(); |
||||||
|
} |
||||||
|
|
||||||
|
public static void info(String s) { |
||||||
|
LOGGER.info(LOG_PREFIX + s); |
||||||
|
} |
||||||
|
|
||||||
|
public static void info(String s, Object... objects) { |
||||||
|
LOGGER.info(LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
|
||||||
|
public static void warn(String s) { |
||||||
|
LOGGER.warn(LOG_PREFIX + s); |
||||||
|
} |
||||||
|
|
||||||
|
public static void warn(String s, Object... objects) { |
||||||
|
LOGGER.warn(LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
|
||||||
|
public static void warn(String s, Throwable throwable) { |
||||||
|
LOGGER.warn(LOG_PREFIX + s, throwable); |
||||||
|
} |
||||||
|
|
||||||
|
public static void warn(Throwable throwable, String s, Object... objects) { |
||||||
|
LOGGER.warn(throwable, LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
|
||||||
|
public static void error(String s) { |
||||||
|
LOGGER.error(LOG_PREFIX + s); |
||||||
|
} |
||||||
|
|
||||||
|
public static void error(String s, Object... objects) { |
||||||
|
LOGGER.error(LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
|
||||||
|
public static void error(String s, Throwable throwable) { |
||||||
|
LOGGER.error(LOG_PREFIX + s, throwable); |
||||||
|
} |
||||||
|
|
||||||
|
public static void error(Throwable throwable, String s, Object... objects) { |
||||||
|
LOGGER.error(throwable, LOG_PREFIX + s, objects); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,13 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @since 2021/12/04 |
||||||
|
*/ |
||||||
|
public class PluginConstants { |
||||||
|
|
||||||
|
public static final String PLUGIN_ID = "com.fr.plugin.xx.zyjn.decode"; |
||||||
|
|
||||||
|
public static final String PLUGIN_NAME = "加密解密"; |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,64 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.decode; |
||||||
|
|
||||||
|
import com.fr.third.jodd.io.StreamUtil; |
||||||
|
|
||||||
|
import javax.servlet.ReadListener; |
||||||
|
import javax.servlet.ServletInputStream; |
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletRequestWrapper; |
||||||
|
import java.io.BufferedReader; |
||||||
|
import java.io.ByteArrayInputStream; |
||||||
|
import java.io.IOException; |
||||||
|
import java.io.InputStreamReader; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2022/03/01 |
||||||
|
*/ |
||||||
|
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper { |
||||||
|
private BufferedReader br; |
||||||
|
private byte[] body; |
||||||
|
|
||||||
|
public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException { |
||||||
|
super(request); |
||||||
|
//this.br = request.getReader();
|
||||||
|
body = StreamUtil.readBytes(request.getInputStream()); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
@Override |
||||||
|
public BufferedReader getReader() throws IOException { |
||||||
|
return new BufferedReader(new InputStreamReader(getInputStream())); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public ServletInputStream getInputStream() throws IOException { |
||||||
|
final ByteArrayInputStream bais = new ByteArrayInputStream(body); |
||||||
|
return new ServletInputStream() { |
||||||
|
|
||||||
|
@Override |
||||||
|
public boolean isFinished() { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public boolean isReady() { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void setReadListener(ReadListener readListener) { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public int read() throws IOException { |
||||||
|
return bais.read(); |
||||||
|
} |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
public void setBody(byte[] body) { |
||||||
|
this.body = body; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,206 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.decode; |
||||||
|
|
||||||
|
import cn.net.drm.edi.client.DrmAgent; |
||||||
|
import cn.net.drm.edi.client.DrmAgentInf; |
||||||
|
import cn.net.drm.edi.exception.DrmException; |
||||||
|
import com.fr.base.ServerConfig; |
||||||
|
import com.fr.base.core.IgnoreBytesInputStream; |
||||||
|
import com.fr.base.core.ParseResult; |
||||||
|
import com.fr.base.core.PostParseUtils; |
||||||
|
import com.fr.data.NetworkHelper; |
||||||
|
import com.fr.decision.authority.data.User; |
||||||
|
import com.fr.decision.fun.impl.AbstractGlobalRequestFilterProvider; |
||||||
|
import com.fr.decision.mobile.terminal.TerminalHandler; |
||||||
|
import com.fr.decision.webservice.v10.config.ConfigService; |
||||||
|
import com.fr.decision.webservice.v10.login.LoginService; |
||||||
|
import com.fr.decision.webservice.v10.login.TokenResource; |
||||||
|
import com.fr.decision.webservice.v10.user.UserService; |
||||||
|
import com.fr.general.CommonIOUtils; |
||||||
|
import com.fr.io.utils.ResourceIOUtils; |
||||||
|
import com.fr.plugin.context.PluginContexts; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.LogUtils; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.PluginConstants; |
||||||
|
import com.fr.plugin.transform.FunctionRecorder; |
||||||
|
import com.fr.stable.fun.Authorize; |
||||||
|
import com.fr.stable.web.Device; |
||||||
|
import com.fr.third.org.apache.commons.io.IOUtils; |
||||||
|
|
||||||
|
import javax.servlet.FilterChain; |
||||||
|
import javax.servlet.ServletException; |
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
import java.io.*; |
||||||
|
import java.util.UUID; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2022/03/01 |
||||||
|
*/ |
||||||
|
@FunctionRecorder |
||||||
|
@Authorize(callSignKey = PluginConstants.PLUGIN_ID) |
||||||
|
public class DecodeFilter extends AbstractGlobalRequestFilterProvider { |
||||||
|
|
||||||
|
private static final byte[] NEW_LINE_BYTES = new byte[]{13, 10}; |
||||||
|
private static final byte[] BOUNDARY_END = new byte[]{45, 45}; |
||||||
|
|
||||||
|
private static final String TMP_PATH = System.getProperty("java.io.tmpdir"); |
||||||
|
|
||||||
|
@Override |
||||||
|
public String filterName() { |
||||||
|
return "decode"; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String[] urlPatterns() { |
||||||
|
String servletPathName = "decision"; |
||||||
|
try { |
||||||
|
servletPathName = ConfigService.getInstance().getBasicParam().getServletPathName(); |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
if (PluginContexts.currentContext().isAvailable()) { |
||||||
|
return new String[]{ |
||||||
|
"/" + servletPathName + "/v10/attach/upload", |
||||||
|
"/" + servletPathName + "/v10/dataset/upload", |
||||||
|
"/" + servletPathName + "/view/report", |
||||||
|
|
||||||
|
}; |
||||||
|
} else { |
||||||
|
return new String[]{"/aaaa"}; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void doFilter(HttpServletRequest req, HttpServletResponse res, FilterChain filterChain) { |
||||||
|
BodyReaderHttpServletRequestWrapper wrapper = null; |
||||||
|
try { |
||||||
|
DrmAgentInf instance; |
||||||
|
instance = DrmAgent.getInstance(); |
||||||
|
wrapper = new BodyReaderHttpServletRequestWrapper(req); |
||||||
|
User user = getUser(req); |
||||||
|
if (user == null) { |
||||||
|
LogUtils.debug4plugin("current user is null,not decrypted"); |
||||||
|
filter(wrapper, res, filterChain); |
||||||
|
return; |
||||||
|
} |
||||||
|
InputStream inputStream = wrapper.getInputStream(); |
||||||
|
ParseResult parseResult = PostParseUtils.parse(inputStream, req.getCharacterEncoding()); |
||||||
|
InputStream fis = new IgnoreBytesInputStream(inputStream, this.concat(this.concat(NEW_LINE_BYTES, parseResult.getBoundary().getBytes()), BOUNDARY_END)); |
||||||
|
LogUtils.debug4plugin("record file request url is {}", req.getRequestURI() + req.getQueryString()); |
||||||
|
File file = saveTmp(fis, parseResult.getFileName()); |
||||||
|
boolean encrypted = instance.isEncrypted(file); |
||||||
|
LogUtils.debug4plugin("current files encrypted is {}", encrypted); |
||||||
|
if (!encrypted) { |
||||||
|
filter(wrapper, res, filterChain); |
||||||
|
return; |
||||||
|
} |
||||||
|
decode(req, wrapper, parseResult, file, instance); |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
if (wrapper != null) { |
||||||
|
filter(wrapper, res, filterChain); |
||||||
|
return; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void decode(HttpServletRequest req, BodyReaderHttpServletRequestWrapper wrapper, ParseResult parseResult, File file, DrmAgentInf instance) throws DrmException, UnsupportedEncodingException, FileNotFoundException { |
||||||
|
//File file = saveTmp(fis,parseResult.getFileName());
|
||||||
|
long decrypt = instance.decrypt(file); |
||||||
|
LogUtils.debug4plugin("decode file size is {}", decrypt); |
||||||
|
wrapper.setBody(handleBody(req, CommonIOUtils.inputStream2Bytes(new FileInputStream(file)), parseResult, req.getCharacterEncoding() == null ? ServerConfig.getInstance().getServerCharset() : req.getCharacterEncoding())); |
||||||
|
file.delete(); |
||||||
|
LogUtils.debug4plugin("delete temp file"); |
||||||
|
} |
||||||
|
|
||||||
|
private User getUser(HttpServletRequest req) { |
||||||
|
try { |
||||||
|
return UserService.getInstance().getUserByRequestCookie(req); |
||||||
|
} catch (Exception e) { |
||||||
|
|
||||||
|
} |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
private File saveTmp(InputStream input, String filename) { |
||||||
|
if (!ResourceIOUtils.isDirectoryExist(TMP_PATH)) { |
||||||
|
ResourceIOUtils.createDirectory(TMP_PATH); |
||||||
|
} |
||||||
|
String path = TMP_PATH + File.separator + UUID.randomUUID() + filename.substring(filename.indexOf(".")); |
||||||
|
File file = new File(path); |
||||||
|
FileOutputStream outputStream = null; |
||||||
|
if (!file.exists()) { |
||||||
|
try { |
||||||
|
file.createNewFile(); |
||||||
|
outputStream = new FileOutputStream(file.getPath()); |
||||||
|
IOUtils.copyLarge(input, outputStream); |
||||||
|
LogUtils.debug4plugin("save tmp file path is {}", path); |
||||||
|
} catch (IOException e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
try { |
||||||
|
input.close(); |
||||||
|
outputStream.flush(); |
||||||
|
outputStream.close(); |
||||||
|
} catch (IOException e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return new File(path); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
private void filter(HttpServletRequest req, HttpServletResponse res, FilterChain filterChain) { |
||||||
|
try { |
||||||
|
filterChain.doFilter(req, res); |
||||||
|
} catch (IOException e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} catch (ServletException e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private boolean isLogin(HttpServletRequest request) { |
||||||
|
String oldToken = TokenResource.COOKIE.getToken(request); |
||||||
|
return oldToken != null && checkTokenValid(request, (String) oldToken); |
||||||
|
} |
||||||
|
|
||||||
|
private boolean checkTokenValid(HttpServletRequest req, String token) { |
||||||
|
try { |
||||||
|
Device device = NetworkHelper.getDevice(req); |
||||||
|
LoginService.getInstance().loginStatusValid(token, TerminalHandler.getTerminal(req, device)); |
||||||
|
return true; |
||||||
|
} catch (Exception ignore) { |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
private byte[] concat(byte[] a, byte[] b) { |
||||||
|
byte[] c = new byte[a.length + b.length]; |
||||||
|
System.arraycopy(a, 0, c, 0, a.length); |
||||||
|
System.arraycopy(b, 0, c, a.length, b.length); |
||||||
|
return c; |
||||||
|
} |
||||||
|
|
||||||
|
private byte[] handleBody(HttpServletRequest req, byte[] body, ParseResult parseResult, String charsetName) throws UnsupportedEncodingException { |
||||||
|
byte[] newBody; |
||||||
|
byte[] start; |
||||||
|
byte[] end; |
||||||
|
StringBuilder stringBuilder = new StringBuilder(); |
||||||
|
stringBuilder.append(parseResult.getBoundary()).append("\r\n") |
||||||
|
.append("Content-Disposition:").append(parseResult.getDispline()) |
||||||
|
.append("; name=\"").append(parseResult.getFieldName()) |
||||||
|
.append("\"; filename=\"").append(parseResult.getFileName()).append("\"\r\n") |
||||||
|
.append("Content-Type: ").append(req.getContentType()).append("\r\n").append("\r\n"); |
||||||
|
start = stringBuilder.toString().getBytes(charsetName); |
||||||
|
StringBuilder stringBuilder1 = new StringBuilder(); |
||||||
|
stringBuilder1.append("\r\n").append(parseResult.getBoundary()).append("--\r\n"); |
||||||
|
end = stringBuilder1.toString().getBytes(charsetName); |
||||||
|
|
||||||
|
newBody = this.concat(this.concat(start, body), end); |
||||||
|
return newBody; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,28 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode; |
||||||
|
|
||||||
|
import com.fr.io.collection.ExportCollection; |
||||||
|
import com.fr.io.exporter.AppExporter; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.LogUtils; |
||||||
|
import com.fr.web.core.ReportSessionIDInfor; |
||||||
|
import com.fr.web.core.reserve.DefaultExportExtension; |
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2022/3/7 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class EncodeExportExtensionProvider extends DefaultExportExtension { |
||||||
|
|
||||||
|
@Override |
||||||
|
public ExportCollection createCollection(HttpServletRequest req, HttpServletResponse res, |
||||||
|
ReportSessionIDInfor info, String format, String filename, boolean isEmbed) throws Exception { |
||||||
|
ExportCollection collection = super.createCollection(req, res, info, format, filename, isEmbed); |
||||||
|
LogUtils.debug4plugin("get encode exporter"); |
||||||
|
AppExporter exporter = collection.getExporter(); |
||||||
|
collection.setExporter(new EncryptExporter(exporter)); |
||||||
|
return collection; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode; |
||||||
|
|
||||||
|
import com.fr.io.exporter.AppExporter; |
||||||
|
import com.fr.main.workbook.ResultWorkBook; |
||||||
|
import com.fr.page.PageSetCreator; |
||||||
|
import com.fr.page.PageSetProvider; |
||||||
|
import com.fr.web.core.ReportRepositoryDeal; |
||||||
|
|
||||||
|
import java.io.OutputStream; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2022/3/7 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class EncryptExporter implements AppExporter { |
||||||
|
|
||||||
|
private AppExporter original; |
||||||
|
|
||||||
|
public EncryptExporter(AppExporter original) { |
||||||
|
this.original = original; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void export(OutputStream out, ResultWorkBook book) throws Exception { |
||||||
|
original.export(new EncryptOutputStream(out), book); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void export(OutputStream out, PageSetProvider pageSet) throws Exception { |
||||||
|
original.export(new EncryptOutputStream(out), pageSet); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void export(OutputStream out, ResultWorkBook book, PageSetCreator creator, ReportRepositoryDeal repo, int[] sheets) throws Exception { |
||||||
|
original.export(new EncryptOutputStream(out), book, creator, repo, sheets); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void export(OutputStream out, ResultWorkBook book, PageSetProvider pageSet, ReportRepositoryDeal repo, int[] sheets) throws Exception { |
||||||
|
original.export(new EncryptOutputStream(out), book, pageSet, repo, sheets); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void setVersion(Object o) { |
||||||
|
original.setVersion(o); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,190 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode; |
||||||
|
|
||||||
|
import cn.net.drm.edi.client.DrmAgent; |
||||||
|
import cn.net.drm.edi.client.DrmAgentInf; |
||||||
|
import cn.net.drm.io.IOUtils; |
||||||
|
import com.fr.general.CommonIOUtils; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.DecodeConfig; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.LogUtils; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.encode.bi.FileUtil; |
||||||
|
import org.apache.tools.zip.ZipOutputStream; |
||||||
|
|
||||||
|
import javax.servlet.ServletOutputStream; |
||||||
|
import javax.servlet.WriteListener; |
||||||
|
import java.io.*; |
||||||
|
import java.util.UUID; |
||||||
|
import java.util.stream.Collectors; |
||||||
|
import java.util.stream.Stream; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2022/3/3 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class EncryptOutputStream extends ServletOutputStream { |
||||||
|
|
||||||
|
private OutputStream original = null; |
||||||
|
private boolean closed = false; |
||||||
|
private ByteArrayOutputStream crypt = new ByteArrayOutputStream(); |
||||||
|
|
||||||
|
public EncryptOutputStream(OutputStream original) { |
||||||
|
this.original = original; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 实际加密的代码就在这里实现了 |
||||||
|
* |
||||||
|
* @param bytes |
||||||
|
* @param ignore |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
public static byte[] encrypt(byte[] bytes, boolean ignore) { |
||||||
|
DecodeConfig config = DecodeConfig.getInstance(); |
||||||
|
/*if (!config.isConfiged()) { |
||||||
|
LogUtils.warn("config not finish, not encrypted"); |
||||||
|
return bytes; |
||||||
|
}*/ |
||||||
|
ByteArrayInputStream bis = new ByteArrayInputStream(bytes); |
||||||
|
byte[] header = new byte[256]; |
||||||
|
bis.mark(256); |
||||||
|
byte[] encode = bytes; |
||||||
|
OutputStream fos = null; |
||||||
|
try { |
||||||
|
DrmAgentInf instance = null; |
||||||
|
instance = DrmAgent.getInstance(); |
||||||
|
File tempFile = File.createTempFile("tmp" + UUID.randomUUID().toString(), null); |
||||||
|
LogUtils.info("create tmp file path is {}", tempFile.getPath()); |
||||||
|
bis.read(header); |
||||||
|
bis.reset(); |
||||||
|
boolean isEncryptFile = instance.isEncrypted(tempFile); |
||||||
|
if (!isEncryptFile) { |
||||||
|
if (FileUtil.isArchiveFile(bytes) && !ignore) { |
||||||
|
encode = encodeZip(bytes, instance, config); |
||||||
|
} else { |
||||||
|
//fos = instance.encryptBasicFile(tempFile, config.getOwnerId(), config.getSecretLevelId());
|
||||||
|
int len = 0; |
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
while ((len = bis.read(buffer)) > 0) { |
||||||
|
fos.write(buffer, 0, len); |
||||||
|
} |
||||||
|
fos.flush(); |
||||||
|
FileInputStream in = new FileInputStream(tempFile); |
||||||
|
encode = CommonIOUtils.inputStream2Bytes(in); |
||||||
|
tempFile.deleteOnExit(); |
||||||
|
} |
||||||
|
} |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
IOUtils.closeQuietly(bis); |
||||||
|
IOUtils.closeQuietly(fos); |
||||||
|
|
||||||
|
} |
||||||
|
//假设逆序就是加密了
|
||||||
|
return encode; |
||||||
|
} |
||||||
|
|
||||||
|
private static byte[] encodeZip(byte[] bytes, DrmAgentInf instance, DecodeConfig config) { |
||||||
|
File zipFile = null, encodeZipFile = null; |
||||||
|
File unzipFileFolder = FileUtil.createTempFolder("tmp_unzip" + UUID.randomUUID().toString()); |
||||||
|
File zipFileFolder = FileUtil.createTempFolder("tmp_zip" + UUID.randomUUID().toString()); |
||||||
|
try { |
||||||
|
encodeZipFile = FileUtil.createTempFile(); |
||||||
|
zipFile = FileUtil.createTempFile(); |
||||||
|
//先缓存文件
|
||||||
|
com.fr.general.IOUtils.copyBinaryTo(new ByteArrayInputStream(bytes), new FileOutputStream(zipFile)); |
||||||
|
LogUtils.debug4plugin("save zip file path:{} size is {}", zipFile.getAbsolutePath(), zipFile.getTotalSpace()); |
||||||
|
//再解压出来
|
||||||
|
com.fr.general.IOUtils.unzip(zipFile, unzipFileFolder.getAbsolutePath(),"GBK"); |
||||||
|
LogUtils.debug4plugin("unzip file path is {} file list is {} unzip size is {}", unzipFileFolder.getAbsolutePath(), Stream.of(unzipFileFolder.listFiles()).map(e->e.getName()).collect(Collectors.joining(",")), unzipFileFolder.listFiles().length); |
||||||
|
File[] files = unzipFileFolder.listFiles(); |
||||||
|
ZipOutputStream zipOutputStream = new ZipOutputStream(encodeZipFile); |
||||||
|
zipOutputStream.setEncoding("GBK"); |
||||||
|
for (int i = 0; i < files.length; i++) { |
||||||
|
File file = files[i]; |
||||||
|
//加密复制到另外一个文件夹
|
||||||
|
String encode = zipFileFolder + File.separator + file.getName(); |
||||||
|
InputStream bis = new FileInputStream(file); |
||||||
|
//加密
|
||||||
|
OutputStream fos =null;// instance.encryptBasicStream(new FileOutputStream(encode), config.getOwnerId(), config.getSecretLevelId());
|
||||||
|
int len = 0; |
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
while ((len = bis.read(buffer)) > 0) { |
||||||
|
fos.write(buffer, 0, len); |
||||||
|
} |
||||||
|
fos.flush(); |
||||||
|
IOUtils.closeQuietly(bis); |
||||||
|
IOUtils.closeQuietly(fos); |
||||||
|
LogUtils.debug4plugin("encode file {} to {}", file.getAbsolutePath(), encode); |
||||||
|
} |
||||||
|
//zip
|
||||||
|
com.fr.general.IOUtils.zip(zipOutputStream, zipFileFolder); |
||||||
|
zipOutputStream.closeEntry(); |
||||||
|
zipOutputStream.close(); |
||||||
|
LogUtils.debug4plugin("zip file folder {} to {}", zipFileFolder, encodeZipFile.getAbsolutePath()); |
||||||
|
bytes = CommonIOUtils.inputStream2Bytes(new FileInputStream(encodeZipFile)); |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
if (zipFile != null) { |
||||||
|
zipFile.deleteOnExit(); |
||||||
|
} |
||||||
|
if (encodeZipFile != null) { |
||||||
|
encodeZipFile.deleteOnExit(); |
||||||
|
} |
||||||
|
unzipFileFolder.deleteOnExit(); |
||||||
|
zipFileFolder.deleteOnExit(); |
||||||
|
zipFile.deleteOnExit(); |
||||||
|
} |
||||||
|
return bytes; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(int b) throws IOException { |
||||||
|
if (closed) { |
||||||
|
throw new IOException("Cannot write to a closed output stream"); |
||||||
|
} |
||||||
|
crypt.write(b); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void close() throws IOException { |
||||||
|
if (!closed) { |
||||||
|
byte[] bytes = crypt.toByteArray(); |
||||||
|
bytes = encrypt(bytes, false); |
||||||
|
original.write(bytes); |
||||||
|
original.flush(); |
||||||
|
original.close(); |
||||||
|
closed = true; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void flush() throws IOException { |
||||||
|
crypt.flush(); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(byte[] data) throws IOException { |
||||||
|
write(data, 0, data.length); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(byte[] data, int off, int len) throws IOException { |
||||||
|
if (closed) { |
||||||
|
throw new IOException("Cannot write to a closed output stream"); |
||||||
|
} |
||||||
|
crypt.write(data, off, len); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public boolean isReady() { |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void setWriteListener(WriteListener listener) { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,25 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode; |
||||||
|
|
||||||
|
import javax.servlet.ServletOutputStream; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
import javax.servlet.http.HttpServletResponseWrapper; |
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2022/3/3 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class EncryptResponse extends HttpServletResponseWrapper { |
||||||
|
|
||||||
|
public EncryptResponse(HttpServletResponse response) { |
||||||
|
super(response); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public ServletOutputStream getOutputStream()throws IOException { |
||||||
|
ServletOutputStream original = super.getOutputStream(); |
||||||
|
return new EncryptOutputStream(original); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,71 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode.bi; |
||||||
|
|
||||||
|
import com.fr.decision.fun.impl.AbstractGlobalRequestFilterProvider; |
||||||
|
import com.fr.decision.webservice.v10.config.ConfigService; |
||||||
|
import com.fr.log.FineLoggerFactory; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.LogUtils; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.util.GZIPUtils; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
|
||||||
|
import javax.servlet.FilterChain; |
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2022/3/3 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class EncodeFilter extends AbstractGlobalRequestFilterProvider { |
||||||
|
@Override |
||||||
|
public String filterName() { |
||||||
|
return "encode"; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String[] urlPatterns() { |
||||||
|
String servletPathName = "decision"; |
||||||
|
try { |
||||||
|
servletPathName = ConfigService.getInstance().getBasicParam().getServletPathName(); |
||||||
|
} catch (Exception e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} |
||||||
|
return new String[]{ |
||||||
|
"/" + servletPathName + "/v5/design/report/data/export", |
||||||
|
"/" + servletPathName + "/v5/design/report/data/global/export/pdf", |
||||||
|
"/" + servletPathName + "/v5/design/report/data/global/export/excel", |
||||||
|
|
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void doFilter(HttpServletRequest req, HttpServletResponse res, FilterChain filterChain) { |
||||||
|
try { |
||||||
|
String method = req.getMethod(); |
||||||
|
if ("POST".equalsIgnoreCase(method)) { |
||||||
|
String uri = req.getRequestURI(); |
||||||
|
String query = req.getQueryString(); |
||||||
|
LogUtils.debug4plugin("BI export url {}",uri + "?" + query); |
||||||
|
ResponseWrapperImpl wrapper = new ResponseWrapperImpl(res); |
||||||
|
filterChain.doFilter(req, wrapper); |
||||||
|
FileUtil.handle(res.getOutputStream(), os -> { |
||||||
|
byte[] bytes = wrapper.getDataStream(); |
||||||
|
if (StringUtils.contains(wrapper.getHeader("Content-Encoding"), "gzip")) { |
||||||
|
FineLoggerFactory.getLogger().info("current url {} is gzip", getUrl(req)); |
||||||
|
bytes = GZIPUtils.decompress(bytes); |
||||||
|
} |
||||||
|
os.write(bytes); |
||||||
|
os.flush(); |
||||||
|
}, req,res); |
||||||
|
} |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private String getUrl(HttpServletRequest req) { |
||||||
|
String uri = req.getRequestURI(); |
||||||
|
String query = req.getQueryString(); |
||||||
|
return StringUtils.isNotBlank(query) ? uri + "?" + query : uri; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,18 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode.bi; |
||||||
|
|
||||||
|
import java.io.OutputStream; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2020/11/20 |
||||||
|
*/ |
||||||
|
@FunctionalInterface |
||||||
|
public interface ExportFunc { |
||||||
|
/** |
||||||
|
* 写入到临时输出流 |
||||||
|
* |
||||||
|
* @param os 临时输出流 |
||||||
|
* @return void |
||||||
|
**/ |
||||||
|
void export(OutputStream os) throws Exception; |
||||||
|
} |
||||||
@ -0,0 +1,169 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode.bi; |
||||||
|
|
||||||
|
import com.fr.general.CommonIOUtils; |
||||||
|
import com.fr.log.FineLoggerFactory; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.LogUtils; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.encode.EncryptOutputStream; |
||||||
|
import com.fr.plugin.xx.zyjn.decode.util.GZIPUtils; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
import com.fr.third.org.apache.commons.io.IOUtils; |
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
import java.io.*; |
||||||
|
import java.util.Arrays; |
||||||
|
import java.util.UUID; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2020/11/20 |
||||||
|
*/ |
||||||
|
public class FileUtil { |
||||||
|
|
||||||
|
/** |
||||||
|
* 文件加密之后再写入指定的输出流 |
||||||
|
* |
||||||
|
* @param file |
||||||
|
* @param stream |
||||||
|
* @return void |
||||||
|
**/ |
||||||
|
public static void handle(File file, OutputStream stream, HttpServletRequest req, HttpServletResponse res) { |
||||||
|
InputStream is = null; |
||||||
|
try { |
||||||
|
FileInputStream inputStream = new FileInputStream(file); |
||||||
|
byte[] encodeFile = encodeFile(CommonIOUtils.inputStream2Bytes(inputStream), req); |
||||||
|
is = new ByteArrayInputStream(encodeFile); |
||||||
|
byte[] bytes = CommonIOUtils.inputStream2Bytes(is); |
||||||
|
if (StringUtils.contains(res.getHeader("Content-Encoding"), "gzip")) { |
||||||
|
bytes = GZIPUtils.compress(bytes); |
||||||
|
} |
||||||
|
IOUtils.copyLarge(new ByteArrayInputStream(bytes), stream); |
||||||
|
} catch (Exception e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
if (is != null) { |
||||||
|
try { |
||||||
|
is.close(); |
||||||
|
} catch (IOException e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
if (stream != null) { |
||||||
|
try { |
||||||
|
stream.flush(); |
||||||
|
} catch (IOException e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private static byte[] encodeFile(byte[] file, HttpServletRequest req) { |
||||||
|
try { |
||||||
|
return EncryptOutputStream.encrypt(file,true); |
||||||
|
} catch (Exception e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} |
||||||
|
return file; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 加密导出文件后写入原始输出流中 |
||||||
|
* 回调函数式处理 |
||||||
|
* |
||||||
|
* @param outputStream 原始输出流 |
||||||
|
* @param func 临时输出流处理 |
||||||
|
* @return void |
||||||
|
**/ |
||||||
|
public static void handle(OutputStream outputStream, ExportFunc func, HttpServletRequest req, HttpServletResponse res) { |
||||||
|
File tempFile = null; |
||||||
|
OutputStream os = null; |
||||||
|
//写入到临时文件
|
||||||
|
try { |
||||||
|
tempFile = FileUtil.createTempFile(); |
||||||
|
os = new FileOutputStream(tempFile); |
||||||
|
//直接写到临时文件里边
|
||||||
|
func.export(os); |
||||||
|
} catch (Exception e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
if (os != null) { |
||||||
|
try { |
||||||
|
os.close(); |
||||||
|
} catch (IOException e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
//处理临时文件，写到原始输出流
|
||||||
|
try { |
||||||
|
handle(tempFile, outputStream, req,res);//临时加密后写到原始输出流中
|
||||||
|
} catch (Exception e) { |
||||||
|
FineLoggerFactory.getLogger().error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
if (tempFile != null) { |
||||||
|
deleteFile(tempFile);//最后删除临时文件
|
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建临时文件 |
||||||
|
* |
||||||
|
* @return java.io.File |
||||||
|
**/ |
||||||
|
public static File createTempFile() throws IOException { |
||||||
|
return File.createTempFile("tmp" + UUID.randomUUID().toString(), null);//创建一个临时文件
|
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 删除文件 |
||||||
|
* |
||||||
|
* @param file |
||||||
|
* @return boolean |
||||||
|
* @author Zhanying |
||||||
|
**/ |
||||||
|
public static boolean deleteFile(File file) { |
||||||
|
boolean flag = file.exists() && file.isFile() && file.delete(); |
||||||
|
// 如果文件路径所对应的文件存在，并且是一个文件，则直接删除
|
||||||
|
return flag; |
||||||
|
} |
||||||
|
|
||||||
|
private static byte[] ZIP_HEADER_1 = new byte[]{80, 75, 3, 4}; |
||||||
|
|
||||||
|
private static byte[] ZIP_HEADER_2 = new byte[]{80, 75, 5, 6}; |
||||||
|
|
||||||
|
public static boolean isArchiveFile(byte[] file) { |
||||||
|
if (file == null) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
boolean isArchive = false; |
||||||
|
InputStream input = null; |
||||||
|
try { |
||||||
|
input = new ByteArrayInputStream(file); |
||||||
|
byte[] buffer = new byte[4]; |
||||||
|
int length = input.read(buffer, 0, 4); |
||||||
|
if (length == 4) { |
||||||
|
isArchive = (Arrays.equals(ZIP_HEADER_1, buffer)) || (Arrays.equals(ZIP_HEADER_2, buffer)); |
||||||
|
} |
||||||
|
} catch (IOException e) { |
||||||
|
LogUtils.error(e.getMessage(), e); |
||||||
|
} finally { |
||||||
|
if (input != null) { |
||||||
|
try { |
||||||
|
input.close(); |
||||||
|
} catch (IOException e) { |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return isArchive; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
public static File createTempFolder(String path) { |
||||||
|
File file = new File(System.getProperty("java.io.tmpdir")+File.separator+path); |
||||||
|
file.mkdir(); |
||||||
|
return file; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,70 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.encode.bi; |
||||||
|
|
||||||
|
import javax.servlet.ServletOutputStream; |
||||||
|
import javax.servlet.WriteListener; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
import javax.servlet.http.HttpServletResponseWrapper; |
||||||
|
import java.io.ByteArrayOutputStream; |
||||||
|
import java.io.DataOutputStream; |
||||||
|
import java.io.IOException; |
||||||
|
import java.io.OutputStream; |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* @author xx |
||||||
|
* @date 2020/11/20 |
||||||
|
*/ |
||||||
|
public class ResponseWrapperImpl extends HttpServletResponseWrapper { |
||||||
|
ByteArrayOutputStream output; |
||||||
|
FilterServletOutputStream filterOutput; |
||||||
|
|
||||||
|
public ResponseWrapperImpl(HttpServletResponse response) { |
||||||
|
super(response); |
||||||
|
output = new ByteArrayOutputStream(); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public ServletOutputStream getOutputStream() throws IOException { |
||||||
|
if (filterOutput == null) { |
||||||
|
filterOutput = new FilterServletOutputStream(output); |
||||||
|
} |
||||||
|
return filterOutput; |
||||||
|
} |
||||||
|
|
||||||
|
public byte[] getDataStream() { |
||||||
|
return output.toByteArray(); |
||||||
|
} |
||||||
|
|
||||||
|
class FilterServletOutputStream extends ServletOutputStream { |
||||||
|
DataOutputStream output; |
||||||
|
|
||||||
|
public FilterServletOutputStream(OutputStream output) { |
||||||
|
this.output = new DataOutputStream(output); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(int arg0) throws IOException { |
||||||
|
output.write(arg0); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(byte[] arg0, int arg1, int arg2) throws IOException { |
||||||
|
output.write(arg0, arg1, arg2); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void write(byte[] arg0) throws IOException { |
||||||
|
output.write(arg0); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public boolean isReady() { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void setWriteListener(WriteListener writeListener) { |
||||||
|
|
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,89 @@ |
|||||||
|
package com.fr.plugin.xx.zyjn.decode.util; |
||||||
|
|
||||||
|
import java.io.ByteArrayInputStream; |
||||||
|
import java.io.ByteArrayOutputStream; |
||||||
|
import java.io.IOException; |
||||||
|
import java.io.OutputStream; |
||||||
|
import java.util.zip.GZIPInputStream; |
||||||
|
import java.util.zip.GZIPOutputStream; |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author xx |
||||||
|
* @Date 2020/9/15 |
||||||
|
* @Description |
||||||
|
**/ |
||||||
|
public class GZIPUtils { |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 数据压缩传输 |
||||||
|
* |
||||||
|
* @param is |
||||||
|
* @param os |
||||||
|
* @throws Exception |
||||||
|
*/ |
||||||
|
public static void compressTransfe(byte[] bytes, OutputStream out) throws IOException { |
||||||
|
GZIPOutputStream gos = null; |
||||||
|
try { |
||||||
|
gos = new GZIPOutputStream(out); |
||||||
|
gos.write(bytes); |
||||||
|
gos.finish(); |
||||||
|
gos.flush(); |
||||||
|
} finally{ |
||||||
|
if(gos != null){ |
||||||
|
gos.close(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 数据压缩 |
||||||
|
* |
||||||
|
* @param is |
||||||
|
* @param os |
||||||
|
* @throws Exception |
||||||
|
*/ |
||||||
|
public static byte[] compress(byte[] bytes) throws IOException { |
||||||
|
ByteArrayOutputStream out = null; |
||||||
|
GZIPOutputStream gos = null; |
||||||
|
try { |
||||||
|
out = new ByteArrayOutputStream(); |
||||||
|
gos = new GZIPOutputStream(out); |
||||||
|
gos.write(bytes); |
||||||
|
gos.finish(); |
||||||
|
gos.flush(); |
||||||
|
} finally{ |
||||||
|
if(gos != null){ |
||||||
|
gos.close(); |
||||||
|
} |
||||||
|
if(out != null){ |
||||||
|
out.close(); |
||||||
|
} |
||||||
|
} |
||||||
|
return out.toByteArray(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 数据解压 |
||||||
|
* |
||||||
|
* @param in |
||||||
|
* @return |
||||||
|
* @throws IOException |
||||||
|
*/ |
||||||
|
public static byte[] decompress(byte[] bytes) throws IOException { |
||||||
|
ByteArrayInputStream in = new ByteArrayInputStream(bytes); |
||||||
|
GZIPInputStream gin = new GZIPInputStream(in); |
||||||
|
ByteArrayOutputStream out = new ByteArrayOutputStream(); |
||||||
|
int count; |
||||||
|
byte data[] = new byte[1024]; |
||||||
|
while ((count = gin.read(data, 0, 1024)) != -1) { |
||||||
|
out.write(data, 0, count); |
||||||
|
} |
||||||
|
out.flush(); |
||||||
|
out.close(); |
||||||
|
gin.close(); |
||||||
|
return out.toByteArray(); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue