automake/autoconf fluctuations
[swftools.git] / pdf2swf / xpdf / Parser.cc
1 //========================================================================
2 //
3 // Parser.cc
4 //
5 // Copyright 1996-2003 Glyph & Cog, LLC
6 //
7 //========================================================================
8
9 #include <aconf.h>
10
11 #ifdef USE_GCC_PRAGMAS
12 #pragma implementation
13 #endif
14
15 #include <stddef.h>
16 #include "Object.h"
17 #include "Array.h"
18 #include "Dict.h"
19 #include "Parser.h"
20 #include "XRef.h"
21 #include "Error.h"
22 #ifndef NO_DECRYPTION
23 #include "Decrypt.h"
24 #endif
25
26 Parser::Parser(XRef *xrefA, Lexer *lexerA) {
27   xref = xrefA;
28   lexer = lexerA;
29   inlineImg = 0;
30   lexer->getObj(&buf1);
31   lexer->getObj(&buf2);
32 }
33
34 Parser::~Parser() {
35   buf1.free();
36   buf2.free();
37   delete lexer;
38 }
39
40 #ifndef NO_DECRYPTION
41 Object *Parser::getObj(Object *obj,
42                        Guchar *fileKey, int keyLength,
43                        int objNum, int objGen) {
44 #else
45 Object *Parser::getObj(Object *obj) {
46 #endif
47   char *key;
48   Stream *str;
49   Object obj2;
50   int num;
51 #ifndef NO_DECRYPTION
52   Decrypt *decrypt;
53   GString *s;
54   char *p;
55   int i;
56 #endif
57
58   // refill buffer after inline image data
59   if (inlineImg == 2) {
60     buf1.free();
61     buf2.free();
62     lexer->getObj(&buf1);
63     lexer->getObj(&buf2);
64     inlineImg = 0;
65   }
66
67   // array
68   if (buf1.isCmd("[")) {
69     shift();
70     obj->initArray(xref);
71     while (!buf1.isCmd("]") && !buf1.isEOF())
72 #ifndef NO_DECRYPTION
73       obj->arrayAdd(getObj(&obj2, fileKey, keyLength, objNum, objGen));
74 #else
75       obj->arrayAdd(getObj(&obj2));
76 #endif
77     if (buf1.isEOF())
78       error(getPos(), "End of file inside array");
79     shift();
80
81   // dictionary or stream
82   } else if (buf1.isCmd("<<")) {
83     shift();
84     obj->initDict(xref);
85     while (!buf1.isCmd(">>") && !buf1.isEOF()) {
86       if (!buf1.isName()) {
87         error(getPos(), "Dictionary key must be a name object");
88         shift();
89       } else {
90         key = copyString(buf1.getName());
91         shift();
92         if (buf1.isEOF() || buf1.isError()) {
93           gfree(key);
94           break;
95         }
96 #ifndef NO_DECRYPTION
97         obj->dictAdd(key, getObj(&obj2, fileKey, keyLength, objNum, objGen));
98 #else
99         obj->dictAdd(key, getObj(&obj2));
100 #endif
101       }
102     }
103     if (buf1.isEOF())
104       error(getPos(), "End of file inside dictionary");
105     if (buf2.isCmd("stream")) {
106       if ((str = makeStream(obj))) {
107         obj->initStream(str);
108 #ifndef NO_DECRYPTION
109         if (fileKey) {
110           str->getBaseStream()->doDecryption(fileKey, keyLength,
111                                              objNum, objGen);
112         }
113 #endif
114       } else {
115         obj->free();
116         obj->initError();
117       }
118     } else {
119       shift();
120     }
121
122   // indirect reference or integer
123   } else if (buf1.isInt()) {
124     num = buf1.getInt();
125     shift();
126     if (buf1.isInt() && buf2.isCmd("R")) {
127       obj->initRef(num, buf1.getInt());
128       shift();
129       shift();
130     } else {
131       obj->initInt(num);
132     }
133
134 #ifndef NO_DECRYPTION
135   // string
136   } else if (buf1.isString() && fileKey) {
137     buf1.copy(obj);
138     s = obj->getString();
139     decrypt = new Decrypt(fileKey, keyLength, objNum, objGen);
140     for (i = 0, p = obj->getString()->getCString();
141          i < s->getLength();
142          ++i, ++p) {
143       *p = decrypt->decryptByte(*p);
144     }
145     delete decrypt;
146     shift();
147 #endif
148
149   // simple object
150   } else {
151     buf1.copy(obj);
152     shift();
153   }
154
155   return obj;
156 }
157
158 Stream *Parser::makeStream(Object *dict) {
159   Object obj;
160   Stream *str;
161   Guint pos, endPos, length;
162
163   // get stream start position
164   lexer->skipToNextLine();
165   pos = lexer->getPos();
166
167   // get length
168   dict->dictLookup("Length", &obj);
169   if (obj.isInt()) {
170     length = (Guint)obj.getInt();
171     obj.free();
172   } else {
173     error(getPos(), "Bad 'Length' attribute in stream");
174     obj.free();
175     return NULL;
176   }
177
178   // check for length in damaged file
179   if (xref && xref->getStreamEnd(pos, &endPos)) {
180     length = endPos - pos;
181   }
182
183   // in badly damaged PDF files, we can run off the end of the input
184   // stream immediately after the "stream" token
185   if (!lexer->getStream()) {
186     return NULL;
187   }
188
189   // make base stream
190   str = lexer->getStream()->getBaseStream()->makeSubStream(pos, gTrue,
191                                                            length, dict);
192
193   // get filters
194   str = str->addFilters(dict);
195
196   // skip over stream data
197   lexer->setPos(pos + length);
198
199   // refill token buffers and check for 'endstream'
200   shift();  // kill '>>'
201   shift();  // kill 'stream'
202   if (buf1.isCmd("endstream")) {
203     shift();
204   } else {
205     error(getPos(), "Missing 'endstream'");
206     str->ignoreLength();
207   }
208
209   return str;
210 }
211
212 void Parser::shift() {
213   if (inlineImg > 0) {
214     if (inlineImg < 2) {
215       ++inlineImg;
216     } else {
217       // in a damaged content stream, if 'ID' shows up in the middle
218       // of a dictionary, we need to reset
219       inlineImg = 0;
220     }
221   } else if (buf2.isCmd("ID")) {
222     lexer->skipChar();          // skip char after 'ID' command
223     inlineImg = 1;
224   }
225   buf1.free();
226   buf1 = buf2;
227   if (inlineImg > 0)            // don't buffer inline image data
228     buf2.initNull();
229   else
230     lexer->getObj(&buf2);
231 }