objc/objc-runtime.h does not exist on all PLATFORM(MAC)
[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 #if __MAC_OS_X_VERSION_MIN_REQUIRED == 1050
34 @interface PDFDocument (Internal)
35 - (CGPDFDocumentRef)documentRef;
36 @end
37 #endif
38
39 static void appendValuesInPDFNameSubtreeToVector(CGPDFDictionaryRef subtree, Vector<CGPDFObjectRef>& values)
40 {
41     CGPDFArrayRef names;
42     if (CGPDFDictionaryGetArray(subtree, "Names", &names)) {
43         size_t nameCount = CGPDFArrayGetCount(names) / 2;
44         for (size_t i = 0; i < nameCount; ++i) {
45             CGPDFObjectRef object;
46             CGPDFArrayGetObject(names, 2 * i + 1, &object);
47             values.append(object);
48         }
49         return;
50     }
51
52     CGPDFArrayRef kids;
53     if (!CGPDFDictionaryGetArray(subtree, "Kids", &kids))
54         return;
55
56     size_t kidCount = CGPDFArrayGetCount(kids);
57     for (size_t i = 0; i < kidCount; ++i) {
58         CGPDFDictionaryRef kid;
59         if (!CGPDFArrayGetDictionary(kids, i, &kid))
60             continue;
61         appendValuesInPDFNameSubtreeToVector(kid, values);
62     }
63 }
64
65 static void getAllValuesInPDFNameTree(CGPDFDictionaryRef tree, Vector<CGPDFObjectRef>& allValues)
66 {
67     appendValuesInPDFNameSubtreeToVector(tree, allValues);
68 }
69
70 NSArray *allScriptsInPDFDocument(PDFDocument *document)
71 {
72     NSMutableArray *scripts = [NSMutableArray array];
73     CGPDFDocumentRef pdfDocument = [document documentRef];
74     if (!pdfDocument)
75         return scripts;
76
77     CGPDFDictionaryRef pdfCatalog = CGPDFDocumentGetCatalog(pdfDocument);
78     if (!pdfCatalog)
79         return scripts;
80
81     // Get the dictionary of all document-level name trees.
82     CGPDFDictionaryRef namesDictionary;
83     if (!CGPDFDictionaryGetDictionary(pdfCatalog, "Names", &namesDictionary))
84         return scripts;
85
86     // Get the document-level "JavaScript" name tree.
87     CGPDFDictionaryRef javaScriptNameTree;
88     if (!CGPDFDictionaryGetDictionary(namesDictionary, "JavaScript", &javaScriptNameTree))
89         return scripts;
90
91     // The names are arbitrary. We are only interested in the values.
92     Vector<CGPDFObjectRef> objects;
93     getAllValuesInPDFNameTree(javaScriptNameTree, objects);
94     size_t objectCount = objects.size();
95
96     for (size_t i = 0; i < objectCount; ++i) {
97         CGPDFDictionaryRef javaScriptAction;
98         if (!CGPDFObjectGetValue(reinterpret_cast<CGPDFObjectRef>(objects[i]), kCGPDFObjectTypeDictionary, &javaScriptAction))
99             continue;
100
101         // A JavaScript action must have an action type of "JavaScript".
102         const char* actionType;
103         if (!CGPDFDictionaryGetName(javaScriptAction, "S", &actionType) || strcmp(actionType, "JavaScript"))
104             continue;
105
106         const UInt8* bytes = 0;
107         CFIndex length;
108         CGPDFStreamRef stream;
109         CGPDFStringRef string;
110         RetainPtr<CFDataRef> data;
111         if (CGPDFDictionaryGetStream(javaScriptAction, "JS", &stream)) {
112             CGPDFDataFormat format;
113             data.adoptCF(CGPDFStreamCopyData(stream, &format));
114             if (!data)
115                 continue;
116             bytes = CFDataGetBytePtr(data.get());
117             length = CFDataGetLength(data.get());
118         } else if (CGPDFDictionaryGetString(javaScriptAction, "JS", &string)) {
119             bytes = CGPDFStringGetBytePtr(string);
120             length = CGPDFStringGetLength(string);
121         }
122         if (!bytes)
123             continue;
124
125         NSStringEncoding encoding = (length > 1 && bytes[0] == 0xFE && bytes[1] == 0xFF) ? NSUnicodeStringEncoding : NSUTF8StringEncoding;
126         NSString *script = [[NSString alloc] initWithBytes:bytes length:length encoding:encoding];
127         if (!script)
128             continue;
129
130         [scripts addObject:script];
131         [script release];
132     }
133
134     return scripts;
135 }