| Index: core/src/fpdfapi/fpdf_edit/fpdf_edit_create.cpp
|
| diff --git a/core/src/fpdfapi/fpdf_edit/fpdf_edit_create.cpp b/core/src/fpdfapi/fpdf_edit/fpdf_edit_create.cpp
|
| index 7b7b61aa025a0c714b3b755f350fc5a0933c96a4..6f193071debfb189af9497de221c1325b04d2474 100644
|
| --- a/core/src/fpdfapi/fpdf_edit/fpdf_edit_create.cpp
|
| +++ b/core/src/fpdfapi/fpdf_edit/fpdf_edit_create.cpp
|
| @@ -8,2129 +8,2161 @@
|
| #include "../../../include/fpdfapi/fpdf_serial.h"
|
| #include "editint.h"
|
|
|
| -#define PDF_OBJECTSTREAM_MAXLENGTH (256 * 1024)
|
| -#define PDF_XREFSTREAM_MAXSIZE 10000
|
| -extern void FlateEncode(const uint8_t* src_buf, FX_DWORD src_data, uint8_t*& dest_buf, FX_DWORD& dest_size);
|
| -extern void FlateEncode(const uint8_t* src_buf, FX_DWORD src_size, int predictor, int Colors, int BitsPerComponent, int Columns,
|
| - uint8_t*& dest_buf, FX_DWORD& dest_size);
|
| +#define PDF_OBJECTSTREAM_MAXLENGTH (256 * 1024)
|
| +#define PDF_XREFSTREAM_MAXSIZE 10000
|
| +extern void FlateEncode(const uint8_t* src_buf,
|
| + FX_DWORD src_data,
|
| + uint8_t*& dest_buf,
|
| + FX_DWORD& dest_size);
|
| +extern void FlateEncode(const uint8_t* src_buf,
|
| + FX_DWORD src_size,
|
| + int predictor,
|
| + int Colors,
|
| + int BitsPerComponent,
|
| + int Columns,
|
| + uint8_t*& dest_buf,
|
| + FX_DWORD& dest_size);
|
| extern FX_BOOL IsSignatureDict(const CPDF_Dictionary* pDict);
|
| -int32_t PDF_CreatorAppendObject(const CPDF_Object* pObj, CFX_FileBufferArchive *pFile, FX_FILESIZE& offset)
|
| -{
|
| - int32_t len = 0;
|
| - if (pObj == NULL) {
|
| - if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 5;
|
| - return 1;
|
| - }
|
| - switch (pObj->GetType()) {
|
| - case PDFOBJ_NULL:
|
| - if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 5;
|
| - break;
|
| - case PDFOBJ_BOOLEAN:
|
| - case PDFOBJ_NUMBER:
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendString(pObj->GetString())) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 1;
|
| - break;
|
| - case PDFOBJ_STRING: {
|
| - CFX_ByteString str = pObj->GetString();
|
| - FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
|
| - if ((len = pFile->AppendString(PDF_EncodeString(str, bHex))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - break;
|
| - }
|
| - case PDFOBJ_NAME: {
|
| - if (pFile->AppendString(FX_BSTRC("/")) < 0) {
|
| - return -1;
|
| - }
|
| - CFX_ByteString str = pObj->GetString();
|
| - if ((len = pFile->AppendString(PDF_NameEncode(str))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 1;
|
| - break;
|
| - }
|
| - case PDFOBJ_REFERENCE: {
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - CPDF_Reference* p = (CPDF_Reference*)pObj;
|
| - if ((len = pFile->AppendDWord(p->GetRefObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 6;
|
| - break;
|
| - }
|
| - case PDFOBJ_ARRAY: {
|
| - if (pFile->AppendString(FX_BSTRC("[")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 1;
|
| - CPDF_Array* p = (CPDF_Array*)pObj;
|
| - for (FX_DWORD i = 0; i < p->GetCount(); i ++) {
|
| - CPDF_Object* pElement = p->GetElement(i);
|
| - if (pElement->GetObjNum()) {
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendDWord(pElement->GetObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 5;
|
| - } else {
|
| - if (PDF_CreatorAppendObject(pElement, pFile, offset) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC("]")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 1;
|
| - break;
|
| - }
|
| - case PDFOBJ_DICTIONARY: {
|
| - if (pFile->AppendString(FX_BSTRC("<<")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 2;
|
| - CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
|
| - FX_POSITION pos = p->GetStartPos();
|
| - while (pos) {
|
| - CFX_ByteString key;
|
| - CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| - if (pFile->AppendString(FX_BSTRC("/")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 1;
|
| - if (pValue->GetObjNum()) {
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 5;
|
| - } else {
|
| - if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(">>")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 2;
|
| - break;
|
| - }
|
| - case PDFOBJ_STREAM: {
|
| - CPDF_Stream* p = (CPDF_Stream*)pObj;
|
| - if (PDF_CreatorAppendObject(p->GetDict(), pFile, offset) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC("stream\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 8;
|
| - CPDF_StreamAcc acc;
|
| - acc.LoadAllData(p, TRUE);
|
| - if (pFile->AppendBlock(acc.GetData(), acc.GetSize()) < 0) {
|
| - return -1;
|
| - }
|
| - offset += acc.GetSize();
|
| - if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - break;
|
| - }
|
| - default:
|
| - ASSERT(FALSE);
|
| - break;
|
| - }
|
| +int32_t PDF_CreatorAppendObject(const CPDF_Object* pObj,
|
| + CFX_FileBufferArchive* pFile,
|
| + FX_FILESIZE& offset) {
|
| + int32_t len = 0;
|
| + if (pObj == NULL) {
|
| + if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 5;
|
| return 1;
|
| -}
|
| -int32_t PDF_CreatorWriteTrailer(CPDF_Document* pDocument, CFX_FileBufferArchive* pFile, CPDF_Array* pIDArray, FX_BOOL bCompress)
|
| -{
|
| - FX_FILESIZE offset = 0;
|
| - int32_t len = 0;
|
| - FXSYS_assert(pDocument && pFile);
|
| - CPDF_Parser *pParser = (CPDF_Parser*)pDocument->GetParser();
|
| - if (pParser) {
|
| - CPDF_Dictionary* p = pParser->GetTrailer();
|
| - FX_POSITION pos = p->GetStartPos();
|
| - while (pos) {
|
| - CFX_ByteString key;
|
| - CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| - if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") || key == FX_BSTRC("Filter") ||
|
| - key == FX_BSTRC("Index") || key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
|
| - key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") || key == FX_BSTRC("Type") || key == FX_BSTRC("ID")) {
|
| - continue;
|
| - }
|
| - if (bCompress && key == FX_BSTRC("DecodeParms")) {
|
| - continue;
|
| - }
|
| - if (pFile->AppendString((FX_BSTRC("/"))) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 1;
|
| - if (pValue->GetObjNum()) {
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + 6;
|
| - } else {
|
| - if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - if (pIDArray) {
|
| - if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 3;
|
| - if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - return offset;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
|
| + }
|
| + switch (pObj->GetType()) {
|
| + case PDFOBJ_NULL:
|
| + if (pFile->AppendString(FX_BSTRC(" null")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 5;
|
| + break;
|
| + case PDFOBJ_BOOLEAN:
|
| + case PDFOBJ_NUMBER:
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendString(pObj->GetString())) < 0) {
|
| return -1;
|
| + }
|
| + offset += len + 1;
|
| + break;
|
| + case PDFOBJ_STRING: {
|
| + CFX_ByteString str = pObj->GetString();
|
| + FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
|
| + if ((len = pFile->AppendString(PDF_EncodeString(str, bHex))) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + break;
|
| }
|
| - if ((len = pFile->AppendDWord(pDocument->GetRoot()->GetObjNum())) < 0) {
|
| + case PDFOBJ_NAME: {
|
| + if (pFile->AppendString(FX_BSTRC("/")) < 0) {
|
| + return -1;
|
| + }
|
| + CFX_ByteString str = pObj->GetString();
|
| + if ((len = pFile->AppendString(PDF_NameEncode(str))) < 0) {
|
| return -1;
|
| + }
|
| + offset += len + 1;
|
| + break;
|
| }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + case PDFOBJ_REFERENCE: {
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| return -1;
|
| + }
|
| + CPDF_Reference* p = (CPDF_Reference*)pObj;
|
| + if ((len = pFile->AppendDWord(p->GetRefObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 6;
|
| + break;
|
| }
|
| - offset += len + 14;
|
| - if (pDocument->GetInfo()) {
|
| - if (pFile->AppendString(FX_BSTRC("/Info ")) < 0) {
|
| + case PDFOBJ_ARRAY: {
|
| + if (pFile->AppendString(FX_BSTRC("[")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 1;
|
| + CPDF_Array* p = (CPDF_Array*)pObj;
|
| + for (FX_DWORD i = 0; i < p->GetCount(); i++) {
|
| + CPDF_Object* pElement = p->GetElement(i);
|
| + if (pElement->GetObjNum()) {
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| return -1;
|
| - }
|
| - if ((len = pFile->AppendDWord(pDocument->GetInfo()->GetObjNum())) < 0) {
|
| + }
|
| + if ((len = pFile->AppendDWord(pElement->GetObjNum())) < 0) {
|
| return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 5;
|
| + } else {
|
| + if (PDF_CreatorAppendObject(pElement, pFile, offset) < 0) {
|
| return -1;
|
| + }
|
| }
|
| - offset += len + 12;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("]")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 1;
|
| + break;
|
| }
|
| - if (pIDArray) {
|
| - if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
|
| - return -1;
|
| + case PDFOBJ_DICTIONARY: {
|
| + if (pFile->AppendString(FX_BSTRC("<<")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 2;
|
| + CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
|
| + FX_POSITION pos = p->GetStartPos();
|
| + while (pos) {
|
| + CFX_ByteString key;
|
| + CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| + if (pFile->AppendString(FX_BSTRC("/")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
|
| + return -1;
|
| }
|
| - offset += 3;
|
| - if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
|
| + offset += len + 1;
|
| + if (pValue->GetObjNum()) {
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 5;
|
| + } else {
|
| + if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
|
| return -1;
|
| + }
|
| }
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(">>")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 2;
|
| + break;
|
| }
|
| - return offset;
|
| + case PDFOBJ_STREAM: {
|
| + CPDF_Stream* p = (CPDF_Stream*)pObj;
|
| + if (PDF_CreatorAppendObject(p->GetDict(), pFile, offset) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("stream\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 8;
|
| + CPDF_StreamAcc acc;
|
| + acc.LoadAllData(p, TRUE);
|
| + if (pFile->AppendBlock(acc.GetData(), acc.GetSize()) < 0) {
|
| + return -1;
|
| + }
|
| + offset += acc.GetSize();
|
| + if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + break;
|
| + }
|
| + default:
|
| + ASSERT(FALSE);
|
| + break;
|
| + }
|
| + return 1;
|
| }
|
| -int32_t PDF_CreatorWriteEncrypt(const CPDF_Dictionary* pEncryptDict, FX_DWORD dwObjNum, CFX_FileBufferArchive *pFile)
|
| -{
|
| - if (!pEncryptDict) {
|
| - return 0;
|
| - }
|
| - FXSYS_assert(pFile);
|
| - FX_FILESIZE offset = 0;
|
| - int32_t len = 0;
|
| - if (pFile->AppendString(FX_BSTRC("/Encrypt")) < 0) {
|
| +int32_t PDF_CreatorWriteTrailer(CPDF_Document* pDocument,
|
| + CFX_FileBufferArchive* pFile,
|
| + CPDF_Array* pIDArray,
|
| + FX_BOOL bCompress) {
|
| + FX_FILESIZE offset = 0;
|
| + int32_t len = 0;
|
| + FXSYS_assert(pDocument && pFile);
|
| + CPDF_Parser* pParser = (CPDF_Parser*)pDocument->GetParser();
|
| + if (pParser) {
|
| + CPDF_Dictionary* p = pParser->GetTrailer();
|
| + FX_POSITION pos = p->GetStartPos();
|
| + while (pos) {
|
| + CFX_ByteString key;
|
| + CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| + if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") ||
|
| + key == FX_BSTRC("Filter") || key == FX_BSTRC("Index") ||
|
| + key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
|
| + key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") ||
|
| + key == FX_BSTRC("Type") || key == FX_BSTRC("ID")) {
|
| + continue;
|
| + }
|
| + if (bCompress && key == FX_BSTRC("DecodeParms")) {
|
| + continue;
|
| + }
|
| + if (pFile->AppendString((FX_BSTRC("/"))) < 0) {
|
| return -1;
|
| - }
|
| - offset += 8;
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| + }
|
| + if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) {
|
| return -1;
|
| + }
|
| + offset += len + 1;
|
| + if (pValue->GetObjNum()) {
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 6;
|
| + } else {
|
| + if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| }
|
| - if ((len = pFile->AppendDWord(dwObjNum)) < 0) {
|
| + if (pIDArray) {
|
| + if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
|
| return -1;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + }
|
| + offset += 3;
|
| + if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
|
| return -1;
|
| + }
|
| }
|
| - offset += len + 6;
|
| return offset;
|
| -}
|
| -FX_BOOL PDF_GenerateFileID(FX_DWORD dwSeed1, FX_DWORD dwSeed2, FX_DWORD* pBuffer)
|
| -{
|
| - if (!pBuffer) {
|
| - return FALSE;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord(pDocument->GetRoot()->GetObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 14;
|
| + if (pDocument->GetInfo()) {
|
| + if (pFile->AppendString(FX_BSTRC("/Info ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord(pDocument->GetInfo()->GetObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + return -1;
|
| }
|
| - void* pContext = FX_Random_MT_Start(dwSeed1);
|
| - int32_t i = 0;
|
| - for (i = 0; i < 2; i++) {
|
| - *pBuffer++ = FX_Random_MT_Generate(pContext);
|
| + offset += len + 12;
|
| + }
|
| + if (pIDArray) {
|
| + if (pFile->AppendString((FX_BSTRC("/ID"))) < 0) {
|
| + return -1;
|
| }
|
| - FX_Random_MT_Close(pContext);
|
| - pContext = FX_Random_MT_Start(dwSeed2);
|
| - for (i = 0; i < 2; i++) {
|
| - *pBuffer++ = FX_Random_MT_Generate(pContext);
|
| + offset += 3;
|
| + if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) {
|
| + return -1;
|
| }
|
| - FX_Random_MT_Close(pContext);
|
| - return TRUE;
|
| + }
|
| + return offset;
|
| }
|
| -class CPDF_FlateEncoder
|
| -{
|
| -public:
|
| - CPDF_FlateEncoder();
|
| - ~CPDF_FlateEncoder();
|
| - FX_BOOL Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode);
|
| - FX_BOOL Initialize(const uint8_t* pBuffer, FX_DWORD size, FX_BOOL bFlateEncode, FX_BOOL bXRefStream = FALSE);
|
| - void CloneDict();
|
| - uint8_t* m_pData;
|
| - FX_DWORD m_dwSize;
|
| - CPDF_Dictionary* m_pDict;
|
| - FX_BOOL m_bCloned;
|
| - FX_BOOL m_bNewData;
|
| - CPDF_StreamAcc m_Acc;
|
| -};
|
| -CPDF_FlateEncoder::CPDF_FlateEncoder()
|
| -{
|
| - m_pData = NULL;
|
| - m_dwSize = 0;
|
| - m_pDict = NULL;
|
| - m_bCloned = FALSE;
|
| - m_bNewData = FALSE;
|
| +int32_t PDF_CreatorWriteEncrypt(const CPDF_Dictionary* pEncryptDict,
|
| + FX_DWORD dwObjNum,
|
| + CFX_FileBufferArchive* pFile) {
|
| + if (!pEncryptDict) {
|
| + return 0;
|
| + }
|
| + FXSYS_assert(pFile);
|
| + FX_FILESIZE offset = 0;
|
| + int32_t len = 0;
|
| + if (pFile->AppendString(FX_BSTRC("/Encrypt")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += 8;
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord(dwObjNum)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 6;
|
| + return offset;
|
| }
|
| -void CPDF_FlateEncoder::CloneDict()
|
| -{
|
| - if (!m_bCloned) {
|
| - m_pDict = (CPDF_Dictionary*)m_pDict->Clone();
|
| - m_bCloned = TRUE;
|
| - }
|
| +FX_BOOL PDF_GenerateFileID(FX_DWORD dwSeed1,
|
| + FX_DWORD dwSeed2,
|
| + FX_DWORD* pBuffer) {
|
| + if (!pBuffer) {
|
| + return FALSE;
|
| + }
|
| + void* pContext = FX_Random_MT_Start(dwSeed1);
|
| + int32_t i = 0;
|
| + for (i = 0; i < 2; i++) {
|
| + *pBuffer++ = FX_Random_MT_Generate(pContext);
|
| + }
|
| + FX_Random_MT_Close(pContext);
|
| + pContext = FX_Random_MT_Start(dwSeed2);
|
| + for (i = 0; i < 2; i++) {
|
| + *pBuffer++ = FX_Random_MT_Generate(pContext);
|
| + }
|
| + FX_Random_MT_Close(pContext);
|
| + return TRUE;
|
| }
|
| -FX_BOOL CPDF_FlateEncoder::Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode)
|
| -{
|
| - m_Acc.LoadAllData(pStream, TRUE);
|
| - if ((pStream && pStream->GetDict() && pStream->GetDict()->KeyExist("Filter")) || !bFlateEncode) {
|
| - if (pStream->GetDict()->KeyExist("Filter") && !bFlateEncode) {
|
| - CPDF_StreamAcc destAcc;
|
| - destAcc.LoadAllData(pStream);
|
| - m_dwSize = destAcc.GetSize();
|
| - m_pData = (uint8_t*)destAcc.DetachData();
|
| - m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
|
| - m_pDict->RemoveAt(FX_BSTRC("Filter"));
|
| - m_bNewData = TRUE;
|
| - m_bCloned = TRUE;
|
| - } else {
|
| - m_pData = (uint8_t*)m_Acc.GetData();
|
| - m_dwSize = m_Acc.GetSize();
|
| - m_pDict = pStream->GetDict();
|
| - }
|
| - return TRUE;
|
| - }
|
| - m_pData = NULL;
|
| - m_dwSize = 0;
|
| - m_bNewData = TRUE;
|
| +class CPDF_FlateEncoder {
|
| + public:
|
| + CPDF_FlateEncoder();
|
| + ~CPDF_FlateEncoder();
|
| + FX_BOOL Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode);
|
| + FX_BOOL Initialize(const uint8_t* pBuffer,
|
| + FX_DWORD size,
|
| + FX_BOOL bFlateEncode,
|
| + FX_BOOL bXRefStream = FALSE);
|
| + void CloneDict();
|
| + uint8_t* m_pData;
|
| + FX_DWORD m_dwSize;
|
| + CPDF_Dictionary* m_pDict;
|
| + FX_BOOL m_bCloned;
|
| + FX_BOOL m_bNewData;
|
| + CPDF_StreamAcc m_Acc;
|
| +};
|
| +CPDF_FlateEncoder::CPDF_FlateEncoder() {
|
| + m_pData = NULL;
|
| + m_dwSize = 0;
|
| + m_pDict = NULL;
|
| + m_bCloned = FALSE;
|
| + m_bNewData = FALSE;
|
| +}
|
| +void CPDF_FlateEncoder::CloneDict() {
|
| + if (!m_bCloned) {
|
| + m_pDict = (CPDF_Dictionary*)m_pDict->Clone();
|
| m_bCloned = TRUE;
|
| - ::FlateEncode(m_Acc.GetData(), m_Acc.GetSize(), m_pData, m_dwSize);
|
| - m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
|
| - m_pDict->SetAtInteger("Length", m_dwSize);
|
| - m_pDict->SetAtName("Filter", "FlateDecode");
|
| - m_pDict->RemoveAt("DecodeParms");
|
| - return TRUE;
|
| + }
|
| }
|
| -FX_BOOL CPDF_FlateEncoder::Initialize(const uint8_t* pBuffer, FX_DWORD size, FX_BOOL bFlateEncode, FX_BOOL bXRefStream)
|
| -{
|
| - if (!bFlateEncode) {
|
| - m_pData = (uint8_t*)pBuffer;
|
| - m_dwSize = size;
|
| - return TRUE;
|
| - }
|
| - m_bNewData = TRUE;
|
| - if (bXRefStream) {
|
| - ::FlateEncode(pBuffer, size, 12, 1, 8, 7, m_pData, m_dwSize);
|
| +FX_BOOL CPDF_FlateEncoder::Initialize(CPDF_Stream* pStream,
|
| + FX_BOOL bFlateEncode) {
|
| + m_Acc.LoadAllData(pStream, TRUE);
|
| + if ((pStream && pStream->GetDict() &&
|
| + pStream->GetDict()->KeyExist("Filter")) ||
|
| + !bFlateEncode) {
|
| + if (pStream->GetDict()->KeyExist("Filter") && !bFlateEncode) {
|
| + CPDF_StreamAcc destAcc;
|
| + destAcc.LoadAllData(pStream);
|
| + m_dwSize = destAcc.GetSize();
|
| + m_pData = (uint8_t*)destAcc.DetachData();
|
| + m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
|
| + m_pDict->RemoveAt(FX_BSTRC("Filter"));
|
| + m_bNewData = TRUE;
|
| + m_bCloned = TRUE;
|
| } else {
|
| - ::FlateEncode(pBuffer, size, m_pData, m_dwSize);
|
| + m_pData = (uint8_t*)m_Acc.GetData();
|
| + m_dwSize = m_Acc.GetSize();
|
| + m_pDict = pStream->GetDict();
|
| }
|
| return TRUE;
|
| + }
|
| + m_pData = NULL;
|
| + m_dwSize = 0;
|
| + m_bNewData = TRUE;
|
| + m_bCloned = TRUE;
|
| + ::FlateEncode(m_Acc.GetData(), m_Acc.GetSize(), m_pData, m_dwSize);
|
| + m_pDict = (CPDF_Dictionary*)pStream->GetDict()->Clone();
|
| + m_pDict->SetAtInteger("Length", m_dwSize);
|
| + m_pDict->SetAtName("Filter", "FlateDecode");
|
| + m_pDict->RemoveAt("DecodeParms");
|
| + return TRUE;
|
| }
|
| -CPDF_FlateEncoder::~CPDF_FlateEncoder()
|
| -{
|
| - if (m_bCloned && m_pDict) {
|
| - m_pDict->Release();
|
| - }
|
| - if (m_bNewData && m_pData) {
|
| - FX_Free(m_pData);
|
| - }
|
| +FX_BOOL CPDF_FlateEncoder::Initialize(const uint8_t* pBuffer,
|
| + FX_DWORD size,
|
| + FX_BOOL bFlateEncode,
|
| + FX_BOOL bXRefStream) {
|
| + if (!bFlateEncode) {
|
| + m_pData = (uint8_t*)pBuffer;
|
| + m_dwSize = size;
|
| + return TRUE;
|
| + }
|
| + m_bNewData = TRUE;
|
| + if (bXRefStream) {
|
| + ::FlateEncode(pBuffer, size, 12, 1, 8, 7, m_pData, m_dwSize);
|
| + } else {
|
| + ::FlateEncode(pBuffer, size, m_pData, m_dwSize);
|
| + }
|
| + return TRUE;
|
| +}
|
| +CPDF_FlateEncoder::~CPDF_FlateEncoder() {
|
| + if (m_bCloned && m_pDict) {
|
| + m_pDict->Release();
|
| + }
|
| + if (m_bNewData && m_pData) {
|
| + FX_Free(m_pData);
|
| + }
|
| }
|
| -class CPDF_Encryptor
|
| -{
|
| -public:
|
| - CPDF_Encryptor();
|
| - ~CPDF_Encryptor();
|
| - FX_BOOL Initialize(CPDF_CryptoHandler* pHandler, int objnum, uint8_t* src_data, FX_DWORD src_size);
|
| - uint8_t* m_pData;
|
| - FX_DWORD m_dwSize;
|
| - FX_BOOL m_bNewBuf;
|
| +class CPDF_Encryptor {
|
| + public:
|
| + CPDF_Encryptor();
|
| + ~CPDF_Encryptor();
|
| + FX_BOOL Initialize(CPDF_CryptoHandler* pHandler,
|
| + int objnum,
|
| + uint8_t* src_data,
|
| + FX_DWORD src_size);
|
| + uint8_t* m_pData;
|
| + FX_DWORD m_dwSize;
|
| + FX_BOOL m_bNewBuf;
|
| };
|
| -CPDF_Encryptor::CPDF_Encryptor()
|
| -{
|
| - m_pData = NULL;
|
| - m_dwSize = 0;
|
| - m_bNewBuf = FALSE;
|
| +CPDF_Encryptor::CPDF_Encryptor() {
|
| + m_pData = NULL;
|
| + m_dwSize = 0;
|
| + m_bNewBuf = FALSE;
|
| }
|
| -FX_BOOL CPDF_Encryptor::Initialize(CPDF_CryptoHandler* pHandler, int objnum, uint8_t* src_data, FX_DWORD src_size)
|
| -{
|
| - if (src_size == 0) {
|
| - return TRUE;
|
| - }
|
| - if (pHandler == NULL) {
|
| - m_pData = (uint8_t*)src_data;
|
| - m_dwSize = src_size;
|
| - m_bNewBuf = FALSE;
|
| - return TRUE;
|
| - }
|
| - m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size);
|
| - m_pData = FX_Alloc(uint8_t, m_dwSize);
|
| - pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize);
|
| - m_bNewBuf = TRUE;
|
| +FX_BOOL CPDF_Encryptor::Initialize(CPDF_CryptoHandler* pHandler,
|
| + int objnum,
|
| + uint8_t* src_data,
|
| + FX_DWORD src_size) {
|
| + if (src_size == 0) {
|
| return TRUE;
|
| + }
|
| + if (pHandler == NULL) {
|
| + m_pData = (uint8_t*)src_data;
|
| + m_dwSize = src_size;
|
| + m_bNewBuf = FALSE;
|
| + return TRUE;
|
| + }
|
| + m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size);
|
| + m_pData = FX_Alloc(uint8_t, m_dwSize);
|
| + pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize);
|
| + m_bNewBuf = TRUE;
|
| + return TRUE;
|
| }
|
| -CPDF_Encryptor::~CPDF_Encryptor()
|
| -{
|
| - if (m_bNewBuf) {
|
| - FX_Free(m_pData);
|
| - }
|
| -}
|
| -CPDF_ObjectStream::CPDF_ObjectStream()
|
| - : m_dwObjNum(0)
|
| - , m_index(0)
|
| -{
|
| +CPDF_Encryptor::~CPDF_Encryptor() {
|
| + if (m_bNewBuf) {
|
| + FX_Free(m_pData);
|
| + }
|
| }
|
| -FX_BOOL CPDF_ObjectStream::Start()
|
| -{
|
| - m_ObjNumArray.RemoveAll();
|
| - m_OffsetArray.RemoveAll();
|
| - m_Buffer.Clear();
|
| - m_dwObjNum = 0;
|
| - m_index = 0;
|
| - return TRUE;
|
| +CPDF_ObjectStream::CPDF_ObjectStream() : m_dwObjNum(0), m_index(0) {}
|
| +FX_BOOL CPDF_ObjectStream::Start() {
|
| + m_ObjNumArray.RemoveAll();
|
| + m_OffsetArray.RemoveAll();
|
| + m_Buffer.Clear();
|
| + m_dwObjNum = 0;
|
| + m_index = 0;
|
| + return TRUE;
|
| }
|
| -int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, const CPDF_Object *pObj)
|
| -{
|
| - m_ObjNumArray.Add(dwObjNum);
|
| - m_OffsetArray.Add(m_Buffer.GetLength());
|
| - m_Buffer << pObj;
|
| - return 1;
|
| +int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum,
|
| + const CPDF_Object* pObj) {
|
| + m_ObjNumArray.Add(dwObjNum);
|
| + m_OffsetArray.Add(m_Buffer.GetLength());
|
| + m_Buffer << pObj;
|
| + return 1;
|
| }
|
| -int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, const uint8_t* pBuffer, FX_DWORD dwSize)
|
| -{
|
| - m_ObjNumArray.Add(dwObjNum);
|
| - m_OffsetArray.Add(m_Buffer.GetLength());
|
| - m_Buffer.AppendBlock(pBuffer, dwSize);
|
| - return 1;
|
| +int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum,
|
| + const uint8_t* pBuffer,
|
| + FX_DWORD dwSize) {
|
| + m_ObjNumArray.Add(dwObjNum);
|
| + m_OffsetArray.Add(m_Buffer.GetLength());
|
| + m_Buffer.AppendBlock(pBuffer, dwSize);
|
| + return 1;
|
| }
|
| -FX_FILESIZE CPDF_ObjectStream::End(CPDF_Creator* pCreator)
|
| -{
|
| - FXSYS_assert(pCreator);
|
| - if (m_ObjNumArray.GetSize() == 0) {
|
| - return 0;
|
| - }
|
| - CFX_FileBufferArchive *pFile = &pCreator->m_File;
|
| - CPDF_CryptoHandler *pHandler = pCreator->m_pCryptoHandler;
|
| - FX_FILESIZE ObjOffset = pCreator->m_Offset;
|
| - if (!m_dwObjNum) {
|
| - m_dwObjNum = ++pCreator->m_dwLastObjNum;
|
| - }
|
| - CFX_ByteTextBuf tempBuffer;
|
| - int32_t iCount = m_ObjNumArray.GetSize();
|
| - for (int32_t i = 0; i < iCount; i++) {
|
| - tempBuffer << m_ObjNumArray.ElementAt(i) << FX_BSTRC(" ") << m_OffsetArray.ElementAt(i) << FX_BSTRC(" ");
|
| - }
|
| - FX_FILESIZE &offset = pCreator->m_Offset;
|
| - int32_t len = pFile->AppendDWord(m_dwObjNum);
|
| - if (len < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - if ((len = pFile->AppendString(FX_BSTRC(" 0 obj\r\n<</Type /ObjStm /N "))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - if ((len = pFile->AppendDWord((FX_DWORD)iCount)) < 0) {
|
| - return -1;
|
| +FX_FILESIZE CPDF_ObjectStream::End(CPDF_Creator* pCreator) {
|
| + FXSYS_assert(pCreator);
|
| + if (m_ObjNumArray.GetSize() == 0) {
|
| + return 0;
|
| + }
|
| + CFX_FileBufferArchive* pFile = &pCreator->m_File;
|
| + CPDF_CryptoHandler* pHandler = pCreator->m_pCryptoHandler;
|
| + FX_FILESIZE ObjOffset = pCreator->m_Offset;
|
| + if (!m_dwObjNum) {
|
| + m_dwObjNum = ++pCreator->m_dwLastObjNum;
|
| + }
|
| + CFX_ByteTextBuf tempBuffer;
|
| + int32_t iCount = m_ObjNumArray.GetSize();
|
| + for (int32_t i = 0; i < iCount; i++) {
|
| + tempBuffer << m_ObjNumArray.ElementAt(i) << FX_BSTRC(" ")
|
| + << m_OffsetArray.ElementAt(i) << FX_BSTRC(" ");
|
| + }
|
| + FX_FILESIZE& offset = pCreator->m_Offset;
|
| + int32_t len = pFile->AppendDWord(m_dwObjNum);
|
| + if (len < 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + if ((len = pFile->AppendString(FX_BSTRC(" 0 obj\r\n<</Type /ObjStm /N "))) <
|
| + 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + if ((len = pFile->AppendDWord((FX_DWORD)iCount)) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + if (pFile->AppendString(FX_BSTRC("/First ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = pFile->AppendDWord((FX_DWORD)tempBuffer.GetLength())) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + 15;
|
| + if (!pCreator->m_bCompress && !pHandler) {
|
| + if ((len = pFile->AppendDWord(
|
| + (FX_DWORD)(tempBuffer.GetLength() + m_Buffer.GetLength()))) < 0) {
|
| + return -1;
|
| }
|
| offset += len;
|
| - if (pFile->AppendString(FX_BSTRC("/First ")) < 0) {
|
| - return -1;
|
| + if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| + return -1;
|
| }
|
| - if ((len = pFile->AppendDWord((FX_DWORD)tempBuffer.GetLength())) < 0) {
|
| - return -1;
|
| + if (pFile->AppendBlock(tempBuffer.GetBuffer(), tempBuffer.GetLength()) <
|
| + 0) {
|
| + return -1;
|
| }
|
| - if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
|
| - return -1;
|
| + if (pFile->AppendBlock(m_Buffer.GetBuffer(), m_Buffer.GetLength()) < 0) {
|
| + return -1;
|
| }
|
| - offset += len + 15;
|
| - if (!pCreator->m_bCompress && !pHandler) {
|
| - if ((len = pFile->AppendDWord((FX_DWORD)(tempBuffer.GetLength() + m_Buffer.GetLength()))) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendBlock(tempBuffer.GetBuffer(), tempBuffer.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendBlock(m_Buffer.GetBuffer(), m_Buffer.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + tempBuffer.GetLength() + m_Buffer.GetLength();
|
| - } else {
|
| - tempBuffer << m_Buffer;
|
| - CPDF_FlateEncoder encoder;
|
| - encoder.Initialize(tempBuffer.GetBuffer(), tempBuffer.GetLength(), pCreator->m_bCompress);
|
| - CPDF_Encryptor encryptor;
|
| - encryptor.Initialize(pHandler, m_dwObjNum, encoder.m_pData, encoder.m_dwSize);
|
| - if ((len = pFile->AppendDWord(encryptor.m_dwSize)) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len;
|
| - if (pCreator->m_bCompress) {
|
| - if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
|
| - return -1;
|
| - }
|
| - offset += 20;
|
| - }
|
| - if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| - return -1;
|
| - }
|
| - offset += len + encryptor.m_dwSize;
|
| + offset += len + tempBuffer.GetLength() + m_Buffer.GetLength();
|
| + } else {
|
| + tempBuffer << m_Buffer;
|
| + CPDF_FlateEncoder encoder;
|
| + encoder.Initialize(tempBuffer.GetBuffer(), tempBuffer.GetLength(),
|
| + pCreator->m_bCompress);
|
| + CPDF_Encryptor encryptor;
|
| + encryptor.Initialize(pHandler, m_dwObjNum, encoder.m_pData,
|
| + encoder.m_dwSize);
|
| + if ((len = pFile->AppendDWord(encryptor.m_dwSize)) < 0) {
|
| + return -1;
|
| }
|
| - if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) < 0) {
|
| + offset += len;
|
| + if (pCreator->m_bCompress) {
|
| + if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
|
| return -1;
|
| + }
|
| + offset += 20;
|
| }
|
| - offset += len;
|
| - return ObjOffset;
|
| + if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| + return -1;
|
| + }
|
| + offset += len + encryptor.m_dwSize;
|
| + }
|
| + if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) <
|
| + 0) {
|
| + return -1;
|
| + }
|
| + offset += len;
|
| + return ObjOffset;
|
| }
|
| CPDF_XRefStream::CPDF_XRefStream()
|
| - : m_PrevOffset(0)
|
| - , m_dwTempObjNum(0)
|
| - , m_iSeg(0)
|
| -{
|
| -}
|
| -FX_BOOL CPDF_XRefStream::Start()
|
| -{
|
| - m_IndexArray.RemoveAll();
|
| - m_Buffer.Clear();
|
| - m_iSeg = 0;
|
| - return TRUE;
|
| + : m_PrevOffset(0), m_dwTempObjNum(0), m_iSeg(0) {}
|
| +FX_BOOL CPDF_XRefStream::Start() {
|
| + m_IndexArray.RemoveAll();
|
| + m_Buffer.Clear();
|
| + m_iSeg = 0;
|
| + return TRUE;
|
| }
|
| -int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, const CPDF_Object *pObj, CPDF_Creator *pCreator)
|
| -{
|
| - if (!pCreator) {
|
| - return 0;
|
| - }
|
| - m_ObjStream.CompressIndirectObject(dwObjNum, pObj);
|
| - if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
|
| - m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
|
| - return 1;
|
| - }
|
| - return EndObjectStream(pCreator);
|
| +int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum,
|
| + const CPDF_Object* pObj,
|
| + CPDF_Creator* pCreator) {
|
| + if (!pCreator) {
|
| + return 0;
|
| + }
|
| + m_ObjStream.CompressIndirectObject(dwObjNum, pObj);
|
| + if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
|
| + m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
|
| + return 1;
|
| + }
|
| + return EndObjectStream(pCreator);
|
| }
|
| -int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, const uint8_t* pBuffer, FX_DWORD dwSize, CPDF_Creator *pCreator)
|
| -{
|
| - if (!pCreator) {
|
| - return 0;
|
| - }
|
| - m_ObjStream.CompressIndirectObject(dwObjNum, pBuffer, dwSize);
|
| - if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
|
| - m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
|
| - return 1;
|
| - }
|
| - return EndObjectStream(pCreator);
|
| +int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum,
|
| + const uint8_t* pBuffer,
|
| + FX_DWORD dwSize,
|
| + CPDF_Creator* pCreator) {
|
| + if (!pCreator) {
|
| + return 0;
|
| + }
|
| + m_ObjStream.CompressIndirectObject(dwObjNum, pBuffer, dwSize);
|
| + if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize &&
|
| + m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) {
|
| + return 1;
|
| + }
|
| + return EndObjectStream(pCreator);
|
| }
|
| -static void _AppendIndex0(CFX_ByteTextBuf& buffer, FX_BOOL bFirstObject = TRUE)
|
| -{
|
| - buffer.AppendByte(0);
|
| - buffer.AppendByte(0);
|
| +static void _AppendIndex0(CFX_ByteTextBuf& buffer,
|
| + FX_BOOL bFirstObject = TRUE) {
|
| + buffer.AppendByte(0);
|
| + buffer.AppendByte(0);
|
| + buffer.AppendByte(0);
|
| + buffer.AppendByte(0);
|
| + buffer.AppendByte(0);
|
| + if (bFirstObject) {
|
| + buffer.AppendByte(0xFF);
|
| + buffer.AppendByte(0xFF);
|
| + } else {
|
| buffer.AppendByte(0);
|
| buffer.AppendByte(0);
|
| - buffer.AppendByte(0);
|
| - if (bFirstObject) {
|
| - buffer.AppendByte(0xFF);
|
| - buffer.AppendByte(0xFF);
|
| - } else {
|
| - buffer.AppendByte(0);
|
| - buffer.AppendByte(0);
|
| - }
|
| + }
|
| }
|
| -static void _AppendIndex1(CFX_ByteTextBuf& buffer, FX_FILESIZE offset)
|
| -{
|
| - buffer.AppendByte(1);
|
| - buffer.AppendByte(FX_GETBYTEOFFSET24(offset));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET16(offset));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET8(offset));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET0(offset));
|
| - buffer.AppendByte(0);
|
| - buffer.AppendByte(0);
|
| +static void _AppendIndex1(CFX_ByteTextBuf& buffer, FX_FILESIZE offset) {
|
| + buffer.AppendByte(1);
|
| + buffer.AppendByte(FX_GETBYTEOFFSET24(offset));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET16(offset));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET8(offset));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET0(offset));
|
| + buffer.AppendByte(0);
|
| + buffer.AppendByte(0);
|
| }
|
| -static void _AppendIndex2(CFX_ByteTextBuf& buffer, FX_DWORD objnum, int32_t index)
|
| -{
|
| - buffer.AppendByte(2);
|
| - buffer.AppendByte(FX_GETBYTEOFFSET24(objnum));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET16(objnum));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET8(objnum));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET0(objnum));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET8(index));
|
| - buffer.AppendByte(FX_GETBYTEOFFSET0(index));
|
| +static void _AppendIndex2(CFX_ByteTextBuf& buffer,
|
| + FX_DWORD objnum,
|
| + int32_t index) {
|
| + buffer.AppendByte(2);
|
| + buffer.AppendByte(FX_GETBYTEOFFSET24(objnum));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET16(objnum));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET8(objnum));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET0(objnum));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET8(index));
|
| + buffer.AppendByte(FX_GETBYTEOFFSET0(index));
|
| }
|
| -int32_t CPDF_XRefStream::EndObjectStream(CPDF_Creator *pCreator, FX_BOOL bEOF)
|
| -{
|
| - FX_FILESIZE objOffset = 0;
|
| - if (bEOF) {
|
| - objOffset = m_ObjStream.End(pCreator);
|
| - if (objOffset < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - FX_DWORD &dwObjStmNum = m_ObjStream.m_dwObjNum;
|
| - if (!dwObjStmNum) {
|
| - dwObjStmNum = ++pCreator->m_dwLastObjNum;
|
| - }
|
| - int32_t iSize = m_ObjStream.m_ObjNumArray.GetSize();
|
| - int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| - if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
|
| - if (m_dwTempObjNum == 0) {
|
| - _AppendIndex0(m_Buffer);
|
| - m_dwTempObjNum++;
|
| - }
|
| - FX_DWORD end_num = m_IndexArray.GetAt((iSeg - 1) * 2) + m_IndexArray.GetAt((iSeg - 1) * 2 + 1);
|
| - int index = 0;
|
| - for (; m_dwTempObjNum < end_num; m_dwTempObjNum++) {
|
| - FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
|
| - if (offset) {
|
| - if (index >= iSize || m_dwTempObjNum != m_ObjStream.m_ObjNumArray[index]) {
|
| - _AppendIndex1(m_Buffer, *offset);
|
| - } else {
|
| - _AppendIndex2(m_Buffer, dwObjStmNum, index++);
|
| - }
|
| - } else {
|
| - _AppendIndex0(m_Buffer, FALSE);
|
| - }
|
| - }
|
| - if (iSize > 0 && bEOF) {
|
| - pCreator->m_ObjectOffset.Add(dwObjStmNum, 1);
|
| - pCreator->m_ObjectSize.Add(dwObjStmNum, 1);
|
| - pCreator->m_ObjectOffset[dwObjStmNum] = objOffset;
|
| - }
|
| - m_iSeg = iSeg;
|
| - if (bEOF) {
|
| - m_ObjStream.Start();
|
| - }
|
| - return 1;
|
| - }
|
| - int32_t &j = m_ObjStream.m_index;
|
| - for (int i = m_iSeg; i < iSeg; i++) {
|
| - FX_DWORD start = m_IndexArray.ElementAt(i * 2);
|
| - FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
|
| - for (FX_DWORD m = start; m < end; m++) {
|
| - if (j >= iSize || m != m_ObjStream.m_ObjNumArray.ElementAt(j)) {
|
| - _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[m]);
|
| - } else {
|
| - _AppendIndex2(m_Buffer, dwObjStmNum, j++);
|
| - }
|
| +int32_t CPDF_XRefStream::EndObjectStream(CPDF_Creator* pCreator, FX_BOOL bEOF) {
|
| + FX_FILESIZE objOffset = 0;
|
| + if (bEOF) {
|
| + objOffset = m_ObjStream.End(pCreator);
|
| + if (objOffset < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| + FX_DWORD& dwObjStmNum = m_ObjStream.m_dwObjNum;
|
| + if (!dwObjStmNum) {
|
| + dwObjStmNum = ++pCreator->m_dwLastObjNum;
|
| + }
|
| + int32_t iSize = m_ObjStream.m_ObjNumArray.GetSize();
|
| + int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| + if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
|
| + if (m_dwTempObjNum == 0) {
|
| + _AppendIndex0(m_Buffer);
|
| + m_dwTempObjNum++;
|
| + }
|
| + FX_DWORD end_num = m_IndexArray.GetAt((iSeg - 1) * 2) +
|
| + m_IndexArray.GetAt((iSeg - 1) * 2 + 1);
|
| + int index = 0;
|
| + for (; m_dwTempObjNum < end_num; m_dwTempObjNum++) {
|
| + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
|
| + if (offset) {
|
| + if (index >= iSize ||
|
| + m_dwTempObjNum != m_ObjStream.m_ObjNumArray[index]) {
|
| + _AppendIndex1(m_Buffer, *offset);
|
| + } else {
|
| + _AppendIndex2(m_Buffer, dwObjStmNum, index++);
|
| }
|
| + } else {
|
| + _AppendIndex0(m_Buffer, FALSE);
|
| + }
|
| }
|
| if (iSize > 0 && bEOF) {
|
| - _AppendIndex1(m_Buffer, objOffset);
|
| - m_IndexArray.Add(dwObjStmNum);
|
| - m_IndexArray.Add(1);
|
| - iSeg += 1;
|
| + pCreator->m_ObjectOffset.Add(dwObjStmNum, 1);
|
| + pCreator->m_ObjectSize.Add(dwObjStmNum, 1);
|
| + pCreator->m_ObjectOffset[dwObjStmNum] = objOffset;
|
| }
|
| m_iSeg = iSeg;
|
| if (bEOF) {
|
| - m_ObjStream.Start();
|
| + m_ObjStream.Start();
|
| }
|
| return 1;
|
| + }
|
| + int32_t& j = m_ObjStream.m_index;
|
| + for (int i = m_iSeg; i < iSeg; i++) {
|
| + FX_DWORD start = m_IndexArray.ElementAt(i * 2);
|
| + FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
|
| + for (FX_DWORD m = start; m < end; m++) {
|
| + if (j >= iSize || m != m_ObjStream.m_ObjNumArray.ElementAt(j)) {
|
| + _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[m]);
|
| + } else {
|
| + _AppendIndex2(m_Buffer, dwObjStmNum, j++);
|
| + }
|
| + }
|
| + }
|
| + if (iSize > 0 && bEOF) {
|
| + _AppendIndex1(m_Buffer, objOffset);
|
| + m_IndexArray.Add(dwObjStmNum);
|
| + m_IndexArray.Add(1);
|
| + iSeg += 1;
|
| + }
|
| + m_iSeg = iSeg;
|
| + if (bEOF) {
|
| + m_ObjStream.Start();
|
| + }
|
| + return 1;
|
| }
|
| -FX_BOOL CPDF_XRefStream::GenerateXRefStream(CPDF_Creator* pCreator, FX_BOOL bEOF)
|
| -{
|
| - FX_FILESIZE offset_tmp = pCreator->m_Offset;
|
| - FX_DWORD objnum = ++pCreator->m_dwLastObjNum;
|
| - CFX_FileBufferArchive *pFile = &pCreator->m_File;
|
| - FX_BOOL bIncremental = (pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
|
| - if (bIncremental) {
|
| - AddObjectNumberToIndexArray(objnum);
|
| - } else {
|
| - for (; m_dwTempObjNum < pCreator->m_dwLastObjNum; m_dwTempObjNum++) {
|
| - FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
|
| - if (offset) {
|
| - _AppendIndex1(m_Buffer, *offset);
|
| - } else {
|
| - _AppendIndex0(m_Buffer, FALSE);
|
| - }
|
| - }
|
| - }
|
| - _AppendIndex1(m_Buffer, offset_tmp);
|
| - FX_FILESIZE &offset = pCreator->m_Offset;
|
| - int32_t len = pFile->AppendDWord(objnum);
|
| - if (len < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len;
|
| - if ((len = pFile->AppendString(FX_BSTRC(" 0 obj\r\n<</Type /XRef/W[1 4 2]/Index["))) < 0) {
|
| - return FALSE;
|
| +FX_BOOL CPDF_XRefStream::GenerateXRefStream(CPDF_Creator* pCreator,
|
| + FX_BOOL bEOF) {
|
| + FX_FILESIZE offset_tmp = pCreator->m_Offset;
|
| + FX_DWORD objnum = ++pCreator->m_dwLastObjNum;
|
| + CFX_FileBufferArchive* pFile = &pCreator->m_File;
|
| + FX_BOOL bIncremental = (pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
|
| + if (bIncremental) {
|
| + AddObjectNumberToIndexArray(objnum);
|
| + } else {
|
| + for (; m_dwTempObjNum < pCreator->m_dwLastObjNum; m_dwTempObjNum++) {
|
| + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum);
|
| + if (offset) {
|
| + _AppendIndex1(m_Buffer, *offset);
|
| + } else {
|
| + _AppendIndex0(m_Buffer, FALSE);
|
| + }
|
| + }
|
| + }
|
| + _AppendIndex1(m_Buffer, offset_tmp);
|
| + FX_FILESIZE& offset = pCreator->m_Offset;
|
| + int32_t len = pFile->AppendDWord(objnum);
|
| + if (len < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len;
|
| + if ((len = pFile->AppendString(
|
| + FX_BSTRC(" 0 obj\r\n<</Type /XRef/W[1 4 2]/Index["))) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len;
|
| + if (!bIncremental) {
|
| + if ((len = pFile->AppendDWord(0)) < 0) {
|
| + return FALSE;
|
| + }
|
| + if ((len = pFile->AppendString(FX_BSTRC(" "))) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len + 1;
|
| + if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
|
| + return FALSE;
|
| }
|
| offset += len;
|
| - if (!bIncremental) {
|
| - if ((len = pFile->AppendDWord(0)) < 0) {
|
| - return FALSE;
|
| - }
|
| - if ((len = pFile->AppendString(FX_BSTRC(" "))) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len + 1;
|
| - if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len;
|
| - } else {
|
| - int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| - for (int32_t i = 0; i < iSeg; i++) {
|
| - if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2))) < 0) {
|
| - return FALSE;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len + 1;
|
| - if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2 + 1))) < 0) {
|
| - return FALSE;
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len + 1;
|
| - }
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC("]/Size ")) < 0) {
|
| + } else {
|
| + int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| + for (int32_t i = 0; i < iSeg; i++) {
|
| + if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2))) < 0) {
|
| return FALSE;
|
| - }
|
| - if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| return FALSE;
|
| - }
|
| - offset += len + 7;
|
| - if (m_PrevOffset > 0) {
|
| - if (pFile->AppendString(FX_BSTRC("/Prev ")) < 0) {
|
| - return FALSE;
|
| - }
|
| - FX_CHAR offset_buf[20];
|
| - FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| - FXSYS_i64toa(m_PrevOffset, offset_buf, 10);
|
| - int32_t len = (int32_t)FXSYS_strlen(offset_buf);
|
| - if (pFile->AppendBlock(offset_buf, len) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len + 6;
|
| - }
|
| - FX_BOOL bPredictor = TRUE;
|
| - CPDF_FlateEncoder encoder;
|
| - encoder.Initialize(m_Buffer.GetBuffer(), m_Buffer.GetLength(), pCreator->m_bCompress, bPredictor);
|
| - if (pCreator->m_bCompress) {
|
| - if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += 20;
|
| - if (bPredictor) {
|
| - if ((len = pFile->AppendString(FX_BSTRC("/DecodeParms<</Columns 7/Predictor 12>>"))) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len;
|
| - }
|
| - }
|
| - if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
|
| + }
|
| + offset += len + 1;
|
| + if ((len = pFile->AppendDWord(m_IndexArray.ElementAt(i * 2 + 1))) < 0) {
|
| return FALSE;
|
| - }
|
| - if ((len = pFile->AppendDWord(encoder.m_dwSize)) < 0) {
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC(" ")) < 0) {
|
| return FALSE;
|
| + }
|
| + offset += len + 1;
|
| + }
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("]/Size ")) < 0) {
|
| + return FALSE;
|
| + }
|
| + if ((len = pFile->AppendDWord(objnum + 1)) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len + 7;
|
| + if (m_PrevOffset > 0) {
|
| + if (pFile->AppendString(FX_BSTRC("/Prev ")) < 0) {
|
| + return FALSE;
|
| }
|
| - offset += len + 8;
|
| - if (bEOF) {
|
| - if ((len = PDF_CreatorWriteTrailer(pCreator->m_pDocument, pFile, pCreator->m_pIDArray, pCreator->m_bCompress)) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len;
|
| - if (pCreator->m_pEncryptDict) {
|
| - FX_DWORD dwEncryptObjNum = pCreator->m_pEncryptDict->GetObjNum();
|
| - if (dwEncryptObjNum == 0) {
|
| - dwEncryptObjNum = pCreator->m_dwEnryptObjNum;
|
| - }
|
| - if ((len = PDF_CreatorWriteEncrypt(pCreator->m_pEncryptDict, dwEncryptObjNum, pFile)) < 0) {
|
| - return FALSE;
|
| - }
|
| - offset += len;
|
| - }
|
| + FX_CHAR offset_buf[20];
|
| + FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| + FXSYS_i64toa(m_PrevOffset, offset_buf, 10);
|
| + int32_t len = (int32_t)FXSYS_strlen(offset_buf);
|
| + if (pFile->AppendBlock(offset_buf, len) < 0) {
|
| + return FALSE;
|
| }
|
| - if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| + offset += len + 6;
|
| + }
|
| + FX_BOOL bPredictor = TRUE;
|
| + CPDF_FlateEncoder encoder;
|
| + encoder.Initialize(m_Buffer.GetBuffer(), m_Buffer.GetLength(),
|
| + pCreator->m_bCompress, bPredictor);
|
| + if (pCreator->m_bCompress) {
|
| + if (pFile->AppendString(FX_BSTRC("/Filter /FlateDecode")) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += 20;
|
| + if (bPredictor) {
|
| + if ((len = pFile->AppendString(
|
| + FX_BSTRC("/DecodeParms<</Columns 7/Predictor 12>>"))) < 0) {
|
| return FALSE;
|
| + }
|
| + offset += len;
|
| + }
|
| + }
|
| + if (pFile->AppendString(FX_BSTRC("/Length ")) < 0) {
|
| + return FALSE;
|
| + }
|
| + if ((len = pFile->AppendDWord(encoder.m_dwSize)) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len + 8;
|
| + if (bEOF) {
|
| + if ((len = PDF_CreatorWriteTrailer(pCreator->m_pDocument, pFile,
|
| + pCreator->m_pIDArray,
|
| + pCreator->m_bCompress)) < 0) {
|
| + return FALSE;
|
| }
|
| offset += len;
|
| - if (pFile->AppendBlock(encoder.m_pData, encoder.m_dwSize) < 0) {
|
| - return FALSE;
|
| - }
|
| - if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) < 0) {
|
| + if (pCreator->m_pEncryptDict) {
|
| + FX_DWORD dwEncryptObjNum = pCreator->m_pEncryptDict->GetObjNum();
|
| + if (dwEncryptObjNum == 0) {
|
| + dwEncryptObjNum = pCreator->m_dwEnryptObjNum;
|
| + }
|
| + if ((len = PDF_CreatorWriteEncrypt(pCreator->m_pEncryptDict,
|
| + dwEncryptObjNum, pFile)) < 0) {
|
| return FALSE;
|
| - }
|
| - offset += encoder.m_dwSize + len;
|
| - m_PrevOffset = offset_tmp;
|
| - return TRUE;
|
| + }
|
| + offset += len;
|
| + }
|
| + }
|
| + if ((len = pFile->AppendString(FX_BSTRC(">>stream\r\n"))) < 0) {
|
| + return FALSE;
|
| + }
|
| + offset += len;
|
| + if (pFile->AppendBlock(encoder.m_pData, encoder.m_dwSize) < 0) {
|
| + return FALSE;
|
| + }
|
| + if ((len = pFile->AppendString(FX_BSTRC("\r\nendstream\r\nendobj\r\n"))) <
|
| + 0) {
|
| + return FALSE;
|
| + }
|
| + offset += encoder.m_dwSize + len;
|
| + m_PrevOffset = offset_tmp;
|
| + return TRUE;
|
| }
|
| -FX_BOOL CPDF_XRefStream::End(CPDF_Creator *pCreator, FX_BOOL bEOF )
|
| -{
|
| - if (EndObjectStream(pCreator, bEOF) < 0) {
|
| - return FALSE;
|
| - }
|
| - return GenerateXRefStream(pCreator, bEOF);
|
| +FX_BOOL CPDF_XRefStream::End(CPDF_Creator* pCreator, FX_BOOL bEOF) {
|
| + if (EndObjectStream(pCreator, bEOF) < 0) {
|
| + return FALSE;
|
| + }
|
| + return GenerateXRefStream(pCreator, bEOF);
|
| }
|
| -FX_BOOL CPDF_XRefStream::EndXRefStream(CPDF_Creator* pCreator)
|
| -{
|
| - if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
|
| - _AppendIndex0(m_Buffer);
|
| - for (FX_DWORD i = 1; i < pCreator->m_dwLastObjNum + 1; i++) {
|
| - FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(i);
|
| - if (offset) {
|
| - _AppendIndex1(m_Buffer, *offset);
|
| - } else {
|
| - _AppendIndex0(m_Buffer, FALSE);
|
| - }
|
| - }
|
| - } else {
|
| - int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| - for (int i = 0; i < iSeg; i++) {
|
| - FX_DWORD start = m_IndexArray.ElementAt(i * 2);
|
| - FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
|
| - for (FX_DWORD j = start; j < end; j++) {
|
| - _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[j]);
|
| - }
|
| - }
|
| - }
|
| - return GenerateXRefStream(pCreator, FALSE);
|
| +FX_BOOL CPDF_XRefStream::EndXRefStream(CPDF_Creator* pCreator) {
|
| + if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) {
|
| + _AppendIndex0(m_Buffer);
|
| + for (FX_DWORD i = 1; i < pCreator->m_dwLastObjNum + 1; i++) {
|
| + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(i);
|
| + if (offset) {
|
| + _AppendIndex1(m_Buffer, *offset);
|
| + } else {
|
| + _AppendIndex0(m_Buffer, FALSE);
|
| + }
|
| + }
|
| + } else {
|
| + int32_t iSeg = m_IndexArray.GetSize() / 2;
|
| + for (int i = 0; i < iSeg; i++) {
|
| + FX_DWORD start = m_IndexArray.ElementAt(i * 2);
|
| + FX_DWORD end = m_IndexArray.ElementAt(i * 2 + 1) + start;
|
| + for (FX_DWORD j = start; j < end; j++) {
|
| + _AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[j]);
|
| + }
|
| + }
|
| + }
|
| + return GenerateXRefStream(pCreator, FALSE);
|
| }
|
| -FX_BOOL CPDF_XRefStream::AddObjectNumberToIndexArray(FX_DWORD objnum)
|
| -{
|
| - int32_t iSize = m_IndexArray.GetSize();
|
| - if (iSize == 0) {
|
| - m_IndexArray.Add(objnum);
|
| - m_IndexArray.Add(1);
|
| +FX_BOOL CPDF_XRefStream::AddObjectNumberToIndexArray(FX_DWORD objnum) {
|
| + int32_t iSize = m_IndexArray.GetSize();
|
| + if (iSize == 0) {
|
| + m_IndexArray.Add(objnum);
|
| + m_IndexArray.Add(1);
|
| + } else {
|
| + FXSYS_assert(iSize > 1);
|
| + FX_DWORD startobjnum = m_IndexArray.ElementAt(iSize - 2);
|
| + int32_t iCount = m_IndexArray.ElementAt(iSize - 1);
|
| + if (objnum == startobjnum + iCount) {
|
| + m_IndexArray[iSize - 1] = iCount + 1;
|
| } else {
|
| - FXSYS_assert(iSize > 1);
|
| - FX_DWORD startobjnum = m_IndexArray.ElementAt(iSize - 2);
|
| - int32_t iCount = m_IndexArray.ElementAt(iSize - 1);
|
| - if (objnum == startobjnum + iCount) {
|
| - m_IndexArray[iSize - 1] = iCount + 1;
|
| - } else {
|
| - m_IndexArray.Add(objnum);
|
| - m_IndexArray.Add(1);
|
| - }
|
| + m_IndexArray.Add(objnum);
|
| + m_IndexArray.Add(1);
|
| }
|
| - return TRUE;
|
| + }
|
| + return TRUE;
|
| }
|
| -CPDF_Creator::CPDF_Creator(CPDF_Document* pDoc)
|
| -{
|
| - m_pDocument = pDoc;
|
| - m_pParser = (CPDF_Parser*)pDoc->m_pParser;
|
| - m_bCompress = TRUE;
|
| - if (m_pParser) {
|
| - m_pEncryptDict = m_pParser->GetEncryptDict();
|
| - m_pCryptoHandler = m_pParser->GetCryptoHandler();
|
| - } else {
|
| - m_pEncryptDict = NULL;
|
| - m_pCryptoHandler = NULL;
|
| - }
|
| - m_bSecurityChanged = FALSE;
|
| - m_bStandardSecurity = FALSE;
|
| - m_pMetadata = NULL;
|
| - m_bEncryptCloned = FALSE;
|
| - m_bEncryptMetadata = FALSE;
|
| - m_Offset = 0;
|
| - m_iStage = -1;
|
| - m_dwFlags = 0;
|
| - m_Pos = NULL;
|
| - m_XrefStart = 0;
|
| - m_pXRefStream = NULL;
|
| - m_ObjectStreamSize = 200;
|
| - m_dwLastObjNum = m_pDocument->GetLastObjNum();
|
| - m_pIDArray = NULL;
|
| - m_FileVersion = 0;
|
| - m_dwEnryptObjNum = 0;
|
| - m_bNewCrypto = FALSE;
|
| +CPDF_Creator::CPDF_Creator(CPDF_Document* pDoc) {
|
| + m_pDocument = pDoc;
|
| + m_pParser = (CPDF_Parser*)pDoc->m_pParser;
|
| + m_bCompress = TRUE;
|
| + if (m_pParser) {
|
| + m_pEncryptDict = m_pParser->GetEncryptDict();
|
| + m_pCryptoHandler = m_pParser->GetCryptoHandler();
|
| + } else {
|
| + m_pEncryptDict = NULL;
|
| + m_pCryptoHandler = NULL;
|
| + }
|
| + m_bSecurityChanged = FALSE;
|
| + m_bStandardSecurity = FALSE;
|
| + m_pMetadata = NULL;
|
| + m_bEncryptCloned = FALSE;
|
| + m_bEncryptMetadata = FALSE;
|
| + m_Offset = 0;
|
| + m_iStage = -1;
|
| + m_dwFlags = 0;
|
| + m_Pos = NULL;
|
| + m_XrefStart = 0;
|
| + m_pXRefStream = NULL;
|
| + m_ObjectStreamSize = 200;
|
| + m_dwLastObjNum = m_pDocument->GetLastObjNum();
|
| + m_pIDArray = NULL;
|
| + m_FileVersion = 0;
|
| + m_dwEnryptObjNum = 0;
|
| + m_bNewCrypto = FALSE;
|
| }
|
| -CPDF_Creator::~CPDF_Creator()
|
| -{
|
| - ResetStandardSecurity();
|
| - if (m_bEncryptCloned && m_pEncryptDict) {
|
| - m_pEncryptDict->Release();
|
| - m_pEncryptDict = NULL;
|
| - }
|
| - Clear();
|
| +CPDF_Creator::~CPDF_Creator() {
|
| + ResetStandardSecurity();
|
| + if (m_bEncryptCloned && m_pEncryptDict) {
|
| + m_pEncryptDict->Release();
|
| + m_pEncryptDict = NULL;
|
| + }
|
| + Clear();
|
| }
|
| -static FX_BOOL _IsXRefNeedEnd(CPDF_XRefStream* pXRef, FX_DWORD flag)
|
| -{
|
| - if (!(flag & FPDFCREATE_INCREMENTAL)) {
|
| - return FALSE;
|
| - }
|
| - int32_t iSize = pXRef->m_IndexArray.GetSize() / 2;
|
| - int32_t iCount = 0;
|
| - for (int32_t i = 0; i < iSize; i++) {
|
| - iCount += pXRef->m_IndexArray.ElementAt(i * 2 + 1);
|
| - }
|
| - return (iCount >= PDF_XREFSTREAM_MAXSIZE);
|
| +static FX_BOOL _IsXRefNeedEnd(CPDF_XRefStream* pXRef, FX_DWORD flag) {
|
| + if (!(flag & FPDFCREATE_INCREMENTAL)) {
|
| + return FALSE;
|
| + }
|
| + int32_t iSize = pXRef->m_IndexArray.GetSize() / 2;
|
| + int32_t iCount = 0;
|
| + for (int32_t i = 0; i < iSize; i++) {
|
| + iCount += pXRef->m_IndexArray.ElementAt(i * 2 + 1);
|
| + }
|
| + return (iCount >= PDF_XREFSTREAM_MAXSIZE);
|
| }
|
| -int32_t CPDF_Creator::WriteIndirectObjectToStream(const CPDF_Object* pObj)
|
| -{
|
| - if (!m_pXRefStream) {
|
| - return 1;
|
| - }
|
| - FX_DWORD objnum = pObj->GetObjNum();
|
| - if (m_pParser && m_pParser->m_ObjVersion.GetSize() > (int32_t)objnum && m_pParser->m_ObjVersion[objnum] > 0) {
|
| - return 1;
|
| - }
|
| - if (pObj->GetType() == PDFOBJ_NUMBER) {
|
| - return 1;
|
| - }
|
| - CPDF_Dictionary *pDict = pObj->GetDict();
|
| - if (pObj->GetType() == PDFOBJ_STREAM) {
|
| - if (pDict && pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("XRef")) {
|
| - return 0;
|
| - }
|
| - return 1;
|
| - }
|
| - if (pDict) {
|
| - if (pDict == m_pDocument->m_pRootDict || pDict == m_pEncryptDict) {
|
| - return 1;
|
| - }
|
| - if (IsSignatureDict(pDict)) {
|
| - return 1;
|
| - }
|
| - if (pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("Page")) {
|
| - return 1;
|
| - }
|
| - }
|
| - m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| - if (m_pXRefStream->CompressIndirectObject(objnum, pObj, this) < 0) {
|
| - return -1;
|
| - }
|
| - if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| - return 0;
|
| - }
|
| - if (!m_pXRefStream->End(this)) {
|
| - return -1;
|
| - }
|
| - if (!m_pXRefStream->Start()) {
|
| - return -1;
|
| +int32_t CPDF_Creator::WriteIndirectObjectToStream(const CPDF_Object* pObj) {
|
| + if (!m_pXRefStream) {
|
| + return 1;
|
| + }
|
| + FX_DWORD objnum = pObj->GetObjNum();
|
| + if (m_pParser && m_pParser->m_ObjVersion.GetSize() > (int32_t)objnum &&
|
| + m_pParser->m_ObjVersion[objnum] > 0) {
|
| + return 1;
|
| + }
|
| + if (pObj->GetType() == PDFOBJ_NUMBER) {
|
| + return 1;
|
| + }
|
| + CPDF_Dictionary* pDict = pObj->GetDict();
|
| + if (pObj->GetType() == PDFOBJ_STREAM) {
|
| + if (pDict && pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("XRef")) {
|
| + return 0;
|
| }
|
| + return 1;
|
| + }
|
| + if (pDict) {
|
| + if (pDict == m_pDocument->m_pRootDict || pDict == m_pEncryptDict) {
|
| + return 1;
|
| + }
|
| + if (IsSignatureDict(pDict)) {
|
| + return 1;
|
| + }
|
| + if (pDict->GetString(FX_BSTRC("Type")) == FX_BSTRC("Page")) {
|
| + return 1;
|
| + }
|
| + }
|
| + m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| + if (m_pXRefStream->CompressIndirectObject(objnum, pObj, this) < 0) {
|
| + return -1;
|
| + }
|
| + if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| return 0;
|
| + }
|
| + if (!m_pXRefStream->End(this)) {
|
| + return -1;
|
| + }
|
| + if (!m_pXRefStream->Start()) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::WriteIndirectObjectToStream(FX_DWORD objnum, const uint8_t* pBuffer, FX_DWORD dwSize)
|
| -{
|
| - if (!m_pXRefStream) {
|
| - return 1;
|
| - }
|
| - m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| - int32_t iRet = m_pXRefStream->CompressIndirectObject(objnum, pBuffer, dwSize, this);
|
| - if (iRet < 1) {
|
| - return iRet;
|
| - }
|
| - if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| - return 0;
|
| - }
|
| - if (!m_pXRefStream->End(this)) {
|
| - return -1;
|
| - }
|
| - if (!m_pXRefStream->Start()) {
|
| - return -1;
|
| - }
|
| +int32_t CPDF_Creator::WriteIndirectObjectToStream(FX_DWORD objnum,
|
| + const uint8_t* pBuffer,
|
| + FX_DWORD dwSize) {
|
| + if (!m_pXRefStream) {
|
| + return 1;
|
| + }
|
| + m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| + int32_t iRet =
|
| + m_pXRefStream->CompressIndirectObject(objnum, pBuffer, dwSize, this);
|
| + if (iRet < 1) {
|
| + return iRet;
|
| + }
|
| + if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| return 0;
|
| + }
|
| + if (!m_pXRefStream->End(this)) {
|
| + return -1;
|
| + }
|
| + if (!m_pXRefStream->Start()) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::AppendObjectNumberToXRef(FX_DWORD objnum)
|
| -{
|
| - if (!m_pXRefStream) {
|
| - return 1;
|
| - }
|
| - m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| - if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| - return 0;
|
| - }
|
| - if (!m_pXRefStream->End(this)) {
|
| - return -1;
|
| - }
|
| - if (!m_pXRefStream->Start()) {
|
| - return -1;
|
| - }
|
| +int32_t CPDF_Creator::AppendObjectNumberToXRef(FX_DWORD objnum) {
|
| + if (!m_pXRefStream) {
|
| + return 1;
|
| + }
|
| + m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| + if (!_IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) {
|
| return 0;
|
| + }
|
| + if (!m_pXRefStream->End(this)) {
|
| + return -1;
|
| + }
|
| + if (!m_pXRefStream->Start()) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::WriteStream(const CPDF_Object* pStream, FX_DWORD objnum, CPDF_CryptoHandler* pCrypto)
|
| -{
|
| - CPDF_FlateEncoder encoder;
|
| - encoder.Initialize((CPDF_Stream*)pStream, pStream == m_pMetadata ? FALSE : m_bCompress);
|
| - CPDF_Encryptor encryptor;
|
| - if(!encryptor.Initialize(pCrypto, objnum, encoder.m_pData, encoder.m_dwSize)) {
|
| +int32_t CPDF_Creator::WriteStream(const CPDF_Object* pStream,
|
| + FX_DWORD objnum,
|
| + CPDF_CryptoHandler* pCrypto) {
|
| + CPDF_FlateEncoder encoder;
|
| + encoder.Initialize((CPDF_Stream*)pStream,
|
| + pStream == m_pMetadata ? FALSE : m_bCompress);
|
| + CPDF_Encryptor encryptor;
|
| + if (!encryptor.Initialize(pCrypto, objnum, encoder.m_pData,
|
| + encoder.m_dwSize)) {
|
| + return -1;
|
| + }
|
| + if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) !=
|
| + encryptor.m_dwSize) {
|
| + encoder.CloneDict();
|
| + encoder.m_pDict->SetAtInteger(FX_BSTRC("Length"), encryptor.m_dwSize);
|
| + }
|
| + if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
|
| + return -1;
|
| + }
|
| + int len = m_File.AppendString(FX_BSTRC("stream\r\n"));
|
| + if (len < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += encryptor.m_dwSize;
|
| + if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + return 1;
|
| +}
|
| +int32_t CPDF_Creator::WriteIndirectObj(FX_DWORD objnum,
|
| + const CPDF_Object* pObj) {
|
| + int32_t len = m_File.AppendDWord(objnum);
|
| + if (len < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + if ((len = m_File.AppendString(FX_BSTRC(" 0 obj\r\n"))) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + if (pObj->GetType() == PDFOBJ_STREAM) {
|
| + CPDF_CryptoHandler* pHandler = NULL;
|
| + pHandler =
|
| + (pObj == m_pMetadata && !m_bEncryptMetadata) ? NULL : m_pCryptoHandler;
|
| + if (WriteStream(pObj, objnum, pHandler) < 0) {
|
| + return -1;
|
| + }
|
| + } else {
|
| + if (WriteDirectObj(objnum, pObj) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| + if ((len = m_File.AppendString(FX_BSTRC("\r\nendobj\r\n"))) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + if (AppendObjectNumberToXRef(objnum) < 0) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| +}
|
| +int32_t CPDF_Creator::WriteIndirectObj(const CPDF_Object* pObj) {
|
| + int32_t iRet = WriteIndirectObjectToStream(pObj);
|
| + if (iRet < 1) {
|
| + return iRet;
|
| + }
|
| + return WriteIndirectObj(pObj->GetObjNum(), pObj);
|
| +}
|
| +int32_t CPDF_Creator::WriteDirectObj(FX_DWORD objnum,
|
| + const CPDF_Object* pObj,
|
| + FX_BOOL bEncrypt) {
|
| + int32_t len = 0;
|
| + if (pObj == NULL) {
|
| + if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 5;
|
| + return 1;
|
| + }
|
| + switch (pObj->GetType()) {
|
| + case PDFOBJ_NULL:
|
| + if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
|
| return -1;
|
| - }
|
| - if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) != encryptor.m_dwSize) {
|
| + }
|
| + m_Offset += 5;
|
| + break;
|
| + case PDFOBJ_BOOLEAN:
|
| + case PDFOBJ_NUMBER:
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = m_File.AppendString(pObj->GetString())) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 1;
|
| + break;
|
| + case PDFOBJ_STRING: {
|
| + CFX_ByteString str = pObj->GetString();
|
| + FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
|
| + if (m_pCryptoHandler == NULL || !bEncrypt) {
|
| + CFX_ByteString content = PDF_EncodeString(str, bHex);
|
| + if ((len = m_File.AppendString(content)) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + break;
|
| + }
|
| + CPDF_Encryptor encryptor;
|
| + encryptor.Initialize(m_pCryptoHandler, objnum, (uint8_t*)str.c_str(),
|
| + str.GetLength());
|
| + CFX_ByteString content = PDF_EncodeString(
|
| + CFX_ByteString((const FX_CHAR*)encryptor.m_pData, encryptor.m_dwSize),
|
| + bHex);
|
| + if ((len = m_File.AppendString(content)) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + break;
|
| + }
|
| + case PDFOBJ_STREAM: {
|
| + CPDF_FlateEncoder encoder;
|
| + encoder.Initialize((CPDF_Stream*)pObj, m_bCompress);
|
| + CPDF_Encryptor encryptor;
|
| + CPDF_CryptoHandler* pHandler = m_pCryptoHandler;
|
| + encryptor.Initialize(pHandler, objnum, encoder.m_pData, encoder.m_dwSize);
|
| + if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) !=
|
| + encryptor.m_dwSize) {
|
| encoder.CloneDict();
|
| encoder.m_pDict->SetAtInteger(FX_BSTRC("Length"), encryptor.m_dwSize);
|
| - }
|
| - if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
|
| + }
|
| + if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
|
| return -1;
|
| - }
|
| - int len = m_File.AppendString(FX_BSTRC("stream\r\n"));
|
| - if (len < 0) {
|
| + }
|
| + if ((len = m_File.AppendString(FX_BSTRC("stream\r\n"))) < 0) {
|
| return -1;
|
| - }
|
| - m_Offset += len;
|
| - if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| + }
|
| + m_Offset += len;
|
| + if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| return -1;
|
| - }
|
| - m_Offset += encryptor.m_dwSize;
|
| - if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| + }
|
| + m_Offset += encryptor.m_dwSize;
|
| + if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| return -1;
|
| + }
|
| + m_Offset += len;
|
| + break;
|
| }
|
| - m_Offset += len;
|
| - return 1;
|
| -}
|
| -int32_t CPDF_Creator::WriteIndirectObj(FX_DWORD objnum, const CPDF_Object* pObj)
|
| -{
|
| - int32_t len = m_File.AppendDWord(objnum);
|
| - if (len < 0) {
|
| + case PDFOBJ_NAME: {
|
| + if (m_File.AppendString(FX_BSTRC("/")) < 0) {
|
| return -1;
|
| + }
|
| + CFX_ByteString str = pObj->GetString();
|
| + if ((len = m_File.AppendString(PDF_NameEncode(str))) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 1;
|
| + break;
|
| }
|
| - m_Offset += len;
|
| - if ((len = m_File.AppendString(FX_BSTRC(" 0 obj\r\n"))) < 0) {
|
| + case PDFOBJ_REFERENCE: {
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + CPDF_Reference* p = (CPDF_Reference*)pObj;
|
| + if ((len = m_File.AppendDWord(p->GetRefObjNum())) < 0) {
|
| return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 5;
|
| + break;
|
| }
|
| - m_Offset += len;
|
| - if (pObj->GetType() == PDFOBJ_STREAM) {
|
| - CPDF_CryptoHandler *pHandler = NULL;
|
| - pHandler = (pObj == m_pMetadata && !m_bEncryptMetadata) ? NULL : m_pCryptoHandler;
|
| - if (WriteStream(pObj, objnum, pHandler) < 0) {
|
| + case PDFOBJ_ARRAY: {
|
| + if (m_File.AppendString(FX_BSTRC("[")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 1;
|
| + CPDF_Array* p = (CPDF_Array*)pObj;
|
| + for (FX_DWORD i = 0; i < p->GetCount(); i++) {
|
| + CPDF_Object* pElement = p->GetElement(i);
|
| + if (pElement->GetObjNum()) {
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = m_File.AppendDWord(pElement->GetObjNum())) < 0) {
|
| return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 5;
|
| + } else {
|
| + if (WriteDirectObj(objnum, pElement) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - } else {
|
| - if (WriteDirectObj(objnum, pObj) < 0) {
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC("]")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 1;
|
| + break;
|
| + }
|
| + case PDFOBJ_DICTIONARY: {
|
| + if (m_pCryptoHandler == NULL || pObj == m_pEncryptDict) {
|
| + return PDF_CreatorAppendObject(pObj, &m_File, m_Offset);
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC("<<")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 2;
|
| + CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
|
| + FX_BOOL bSignDict = IsSignatureDict(p);
|
| + FX_POSITION pos = p->GetStartPos();
|
| + while (pos) {
|
| + FX_BOOL bSignValue = FALSE;
|
| + CFX_ByteString key;
|
| + CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| + if (m_File.AppendString(FX_BSTRC("/")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = m_File.AppendString(PDF_NameEncode(key))) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 1;
|
| + if (bSignDict && key == FX_BSTRC("Contents")) {
|
| + bSignValue = TRUE;
|
| + }
|
| + if (pValue->GetObjNum()) {
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if ((len = m_File.AppendDWord(pValue->GetObjNum())) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| return -1;
|
| + }
|
| + m_Offset += len + 6;
|
| + } else {
|
| + if (WriteDirectObj(objnum, pValue, !bSignValue) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - }
|
| - if ((len = m_File.AppendString(FX_BSTRC("\r\nendobj\r\n"))) < 0) {
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
|
| return -1;
|
| + }
|
| + m_Offset += 2;
|
| + break;
|
| }
|
| - m_Offset += len;
|
| - if (AppendObjectNumberToXRef(objnum) < 0) {
|
| + }
|
| + return 1;
|
| +}
|
| +int32_t CPDF_Creator::WriteOldIndirectObject(FX_DWORD objnum) {
|
| + if (m_pParser->m_V5Type[objnum] == 0 || m_pParser->m_V5Type[objnum] == 255) {
|
| + return 0;
|
| + }
|
| + m_ObjectOffset[objnum] = m_Offset;
|
| + void* valuetemp = NULL;
|
| + FX_BOOL bExistInMap =
|
| + m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, valuetemp);
|
| + FX_BOOL bObjStm =
|
| + (m_pParser->m_V5Type[objnum] == 2) && m_pEncryptDict && !m_pXRefStream;
|
| + if (m_pParser->m_bVersionUpdated || m_bSecurityChanged || bExistInMap ||
|
| + bObjStm) {
|
| + CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum);
|
| + if (pObj == NULL) {
|
| + m_ObjectOffset[objnum] = 0;
|
| + m_ObjectSize[objnum] = 0;
|
| + return 0;
|
| + }
|
| + if (WriteIndirectObj(pObj)) {
|
| + return -1;
|
| + }
|
| + if (!bExistInMap) {
|
| + m_pDocument->ReleaseIndirectObject(objnum);
|
| + }
|
| + } else {
|
| + uint8_t* pBuffer;
|
| + FX_DWORD size;
|
| + m_pParser->GetIndirectBinary(objnum, pBuffer, size);
|
| + if (pBuffer == NULL) {
|
| + return 0;
|
| + }
|
| + if (m_pParser->m_V5Type[objnum] == 2) {
|
| + if (m_pXRefStream) {
|
| + if (WriteIndirectObjectToStream(objnum, pBuffer, size) < 0) {
|
| + FX_Free(pBuffer);
|
| + return -1;
|
| + }
|
| + } else {
|
| + int32_t len = m_File.AppendDWord(objnum);
|
| + if (len < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 obj ")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len + 7;
|
| + if (m_File.AppendBlock(pBuffer, size) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += size;
|
| + if (m_File.AppendString(FX_BSTRC("\r\nendobj\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 10;
|
| + }
|
| + } else {
|
| + if (m_File.AppendBlock(pBuffer, size) < 0) {
|
| return -1;
|
| + }
|
| + m_Offset += size;
|
| + if (AppendObjectNumberToXRef(objnum) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - return 0;
|
| + FX_Free(pBuffer);
|
| + }
|
| + return 1;
|
| }
|
| -int32_t CPDF_Creator::WriteIndirectObj(const CPDF_Object* pObj)
|
| -{
|
| - int32_t iRet = WriteIndirectObjectToStream(pObj);
|
| - if (iRet < 1) {
|
| - return iRet;
|
| - }
|
| - return WriteIndirectObj(pObj->GetObjNum(), pObj);
|
| +int32_t CPDF_Creator::WriteOldObjs(IFX_Pause* pPause) {
|
| + FX_DWORD nOldSize = m_pParser->m_CrossRef.GetSize();
|
| + FX_DWORD objnum = (FX_DWORD)(uintptr_t)m_Pos;
|
| + for (; objnum < nOldSize; objnum++) {
|
| + int32_t iRet = WriteOldIndirectObject(objnum);
|
| + if (!iRet) {
|
| + continue;
|
| + }
|
| + if (iRet < 0) {
|
| + return iRet;
|
| + }
|
| + m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
|
| + if (pPause && pPause->NeedToPauseNow()) {
|
| + m_Pos = (void*)(uintptr_t)(objnum + 1);
|
| + return 1;
|
| + }
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::WriteDirectObj(FX_DWORD objnum, const CPDF_Object* pObj, FX_BOOL bEncrypt)
|
| -{
|
| - int32_t len = 0;
|
| - if (pObj == NULL) {
|
| - if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 5;
|
| - return 1;
|
| +int32_t CPDF_Creator::WriteNewObjs(FX_BOOL bIncremental, IFX_Pause* pPause) {
|
| + int32_t iCount = m_NewObjNumArray.GetSize();
|
| + int32_t index = (int32_t)(uintptr_t)m_Pos;
|
| + while (index < iCount) {
|
| + FX_DWORD objnum = m_NewObjNumArray.ElementAt(index);
|
| + CPDF_Object* pObj = NULL;
|
| + m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, (void*&)pObj);
|
| + if (NULL == pObj) {
|
| + ++index;
|
| + continue;
|
| }
|
| - switch (pObj->GetType()) {
|
| - case PDFOBJ_NULL:
|
| - if (m_File.AppendString(FX_BSTRC(" null")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 5;
|
| - break;
|
| - case PDFOBJ_BOOLEAN:
|
| - case PDFOBJ_NUMBER:
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = m_File.AppendString(pObj->GetString())) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 1;
|
| - break;
|
| - case PDFOBJ_STRING: {
|
| - CFX_ByteString str = pObj->GetString();
|
| - FX_BOOL bHex = ((CPDF_String*)pObj)->IsHex();
|
| - if (m_pCryptoHandler == NULL || !bEncrypt) {
|
| - CFX_ByteString content = PDF_EncodeString(str, bHex);
|
| - if ((len = m_File.AppendString(content)) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - break;
|
| - }
|
| - CPDF_Encryptor encryptor;
|
| - encryptor.Initialize(m_pCryptoHandler, objnum, (uint8_t*)str.c_str(), str.GetLength());
|
| - CFX_ByteString content = PDF_EncodeString(CFX_ByteString((const FX_CHAR*)encryptor.m_pData, encryptor.m_dwSize), bHex);
|
| - if ((len = m_File.AppendString(content)) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - break;
|
| - }
|
| - case PDFOBJ_STREAM: {
|
| - CPDF_FlateEncoder encoder;
|
| - encoder.Initialize((CPDF_Stream*)pObj, m_bCompress);
|
| - CPDF_Encryptor encryptor;
|
| - CPDF_CryptoHandler* pHandler = m_pCryptoHandler;
|
| - encryptor.Initialize(pHandler, objnum, encoder.m_pData, encoder.m_dwSize);
|
| - if ((FX_DWORD)encoder.m_pDict->GetInteger(FX_BSTRC("Length")) != encryptor.m_dwSize) {
|
| - encoder.CloneDict();
|
| - encoder.m_pDict->SetAtInteger(FX_BSTRC("Length"), encryptor.m_dwSize);
|
| - }
|
| - if (WriteDirectObj(objnum, encoder.m_pDict) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = m_File.AppendString(FX_BSTRC("stream\r\n"))) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += encryptor.m_dwSize;
|
| - if ((len = m_File.AppendString(FX_BSTRC("\r\nendstream"))) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - break;
|
| - }
|
| - case PDFOBJ_NAME: {
|
| - if (m_File.AppendString(FX_BSTRC("/")) < 0) {
|
| - return -1;
|
| - }
|
| - CFX_ByteString str = pObj->GetString();
|
| - if ((len = m_File.AppendString(PDF_NameEncode(str))) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 1;
|
| - break;
|
| - }
|
| - case PDFOBJ_REFERENCE: {
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - CPDF_Reference* p = (CPDF_Reference*)pObj;
|
| - if ((len = m_File.AppendDWord(p->GetRefObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 5;
|
| - break;
|
| - }
|
| - case PDFOBJ_ARRAY: {
|
| - if (m_File.AppendString(FX_BSTRC("[")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 1;
|
| - CPDF_Array* p = (CPDF_Array*)pObj;
|
| - for (FX_DWORD i = 0; i < p->GetCount(); i ++) {
|
| - CPDF_Object* pElement = p->GetElement(i);
|
| - if (pElement->GetObjNum()) {
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = m_File.AppendDWord(pElement->GetObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 5;
|
| - } else {
|
| - if (WriteDirectObj(objnum, pElement) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("]")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 1;
|
| - break;
|
| - }
|
| - case PDFOBJ_DICTIONARY: {
|
| - if (m_pCryptoHandler == NULL || pObj == m_pEncryptDict) {
|
| - return PDF_CreatorAppendObject(pObj, &m_File, m_Offset);
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("<<")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 2;
|
| - CPDF_Dictionary* p = (CPDF_Dictionary*)pObj;
|
| - FX_BOOL bSignDict = IsSignatureDict(p);
|
| - FX_POSITION pos = p->GetStartPos();
|
| - while (pos) {
|
| - FX_BOOL bSignValue = FALSE;
|
| - CFX_ByteString key;
|
| - CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| - if (m_File.AppendString(FX_BSTRC("/")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = m_File.AppendString(PDF_NameEncode(key))) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 1;
|
| - if (bSignDict && key == FX_BSTRC("Contents")) {
|
| - bSignValue = TRUE;
|
| - }
|
| - if (pValue->GetObjNum()) {
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((len = m_File.AppendDWord(pValue->GetObjNum())) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 6;
|
| - } else {
|
| - if (WriteDirectObj(objnum, pValue, !bSignValue) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 2;
|
| - break;
|
| - }
|
| + m_ObjectOffset[objnum] = m_Offset;
|
| + if (WriteIndirectObj(pObj)) {
|
| + return -1;
|
| }
|
| - return 1;
|
| + m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
|
| + index++;
|
| + if (pPause && pPause->NeedToPauseNow()) {
|
| + m_Pos = (FX_POSITION)(uintptr_t)index;
|
| + return 1;
|
| + }
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::WriteOldIndirectObject(FX_DWORD objnum)
|
| -{
|
| - if(m_pParser->m_V5Type[objnum] == 0 || m_pParser->m_V5Type[objnum] == 255) {
|
| - return 0;
|
| +void CPDF_Creator::InitOldObjNumOffsets() {
|
| + if (!m_pParser) {
|
| + return;
|
| + }
|
| + FX_DWORD j = 0;
|
| + FX_DWORD dwStart = 0;
|
| + FX_DWORD dwEnd = m_pParser->GetLastObjNum();
|
| + while (dwStart <= dwEnd) {
|
| + while (dwStart <= dwEnd && (m_pParser->m_V5Type[dwStart] == 0 ||
|
| + m_pParser->m_V5Type[dwStart] == 255)) {
|
| + dwStart++;
|
| + }
|
| + if (dwStart > dwEnd) {
|
| + break;
|
| + }
|
| + j = dwStart;
|
| + while (j <= dwEnd && m_pParser->m_V5Type[j] != 0 &&
|
| + m_pParser->m_V5Type[j] != 255) {
|
| + j++;
|
| + }
|
| + m_ObjectOffset.Add(dwStart, j - dwStart);
|
| + m_ObjectSize.Add(dwStart, j - dwStart);
|
| + dwStart = j;
|
| + }
|
| +}
|
| +void CPDF_Creator::InitNewObjNumOffsets() {
|
| + FX_BOOL bIncremental = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
|
| + FX_BOOL bNoOriginal = (m_dwFlags & FPDFCREATE_NO_ORIGINAL) != 0;
|
| + FX_DWORD nOldSize = m_pParser ? m_pParser->m_CrossRef.GetSize() : 0;
|
| + FX_POSITION pos = m_pDocument->m_IndirectObjs.GetStartPosition();
|
| + while (pos) {
|
| + size_t key = 0;
|
| + CPDF_Object* pObj;
|
| + m_pDocument->m_IndirectObjs.GetNextAssoc(pos, (void*&)key, (void*&)pObj);
|
| + FX_DWORD objnum = (FX_DWORD)key;
|
| + if (pObj->GetObjNum() == -1) {
|
| + continue;
|
| }
|
| - m_ObjectOffset[objnum] = m_Offset;
|
| - void* valuetemp = NULL;
|
| - FX_BOOL bExistInMap = m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, valuetemp);
|
| - FX_BOOL bObjStm = (m_pParser->m_V5Type[objnum] == 2) && m_pEncryptDict && !m_pXRefStream;
|
| - if(m_pParser->m_bVersionUpdated || m_bSecurityChanged || bExistInMap || bObjStm) {
|
| - CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum);
|
| - if (pObj == NULL) {
|
| - m_ObjectOffset[objnum] = 0;
|
| - m_ObjectSize[objnum] = 0;
|
| - return 0;
|
| - }
|
| - if (WriteIndirectObj(pObj)) {
|
| - return -1;
|
| - }
|
| - if (!bExistInMap) {
|
| - m_pDocument->ReleaseIndirectObject(objnum);
|
| - }
|
| + if (bIncremental) {
|
| + if (!pObj->IsModified()) {
|
| + continue;
|
| + }
|
| } else {
|
| - uint8_t* pBuffer;
|
| - FX_DWORD size;
|
| - m_pParser->GetIndirectBinary(objnum, pBuffer, size);
|
| - if (pBuffer == NULL) {
|
| - return 0;
|
| - }
|
| - if (m_pParser->m_V5Type[objnum] == 2) {
|
| - if (m_pXRefStream) {
|
| - if (WriteIndirectObjectToStream(objnum, pBuffer, size) < 0) {
|
| - FX_Free(pBuffer);
|
| - return -1;
|
| - }
|
| - } else {
|
| - int32_t len = m_File.AppendDWord(objnum);
|
| - if (len < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 obj ")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len + 7;
|
| - if (m_File.AppendBlock(pBuffer, size) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += size;
|
| - if (m_File.AppendString(FX_BSTRC("\r\nendobj\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 10;
|
| - }
|
| - } else {
|
| - if (m_File.AppendBlock(pBuffer, size) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += size;
|
| - if(AppendObjectNumberToXRef(objnum) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - FX_Free(pBuffer);
|
| + if (objnum < nOldSize && m_pParser->m_V5Type[objnum] != 0) {
|
| + continue;
|
| + }
|
| + }
|
| + AppendNewObjNum(objnum);
|
| + }
|
| + int32_t iCount = m_NewObjNumArray.GetSize();
|
| + if (iCount == 0) {
|
| + return;
|
| + }
|
| + int32_t i = 0;
|
| + FX_DWORD dwStartObjNum = 0;
|
| + FX_BOOL bCrossRefValid = m_pParser && m_pParser->GetLastXRefOffset() > 0;
|
| + while (i < iCount) {
|
| + dwStartObjNum = m_NewObjNumArray.ElementAt(i);
|
| + if ((bIncremental && (bNoOriginal || bCrossRefValid)) ||
|
| + !m_ObjectOffset.GetPtrAt(dwStartObjNum)) {
|
| + break;
|
| }
|
| - return 1;
|
| + i++;
|
| + }
|
| + if (i >= iCount) {
|
| + return;
|
| + }
|
| + FX_DWORD dwLastObjNum = dwStartObjNum;
|
| + i++;
|
| + FX_BOOL bNewStart = FALSE;
|
| + for (; i < iCount; i++) {
|
| + FX_DWORD dwCurObjNum = m_NewObjNumArray.ElementAt(i);
|
| + FX_BOOL bExist = (dwCurObjNum < nOldSize &&
|
| + m_ObjectOffset.GetPtrAt(dwCurObjNum) != NULL);
|
| + if (bExist || dwCurObjNum - dwLastObjNum > 1) {
|
| + if (!bNewStart) {
|
| + m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| + m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| + }
|
| + dwStartObjNum = dwCurObjNum;
|
| + }
|
| + if (bNewStart) {
|
| + dwStartObjNum = dwCurObjNum;
|
| + }
|
| + bNewStart = bExist;
|
| + dwLastObjNum = dwCurObjNum;
|
| + }
|
| + m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| + m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| }
|
| -int32_t CPDF_Creator::WriteOldObjs(IFX_Pause *pPause)
|
| -{
|
| - FX_DWORD nOldSize = m_pParser->m_CrossRef.GetSize();
|
| - FX_DWORD objnum = (FX_DWORD)(uintptr_t)m_Pos;
|
| - for(; objnum < nOldSize; objnum ++) {
|
| - int32_t iRet = WriteOldIndirectObject(objnum);
|
| - if (!iRet) {
|
| - continue;
|
| - }
|
| - if (iRet < 0) {
|
| - return iRet;
|
| - }
|
| - m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
|
| - if (pPause && pPause->NeedToPauseNow()) {
|
| - m_Pos = (void*)(uintptr_t)(objnum + 1);
|
| - return 1;
|
| - }
|
| - }
|
| - return 0;
|
| +void CPDF_Creator::AppendNewObjNum(FX_DWORD objbum) {
|
| + int32_t iStart = 0, iFind = 0;
|
| + int32_t iEnd = m_NewObjNumArray.GetUpperBound();
|
| + while (iStart <= iEnd) {
|
| + int32_t iMid = (iStart + iEnd) / 2;
|
| + FX_DWORD dwMid = m_NewObjNumArray.ElementAt(iMid);
|
| + if (objbum < dwMid) {
|
| + iEnd = iMid - 1;
|
| + } else {
|
| + if (iMid == iEnd) {
|
| + iFind = iMid + 1;
|
| + break;
|
| + }
|
| + FX_DWORD dwNext = m_NewObjNumArray.ElementAt(iMid + 1);
|
| + if (objbum < dwNext) {
|
| + iFind = iMid + 1;
|
| + break;
|
| + }
|
| + iStart = iMid + 1;
|
| + }
|
| + }
|
| + m_NewObjNumArray.InsertAt(iFind, objbum);
|
| }
|
| -int32_t CPDF_Creator::WriteNewObjs(FX_BOOL bIncremental, IFX_Pause *pPause)
|
| -{
|
| - int32_t iCount = m_NewObjNumArray.GetSize();
|
| - int32_t index = (int32_t)(uintptr_t)m_Pos;
|
| - while (index < iCount) {
|
| - FX_DWORD objnum = m_NewObjNumArray.ElementAt(index);
|
| - CPDF_Object *pObj = NULL;
|
| - m_pDocument->m_IndirectObjs.Lookup((void*)(uintptr_t)objnum, (void*&)pObj);
|
| - if (NULL == pObj) {
|
| - ++index;
|
| - continue;
|
| - }
|
| - m_ObjectOffset[objnum] = m_Offset;
|
| - if (WriteIndirectObj(pObj)) {
|
| - return -1;
|
| - }
|
| - m_ObjectSize[objnum] = (FX_DWORD)(m_Offset - m_ObjectOffset[objnum]);
|
| - index++;
|
| +int32_t CPDF_Creator::WriteDoc_Stage1(IFX_Pause* pPause) {
|
| + FXSYS_assert(m_iStage > -1 || m_iStage < 20);
|
| + if (m_iStage == 0) {
|
| + if (m_pParser == NULL) {
|
| + m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
|
| + }
|
| + if (m_bSecurityChanged && (m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0) {
|
| + m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
|
| + }
|
| + CPDF_Dictionary* pDict = m_pDocument->GetRoot();
|
| + m_pMetadata = pDict ? pDict->GetElementValue(FX_BSTRC("Metadata")) : NULL;
|
| + if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
|
| + m_pXRefStream = new CPDF_XRefStream;
|
| + m_pXRefStream->Start();
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser) {
|
| + FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
|
| + m_pXRefStream->m_PrevOffset = prev;
|
| + }
|
| + }
|
| + m_iStage = 10;
|
| + }
|
| + if (m_iStage == 10) {
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0) {
|
| + if (m_File.AppendString(FX_BSTRC("%PDF-1.")) < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += 7;
|
| + int32_t version = 7;
|
| + if (m_FileVersion) {
|
| + version = m_FileVersion;
|
| + } else if (m_pParser) {
|
| + version = m_pParser->GetFileVersion();
|
| + }
|
| + int32_t len = m_File.AppendDWord(version % 10);
|
| + if (len < 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + if ((len = m_File.AppendString(FX_BSTRC("\r\n%\xA1\xB3\xC5\xD7\r\n"))) <
|
| + 0) {
|
| + return -1;
|
| + }
|
| + m_Offset += len;
|
| + InitOldObjNumOffsets();
|
| + m_iStage = 20;
|
| + } else {
|
| + IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
|
| + m_Offset = pSrcFile->GetSize();
|
| + m_Pos = (void*)(uintptr_t)m_Offset;
|
| + m_iStage = 15;
|
| + }
|
| + }
|
| + if (m_iStage == 15) {
|
| + if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_Pos) {
|
| + IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
|
| + uint8_t buffer[4096];
|
| + FX_DWORD src_size = (FX_DWORD)(uintptr_t)m_Pos;
|
| + while (src_size) {
|
| + FX_DWORD block_size = src_size > 4096 ? 4096 : src_size;
|
| + if (!pSrcFile->ReadBlock(buffer, m_Offset - src_size, block_size)) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendBlock(buffer, block_size) < 0) {
|
| + return -1;
|
| + }
|
| + src_size -= block_size;
|
| if (pPause && pPause->NeedToPauseNow()) {
|
| - m_Pos = (FX_POSITION)(uintptr_t)index;
|
| - return 1;
|
| - }
|
| - }
|
| - return 0;
|
| + m_Pos = (void*)(uintptr_t)src_size;
|
| + return 1;
|
| + }
|
| + }
|
| + }
|
| + if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 &&
|
| + m_pParser->GetLastXRefOffset() == 0) {
|
| + InitOldObjNumOffsets();
|
| + FX_DWORD dwEnd = m_pParser->GetLastObjNum();
|
| + FX_BOOL bObjStm = (m_dwFlags & FPDFCREATE_OBJECTSTREAM) != 0;
|
| + for (FX_DWORD objnum = 0; objnum <= dwEnd; objnum++) {
|
| + if (m_pParser->m_V5Type[objnum] == 0 ||
|
| + m_pParser->m_V5Type[objnum] == 255) {
|
| + continue;
|
| + }
|
| + m_ObjectOffset[objnum] = m_pParser->m_CrossRef[objnum];
|
| + if (bObjStm) {
|
| + m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| + }
|
| + }
|
| + if (bObjStm) {
|
| + m_pXRefStream->EndXRefStream(this);
|
| + m_pXRefStream->Start();
|
| + }
|
| + }
|
| + m_iStage = 20;
|
| + }
|
| + InitNewObjNumOffsets();
|
| + return m_iStage;
|
| }
|
| -void CPDF_Creator::InitOldObjNumOffsets()
|
| -{
|
| - if (!m_pParser) {
|
| - return;
|
| - }
|
| - FX_DWORD j = 0;
|
| - FX_DWORD dwStart = 0;
|
| - FX_DWORD dwEnd = m_pParser->GetLastObjNum();
|
| - while (dwStart <= dwEnd) {
|
| - while (dwStart <= dwEnd && (m_pParser->m_V5Type[dwStart] == 0 || m_pParser->m_V5Type[dwStart] == 255)) {
|
| - dwStart++;
|
| - }
|
| - if (dwStart > dwEnd) {
|
| - break;
|
| - }
|
| - j = dwStart;
|
| - while (j <= dwEnd && m_pParser->m_V5Type[j] != 0 && m_pParser->m_V5Type[j] != 255) {
|
| - j++;
|
| - }
|
| - m_ObjectOffset.Add(dwStart, j - dwStart);
|
| - m_ObjectSize.Add(dwStart, j - dwStart);
|
| - dwStart = j;
|
| - }
|
| +int32_t CPDF_Creator::WriteDoc_Stage2(IFX_Pause* pPause) {
|
| + FXSYS_assert(m_iStage >= 20 || m_iStage < 30);
|
| + if (m_iStage == 20) {
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 && m_pParser) {
|
| + m_Pos = (void*)(uintptr_t)0;
|
| + m_iStage = 21;
|
| + } else {
|
| + m_iStage = 25;
|
| + }
|
| + }
|
| + if (m_iStage == 21) {
|
| + int32_t iRet = WriteOldObjs(pPause);
|
| + if (iRet) {
|
| + return iRet;
|
| + }
|
| + m_iStage = 25;
|
| + }
|
| + if (m_iStage == 25) {
|
| + m_Pos = (void*)(uintptr_t)0;
|
| + m_iStage = 26;
|
| + }
|
| + if (m_iStage == 26) {
|
| + int32_t iRet =
|
| + WriteNewObjs((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0, pPause);
|
| + if (iRet) {
|
| + return iRet;
|
| + }
|
| + m_iStage = 27;
|
| + }
|
| + if (m_iStage == 27) {
|
| + if (NULL != m_pEncryptDict && 0 == m_pEncryptDict->GetObjNum()) {
|
| + m_dwLastObjNum += 1;
|
| + FX_FILESIZE saveOffset = m_Offset;
|
| + if (WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict) < 0) {
|
| + return -1;
|
| + }
|
| + m_ObjectOffset.Add(m_dwLastObjNum, 1);
|
| + m_ObjectOffset[m_dwLastObjNum] = saveOffset;
|
| + m_ObjectSize.Add(m_dwLastObjNum, 1);
|
| + m_ObjectSize[m_dwLastObjNum] = m_Offset - saveOffset;
|
| + m_dwEnryptObjNum = m_dwLastObjNum;
|
| + if (m_dwFlags & FPDFCREATE_INCREMENTAL) {
|
| + m_NewObjNumArray.Add(m_dwLastObjNum);
|
| + }
|
| + }
|
| + m_iStage = 80;
|
| + }
|
| + return m_iStage;
|
| }
|
| -void CPDF_Creator::InitNewObjNumOffsets()
|
| -{
|
| - FX_BOOL bIncremental = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0;
|
| - FX_BOOL bNoOriginal = (m_dwFlags & FPDFCREATE_NO_ORIGINAL) != 0;
|
| - FX_DWORD nOldSize = m_pParser ? m_pParser->m_CrossRef.GetSize() : 0;
|
| - FX_POSITION pos = m_pDocument->m_IndirectObjs.GetStartPosition();
|
| - while (pos) {
|
| - size_t key = 0;
|
| - CPDF_Object* pObj;
|
| - m_pDocument->m_IndirectObjs.GetNextAssoc(pos, (void*&)key, (void*&)pObj);
|
| - FX_DWORD objnum = (FX_DWORD)key;
|
| - if (pObj->GetObjNum() == -1) {
|
| - continue;
|
| - }
|
| - if (bIncremental) {
|
| - if (!pObj->IsModified()) {
|
| - continue;
|
| - }
|
| - } else {
|
| - if (objnum < nOldSize && m_pParser->m_V5Type[objnum] != 0) {
|
| - continue;
|
| - }
|
| +int32_t CPDF_Creator::WriteDoc_Stage3(IFX_Pause* pPause) {
|
| + FXSYS_assert(m_iStage >= 80 || m_iStage < 90);
|
| + FX_DWORD dwLastObjNum = m_dwLastObjNum;
|
| + if (m_iStage == 80) {
|
| + m_XrefStart = m_Offset;
|
| + if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
|
| + m_pXRefStream->End(this, TRUE);
|
| + m_XrefStart = m_pXRefStream->m_PrevOffset;
|
| + m_iStage = 90;
|
| + } else if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 ||
|
| + !m_pParser->IsXRefStream()) {
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 ||
|
| + m_pParser->GetLastXRefOffset() == 0) {
|
| + CFX_ByteString str;
|
| + str = m_ObjectOffset.GetPtrAt(1)
|
| + ? FX_BSTRC("xref\r\n")
|
| + : FX_BSTRC("xref\r\n0 1\r\n0000000000 65536 f\r\n");
|
| + if (m_File.AppendString(str) < 0) {
|
| + return -1;
|
| + }
|
| + m_Pos = (void*)(uintptr_t)1;
|
| + m_iStage = 81;
|
| + } else {
|
| + if (m_File.AppendString(FX_BSTRC("xref\r\n")) < 0) {
|
| + return -1;
|
| }
|
| - AppendNewObjNum(objnum);
|
| + m_Pos = (void*)(uintptr_t)0;
|
| + m_iStage = 82;
|
| + }
|
| + } else {
|
| + m_iStage = 90;
|
| + }
|
| + }
|
| + if (m_iStage == 81) {
|
| + CFX_ByteString str;
|
| + FX_DWORD i = (FX_DWORD)(uintptr_t)m_Pos, j;
|
| + while (i <= dwLastObjNum) {
|
| + while (i <= dwLastObjNum && !m_ObjectOffset.GetPtrAt(i)) {
|
| + i++;
|
| + }
|
| + if (i > dwLastObjNum) {
|
| + break;
|
| + }
|
| + j = i;
|
| + while (j <= dwLastObjNum && m_ObjectOffset.GetPtrAt(j)) {
|
| + j++;
|
| + }
|
| + if (i == 1) {
|
| + str.Format("0 %d\r\n0000000000 65536 f\r\n", j);
|
| + } else {
|
| + str.Format("%d %d\r\n", i, j - i);
|
| + }
|
| + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| + return -1;
|
| + }
|
| + while (i < j) {
|
| + str.Format("%010d 00000 n\r\n", m_ObjectOffset[i++]);
|
| + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| + if (i > dwLastObjNum) {
|
| + break;
|
| + }
|
| + if (pPause && pPause->NeedToPauseNow()) {
|
| + m_Pos = (void*)(uintptr_t)i;
|
| + return 1;
|
| + }
|
| }
|
| + m_iStage = 90;
|
| + }
|
| + if (m_iStage == 82) {
|
| + CFX_ByteString str;
|
| int32_t iCount = m_NewObjNumArray.GetSize();
|
| - if (iCount == 0) {
|
| - return;
|
| - }
|
| - int32_t i = 0;
|
| - FX_DWORD dwStartObjNum = 0;
|
| - FX_BOOL bCrossRefValid = m_pParser && m_pParser->GetLastXRefOffset() > 0;
|
| + int32_t i = (int32_t)(uintptr_t)m_Pos;
|
| while (i < iCount) {
|
| - dwStartObjNum = m_NewObjNumArray.ElementAt(i);
|
| - if ((bIncremental && (bNoOriginal || bCrossRefValid)) || !m_ObjectOffset.GetPtrAt(dwStartObjNum)) {
|
| - break;
|
| - }
|
| - i++;
|
| - }
|
| - if (i >= iCount) {
|
| - return;
|
| - }
|
| - FX_DWORD dwLastObjNum = dwStartObjNum;
|
| - i++;
|
| - FX_BOOL bNewStart = FALSE;
|
| - for (; i < iCount; i++) {
|
| - FX_DWORD dwCurObjNum = m_NewObjNumArray.ElementAt(i);
|
| - FX_BOOL bExist = (dwCurObjNum < nOldSize && m_ObjectOffset.GetPtrAt(dwCurObjNum) != NULL);
|
| - if (bExist || dwCurObjNum - dwLastObjNum > 1) {
|
| - if (!bNewStart) {
|
| - m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| - m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| - }
|
| - dwStartObjNum = dwCurObjNum;
|
| - }
|
| - if (bNewStart) {
|
| - dwStartObjNum = dwCurObjNum;
|
| - }
|
| - bNewStart = bExist;
|
| - dwLastObjNum = dwCurObjNum;
|
| + int32_t j = i;
|
| + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| + while (j < iCount) {
|
| + if (++j == iCount) {
|
| + break;
|
| + }
|
| + FX_DWORD dwCurrent = m_NewObjNumArray.ElementAt(j);
|
| + if (dwCurrent - objnum > 1) {
|
| + break;
|
| + }
|
| + objnum = dwCurrent;
|
| + }
|
| + objnum = m_NewObjNumArray.ElementAt(i);
|
| + if (objnum == 1) {
|
| + str.Format("0 %d\r\n0000000000 65536 f\r\n", j - i + 1);
|
| + } else {
|
| + str.Format("%d %d\r\n", objnum, j - i);
|
| + }
|
| + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| + return -1;
|
| + }
|
| + while (i < j) {
|
| + objnum = m_NewObjNumArray.ElementAt(i++);
|
| + str.Format("%010d 00000 n\r\n", m_ObjectOffset[objnum]);
|
| + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| + if (pPause && (i % 100) == 0 && pPause->NeedToPauseNow()) {
|
| + m_Pos = (void*)(uintptr_t)i;
|
| + return 1;
|
| + }
|
| }
|
| - m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| - m_ObjectSize.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1);
|
| + m_iStage = 90;
|
| + }
|
| + return m_iStage;
|
| }
|
| -void CPDF_Creator::AppendNewObjNum(FX_DWORD objbum)
|
| -{
|
| - int32_t iStart = 0, iFind = 0;
|
| - int32_t iEnd = m_NewObjNumArray.GetUpperBound();
|
| - while (iStart <= iEnd) {
|
| - int32_t iMid = (iStart + iEnd) / 2;
|
| - FX_DWORD dwMid = m_NewObjNumArray.ElementAt(iMid);
|
| - if (objbum < dwMid) {
|
| - iEnd = iMid - 1;
|
| - } else {
|
| - if (iMid == iEnd) {
|
| - iFind = iMid + 1;
|
| - break;
|
| - }
|
| - FX_DWORD dwNext = m_NewObjNumArray.ElementAt(iMid + 1);
|
| - if (objbum < dwNext) {
|
| - iFind = iMid + 1;
|
| - break;
|
| - }
|
| - iStart = iMid + 1;
|
| - }
|
| - }
|
| - m_NewObjNumArray.InsertAt(iFind, objbum);
|
| +static int32_t _OutPutIndex(CFX_FileBufferArchive* pFile, FX_FILESIZE offset) {
|
| + FXSYS_assert(pFile);
|
| + if (sizeof(offset) > 4) {
|
| + if (FX_GETBYTEOFFSET32(offset)) {
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET56(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET48(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET40(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET32(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET24(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET16(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET8(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(FX_GETBYTEOFFSET0(offset)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pFile->AppendByte(0) < 0) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| }
|
| -int32_t CPDF_Creator::WriteDoc_Stage1(IFX_Pause *pPause)
|
| -{
|
| - FXSYS_assert(m_iStage > -1 || m_iStage < 20);
|
| - if (m_iStage == 0) {
|
| - if (m_pParser == NULL) {
|
| - m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
|
| - }
|
| - if (m_bSecurityChanged && (m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0) {
|
| - m_dwFlags &= ~FPDFCREATE_INCREMENTAL;
|
| - }
|
| - CPDF_Dictionary* pDict = m_pDocument->GetRoot();
|
| - m_pMetadata = pDict ? pDict->GetElementValue(FX_BSTRC("Metadata")) : NULL;
|
| - if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
|
| - m_pXRefStream = new CPDF_XRefStream;
|
| - m_pXRefStream->Start();
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser) {
|
| - FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
|
| - m_pXRefStream->m_PrevOffset = prev;
|
| - }
|
| - }
|
| - m_iStage = 10;
|
| - }
|
| - if (m_iStage == 10) {
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0) {
|
| - if (m_File.AppendString(FX_BSTRC("%PDF-1.")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += 7;
|
| - int32_t version = 7;
|
| - if (m_FileVersion) {
|
| - version = m_FileVersion;
|
| - } else if (m_pParser) {
|
| - version = m_pParser->GetFileVersion();
|
| - }
|
| - int32_t len = m_File.AppendDWord(version % 10);
|
| - if (len < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - if ((len = m_File.AppendString(FX_BSTRC("\r\n%\xA1\xB3\xC5\xD7\r\n"))) < 0) {
|
| - return -1;
|
| - }
|
| - m_Offset += len;
|
| - InitOldObjNumOffsets();
|
| - m_iStage = 20;
|
| - } else {
|
| - IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
|
| - m_Offset = pSrcFile->GetSize();
|
| - m_Pos = (void*)(uintptr_t)m_Offset;
|
| - m_iStage = 15;
|
| - }
|
| - }
|
| - if (m_iStage == 15) {
|
| - if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_Pos) {
|
| - IFX_FileRead* pSrcFile = m_pParser->GetFileAccess();
|
| - uint8_t buffer[4096];
|
| - FX_DWORD src_size = (FX_DWORD)(uintptr_t)m_Pos;
|
| - while (src_size) {
|
| - FX_DWORD block_size = src_size > 4096 ? 4096 : src_size;
|
| - if (!pSrcFile->ReadBlock(buffer, m_Offset - src_size, block_size)) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendBlock(buffer, block_size) < 0) {
|
| - return -1;
|
| - }
|
| - src_size -= block_size;
|
| - if (pPause && pPause->NeedToPauseNow()) {
|
| - m_Pos = (void*)(uintptr_t)src_size;
|
| - return 1;
|
| - }
|
| - }
|
| - }
|
| - if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_pParser->GetLastXRefOffset() == 0) {
|
| - InitOldObjNumOffsets();
|
| - FX_DWORD dwEnd = m_pParser->GetLastObjNum();
|
| - FX_BOOL bObjStm = (m_dwFlags & FPDFCREATE_OBJECTSTREAM) != 0;
|
| - for (FX_DWORD objnum = 0; objnum <= dwEnd; objnum++) {
|
| - if (m_pParser->m_V5Type[objnum] == 0 || m_pParser->m_V5Type[objnum] == 255) {
|
| - continue;
|
| - }
|
| - m_ObjectOffset[objnum] = m_pParser->m_CrossRef[objnum];
|
| - if (bObjStm) {
|
| - m_pXRefStream->AddObjectNumberToIndexArray(objnum);
|
| - }
|
| - }
|
| - if (bObjStm) {
|
| - m_pXRefStream->EndXRefStream(this);
|
| - m_pXRefStream->Start();
|
| - }
|
| - }
|
| - m_iStage = 20;
|
| +int32_t CPDF_Creator::WriteDoc_Stage4(IFX_Pause* pPause) {
|
| + FXSYS_assert(m_iStage >= 90);
|
| + if ((m_dwFlags & FPDFCREATE_OBJECTSTREAM) == 0) {
|
| + FX_BOOL bXRefStream =
|
| + (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser->IsXRefStream();
|
| + if (!bXRefStream) {
|
| + if (m_File.AppendString(FX_BSTRC("trailer\r\n<<")) < 0) {
|
| + return -1;
|
| + }
|
| + } else {
|
| + if (m_File.AppendDWord(m_pDocument->m_LastObjNum + 1) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 obj <<")) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - InitNewObjNumOffsets();
|
| - return m_iStage;
|
| -}
|
| -int32_t CPDF_Creator::WriteDoc_Stage2(IFX_Pause *pPause)
|
| -{
|
| - FXSYS_assert(m_iStage >= 20 || m_iStage < 30);
|
| - if (m_iStage == 20) {
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 && m_pParser) {
|
| - m_Pos = (void*)(uintptr_t)0;
|
| - m_iStage = 21;
|
| + if (m_pParser) {
|
| + CPDF_Dictionary* p = m_pParser->m_pTrailer;
|
| + FX_POSITION pos = p->GetStartPos();
|
| + while (pos) {
|
| + CFX_ByteString key;
|
| + CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| + if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") ||
|
| + key == FX_BSTRC("Filter") || key == FX_BSTRC("Index") ||
|
| + key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
|
| + key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") ||
|
| + key == FX_BSTRC("ID")) {
|
| + continue;
|
| + }
|
| + if (m_File.AppendString((FX_BSTRC("/"))) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(PDF_NameEncode(key)) < 0) {
|
| + return -1;
|
| + }
|
| + if (pValue->GetObjNum()) {
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendDWord(pValue->GetObjNum()) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + return -1;
|
| + }
|
| } else {
|
| - m_iStage = 25;
|
| + FX_FILESIZE offset = 0;
|
| + if (PDF_CreatorAppendObject(pValue, &m_File, offset) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - }
|
| - if (m_iStage == 21) {
|
| - int32_t iRet = WriteOldObjs(pPause);
|
| - if (iRet) {
|
| - return iRet;
|
| + }
|
| + } else {
|
| + if (m_File.AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendDWord(m_pDocument->m_pRootDict->GetObjNum()) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_pDocument->m_pInfoDict) {
|
| + if (m_File.AppendString(FX_BSTRC("/Info ")) < 0) {
|
| + return -1;
|
| }
|
| - m_iStage = 25;
|
| - }
|
| - if (m_iStage == 25) {
|
| - m_Pos = (void*)(uintptr_t)0;
|
| - m_iStage = 26;
|
| - }
|
| - if (m_iStage == 26) {
|
| - int32_t iRet = WriteNewObjs((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0, pPause);
|
| - if (iRet) {
|
| - return iRet;
|
| + if (m_File.AppendDWord(m_pDocument->m_pInfoDict->GetObjNum()) < 0) {
|
| + return -1;
|
| }
|
| - m_iStage = 27;
|
| - }
|
| - if (m_iStage == 27) {
|
| - if (NULL != m_pEncryptDict && 0 == m_pEncryptDict->GetObjNum()) {
|
| - m_dwLastObjNum += 1;
|
| - FX_FILESIZE saveOffset = m_Offset;
|
| - if (WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict) < 0) {
|
| - return -1;
|
| - }
|
| - m_ObjectOffset.Add(m_dwLastObjNum, 1);
|
| - m_ObjectOffset[m_dwLastObjNum] = saveOffset;
|
| - m_ObjectSize.Add(m_dwLastObjNum, 1);
|
| - m_ObjectSize[m_dwLastObjNum] = m_Offset - saveOffset;
|
| - m_dwEnryptObjNum = m_dwLastObjNum;
|
| - if (m_dwFlags & FPDFCREATE_INCREMENTAL) {
|
| - m_NewObjNumArray.Add(m_dwLastObjNum);
|
| - }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| + return -1;
|
| }
|
| - m_iStage = 80;
|
| + }
|
| }
|
| - return m_iStage;
|
| -}
|
| -int32_t CPDF_Creator::WriteDoc_Stage3(IFX_Pause *pPause)
|
| -{
|
| - FXSYS_assert(m_iStage >= 80 || m_iStage < 90);
|
| - FX_DWORD dwLastObjNum = m_dwLastObjNum;
|
| - if (m_iStage == 80) {
|
| - m_XrefStart = m_Offset;
|
| - if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
|
| - m_pXRefStream->End(this, TRUE);
|
| - m_XrefStart = m_pXRefStream->m_PrevOffset;
|
| - m_iStage = 90;
|
| - } else if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || !m_pParser->IsXRefStream()) {
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || m_pParser->GetLastXRefOffset() == 0) {
|
| - CFX_ByteString str;
|
| - str = m_ObjectOffset.GetPtrAt(1) ? FX_BSTRC("xref\r\n") : FX_BSTRC("xref\r\n0 1\r\n0000000000 65536 f\r\n");
|
| - if (m_File.AppendString(str) < 0) {
|
| - return -1;
|
| - }
|
| - m_Pos = (void*)(uintptr_t)1;
|
| - m_iStage = 81;
|
| - } else {
|
| - if (m_File.AppendString(FX_BSTRC("xref\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - m_Pos = (void*)(uintptr_t)0;
|
| - m_iStage = 82;
|
| - }
|
| - } else {
|
| - m_iStage = 90;
|
| - }
|
| + if (m_pEncryptDict) {
|
| + if (m_File.AppendString(FX_BSTRC("/Encrypt")) < 0) {
|
| + return -1;
|
| + }
|
| + FX_DWORD dwObjNum = m_pEncryptDict->GetObjNum();
|
| + if (dwObjNum == 0) {
|
| + dwObjNum = m_pDocument->GetLastObjNum() + 1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendDWord(dwObjNum) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - if (m_iStage == 81) {
|
| - CFX_ByteString str;
|
| - FX_DWORD i = (FX_DWORD)(uintptr_t)m_Pos, j;
|
| - while (i <= dwLastObjNum) {
|
| - while (i <= dwLastObjNum && !m_ObjectOffset.GetPtrAt(i)) {
|
| - i++;
|
| - }
|
| - if (i > dwLastObjNum) {
|
| - break;
|
| - }
|
| - j = i;
|
| - while (j <= dwLastObjNum && m_ObjectOffset.GetPtrAt(j)) {
|
| - j++;
|
| - }
|
| - if (i == 1) {
|
| - str.Format("0 %d\r\n0000000000 65536 f\r\n", j);
|
| - } else {
|
| - str.Format("%d %d\r\n", i, j - i);
|
| - }
|
| - if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - while (i < j) {
|
| - str.Format("%010d 00000 n\r\n", m_ObjectOffset[i ++]);
|
| - if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - if (i > dwLastObjNum) {
|
| - break;
|
| - }
|
| - if (pPause && pPause->NeedToPauseNow()) {
|
| - m_Pos = (void*)(uintptr_t)i;
|
| - return 1;
|
| - }
|
| - }
|
| - m_iStage = 90;
|
| + if (m_File.AppendString(FX_BSTRC("/Size ")) < 0) {
|
| + return -1;
|
| }
|
| - if (m_iStage == 82) {
|
| - CFX_ByteString str;
|
| - int32_t iCount = m_NewObjNumArray.GetSize();
|
| - int32_t i = (int32_t)(uintptr_t)m_Pos;
|
| - while (i < iCount) {
|
| - int32_t j = i;
|
| - FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| - while (j < iCount) {
|
| - if (++j == iCount) {
|
| - break;
|
| - }
|
| - FX_DWORD dwCurrent = m_NewObjNumArray.ElementAt(j);
|
| - if (dwCurrent - objnum > 1) {
|
| - break;
|
| - }
|
| - objnum = dwCurrent;
|
| - }
|
| - objnum = m_NewObjNumArray.ElementAt(i);
|
| - if (objnum == 1) {
|
| - str.Format("0 %d\r\n0000000000 65536 f\r\n", j - i + 1);
|
| - } else {
|
| - str.Format("%d %d\r\n", objnum, j - i);
|
| - }
|
| - if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - while (i < j) {
|
| - objnum = m_NewObjNumArray.ElementAt(i++);
|
| - str.Format("%010d 00000 n\r\n", m_ObjectOffset[objnum]);
|
| - if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - if (pPause && (i % 100) == 0 && pPause->NeedToPauseNow()) {
|
| - m_Pos = (void*)(uintptr_t)i;
|
| - return 1;
|
| - }
|
| - }
|
| - m_iStage = 90;
|
| + if (m_File.AppendDWord(m_dwLastObjNum + (bXRefStream ? 2 : 1)) < 0) {
|
| + return -1;
|
| }
|
| - return m_iStage;
|
| -}
|
| -static int32_t _OutPutIndex(CFX_FileBufferArchive* pFile, FX_FILESIZE offset)
|
| -{
|
| - FXSYS_assert(pFile);
|
| - if (sizeof(offset) > 4) {
|
| - if (FX_GETBYTEOFFSET32(offset)) {
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET56(offset)) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET48(offset)) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET40(offset)) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET32(offset)) < 0) {
|
| - return -1;
|
| - }
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0) {
|
| + FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
|
| + if (prev) {
|
| + if (m_File.AppendString(FX_BSTRC("/Prev ")) < 0) {
|
| + return -1;
|
| }
|
| + FX_CHAR offset_buf[20];
|
| + FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| + FXSYS_i64toa(prev, offset_buf, 10);
|
| + if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
|
| + return -1;
|
| + }
|
| + }
|
| }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET24(offset)) < 0) {
|
| - return -1;
|
| - }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET16(offset)) < 0) {
|
| + if (m_pIDArray) {
|
| + if (m_File.AppendString((FX_BSTRC("/ID"))) < 0) {
|
| return -1;
|
| - }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET8(offset)) < 0) {
|
| + }
|
| + FX_FILESIZE offset = 0;
|
| + if (PDF_CreatorAppendObject(m_pIDArray, &m_File, offset) < 0) {
|
| return -1;
|
| + }
|
| }
|
| - if (pFile->AppendByte(FX_GETBYTEOFFSET0(offset)) < 0) {
|
| + if (!bXRefStream) {
|
| + if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
|
| return -1;
|
| - }
|
| - if (pFile->AppendByte(0) < 0) {
|
| + }
|
| + } else {
|
| + if (m_File.AppendString(FX_BSTRC("/W[0 4 1]/Index[")) < 0) {
|
| return -1;
|
| - }
|
| - return 0;
|
| -}
|
| -int32_t CPDF_Creator::WriteDoc_Stage4(IFX_Pause *pPause)
|
| -{
|
| - FXSYS_assert(m_iStage >= 90);
|
| - if ((m_dwFlags & FPDFCREATE_OBJECTSTREAM) == 0) {
|
| - FX_BOOL bXRefStream = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser->IsXRefStream();
|
| - if (!bXRefStream) {
|
| - if (m_File.AppendString(FX_BSTRC("trailer\r\n<<")) < 0) {
|
| - return -1;
|
| - }
|
| - } else {
|
| - if (m_File.AppendDWord(m_pDocument->m_LastObjNum + 1) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 obj <<")) < 0) {
|
| - return -1;
|
| - }
|
| + }
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser &&
|
| + m_pParser->GetLastXRefOffset() == 0) {
|
| + FX_DWORD i = 0;
|
| + for (i = 0; i < m_dwLastObjNum; i++) {
|
| + if (!m_ObjectOffset.GetPtrAt(i)) {
|
| + continue;
|
| + }
|
| + if (m_File.AppendDWord(i) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
|
| + return -1;
|
| + }
|
| }
|
| - if (m_pParser) {
|
| - CPDF_Dictionary* p = m_pParser->m_pTrailer;
|
| - FX_POSITION pos = p->GetStartPos();
|
| - while (pos) {
|
| - CFX_ByteString key;
|
| - CPDF_Object* pValue = p->GetNextElement(pos, key);
|
| - if (key == FX_BSTRC("Encrypt") || key == FX_BSTRC("Size") || key == FX_BSTRC("Filter") ||
|
| - key == FX_BSTRC("Index") || key == FX_BSTRC("Length") || key == FX_BSTRC("Prev") ||
|
| - key == FX_BSTRC("W") || key == FX_BSTRC("XRefStm") || key == FX_BSTRC("ID")) {
|
| - continue;
|
| - }
|
| - if (m_File.AppendString((FX_BSTRC("/"))) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(PDF_NameEncode(key)) < 0) {
|
| - return -1;
|
| - }
|
| - if (pValue->GetObjNum()) {
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(pValue->GetObjNum()) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| - return -1;
|
| - }
|
| - } else {
|
| - FX_FILESIZE offset = 0;
|
| - if (PDF_CreatorAppendObject(pValue, &m_File, offset) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - }
|
| - } else {
|
| - if (m_File.AppendString(FX_BSTRC("\r\n/Root ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(m_pDocument->m_pRootDict->GetObjNum()) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_pDocument->m_pInfoDict) {
|
| - if (m_File.AppendString(FX_BSTRC("/Info ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(m_pDocument->m_pInfoDict->GetObjNum()) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| + if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
|
| + return -1;
|
| }
|
| - if (m_pEncryptDict) {
|
| - if (m_File.AppendString(FX_BSTRC("/Encrypt")) < 0) {
|
| - return -1;
|
| - }
|
| - FX_DWORD dwObjNum = m_pEncryptDict->GetObjNum();
|
| - if (dwObjNum == 0) {
|
| - dwObjNum = m_pDocument->GetLastObjNum() + 1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(dwObjNum) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 0 R ")) < 0) {
|
| - return -1;
|
| - }
|
| + if (m_File.AppendDWord(m_dwLastObjNum * 5) < 0) {
|
| + return -1;
|
| }
|
| - if (m_File.AppendString(FX_BSTRC("/Size ")) < 0) {
|
| - return -1;
|
| + if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
|
| + return -1;
|
| }
|
| - if (m_File.AppendDWord(m_dwLastObjNum + (bXRefStream ? 2 : 1)) < 0) {
|
| + for (i = 0; i < m_dwLastObjNum; i++) {
|
| + FX_FILESIZE* offset = m_ObjectOffset.GetPtrAt(i);
|
| + if (!offset) {
|
| + continue;
|
| + }
|
| + _OutPutIndex(&m_File, *offset);
|
| + }
|
| + } else {
|
| + int count = m_NewObjNumArray.GetSize();
|
| + int32_t i = 0;
|
| + for (i = 0; i < count; i++) {
|
| + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| + if (m_File.AppendDWord(objnum) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
|
| return -1;
|
| + }
|
| }
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0) {
|
| - FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
|
| - if (prev) {
|
| - if (m_File.AppendString(FX_BSTRC("/Prev ")) < 0) {
|
| - return -1;
|
| - }
|
| - FX_CHAR offset_buf[20];
|
| - FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| - FXSYS_i64toa(prev, offset_buf, 10);
|
| - if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| + if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
|
| + return -1;
|
| }
|
| - if (m_pIDArray) {
|
| - if (m_File.AppendString((FX_BSTRC("/ID"))) < 0) {
|
| - return -1;
|
| - }
|
| - FX_FILESIZE offset = 0;
|
| - if (PDF_CreatorAppendObject(m_pIDArray, &m_File, offset) < 0) {
|
| - return -1;
|
| - }
|
| + if (m_File.AppendDWord(count * 5) < 0) {
|
| + return -1;
|
| }
|
| - if (!bXRefStream) {
|
| - if (m_File.AppendString(FX_BSTRC(">>")) < 0) {
|
| - return -1;
|
| - }
|
| - } else {
|
| - if (m_File.AppendString(FX_BSTRC("/W[0 4 1]/Index[")) < 0) {
|
| - return -1;
|
| - }
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser && m_pParser->GetLastXRefOffset() == 0) {
|
| - FX_DWORD i = 0;
|
| - for (i = 0; i < m_dwLastObjNum; i++) {
|
| - if (!m_ObjectOffset.GetPtrAt(i)) {
|
| - continue;
|
| - }
|
| - if (m_File.AppendDWord(i) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(m_dwLastObjNum * 5) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - for (i = 0; i < m_dwLastObjNum; i++) {
|
| - FX_FILESIZE* offset = m_ObjectOffset.GetPtrAt(i);
|
| - if (!offset) {
|
| - continue;
|
| - }
|
| - _OutPutIndex(&m_File, *offset);
|
| - }
|
| - } else {
|
| - int count = m_NewObjNumArray.GetSize();
|
| - int32_t i = 0;
|
| - for (i = 0; i < count; i++) {
|
| - FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| - if (m_File.AppendDWord(objnum) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(" 1 ")) < 0) {
|
| - return -1;
|
| - }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("]/Length ")) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendDWord(count * 5) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - for (i = 0; i < count; i++) {
|
| - FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| - FX_FILESIZE offset = m_ObjectOffset[objnum];
|
| - _OutPutIndex(&m_File, offset);
|
| - }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("\r\nendstream")) < 0) {
|
| - return -1;
|
| - }
|
| + if (m_File.AppendString(FX_BSTRC(">>stream\r\n")) < 0) {
|
| + return -1;
|
| }
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("\r\nstartxref\r\n")) < 0) {
|
| - return -1;
|
| - }
|
| - FX_CHAR offset_buf[20];
|
| - FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| - FXSYS_i64toa(m_XrefStart, offset_buf, 10);
|
| - if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
|
| - return -1;
|
| - }
|
| - if (m_File.AppendString(FX_BSTRC("\r\n%%EOF\r\n")) < 0) {
|
| + for (i = 0; i < count; i++) {
|
| + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i);
|
| + FX_FILESIZE offset = m_ObjectOffset[objnum];
|
| + _OutPutIndex(&m_File, offset);
|
| + }
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC("\r\nendstream")) < 0) {
|
| return -1;
|
| - }
|
| - m_File.Flush();
|
| - return m_iStage = 100;
|
| + }
|
| + }
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC("\r\nstartxref\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + FX_CHAR offset_buf[20];
|
| + FXSYS_memset(offset_buf, 0, sizeof(offset_buf));
|
| + FXSYS_i64toa(m_XrefStart, offset_buf, 10);
|
| + if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) {
|
| + return -1;
|
| + }
|
| + if (m_File.AppendString(FX_BSTRC("\r\n%%EOF\r\n")) < 0) {
|
| + return -1;
|
| + }
|
| + m_File.Flush();
|
| + return m_iStage = 100;
|
| }
|
| -void CPDF_Creator::Clear()
|
| -{
|
| - delete m_pXRefStream;
|
| - m_pXRefStream = NULL;
|
| - m_File.Clear();
|
| - m_NewObjNumArray.RemoveAll();
|
| - if (m_pIDArray) {
|
| - m_pIDArray->Release();
|
| - m_pIDArray = NULL;
|
| - }
|
| +void CPDF_Creator::Clear() {
|
| + delete m_pXRefStream;
|
| + m_pXRefStream = NULL;
|
| + m_File.Clear();
|
| + m_NewObjNumArray.RemoveAll();
|
| + if (m_pIDArray) {
|
| + m_pIDArray->Release();
|
| + m_pIDArray = NULL;
|
| + }
|
| }
|
| -FX_BOOL CPDF_Creator::Create(const FX_CHAR* filename, FX_DWORD flags)
|
| -{
|
| - if (!m_File.AttachFile(filename)) {
|
| - return FALSE;
|
| - }
|
| - FX_BOOL bRet = Create(flags);
|
| - if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
|
| - Clear();
|
| - }
|
| - return bRet;
|
| +FX_BOOL CPDF_Creator::Create(const FX_CHAR* filename, FX_DWORD flags) {
|
| + if (!m_File.AttachFile(filename)) {
|
| + return FALSE;
|
| + }
|
| + FX_BOOL bRet = Create(flags);
|
| + if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
|
| + Clear();
|
| + }
|
| + return bRet;
|
| }
|
| -FX_BOOL CPDF_Creator::Create(const FX_WCHAR* filename, FX_DWORD flags)
|
| -{
|
| - if (!m_File.AttachFile(filename)) {
|
| - return FALSE;
|
| - }
|
| - FX_BOOL bRet = Create(flags);
|
| - if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
|
| - Clear();
|
| - }
|
| - return bRet;
|
| +FX_BOOL CPDF_Creator::Create(const FX_WCHAR* filename, FX_DWORD flags) {
|
| + if (!m_File.AttachFile(filename)) {
|
| + return FALSE;
|
| + }
|
| + FX_BOOL bRet = Create(flags);
|
| + if (!bRet || !(flags & FPDFCREATE_PROGRESSIVE)) {
|
| + Clear();
|
| + }
|
| + return bRet;
|
| }
|
| -FX_BOOL CPDF_Creator::Create(IFX_StreamWrite* pFile, FX_DWORD flags)
|
| -{
|
| - if (!pFile) {
|
| - return FALSE;
|
| - }
|
| - if (!m_File.AttachFile(pFile, FALSE)) {
|
| - return FALSE;
|
| - }
|
| - return Create(flags);
|
| +FX_BOOL CPDF_Creator::Create(IFX_StreamWrite* pFile, FX_DWORD flags) {
|
| + if (!pFile) {
|
| + return FALSE;
|
| + }
|
| + if (!m_File.AttachFile(pFile, FALSE)) {
|
| + return FALSE;
|
| + }
|
| + return Create(flags);
|
| }
|
| -FX_BOOL CPDF_Creator::Create(FX_DWORD flags)
|
| -{
|
| - m_dwFlags = flags;
|
| - m_iStage = 0;
|
| - m_Offset = 0;
|
| - m_dwLastObjNum = m_pDocument->GetLastObjNum();
|
| - m_ObjectOffset.Clear();
|
| - m_ObjectSize.Clear();
|
| - m_NewObjNumArray.RemoveAll();
|
| - InitID();
|
| - if (flags & FPDFCREATE_PROGRESSIVE) {
|
| - return TRUE;
|
| - }
|
| - return Continue(NULL) > -1;
|
| +FX_BOOL CPDF_Creator::Create(FX_DWORD flags) {
|
| + m_dwFlags = flags;
|
| + m_iStage = 0;
|
| + m_Offset = 0;
|
| + m_dwLastObjNum = m_pDocument->GetLastObjNum();
|
| + m_ObjectOffset.Clear();
|
| + m_ObjectSize.Clear();
|
| + m_NewObjNumArray.RemoveAll();
|
| + InitID();
|
| + if (flags & FPDFCREATE_PROGRESSIVE) {
|
| + return TRUE;
|
| + }
|
| + return Continue(NULL) > -1;
|
| }
|
| -void CPDF_Creator::InitID(FX_BOOL bDefault )
|
| -{
|
| - CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : NULL;
|
| - FX_BOOL bNewId = !m_pIDArray;
|
| - if (!m_pIDArray) {
|
| - FX_DWORD* pBuffer = NULL;
|
| - m_pIDArray = CPDF_Array::Create();
|
| - CPDF_Object* pID1 = pOldIDArray ? pOldIDArray->GetElement(0) : NULL;
|
| - if (pID1) {
|
| - m_pIDArray->Add(pID1->Clone());
|
| - } else {
|
| - pBuffer = FX_Alloc(FX_DWORD, 4);
|
| - PDF_GenerateFileID((FX_DWORD)(uintptr_t)this, m_dwLastObjNum, pBuffer);
|
| - CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
|
| - m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
|
| - }
|
| - if (pBuffer) {
|
| - FX_Free(pBuffer);
|
| - }
|
| - }
|
| - if (!bDefault) {
|
| - return;
|
| - }
|
| - if (pOldIDArray) {
|
| - CPDF_Object* pID2 = pOldIDArray->GetElement(1);
|
| - if ((m_dwFlags & FPDFCREATE_INCREMENTAL) && m_pEncryptDict && pID2) {
|
| - m_pIDArray->Add(pID2->Clone());
|
| - return;
|
| - }
|
| - FX_DWORD* pBuffer = FX_Alloc(FX_DWORD, 4);
|
| - PDF_GenerateFileID((FX_DWORD)(uintptr_t)this, m_dwLastObjNum, pBuffer);
|
| - CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
|
| - m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
|
| - FX_Free(pBuffer);
|
| - return;
|
| - }
|
| - m_pIDArray->Add(m_pIDArray->GetElement(0)->Clone());
|
| - if (m_pEncryptDict && !pOldIDArray && m_pParser && bNewId) {
|
| - if (m_pEncryptDict->GetString(FX_BSTRC("Filter")) == FX_BSTRC("Standard")) {
|
| - CPDF_StandardSecurityHandler handler;
|
| - CFX_ByteString user_pass = m_pParser->GetPassword();
|
| - FX_DWORD flag = PDF_ENCRYPT_CONTENT;
|
| - handler.OnCreate(m_pEncryptDict, m_pIDArray, (const uint8_t*)user_pass, user_pass.GetLength(), flag);
|
| - if (m_bNewCrypto) {
|
| - delete m_pCryptoHandler;
|
| - }
|
| - m_pCryptoHandler = new CPDF_StandardCryptoHandler;
|
| - m_pCryptoHandler->Init(m_pEncryptDict, &handler);
|
| - m_bNewCrypto = TRUE;
|
| - m_bSecurityChanged = TRUE;
|
| - }
|
| +void CPDF_Creator::InitID(FX_BOOL bDefault) {
|
| + CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : NULL;
|
| + FX_BOOL bNewId = !m_pIDArray;
|
| + if (!m_pIDArray) {
|
| + FX_DWORD* pBuffer = NULL;
|
| + m_pIDArray = CPDF_Array::Create();
|
| + CPDF_Object* pID1 = pOldIDArray ? pOldIDArray->GetElement(0) : NULL;
|
| + if (pID1) {
|
| + m_pIDArray->Add(pID1->Clone());
|
| + } else {
|
| + pBuffer = FX_Alloc(FX_DWORD, 4);
|
| + PDF_GenerateFileID((FX_DWORD)(uintptr_t) this, m_dwLastObjNum, pBuffer);
|
| + CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
|
| + m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
|
| + }
|
| + if (pBuffer) {
|
| + FX_Free(pBuffer);
|
| + }
|
| + }
|
| + if (!bDefault) {
|
| + return;
|
| + }
|
| + if (pOldIDArray) {
|
| + CPDF_Object* pID2 = pOldIDArray->GetElement(1);
|
| + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) && m_pEncryptDict && pID2) {
|
| + m_pIDArray->Add(pID2->Clone());
|
| + return;
|
| + }
|
| + FX_DWORD* pBuffer = FX_Alloc(FX_DWORD, 4);
|
| + PDF_GenerateFileID((FX_DWORD)(uintptr_t) this, m_dwLastObjNum, pBuffer);
|
| + CFX_ByteStringC bsBuffer((const uint8_t*)pBuffer, 4 * sizeof(FX_DWORD));
|
| + m_pIDArray->Add(CPDF_String::Create(bsBuffer, TRUE), m_pDocument);
|
| + FX_Free(pBuffer);
|
| + return;
|
| + }
|
| + m_pIDArray->Add(m_pIDArray->GetElement(0)->Clone());
|
| + if (m_pEncryptDict && !pOldIDArray && m_pParser && bNewId) {
|
| + if (m_pEncryptDict->GetString(FX_BSTRC("Filter")) == FX_BSTRC("Standard")) {
|
| + CPDF_StandardSecurityHandler handler;
|
| + CFX_ByteString user_pass = m_pParser->GetPassword();
|
| + FX_DWORD flag = PDF_ENCRYPT_CONTENT;
|
| + handler.OnCreate(m_pEncryptDict, m_pIDArray, (const uint8_t*)user_pass,
|
| + user_pass.GetLength(), flag);
|
| + if (m_bNewCrypto) {
|
| + delete m_pCryptoHandler;
|
| + }
|
| + m_pCryptoHandler = new CPDF_StandardCryptoHandler;
|
| + m_pCryptoHandler->Init(m_pEncryptDict, &handler);
|
| + m_bNewCrypto = TRUE;
|
| + m_bSecurityChanged = TRUE;
|
| }
|
| + }
|
| }
|
| -int32_t CPDF_Creator::Continue(IFX_Pause *pPause)
|
| -{
|
| - if (m_iStage < 0) {
|
| - return m_iStage;
|
| - }
|
| - int32_t iRet = 0;
|
| - while (m_iStage < 100) {
|
| - if (m_iStage < 20) {
|
| - iRet = WriteDoc_Stage1(pPause);
|
| - } else if (m_iStage < 30) {
|
| - iRet = WriteDoc_Stage2(pPause);
|
| - } else if (m_iStage < 90) {
|
| - iRet = WriteDoc_Stage3(pPause);
|
| - } else {
|
| - iRet = WriteDoc_Stage4(pPause);
|
| - }
|
| - if (iRet < m_iStage) {
|
| - break;
|
| - }
|
| +int32_t CPDF_Creator::Continue(IFX_Pause* pPause) {
|
| + if (m_iStage < 0) {
|
| + return m_iStage;
|
| + }
|
| + int32_t iRet = 0;
|
| + while (m_iStage < 100) {
|
| + if (m_iStage < 20) {
|
| + iRet = WriteDoc_Stage1(pPause);
|
| + } else if (m_iStage < 30) {
|
| + iRet = WriteDoc_Stage2(pPause);
|
| + } else if (m_iStage < 90) {
|
| + iRet = WriteDoc_Stage3(pPause);
|
| + } else {
|
| + iRet = WriteDoc_Stage4(pPause);
|
| }
|
| - if (iRet < 1 || m_iStage == 100) {
|
| - m_iStage = -1;
|
| - Clear();
|
| - return iRet > 99 ? 0 : (iRet < 1 ? -1 : iRet);
|
| + if (iRet < m_iStage) {
|
| + break;
|
| }
|
| - return m_iStage;
|
| + }
|
| + if (iRet < 1 || m_iStage == 100) {
|
| + m_iStage = -1;
|
| + Clear();
|
| + return iRet > 99 ? 0 : (iRet < 1 ? -1 : iRet);
|
| + }
|
| + return m_iStage;
|
| }
|
| -FX_BOOL CPDF_Creator::SetFileVersion(int32_t fileVersion )
|
| -{
|
| - if (fileVersion < 10 || fileVersion > 17) {
|
| - return FALSE;
|
| - }
|
| - m_FileVersion = fileVersion;
|
| - return TRUE;
|
| +FX_BOOL CPDF_Creator::SetFileVersion(int32_t fileVersion) {
|
| + if (fileVersion < 10 || fileVersion > 17) {
|
| + return FALSE;
|
| + }
|
| + m_FileVersion = fileVersion;
|
| + return TRUE;
|
| }
|
| -void CPDF_Creator::RemoveSecurity()
|
| -{
|
| - ResetStandardSecurity();
|
| - m_bSecurityChanged = TRUE;
|
| - m_pEncryptDict = NULL;
|
| - m_pCryptoHandler = NULL;
|
| +void CPDF_Creator::RemoveSecurity() {
|
| + ResetStandardSecurity();
|
| + m_bSecurityChanged = TRUE;
|
| + m_pEncryptDict = NULL;
|
| + m_pCryptoHandler = NULL;
|
| }
|
| -void CPDF_Creator::ResetStandardSecurity()
|
| -{
|
| - if (m_bStandardSecurity || m_bNewCrypto) {
|
| - delete m_pCryptoHandler;
|
| - m_pCryptoHandler = NULL;
|
| - }
|
| - m_bNewCrypto = FALSE;
|
| - if (!m_bStandardSecurity) {
|
| - return;
|
| - }
|
| - if (m_pEncryptDict) {
|
| - m_pEncryptDict->Release();
|
| - m_pEncryptDict = NULL;
|
| - }
|
| - m_bStandardSecurity = FALSE;
|
| +void CPDF_Creator::ResetStandardSecurity() {
|
| + if (m_bStandardSecurity || m_bNewCrypto) {
|
| + delete m_pCryptoHandler;
|
| + m_pCryptoHandler = NULL;
|
| + }
|
| + m_bNewCrypto = FALSE;
|
| + if (!m_bStandardSecurity) {
|
| + return;
|
| + }
|
| + if (m_pEncryptDict) {
|
| + m_pEncryptDict->Release();
|
| + m_pEncryptDict = NULL;
|
| + }
|
| + m_bStandardSecurity = FALSE;
|
| }
|
|
|