修复漏洞——文件为空时(敏感词模板下载)
This commit is contained in:
@@ -159,21 +159,27 @@ public class SensitiveWordDataController extends BaseController {
|
|||||||
response.setHeader("Pragma", "public");
|
response.setHeader("Pragma", "public");
|
||||||
response.setHeader("Cache-Control", "max-age=0");
|
response.setHeader("Cache-Control", "max-age=0");
|
||||||
InputStream in = null;
|
InputStream in = null;
|
||||||
|
OutputStream ou =null;
|
||||||
try {
|
try {
|
||||||
in = new FileInputStream(url);
|
in = new FileInputStream(url);
|
||||||
} catch (FileNotFoundException e1) {
|
ou = response.getOutputStream();
|
||||||
resMsg = "文件未找到";
|
|
||||||
e1.printStackTrace();
|
|
||||||
response.getWriter().write(resMsg + ":" + name);
|
|
||||||
}
|
|
||||||
OutputStream ou = response.getOutputStream();
|
|
||||||
byte[] buffer = new byte[1024];
|
byte[] buffer = new byte[1024];
|
||||||
int i = -1;
|
int i = -1;
|
||||||
while ((i = in.read(buffer)) != -1) {
|
while ((i = in.read(buffer)) != -1) {
|
||||||
ou.write(buffer, 0, i);
|
ou.write(buffer, 0, i);
|
||||||
}
|
}
|
||||||
ou.flush();
|
ou.flush();
|
||||||
|
} catch (FileNotFoundException e1) {
|
||||||
|
resMsg = "文件未找到";
|
||||||
|
e1.printStackTrace();
|
||||||
|
response.getWriter().write(resMsg + ":" + name);
|
||||||
|
}finally {
|
||||||
|
if (ou != null) {
|
||||||
ou.close();
|
ou.close();
|
||||||
|
}
|
||||||
|
if (in != null) {
|
||||||
in.close();
|
in.close();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user