1 /*
  2  * Copyright (c) 2022, 2023, Oracle and/or its affiliates. All rights reserved.
  3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  4  *
  5  * This code is free software; you can redistribute it and/or modify it
  6  * under the terms of the GNU General Public License version 2 only, as
  7  * published by the Free Software Foundation.
  8  *
  9  * This code is distributed in the hope that it will be useful, but WITHOUT
 10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 12  * version 2 for more details (a copy is included in the LICENSE file that
 13  * accompanied this code).
 14  *
 15  * You should have received a copy of the GNU General Public License version
 16  * 2 along with this work; if not, write to the Free Software Foundation,
 17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 18  *
 19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 20  * or visit www.oracle.com if you need additional information or have any
 21  * questions.
 22  *
 23  */
 24 
 25 #ifndef CPU_AARCH64_CONTINUATIONHELPER_AARCH64_INLINE_HPP
 26 #define CPU_AARCH64_CONTINUATIONHELPER_AARCH64_INLINE_HPP
 27 
 28 #include "runtime/continuationHelper.hpp"
 29 
 30 #include "runtime/continuationEntry.inline.hpp"
 31 #include "runtime/frame.inline.hpp"
 32 #include "runtime/registerMap.hpp"
 33 #include "utilities/macros.hpp"
 34 
 35 template<typename FKind>
 36 static inline intptr_t** link_address(const frame& f) {
 37   assert(FKind::is_instance(f), "");
 38   return FKind::interpreted
 39             ? (intptr_t**)(f.fp() + frame::link_offset)
 40             : (intptr_t**)(f.unextended_sp() + f.cb()->frame_size() - frame::sender_sp_offset);
 41 }
 42 
 43 static inline void patch_return_pc_with_preempt_stub(frame& f) {
 44   // Unlike x86 we don't know where in the callee frame the return pc is
 45   // saved so we can't patch the return from the VM call back to Java. If
 46   // the target is coming from compiled code we will patch the return from
 47   // the safepoint handler blob back to the compiled method instead. If
 48   // it's coming from the interpreter, the target will check for preemption
 49   // once it returns to the interpreter and will manually jump to the
 50   // preempt stub.
 51   if (!f.is_interpreted_frame()) {
 52     assert(f.is_runtime_frame(), "invariant");
 53     intptr_t* caller_sp = f.sp() + f.cb()->frame_size();
 54     caller_sp[-1] = (intptr_t)StubRoutines::cont_preempt_stub();
 55   } else {
 56     JavaThread *thread = JavaThread::current();
 57     thread->set_preempt_alternate_return(StubRoutines::cont_preempt_stub());
 58     thread->set_preempt_alternate_return_sp((address)f.sp());
 59   }
 60 }
 61 
 62 inline int ContinuationHelper::frame_align_words(int size) {
 63 #ifdef _LP64
 64   return size & 1;
 65 #else
 66   return 0;
 67 #endif
 68 }
 69 
 70 inline intptr_t* ContinuationHelper::frame_align_pointer(intptr_t* sp) {
 71 #ifdef _LP64
 72   sp = align_down(sp, frame::frame_alignment);
 73 #endif
 74   return sp;
 75 }
 76 
 77 template<typename FKind>
 78 inline void ContinuationHelper::update_register_map(const frame& f, RegisterMap* map) {
 79   frame::update_map_with_saved_link(map, link_address<FKind>(f));
 80 }
 81 
 82 inline void ContinuationHelper::update_register_map_with_callee(const frame& f, RegisterMap* map) {
 83   frame::update_map_with_saved_link(map, ContinuationHelper::Frame::callee_link_address(f));
 84 }
 85 
 86 inline void ContinuationHelper::push_pd(const frame& f) {
 87   *(intptr_t**)(f.sp() - frame::sender_sp_offset) = f.fp();
 88 }
 89 
 90 #define CPU_OVERRIDES_RETURN_ADDRESS_ACCESSORS
 91 
 92 inline address ContinuationHelper::return_address_at(intptr_t* sp) {
 93   return pauth_strip_verifiable(*(address*)sp);
 94 }
 95 
 96 inline void ContinuationHelper::patch_return_address_at(intptr_t* sp,
 97                                                         address pc) {
 98   *(address*)sp = pauth_sign_return_address(pc);
 99 }
100 
101 inline void ContinuationHelper::set_anchor_to_entry_pd(JavaFrameAnchor* anchor, ContinuationEntry* entry) {
102   anchor->set_last_Java_fp(entry->entry_fp());
103 }
104 
105 inline void ContinuationHelper::set_anchor_pd(JavaFrameAnchor* anchor, intptr_t* sp) {
106   intptr_t* fp = *(intptr_t**)(sp - frame::sender_sp_offset);
107   anchor->set_last_Java_fp(fp);
108 }
109 
110 #ifdef ASSERT
111 inline bool ContinuationHelper::Frame::assert_frame_laid_out(frame f) {
112   intptr_t* sp = f.sp();
113   address pc = ContinuationHelper::return_address_at(
114                  sp - frame::sender_sp_ret_address_offset());
115   intptr_t* fp = *(intptr_t**)(sp - frame::sender_sp_offset);
116   assert(f.raw_pc() == pc, "f.ra_pc: " INTPTR_FORMAT " actual: " INTPTR_FORMAT, p2i(f.raw_pc()), p2i(pc));
117   assert(f.fp() == fp, "f.fp: " INTPTR_FORMAT " actual: " INTPTR_FORMAT, p2i(f.fp()), p2i(fp));
118   return f.raw_pc() == pc && f.fp() == fp;
119 }
120 #endif
121 
122 inline intptr_t** ContinuationHelper::Frame::callee_link_address(const frame& f) {
123   return (intptr_t**)(f.sp() - frame::sender_sp_offset);
124 }
125 
126 inline address* ContinuationHelper::Frame::return_pc_address(const frame& f) {
127   return (address*)(f.real_fp() - 1);
128 }
129 
130 inline address* ContinuationHelper::InterpretedFrame::return_pc_address(const frame& f) {
131   return (address*)(f.fp() + frame::return_addr_offset);
132 }
133 
134 inline void ContinuationHelper::InterpretedFrame::patch_sender_sp(frame& f, const frame& caller) {
135   intptr_t* sp = caller.unextended_sp();
136   assert(f.is_interpreted_frame(), "");
137   intptr_t* la = f.addr_at(frame::interpreter_frame_sender_sp_offset);
138   *la = f.is_heap_frame() ? (intptr_t)(sp - f.fp()) : (intptr_t)sp;
139 }
140 
141 inline address ContinuationHelper::Frame::real_pc(const frame& f) {
142   // Always used in assertions. Just strip it.
143   address* pc_addr = &(((address*) f.sp())[-1]);
144   return pauth_strip_pointer(*pc_addr);
145 }
146 
147 inline void ContinuationHelper::Frame::patch_pc(const frame& f, address pc) {
148   address* pc_addr = &(((address*) f.sp())[-1]);
149   *pc_addr = pauth_sign_return_address(pc);
150 }
151 
152 inline intptr_t* ContinuationHelper::InterpretedFrame::frame_top(const frame& f, InterpreterOopMap* mask) { // inclusive; this will be copied with the frame
153   // interpreter_frame_last_sp_offset, points to unextended_sp includes arguments in the frame
154   // interpreter_frame_initial_sp_offset excludes expression stack slots
155   int expression_stack_sz = expression_stack_size(f, mask);
156   intptr_t* res = (intptr_t*)f.at_relative(frame::interpreter_frame_initial_sp_offset) - expression_stack_sz;
157   assert(res == (intptr_t*)f.interpreter_frame_monitor_end() - expression_stack_sz, "");
158   assert(res >= f.unextended_sp(),
159     "res: " INTPTR_FORMAT " initial_sp: " INTPTR_FORMAT " last_sp: " INTPTR_FORMAT " unextended_sp: " INTPTR_FORMAT " expression_stack_size: %d",
160     p2i(res), p2i(f.addr_at(frame::interpreter_frame_initial_sp_offset)), f.at_relative_or_null(frame::interpreter_frame_last_sp_offset),
161     p2i(f.unextended_sp()), expression_stack_sz);
162   return res;
163 }
164 
165 inline intptr_t* ContinuationHelper::InterpretedFrame::frame_bottom(const frame& f) { // exclusive; this will not be copied with the frame
166   return (intptr_t*)f.at_relative(frame::interpreter_frame_locals_offset) + 1; // exclusive, so we add 1 word
167 }
168 
169 inline intptr_t* ContinuationHelper::InterpretedFrame::frame_top(const frame& f, int callee_argsize, bool callee_interpreted) {
170   return f.unextended_sp() + (callee_interpreted ? callee_argsize : 0);
171 }
172 
173 inline intptr_t* ContinuationHelper::InterpretedFrame::callers_sp(const frame& f) {
174   return f.fp() + frame::metadata_words;
175 }
176 
177 #endif // CPU_AARCH64_CONTINUATIONHELPER_AARCH64_INLINE_HPP