--- /dev/null
+/* ====================================================================
+ Licensed to the Apache Software Foundation (ASF) under one or more
+ contributor license agreements. See the NOTICE file distributed with
+ this work for additional information regarding copyright ownership.
+ The ASF licenses this file to You under the Apache License, Version 2.0
+ (the "License"); you may not use this file except in compliance with
+ the License. You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+==================================================================== */
+
+package org.apache.poi.poifs.filesystem;
+
+import org.apache.poi.EncryptedDocumentException;
+import org.apache.poi.poifs.common.POIFSConstants;
+import org.apache.poi.poifs.crypt.Decryptor;
+import org.apache.poi.poifs.crypt.EncryptionInfo;
+import org.apache.poi.util.IOUtils;
+
+import java.io.FilterInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.PushbackInputStream;
+import java.security.GeneralSecurityException;
+
+/**
+ * A small base class for the various factories, e.g. WorkbookFactory,
+ * SlideShowFactory to combine common code here.
+ */
+public class DocumentFactoryHelper {
+ /**
+ * Wrap the OLE2 data in the NPOIFSFileSystem into a decrypted stream by using
+ * the given password.
+ *
+ * @param fs The OLE2 stream for the document
+ * @param password The password, null if the default password should be used
+ * @return A stream for reading the decrypted data
+ * @throws IOException If an error occurs while decrypting or if the password does not match
+ */
+ public static InputStream getDecryptedStream(final NPOIFSFileSystem fs, String password)
+ throws IOException {
+ EncryptionInfo info = new EncryptionInfo(fs);
+ Decryptor d = Decryptor.getInstance(info);
+
+ try {
+ boolean passwordCorrect = false;
+ if (password != null && d.verifyPassword(password)) {
+ passwordCorrect = true;
+ }
+ if (!passwordCorrect && d.verifyPassword(Decryptor.DEFAULT_PASSWORD)) {
+ passwordCorrect = true;
+ }
+
+ if (passwordCorrect) {
+ // wrap the stream in a FilterInputStream to close the NPOIFSFileSystem
+ // as well when the resulting OPCPackage is closed
+ return new FilterInputStream(d.getDataStream(fs.getRoot())) {
+ @Override
+ public void close() throws IOException {
+ fs.close();
+
+ super.close();
+ }
+ };
+ } else {
+ if (password != null)
+ throw new EncryptedDocumentException("Password incorrect");
+ else
+ throw new EncryptedDocumentException("The supplied spreadsheet is protected, but no password was supplied");
+ }
+ } catch (GeneralSecurityException e) {
+ throw new IOException(e);
+ }
+ }
+
+ /**
+ * Checks that the supplied InputStream (which MUST
+ * support mark and reset, or be a PushbackInputStream)
+ * has a OOXML (zip) header at the start of it.
+ * If your InputStream does not support mark / reset,
+ * then wrap it in a PushBackInputStream, then be
+ * sure to always use that, and not the original!
+ * @param inp An InputStream which supports either mark/reset, or is a PushbackInputStream
+ */
+ public static boolean hasOOXMLHeader(InputStream inp) throws IOException {
+ // We want to peek at the first 4 bytes
+ inp.mark(4);
+
+ byte[] header = new byte[4];
+ int bytesRead = IOUtils.readFully(inp, header);
+
+ // Wind back those 4 bytes
+ if(inp instanceof PushbackInputStream) {
+ PushbackInputStream pin = (PushbackInputStream)inp;
+ pin.unread(header, 0, bytesRead);
+ } else {
+ inp.reset();
+ }
+
+ // Did it match the ooxml zip signature?
+ return (
+ bytesRead == 4 &&
+ header[0] == POIFSConstants.OOXML_FILE_HEADER[0] &&
+ header[1] == POIFSConstants.OOXML_FILE_HEADER[1] &&
+ header[2] == POIFSConstants.OOXML_FILE_HEADER[2] &&
+ header[3] == POIFSConstants.OOXML_FILE_HEADER[3]
+ );
+ }
+
+}
==================================================================== */\r
package org.apache.poi.sl.usermodel;\r
\r
-import java.io.File;\r
-import java.io.FileNotFoundException;\r
-import java.io.IOException;\r
-import java.io.InputStream;\r
-import java.io.PushbackInputStream;\r
+import java.io.*;\r
import java.lang.reflect.InvocationTargetException;\r
import java.lang.reflect.Method;\r
-import java.security.GeneralSecurityException;\r
\r
import org.apache.poi.EncryptedDocumentException;\r
import org.apache.poi.OldFileFormatException;\r
+import org.apache.poi.poifs.filesystem.DocumentFactoryHelper;\r
import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey;\r
import org.apache.poi.poifs.crypt.Decryptor;\r
-import org.apache.poi.poifs.crypt.EncryptionInfo;\r
import org.apache.poi.poifs.filesystem.DirectoryNode;\r
import org.apache.poi.poifs.filesystem.NPOIFSFileSystem;\r
import org.apache.poi.poifs.filesystem.OfficeXmlFileException;\r
import org.apache.poi.util.IOUtils;\r
\r
public class SlideShowFactory {\r
- /** The first 4 bytes of an OOXML file, used in detection */\r
- private static final byte[] OOXML_FILE_HEADER = { 0x50, 0x4b, 0x03, 0x04 };\r
- \r
/**\r
* Creates a SlideShow from the given NPOIFSFileSystem.\r
*\r
*\r
* @throws IOException if an error occurs while reading the data\r
*/\r
- public static SlideShow<?,?> create(NPOIFSFileSystem fs, String password) throws IOException {\r
+ public static SlideShow<?,?> create(final NPOIFSFileSystem fs, String password) throws IOException {\r
DirectoryNode root = fs.getRoot();\r
\r
// Encrypted OOXML files go inside OLE2 containers, is this one?\r
if (root.hasEntry(Decryptor.DEFAULT_POIFS_ENTRY)) {\r
- EncryptionInfo info = new EncryptionInfo(fs);\r
- Decryptor d = Decryptor.getInstance(info);\r
-\r
- boolean passwordCorrect = false;\r
InputStream stream = null;\r
try {\r
- if (password != null && d.verifyPassword(password)) {\r
- passwordCorrect = true;\r
- }\r
- if (!passwordCorrect && d.verifyPassword(Decryptor.DEFAULT_PASSWORD)) {\r
- passwordCorrect = true;\r
- }\r
- if (passwordCorrect) {\r
- stream = d.getDataStream(root);\r
- }\r
-\r
- if (!passwordCorrect) {\r
- String err = (password != null)\r
- ? "Password incorrect"\r
- : "The supplied spreadsheet is protected, but no password was supplied";\r
- throw new EncryptedDocumentException(err);\r
- }\r
+ stream = DocumentFactoryHelper.getDecryptedStream(fs, password);\r
\r
return createXSLFSlideShow(stream);\r
- } catch (GeneralSecurityException e) {\r
- throw new IOException(e);\r
} finally {\r
if (stream != null) stream.close();\r
}\r
NPOIFSFileSystem fs = new NPOIFSFileSystem(inp);\r
return create(fs, password);\r
}\r
- if (hasOOXMLHeader(inp)) {\r
+ if (DocumentFactoryHelper.hasOOXMLHeader(inp)) {\r
return createXSLFSlideShow(inp);\r
}\r
throw new IllegalArgumentException("Your InputStream was neither an OLE2 stream, nor an OOXML stream");\r
throw new IOException(e);\r
}\r
}\r
-\r
- /**\r
- * This copied over from ooxml, because we can't rely on these classes in the main package\r
- * \r
- * @see org.apache.poi.POIXMLDocument#hasOOXMLHeader(InputStream)\r
- */\r
- protected static boolean hasOOXMLHeader(InputStream inp) throws IOException {\r
- // We want to peek at the first 4 bytes\r
- inp.mark(4);\r
-\r
- byte[] header = new byte[4];\r
- int bytesRead = IOUtils.readFully(inp, header);\r
-\r
- // Wind back those 4 bytes\r
- if(inp instanceof PushbackInputStream) {\r
- PushbackInputStream pin = (PushbackInputStream)inp;\r
- pin.unread(header, 0, bytesRead);\r
- } else {\r
- inp.reset();\r
- }\r
-\r
- // Did it match the ooxml zip signature?\r
- return (\r
- bytesRead == 4 &&\r
- header[0] == OOXML_FILE_HEADER[0] &&\r
- header[1] == OOXML_FILE_HEADER[1] &&\r
- header[2] == OOXML_FILE_HEADER[2] &&\r
- header[3] == OOXML_FILE_HEADER[3]\r
- );\r
- }\r
-\r
}\r
import org.apache.poi.openxml4j.opc.PackageRelationship;
import org.apache.poi.openxml4j.opc.PackageRelationshipCollection;
import org.apache.poi.poifs.common.POIFSConstants;
+import org.apache.poi.poifs.filesystem.DocumentFactoryHelper;
import org.apache.poi.util.IOUtils;
import org.apache.xmlbeans.impl.common.SystemCache;
* then wrap it in a PushBackInputStream, then be
* sure to always use that, and not the original!
* @param inp An InputStream which supports either mark/reset, or is a PushbackInputStream
+ *
+ * @deprecated use the method from DocumentFactoryHelper, deprecated as of 3.15-beta1, therefore eligible for removal in 3.17
*/
public static boolean hasOOXMLHeader(InputStream inp) throws IOException {
- // We want to peek at the first 4 bytes
- inp.mark(4);
-
- byte[] header = new byte[4];
- int bytesRead = IOUtils.readFully(inp, header);
-
- // Wind back those 4 bytes
- if(inp instanceof PushbackInputStream) {
- PushbackInputStream pin = (PushbackInputStream)inp;
- pin.unread(header, 0, bytesRead);
- } else {
- inp.reset();
- }
-
- // Did it match the ooxml zip signature?
- return (
- bytesRead == 4 &&
- header[0] == POIFSConstants.OOXML_FILE_HEADER[0] &&
- header[1] == POIFSConstants.OOXML_FILE_HEADER[1] &&
- header[2] == POIFSConstants.OOXML_FILE_HEADER[2] &&
- header[3] == POIFSConstants.OOXML_FILE_HEADER[3]
- );
+ return DocumentFactoryHelper.hasOOXMLHeader(inp);
}
/**
import org.apache.poi.openxml4j.opc.PackagePart;
import org.apache.poi.openxml4j.opc.PackageRelationshipCollection;
import org.apache.poi.openxml4j.opc.PackageRelationshipTypes;
-import org.apache.poi.poifs.filesystem.DirectoryEntry;
-import org.apache.poi.poifs.filesystem.DirectoryNode;
-import org.apache.poi.poifs.filesystem.Entry;
-import org.apache.poi.poifs.filesystem.NPOIFSFileSystem;
-import org.apache.poi.poifs.filesystem.NotOLE2FileException;
-import org.apache.poi.poifs.filesystem.OPOIFSFileSystem;
-import org.apache.poi.poifs.filesystem.OfficeXmlFileException;
-import org.apache.poi.poifs.filesystem.POIFSFileSystem;
+import org.apache.poi.poifs.filesystem.*;
import org.apache.poi.xdgf.extractor.XDGFVisioExtractor;
import org.apache.poi.xslf.extractor.XSLFPowerPointExtractor;
import org.apache.poi.xslf.usermodel.XSLFRelation;
if(NPOIFSFileSystem.hasPOIFSHeader(inp)) {
return createExtractor(new NPOIFSFileSystem(inp));
}
- if(POIXMLDocument.hasOOXMLHeader(inp)) {
+ if(DocumentFactoryHelper.hasOOXMLHeader(inp)) {
return createExtractor(OPCPackage.open(inp));
}
throw new IllegalArgumentException("Your InputStream was neither an OLE2 stream, nor an OOXML stream");
package org.apache.poi.ss.usermodel;
import java.io.*;
-import java.security.GeneralSecurityException;
import org.apache.poi.EmptyFileException;
import org.apache.poi.EncryptedDocumentException;
-import org.apache.poi.POIXMLDocument;
+import org.apache.poi.poifs.filesystem.DocumentFactoryHelper;
import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.openxml4j.opc.PackageAccess;
import org.apache.poi.poifs.crypt.Decryptor;
-import org.apache.poi.poifs.crypt.EncryptionInfo;
import org.apache.poi.poifs.filesystem.DirectoryNode;
import org.apache.poi.poifs.filesystem.NPOIFSFileSystem;
import org.apache.poi.poifs.filesystem.OfficeXmlFileException;
// Encrypted OOXML files go inside OLE2 containers, is this one?
if (root.hasEntry(Decryptor.DEFAULT_POIFS_ENTRY)) {
- EncryptionInfo info = new EncryptionInfo(fs);
- Decryptor d = Decryptor.getInstance(info);
-
- boolean passwordCorrect = false;
- InputStream stream = null;
- try {
- if (password != null && d.verifyPassword(password)) {
- passwordCorrect = true;
- }
- if (!passwordCorrect && d.verifyPassword(Decryptor.DEFAULT_PASSWORD)) {
- passwordCorrect = true;
- }
- if (passwordCorrect) {
- // wrap the stream in a FilterInputStream to close the NPOIFSFileSystem
- // as well when the resulting OPCPackage is closed
- stream = new FilterInputStream(d.getDataStream(root)) {
- @Override
- public void close() throws IOException {
- fs.close();
-
- super.close();
- }
- };
- }
- } catch (GeneralSecurityException e) {
- throw new IOException(e);
- }
-
- if (! passwordCorrect) {
- if (password != null)
- throw new EncryptedDocumentException("Password incorrect");
- else
- throw new EncryptedDocumentException("The supplied spreadsheet is protected, but no password was supplied");
- }
+ InputStream stream = DocumentFactoryHelper.getDecryptedStream(fs, password);
OPCPackage pkg = OPCPackage.open(stream);
return create(pkg);
NPOIFSFileSystem fs = new NPOIFSFileSystem(inp);
return create(fs, password);
}
- if (POIXMLDocument.hasOOXMLHeader(inp)) {
+ if (DocumentFactoryHelper.hasOOXMLHeader(inp)) {
return new XSSFWorkbook(OPCPackage.open(inp));
}
throw new InvalidFormatException("Your InputStream was neither an OLE2 stream, nor an OOXML stream");
import org.apache.poi.hssf.HSSFTestDataSamples;
import org.apache.poi.openxml4j.opc.OPCPackage;
+import org.apache.poi.poifs.filesystem.DocumentFactoryHelper;
/**
* Class to test that HXF correctly detects OOXML
in = new PushbackInputStream(
HSSFTestDataSamples.openSampleFileStream("SampleSS.xlsx"), 10
);
- assertTrue(POIXMLDocument.hasOOXMLHeader(in));
+ assertTrue(DocumentFactoryHelper.hasOOXMLHeader(in));
in.close();
// xls file isn't
in = new PushbackInputStream(
HSSFTestDataSamples.openSampleFileStream("SampleSS.xls"), 10
);
- assertFalse(POIXMLDocument.hasOOXMLHeader(in));
+ assertFalse(DocumentFactoryHelper.hasOOXMLHeader(in));
in.close();
// text file isn't
in = new PushbackInputStream(
HSSFTestDataSamples.openSampleFileStream("SampleSS.txt"), 10
);
- assertFalse(POIXMLDocument.hasOOXMLHeader(in));
+ assertFalse(DocumentFactoryHelper.hasOOXMLHeader(in));
in.close();
}
// detect header
InputStream in = new PushbackInputStream(testInput, 10);
- assertFalse(POIXMLDocument.hasOOXMLHeader(in));
+ assertFalse(DocumentFactoryHelper.hasOOXMLHeader(in));
+ //noinspection deprecation
+ assertFalse(POIXMLDocument.hasOOXMLHeader(in));
// check if InputStream is still intact
byte[] test = new byte[3];
- in.read(test);
+ assertEquals(3, in.read(test));
assertTrue(Arrays.equals(testData, test));
assertEquals(-1, in.read());
}
-
}
projects / poi.git / commitdiff