Examples with COSName com.tom_roush.pdfbox.cos.COSName used on opensource projects

Example 41 with COSName

use of com.tom_roush.pdfbox.cos.COSName in project PdfBox-Android by TomRoush.

the class SecurityHandler method decryptDictionary.

/**
 * This will decrypt a dictionary.
 *
 * @param dictionary The dictionary to decrypt.
 * @param objNum The object number.
 * @param genNum The object generation number.
 *
 * @throws IOException If there is an error creating a new string.
 */
private void decryptDictionary(COSDictionary dictionary, long objNum, long genNum) throws IOException {
    if (dictionary.getItem(COSName.CF) != null) {
        // PDFBOX-2936: avoid orphan /CF dictionaries found in US govt "I-" files
        return;
    }
    COSBase type = dictionary.getDictionaryObject(COSName.TYPE);
    boolean isSignature = COSName.SIG.equals(type) || COSName.DOC_TIME_STAMP.equals(type) || // https://ec.europa.eu/cefdigital/tracker/browse/DSS-1538
    (dictionary.getDictionaryObject(COSName.CONTENTS) instanceof COSString && dictionary.getDictionaryObject(COSName.BYTERANGE) instanceof COSArray);
    for (Map.Entry<COSName, COSBase> entry : dictionary.entrySet()) {
        if (isSignature && COSName.CONTENTS.equals(entry.getKey())) {
            // do not decrypt the signature contents string
            continue;
        }
        COSBase value = entry.getValue();
        // within a dictionary only the following kind of COS objects have to be decrypted
        if (value instanceof COSString || value instanceof COSArray || value instanceof COSDictionary) {
            decrypt(value, objNum, genNum);
        }
    }
}

Example 42 with COSName

use of com.tom_roush.pdfbox.cos.COSName in project PdfBox-Android by TomRoush.

the class PDStructureElement method addAttribute.

/**
 * Adds an attribute object.
 *
 * @param attributeObject the attribute object
 */
public void addAttribute(PDAttributeObject attributeObject) {
    COSName key = COSName.A;
    attributeObject.setStructureElement(this);
    COSBase a = this.getCOSObject().getDictionaryObject(key);
    COSArray array;
    if (a instanceof COSArray) {
        array = (COSArray) a;
    } else {
        array = new COSArray();
        if (a != null) {
            array.add(a);
            array.add(COSInteger.get(0));
        }
    }
    this.getCOSObject().setItem(key, array);
    array.add(attributeObject);
    array.add(COSInteger.get(this.getRevisionNumber()));
}

Example 43 with COSName

use of com.tom_roush.pdfbox.cos.COSName in project PdfBox-Android by TomRoush.

the class PDStructureElement method getClassNames.

/**
 * Returns the class names together with their revision numbers (C).
 *
 * @return the class names as a list, never null.
 */
public Revisions<String> getClassNames() {
    COSName key = COSName.C;
    Revisions<String> classNames = new Revisions<String>();
    COSBase c = this.getCOSObject().getDictionaryObject(key);
    if (c instanceof COSName) {
        classNames.addObject(((COSName) c).getName(), 0);
    }
    if (c instanceof COSArray) {
        COSArray array = (COSArray) c;
        Iterator<COSBase> it = array.iterator();
        String className = null;
        while (it.hasNext()) {
            COSBase item = it.next();
            if (item instanceof COSObject) {
                item = ((COSObject) item).getObject();
            }
            if (item instanceof COSName) {
                className = ((COSName) item).getName();
                classNames.addObject(className, 0);
            } else if (item instanceof COSInteger) {
                classNames.setRevisionNumber(className, ((COSInteger) item).intValue());
            }
        }
    }
    return classNames;
}

Example 44 with COSName

use of com.tom_roush.pdfbox.cos.COSName in project PdfBox-Android by TomRoush.

the class PDStandardAttributeObject method getNameOrArrayOfName.

/**
 * Gets a name value or array of name values.
 *
 * @param name the attribute name
 * @param defaultValue the default value
 * @return a String or array of Strings
 */
protected Object getNameOrArrayOfName(String name, String defaultValue) {
    COSBase v = this.getCOSObject().getDictionaryObject(name);
    if (v instanceof COSArray) {
        COSArray array = (COSArray) v;
        String[] names = new String[array.size()];
        for (int i = 0; i < array.size(); i++) {
            COSBase item = array.getObject(i);
            if (item instanceof COSName) {
                names[i] = ((COSName) item).getName();
            }
        }
        return names;
    }
    if (v instanceof COSName) {
        return ((COSName) v).getName();
    }
    return defaultValue;
}

Example 45 with COSName

use of com.tom_roush.pdfbox.cos.COSName in project PdfBox-Android by TomRoush.

the class PDFMergerUtility method appendDocument.

/**
 * append all pages from source to destination.
 *
 * @param destination the document to receive the pages
 * @param source the document originating the new pages
 *
 * @throws IOException If there is an error accessing data from either
 * document.
 */
public void appendDocument(PDDocument destination, PDDocument source) throws IOException {
    if (source.getDocument().isClosed()) {
        throw new IOException("Error: source PDF is closed.");
    }
    if (destination.getDocument().isClosed()) {
        throw new IOException("Error: destination PDF is closed.");
    }
    PDDocumentCatalog destCatalog = destination.getDocumentCatalog();
    PDDocumentCatalog srcCatalog = source.getDocumentCatalog();
    if (isDynamicXfa(srcCatalog.getAcroForm())) {
        throw new IOException("Error: can't merge source document containing dynamic XFA form content.");
    }
    PDDocumentInformation destInfo = destination.getDocumentInformation();
    PDDocumentInformation srcInfo = source.getDocumentInformation();
    mergeInto(srcInfo.getCOSObject(), destInfo.getCOSObject(), Collections.<COSName>emptySet());
    // use the highest version number for the resulting pdf
    float destVersion = destination.getVersion();
    float srcVersion = source.getVersion();
    if (destVersion < srcVersion) {
        destination.setVersion(srcVersion);
    }
    int pageIndexOpenActionDest = -1;
    if (destCatalog.getOpenAction() == null) {
        // PDFBOX-3972: get local dest page index, it must be reassigned after the page cloning
        PDDestinationOrAction openAction = null;
        try {
            openAction = srcCatalog.getOpenAction();
        } catch (IOException ex) {
            // PDFBOX-4223
            Log.e("PdfBox-Android", "Invalid OpenAction ignored", ex);
        }
        PDDestination openActionDestination = null;
        if (openAction instanceof PDActionGoTo) {
            openActionDestination = ((PDActionGoTo) openAction).getDestination();
        } else if (openAction instanceof PDDestination) {
            openActionDestination = (PDDestination) openAction;
        }
        if (openActionDestination instanceof PDPageDestination) {
            PDPage page = ((PDPageDestination) openActionDestination).getPage();
            if (page != null) {
                pageIndexOpenActionDest = srcCatalog.getPages().indexOf(page);
            }
        }
        destCatalog.setOpenAction(openAction);
    }
    PDFCloneUtility cloner = new PDFCloneUtility(destination);
    mergeAcroForm(cloner, destCatalog, srcCatalog);
    COSArray destThreads = (COSArray) destCatalog.getCOSObject().getDictionaryObject(COSName.THREADS);
    COSArray srcThreads = (COSArray) cloner.cloneForNewDocument(destCatalog.getCOSObject().getDictionaryObject(COSName.THREADS));
    if (destThreads == null) {
        destCatalog.getCOSObject().setItem(COSName.THREADS, srcThreads);
    } else {
        destThreads.addAll(srcThreads);
    }
    PDDocumentNameDictionary destNames = destCatalog.getNames();
    PDDocumentNameDictionary srcNames = srcCatalog.getNames();
    if (srcNames != null) {
        if (destNames == null) {
            destCatalog.getCOSObject().setItem(COSName.NAMES, cloner.cloneForNewDocument(srcNames));
        } else {
            cloner.cloneMerge(srcNames, destNames);
        }
    }
    if (destNames != null) {
        // found in 054080.pdf from PDFBOX-4417 and doesn't belong there
        destNames.getCOSObject().removeItem(COSName.ID_TREE);
        Log.w("PdfBox-Android", "Removed /IDTree from /Names dictionary, doesn't belong there");
    }
    PDDocumentNameDestinationDictionary destDests = destCatalog.getDests();
    PDDocumentNameDestinationDictionary srcDests = srcCatalog.getDests();
    if (srcDests != null) {
        if (destDests == null) {
            destCatalog.getCOSObject().setItem(COSName.DESTS, cloner.cloneForNewDocument(srcDests));
        } else {
            cloner.cloneMerge(srcDests, destDests);
        }
    }
    PDDocumentOutline destOutline = destCatalog.getDocumentOutline();
    PDDocumentOutline srcOutline = srcCatalog.getDocumentOutline();
    if (srcOutline != null) {
        if (destOutline == null || destOutline.getFirstChild() == null) {
            PDDocumentOutline cloned = new PDDocumentOutline((COSDictionary) cloner.cloneForNewDocument(srcOutline));
            destCatalog.setDocumentOutline(cloned);
        } else {
            // search last sibling for dest, because /Last entry is sometimes wrong
            PDOutlineItem destLastOutlineItem = destOutline.getFirstChild();
            while (destLastOutlineItem.getNextSibling() != null) {
                destLastOutlineItem = destLastOutlineItem.getNextSibling();
            }
            for (PDOutlineItem item : srcOutline.children()) {
                // get each child, clone its dictionary, remove siblings info,
                // append outline item created from there
                COSDictionary clonedDict = (COSDictionary) cloner.cloneForNewDocument(item);
                clonedDict.removeItem(COSName.PREV);
                clonedDict.removeItem(COSName.NEXT);
                PDOutlineItem clonedItem = new PDOutlineItem(clonedDict);
                destLastOutlineItem.insertSiblingAfter(clonedItem);
                destLastOutlineItem = destLastOutlineItem.getNextSibling();
            }
        }
    }
    PageMode destPageMode = destCatalog.getPageMode();
    PageMode srcPageMode = srcCatalog.getPageMode();
    if (destPageMode == null) {
        destCatalog.setPageMode(srcPageMode);
    }
    COSDictionary destLabels = destCatalog.getCOSObject().getCOSDictionary(COSName.PAGE_LABELS);
    COSDictionary srcLabels = srcCatalog.getCOSObject().getCOSDictionary(COSName.PAGE_LABELS);
    if (srcLabels != null) {
        int destPageCount = destination.getNumberOfPages();
        COSArray destNums;
        if (destLabels == null) {
            destLabels = new COSDictionary();
            destNums = new COSArray();
            destLabels.setItem(COSName.NUMS, destNums);
            destCatalog.getCOSObject().setItem(COSName.PAGE_LABELS, destLabels);
        } else {
            destNums = (COSArray) destLabels.getDictionaryObject(COSName.NUMS);
        }
        COSArray srcNums = (COSArray) srcLabels.getDictionaryObject(COSName.NUMS);
        if (srcNums != null) {
            int startSize = destNums.size();
            for (int i = 0; i < srcNums.size(); i += 2) {
                COSBase base = srcNums.getObject(i);
                if (!(base instanceof COSNumber)) {
                    Log.e("PdfBox-Android", "page labels ignored, index " + i + " should be a number, but is " + base);
                    // remove what we added
                    while (destNums.size() > startSize) {
                        destNums.remove(startSize);
                    }
                    break;
                }
                COSNumber labelIndex = (COSNumber) base;
                long labelIndexValue = labelIndex.intValue();
                destNums.add(COSInteger.get(labelIndexValue + destPageCount));
                destNums.add(cloner.cloneForNewDocument(srcNums.getObject(i + 1)));
            }
        }
    }
    COSStream destMetadata = destCatalog.getCOSObject().getCOSStream(COSName.METADATA);
    COSStream srcMetadata = srcCatalog.getCOSObject().getCOSStream(COSName.METADATA);
    if (destMetadata == null && srcMetadata != null) {
        try {
            PDStream newStream = new PDStream(destination, srcMetadata.createInputStream(), (COSName) null);
            mergeInto(srcMetadata, newStream.getCOSObject(), new HashSet<COSName>(Arrays.asList(COSName.FILTER, COSName.LENGTH)));
            destCatalog.getCOSObject().setItem(COSName.METADATA, newStream);
        } catch (IOException ex) {
            // PDFBOX-4227 cleartext XMP stream with /Flate
            Log.e("PdfBox-Android", "Metadata skipped because it could not be read", ex);
        }
    }
    COSDictionary destOCP = destCatalog.getCOSObject().getCOSDictionary(COSName.OCPROPERTIES);
    COSDictionary srcOCP = srcCatalog.getCOSObject().getCOSDictionary(COSName.OCPROPERTIES);
    if (destOCP == null && srcOCP != null) {
        destCatalog.getCOSObject().setItem(COSName.OCPROPERTIES, cloner.cloneForNewDocument(srcOCP));
    } else if (destOCP != null && srcOCP != null) {
        cloner.cloneMerge(srcOCP, destOCP);
    }
    mergeOutputIntents(cloner, srcCatalog, destCatalog);
    // merge logical structure hierarchy
    boolean mergeStructTree = false;
    int destParentTreeNextKey = -1;
    Map<Integer, COSObjectable> srcNumberTreeAsMap = null;
    Map<Integer, COSObjectable> destNumberTreeAsMap = null;
    PDStructureTreeRoot srcStructTree = srcCatalog.getStructureTreeRoot();
    PDStructureTreeRoot destStructTree = destCatalog.getStructureTreeRoot();
    if (destStructTree == null && srcStructTree != null) {
        // create a dummy structure tree in the destination, so that the source
        // tree is cloned. (We can't just copy the tree reference due to PDFBOX-3999)
        destStructTree = new PDStructureTreeRoot();
        destCatalog.setStructureTreeRoot(destStructTree);
        destStructTree.setParentTree(new PDNumberTreeNode(PDParentTreeValue.class));
        // PDFBOX-4429: remove bogus StructParent(s)
        for (PDPage page : destCatalog.getPages()) {
            page.getCOSObject().removeItem(COSName.STRUCT_PARENTS);
            for (PDAnnotation ann : page.getAnnotations()) {
                ann.getCOSObject().removeItem(COSName.STRUCT_PARENT);
            }
        }
    }
    if (destStructTree != null) {
        PDNumberTreeNode destParentTree = destStructTree.getParentTree();
        destParentTreeNextKey = destStructTree.getParentTreeNextKey();
        if (destParentTree != null) {
            destNumberTreeAsMap = getNumberTreeAsMap(destParentTree);
            if (destParentTreeNextKey < 0) {
                if (destNumberTreeAsMap.isEmpty()) {
                    destParentTreeNextKey = 0;
                } else {
                    destParentTreeNextKey = Collections.max(destNumberTreeAsMap.keySet()) + 1;
                }
            }
            if (destParentTreeNextKey >= 0 && srcStructTree != null) {
                PDNumberTreeNode srcParentTree = srcStructTree.getParentTree();
                if (srcParentTree != null) {
                    srcNumberTreeAsMap = getNumberTreeAsMap(srcParentTree);
                    if (!srcNumberTreeAsMap.isEmpty()) {
                        mergeStructTree = true;
                    }
                }
            }
        }
    }
    Map<COSDictionary, COSDictionary> objMapping = new HashMap<COSDictionary, COSDictionary>();
    int pageIndex = 0;
    for (PDPage page : srcCatalog.getPages()) {
        PDPage newPage = new PDPage((COSDictionary) cloner.cloneForNewDocument(page.getCOSObject()));
        if (!mergeStructTree) {
            // PDFBOX-4429: remove bogus StructParent(s)
            newPage.getCOSObject().removeItem(COSName.STRUCT_PARENTS);
            for (PDAnnotation ann : newPage.getAnnotations()) {
                ann.getCOSObject().removeItem(COSName.STRUCT_PARENT);
            }
        }
        newPage.setCropBox(page.getCropBox());
        newPage.setMediaBox(page.getMediaBox());
        newPage.setRotation(page.getRotation());
        PDResources resources = page.getResources();
        if (resources != null) {
            // this is smart enough to just create references for resources that are used on multiple pages
            newPage.setResources(new PDResources((COSDictionary) cloner.cloneForNewDocument(resources)));
        } else {
            newPage.setResources(new PDResources());
        }
        if (mergeStructTree) {
            // add the value of the destination ParentTreeNextKey to every source element
            // StructParent(s) value so that these don't overlap with the existing values
            updateStructParentEntries(newPage, destParentTreeNextKey);
            objMapping.put(page.getCOSObject(), newPage.getCOSObject());
            List<PDAnnotation> oldAnnots = page.getAnnotations();
            List<PDAnnotation> newAnnots = newPage.getAnnotations();
            for (int i = 0; i < oldAnnots.size(); i++) {
                objMapping.put(oldAnnots.get(i).getCOSObject(), newAnnots.get(i).getCOSObject());
            }
        // TODO update mapping for XObjects
        }
        destination.addPage(newPage);
        if (pageIndex == pageIndexOpenActionDest) {
            // PDFBOX-3972: reassign the page.
            // The openAction is either a PDActionGoTo or a PDPageDestination
            PDDestinationOrAction openAction = destCatalog.getOpenAction();
            PDPageDestination pageDestination;
            if (openAction instanceof PDActionGoTo) {
                pageDestination = (PDPageDestination) ((PDActionGoTo) openAction).getDestination();
            } else {
                pageDestination = (PDPageDestination) openAction;
            }
            pageDestination.setPage(newPage);
        }
        ++pageIndex;
    }
    if (mergeStructTree) {
        updatePageReferences(cloner, srcNumberTreeAsMap, objMapping);
        int maxSrcKey = -1;
        for (Map.Entry<Integer, COSObjectable> entry : srcNumberTreeAsMap.entrySet()) {
            int srcKey = entry.getKey();
            maxSrcKey = Math.max(srcKey, maxSrcKey);
            destNumberTreeAsMap.put(destParentTreeNextKey + srcKey, cloner.cloneForNewDocument(entry.getValue()));
        }
        destParentTreeNextKey += maxSrcKey + 1;
        PDNumberTreeNode newParentTreeNode = new PDNumberTreeNode(PDParentTreeValue.class);
        // Note that all elements are stored flatly. This could become a problem for large files
        // when these are opened in a viewer that uses the tagging information.
        // If this happens, then ​PDNumberTreeNode should be improved with a convenience method that
        // stores the map into a B+Tree, see https://en.wikipedia.org/wiki/B+_tree
        newParentTreeNode.setNumbers(destNumberTreeAsMap);
        destStructTree.setParentTree(newParentTreeNode);
        destStructTree.setParentTreeNextKey(destParentTreeNextKey);
        mergeKEntries(cloner, srcStructTree, destStructTree);
        mergeRoleMap(srcStructTree, destStructTree);
        mergeIDTree(cloner, srcStructTree, destStructTree);
        mergeMarkInfo(destCatalog, srcCatalog);
        mergeLanguage(destCatalog, srcCatalog);
        mergeViewerPreferences(destCatalog, srcCatalog);
    }
}