moparisthebest/poi
1
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

Allow optional dumping of the raw ministream as well

Browse Source 
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1688533 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Nick Burch 2015-06-30 23:25:41 +00:00
parent f5e3e5d64d
commit 51af6e58b4
1 changed files with 38 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/java/org/apache/poi/poifs/dev/POIFSDump.java
View File

@ -25,31 +25,39 @@ import java.lang.reflect.Field;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.Iterator; import java.util.Iterator;
import org.apache.poi.poifs.common.POIFSConstants;
import org.apache.poi.poifs.filesystem.DirectoryEntry; import org.apache.poi.poifs.filesystem.DirectoryEntry;
import org.apache.poi.poifs.filesystem.DocumentInputStream; import org.apache.poi.poifs.filesystem.DocumentInputStream;
import org.apache.poi.poifs.filesystem.DocumentNode; import org.apache.poi.poifs.filesystem.DocumentNode;
import org.apache.poi.poifs.filesystem.Entry; import org.apache.poi.poifs.filesystem.Entry;
import org.apache.poi.poifs.filesystem.NPOIFSFileSystem; import org.apache.poi.poifs.filesystem.NPOIFSFileSystem;
import org.apache.poi.poifs.filesystem.NPOIFSStream; import org.apache.poi.poifs.filesystem.NPOIFSStream;
import org.apache.poi.poifs.property.NPropertyTable;
import org.apache.poi.poifs.storage.HeaderBlock; import org.apache.poi.poifs.storage.HeaderBlock;
/** /**
* Dump internal structure of a OLE2 file into file system * Dump internal structure of a OLE2 file into file system
*/ */
public class POIFSDump { public class POIFSDump {
public static void main(String[] args) throws Exception { public static void main(String[] args) throws Exception {
if (args.length == 0) { if (args.length == 0) {
System.err.println("Must specify at least one file to dump"); System.err.println("Must specify at least one file to dump");
System.exit(1); System.exit(1);
} }
boolean withProps = false; boolean dumpProps = false, dumpMini = false;
for (int i = 0; i < args.length; i++) { for (int i = 0; i < args.length; i++) {
if (args[i].equalsIgnoreCase("-dumprops") || if (args[i].equalsIgnoreCase("-dumprops") ||
args[i].equalsIgnoreCase("-dump-props") || args[i].equalsIgnoreCase("-dump-props") ||
args[i].equalsIgnoreCase("-dump-properties")) { args[i].equalsIgnoreCase("-dump-properties")) {
withProps = true; dumpProps = true;
continue;
}
if (args[i].equalsIgnoreCase("-dumpmini") ||
args[i].equalsIgnoreCase("-dump-mini") ||
args[i].equalsIgnoreCase("-dump-ministream") ||
args[i].equalsIgnoreCase("-dump-mini-stream")) {
dumpMini = true;
continue; continue;
} }
@ -64,15 +72,34 @@ public class POIFSDump {
dump(root, file); dump(root, file);
if (withProps) { if (dumpProps) {
Field headerF = NPOIFSFileSystem.class.getDeclaredField("_header"); HeaderBlock header = getHeaderBlock(fs);
headerF.setAccessible(true); dump(fs, header.getPropertyStart(), "properties", file);
HeaderBlock header = (HeaderBlock)headerF.get(fs); }
dump(fs, header.getPropertyStart(), file); if (dumpMini) {
NPropertyTable props = getPropertyTable(fs);
int startBlock = props.getRoot().getStartBlock();
if (startBlock == POIFSConstants.END_OF_CHAIN) {
System.err.println("No Mini Stream in file");
} else {
dump(fs, startBlock, "mini-stream", file);
}
} }
} }
} }
protected static HeaderBlock getHeaderBlock(NPOIFSFileSystem fs) throws Exception {
Field headerF = NPOIFSFileSystem.class.getDeclaredField("_header");
headerF.setAccessible(true);
HeaderBlock header = (HeaderBlock)headerF.get(fs);
return header;
}
protected static NPropertyTable getPropertyTable(NPOIFSFileSystem fs) throws Exception {
Field ptF = NPOIFSFileSystem.class.getDeclaredField("_property_table");
ptF.setAccessible(true);
NPropertyTable table = (NPropertyTable)ptF.get(fs);
return table;
}
public static void dump(DirectoryEntry root, File parent) throws IOException { public static void dump(DirectoryEntry root, File parent) throws IOException {
for(Iterator<Entry> it = root.getEntries(); it.hasNext();){ for(Iterator<Entry> it = root.getEntries(); it.hasNext();){
@ -100,8 +127,8 @@ public class POIFSDump {
} }
} }
} }
public static void dump(NPOIFSFileSystem fs, int startBlock, File parent) throws IOException { public static void dump(NPOIFSFileSystem fs, int startBlock, String name, File parent) throws IOException {
File file = new File(parent, "properties"); File file = new File(parent, name);
FileOutputStream out = new FileOutputStream(file); FileOutputStream out = new FileOutputStream(file);
NPOIFSStream stream = new NPOIFSStream(fs, startBlock); NPOIFSStream stream = new NPOIFSStream(fs, startBlock);