aboutsummaryrefslogtreecommitdiffstats
path: root/src/java/org/apache/fop/pdf/FileIDGenerator.java
blob: 00aad44265798336ceb06ff6d149cc896bf2a73b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/*
 * 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.
 */

/* $Id$ */

package org.apache.fop.pdf;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Random;

/**
 * A class to generate the File Identifier of a PDF document (the ID entry of the file
 * trailer dictionary).
 */
abstract class FileIDGenerator {

    abstract byte[] getOriginalFileID();

    abstract byte[] getUpdatedFileID();

    private static final class RandomFileIDGenerator extends FileIDGenerator {

        private byte[] fileID;

        private RandomFileIDGenerator() {
            Random random = new Random();
            fileID = new byte[16];
            random.nextBytes(fileID);
        }

        @Override
        byte[] getOriginalFileID() {
            return fileID;
        }

        @Override
        byte[] getUpdatedFileID() {
            return fileID;
        }

    }

    private static final class DigestFileIDGenerator extends FileIDGenerator {

        private byte[] fileID;

        private final PDFDocument document;

        private final MessageDigest digest;

        DigestFileIDGenerator(PDFDocument document) throws NoSuchAlgorithmException {
            this.document = document;
            this.digest = MessageDigest.getInstance("MD5");
        }

        @Override
        byte[] getOriginalFileID() {
            if (fileID == null) {
                generateFileID();
            }
            return fileID;
        }

        @Override
        byte[] getUpdatedFileID() {
            return getOriginalFileID();
        }

        private void generateFileID() {
            DateFormat df = new SimpleDateFormat("yyyy'-'MM'-'dd'T'HH':'mm':'ss'.'SSS");
            digest.update(PDFDocument.encode(df.format(new Date())));
            // Ignoring the filename here for simplicity even though it's recommended
            // by the PDF spec
            digest.update(PDFDocument.encode(String.valueOf(document.getCurrentFileSize())));
            digest.update(document.getInfo().toPDF());
            fileID = digest.digest();
        }

    }

    /**
     * Use this method when the file ID is needed before the document is finalized. The
     * digest method recommended by the PDF Reference is based, among other things, on the
     * file size.
     *
     * @return an instance that generates a random sequence of bytes for the File
     * Identifier
     */
    static FileIDGenerator getRandomFileIDGenerator() {
        return new RandomFileIDGenerator();
    }

    /**
     * Returns an instance that generates a file ID using the digest method recommended by
     * the PDF Reference. To properly follow the Reference, the size of the document must
     * no longer change after this method is called.
     *
     * @param document the document whose File Identifier must be generated
     * @return the generator
     * @throws NoSuchAlgorithmException if the MD5 Digest algorithm is not available
     */
    static FileIDGenerator getDigestFileIDGenerator(PDFDocument document)
            throws NoSuchAlgorithmException {
        return new DigestFileIDGenerator(document);
    }
}