123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200 |
- /* ====================================================================
- 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.hslf.record;
-
- import java.io.ByteArrayInputStream;
- import java.io.ByteArrayOutputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.OutputStream;
- import java.util.Map;
- import java.util.zip.DeflaterOutputStream;
- import java.util.zip.InflaterInputStream;
-
- import org.apache.poi.util.BoundedInputStream;
- import org.apache.poi.util.LittleEndian;
-
- /**
- * Storage for embedded OLE objects.
- *
- * @author Daniel Noll
- */
- public class ExOleObjStg extends RecordAtom implements PositionDependentRecord, PersistRecord {
-
- private int _persistId; // Found from PersistPtrHolder
-
- /**
- * Record header.
- */
- private final byte[] _header;
-
- /**
- * Record data.
- */
- private byte[] _data;
-
- /**
- * Constructs a new empty storage container.
- */
- public ExOleObjStg() {
- _header = new byte[8];
- _data = new byte[0];
-
- LittleEndian.putShort(_header, 0, (short)0x10);
- LittleEndian.putShort(_header, 2, (short)getRecordType());
- LittleEndian.putInt(_header, 4, _data.length);
- }
-
- /**
- * Constructs the link related atom record from its
- * source data.
- *
- * @param source the source data as a byte array.
- * @param start the start offset into the byte array.
- * @param len the length of the slice in the byte array.
- */
- protected ExOleObjStg(byte[] source, int start, int len) {
- // Get the header.
- _header = new byte[8];
- System.arraycopy(source,start,_header,0,8);
-
- // Get the record data.
- _data = new byte[len-8];
- System.arraycopy(source,start+8,_data,0,len-8);
- }
-
- public boolean isCompressed() {
- return LittleEndian.getShort(_header, 0)!=0;
- }
-
- /**
- * Gets the uncompressed length of the data.
- *
- * @return the uncompressed length of the data.
- */
- public int getDataLength() {
- if (isCompressed()) {
- return LittleEndian.getInt(_data, 0);
- } else {
- return _data.length;
- }
- }
-
- /**
- * Opens an input stream which will decompress the data on the fly.
- *
- * @return the data input stream.
- */
- public InputStream getData() {
- if (isCompressed()) {
- int size = LittleEndian.getInt(_data);
-
- InputStream compressedStream = new ByteArrayInputStream(_data, 4, _data.length);
- return new BoundedInputStream(new InflaterInputStream(compressedStream), size);
- } else {
- return new ByteArrayInputStream(_data, 0, _data.length);
- }
- }
-
- public byte[] getRawData() {
- return _data;
- }
-
- /**
- * Sets the embedded data.
- *
- * @param data the embedded data.
- */
- public void setData(byte[] data) throws IOException {
- ByteArrayOutputStream out = new ByteArrayOutputStream();
- //first four bytes is the length of the raw data
- byte[] b = new byte[4];
- LittleEndian.putInt(b, 0, data.length);
- out.write(b);
-
- DeflaterOutputStream def = new DeflaterOutputStream(out);
- def.write(data, 0, data.length);
- def.finish();
- _data = out.toByteArray();
- LittleEndian.putInt(_header, 4, _data.length);
- }
-
- /**
- * Gets the record type.
- *
- * @return the record type.
- */
- public long getRecordType() {
- return RecordTypes.ExOleObjStg.typeID;
- }
-
- /**
- * Gets the record instance from the header
- *
- * @return record instance
- */
- public int getRecordInstance() {
- return (LittleEndian.getUShort(_header, 0) >>> 4);
- }
-
- /**
- * Write the contents of the record back, so it can be written
- * to disk.
- *
- * @param out the output stream to write to.
- * @throws IOException if an error occurs.
- */
- public void writeOut(OutputStream out) throws IOException {
- out.write(_header);
- out.write(_data);
- }
-
- /**
- * Fetch our sheet ID, as found from a PersistPtrHolder.
- * Should match the RefId of our matching SlidePersistAtom
- */
- public int getPersistId() {
- return _persistId;
- }
-
- /**
- * Set our sheet ID, as found from a PersistPtrHolder
- */
- public void setPersistId(int id) {
- _persistId = id;
- }
-
- /** Our location on the disk, as of the last write out */
- protected int myLastOnDiskOffset;
-
- /** Fetch our location on the disk, as of the last write out */
- public int getLastOnDiskOffset() { return myLastOnDiskOffset; }
-
- /**
- * Update the Record's idea of where on disk it lives, after a write out.
- * Use with care...
- */
- public void setLastOnDiskOffset(int offset) {
- myLastOnDiskOffset = offset;
- }
-
- @Override
- public void updateOtherRecordReferences(Map<Integer,Integer> oldToNewReferencesLookup) {
- // nothing to update
- }
- }
|