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

refix 51686 - fix possible NPE, add writeTo() to ObjectPoolImpl.

Browse Source 
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1160152 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Sergey Vladimirov 2011-08-22 08:55:21 +00:00
parent c3235cc745
commit 2ae382b2c9
4 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/scratchpad/src/org/apache/poi/hwpf/HWPFDocument.java
View File

@ -69,6 +69,7 @@ import org.apache.poi.poifs.filesystem.DocumentEntry;
import org.apache.poi.poifs.filesystem.Entry;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.util.Internal;
import org.apache.poi.util.POIUtils;
/**
@ -947,6 +948,7 @@ public final class HWPFDocument extends HWPFDocumentCore
POIFSFileSystem pfs = new POIFSFileSystem();
boolean docWritten = false;
boolean dataWritten = false;
boolean objectPoolWritten = false;
boolean tableWritten = false;
boolean propertiesWritten = false;
for ( Iterator<Entry> iter = directory.getEntries(); iter.hasNext(); )
@ -961,6 +963,14 @@ public final class HWPFDocument extends HWPFDocumentCore
docWritten = true;
}
}
else if ( entry.getName().equals( STREAM_OBJECT_POOL ) )
{
if ( !objectPoolWritten )
{
_objectPool.writeTo( pfs.getRoot() );
objectPoolWritten = true;
}
}
else if ( entry.getName().equals( STREAM_TABLE_0 )
|| entry.getName().equals( STREAM_TABLE_1 ) )
{
@ -993,7 +1003,7 @@ public final class HWPFDocument extends HWPFDocumentCore
}
else
{
copyNodeRecursively( entry, pfs.getRoot() );
POIUtils.copyNodeRecursively( entry, pfs.getRoot() );
}
}
@ -1008,6 +1018,8 @@ public final class HWPFDocument extends HWPFDocumentCore
if ( !dataWritten )
pfs.createDocument( new ByteArrayInputStream( dataBuf ),
STREAM_DATA );
if ( !objectPoolWritten )
_objectPool.writeTo( pfs.getRoot() );
pfs.writeFilesystem( out );
this.directory = pfs.getRoot();

6
src/scratchpad/src/org/apache/poi/hwpf/HWPFDocumentCore.java
View File

@ -158,12 +158,14 @@ public abstract class HWPFDocumentCore extends POIDocument
throw new EncryptedDocumentException("Cannot process encrypted word files!");
}
DirectoryEntry objectPoolEntry;
try {
DirectoryEntry objectPoolEntry = (DirectoryEntry) directory
objectPoolEntry = (DirectoryEntry) directory
.getEntry(STREAM_OBJECT_POOL);
_objectPool = new ObjectPoolImpl(objectPoolEntry);
} catch (FileNotFoundException exc) {
objectPoolEntry = null;
}
_objectPool = new ObjectPoolImpl(objectPoolEntry);
}
/**

9
src/scratchpad/src/org/apache/poi/hwpf/usermodel/ObjectPoolImpl.java
View File

@ -17,10 +17,12 @@
package org.apache.poi.hwpf.usermodel;
import java.io.FileNotFoundException;
import java.io.IOException;
import org.apache.poi.poifs.filesystem.DirectoryEntry;
import org.apache.poi.poifs.filesystem.Entry;
import org.apache.poi.util.Internal;
import org.apache.poi.util.POIUtils;
@Internal
public class ObjectPoolImpl implements ObjectsPool
@ -47,4 +49,11 @@ public class ObjectPoolImpl implements ObjectsPool
return null;
}
}
@Internal
public void writeTo( DirectoryEntry directoryEntry ) throws IOException
{
if ( _objectPool != null )
POIUtils.copyNodeRecursively( _objectPool, directoryEntry );
}
}

2
src/scratchpad/testcases/org/apache/poi/hwpf/extractor/TestWordExtractor.java
View File

@ -362,7 +362,7 @@ public final class TestWordExtractor extends TestCase {
* [RESOLVED FIXED] Bug 51686 - Update to POI 3.8 beta 4 causes
* ConcurrentModificationException in Tika's OfficeParser
*/
public void testRootEntiesNavigation() throws IOException
public void testBug51686() throws IOException
{
InputStream is = POIDataSamples.getDocumentInstance()
.openResourceAsStream( "Bug51686.doc" );