PointerFactory.java

/* ====================================================================
   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.hdgf.pointers;

import org.apache.poi.hdgf.exceptions.OldVisioFormatException;
import org.apache.poi.hdgf.streams.PointerContainingStream;
import org.apache.poi.util.ArrayUtil;
import org.apache.poi.util.IOUtils;
import org.apache.poi.util.LittleEndian;
import org.apache.poi.util.RecordFormatException;

/**
 * Factor class to create the appropriate pointers, based on the version
 *  of the file
 */
public final class PointerFactory {
    private static final int MAX_NUMBER_OF_POINTERS = 100_000;

    private final int version;

    public PointerFactory(int version) {
        this.version = version;
    }

    public int getVersion() { return version; }

    /**
     * Creates a single Pointer from the data at the given offset
     */
    public Pointer createPointer(byte[] data, int offset) {
        Pointer p;
        if(version >= 6) {
            p = new PointerV6();
            p.setType(LittleEndian.getInt(data, offset));
            p.setAddress((int)LittleEndian.getUInt(data, offset+4));
            // Offset and Length flow into Stream.createStream as the (offset, length)
            // pair handed to StreamStore / CompressedStreamStore -> IOUtils.safelyClone.
            // Validate the uint32 values up-front: throw RecordFormatException with the
            // offending value rather than letting a wrapped int reach the downstream
            // bounds check (matches the EMF header description fix in PR #1060).
            final long v6Offset = LittleEndian.getUInt(data, offset+8);
            final long v6Length = LittleEndian.getUInt(data, offset+12);
            checkPointerOffset(v6Offset);
            ArrayUtil.strictAllocateCheck(v6Length, Integer.MAX_VALUE);
            p.setOffset((int)v6Offset);
            p.setLength((int)v6Length);
            p.setFormat(LittleEndian.getShort(data, offset+16));

            return p;
        } else if(version == 5) {
            p = new PointerV5();
            p.setType(LittleEndian.getShort(data, offset));
            p.setFormat(LittleEndian.getShort(data, offset+2));
            p.setAddress((int)LittleEndian.getUInt(data, offset+4));
            final long v5Offset = LittleEndian.getUInt(data, offset+8);
            final long v5Length = LittleEndian.getUInt(data, offset+12);
            checkPointerOffset(v5Offset);
            IOUtils.safelyAllocateCheck(v5Length, Integer.MAX_VALUE);
            p.setOffset((int)v5Offset);
            p.setLength((int)v5Length);

            return p;
        } else {
            throw new OldVisioFormatException("Visio files with versions below 5 are not supported, yours was " + version);
        }
    }

    private static void checkPointerOffset(long unsignedOffset) {
        if (unsignedOffset > Integer.MAX_VALUE) {
            throw new RecordFormatException(
                    "HDGF Pointer offset " + unsignedOffset + " exceeds Integer.MAX_VALUE");
        }
    }

    /**
     * Parsers the {@link PointerContainingStream} contents and
     *  creates all the child Pointers for it
     */
    public Pointer[] createContainerPointers(Pointer parent, byte[] data) {
        // Where in the stream does the "number of pointers" offset live?
        int numPointersOffset = parent.getNumPointersOffset(data);
        // How many do we have?
        int numPointers = parent.getNumPointers(numPointersOffset, data);
        // How much to skip for the num pointers + any extra data?
        int skip = parent.getPostNumPointersSkip();

        if (numPointers < 0) {
            throw new IllegalArgumentException("Cannot create container pointers with negative count: " + numPointers);
        }

        ArrayUtil.strictAllocateCheck(numPointers, MAX_NUMBER_OF_POINTERS);

        // Create
        int pos = numPointersOffset + skip;
        Pointer[] childPointers = new Pointer[numPointers];
        for(int i=0; i<numPointers; i++) {
            childPointers[i] = this.createPointer(data, pos);
            pos += childPointers[i].getSizeInBytes();
        }

        return childPointers;
    }
}