| OLD | NEW |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #ifndef CORE_SRC_FXCRT_EXTENSION_H_ | 7 #ifndef CORE_SRC_FXCRT_EXTENSION_H_ |
| 8 #define CORE_SRC_FXCRT_EXTENSION_H_ | 8 #define CORE_SRC_FXCRT_EXTENSION_H_ |
| 9 | 9 |
| 10 #include "../../include/fxcrt/fx_safe_types.h" | 10 #include "../../include/fxcrt/fx_safe_types.h" |
| (...skipping 128 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 CFX_MemoryStream(FX_BOOL bConsecutive) | 139 CFX_MemoryStream(FX_BOOL bConsecutive) |
| 140 : m_dwCount(1) | 140 : m_dwCount(1) |
| 141 , m_nTotalSize(0) | 141 , m_nTotalSize(0) |
| 142 , m_nCurSize(0) | 142 , m_nCurSize(0) |
| 143 , m_nCurPos(0) | 143 , m_nCurPos(0) |
| 144 , m_nGrowSize(FX_MEMSTREAM_BlockSize) | 144 , m_nGrowSize(FX_MEMSTREAM_BlockSize) |
| 145 , m_bUseRange(FALSE) | 145 , m_bUseRange(FALSE) |
| 146 { | 146 { |
| 147 m_dwFlags = FX_MEMSTREAM_TakeOver | (bConsecutive ? FX_MEMSTREAM_Consecu
tive : 0); | 147 m_dwFlags = FX_MEMSTREAM_TakeOver | (bConsecutive ? FX_MEMSTREAM_Consecu
tive : 0); |
| 148 } | 148 } |
| 149 CFX_MemoryStream(FX_LPBYTE pBuffer, size_t nSize, FX_BOOL bTakeOver) | 149 CFX_MemoryStream(uint8_t* pBuffer, size_t nSize, FX_BOOL bTakeOver) |
| 150 : m_dwCount(1) | 150 : m_dwCount(1) |
| 151 , m_nTotalSize(nSize) | 151 , m_nTotalSize(nSize) |
| 152 , m_nCurSize(nSize) | 152 , m_nCurSize(nSize) |
| 153 , m_nCurPos(0) | 153 , m_nCurPos(0) |
| 154 , m_nGrowSize(FX_MEMSTREAM_BlockSize) | 154 , m_nGrowSize(FX_MEMSTREAM_BlockSize) |
| 155 , m_bUseRange(FALSE) | 155 , m_bUseRange(FALSE) |
| 156 { | 156 { |
| 157 m_Blocks.Add(pBuffer); | 157 m_Blocks.Add(pBuffer); |
| 158 m_dwFlags = FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOve
r : 0); | 158 m_dwFlags = FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOve
r : 0); |
| 159 } | 159 } |
| 160 ~CFX_MemoryStream() | 160 ~CFX_MemoryStream() |
| 161 { | 161 { |
| 162 if (m_dwFlags & FX_MEMSTREAM_TakeOver) { | 162 if (m_dwFlags & FX_MEMSTREAM_TakeOver) { |
| 163 for (int32_t i = 0; i < m_Blocks.GetSize(); i++) { | 163 for (int32_t i = 0; i < m_Blocks.GetSize(); i++) { |
| 164 FX_Free((FX_LPBYTE)m_Blocks[i]); | 164 FX_Free((uint8_t*)m_Blocks[i]); |
| 165 } | 165 } |
| 166 } | 166 } |
| 167 m_Blocks.RemoveAll(); | 167 m_Blocks.RemoveAll(); |
| 168 } | 168 } |
| 169 virtual IFX_FileStream* Retain() override | 169 virtual IFX_FileStream* Retain() override |
| 170 { | 170 { |
| 171 m_dwCount ++; | 171 m_dwCount ++; |
| 172 return this; | 172 return this; |
| 173 } | 173 } |
| 174 virtual void Release() override | 174 virtual void Release() override |
| (...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 233 offset = safeOffset.ValueOrDie(); | 233 offset = safeOffset.ValueOrDie(); |
| 234 | 234 |
| 235 FX_SAFE_SIZE_T newPos = size; | 235 FX_SAFE_SIZE_T newPos = size; |
| 236 newPos += offset; | 236 newPos += offset; |
| 237 if (!newPos.IsValid() || newPos.ValueOrDefault(0) == 0 || newPos.ValueOr
Die() > m_nCurSize) { | 237 if (!newPos.IsValid() || newPos.ValueOrDefault(0) == 0 || newPos.ValueOr
Die() > m_nCurSize) { |
| 238 return FALSE; | 238 return FALSE; |
| 239 } | 239 } |
| 240 | 240 |
| 241 m_nCurPos = newPos.ValueOrDie(); | 241 m_nCurPos = newPos.ValueOrDie(); |
| 242 if (m_dwFlags & FX_MEMSTREAM_Consecutive) { | 242 if (m_dwFlags & FX_MEMSTREAM_Consecutive) { |
| 243 FXSYS_memcpy32(buffer, (FX_LPBYTE)m_Blocks[0] + (size_t)offset, size
); | 243 FXSYS_memcpy32(buffer, (uint8_t*)m_Blocks[0] + (size_t)offset, size)
; |
| 244 return TRUE; | 244 return TRUE; |
| 245 } | 245 } |
| 246 size_t nStartBlock = (size_t)offset / m_nGrowSize; | 246 size_t nStartBlock = (size_t)offset / m_nGrowSize; |
| 247 offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); | 247 offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); |
| 248 while (size) { | 248 while (size) { |
| 249 size_t nRead = m_nGrowSize - (size_t)offset; | 249 size_t nRead = m_nGrowSize - (size_t)offset; |
| 250 if (nRead > size) { | 250 if (nRead > size) { |
| 251 nRead = size; | 251 nRead = size; |
| 252 } | 252 } |
| 253 FXSYS_memcpy32(buffer, (FX_LPBYTE)m_Blocks[(int)nStartBlock] + (size
_t)offset, nRead); | 253 FXSYS_memcpy32(buffer, (uint8_t*)m_Blocks[(int)nStartBlock] + (size_
t)offset, nRead); |
| 254 buffer = ((FX_LPBYTE)buffer) + nRead; | 254 buffer = ((uint8_t*)buffer) + nRead; |
| 255 size -= nRead; | 255 size -= nRead; |
| 256 nStartBlock ++; | 256 nStartBlock ++; |
| 257 offset = 0; | 257 offset = 0; |
| 258 } | 258 } |
| 259 return TRUE; | 259 return TRUE; |
| 260 } | 260 } |
| 261 virtual size_t ReadBlock(void* buffer, size_t s
ize) override | 261 virtual size_t ReadBlock(void* buffer, size_t s
ize) override |
| 262 { | 262 { |
| 263 if (m_nCurPos >= m_nCurSize) { | 263 if (m_nCurPos >= m_nCurSize) { |
| 264 return 0; | 264 return 0; |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 296 void* block = FX_Alloc(uint8_t, m_nTotalSize); | 296 void* block = FX_Alloc(uint8_t, m_nTotalSize); |
| 297 m_Blocks.Add(block); | 297 m_Blocks.Add(block); |
| 298 } else { | 298 } else { |
| 299 m_Blocks[0] = FX_Realloc(uint8_t, m_Blocks[0], m_nTotalSize)
; | 299 m_Blocks[0] = FX_Realloc(uint8_t, m_Blocks[0], m_nTotalSize)
; |
| 300 } | 300 } |
| 301 if (!m_Blocks[0]) { | 301 if (!m_Blocks[0]) { |
| 302 m_Blocks.RemoveAll(); | 302 m_Blocks.RemoveAll(); |
| 303 return FALSE; | 303 return FALSE; |
| 304 } | 304 } |
| 305 } | 305 } |
| 306 FXSYS_memcpy32((FX_LPBYTE)m_Blocks[0] + (size_t)offset, buffer, size
); | 306 FXSYS_memcpy32((uint8_t*)m_Blocks[0] + (size_t)offset, buffer, size)
; |
| 307 if (m_nCurSize < m_nCurPos) { | 307 if (m_nCurSize < m_nCurPos) { |
| 308 m_nCurSize = m_nCurPos; | 308 m_nCurSize = m_nCurPos; |
| 309 } | 309 } |
| 310 return TRUE; | 310 return TRUE; |
| 311 } | 311 } |
| 312 | 312 |
| 313 FX_SAFE_SIZE_T newPos = size; | 313 FX_SAFE_SIZE_T newPos = size; |
| 314 newPos += offset; | 314 newPos += offset; |
| 315 if (!newPos.IsValid()) { | 315 if (!newPos.IsValid()) { |
| 316 return FALSE; | 316 return FALSE; |
| 317 } | 317 } |
| 318 | 318 |
| 319 if (!ExpandBlocks(newPos.ValueOrDie())) { | 319 if (!ExpandBlocks(newPos.ValueOrDie())) { |
| 320 return FALSE; | 320 return FALSE; |
| 321 } | 321 } |
| 322 m_nCurPos = newPos.ValueOrDie(); | 322 m_nCurPos = newPos.ValueOrDie(); |
| 323 size_t nStartBlock = (size_t)offset / m_nGrowSize; | 323 size_t nStartBlock = (size_t)offset / m_nGrowSize; |
| 324 offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); | 324 offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); |
| 325 while (size) { | 325 while (size) { |
| 326 size_t nWrite = m_nGrowSize - (size_t)offset; | 326 size_t nWrite = m_nGrowSize - (size_t)offset; |
| 327 if (nWrite > size) { | 327 if (nWrite > size) { |
| 328 nWrite = size; | 328 nWrite = size; |
| 329 } | 329 } |
| 330 FXSYS_memcpy32((FX_LPBYTE)m_Blocks[(int)nStartBlock] + (size_t)offse
t, buffer, nWrite); | 330 FXSYS_memcpy32((uint8_t*)m_Blocks[(int)nStartBlock] + (size_t)offset
, buffer, nWrite); |
| 331 buffer = ((FX_LPBYTE)buffer) + nWrite; | 331 buffer = ((uint8_t*)buffer) + nWrite; |
| 332 size -= nWrite; | 332 size -= nWrite; |
| 333 nStartBlock ++; | 333 nStartBlock ++; |
| 334 offset = 0; | 334 offset = 0; |
| 335 } | 335 } |
| 336 return TRUE; | 336 return TRUE; |
| 337 } | 337 } |
| 338 virtual FX_BOOL Flush() override | 338 virtual FX_BOOL Flush() override |
| 339 { | 339 { |
| 340 return TRUE; | 340 return TRUE; |
| 341 } | 341 } |
| 342 virtual FX_BOOL IsConsecutive() const override | 342 virtual FX_BOOL IsConsecutive() const override |
| 343 { | 343 { |
| 344 return m_dwFlags & FX_MEMSTREAM_Consecutive; | 344 return m_dwFlags & FX_MEMSTREAM_Consecutive; |
| 345 } | 345 } |
| 346 virtual void EstimateSize(size_t nInitSize, s
ize_t nGrowSize) override | 346 virtual void EstimateSize(size_t nInitSize, s
ize_t nGrowSize) override |
| 347 { | 347 { |
| 348 if (m_dwFlags & FX_MEMSTREAM_Consecutive) { | 348 if (m_dwFlags & FX_MEMSTREAM_Consecutive) { |
| 349 if (m_Blocks.GetSize() < 1) { | 349 if (m_Blocks.GetSize() < 1) { |
| 350 FX_LPBYTE pBlock = FX_Alloc(uint8_t, FX_MAX(nInitSize, 4096)); | 350 uint8_t* pBlock = FX_Alloc(uint8_t, FX_MAX(nInitSize, 4096)); |
| 351 m_Blocks.Add(pBlock); | 351 m_Blocks.Add(pBlock); |
| 352 } | 352 } |
| 353 m_nGrowSize = FX_MAX(nGrowSize, 4096); | 353 m_nGrowSize = FX_MAX(nGrowSize, 4096); |
| 354 } else if (m_Blocks.GetSize() < 1) { | 354 } else if (m_Blocks.GetSize() < 1) { |
| 355 m_nGrowSize = FX_MAX(nGrowSize, 4096); | 355 m_nGrowSize = FX_MAX(nGrowSize, 4096); |
| 356 } | 356 } |
| 357 } | 357 } |
| 358 virtual FX_LPBYTE» » » GetBuffer() const override | 358 virtual uint8_t*» » » GetBuffer() const override |
| 359 { | 359 { |
| 360 return m_Blocks.GetSize() ? (FX_LPBYTE)m_Blocks[0] : NULL; | 360 return m_Blocks.GetSize() ? (uint8_t*)m_Blocks[0] : NULL; |
| 361 } | 361 } |
| 362 virtual void» » » » AttachBuffer(FX_LPBYTE pBuffer,
size_t nSize, FX_BOOL bTakeOver = FALSE) override | 362 virtual void» » » » AttachBuffer(uint8_t* pBuffer, s
ize_t nSize, FX_BOOL bTakeOver = FALSE) override |
| 363 { | 363 { |
| 364 if (!(m_dwFlags & FX_MEMSTREAM_Consecutive)) { | 364 if (!(m_dwFlags & FX_MEMSTREAM_Consecutive)) { |
| 365 return; | 365 return; |
| 366 } | 366 } |
| 367 m_Blocks.RemoveAll(); | 367 m_Blocks.RemoveAll(); |
| 368 m_Blocks.Add(pBuffer); | 368 m_Blocks.Add(pBuffer); |
| 369 m_nTotalSize = m_nCurSize = nSize; | 369 m_nTotalSize = m_nCurSize = nSize; |
| 370 m_nCurPos = 0; | 370 m_nCurPos = 0; |
| 371 m_dwFlags = FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOve
r : 0); | 371 m_dwFlags = FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOve
r : 0); |
| 372 ClearRange(); | 372 ClearRange(); |
| (...skipping 24 matching lines...) Expand all Loading... |
| 397 if (m_nCurSize < size) { | 397 if (m_nCurSize < size) { |
| 398 m_nCurSize = size; | 398 m_nCurSize = size; |
| 399 } | 399 } |
| 400 if (size <= m_nTotalSize) { | 400 if (size <= m_nTotalSize) { |
| 401 return TRUE; | 401 return TRUE; |
| 402 } | 402 } |
| 403 int32_t iCount = m_Blocks.GetSize(); | 403 int32_t iCount = m_Blocks.GetSize(); |
| 404 size = (size - m_nTotalSize + m_nGrowSize - 1) / m_nGrowSize; | 404 size = (size - m_nTotalSize + m_nGrowSize - 1) / m_nGrowSize; |
| 405 m_Blocks.SetSize(m_Blocks.GetSize() + (int32_t)size); | 405 m_Blocks.SetSize(m_Blocks.GetSize() + (int32_t)size); |
| 406 while (size --) { | 406 while (size --) { |
| 407 FX_LPBYTE pBlock = FX_Alloc(uint8_t, m_nGrowSize); | 407 uint8_t* pBlock = FX_Alloc(uint8_t, m_nGrowSize); |
| 408 m_Blocks.SetAt(iCount ++, pBlock); | 408 m_Blocks.SetAt(iCount ++, pBlock); |
| 409 m_nTotalSize += m_nGrowSize; | 409 m_nTotalSize += m_nGrowSize; |
| 410 } | 410 } |
| 411 return TRUE; | 411 return TRUE; |
| 412 } | 412 } |
| 413 }; | 413 }; |
| 414 #ifdef __cplusplus | 414 #ifdef __cplusplus |
| 415 extern "C" { | 415 extern "C" { |
| 416 #endif | 416 #endif |
| 417 #define MT_N 848 | 417 #define MT_N 848 |
| 418 #define MT_M 456 | 418 #define MT_M 456 |
| 419 #define MT_Matrix_A 0x9908b0df | 419 #define MT_Matrix_A 0x9908b0df |
| 420 #define MT_Upper_Mask 0x80000000 | 420 #define MT_Upper_Mask 0x80000000 |
| 421 #define MT_Lower_Mask 0x7fffffff | 421 #define MT_Lower_Mask 0x7fffffff |
| 422 typedef struct _FX_MTRANDOMCONTEXT { | 422 typedef struct _FX_MTRANDOMCONTEXT { |
| 423 _FX_MTRANDOMCONTEXT() | 423 _FX_MTRANDOMCONTEXT() |
| 424 { | 424 { |
| 425 mti = MT_N + 1; | 425 mti = MT_N + 1; |
| 426 bHaveSeed = FALSE; | 426 bHaveSeed = FALSE; |
| 427 } | 427 } |
| 428 FX_DWORD mti; | 428 FX_DWORD mti; |
| 429 FX_BOOL bHaveSeed; | 429 FX_BOOL bHaveSeed; |
| 430 FX_DWORD mt[MT_N]; | 430 FX_DWORD mt[MT_N]; |
| 431 } FX_MTRANDOMCONTEXT, * FX_LPMTRANDOMCONTEXT; | 431 } FX_MTRANDOMCONTEXT, * FX_LPMTRANDOMCONTEXT; |
| 432 typedef FX_MTRANDOMCONTEXT const * FX_LPCMTRANDOMCONTEXT; | 432 typedef FX_MTRANDOMCONTEXT const * FX_LPCMTRANDOMCONTEXT; |
| 433 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ | 433 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ |
| 434 FX_BOOL FX_GenerateCryptoRandom(FX_LPDWORD pBuffer, int32_t iCount); | 434 FX_BOOL FX_GenerateCryptoRandom(FX_DWORD* pBuffer, int32_t iCount); |
| 435 #endif | 435 #endif |
| 436 #ifdef __cplusplus | 436 #ifdef __cplusplus |
| 437 } | 437 } |
| 438 #endif | 438 #endif |
| 439 | 439 |
| 440 #endif // CORE_SRC_FXCRT_EXTENSION_H_ | 440 #endif // CORE_SRC_FXCRT_EXTENSION_H_ |
| OLD | NEW |