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