text: /*
   * Copyright (c) 2021 Huawei Device Co., Ltd.
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *     http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  #include "ecmascript/mem/semi_space_marker.h"
  
  #include "ecmascript/free_object.h"
  #include "ecmascript/js_hclass-inl.h"
  #include "ecmascript/mem/region.h"
  #include "ecmascript/mem/semi_space_collector-inl.h"
  #include "ecmascript/mem/semi_space_worker.h"
  #include "ecmascript/mem/tlab_allocator-inl.h"
  
  namespace panda::ecmascript {
  constexpr int HEAD_SIZE = TaggedObject::TaggedObjectSize();
  
  SemiSpaceMarker::SemiSpaceMarker(SemiSpaceCollector *semiSpaceCollector) : collector_(semiSpaceCollector) {}
  
  SlotStatus SemiSpaceMarker::MarkObject(uint32_t threadId, TaggedObject *object, ObjectSlot slot)
  {
      Region *objectRegion = Region::ObjectAddressToRange(object);
      if (!objectRegion->InYoungGeneration()) {
          return SlotStatus::CLEAR_SLOT;
      }
      return MarkYoungGenerationObject(threadId, object, slot);
  }
  
  SlotStatus SemiSpaceMarker::MarkYoungGenerationObject(uint32_t threadId, TaggedObject *object, ObjectSlot slot)
  {
      MarkWord markWord(object);
      if (markWord.IsForwardingAddress()) {
          TaggedObject *dst = markWord.ToForwardingAddress();
          slot.Update(dst);
          Region *valueRegion = Region::ObjectAddressToRange(dst);
          return valueRegion->InYoungGeneration() ? SlotStatus::KEEP_SLOT : SlotStatus::CLEAR_SLOT;
      }
      return EvacuateObject(threadId, object, markWord, slot);
  }
  
  SlotStatus SemiSpaceMarker::EvacuateObject(uint32_t threadId, TaggedObject *object, const MarkWord &markWord,
                                             ObjectSlot slot)
  {
      auto klass = markWord.GetJSHClass();
      JSType jsType = klass->GetObjectType();
      auto size = klass->SizeFromJSHClass(jsType, object);
      bool isPromoted = ShouldBePromoted(object);
      uintptr_t forwardAddress;
      if (isPromoted) {
          forwardAddress = collector_->workList_->GetTlabAllocator(threadId)->Allocate(size, SpaceAlloc::OLD_SPACE);
          if (UNLIKELY(forwardAddress == 0)) {
              forwardAddress = collector_->workList_->GetTlabAllocator(threadId)->Allocate(size, SpaceAlloc::YOUNG_SPACE);
              if (UNLIKELY(forwardAddress == 0)) {
                  LOG_ECMA_MEM(FATAL) << "alloc failed";
                  UNREACHABLE();
              }
              isPromoted = false;
          }
      } else {
          forwardAddress = collector_->workList_->GetTlabAllocator(threadId)->Allocate(size, SpaceAlloc::YOUNG_SPACE);
          if (UNLIKELY(forwardAddress == 0)) {
              forwardAddress = collector_->workList_->GetTlabAllocator(threadId)->Allocate(size, SpaceAlloc::OLD_SPACE);
              if (UNLIKELY(forwardAddress == 0)) {
                  LOG_ECMA_MEM(FATAL) << "alloc failed";
                  UNREACHABLE();
              }
              isPromoted = true;
          }
      }
  
      // NOLINTNEXTLINE(clang-analyzer-core.CallAndMessage, hicpp-signed-bitwise)
      bool result = Barriers::AtomicSetDynPrimitive(object, 0, markWord.GetValue(),
                                                    MarkWord::FromForwardingAddress(forwardAddress));
      if (result) {
          CopyObjectWithoutHeader(object, forwardAddress, size);
          *reinterpret_cast<MarkWordType *>(forwardAddress) = markWord.GetValue();
          collector_->workList_->AddAliveSize(threadId, size);
          if (isPromoted) {
              collector_->workList_->AddPromoteSize(threadId, size);
          }
          collector_->heap_->OnMoveEvent(reinterpret_cast<intptr_t>(object), forwardAddress);
          if (klass->HasReferenceField(jsType)) {
              collector_->workList_->Push(threadId, reinterpret_cast<TaggedObject *>(forwardAddress));
          }
          slot.Update(reinterpret_cast<TaggedObject *>(forwardAddress));
          return isPromoted ? SlotStatus::CLEAR_SLOT : SlotStatus::KEEP_SLOT;
      }
      FreeObject::FillFreeObject(collector_->heap_->GetEcmaVM(), forwardAddress,
                                 AlignUp(size, static_cast<size_t>(MemAlignment::MEM_ALIGN_OBJECT)));
      auto dst = MarkWord(object).ToForwardingAddress();
      slot.Update(dst);
      return Region::ObjectAddressToRange(dst)->InYoungGeneration() ? SlotStatus::KEEP_SLOT : SlotStatus::CLEAR_SLOT;
  }
  
  void SemiSpaceMarker::CopyObjectWithoutHeader(TaggedObject *object, uintptr_t address, size_t size)
  {
      if (memcpy_s(ToVoidPtr(address + HEAD_SIZE), size - HEAD_SIZE, ToVoidPtr(ToUintPtr(object) + HEAD_SIZE),
          size - HEAD_SIZE) != EOK) {
          LOG_ECMA(FATAL) << "memcpy_s failed";
          UNREACHABLE();
      }
  }
  
  bool SemiSpaceMarker::ShouldBePromoted(TaggedObject *object)
  {
      Region *region = Region::ObjectAddressToRange(object);
      return (region->BelowAgeMark() || (region->HasAgeMark() && collector_->BlowAgeMark(ToUintPtr(object))));
  }
  }  // namespace panda::ecmascript
  