public void TestReadWrite()
{
FileInformationBlock fib = _hWPFDocFixture._fib;
byte[] mainStream = _hWPFDocFixture._mainStream;
byte[] tableStream = _hWPFDocFixture._tableStream;
int fcMin = fib.GetFcMin();
CPSplitCalculator cps = new CPSplitCalculator(fib);
ComplexFileTable cft = new ComplexFileTable(mainStream, tableStream, fib.GetFcClx(), fcMin);
TextPieceTable tpt = cft.GetTextPieceTable();
SectionTable sectionTable = new SectionTable(mainStream, tableStream,
fib.GetFcPlcfsed(),
fib.GetLcbPlcfsed(),
fcMin, tpt, cps);
HWPFFileSystem fileSys = new HWPFFileSystem();
sectionTable.WriteTo(fileSys, 0);
MemoryStream tableOut = fileSys.GetStream("1Table");
MemoryStream mainOut = fileSys.GetStream("WordDocument");
byte[] newTableStream = tableOut.ToArray();
byte[] newMainStream = mainOut.ToArray();
SectionTable newSectionTable = new SectionTable(
newMainStream, newTableStream, 0,
newTableStream.Length, 0, tpt, cps);
List<SEPX> oldSections = sectionTable.GetSections();
List<SEPX> newSections = newSectionTable.GetSections();
Assert.AreEqual(oldSections.Count, newSections.Count);
//test for proper char offset conversions
PlexOfCps oldSedPlex = new PlexOfCps(tableStream, fib.GetFcPlcfsed(),
fib.GetLcbPlcfsed(), 12);
PlexOfCps newSedPlex = new PlexOfCps(newTableStream, 0,
newTableStream.Length, 12);
Assert.AreEqual(oldSedPlex.Length, newSedPlex.Length);
for (int x = 0; x < oldSedPlex.Length; x++)
{
Assert.AreEqual(oldSedPlex.GetProperty(x).Start, newSedPlex.GetProperty(x).Start);
Assert.AreEqual(oldSedPlex.GetProperty(x).End, newSedPlex.GetProperty(x).End);
}
int size = oldSections.Count;
for (int x = 0; x < size; x++)
{
PropertyNode oldNode = (PropertyNode)oldSections[x];
PropertyNode newNode = (PropertyNode)newSections[x];
Assert.AreEqual(oldNode, newNode);
}
}
public SectionTable(byte[] documentStream, byte[] tableStream, int OffSet,
int size, int fcMin,
TextPieceTable tpt, CPSplitCalculator cps)
{
PlexOfCps sedPlex = new PlexOfCps(tableStream, OffSet, size, SED_SIZE);
this.tpt = tpt;
this._text = tpt.TextPieces;
int length = sedPlex.Length;
for (int x = 0; x < length; x++)
{
GenericPropertyNode node = sedPlex.GetProperty(x);
SectionDescriptor sed = new SectionDescriptor(node.Bytes, 0);
int fileOffset = sed.GetFc();
int startAt = CPtoFC(node.Start);
int endAt = CPtoFC(node.End);
// check for the optimization
if (fileOffset == unchecked((int)0xffffffff))
{
_sections.Add(new SEPX(sed, startAt, endAt, new byte[0]));
}
else
{
// The first short at the offset is the size of the grpprl.
int sepxSize = LittleEndian.GetShort(documentStream, fileOffset);
byte[] buf = new byte[sepxSize];
fileOffset += LittleEndianConsts.SHORT_SIZE;
Array.Copy(documentStream, fileOffset, buf, 0, buf.Length);
_sections.Add(new SEPX(sed, startAt, endAt, buf));
}
}
// Some files seem to lie about their unicode status, which
// is very very pesky. Try to work around these, but this
// is Getting on for black magic...
int mainEndsAt = cps.GetMainDocumentEnd();
bool matchAt = false;
bool matchHalf = false;
for (int i = 0; i < _sections.Count; i++)
{
SEPX s = _sections[i];
if (s.End == mainEndsAt)
{
matchAt = true;
}
else if (s.EndBytes == mainEndsAt || s.EndBytes == mainEndsAt - 1)
{
matchHalf = true;
}
}
if (!matchAt && matchHalf)
{
//System.err.println("Your document seemed to be mostly unicode, but the section defInition was in bytes! Trying anyway, but things may well go wrong!");
for (int i = 0; i < _sections.Count; i++)
{
SEPX s = _sections[i];
GenericPropertyNode node = sedPlex.GetProperty(i);
int startAt = node.Start;
int endAt = node.End;
s.Start = (startAt);
s.End = (endAt);
}
}
}
/// <summary>
/// Initializes a new instance of the <see cref="HWPFDocument"/> class.
/// </summary>
/// <param name="directory">The directory.</param>
public HWPFDocument(DirectoryNode directory)
: base(directory)
{
_endnotes = new NotesImpl(_endnotesTables);
_footnotes = new NotesImpl(_footnotesTables);
// Load the main stream and FIB
// Also handles HPSF bits
// Do the CP Split
_cpSplit = new CPSplitCalculator(_fib);
// Is this document too old for us?
if (_fib.GetNFib() < 106)
{
throw new OldWordFileFormatException("The document is too old - Word 95 or older. Try HWPFOldDocument instead?");
}
// use the fib to determine the name of the table stream.
String name = "0Table";
if (_fib.IsFWhichTblStm())
{
name = "1Table";
}
// Grab the table stream.
DocumentEntry tableProps;
try
{
tableProps =
(DocumentEntry)directory.GetEntry(name);
}
catch (FileNotFoundException)
{
throw new InvalidOperationException("Table Stream '" + name + "' wasn't found - Either the document is corrupt, or is Word95 (or earlier)");
}
// read in the table stream.
_tableStream = new byte[tableProps.Size];
directory.CreatePOIFSDocumentReader(name).Read(_tableStream);
_fib.FillVariableFields(_mainStream, _tableStream);
// read in the data stream.
try
{
DocumentEntry dataProps =
(DocumentEntry)directory.GetEntry("Data");
_dataStream = new byte[dataProps.Size];
directory.CreatePOIFSDocumentReader("Data").Read(_dataStream);
}
catch (FileNotFoundException)
{
_dataStream = new byte[0];
}
// Get the cp of the start of text in the main stream
// The latest spec doc says this is always zero!
int fcMin = 0;
//fcMin = _fib.GetFcMin()
// Start to load up our standard structures.
_dop = new DocumentProperties(_tableStream, _fib.GetFcDop());
_cft = new ComplexFileTable(_mainStream, _tableStream, _fib.GetFcClx(), fcMin);
TextPieceTable _tpt = _cft.GetTextPieceTable();
// Now load the rest of the properties, which need to be adjusted
// for where text really begin
_cbt = new CHPBinTable(_mainStream, _tableStream, _fib.GetFcPlcfbteChpx(), _fib.GetLcbPlcfbteChpx(), _tpt);
_pbt = new PAPBinTable(_mainStream, _tableStream, _dataStream, _fib.GetFcPlcfbtePapx(), _fib.GetLcbPlcfbtePapx(), _tpt);
_text = _tpt.Text;
/*
* in this mode we preserving PAPX/CHPX structure from file, so text may
* miss from output, and text order may be corrupted
*/
bool preserveBinTables = false;
try
{
preserveBinTables = Boolean.Parse(
ConfigurationManager.AppSettings[PROPERTY_PRESERVE_BIN_TABLES]);
}
catch (Exception)
{
// ignore;
}
if (!preserveBinTables)
{
_cbt.Rebuild(_cft);
_pbt.Rebuild(_text, _cft);
}
/*
* Property to disable text rebuilding. In this mode changing the text
* will lead to unpredictable behavior
*/
bool preserveTextTable = false;
try
{
preserveTextTable = Boolean.Parse(
ConfigurationManager.AppSettings[PROPERTY_PRESERVE_TEXT_TABLE]);
}
catch (Exception)
{
// ignore;
}
if (!preserveTextTable)
{
_cft = new ComplexFileTable();
_tpt = _cft.GetTextPieceTable();
TextPiece textPiece = new SinglentonTextPiece(_text);
_tpt.Add(textPiece);
_text = textPiece.GetStringBuilder();
}
// Read FSPA and Escher information
// _fspa = new FSPATable(_tableStream, _fib.getFcPlcspaMom(),
// _fib.getLcbPlcspaMom(), getTextTable().getTextPieces());
_fspaHeaders = new FSPATable(_tableStream, _fib,
FSPADocumentPart.HEADER);
_fspaMain = new FSPATable(_tableStream, _fib, FSPADocumentPart.MAIN);
if (_fib.GetFcDggInfo() != 0)
{
_dgg = new EscherRecordHolder(_tableStream, _fib.GetFcDggInfo(), _fib.GetLcbDggInfo());
}
else
{
_dgg = new EscherRecordHolder();
}
// read in the pictures stream
_pictures = new PicturesTable(this, _dataStream, _mainStream, _fspa, _dgg);
// And the art shapes stream
_officeArts = new ShapesTable(_tableStream, _fib);
// And escher pictures
_officeDrawingsHeaders = new OfficeDrawingsImpl(_fspaHeaders, _dgg, _mainStream);
_officeDrawingsMain = new OfficeDrawingsImpl(_fspaMain, _dgg, _mainStream);
_st = new SectionTable(_mainStream, _tableStream, _fib.GetFcPlcfsed(), _fib.GetLcbPlcfsed(), fcMin, _tpt, _cpSplit);
_ss = new StyleSheet(_tableStream, _fib.GetFcStshf());
_ft = new FontTable(_tableStream, _fib.GetFcSttbfffn(), _fib.GetLcbSttbfffn());
int listOffset = _fib.GetFcPlcfLst();
int lfoOffset = _fib.GetFcPlfLfo();
if (listOffset != 0 && _fib.GetLcbPlcfLst() != 0)
{
_lt = new ListTables(_tableStream, _fib.GetFcPlcfLst(), _fib.GetFcPlfLfo());
}
int sbtOffset = _fib.GetFcSttbSavedBy();
int sbtLength = _fib.GetLcbSttbSavedBy();
if (sbtOffset != 0 && sbtLength != 0)
{
_sbt = new SavedByTable(_tableStream, sbtOffset, sbtLength);
}
int rmarkOffset = _fib.GetFcSttbfRMark();
int rmarkLength = _fib.GetLcbSttbfRMark();
if (rmarkOffset != 0 && rmarkLength != 0)
{
_rmat = new RevisionMarkAuthorTable(_tableStream, rmarkOffset, rmarkLength);
}
_bookmarksTables = new BookmarksTables(_tableStream, _fib);
_bookmarks = new BookmarksImpl(_bookmarksTables);
_endnotesTables = new NotesTables(NoteType.ENDNOTE, _tableStream, _fib);
_endnotes = new NotesImpl(_endnotesTables);
_footnotesTables = new NotesTables(NoteType.FOOTNOTE, _tableStream, _fib);
_footnotes = new NotesImpl(_footnotesTables);
_fieldsTables = new FieldsTables(_tableStream, _fib);
_fields = new FieldsImpl(_fieldsTables);
}
public SectionTable(byte[] documentStream, byte[] tableStream, int OffSet,
int size, int fcMin,
TextPieceTable tpt, CPSplitCalculator cps)
{
PlexOfCps sedPlex = new PlexOfCps(tableStream, OffSet, size, SED_SIZE);
this.tpt = tpt;
this._text = tpt.TextPieces;
int length = sedPlex.Length;
for (int x = 0; x < length; x++)
{
GenericPropertyNode node = sedPlex.GetProperty(x);
SectionDescriptor sed = new SectionDescriptor(node.Bytes, 0);
int fileOffset = sed.GetFc();
int startAt = CPtoFC(node.Start);
int endAt = CPtoFC(node.End);
// check for the optimization
if (fileOffset == unchecked ((int)0xffffffff))
{
_sections.Add(new SEPX(sed, startAt, endAt, new byte[0]));
}
else
{
// The first short at the offset is the size of the grpprl.
int sepxSize = LittleEndian.GetShort(documentStream, fileOffset);
byte[] buf = new byte[sepxSize];
fileOffset += LittleEndianConsts.SHORT_SIZE;
Array.Copy(documentStream, fileOffset, buf, 0, buf.Length);
_sections.Add(new SEPX(sed, startAt, endAt, buf));
}
}
// Some files seem to lie about their unicode status, which
// is very very pesky. Try to work around these, but this
// is Getting on for black magic...
int mainEndsAt = cps.GetMainDocumentEnd();
bool matchAt = false;
bool matchHalf = false;
for (int i = 0; i < _sections.Count; i++)
{
SEPX s = _sections[i];
if (s.End == mainEndsAt)
{
matchAt = true;
}
else if (s.EndBytes == mainEndsAt || s.EndBytes == mainEndsAt - 1)
{
matchHalf = true;
}
}
if (!matchAt && matchHalf)
{
//System.err.println("Your document seemed to be mostly unicode, but the section defInition was in bytes! Trying anyway, but things may well go wrong!");
for (int i = 0; i < _sections.Count; i++)
{
SEPX s = _sections[i];
GenericPropertyNode node = sedPlex.GetProperty(i);
int startAt = node.Start;
int endAt = node.End;
s.Start = (startAt);
s.End = (endAt);
}
}
}
