src/hotspot/cpu/aarch64/gc/shenandoah/shenandoahBarrierSetAssembler_aarch64.hpp

@@ -1,7 +1,8 @@
  /*
   * Copyright (c) 2018, 2021, Red Hat, Inc. All rights reserved.
+  * Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
   * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   *
   * This code is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License version 2 only, as
   * published by the Free Software Foundation.

@@ -53,14 +54,20 @@
                                      Register thread,
                                      Register tmp,
                                      bool tosca_live,
                                      bool expand_call);
  
+   void store_check(MacroAssembler* masm, Register obj);
+ 
    void resolve_forward_pointer(MacroAssembler* masm, Register dst, Register tmp = noreg);
    void resolve_forward_pointer_not_null(MacroAssembler* masm, Register dst, Register tmp = noreg);
    void load_reference_barrier(MacroAssembler* masm, Register dst, Address load_addr, DecoratorSet decorators);
  
+   void gen_write_ref_array_post_barrier(MacroAssembler* masm, DecoratorSet decorators,
+                                         Register start, Register count,
+                                         Register scratch, RegSet saved_regs);
+ 
  public:
    virtual NMethodPatchingType nmethod_patching_type() { return NMethodPatchingType::conc_data_patch; }
  
  #ifdef COMPILER1
    void gen_pre_barrier_stub(LIR_Assembler* ce, ShenandoahPreBarrierStub* stub);

@@ -69,10 +76,12 @@
    void generate_c1_load_reference_barrier_runtime_stub(StubAssembler* sasm, DecoratorSet decorators);
  #endif
  
    virtual void arraycopy_prologue(MacroAssembler* masm, DecoratorSet decorators, bool is_oop,
                                    Register src, Register dst, Register count, RegSet saved_regs);
+   virtual void arraycopy_epilogue(MacroAssembler* masm, DecoratorSet decorators, bool is_oop,
+                                   Register start, Register count, Register tmp, RegSet saved_regs);
    virtual void load_at(MacroAssembler* masm, DecoratorSet decorators, BasicType type,
                         Register dst, Address src, Register tmp1, Register tmp2);
    virtual void store_at(MacroAssembler* masm, DecoratorSet decorators, BasicType type,
                          Address dst, Register val, Register tmp1, Register tmp2, Register tmp3);
    virtual void try_resolve_jobject_in_native(MacroAssembler* masm, Register jni_env,