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