修复升级poi版本后判断office是否受密码保护方法

pull/56/head
陈精华 2 years ago
parent b225cebc95
commit 764ea702d0

@ -1,12 +1,12 @@
package cn.keking.utils; package cn.keking.utils;
import org.apache.commons.lang3.exception.ExceptionUtils; import org.apache.commons.lang3.exception.ExceptionUtils;
import org.apache.poi.EncryptedDocumentException;
import org.apache.poi.extractor.ExtractorFactory; import org.apache.poi.extractor.ExtractorFactory;
import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey; import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey;
import org.springframework.lang.Nullable; import org.springframework.lang.Nullable;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.IOException;
/** /**
* Office * Office
@ -16,6 +16,8 @@ import java.io.FileInputStream;
*/ */
public class OfficeUtils { public class OfficeUtils {
private static final String POI_INVALID_PASSWORD_MSG = "Invalid password specified";
/** /**
* officeword,excel,ppt * officeword,excel,ppt
* *
@ -25,17 +27,18 @@ public class OfficeUtils {
public static boolean isPwdProtected(String path) { public static boolean isPwdProtected(String path) {
try { try {
ExtractorFactory.createExtractor(new FileInputStream(path)); ExtractorFactory.createExtractor(new FileInputStream(path));
} catch (EncryptedDocumentException e) { } catch (IOException e) {
if (POI_INVALID_PASSWORD_MSG.equals(e.getMessage())) {
return true; return true;
}
} catch (Exception e) { } catch (Exception e) {
Throwable[] throwables = ExceptionUtils.getThrowables(e); Throwable[] throwableArray = ExceptionUtils.getThrowables(e);
for (Throwable throwable : throwables) { for (Throwable throwable : throwableArray) {
if (throwable instanceof EncryptedDocumentException) { if (throwable instanceof IOException && POI_INVALID_PASSWORD_MSG.equals(throwable.getMessage())) {
return true; return true;
} }
} }
} }
return false; return false;
} }
@ -55,7 +58,6 @@ public class OfficeUtils {
} finally { } finally {
Biff8EncryptionKey.setCurrentUserPassword(null); Biff8EncryptionKey.setCurrentUserPassword(null);
} }
return true; return true;
} }

Loading…
Cancel
Save