1 files changed, 446 insertions, 0 deletions
diff --git a/xfa/src/fgas/crt/fgas_memory.cpp b/xfa/src/fgas/crt/fgas_memory.cpp
new file mode 100644
index 0000000000..c88c7e3aac
--- /dev/null
+++ b/xfa/src/fgas/crt/fgas_memory.cpp
@@ -0,0 +1,446 @@
+// Copyright 2014 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
+
+#include "xfa/src/fgas/crt/fgas_memory.h"
+
+#include <algorithm>
+
+namespace {
+
+class CFX_DefStore : public IFX_MEMAllocator, public CFX_Target {
+ public:
+  CFX_DefStore() {}
+  ~CFX_DefStore() {}
+  virtual void Release() { delete this; }
+  virtual void* Alloc(size_t size) { return FX_Alloc(uint8_t, size); }
+  virtual void Free(void* pBlock) { FX_Free(pBlock); }
+  virtual size_t GetBlockSize() const { return 0; }
+  virtual size_t GetDefChunkSize() const { return 0; }
+  virtual size_t SetDefChunkSize(size_t size) { return 0; }
+  virtual size_t GetCurrentDataSize() const { return 0; }
+};
+
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(push, 1)
+#endif
+struct FX_STATICSTORECHUNK {
+  FX_STATICSTORECHUNK* pNextChunk;
+  size_t iChunkSize;
+  size_t iFreeSize;
+};
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(pop)
+#endif
+
+class CFX_StaticStore : public IFX_MEMAllocator, public CFX_Target {
+ public:
+  CFX_StaticStore(size_t iDefChunkSize = 4096);
+  ~CFX_StaticStore();
+  virtual void Release() { delete this; }
+  virtual void* Alloc(size_t size);
+  virtual void Free(void* pBlock) {}
+  virtual size_t GetBlockSize() const { return 0; }
+  virtual size_t GetDefChunkSize() const { return m_iDefChunkSize; }
+  virtual size_t SetDefChunkSize(size_t size);
+  virtual size_t GetCurrentDataSize() const { return m_iAllocatedSize; }
+
+ protected:
+  size_t m_iAllocatedSize;
+  size_t m_iDefChunkSize;
+  FX_STATICSTORECHUNK* m_pChunk;
+  FX_STATICSTORECHUNK* m_pLastChunk;
+  FX_STATICSTORECHUNK* AllocChunk(size_t size);
+  FX_STATICSTORECHUNK* FindChunk(size_t size);
+};
+
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(push, 1)
+#endif
+struct FX_FIXEDSTORECHUNK {
+  uint8_t* FirstFlag() { return reinterpret_cast<uint8_t*>(this + 1); }
+  uint8_t* FirstBlock() { return FirstFlag() + iChunkSize; }
+
+  FX_FIXEDSTORECHUNK* pNextChunk;
+  size_t iChunkSize;
+  size_t iFreeNum;
+};
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(pop)
+#endif
+
+class CFX_FixedStore : public IFX_MEMAllocator, public CFX_Target {
+ public:
+  CFX_FixedStore(size_t iBlockSize, size_t iBlockNumsInChunk);
+  virtual ~CFX_FixedStore();
+  virtual void Release() { delete this; }
+  virtual void* Alloc(size_t size);
+  virtual void Free(void* pBlock);
+  virtual size_t GetBlockSize() const { return m_iBlockSize; }
+  virtual size_t GetDefChunkSize() const { return m_iDefChunkSize; }
+  virtual size_t SetDefChunkSize(size_t iChunkSize);
+  virtual size_t GetCurrentDataSize() const { return 0; }
+
+ protected:
+  FX_FIXEDSTORECHUNK* AllocChunk();
+
+  size_t m_iBlockSize;
+  size_t m_iDefChunkSize;
+  FX_FIXEDSTORECHUNK* m_pChunk;
+};
+
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(push, 1)
+#endif
+struct FX_DYNAMICSTOREBLOCK {
+  uint8_t* Data() { return reinterpret_cast<uint8_t*>(this + 1); }
+  FX_DYNAMICSTOREBLOCK* NextBlock() {
+    return reinterpret_cast<FX_DYNAMICSTOREBLOCK*>(Data() + iBlockSize);
+  }
+  size_t iBlockSize;
+  FX_BOOL bUsed;
+};
+
+struct FX_DYNAMICSTORECHUNK {
+  FX_DYNAMICSTOREBLOCK* FirstBlock() {
+    return reinterpret_cast<FX_DYNAMICSTOREBLOCK*>(this + 1);
+  }
+  FX_DYNAMICSTORECHUNK* pNextChunk;
+  size_t iChunkSize;
+  size_t iFreeSize;
+};
+#if _FX_OS_ != _FX_ANDROID_
+#pragma pack(pop)
+#endif
+
+class CFX_DynamicStore : public IFX_MEMAllocator, public CFX_Target {
+ public:
+  CFX_DynamicStore(size_t iDefChunkSize = 4096);
+  virtual ~CFX_DynamicStore();
+  virtual void Release() { delete this; }
+  virtual void* Alloc(size_t size);
+  virtual void Free(void* pBlock);
+  virtual size_t GetBlockSize() const { return 0; }
+  virtual size_t GetDefChunkSize() const { return m_iDefChunkSize; }
+  virtual size_t SetDefChunkSize(size_t size);
+  virtual size_t GetCurrentDataSize() const { return 0; }
+
+ protected:
+  FX_DYNAMICSTORECHUNK* AllocChunk(size_t size);
+
+  size_t m_iDefChunkSize;
+  FX_DYNAMICSTORECHUNK* m_pChunk;
+};
+
+}  // namespace
+
+#define FX_4BYTEALIGN(size) (((size) + 3) / 4 * 4)
+
+IFX_MEMAllocator* FX_CreateAllocator(FX_ALLOCTYPE eType,
+                                     size_t chunkSize,
+                                     size_t blockSize) {
+  switch (eType) {
+    case FX_ALLOCTYPE_Dynamic:
+      return new CFX_DynamicStore(chunkSize);
+    case FX_ALLOCTYPE_Default:
+      return new CFX_DefStore();
+    case FX_ALLOCTYPE_Static:
+      return new CFX_StaticStore(chunkSize);
+    case FX_ALLOCTYPE_Fixed:
+      return new CFX_FixedStore(blockSize, chunkSize);
+    default:
+      return NULL;
+  }
+}
+CFX_StaticStore::CFX_StaticStore(size_t iDefChunkSize)
+    : m_iAllocatedSize(0),
+      m_iDefChunkSize(iDefChunkSize),
+      m_pChunk(NULL),
+      m_pLastChunk(NULL) {
+  FXSYS_assert(m_iDefChunkSize != 0);
+}
+CFX_StaticStore::~CFX_StaticStore() {
+  FX_STATICSTORECHUNK* pChunk = m_pChunk;
+  while (pChunk) {
+    FX_STATICSTORECHUNK* pNext = pChunk->pNextChunk;
+    FX_Free(pChunk);
+    pChunk = pNext;
+  }
+}
+FX_STATICSTORECHUNK* CFX_StaticStore::AllocChunk(size_t size) {
+  FXSYS_assert(size != 0);
+  FX_STATICSTORECHUNK* pChunk = (FX_STATICSTORECHUNK*)FX_Alloc(
+      uint8_t, sizeof(FX_STATICSTORECHUNK) + size);
+  pChunk->iChunkSize = size;
+  pChunk->iFreeSize = size;
+  pChunk->pNextChunk = NULL;
+  if (m_pLastChunk == NULL) {
+    m_pChunk = pChunk;
+  } else {
+    m_pLastChunk->pNextChunk = pChunk;
+  }
+  m_pLastChunk = pChunk;
+  return pChunk;
+}
+FX_STATICSTORECHUNK* CFX_StaticStore::FindChunk(size_t size) {
+  FXSYS_assert(size != 0);
+  if (m_pLastChunk == NULL || m_pLastChunk->iFreeSize < size) {
+    return AllocChunk(std::max(m_iDefChunkSize, size));
+  }
+  return m_pLastChunk;
+}
+void* CFX_StaticStore::Alloc(size_t size) {
+  size = FX_4BYTEALIGN(size);
+  FXSYS_assert(size != 0);
+  FX_STATICSTORECHUNK* pChunk = FindChunk(size);
+  FXSYS_assert(pChunk->iFreeSize >= size);
+  uint8_t* p = (uint8_t*)pChunk;
+  p += sizeof(FX_STATICSTORECHUNK) + pChunk->iChunkSize - pChunk->iFreeSize;
+  pChunk->iFreeSize -= size;
+  m_iAllocatedSize += size;
+  return p;
+}
+size_t CFX_StaticStore::SetDefChunkSize(size_t size) {
+  FXSYS_assert(size != 0);
+  size_t v = m_iDefChunkSize;
+  m_iDefChunkSize = size;
+  return v;
+}
+CFX_FixedStore::CFX_FixedStore(size_t iBlockSize, size_t iBlockNumsInChunk)
+    : m_iBlockSize(FX_4BYTEALIGN(iBlockSize)),
+      m_iDefChunkSize(FX_4BYTEALIGN(iBlockNumsInChunk)),
+      m_pChunk(NULL) {
+  FXSYS_assert(m_iBlockSize != 0 && m_iDefChunkSize != 0);
+}
+CFX_FixedStore::~CFX_FixedStore() {
+  FX_FIXEDSTORECHUNK* pChunk = m_pChunk;
+  while (pChunk) {
+    FX_FIXEDSTORECHUNK* pNext = pChunk->pNextChunk;
+    FX_Free(pChunk);
+    pChunk = pNext;
+  }
+}
+FX_FIXEDSTORECHUNK* CFX_FixedStore::AllocChunk() {
+  int32_t iTotalSize = sizeof(FX_FIXEDSTORECHUNK) + m_iDefChunkSize +
+                       m_iBlockSize * m_iDefChunkSize;
+  FX_FIXEDSTORECHUNK* pChunk =
+      (FX_FIXEDSTORECHUNK*)FX_Alloc(uint8_t, iTotalSize);
+  if (pChunk == NULL) {
+    return NULL;
+  }
+  FXSYS_memset(pChunk->FirstFlag(), 0, m_iDefChunkSize);
+  pChunk->pNextChunk = m_pChunk;
+  pChunk->iChunkSize = m_iDefChunkSize;
+  pChunk->iFreeNum = m_iDefChunkSize;
+  m_pChunk = pChunk;
+  return pChunk;
+}
+void* CFX_FixedStore::Alloc(size_t size) {
+  if (size > m_iBlockSize) {
+    return NULL;
+  }
+  FX_FIXEDSTORECHUNK* pChunk = m_pChunk;
+  while (pChunk != NULL) {
+    if (pChunk->iFreeNum > 0) {
+      break;
+    }
+    pChunk = pChunk->pNextChunk;
+  }
+  if (pChunk == NULL) {
+    pChunk = AllocChunk();
+  }
+  FXSYS_assert(pChunk != NULL);
+  uint8_t* pFlags = pChunk->FirstFlag();
+  size_t i = 0;
+  for (; i < pChunk->iChunkSize; i++)
+    if (pFlags[i] == 0) {
+      break;
+    }
+  FXSYS_assert(i < pChunk->iChunkSize);
+  pFlags[i] = 1;
+  pChunk->iFreeNum--;
+  return pChunk->FirstBlock() + i * m_iBlockSize;
+}
+void CFX_FixedStore::Free(void* pBlock) {
+  FXSYS_assert(pBlock != NULL);
+  FX_FIXEDSTORECHUNK* pPrior = NULL;
+  FX_FIXEDSTORECHUNK* pChunk = m_pChunk;
+  uint8_t* pStart = NULL;
+  uint8_t* pEnd;
+  while (pChunk != NULL) {
+    pStart = pChunk->FirstBlock();
+    if (pBlock >= pStart) {
+      pEnd = pStart + m_iBlockSize * pChunk->iChunkSize;
+      if (pBlock < pEnd) {
+        break;
+      }
+    }
+    pPrior = pChunk, pChunk = pChunk->pNextChunk;
+  }
+  FXSYS_assert(pChunk != NULL);
+  size_t iPos = ((uint8_t*)pBlock - pStart) / m_iBlockSize;
+  FXSYS_assert(iPos < pChunk->iChunkSize);
+  uint8_t* pFlags = pChunk->FirstFlag();
+  if (pFlags[iPos] == 0) {
+    return;
+  }
+  pFlags[iPos] = 0;
+  pChunk->iFreeNum++;
+  if (pChunk->iFreeNum == pChunk->iChunkSize) {
+    if (pPrior == NULL) {
+      m_pChunk = pChunk->pNextChunk;
+    } else {
+      pPrior->pNextChunk = pChunk->pNextChunk;
+    }
+    FX_Free(pChunk);
+  }
+}
+size_t CFX_FixedStore::SetDefChunkSize(size_t iChunkSize) {
+  FXSYS_assert(iChunkSize != 0);
+  size_t v = m_iDefChunkSize;
+  m_iDefChunkSize = FX_4BYTEALIGN(iChunkSize);
+  return v;
+}
+CFX_DynamicStore::CFX_DynamicStore(size_t iDefChunkSize)
+    : m_iDefChunkSize(iDefChunkSize), m_pChunk(NULL) {
+  FXSYS_assert(m_iDefChunkSize != 0);
+}
+CFX_DynamicStore::~CFX_DynamicStore() {
+  FX_DYNAMICSTORECHUNK* pChunk = m_pChunk;
+  while (pChunk) {
+    FX_DYNAMICSTORECHUNK* pNext = pChunk->pNextChunk;
+    FX_Free(pChunk);
+    pChunk = pNext;
+  }
+}
+FX_DYNAMICSTORECHUNK* CFX_DynamicStore::AllocChunk(size_t size) {
+  FXSYS_assert(size != 0);
+  FX_DYNAMICSTORECHUNK* pChunk = (FX_DYNAMICSTORECHUNK*)FX_Alloc(
+      uint8_t,
+      sizeof(FX_DYNAMICSTORECHUNK) + sizeof(FX_DYNAMICSTOREBLOCK) * 2 + size);
+  if (pChunk == NULL) {
+    return NULL;
+  }
+  pChunk->iChunkSize = size;
+  pChunk->iFreeSize = size;
+  FX_DYNAMICSTOREBLOCK* pBlock = pChunk->FirstBlock();
+  pBlock->iBlockSize = size;
+  pBlock->bUsed = FALSE;
+  pBlock = pBlock->NextBlock();
+  pBlock->iBlockSize = 0;
+  pBlock->bUsed = TRUE;
+  if (m_pChunk != NULL && size >= m_iDefChunkSize) {
+    FX_DYNAMICSTORECHUNK* pLast = m_pChunk;
+    while (pLast->pNextChunk != NULL) {
+      pLast = pLast->pNextChunk;
+    }
+    pLast->pNextChunk = pChunk;
+    pChunk->pNextChunk = NULL;
+  } else {
+    pChunk->pNextChunk = m_pChunk;
+    m_pChunk = pChunk;
+  }
+  return pChunk;
+}
+void* CFX_DynamicStore::Alloc(size_t size) {
+  size = FX_4BYTEALIGN(size);
+  FXSYS_assert(size != 0);
+  FX_DYNAMICSTORECHUNK* pChunk = m_pChunk;
+  FX_DYNAMICSTOREBLOCK* pBlock = NULL;
+  while (pChunk != NULL) {
+    if (pChunk->iFreeSize >= size) {
+      pBlock = pChunk->FirstBlock();
+      FX_BOOL bFind = FALSE;
+      while (pBlock->iBlockSize != 0) {
+        if (!pBlock->bUsed && pBlock->iBlockSize >= size) {
+          bFind = TRUE;
+          break;
+        }
+        pBlock = pBlock->NextBlock();
+      }
+      if (bFind) {
+        break;
+      }
+    }
+    pChunk = pChunk->pNextChunk;
+  }
+  if (pChunk == NULL) {
+    pChunk = AllocChunk(std::max(m_iDefChunkSize, size));
+    pBlock = pChunk->FirstBlock();
+  }
+  FXSYS_assert(pChunk != NULL && pBlock != NULL);
+  size_t m = size + sizeof(FX_DYNAMICSTOREBLOCK);
+  pBlock->bUsed = TRUE;
+  if (pBlock->iBlockSize > m) {
+    size_t n = pBlock->iBlockSize;
+    pBlock->iBlockSize = size;
+    FX_DYNAMICSTOREBLOCK* pNextBlock = pBlock->NextBlock();
+    pNextBlock->bUsed = FALSE;
+    pNextBlock->iBlockSize = n - size - sizeof(FX_DYNAMICSTOREBLOCK);
+    pChunk->iFreeSize -= size + sizeof(FX_DYNAMICSTOREBLOCK);
+  } else {
+    pChunk->iFreeSize -= pBlock->iBlockSize;
+  }
+  return pBlock->Data();
+}
+void CFX_DynamicStore::Free(void* pBlock) {
+  FXSYS_assert(pBlock != NULL);
+  FX_DYNAMICSTORECHUNK* pPriorChunk = NULL;
+  FX_DYNAMICSTORECHUNK* pChunk = m_pChunk;
+  while (pChunk != NULL) {
+    if (pBlock > pChunk &&
+        pBlock <= ((uint8_t*)pChunk + sizeof(FX_DYNAMICSTORECHUNK) +
+                   pChunk->iChunkSize)) {
+      break;
+    }
+    pPriorChunk = pChunk, pChunk = pChunk->pNextChunk;
+  }
+  FXSYS_assert(pChunk != NULL);
+  FX_DYNAMICSTOREBLOCK* pPriorBlock = NULL;
+  FX_DYNAMICSTOREBLOCK* pFindBlock = pChunk->FirstBlock();
+  while (pFindBlock->iBlockSize != 0) {
+    if (pBlock == (void*)pFindBlock->Data()) {
+      break;
+    }
+    pPriorBlock = pFindBlock;
+    pFindBlock = pFindBlock->NextBlock();
+  }
+  FXSYS_assert(pFindBlock->iBlockSize != 0 && pFindBlock->bUsed &&
+               pBlock == (void*)pFindBlock->Data());
+  pFindBlock->bUsed = FALSE;
+  pChunk->iFreeSize += pFindBlock->iBlockSize;
+  if (pPriorBlock == NULL) {
+    pPriorBlock = pChunk->FirstBlock();
+  } else if (pPriorBlock->bUsed) {
+    pPriorBlock = pFindBlock;
+  }
+  pFindBlock = pPriorBlock;
+  size_t sizeFree = 0;
+  size_t sizeBlock = 0;
+  while (pFindBlock->iBlockSize != 0 && !pFindBlock->bUsed) {
+    if (pFindBlock != pPriorBlock) {
+      sizeFree += sizeof(FX_DYNAMICSTOREBLOCK);
+      sizeBlock += sizeof(FX_DYNAMICSTOREBLOCK);
+    }
+    sizeBlock += pFindBlock->iBlockSize;
+    pFindBlock = pFindBlock->NextBlock();
+  }
+  pPriorBlock->iBlockSize = sizeBlock;
+  pChunk->iFreeSize += sizeFree;
+  if (pChunk->iFreeSize == pChunk->iChunkSize) {
+    if (pPriorChunk == NULL) {
+      m_pChunk = pChunk->pNextChunk;
+    } else {
+      pPriorChunk->pNextChunk = pChunk->pNextChunk;
+    }
+    FX_Free(pChunk);
+  }
+}
+size_t CFX_DynamicStore::SetDefChunkSize(size_t size) {
+  FXSYS_assert(size != 0);
+  size_t v = m_iDefChunkSize;
+  m_iDefChunkSize = size;
+  return v;
+}