wangtao
/
file-online-preview
mirror of https://gitee.com/kekingcn/file-online-preview.git
1
0
Fork 0
Code Issues Releases Wiki Activity

!145 优化跨域方法代码 

Merge pull request !145 from 高雄/kuayu
This commit is contained in:
陈精华 2023-06-19 02:32:27 +00:00 committed by Gitee
parent c0d9a8ebad 127f759f16
commit c145f4ca0e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
server/src/main/java/cn/keking/web/controller/OnlinePreviewController.java
View File

@ -109,9 +109,10 @@ public class OnlinePreviewController {
logger.error(String.format(BASE64_DECODE_ERROR_MSG, urlPath),ex); logger.error(String.format(BASE64_DECODE_ERROR_MSG, urlPath),ex);
return; return;
} }
HttpURLConnection urlcon; HttpURLConnection urlcon = null;
InputStream inputStream = null; InputStream inputStream = null;
if (urlPath.toLowerCase().startsWith("file:") || urlPath.toLowerCase().startsWith("file%3")) { assert urlPath != null;
if (!urlPath.toLowerCase().startsWith("http") && !urlPath.toLowerCase().startsWith("https") && !urlPath.toLowerCase().startsWith("ftp")) {
logger.info("读取跨域文件异常，可能存在非法访问，urlPath：{}", urlPath); logger.info("读取跨域文件异常，可能存在非法访问，urlPath：{}", urlPath);
return; return;
} }
@ -123,26 +124,25 @@ public class OnlinePreviewController {
urlcon.setConnectTimeout(30000); urlcon.setConnectTimeout(30000);
urlcon.setReadTimeout(30000); urlcon.setReadTimeout(30000);
urlcon.setInstanceFollowRedirects(false); urlcon.setInstanceFollowRedirects(false);
if (urlcon.getResponseCode() == 302 || urlcon.getResponseCode() == 301) { int responseCode = urlcon.getResponseCode();
urlcon.disconnect(); if (responseCode == HttpURLConnection.HTTP_MOVED_PERM || responseCode == HttpURLConnection.HTTP_MOVED_TEMP) { //301 302
url =new URL(urlcon.getHeaderField("Location")); url =new URL(urlcon.getHeaderField("Location"));
urlcon=(HttpURLConnection)url.openConnection(); urlcon=(HttpURLConnection)url.openConnection();
} }
if (urlcon.getResponseCode() == 404 || urlcon.getResponseCode() == 403 || urlcon.getResponseCode() == 500 ) { if (responseCode == HttpURLConnection.HTTP_NOT_FOUND ||responseCode == HttpURLConnection.HTTP_FORBIDDEN || responseCode == HttpURLConnection.HTTP_INTERNAL_ERROR ) { //403 404 500
logger.error("读取跨域文件异常，url：{}", urlPath); logger.error("读取跨域文件异常，url：{}，错误：{}", urlPath,responseCode);
return ;
} else { } else {
if(urlPath.contains( ".svg")) { if(urlPath.contains( ".svg")) {
response.setContentType("image/svg+xml"); response.setContentType("image/svg+xml");
} }
inputStream=(url).openStream(); inputStream=(url).openStream();
IOUtils.copy(inputStream, response.getOutputStream()); IOUtils.copy(inputStream, response.getOutputStream());
urlcon.disconnect();
} }
} catch (IOException | GalimatiasParseException e) { } catch (IOException | GalimatiasParseException e) {
logger.error("读取跨域文件异常，url：{}", urlPath); logger.error("读取跨域文件异常，url：{}", urlPath);
return ;
} finally { } finally {
assert urlcon != null;
urlcon.disconnect();
IOUtils.closeQuietly(inputStream); IOUtils.closeQuietly(inputStream);
} }
} else { } else {
@ -155,7 +155,6 @@ public class OnlinePreviewController {
IOUtils.copy(inputStream, response.getOutputStream()); IOUtils.copy(inputStream, response.getOutputStream());
} catch (IOException | GalimatiasParseException e) { } catch (IOException | GalimatiasParseException e) {
logger.error("读取跨域文件异常，url：{}", urlPath); logger.error("读取跨域文件异常，url：{}", urlPath);
return ;
} finally { } finally {
IOUtils.closeQuietly(inputStream); IOUtils.closeQuietly(inputStream);
} }