Fix FX_BOOL compilation errors under windows
[pdfium.git] / core / src / fpdfapi / fpdf_edit / fpdf_edit_create.cpp
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "../../../include/fxcrt/fx_ext.h"
8 #include "../../../include/fpdfapi/fpdf_serial.h"
9 #include "editint.h"
10
11 #define PDF_OBJECTSTREAM_MAXLENGTH      (256 * 1024)
12 #define PDF_XREFSTREAM_MAXSIZE          10000
13 extern void FlateEncode(const uint8_t* src_buf, FX_DWORD src_data, uint8_t*& dest_buf, FX_DWORD& dest_size);
14 extern void FlateEncode(const uint8_t* src_buf, FX_DWORD src_size, int predictor, int Colors, int BitsPerComponent, int Columns,
15                         uint8_t*& dest_buf, FX_DWORD& dest_size);
16 extern FX_BOOL IsSignatureDict(const CPDF_Dictionary* pDict);
17 int32_t PDF_CreatorAppendObject(const CPDF_Object* pObj, CFX_FileBufferArchive *pFile, FX_FILESIZE& offset)
18 {
19     int32_t len = 0;
20     if (pObj == NULL) {
21         if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
22             return -1;
23         }
24         offset += 5;
25         return 1;
26     }
27     switch (pObj->GetType()) {
28         case PDFOBJ_NULL:
29             if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
30                 return -1;
31             }
32             offset += 5;
33             break;
34         case PDFOBJ_BOOLEAN:
35         case PDFOBJ_NUMBER:
36             if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
37                 return -1;
38             }
39             if ((len = pFile->AppendString(pObj->GetString())) < 0) {
40                 return -1;
41             }
42             offset += len + 1;
43             break;
44         case PDFOBJ_STRING: {
45                 CFX_ByteString str = pObj->GetString();
46                 FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
47                 if ((len = pFile->AppendString(PDF_EncodeString(str, bHex))) < 0) {
48                     return -1;
49                 }
50                 offset += len;
51                 break;
52             }
53         case PDFOBJ_NAME: {
54                 if (pFile->AppendString(FX_BSTRC("/")) < 0) {
55                     return -1;
56                 }
57                 CFX_ByteString str = pObj->GetString();
58                 if ((len = pFile->AppendString(PDF_NameEncode(str))) < 0) {
59                     return -1;
60                 }
61                 offset += len + 1;
62                 break;
63             }
64         case PDFOBJ_REFERENCE: {
65                 if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
66                     return -1;
67                 }
68                 CPDF_Reference* p = (CPDF_Reference*)pObj;
69                 if ((len = pFile->AppendDWord(p->GetRefObjNum())) < 0) {
70                     return -1;
71                 }
72                 if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
73                     return -1;
74                 }
75                 offset += len + 6;
76                 break;
77             }
78         case PDFOBJ_ARRAY: {
79                 if (pFile->AppendString(FX_BSTRC("[")) < 0) {
80                     return -1;
81                 }
82                 offset += 1;
83                 CPDF_Array* p = (CPDF_Array*)pObj;
84                 for (FX_DWORD i = 0; i < p->GetCount(); i ++) {
85                     CPDF_Object* pElement = p->GetElement(i);
86                     if (pElement->GetObjNum()) {
87                         if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
88                             return -1;
89                         }
90                         if ((len = pFile->AppendDWord(pElement->GetObjNum())) < 0) {
91                             return -1;
92                         }
93                         if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
94                             return -1;
95                         }
96                         offset += len + 5;
97                     } else {
98                         if (PDF_CreatorAppendObject(pElement, pFile, offset) < 0) {
99                             return -1;
100                         }
101                     }
102                 }
103                 if (pFile->AppendString(FX_BSTRC("]")) < 0) {
104                     return -1;
105                 }
106                 offset += 1;
107                 break;
108             }
109         case PDFOBJ_DICTIONARY: {
110                 if (pFile->AppendString(FX_BSTRC("<<")) < 0) {
111                     return -1;
112                 }
113                 offset += 2;
114                 CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
115                 FX_POSITION pos = p->GetStartPos();
116                 while (pos) {
117                     CFX_ByteString key;
118                     CPDF_Object* pValue = p->GetNextElement(pos, key);
119                     if (pFile->AppendString(FX_BSTRC("/")) < 0) {
120                         return -1;
121                     }
122                     if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
123                         return -1;
124                     }
125                     offset += len + 1;
126                     if (pValue->GetObjNum()) {
127                         if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
128                             return -1;
129                         }
130                         if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
131                             return -1;
132                         }
133                         if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
134                             return -1;
135                         }
136                         offset += len + 5;
137                     } else {
138                         if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
139                             return -1;
140                         }
141                     }
142                 }
143                 if (pFile->AppendString(FX_BSTRC(">>")) < 0) {
144                     return -1;
145                 }
146                 offset += 2;
147                 break;
148             }
149         case PDFOBJ_STREAM: {
150                 CPDF_Stream* p = (CPDF_Stream*)pObj;
151                 if (PDF_CreatorAppendObject(p->GetDict(), pFile, offset) < 0) {
152                     return -1;
153                 }
154                 if (pFile->AppendString(FX_BSTRC("stream\r\n")) < 0) {
155                     return -1;
156                 }
157                 offset += 8;
158                 CPDF_StreamAcc acc;
159                 acc.LoadAllData(p, TRUE);
160                 if (pFile->AppendBlock(acc.GetData(), acc.GetSize()) < 0) {
161                     return -1;
162                 }
163                 offset += acc.GetSize();
164                 if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
165                     return -1;
166                 }
167                 offset += len;
168                 break;
169             }
170         default:
171             ASSERT(FALSE);
172             break;
173     }
174     return 1;
175 }
176 int32_t PDF_CreatorWriteTrailer(CPDF_Document* pDocument, CFX_FileBufferArchive* pFile, CPDF_Array* pIDArray, FX_BOOL bCompress)
177 {
178     FX_FILESIZE offset = 0;
179     int32_t len = 0;
180     FXSYS_assert(pDocument && pFile);
181     CPDF_Parser *pParser = (CPDF_Parser*)pDocument->GetParser();
182     if (pParser) {
183         CPDF_Dictionary* p = pParser->GetTrailer();
184         FX_POSITION pos = p->GetStartPos();
185         while (pos) {
186             CFX_ByteString key;
187             CPDF_Object* pValue = p->GetNextElement(pos, key);
188             if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") || key == FX_BSTRC("Filter") ||
189                     key == FX_BSTRC("Index") || key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
190                     key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") || key == FX_BSTRC("Type") || key == FX_BSTRC("ID")) {
191                 continue;
192             }
193             if (bCompress && key == FX_BSTRC("DecodeParms")) {
194                 continue;
195             }
196             if (pFile->AppendString((FX_BSTRC("/"))) < 0) {
197                 return -1;
198             }
199             if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
200                 return -1;
201             }
202             offset += len + 1;
203             if (pValue->GetObjNum()) {
204                 if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
205                     return -1;
206                 }
207                 if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
208                     return -1;
209                 }
210                 if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
211                     return -1;
212                 }
213                 offset += len + 6;
214             } else {
215                 if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
216                     return -1;
217                 }
218             }
219         }
220         if (pIDArray) {
221             if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
222                 return -1;
223             }
224             offset += 3;
225             if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
226                 return -1;
227             }
228         }
229         return offset;
230     }
231     if (pFile->AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
232         return -1;
233     }
234     if ((len = pFile->AppendDWord(pDocument->GetRoot()->GetObjNum())) < 0) {
235         return -1;
236     }
237     if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
238         return -1;
239     }
240     offset += len + 14;
241     if (pDocument->GetInfo()) {
242         if (pFile->AppendString(FX_BSTRC("/Info ")) < 0) {
243             return -1;
244         }
245         if ((len = pFile->AppendDWord(pDocument->GetInfo()->GetObjNum())) < 0) {
246             return -1;
247         }
248         if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
249             return -1;
250         }
251         offset += len + 12;
252     }
253     if (pIDArray) {
254         if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
255             return -1;
256         }
257         offset += 3;
258         if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
259             return -1;
260         }
261     }
262     return offset;
263 }
264 int32_t PDF_CreatorWriteEncrypt(const CPDF_Dictionary* pEncryptDict, FX_DWORD dwObjNum, CFX_FileBufferArchive *pFile)
265 {
266     if (!pEncryptDict) {
267         return 0;
268     }
269     FXSYS_assert(pFile);
270     FX_FILESIZE offset = 0;
271     int32_t len = 0;
272     if (pFile->AppendString(FX_BSTRC("/Encrypt")) < 0) {
273         return -1;
274     }
275     offset += 8;
276     if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
277         return -1;
278     }
279     if ((len = pFile->AppendDWord(dwObjNum)) < 0) {
280         return -1;
281     }
282     if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
283         return -1;
284     }
285     offset += len + 6;
286     return offset;
287 }
288 FX_BOOL PDF_GenerateFileID(FX_DWORD dwSeed1, FX_DWORD dwSeed2, FX_DWORD* pBuffer)
289 {
290     if (!pBuffer) {
291         return FALSE;
292     }
293     void* pContext = FX_Random_MT_Start(dwSeed1);
294     int32_t i = 0;
295     for (i = 0; i < 2; i++) {
296         *pBuffer++ = FX_Random_MT_Generate(pContext);
297     }
298     FX_Random_MT_Close(pContext);
299     pContext = FX_Random_MT_Start(dwSeed2);
300     for (i = 0; i < 2; i++) {
301         *pBuffer++ = FX_Random_MT_Generate(pContext);
302     }
303     FX_Random_MT_Close(pContext);
304     return TRUE;
305 }
306 class CPDF_FlateEncoder
307 {
308 public:
309     CPDF_FlateEncoder();
310     ~CPDF_FlateEncoder();
311     FX_BOOL             Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode);
312     FX_BOOL             Initialize(const uint8_t* pBuffer, FX_DWORD size, FX_BOOL bFlateEncode, FX_BOOL bXRefStream = FALSE);
313     void                CloneDict();
314     uint8_t*                    m_pData;
315     FX_DWORD                    m_dwSize;
316     CPDF_Dictionary*    m_pDict;
317     FX_BOOL                             m_bCloned;
318     FX_BOOL                             m_bNewData;
319     CPDF_StreamAcc              m_Acc;
320 };
321 CPDF_FlateEncoder::CPDF_FlateEncoder()
322 {
323     m_pData = NULL;
324     m_dwSize = 0;
325     m_pDict = NULL;
326     m_bCloned = FALSE;
327     m_bNewData = FALSE;
328 }
329 void CPDF_FlateEncoder::CloneDict()
330 {
331     if (!m_bCloned) {
332         m_pDict = (CPDF_Dictionary*)m_pDict->Clone();
333         m_bCloned = TRUE;
334     }
335 }
336 FX_BOOL CPDF_FlateEncoder::Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode)
337 {
338     m_Acc.LoadAllData(pStream, TRUE);
339     if ((pStream && pStream->GetDict() && pStream->GetDict()->KeyExist("Filter")) || !bFlateEncode) {
340         if (pStream->GetDict()->KeyExist("Filter") && !bFlateEncode) {
341             CPDF_StreamAcc destAcc;
342             destAcc.LoadAllData(pStream);
343             m_dwSize = destAcc.GetSize();
344             m_pData = (uint8_t*)destAcc.DetachData();
345             m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
346             m_pDict->RemoveAt(FX_BSTRC("Filter"));
347             m_bNewData = TRUE;
348             m_bCloned = TRUE;
349         } else {
350             m_pData = (uint8_t*)m_Acc.GetData();
351             m_dwSize = m_Acc.GetSize();
352             m_pDict = pStream->GetDict();
353         }
354         return TRUE;
355     }
356     m_pData = NULL;
357     m_dwSize = 0;
358     m_bNewData = TRUE;
359     m_bCloned = TRUE;
360     ::FlateEncode(m_Acc.GetData(), m_Acc.GetSize(), m_pData, m_dwSize);
361     m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
362     m_pDict->SetAtInteger("Length", m_dwSize);
363     m_pDict->SetAtName("Filter", "FlateDecode");
364     m_pDict->RemoveAt("DecodeParms");
365     return TRUE;
366 }
367 FX_BOOL CPDF_FlateEncoder::Initialize(const uint8_t* pBuffer, FX_DWORD size, FX_BOOL bFlateEncode, FX_BOOL bXRefStream)
368 {
369     if (!bFlateEncode) {
370         m_pData = (uint8_t*)pBuffer;
371         m_dwSize = size;
372         return TRUE;
373     }
374     m_bNewData = TRUE;
375     if (bXRefStream) {
376         ::FlateEncode(pBuffer, size, 12, 1, 8, 7, m_pData, m_dwSize);
377     } else {
378         ::FlateEncode(pBuffer, size, m_pData, m_dwSize);
379     }
380     return TRUE;
381 }
382 CPDF_FlateEncoder::~CPDF_FlateEncoder()
383 {
384     if (m_bCloned && m_pDict) {
385         m_pDict->Release();
386     }
387     if (m_bNewData && m_pData) {
388         FX_Free(m_pData);
389     }
390 }
391 class CPDF_Encryptor
392 {
393 public:
394     CPDF_Encryptor();
395     ~CPDF_Encryptor();
396     FX_BOOL             Initialize(CPDF_CryptoHandler* pHandler, int objnum, uint8_t* src_data, FX_DWORD src_size);
397     uint8_t*                    m_pData;
398     FX_DWORD                    m_dwSize;
399     FX_BOOL                             m_bNewBuf;
400 };
401 CPDF_Encryptor::CPDF_Encryptor()
402 {
403     m_pData = NULL;
404     m_dwSize = 0;
405     m_bNewBuf = FALSE;
406 }
407 FX_BOOL CPDF_Encryptor::Initialize(CPDF_CryptoHandler* pHandler, int objnum, uint8_t* src_data, FX_DWORD src_size)
408 {
409     if (src_size == 0) {
410         return TRUE;
411     }
412     if (pHandler == NULL) {
413         m_pData = (uint8_t*)src_data;
414         m_dwSize = src_size;
415         m_bNewBuf = FALSE;
416         return TRUE;
417     }
418     m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size);
419     m_pData = FX_Alloc(uint8_t, m_dwSize);
420     pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize);
421     m_bNewBuf = TRUE;
422     return TRUE;
423 }
424 CPDF_Encryptor::~CPDF_Encryptor()
425 {
426     if (m_bNewBuf) {
427         FX_Free(m_pData);
428     }
429 }
430 CPDF_ObjectStream::CPDF_ObjectStream()
431     : m_dwObjNum(0)
432     , m_index(0)
433 {
434 }
435 FX_BOOL CPDF_ObjectStream::Start()
436 {
437     m_ObjNumArray.RemoveAll();
438     m_OffsetArray.RemoveAll();
439     m_Buffer.Clear();
440     m_dwObjNum = 0;
441     m_index = 0;
442     return TRUE;
443 }
444 int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, const CPDF_Object *pObj)
445 {
446     m_ObjNumArray.Add(dwObjNum);
447     m_OffsetArray.Add(m_Buffer.GetLength());
448     m_Buffer << pObj;
449     return 1;
450 }
451 int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, const uint8_t* pBuffer, FX_DWORD dwSize)
452 {
453     m_ObjNumArray.Add(dwObjNum);
454     m_OffsetArray.Add(m_Buffer.GetLength());
455     m_Buffer.AppendBlock(pBuffer, dwSize);
456     return 1;
457 }
458 FX_FILESIZE CPDF_ObjectStream::End(CPDF_Creator* pCreator)
459 {
460     FXSYS_assert(pCreator);
461     if (m_ObjNumArray.GetSize() == 0) {
462         return 0;
463     }
464     CFX_FileBufferArchive *pFile = &pCreator->m_File;
465     CPDF_CryptoHandler *pHandler = pCreator->m_pCryptoHandler;
466     FX_FILESIZE ObjOffset = pCreator->m_Offset;
467     if (!m_dwObjNum) {
468         m_dwObjNum = ++pCreator->m_dwLastObjNum;
469     }
470     CFX_ByteTextBuf tempBuffer;
471     int32_t iCount = m_ObjNumArray.GetSize();
472     for (int32_t i = 0; i < iCount; i++) {
473         tempBuffer << m_ObjNumArray.ElementAt(i) << FX_BSTRC(" ") << m_OffsetArray.ElementAt(i) << FX_BSTRC(" ");
474     }
475     FX_FILESIZE &offset = pCreator->m_Offset;
476     int32_t len = pFile->AppendDWord(m_dwObjNum);
477     if (len < 0) {
478         return -1;
479     }
480     offset += len;
481     if ((len = pFile->AppendString(FX_BSTRC(" 0 obj\r\n<</Type /ObjStm /N "))) < 0) {
482         return -1;
483     }
484     offset += len;
485     if ((len = pFile->AppendDWord((FX_DWORD)iCount)) < 0) {
486         return -1;
487     }
488     offset += len;
489     if (pFile->AppendString(FX_BSTRC("/First ")) < 0) {
490         return -1;
491     }
492     if ((len = pFile->AppendDWord((FX_DWORD)tempBuffer.GetLength())) < 0) {
493         return -1;
494     }
495     if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
496         return -1;
497     }
498     offset += len + 15;
499     if (!pCreator->m_bCompress && !pHandler) {
500         if ((len = pFile->AppendDWord((FX_DWORD)(tempBuffer.GetLength() + m_Buffer.GetLength()))) < 0) {
501             return -1;
502         }
503         offset += len;
504         if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
505             return -1;
506         }
507         if (pFile->AppendBlock(tempBuffer.GetBuffer(), tempBuffer.GetLength()) < 0) {
508             return -1;
509         }
510         if (pFile->AppendBlock(m_Buffer.GetBuffer(), m_Buffer.GetLength()) < 0) {
511             return -1;
512         }
513         offset += len + tempBuffer.GetLength() + m_Buffer.GetLength();
514     } else {
515         tempBuffer << m_Buffer;
516         CPDF_FlateEncoder encoder;
517         encoder.Initialize(tempBuffer.GetBuffer(), tempBuffer.GetLength(), pCreator->m_bCompress);
518         CPDF_Encryptor encryptor;
519         encryptor.Initialize(pHandler, m_dwObjNum, encoder.m_pData, encoder.m_dwSize);
520         if ((len = pFile->AppendDWord(encryptor.m_dwSize)) < 0) {
521             return -1;
522         }
523         offset += len;
524         if (pCreator->m_bCompress) {
525             if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
526                 return -1;
527             }
528             offset += 20;
529         }
530         if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
531             return -1;
532         }
533         if (pFile->AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
534             return -1;
535         }
536         offset += len + encryptor.m_dwSize;
537     }
538     if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) < 0) {
539         return -1;
540     }
541     offset += len;
542     return ObjOffset;
543 }
544 CPDF_XRefStream::CPDF_XRefStream()
545     : m_PrevOffset(0)
546     , m_dwTempObjNum(0)
547     , m_iSeg(0)
548 {
549 }
550 FX_BOOL CPDF_XRefStream::Start()
551 {
552     m_IndexArray.RemoveAll();
553     m_Buffer.Clear();
554     m_iSeg = 0;
555     return TRUE;
556 }
557 int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, const CPDF_Object *pObj, CPDF_Creator *pCreator)
558 {
559     if (!pCreator) {
560         return 0;
561     }
562     m_ObjStream.CompressIndirectObject(dwObjNum, pObj);
563     if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
564             m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
565         return 1;
566     }
567     return EndObjectStream(pCreator);
568 }
569 int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, const uint8_t* pBuffer, FX_DWORD dwSize, CPDF_Creator *pCreator)
570 {
571     if (!pCreator) {
572         return 0;
573     }
574     m_ObjStream.CompressIndirectObject(dwObjNum, pBuffer, dwSize);
575     if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
576             m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
577         return 1;
578     }
579     return EndObjectStream(pCreator);
580 }
581 static void _AppendIndex0(CFX_ByteTextBuf& buffer, FX_BOOL bFirstObject =  TRUE)
582 {
583     buffer.AppendByte(0);
584     buffer.AppendByte(0);
585     buffer.AppendByte(0);
586     buffer.AppendByte(0);
587     buffer.AppendByte(0);
588     if (bFirstObject) {
589         buffer.AppendByte(0xFF);
590         buffer.AppendByte(0xFF);
591     } else {
592         buffer.AppendByte(0);
593         buffer.AppendByte(0);
594     }
595 }
596 static void _AppendIndex1(CFX_ByteTextBuf& buffer, FX_FILESIZE offset)
597 {
598     buffer.AppendByte(1);
599     buffer.AppendByte(FX_GETBYTEOFFSET24(offset));
600     buffer.AppendByte(FX_GETBYTEOFFSET16(offset));
601     buffer.AppendByte(FX_GETBYTEOFFSET8(offset));
602     buffer.AppendByte(FX_GETBYTEOFFSET0(offset));
603     buffer.AppendByte(0);
604     buffer.AppendByte(0);
605 }
606 static void _AppendIndex2(CFX_ByteTextBuf& buffer, FX_DWORD objnum, int32_t index)
607 {
608     buffer.AppendByte(2);
609     buffer.AppendByte(FX_GETBYTEOFFSET24(objnum));
610     buffer.AppendByte(FX_GETBYTEOFFSET16(objnum));
611     buffer.AppendByte(FX_GETBYTEOFFSET8(objnum));
612     buffer.AppendByte(FX_GETBYTEOFFSET0(objnum));
613     buffer.AppendByte(FX_GETBYTEOFFSET8(index));
614     buffer.AppendByte(FX_GETBYTEOFFSET0(index));
615 }
616 int32_t CPDF_XRefStream::EndObjectStream(CPDF_Creator *pCreator, FX_BOOL bEOF)
617 {
618     FX_FILESIZE objOffset = 0;
619     if (bEOF) {
620         objOffset = m_ObjStream.End(pCreator);
621         if (objOffset < 0) {
622             return -1;
623         }
624     }
625     FX_DWORD &dwObjStmNum = m_ObjStream.m_dwObjNum;
626     if (!dwObjStmNum) {
627         dwObjStmNum = ++pCreator->m_dwLastObjNum;
628     }
629     int32_t iSize = m_ObjStream.m_ObjNumArray.GetSize();
630     int32_t iSeg = m_IndexArray.GetSize() / 2;
631     if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
632         if (m_dwTempObjNum == 0) {
633             _AppendIndex0(m_Buffer);
634             m_dwTempObjNum++;
635         }
636         FX_DWORD end_num = m_IndexArray.GetAt((iSeg - 1) * 2) + m_IndexArray.GetAt((iSeg - 1) * 2 + 1);
637         int index = 0;
638         for (; m_dwTempObjNum < end_num; m_dwTempObjNum++) {
639             FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
640             if (offset) {
641                 if (index >= iSize || m_dwTempObjNum != m_ObjStream.m_ObjNumArray[index]) {
642                     _AppendIndex1(m_Buffer, *offset);
643                 } else {
644                     _AppendIndex2(m_Buffer, dwObjStmNum, index++);
645                 }
646             } else {
647                 _AppendIndex0(m_Buffer, FALSE);
648             }
649         }
650         if (iSize > 0 && bEOF) {
651             pCreator->m_ObjectOffset.Add(dwObjStmNum, 1);
652             pCreator->m_ObjectSize.Add(dwObjStmNum, 1);
653             pCreator->m_ObjectOffset[dwObjStmNum] = objOffset;
654         }
655         m_iSeg = iSeg;
656         if (bEOF) {
657             m_ObjStream.Start();
658         }
659         return 1;
660     }
661     int32_t &j = m_ObjStream.m_index;
662     for (int i = m_iSeg; i < iSeg; i++) {
663         FX_DWORD start = m_IndexArray.ElementAt(i * 2);
664         FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
665         for (FX_DWORD m = start; m < end; m++) {
666             if (j >= iSize || m != m_ObjStream.m_ObjNumArray.ElementAt(j)) {
667                 _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[m]);
668             } else {
669                 _AppendIndex2(m_Buffer, dwObjStmNum, j++);
670             }
671         }
672     }
673     if (iSize > 0 && bEOF) {
674         _AppendIndex1(m_Buffer, objOffset);
675         m_IndexArray.Add(dwObjStmNum);
676         m_IndexArray.Add(1);
677         iSeg += 1;
678     }
679     m_iSeg = iSeg;
680     if (bEOF) {
681         m_ObjStream.Start();
682     }
683     return 1;
684 }
685 FX_BOOL CPDF_XRefStream::GenerateXRefStream(CPDF_Creator* pCreator, FX_BOOL bEOF)
686 {
687     FX_FILESIZE offset_tmp = pCreator->m_Offset;
688     FX_DWORD objnum = ++pCreator->m_dwLastObjNum;
689     CFX_FileBufferArchive *pFile = &pCreator->m_File;
690     FX_BOOL bIncremental = (pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
691     if (bIncremental) {
692         AddObjectNumberToIndexArray(objnum);
693     } else {
694         for (; m_dwTempObjNum < pCreator->m_dwLastObjNum; m_dwTempObjNum++) {
695             FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
696             if (offset) {
697                 _AppendIndex1(m_Buffer, *offset);
698             } else {
699                 _AppendIndex0(m_Buffer, FALSE);
700             }
701         }
702     }
703     _AppendIndex1(m_Buffer, offset_tmp);
704     FX_FILESIZE &offset = pCreator->m_Offset;
705     int32_t len = pFile->AppendDWord(objnum);
706     if (len < 0) {
707         return FALSE;
708     }
709     offset += len;
710     if ((len = pFile->AppendString(FX_BSTRC(" 0 obj\r\n<</Type /XRef/W[1 4 2]/Index["))) < 0) {
711         return FALSE;
712     }
713     offset += len;
714     if (!bIncremental) {
715         if ((len = pFile->AppendDWord(0)) < 0) {
716             return FALSE;
717         }
718         if ((len = pFile->AppendString(FX_BSTRC(" "))) < 0) {
719             return FALSE;
720         }
721         offset += len + 1;
722         if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
723             return FALSE;
724         }
725         offset += len;
726     } else {
727         int32_t iSeg = m_IndexArray.GetSize() / 2;
728         for (int32_t i = 0; i < iSeg; i++) {
729             if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2))) < 0) {
730                 return FALSE;
731             }
732             if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
733                 return FALSE;
734             }
735             offset += len + 1;
736             if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2 + 1))) < 0) {
737                 return FALSE;
738             }
739             if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
740                 return FALSE;
741             }
742             offset += len + 1;
743         }
744     }
745     if (pFile->AppendString(FX_BSTRC("]/Size ")) < 0) {
746         return FALSE;
747     }
748     if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
749         return FALSE;
750     }
751     offset += len + 7;
752     if (m_PrevOffset > 0) {
753         if (pFile->AppendString(FX_BSTRC("/Prev ")) < 0) {
754             return FALSE;
755         }
756         FX_CHAR offset_buf[20];
757         FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
758         FXSYS_i64toa(m_PrevOffset, offset_buf, 10);
759         int32_t len = (int32_t)FXSYS_strlen(offset_buf);
760         if (pFile->AppendBlock(offset_buf, len) < 0) {
761             return FALSE;
762         }
763         offset += len + 6;
764     }
765     FX_BOOL bPredictor = TRUE;
766     CPDF_FlateEncoder encoder;
767     encoder.Initialize(m_Buffer.GetBuffer(), m_Buffer.GetLength(), pCreator->m_bCompress, bPredictor);
768     if (pCreator->m_bCompress) {
769         if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
770             return FALSE;
771         }
772         offset += 20;
773         if (bPredictor) {
774             if ((len = pFile->AppendString(FX_BSTRC("/DecodeParms<</Columns 7/Predictor 12>>"))) < 0) {
775                 return FALSE;
776             }
777             offset += len;
778         }
779     }
780     if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
781         return FALSE;
782     }
783     if ((len = pFile->AppendDWord(encoder.m_dwSize)) < 0) {
784         return FALSE;
785     }
786     offset += len + 8;
787     if (bEOF) {
788         if ((len = PDF_CreatorWriteTrailer(pCreator->m_pDocument, pFile, pCreator->m_pIDArray, pCreator->m_bCompress)) < 0) {
789             return FALSE;
790         }
791         offset += len;
792         if (pCreator->m_pEncryptDict) {
793             FX_DWORD dwEncryptObjNum = pCreator->m_pEncryptDict->GetObjNum();
794             if (dwEncryptObjNum == 0) {
795                 dwEncryptObjNum = pCreator->m_dwEnryptObjNum;
796             }
797             if ((len = PDF_CreatorWriteEncrypt(pCreator->m_pEncryptDict, dwEncryptObjNum, pFile)) < 0) {
798                 return FALSE;
799             }
800             offset += len;
801         }
802     }
803     if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
804         return FALSE;
805     }
806     offset += len;
807     if (pFile->AppendBlock(encoder.m_pData, encoder.m_dwSize) < 0) {
808         return FALSE;
809     }
810     if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) < 0) {
811         return FALSE;
812     }
813     offset += encoder.m_dwSize + len;
814     m_PrevOffset = offset_tmp;
815     return TRUE;
816 }
817 FX_BOOL CPDF_XRefStream::End(CPDF_Creator *pCreator, FX_BOOL bEOF )
818 {
819     if (EndObjectStream(pCreator, bEOF) < 0) {
820         return FALSE;
821     }
822     return GenerateXRefStream(pCreator, bEOF);
823 }
824 FX_BOOL CPDF_XRefStream::EndXRefStream(CPDF_Creator* pCreator)
825 {
826     if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
827         _AppendIndex0(m_Buffer);
828         for (FX_DWORD i = 1; i < pCreator->m_dwLastObjNum + 1; i++) {
829             FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(i);
830             if (offset) {
831                 _AppendIndex1(m_Buffer, *offset);
832             } else {
833                 _AppendIndex0(m_Buffer, FALSE);
834             }
835         }
836     } else {
837         int32_t iSeg = m_IndexArray.GetSize() / 2;
838         for (int i = 0; i < iSeg; i++) {
839             FX_DWORD start = m_IndexArray.ElementAt(i * 2);
840             FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
841             for (FX_DWORD j = start; j < end; j++) {
842                 _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[j]);
843             }
844         }
845     }
846     return GenerateXRefStream(pCreator, FALSE);
847 }
848 FX_BOOL CPDF_XRefStream::AddObjectNumberToIndexArray(FX_DWORD objnum)
849 {
850     int32_t iSize = m_IndexArray.GetSize();
851     if (iSize == 0) {
852         m_IndexArray.Add(objnum);
853         m_IndexArray.Add(1);
854     } else {
855         FXSYS_assert(iSize > 1);
856         FX_DWORD startobjnum = m_IndexArray.ElementAt(iSize - 2);
857         int32_t iCount = m_IndexArray.ElementAt(iSize - 1);
858         if (objnum == startobjnum + iCount) {
859             m_IndexArray[iSize - 1] = iCount + 1;
860         } else {
861             m_IndexArray.Add(objnum);
862             m_IndexArray.Add(1);
863         }
864     }
865     return TRUE;
866 }
867 CPDF_Creator::CPDF_Creator(CPDF_Document* pDoc)
868 {
869     m_pDocument = pDoc;
870     m_pParser = (CPDF_Parser*)pDoc->m_pParser;
871     m_bCompress = TRUE;
872     if (m_pParser) {
873         m_pEncryptDict = m_pParser->GetEncryptDict();
874         m_pCryptoHandler = m_pParser->GetCryptoHandler();
875     } else {
876         m_pEncryptDict = NULL;
877         m_pCryptoHandler = NULL;
878     }
879     m_bSecurityChanged = FALSE;
880     m_bStandardSecurity = FALSE;
881     m_pMetadata = NULL;
882     m_bEncryptCloned = FALSE;
883     m_bEncryptMetadata = FALSE;
884     m_Offset = 0;
885     m_iStage = -1;
886     m_dwFlags = 0;
887     m_Pos = NULL;
888     m_XrefStart = 0;
889     m_pXRefStream = NULL;
890     m_ObjectStreamSize = 200;
891     m_dwLastObjNum = m_pDocument->GetLastObjNum();
892     m_pIDArray = NULL;
893     m_FileVersion = 0;
894     m_dwEnryptObjNum = 0;
895     m_bNewCrypto = FALSE;
896 }
897 CPDF_Creator::~CPDF_Creator()
898 {
899     ResetStandardSecurity();
900     if (m_bEncryptCloned && m_pEncryptDict) {
901         m_pEncryptDict->Release();
902         m_pEncryptDict = NULL;
903     }
904     Clear();
905 }
906 static FX_BOOL _IsXRefNeedEnd(CPDF_XRefStream* pXRef, FX_DWORD flag)
907 {
908     if (!(flag & FPDFCREATE_INCREMENTAL)) {
909         return FALSE;
910     }
911     int32_t iSize = pXRef->m_IndexArray.GetSize() / 2;
912     int32_t iCount = 0;
913     for (int32_t i = 0; i < iSize; i++) {
914         iCount += pXRef->m_IndexArray.ElementAt(i * 2 + 1);
915     }
916     return (iCount >= PDF_XREFSTREAM_MAXSIZE);
917 }
918 int32_t CPDF_Creator::WriteIndirectObjectToStream(const CPDF_Object* pObj)
919 {
920     if (!m_pXRefStream) {
921         return 1;
922     }
923     FX_DWORD objnum = pObj->GetObjNum();
924     if (m_pParser && m_pParser->m_ObjVersion.GetSize() > (int32_t)objnum && m_pParser->m_ObjVersion[objnum] > 0) {
925         return 1;
926     }
927     if (pObj->GetType() == PDFOBJ_NUMBER) {
928         return 1;
929     }
930     CPDF_Dictionary *pDict = pObj->GetDict();
931     if (pObj->GetType() == PDFOBJ_STREAM) {
932         if (pDict && pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("XRef")) {
933             return 0;
934         }
935         return 1;
936     }
937     if (pDict) {
938         if (pDict == m_pDocument->m_pRootDict || pDict == m_pEncryptDict) {
939             return 1;
940         }
941         if (IsSignatureDict(pDict)) {
942             return 1;
943         }
944         if (pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("Page")) {
945             return 1;
946         }
947     }
948     m_pXRefStream->AddObjectNumberToIndexArray(objnum);
949     if (m_pXRefStream->CompressIndirectObject(objnum, pObj, this) < 0) {
950         return -1;
951     }
952     if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
953         return 0;
954     }
955     if (!m_pXRefStream->End(this)) {
956         return -1;
957     }
958     if (!m_pXRefStream->Start()) {
959         return -1;
960     }
961     return 0;
962 }
963 int32_t CPDF_Creator::WriteIndirectObjectToStream(FX_DWORD objnum, const uint8_t* pBuffer, FX_DWORD dwSize)
964 {
965     if (!m_pXRefStream) {
966         return 1;
967     }
968     m_pXRefStream->AddObjectNumberToIndexArray(objnum);
969     int32_t iRet = m_pXRefStream->CompressIndirectObject(objnum, pBuffer, dwSize, this);
970     if (iRet < 1) {
971         return iRet;
972     }
973     if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
974         return 0;
975     }
976     if (!m_pXRefStream->End(this)) {
977         return -1;
978     }
979     if (!m_pXRefStream->Start()) {
980         return -1;
981     }
982     return 0;
983 }
984 int32_t CPDF_Creator::AppendObjectNumberToXRef(FX_DWORD objnum)
985 {
986     if (!m_pXRefStream) {
987         return 1;
988     }
989     m_pXRefStream->AddObjectNumberToIndexArray(objnum);
990     if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
991         return 0;
992     }
993     if (!m_pXRefStream->End(this)) {
994         return -1;
995     }
996     if (!m_pXRefStream->Start()) {
997         return -1;
998     }
999     return 0;
1000 }
1001 int32_t CPDF_Creator::WriteStream(const CPDF_Object* pStream, FX_DWORD objnum, CPDF_CryptoHandler* pCrypto)
1002 {
1003     CPDF_FlateEncoder encoder;
1004     encoder.Initialize((CPDF_Stream*)pStream, pStream == m_pMetadata ? FALSE : m_bCompress);
1005     CPDF_Encryptor encryptor;
1006     if(!encryptor.Initialize(pCrypto, objnum, encoder.m_pData, encoder.m_dwSize)) {
1007         return -1;
1008     }
1009     if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) != encryptor.m_dwSize) {
1010         encoder.CloneDict();
1011         encoder.m_pDict->SetAtInteger(FX_BSTRC("Length"), encryptor.m_dwSize);
1012     }
1013     if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
1014         return -1;
1015     }
1016     int len = m_File.AppendString(FX_BSTRC("stream\r\n"));
1017     if (len < 0) {
1018         return -1;
1019     }
1020     m_Offset += len;
1021     if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
1022         return -1;
1023     }
1024     m_Offset += encryptor.m_dwSize;
1025     if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
1026         return -1;
1027     }
1028     m_Offset += len;
1029     return 1;
1030 }
1031 int32_t CPDF_Creator::WriteIndirectObj(FX_DWORD objnum, const CPDF_Object* pObj)
1032 {
1033     int32_t len = m_File.AppendDWord(objnum);
1034     if (len < 0) {
1035         return -1;
1036     }
1037     m_Offset += len;
1038     if ((len = m_File.AppendString(FX_BSTRC(" 0 obj\r\n"))) < 0) {
1039         return -1;
1040     }
1041     m_Offset += len;
1042     if (pObj->GetType() == PDFOBJ_STREAM) {
1043         CPDF_CryptoHandler *pHandler = NULL;
1044         pHandler = (pObj == m_pMetadata && !m_bEncryptMetadata) ? NULL : m_pCryptoHandler;
1045         if (WriteStream(pObj, objnum, pHandler) < 0) {
1046             return -1;
1047         }
1048     } else {
1049         if (WriteDirectObj(objnum, pObj) < 0) {
1050             return -1;
1051         }
1052     }
1053     if ((len = m_File.AppendString(FX_BSTRC("\r\nendobj\r\n"))) < 0) {
1054         return -1;
1055     }
1056     m_Offset += len;
1057     if (AppendObjectNumberToXRef(objnum) < 0) {
1058         return -1;
1059     }
1060     return 0;
1061 }
1062 int32_t CPDF_Creator::WriteIndirectObj(const CPDF_Object* pObj)
1063 {
1064     int32_t iRet = WriteIndirectObjectToStream(pObj);
1065     if (iRet < 1) {
1066         return iRet;
1067     }
1068     return WriteIndirectObj(pObj->GetObjNum(), pObj);
1069 }
1070 int32_t CPDF_Creator::WriteDirectObj(FX_DWORD objnum, const CPDF_Object* pObj, FX_BOOL bEncrypt)
1071 {
1072     int32_t len = 0;
1073     if (pObj == NULL) {
1074         if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
1075             return -1;
1076         }
1077         m_Offset += 5;
1078         return 1;
1079     }
1080     switch (pObj->GetType()) {
1081         case PDFOBJ_NULL:
1082             if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
1083                 return -1;
1084             }
1085             m_Offset += 5;
1086             break;
1087         case PDFOBJ_BOOLEAN:
1088         case PDFOBJ_NUMBER:
1089             if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1090                 return -1;
1091             }
1092             if ((len = m_File.AppendString(pObj->GetString())) < 0) {
1093                 return -1;
1094             }
1095             m_Offset += len + 1;
1096             break;
1097         case PDFOBJ_STRING: {
1098                 CFX_ByteString str = pObj->GetString();
1099                 FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
1100                 if (m_pCryptoHandler == NULL || !bEncrypt) {
1101                     CFX_ByteString content = PDF_EncodeString(str, bHex);
1102                     if ((len = m_File.AppendString(content)) < 0) {
1103                         return -1;
1104                     }
1105                     m_Offset += len;
1106                     break;
1107                 }
1108                 CPDF_Encryptor encryptor;
1109                 encryptor.Initialize(m_pCryptoHandler, objnum, (uint8_t*)str.c_str(), str.GetLength());
1110                 CFX_ByteString content = PDF_EncodeString(CFX_ByteString((const FX_CHAR*)encryptor.m_pData, encryptor.m_dwSize), bHex);
1111                 if ((len = m_File.AppendString(content)) < 0) {
1112                     return -1;
1113                 }
1114                 m_Offset += len;
1115                 break;
1116             }
1117         case PDFOBJ_STREAM: {
1118                 CPDF_FlateEncoder encoder;
1119                 encoder.Initialize((CPDF_Stream*)pObj, m_bCompress);
1120                 CPDF_Encryptor encryptor;
1121                 CPDF_CryptoHandler* pHandler = m_pCryptoHandler;
1122                 encryptor.Initialize(pHandler, objnum, encoder.m_pData, encoder.m_dwSize);
1123                 if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) != encryptor.m_dwSize) {
1124                     encoder.CloneDict();
1125                     encoder.m_pDict->SetAtInteger(FX_BSTRC("Length"), encryptor.m_dwSize);
1126                 }
1127                 if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
1128                     return -1;
1129                 }
1130                 if ((len = m_File.AppendString(FX_BSTRC("stream\r\n"))) < 0) {
1131                     return -1;
1132                 }
1133                 m_Offset += len;
1134                 if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
1135                     return -1;
1136                 }
1137                 m_Offset += encryptor.m_dwSize;
1138                 if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
1139                     return -1;
1140                 }
1141                 m_Offset += len;
1142                 break;
1143             }
1144         case PDFOBJ_NAME: {
1145                 if (m_File.AppendString(FX_BSTRC("/")) < 0) {
1146                     return -1;
1147                 }
1148                 CFX_ByteString str = pObj->GetString();
1149                 if ((len = m_File.AppendString(PDF_NameEncode(str))) < 0) {
1150                     return -1;
1151                 }
1152                 m_Offset += len + 1;
1153                 break;
1154             }
1155         case PDFOBJ_REFERENCE: {
1156                 if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1157                     return -1;
1158                 }
1159                 CPDF_Reference* p = (CPDF_Reference*)pObj;
1160                 if ((len = m_File.AppendDWord(p->GetRefObjNum())) < 0) {
1161                     return -1;
1162                 }
1163                 if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
1164                     return -1;
1165                 }
1166                 m_Offset += len + 5;
1167                 break;
1168             }
1169         case PDFOBJ_ARRAY: {
1170                 if (m_File.AppendString(FX_BSTRC("[")) < 0) {
1171                     return -1;
1172                 }
1173                 m_Offset += 1;
1174                 CPDF_Array* p = (CPDF_Array*)pObj;
1175                 for (FX_DWORD i = 0; i < p->GetCount(); i ++) {
1176                     CPDF_Object* pElement = p->GetElement(i);
1177                     if (pElement->GetObjNum()) {
1178                         if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1179                             return -1;
1180                         }
1181                         if ((len = m_File.AppendDWord(pElement->GetObjNum())) < 0) {
1182                             return -1;
1183                         }
1184                         if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
1185                             return -1;
1186                         }
1187                         m_Offset += len + 5;
1188                     } else {
1189                         if (WriteDirectObj(objnum, pElement) < 0) {
1190                             return -1;
1191                         }
1192                     }
1193                 }
1194                 if (m_File.AppendString(FX_BSTRC("]")) < 0) {
1195                     return -1;
1196                 }
1197                 m_Offset += 1;
1198                 break;
1199             }
1200         case PDFOBJ_DICTIONARY: {
1201                 if (m_pCryptoHandler == NULL || pObj == m_pEncryptDict) {
1202                     return PDF_CreatorAppendObject(pObj, &m_File, m_Offset);
1203                 }
1204                 if (m_File.AppendString(FX_BSTRC("<<")) < 0) {
1205                     return -1;
1206                 }
1207                 m_Offset += 2;
1208                 CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
1209                 FX_BOOL bSignDict = IsSignatureDict(p);
1210                 FX_POSITION pos = p->GetStartPos();
1211                 while (pos) {
1212                     FX_BOOL bSignValue = FALSE;
1213                     CFX_ByteString key;
1214                     CPDF_Object* pValue = p->GetNextElement(pos, key);
1215                     if (m_File.AppendString(FX_BSTRC("/")) < 0) {
1216                         return -1;
1217                     }
1218                     if ((len = m_File.AppendString(PDF_NameEncode(key))) < 0) {
1219                         return -1;
1220                     }
1221                     m_Offset += len + 1;
1222                     if (bSignDict && key == FX_BSTRC("Contents")) {
1223                         bSignValue = TRUE;
1224                     }
1225                     if (pValue->GetObjNum()) {
1226                         if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1227                             return -1;
1228                         }
1229                         if ((len = m_File.AppendDWord(pValue->GetObjNum())) < 0) {
1230                             return -1;
1231                         }
1232                         if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
1233                             return -1;
1234                         }
1235                         m_Offset += len + 6;
1236                     } else {
1237                         if (WriteDirectObj(objnum, pValue, !bSignValue) < 0) {
1238                             return -1;
1239                         }
1240                     }
1241                 }
1242                 if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
1243                     return -1;
1244                 }
1245                 m_Offset += 2;
1246                 break;
1247             }
1248     }
1249     return 1;
1250 }
1251 int32_t CPDF_Creator::WriteOldIndirectObject(FX_DWORD objnum)
1252 {
1253     if(m_pParser->m_V5Type[objnum] == 0 || m_pParser->m_V5Type[objnum] == 255) {
1254         return 0;
1255     }
1256     m_ObjectOffset[objnum] = m_Offset;
1257     void* valuetemp = NULL;
1258     FX_BOOL bExistInMap = m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, valuetemp);
1259     FX_BOOL bObjStm = (m_pParser->m_V5Type[objnum] == 2) && m_pEncryptDict && !m_pXRefStream;
1260     if(m_pParser->m_bVersionUpdated || m_bSecurityChanged || bExistInMap || bObjStm) {
1261         CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum);
1262         if (pObj == NULL) {
1263             m_ObjectOffset[objnum] = 0;
1264             m_ObjectSize[objnum] = 0;
1265             return 0;
1266         }
1267         if (WriteIndirectObj(pObj)) {
1268             return -1;
1269         }
1270         if (!bExistInMap) {
1271             m_pDocument->ReleaseIndirectObject(objnum);
1272         }
1273     } else {
1274         uint8_t* pBuffer;
1275         FX_DWORD size;
1276         m_pParser->GetIndirectBinary(objnum, pBuffer, size);
1277         if (pBuffer == NULL) {
1278             return 0;
1279         }
1280         if (m_pParser->m_V5Type[objnum] == 2) {
1281             if (m_pXRefStream) {
1282                 if (WriteIndirectObjectToStream(objnum, pBuffer, size) < 0) {
1283                     FX_Free(pBuffer);
1284                     return -1;
1285                 }
1286             } else {
1287                 int32_t len = m_File.AppendDWord(objnum);
1288                 if (len < 0) {
1289                     return -1;
1290                 }
1291                 if (m_File.AppendString(FX_BSTRC(" 0 obj ")) < 0) {
1292                     return -1;
1293                 }
1294                 m_Offset += len + 7;
1295                 if (m_File.AppendBlock(pBuffer, size) < 0) {
1296                     return -1;
1297                 }
1298                 m_Offset += size;
1299                 if (m_File.AppendString(FX_BSTRC("\r\nendobj\r\n")) < 0) {
1300                     return -1;
1301                 }
1302                 m_Offset += 10;
1303             }
1304         } else {
1305             if (m_File.AppendBlock(pBuffer, size) < 0) {
1306                 return -1;
1307             }
1308             m_Offset += size;
1309             if(AppendObjectNumberToXRef(objnum) < 0) {
1310                 return -1;
1311             }
1312         }
1313         FX_Free(pBuffer);
1314     }
1315     return 1;
1316 }
1317 int32_t CPDF_Creator::WriteOldObjs(IFX_Pause *pPause)
1318 {
1319     FX_DWORD nOldSize = m_pParser->m_CrossRef.GetSize();
1320     FX_DWORD objnum = (FX_DWORD)(uintptr_t)m_Pos;
1321     for(; objnum < nOldSize; objnum ++) {
1322         int32_t iRet = WriteOldIndirectObject(objnum);
1323         if (!iRet) {
1324             continue;
1325         }
1326         if (iRet < 0) {
1327             return iRet;
1328         }
1329         m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
1330         if (pPause && pPause->NeedToPauseNow()) {
1331             m_Pos = (void*)(uintptr_t)(objnum + 1);
1332             return 1;
1333         }
1334     }
1335     return 0;
1336 }
1337 int32_t CPDF_Creator::WriteNewObjs(FX_BOOL bIncremental, IFX_Pause *pPause)
1338 {
1339     int32_t iCount = m_NewObjNumArray.GetSize();
1340     int32_t index = (int32_t)(uintptr_t)m_Pos;
1341     while (index < iCount) {
1342         FX_DWORD objnum = m_NewObjNumArray.ElementAt(index);
1343         CPDF_Object *pObj = NULL;
1344         m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, (void*&)pObj);
1345         if (NULL == pObj) {
1346             ++index;
1347             continue;
1348         }
1349         m_ObjectOffset[objnum] = m_Offset;
1350         if (WriteIndirectObj(pObj)) {
1351             return -1;
1352         }
1353         m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
1354         index++;
1355         if (pPause && pPause->NeedToPauseNow()) {
1356             m_Pos = (FX_POSITION)(uintptr_t)index;
1357             return 1;
1358         }
1359     }
1360     return 0;
1361 }
1362 void CPDF_Creator::InitOldObjNumOffsets()
1363 {
1364     if (!m_pParser) {
1365         return;
1366     }
1367     FX_DWORD j = 0;
1368     FX_DWORD dwStart = 0;
1369     FX_DWORD dwEnd = m_pParser->GetLastObjNum();
1370     while (dwStart <= dwEnd) {
1371         while (dwStart <= dwEnd && (m_pParser->m_V5Type[dwStart] == 0 || m_pParser->m_V5Type[dwStart] == 255)) {
1372             dwStart++;
1373         }
1374         if (dwStart > dwEnd) {
1375             break;
1376         }
1377         j = dwStart;
1378         while (j <= dwEnd && m_pParser->m_V5Type[j] != 0 && m_pParser->m_V5Type[j] != 255) {
1379             j++;
1380         }
1381         m_ObjectOffset.Add(dwStart, j - dwStart);
1382         m_ObjectSize.Add(dwStart, j - dwStart);
1383         dwStart = j;
1384     }
1385 }
1386 void CPDF_Creator::InitNewObjNumOffsets()
1387 {
1388     FX_BOOL bIncremental = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
1389     FX_BOOL bNoOriginal = (m_dwFlags & FPDFCREATE_NO_ORIGINAL) != 0;
1390     FX_DWORD nOldSize = m_pParser ? m_pParser->m_CrossRef.GetSize() : 0;
1391     FX_POSITION pos = m_pDocument->m_IndirectObjs.GetStartPosition();
1392     while (pos) {
1393         size_t key = 0;
1394         CPDF_Object* pObj;
1395         m_pDocument->m_IndirectObjs.GetNextAssoc(pos, (void*&)key, (void*&)pObj);
1396         FX_DWORD objnum = (FX_DWORD)key;
1397         if (pObj->GetObjNum() == -1) {
1398             continue;
1399         }
1400         if (bIncremental) {
1401             if (!pObj->IsModified()) {
1402                 continue;
1403             }
1404         } else {
1405             if (objnum < nOldSize && m_pParser->m_V5Type[objnum] != 0) {
1406                 continue;
1407             }
1408         }
1409         AppendNewObjNum(objnum);
1410     }
1411     int32_t iCount = m_NewObjNumArray.GetSize();
1412     if (iCount == 0) {
1413         return;
1414     }
1415     int32_t i = 0;
1416     FX_DWORD dwStartObjNum = 0;
1417     FX_BOOL bCrossRefValid = m_pParser && m_pParser->GetLastXRefOffset() > 0;
1418     while (i < iCount) {
1419         dwStartObjNum = m_NewObjNumArray.ElementAt(i);
1420         if ((bIncremental && (bNoOriginal || bCrossRefValid)) || !m_ObjectOffset.GetPtrAt(dwStartObjNum)) {
1421             break;
1422         }
1423         i++;
1424     }
1425     if (i >= iCount) {
1426         return;
1427     }
1428     FX_DWORD dwLastObjNum = dwStartObjNum;
1429     i++;
1430     FX_BOOL bNewStart = FALSE;
1431     for (; i < iCount; i++) {
1432         FX_DWORD dwCurObjNum = m_NewObjNumArray.ElementAt(i);
1433         FX_BOOL bExist = (dwCurObjNum < nOldSize && m_ObjectOffset.GetPtrAt(dwCurObjNum) != NULL);
1434         if (bExist || dwCurObjNum - dwLastObjNum > 1) {
1435             if (!bNewStart) {
1436                 m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
1437                 m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
1438             }
1439             dwStartObjNum = dwCurObjNum;
1440         }
1441         if (bNewStart) {
1442             dwStartObjNum = dwCurObjNum;
1443         }
1444         bNewStart = bExist;
1445         dwLastObjNum = dwCurObjNum;
1446     }
1447     m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
1448     m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
1449 }
1450 void CPDF_Creator::AppendNewObjNum(FX_DWORD objbum)
1451 {
1452     int32_t iStart = 0, iFind = 0;
1453     int32_t iEnd = m_NewObjNumArray.GetUpperBound();
1454     while (iStart <= iEnd) {
1455         int32_t iMid = (iStart + iEnd) / 2;
1456         FX_DWORD dwMid = m_NewObjNumArray.ElementAt(iMid);
1457         if (objbum < dwMid) {
1458             iEnd = iMid - 1;
1459         } else {
1460             if (iMid == iEnd) {
1461                 iFind = iMid + 1;
1462                 break;
1463             }
1464             FX_DWORD dwNext = m_NewObjNumArray.ElementAt(iMid + 1);
1465             if (objbum < dwNext) {
1466                 iFind = iMid + 1;
1467                 break;
1468             }
1469             iStart = iMid + 1;
1470         }
1471     }
1472     m_NewObjNumArray.InsertAt(iFind, objbum);
1473 }
1474 int32_t CPDF_Creator::WriteDoc_Stage1(IFX_Pause *pPause)
1475 {
1476     FXSYS_assert(m_iStage > -1 || m_iStage < 20);
1477     if (m_iStage == 0) {
1478         if (m_pParser == NULL) {
1479             m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
1480         }
1481         if (m_bSecurityChanged && (m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0) {
1482             m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
1483         }
1484         CPDF_Dictionary* pDict = m_pDocument->GetRoot();
1485         m_pMetadata = pDict ? pDict->GetElementValue(FX_BSTRC("Metadata")) : NULL;
1486         if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
1487             m_pXRefStream = new CPDF_XRefStream;
1488             m_pXRefStream->Start();
1489             if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser) {
1490                 FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
1491                 m_pXRefStream->m_PrevOffset = prev;
1492             }
1493         }
1494         m_iStage = 10;
1495     }
1496     if (m_iStage == 10) {
1497         if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0) {
1498             if (m_File.AppendString(FX_BSTRC("%PDF-1.")) < 0) {
1499                 return -1;
1500             }
1501             m_Offset += 7;
1502             int32_t version = 7;
1503             if (m_FileVersion) {
1504                 version = m_FileVersion;
1505             } else if (m_pParser) {
1506                 version = m_pParser->GetFileVersion();
1507             }
1508             int32_t len = m_File.AppendDWord(version % 10);
1509             if (len < 0) {
1510                 return -1;
1511             }
1512             m_Offset += len;
1513             if ((len = m_File.AppendString(FX_BSTRC("\r\n%\xA1\xB3\xC5\xD7\r\n"))) < 0) {
1514                 return -1;
1515             }
1516             m_Offset += len;
1517             InitOldObjNumOffsets();
1518             m_iStage = 20;
1519         } else {
1520             IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
1521             m_Offset = pSrcFile->GetSize();
1522             m_Pos = (void*)(uintptr_t)m_Offset;
1523             m_iStage = 15;
1524         }
1525     }
1526     if (m_iStage == 15) {
1527         if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_Pos) {
1528             IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
1529             uint8_t buffer[4096];
1530             FX_DWORD src_size = (FX_DWORD)(uintptr_t)m_Pos;
1531             while (src_size) {
1532                 FX_DWORD block_size = src_size > 4096 ? 4096 : src_size;
1533                 if (!pSrcFile->ReadBlock(buffer, m_Offset - src_size, block_size)) {
1534                     return -1;
1535                 }
1536                 if (m_File.AppendBlock(buffer, block_size) < 0) {
1537                     return -1;
1538                 }
1539                 src_size -= block_size;
1540                 if (pPause && pPause->NeedToPauseNow()) {
1541                     m_Pos = (void*)(uintptr_t)src_size;
1542                     return 1;
1543                 }
1544             }
1545         }
1546         if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_pParser->GetLastXRefOffset() == 0) {
1547             InitOldObjNumOffsets();
1548             FX_DWORD dwEnd = m_pParser->GetLastObjNum();
1549             FX_BOOL bObjStm = (m_dwFlags & FPDFCREATE_OBJECTSTREAM) != 0;
1550             for (FX_DWORD objnum = 0; objnum <= dwEnd; objnum++) {
1551                 if (m_pParser->m_V5Type[objnum] == 0 || m_pParser->m_V5Type[objnum] == 255) {
1552                     continue;
1553                 }
1554                 m_ObjectOffset[objnum] = m_pParser->m_CrossRef[objnum];
1555                 if (bObjStm) {
1556                     m_pXRefStream->AddObjectNumberToIndexArray(objnum);
1557                 }
1558             }
1559             if (bObjStm) {
1560                 m_pXRefStream->EndXRefStream(this);
1561                 m_pXRefStream->Start();
1562             }
1563         }
1564         m_iStage = 20;
1565     }
1566     InitNewObjNumOffsets();
1567     return m_iStage;
1568 }
1569 int32_t CPDF_Creator::WriteDoc_Stage2(IFX_Pause *pPause)
1570 {
1571     FXSYS_assert(m_iStage >= 20 || m_iStage < 30);
1572     if (m_iStage == 20) {
1573         if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 && m_pParser) {
1574             m_Pos = (void*)(uintptr_t)0;
1575             m_iStage = 21;
1576         } else {
1577             m_iStage = 25;
1578         }
1579     }
1580     if (m_iStage == 21) {
1581         int32_t iRet = WriteOldObjs(pPause);
1582         if (iRet) {
1583             return iRet;
1584         }
1585         m_iStage = 25;
1586     }
1587     if (m_iStage == 25) {
1588         m_Pos = (void*)(uintptr_t)0;
1589         m_iStage = 26;
1590     }
1591     if (m_iStage == 26) {
1592         int32_t iRet = WriteNewObjs((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0, pPause);
1593         if (iRet) {
1594             return iRet;
1595         }
1596         m_iStage = 27;
1597     }
1598     if (m_iStage == 27) {
1599         if (NULL != m_pEncryptDict && 0 == m_pEncryptDict->GetObjNum()) {
1600             m_dwLastObjNum += 1;
1601             FX_FILESIZE saveOffset = m_Offset;
1602             if (WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict) < 0) {
1603                 return -1;
1604             }
1605             m_ObjectOffset.Add(m_dwLastObjNum, 1);
1606             m_ObjectOffset[m_dwLastObjNum] = saveOffset;
1607             m_ObjectSize.Add(m_dwLastObjNum, 1);
1608             m_ObjectSize[m_dwLastObjNum] = m_Offset - saveOffset;
1609             m_dwEnryptObjNum = m_dwLastObjNum;
1610             if (m_dwFlags & FPDFCREATE_INCREMENTAL) {
1611                 m_NewObjNumArray.Add(m_dwLastObjNum);
1612             }
1613         }
1614         m_iStage = 80;
1615     }
1616     return m_iStage;
1617 }
1618 int32_t CPDF_Creator::WriteDoc_Stage3(IFX_Pause *pPause)
1619 {
1620     FXSYS_assert(m_iStage >= 80 || m_iStage < 90);
1621     FX_DWORD dwLastObjNum = m_dwLastObjNum;
1622     if (m_iStage == 80) {
1623         m_XrefStart = m_Offset;
1624         if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
1625             m_pXRefStream->End(this, TRUE);
1626             m_XrefStart = m_pXRefStream->m_PrevOffset;
1627             m_iStage = 90;
1628         } else if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || !m_pParser->IsXRefStream()) {
1629             if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || m_pParser->GetLastXRefOffset() == 0) {
1630                 CFX_ByteString str;
1631                 str = m_ObjectOffset.GetPtrAt(1) ? FX_BSTRC("xref\r\n") : FX_BSTRC("xref\r\n0 1\r\n0000000000 65536 f\r\n");
1632                 if (m_File.AppendString(str) < 0) {
1633                     return -1;
1634                 }
1635                 m_Pos = (void*)(uintptr_t)1;
1636                 m_iStage = 81;
1637             } else {
1638                 if (m_File.AppendString(FX_BSTRC("xref\r\n")) < 0) {
1639                     return -1;
1640                 }
1641                 m_Pos = (void*)(uintptr_t)0;
1642                 m_iStage = 82;
1643             }
1644         } else {
1645             m_iStage = 90;
1646         }
1647     }
1648     if (m_iStage == 81) {
1649         CFX_ByteString str;
1650         FX_DWORD i = (FX_DWORD)(uintptr_t)m_Pos, j;
1651         while (i <= dwLastObjNum) {
1652             while (i <= dwLastObjNum && !m_ObjectOffset.GetPtrAt(i)) {
1653                 i++;
1654             }
1655             if (i > dwLastObjNum) {
1656                 break;
1657             }
1658             j = i;
1659             while (j <= dwLastObjNum && m_ObjectOffset.GetPtrAt(j)) {
1660                 j++;
1661             }
1662             if (i == 1) {
1663                 str.Format("0 %d\r\n0000000000 65536 f\r\n", j);
1664             } else {
1665                 str.Format("%d %d\r\n", i, j - i);
1666             }
1667             if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
1668                 return -1;
1669             }
1670             while (i < j) {
1671                 str.Format("%010d 00000 n\r\n", m_ObjectOffset[i ++]);
1672                 if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
1673                     return -1;
1674                 }
1675             }
1676             if (i > dwLastObjNum) {
1677                 break;
1678             }
1679             if (pPause && pPause->NeedToPauseNow()) {
1680                 m_Pos = (void*)(uintptr_t)i;
1681                 return 1;
1682             }
1683         }
1684         m_iStage = 90;
1685     }
1686     if (m_iStage == 82) {
1687         CFX_ByteString str;
1688         int32_t iCount = m_NewObjNumArray.GetSize();
1689         int32_t i = (int32_t)(uintptr_t)m_Pos;
1690         while (i < iCount) {
1691             int32_t j = i;
1692             FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
1693             while (j < iCount) {
1694                 if (++j == iCount) {
1695                     break;
1696                 }
1697                 FX_DWORD dwCurrent = m_NewObjNumArray.ElementAt(j);
1698                 if (dwCurrent - objnum > 1) {
1699                     break;
1700                 }
1701                 objnum = dwCurrent;
1702             }
1703             objnum = m_NewObjNumArray.ElementAt(i);
1704             if (objnum == 1) {
1705                 str.Format("0 %d\r\n0000000000 65536 f\r\n", j - i + 1);
1706             } else {
1707                 str.Format("%d %d\r\n", objnum, j - i);
1708             }
1709             if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
1710                 return -1;
1711             }
1712             while (i < j) {
1713                 objnum = m_NewObjNumArray.ElementAt(i++);
1714                 str.Format("%010d 00000 n\r\n", m_ObjectOffset[objnum]);
1715                 if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
1716                     return -1;
1717                 }
1718             }
1719             if (pPause && (i % 100) == 0 && pPause->NeedToPauseNow()) {
1720                 m_Pos = (void*)(uintptr_t)i;
1721                 return 1;
1722             }
1723         }
1724         m_iStage = 90;
1725     }
1726     return m_iStage;
1727 }
1728 static int32_t _OutPutIndex(CFX_FileBufferArchive* pFile, FX_FILESIZE offset)
1729 {
1730     FXSYS_assert(pFile);
1731     if (sizeof(offset) > 4) {
1732         if (FX_GETBYTEOFFSET32(offset)) {
1733             if (pFile->AppendByte(FX_GETBYTEOFFSET56(offset)) < 0) {
1734                 return -1;
1735             }
1736             if (pFile->AppendByte(FX_GETBYTEOFFSET48(offset)) < 0) {
1737                 return -1;
1738             }
1739             if (pFile->AppendByte(FX_GETBYTEOFFSET40(offset)) < 0) {
1740                 return -1;
1741             }
1742             if (pFile->AppendByte(FX_GETBYTEOFFSET32(offset)) < 0) {
1743                 return -1;
1744             }
1745         }
1746     }
1747     if (pFile->AppendByte(FX_GETBYTEOFFSET24(offset)) < 0) {
1748         return -1;
1749     }
1750     if (pFile->AppendByte(FX_GETBYTEOFFSET16(offset)) < 0) {
1751         return -1;
1752     }
1753     if (pFile->AppendByte(FX_GETBYTEOFFSET8(offset)) < 0) {
1754         return -1;
1755     }
1756     if (pFile->AppendByte(FX_GETBYTEOFFSET0(offset)) < 0) {
1757         return -1;
1758     }
1759     if (pFile->AppendByte(0) < 0) {
1760         return -1;
1761     }
1762     return 0;
1763 }
1764 int32_t CPDF_Creator::WriteDoc_Stage4(IFX_Pause *pPause)
1765 {
1766     FXSYS_assert(m_iStage >= 90);
1767     if ((m_dwFlags & FPDFCREATE_OBJECTSTREAM) == 0) {
1768         FX_BOOL bXRefStream = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser->IsXRefStream();
1769         if (!bXRefStream) {
1770             if (m_File.AppendString(FX_BSTRC("trailer\r\n<<")) < 0) {
1771                 return -1;
1772             }
1773         } else {
1774             if (m_File.AppendDWord(m_pDocument->m_LastObjNum + 1) < 0) {
1775                 return -1;
1776             }
1777             if (m_File.AppendString(FX_BSTRC(" 0 obj <<")) < 0) {
1778                 return -1;
1779             }
1780         }
1781         if (m_pParser) {
1782             CPDF_Dictionary* p = m_pParser->m_pTrailer;
1783             FX_POSITION pos = p->GetStartPos();
1784             while (pos) {
1785                 CFX_ByteString key;
1786                 CPDF_Object* pValue = p->GetNextElement(pos, key);
1787                 if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") || key == FX_BSTRC("Filter") ||
1788                         key == FX_BSTRC("Index") || key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
1789                         key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") || key == FX_BSTRC("ID")) {
1790                     continue;
1791                 }
1792                 if (m_File.AppendString((FX_BSTRC("/"))) < 0) {
1793                     return -1;
1794                 }
1795                 if (m_File.AppendString(PDF_NameEncode(key)) < 0) {
1796                     return -1;
1797                 }
1798                 if (pValue->GetObjNum()) {
1799                     if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1800                         return -1;
1801                     }
1802                     if (m_File.AppendDWord(pValue->GetObjNum()) < 0) {
1803                         return -1;
1804                     }
1805                     if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
1806                         return -1;
1807                     }
1808                 } else {
1809                     FX_FILESIZE offset = 0;
1810                     if (PDF_CreatorAppendObject(pValue, &m_File, offset) < 0) {
1811                         return -1;
1812                     }
1813                 }
1814             }
1815         } else {
1816             if (m_File.AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
1817                 return -1;
1818             }
1819             if (m_File.AppendDWord(m_pDocument->m_pRootDict->GetObjNum()) < 0) {
1820                 return -1;
1821             }
1822             if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
1823                 return -1;
1824             }
1825             if (m_pDocument->m_pInfoDict) {
1826                 if (m_File.AppendString(FX_BSTRC("/Info ")) < 0) {
1827                     return -1;
1828                 }
1829                 if (m_File.AppendDWord(m_pDocument->m_pInfoDict->GetObjNum()) < 0) {
1830                     return -1;
1831                 }
1832                 if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
1833                     return -1;
1834                 }
1835             }
1836         }
1837         if (m_pEncryptDict) {
1838             if (m_File.AppendString(FX_BSTRC("/Encrypt")) < 0) {
1839                 return -1;
1840             }
1841             FX_DWORD dwObjNum = m_pEncryptDict->GetObjNum();
1842             if (dwObjNum == 0) {
1843                 dwObjNum = m_pDocument->GetLastObjNum() + 1;
1844             }
1845             if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
1846                 return -1;
1847             }
1848             if (m_File.AppendDWord(dwObjNum) < 0) {
1849                 return -1;
1850             }
1851             if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
1852                 return -1;
1853             }
1854         }
1855         if (m_File.AppendString(FX_BSTRC("/Size ")) < 0) {
1856             return -1;
1857         }
1858         if (m_File.AppendDWord(m_dwLastObjNum + (bXRefStream ? 2 : 1)) < 0) {
1859             return -1;
1860         }
1861         if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0) {
1862             FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
1863             if (prev) {
1864                 if (m_File.AppendString(FX_BSTRC("/Prev ")) < 0) {
1865                     return -1;
1866                 }
1867                 FX_CHAR offset_buf[20];
1868                 FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
1869                 FXSYS_i64toa(prev, offset_buf, 10);
1870                 if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
1871                     return -1;
1872                 }
1873             }
1874         }
1875         if (m_pIDArray) {
1876             if (m_File.AppendString((FX_BSTRC("/ID"))) < 0) {
1877                 return -1;
1878             }
1879             FX_FILESIZE offset = 0;
1880             if (PDF_CreatorAppendObject(m_pIDArray, &m_File, offset) < 0) {
1881                 return -1;
1882             }
1883         }
1884         if (!bXRefStream) {
1885             if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
1886                 return -1;
1887             }
1888         } else {
1889             if (m_File.AppendString(FX_BSTRC("/W[0 4 1]/Index[")) < 0) {
1890                 return -1;
1891             }
1892             if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser && m_pParser->GetLastXRefOffset() == 0) {
1893                 FX_DWORD i = 0;
1894                 for (i = 0; i < m_dwLastObjNum; i++) {
1895                     if (!m_ObjectOffset.GetPtrAt(i)) {
1896                         continue;
1897                     }
1898                     if (m_File.AppendDWord(i) < 0) {
1899                         return -1;
1900                     }
1901                     if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
1902                         return -1;
1903                     }
1904                 }
1905                 if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
1906                     return -1;
1907                 }
1908                 if (m_File.AppendDWord(m_dwLastObjNum * 5) < 0) {
1909                     return -1;
1910                 }
1911                 if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
1912                     return -1;
1913                 }
1914                 for (i = 0; i < m_dwLastObjNum; i++) {
1915                     FX_FILESIZE* offset = m_ObjectOffset.GetPtrAt(i);
1916                     if (!offset) {
1917                         continue;
1918                     }
1919                     _OutPutIndex(&m_File, *offset);
1920                 }
1921             } else {
1922                 int count = m_NewObjNumArray.GetSize();
1923                 int32_t i = 0;
1924                 for (i = 0; i < count; i++) {
1925                     FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
1926                     if (m_File.AppendDWord(objnum) < 0) {
1927                         return -1;
1928                     }
1929                     if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
1930                         return -1;
1931                     }
1932                 }
1933                 if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
1934                     return -1;
1935                 }
1936                 if (m_File.AppendDWord(count * 5) < 0) {
1937                     return -1;
1938                 }
1939                 if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
1940                     return -1;
1941                 }
1942                 for (i = 0; i < count; i++) {
1943                     FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
1944                     FX_FILESIZE offset = m_ObjectOffset[objnum];
1945                     _OutPutIndex(&m_File, offset);
1946                 }
1947             }
1948             if (m_File.AppendString(FX_BSTRC("\r\nendstream")) < 0) {
1949                 return -1;
1950             }
1951         }
1952     }
1953     if (m_File.AppendString(FX_BSTRC("\r\nstartxref\r\n")) < 0) {
1954         return -1;
1955     }
1956     FX_CHAR offset_buf[20];
1957     FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
1958     FXSYS_i64toa(m_XrefStart, offset_buf, 10);
1959     if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
1960         return -1;
1961     }
1962     if (m_File.AppendString(FX_BSTRC("\r\n%%EOF\r\n")) < 0) {
1963         return -1;
1964     }
1965     m_File.Flush();
1966     return m_iStage = 100;
1967 }
1968 void CPDF_Creator::Clear()
1969 {
1970     delete m_pXRefStream;
1971     m_pXRefStream = NULL;
1972     m_File.Clear();
1973     m_NewObjNumArray.RemoveAll();
1974     if (m_pIDArray) {
1975         m_pIDArray->Release();
1976         m_pIDArray = NULL;
1977     }
1978 }
1979 FX_BOOL CPDF_Creator::Create(const FX_CHAR* filename, FX_DWORD flags)
1980 {
1981     if (!m_File.AttachFile(filename)) {
1982         return FALSE;
1983     }
1984     FX_BOOL bRet = Create(flags);
1985     if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
1986         Clear();
1987     }
1988     return bRet;
1989 }
1990 FX_BOOL CPDF_Creator::Create(const FX_WCHAR* filename, FX_DWORD flags)
1991 {
1992     if (!m_File.AttachFile(filename)) {
1993         return FALSE;
1994     }
1995     FX_BOOL bRet = Create(flags);
1996     if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
1997         Clear();
1998     }
1999     return bRet;
2000 }
2001 FX_BOOL CPDF_Creator::Create(IFX_StreamWrite* pFile, FX_DWORD flags)
2002 {
2003     if (!pFile) {
2004         return FALSE;
2005     }
2006     if (!m_File.AttachFile(pFile, FALSE)) {
2007         return FALSE;
2008     }
2009     return Create(flags);
2010 }
2011 FX_BOOL CPDF_Creator::Create(FX_DWORD flags)
2012 {
2013     m_dwFlags = flags;
2014     m_iStage = 0;
2015     m_Offset = 0;
2016     m_dwLastObjNum = m_pDocument->GetLastObjNum();
2017     m_ObjectOffset.Clear();
2018     m_ObjectSize.Clear();
2019     m_NewObjNumArray.RemoveAll();
2020     InitID();
2021     if (flags & FPDFCREATE_PROGRESSIVE) {
2022         return TRUE;
2023     }
2024     return Continue(NULL) > -1;
2025 }
2026 void CPDF_Creator::InitID(FX_BOOL bDefault )
2027 {
2028     CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : NULL;
2029     FX_BOOL bNewId = !m_pIDArray;
2030     if (!m_pIDArray) {
2031         FX_DWORD* pBuffer = NULL;
2032         m_pIDArray = CPDF_Array::Create();
2033         CPDF_Object* pID1 = pOldIDArray ? pOldIDArray->GetElement(0) : NULL;
2034         if (pID1) {
2035             m_pIDArray->Add(pID1->Clone());
2036         } else {
2037             pBuffer = FX_Alloc(FX_DWORD, 4);
2038             PDF_GenerateFileID((FX_DWORD)(uintptr_t)this, m_dwLastObjNum, pBuffer);
2039             CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
2040             m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
2041         }
2042         if (pBuffer) {
2043             FX_Free(pBuffer);
2044         }
2045     }
2046     if (!bDefault) {
2047         return;
2048     }
2049     if (pOldIDArray) {
2050         CPDF_Object* pID2 = pOldIDArray->GetElement(1);
2051         if ((m_dwFlags & FPDFCREATE_INCREMENTAL) && m_pEncryptDict && pID2) {
2052             m_pIDArray->Add(pID2->Clone());
2053             return;
2054         }
2055         FX_DWORD* pBuffer = FX_Alloc(FX_DWORD, 4);
2056         PDF_GenerateFileID((FX_DWORD)(uintptr_t)this, m_dwLastObjNum, pBuffer);
2057         CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
2058         m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
2059         FX_Free(pBuffer);
2060         return;
2061     }
2062     m_pIDArray->Add(m_pIDArray->GetElement(0)->Clone());
2063     if (m_pEncryptDict && !pOldIDArray && m_pParser && bNewId) {
2064         if (m_pEncryptDict->GetString(FX_BSTRC("Filter")) == FX_BSTRC("Standard")) {
2065             CPDF_StandardSecurityHandler handler;
2066             CFX_ByteString user_pass = m_pParser->GetPassword();
2067             FX_DWORD flag = PDF_ENCRYPT_CONTENT;
2068             handler.OnCreate(m_pEncryptDict, m_pIDArray, (const uint8_t*)user_pass, user_pass.GetLength(), flag);
2069             if (m_bNewCrypto) {
2070                 delete m_pCryptoHandler;
2071             }
2072             m_pCryptoHandler = new CPDF_StandardCryptoHandler;
2073             m_pCryptoHandler->Init(m_pEncryptDict, &handler);
2074             m_bNewCrypto = TRUE;
2075             m_bSecurityChanged = TRUE;
2076         }
2077     }
2078 }
2079 int32_t CPDF_Creator::Continue(IFX_Pause *pPause)
2080 {
2081     if (m_iStage < 0) {
2082         return m_iStage;
2083     }
2084     int32_t iRet = 0;
2085     while (m_iStage < 100) {
2086         if (m_iStage < 20) {
2087             iRet = WriteDoc_Stage1(pPause);
2088         } else if (m_iStage < 30) {
2089             iRet = WriteDoc_Stage2(pPause);
2090         } else if (m_iStage < 90) {
2091             iRet = WriteDoc_Stage3(pPause);
2092         } else {
2093             iRet = WriteDoc_Stage4(pPause);
2094         }
2095         if (iRet < m_iStage) {
2096             break;
2097         }
2098     }
2099     if (iRet < 1 || m_iStage == 100) {
2100         m_iStage = -1;
2101         Clear();
2102         return iRet > 99 ? 0 : (iRet < 1 ? -1 : iRet);
2103     }
2104     return m_iStage;
2105 }
2106 FX_BOOL CPDF_Creator::SetFileVersion(int32_t fileVersion )
2107 {
2108     if (fileVersion < 10 || fileVersion > 17) {
2109         return FALSE;
2110     }
2111     m_FileVersion = fileVersion;
2112     return TRUE;
2113 }
2114 void CPDF_Creator::RemoveSecurity()
2115 {
2116     ResetStandardSecurity();
2117     m_bSecurityChanged = TRUE;
2118     m_pEncryptDict = NULL;
2119     m_pCryptoHandler = NULL;
2120 }
2121 void CPDF_Creator::ResetStandardSecurity()
2122 {
2123     if (m_bStandardSecurity || m_bNewCrypto) {
2124         delete m_pCryptoHandler;
2125         m_pCryptoHandler = NULL;
2126     }
2127     m_bNewCrypto = FALSE;
2128     if (!m_bStandardSecurity) {
2129         return;
2130     }
2131     if (m_pEncryptDict) {
2132         m_pEncryptDict->Release();
2133         m_pEncryptDict = NULL;
2134     }
2135     m_bStandardSecurity = FALSE;
2136 }