Move from constructor and member function adoptCF/NS to free function adoptCF/NS.
[WebKit-https.git] / Source / WebKit / mac / WebView / WebPDFDocumentExtras.mm
1 /*
2  * Copyright (C) 2009 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #import "WebPDFDocumentExtras.h"
27
28 #import "WebTypesInternal.h"
29 #import <wtf/Vector.h>
30 #import <wtf/RetainPtr.h>
31 #import <PDFKit/PDFDocument.h>
32
33 static void appendValuesInPDFNameSubtreeToVector(CGPDFDictionaryRef subtree, Vector<CGPDFObjectRef>& values)
34 {
35     CGPDFArrayRef names;
36     if (CGPDFDictionaryGetArray(subtree, "Names", &names)) {
37         size_t nameCount = CGPDFArrayGetCount(names) / 2;
38         for (size_t i = 0; i < nameCount; ++i) {
39             CGPDFObjectRef object;
40             CGPDFArrayGetObject(names, 2 * i + 1, &object);
41             values.append(object);
42         }
43         return;
44     }
45
46     CGPDFArrayRef kids;
47     if (!CGPDFDictionaryGetArray(subtree, "Kids", &kids))
48         return;
49
50     size_t kidCount = CGPDFArrayGetCount(kids);
51     for (size_t i = 0; i < kidCount; ++i) {
52         CGPDFDictionaryRef kid;
53         if (!CGPDFArrayGetDictionary(kids, i, &kid))
54             continue;
55         appendValuesInPDFNameSubtreeToVector(kid, values);
56     }
57 }
58
59 static void getAllValuesInPDFNameTree(CGPDFDictionaryRef tree, Vector<CGPDFObjectRef>& allValues)
60 {
61     appendValuesInPDFNameSubtreeToVector(tree, allValues);
62 }
63
64 NSArray *allScriptsInPDFDocument(PDFDocument *document)
65 {
66     NSMutableArray *scripts = [NSMutableArray array];
67     CGPDFDocumentRef pdfDocument = [document documentRef];
68     if (!pdfDocument)
69         return scripts;
70
71     CGPDFDictionaryRef pdfCatalog = CGPDFDocumentGetCatalog(pdfDocument);
72     if (!pdfCatalog)
73         return scripts;
74
75     // Get the dictionary of all document-level name trees.
76     CGPDFDictionaryRef namesDictionary;
77     if (!CGPDFDictionaryGetDictionary(pdfCatalog, "Names", &namesDictionary))
78         return scripts;
79
80     // Get the document-level "JavaScript" name tree.
81     CGPDFDictionaryRef javaScriptNameTree;
82     if (!CGPDFDictionaryGetDictionary(namesDictionary, "JavaScript", &javaScriptNameTree))
83         return scripts;
84
85     // The names are arbitrary. We are only interested in the values.
86     Vector<CGPDFObjectRef> objects;
87     getAllValuesInPDFNameTree(javaScriptNameTree, objects);
88     size_t objectCount = objects.size();
89
90     for (size_t i = 0; i < objectCount; ++i) {
91         CGPDFDictionaryRef javaScriptAction;
92         if (!CGPDFObjectGetValue(reinterpret_cast<CGPDFObjectRef>(objects[i]), kCGPDFObjectTypeDictionary, &javaScriptAction))
93             continue;
94
95         // A JavaScript action must have an action type of "JavaScript".
96         const char* actionType;
97         if (!CGPDFDictionaryGetName(javaScriptAction, "S", &actionType) || strcmp(actionType, "JavaScript"))
98             continue;
99
100         const UInt8* bytes = 0;
101         CFIndex length;
102         CGPDFStreamRef stream;
103         CGPDFStringRef string;
104         RetainPtr<CFDataRef> data;
105         if (CGPDFDictionaryGetStream(javaScriptAction, "JS", &stream)) {
106             CGPDFDataFormat format;
107             data = adoptCF(CGPDFStreamCopyData(stream, &format));
108             if (!data)
109                 continue;
110             bytes = CFDataGetBytePtr(data.get());
111             length = CFDataGetLength(data.get());
112         } else if (CGPDFDictionaryGetString(javaScriptAction, "JS", &string)) {
113             bytes = CGPDFStringGetBytePtr(string);
114             length = CGPDFStringGetLength(string);
115         }
116         if (!bytes)
117             continue;
118
119         NSStringEncoding encoding = (length > 1 && bytes[0] == 0xFE && bytes[1] == 0xFF) ? NSUnicodeStringEncoding : NSUTF8StringEncoding;
120         NSString *script = [[NSString alloc] initWithBytes:bytes length:length encoding:encoding];
121         if (!script)
122             continue;
123
124         [scripts addObject:script];
125         [script release];
126     }
127
128     return scripts;
129 }