130fdc8d8SChris Lattner //===-- ThreadPlanStepOut.cpp -----------------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
330fdc8d8SChris Lattner //                     The LLVM Compiler Infrastructure
430fdc8d8SChris Lattner //
530fdc8d8SChris Lattner // This file is distributed under the University of Illinois Open Source
630fdc8d8SChris Lattner // License. See LICENSE.TXT for details.
730fdc8d8SChris Lattner //
830fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
930fdc8d8SChris Lattner 
1030fdc8d8SChris Lattner #include "lldb/Target/ThreadPlanStepOut.h"
1130fdc8d8SChris Lattner 
1230fdc8d8SChris Lattner // C Includes
1330fdc8d8SChris Lattner // C++ Includes
1430fdc8d8SChris Lattner // Other libraries and framework includes
1530fdc8d8SChris Lattner // Project includes
1630fdc8d8SChris Lattner #include "lldb/Breakpoint/Breakpoint.h"
1730fdc8d8SChris Lattner #include "lldb/lldb-private-log.h"
1830fdc8d8SChris Lattner #include "lldb/Core/Log.h"
1930fdc8d8SChris Lattner #include "lldb/Target/Process.h"
2030fdc8d8SChris Lattner #include "lldb/Target/RegisterContext.h"
21f4b47e15SGreg Clayton #include "lldb/Target/StopInfo.h"
2230fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2330fdc8d8SChris Lattner 
2430fdc8d8SChris Lattner using namespace lldb;
2530fdc8d8SChris Lattner using namespace lldb_private;
2630fdc8d8SChris Lattner 
2730fdc8d8SChris Lattner //----------------------------------------------------------------------
2830fdc8d8SChris Lattner // ThreadPlanStepOut: Step out of the current frame
2930fdc8d8SChris Lattner //----------------------------------------------------------------------
3030fdc8d8SChris Lattner 
3130fdc8d8SChris Lattner ThreadPlanStepOut::ThreadPlanStepOut
3230fdc8d8SChris Lattner (
3330fdc8d8SChris Lattner     Thread &thread,
3430fdc8d8SChris Lattner     SymbolContext *context,
3530fdc8d8SChris Lattner     bool first_insn,
3630fdc8d8SChris Lattner     bool stop_others,
3730fdc8d8SChris Lattner     Vote stop_vote,
38*481cef25SGreg Clayton     Vote run_vote,
39*481cef25SGreg Clayton     uint32_t frame_idx
4030fdc8d8SChris Lattner ) :
41b01e742aSJim Ingham     ThreadPlan (ThreadPlan::eKindStepOut, "Step out", thread, stop_vote, run_vote),
4230fdc8d8SChris Lattner     m_step_from_context (context),
4330fdc8d8SChris Lattner     m_step_from_insn (LLDB_INVALID_ADDRESS),
441ee0d4f7SBenjamin Kramer     m_return_bp_id (LLDB_INVALID_BREAK_ID),
4530fdc8d8SChris Lattner     m_return_addr (LLDB_INVALID_ADDRESS),
4630fdc8d8SChris Lattner     m_first_insn (first_insn),
4730fdc8d8SChris Lattner     m_stop_others (stop_others)
4830fdc8d8SChris Lattner {
4930fdc8d8SChris Lattner     m_step_from_insn = m_thread.GetRegisterContext()->GetPC(0);
5030fdc8d8SChris Lattner 
5130fdc8d8SChris Lattner     // Find the return address and set a breakpoint there:
5230fdc8d8SChris Lattner     // FIXME - can we do this more securely if we know first_insn?
5330fdc8d8SChris Lattner 
54*481cef25SGreg Clayton     StackFrameSP return_frame_sp (m_thread.GetStackFrameAtIndex(frame_idx + 1));
55*481cef25SGreg Clayton     if (return_frame_sp)
5630fdc8d8SChris Lattner     {
579da7bd07SGreg Clayton         // TODO: check for inlined frames and do the right thing...
58*481cef25SGreg Clayton         m_return_addr = return_frame_sp->GetRegisterContext()->GetPC();
5930fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().CreateBreakpoint (m_return_addr, true).get();
6030fdc8d8SChris Lattner         if (return_bp != NULL)
6130fdc8d8SChris Lattner         {
6230fdc8d8SChris Lattner             return_bp->SetThreadID(m_thread.GetID());
6330fdc8d8SChris Lattner             m_return_bp_id = return_bp->GetID();
6430fdc8d8SChris Lattner         }
6530fdc8d8SChris Lattner     }
6630fdc8d8SChris Lattner 
67*481cef25SGreg Clayton     m_stack_depth = m_thread.GetStackFrameCount() - frame_idx;
6830fdc8d8SChris Lattner }
6930fdc8d8SChris Lattner 
7030fdc8d8SChris Lattner ThreadPlanStepOut::~ThreadPlanStepOut ()
7130fdc8d8SChris Lattner {
7230fdc8d8SChris Lattner     if (m_return_bp_id != LLDB_INVALID_BREAK_ID)
7330fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID(m_return_bp_id);
7430fdc8d8SChris Lattner }
7530fdc8d8SChris Lattner 
7630fdc8d8SChris Lattner void
7730fdc8d8SChris Lattner ThreadPlanStepOut::GetDescription (Stream *s, lldb::DescriptionLevel level)
7830fdc8d8SChris Lattner {
7930fdc8d8SChris Lattner     if (level == lldb::eDescriptionLevelBrief)
8030fdc8d8SChris Lattner         s->Printf ("step out");
8130fdc8d8SChris Lattner     else
8230fdc8d8SChris Lattner     {
8330fdc8d8SChris Lattner         s->Printf ("Stepping out from address 0x%llx to return address 0x%llx using breakpoint site %d",
8430fdc8d8SChris Lattner                    (uint64_t)m_step_from_insn,
8530fdc8d8SChris Lattner                    (uint64_t)m_return_addr,
8630fdc8d8SChris Lattner                    m_return_bp_id);
8730fdc8d8SChris Lattner     }
8830fdc8d8SChris Lattner }
8930fdc8d8SChris Lattner 
9030fdc8d8SChris Lattner bool
9130fdc8d8SChris Lattner ThreadPlanStepOut::ValidatePlan (Stream *error)
9230fdc8d8SChris Lattner {
9330fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
9430fdc8d8SChris Lattner         return false;
9530fdc8d8SChris Lattner     else
9630fdc8d8SChris Lattner         return true;
9730fdc8d8SChris Lattner }
9830fdc8d8SChris Lattner 
9930fdc8d8SChris Lattner bool
10030fdc8d8SChris Lattner ThreadPlanStepOut::PlanExplainsStop ()
10130fdc8d8SChris Lattner {
10230fdc8d8SChris Lattner     // We don't explain signals or breakpoints (breakpoints that handle stepping in or
10330fdc8d8SChris Lattner     // out will be handled by a child plan.
104b15bfc75SJim Ingham     StopInfoSP stop_info_sp = GetPrivateStopReason();
105b15bfc75SJim Ingham     if (stop_info_sp)
10630fdc8d8SChris Lattner     {
107b15bfc75SJim Ingham         StopReason reason = stop_info_sp->GetStopReason();
10830fdc8d8SChris Lattner         switch (reason)
10930fdc8d8SChris Lattner         {
11030fdc8d8SChris Lattner         case eStopReasonBreakpoint:
11130fdc8d8SChris Lattner         {
11230fdc8d8SChris Lattner             // If this is OUR breakpoint, we're fine, otherwise we don't know why this happened...
113b15bfc75SJim Ingham             BreakpointSiteSP site_sp (m_thread.GetProcess().GetBreakpointSiteList().FindByID (stop_info_sp->GetValue()));
114f4b47e15SGreg Clayton             if (site_sp && site_sp->IsBreakpointAtThisSite (m_return_bp_id))
11530fdc8d8SChris Lattner             {
116*481cef25SGreg Clayton                 const uint32_t num_frames = m_thread.GetStackFrameCount();
117*481cef25SGreg Clayton                 if (m_stack_depth > num_frames);
118*481cef25SGreg Clayton                     SetPlanComplete();
119*481cef25SGreg Clayton 
12030fdc8d8SChris Lattner                 // If there was only one owner, then we're done.  But if we also hit some
12130fdc8d8SChris Lattner                 // user breakpoint on our way out, we should mark ourselves as done, but
12230fdc8d8SChris Lattner                 // also not claim to explain the stop, since it is more important to report
12330fdc8d8SChris Lattner                 // the user breakpoint than the step out completion.
12430fdc8d8SChris Lattner 
125f4b47e15SGreg Clayton                 if (site_sp->GetNumberOfOwners() == 1)
12630fdc8d8SChris Lattner                     return true;
127f4b47e15SGreg Clayton 
12830fdc8d8SChris Lattner             }
12930fdc8d8SChris Lattner             return false;
13030fdc8d8SChris Lattner         }
13130fdc8d8SChris Lattner         case eStopReasonWatchpoint:
13230fdc8d8SChris Lattner         case eStopReasonSignal:
13330fdc8d8SChris Lattner         case eStopReasonException:
13430fdc8d8SChris Lattner             return false;
135f4b47e15SGreg Clayton 
13630fdc8d8SChris Lattner         default:
13730fdc8d8SChris Lattner             return true;
13830fdc8d8SChris Lattner         }
13930fdc8d8SChris Lattner     }
14030fdc8d8SChris Lattner     return true;
14130fdc8d8SChris Lattner }
14230fdc8d8SChris Lattner 
14330fdc8d8SChris Lattner bool
14430fdc8d8SChris Lattner ThreadPlanStepOut::ShouldStop (Event *event_ptr)
14530fdc8d8SChris Lattner {
146*481cef25SGreg Clayton     if (IsPlanComplete() || m_stack_depth > m_thread.GetStackFrameCount())
14730fdc8d8SChris Lattner     {
14830fdc8d8SChris Lattner         SetPlanComplete();
14930fdc8d8SChris Lattner         return true;
15030fdc8d8SChris Lattner     }
15130fdc8d8SChris Lattner     else
15230fdc8d8SChris Lattner         return false;
15330fdc8d8SChris Lattner }
15430fdc8d8SChris Lattner 
15530fdc8d8SChris Lattner bool
15630fdc8d8SChris Lattner ThreadPlanStepOut::StopOthers ()
15730fdc8d8SChris Lattner {
15830fdc8d8SChris Lattner     return m_stop_others;
15930fdc8d8SChris Lattner }
16030fdc8d8SChris Lattner 
16130fdc8d8SChris Lattner StateType
16206e827ccSJim Ingham ThreadPlanStepOut::GetPlanRunState ()
16330fdc8d8SChris Lattner {
16430fdc8d8SChris Lattner     return eStateRunning;
16530fdc8d8SChris Lattner }
16630fdc8d8SChris Lattner 
16730fdc8d8SChris Lattner bool
16830fdc8d8SChris Lattner ThreadPlanStepOut::WillResume (StateType resume_state, bool current_plan)
16930fdc8d8SChris Lattner {
17030fdc8d8SChris Lattner     ThreadPlan::WillResume (resume_state, current_plan);
17130fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
17230fdc8d8SChris Lattner         return false;
17330fdc8d8SChris Lattner 
17430fdc8d8SChris Lattner     if (current_plan)
17530fdc8d8SChris Lattner     {
17630fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
17730fdc8d8SChris Lattner         if (return_bp != NULL)
17830fdc8d8SChris Lattner             return_bp->SetEnabled (true);
17930fdc8d8SChris Lattner     }
18030fdc8d8SChris Lattner     return true;
18130fdc8d8SChris Lattner }
18230fdc8d8SChris Lattner 
18330fdc8d8SChris Lattner bool
18430fdc8d8SChris Lattner ThreadPlanStepOut::WillStop ()
18530fdc8d8SChris Lattner {
18630fdc8d8SChris Lattner     Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
18730fdc8d8SChris Lattner     if (return_bp != NULL)
18830fdc8d8SChris Lattner         return_bp->SetEnabled (false);
18930fdc8d8SChris Lattner     return true;
19030fdc8d8SChris Lattner }
19130fdc8d8SChris Lattner 
19230fdc8d8SChris Lattner bool
19330fdc8d8SChris Lattner ThreadPlanStepOut::MischiefManaged ()
19430fdc8d8SChris Lattner {
19530fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
19630fdc8d8SChris Lattner     {
19730fdc8d8SChris Lattner         // If I couldn't set this breakpoint, then I'm just going to jettison myself.
19830fdc8d8SChris Lattner         return true;
19930fdc8d8SChris Lattner     }
20030fdc8d8SChris Lattner     else if (IsPlanComplete())
20130fdc8d8SChris Lattner     {
20230fdc8d8SChris Lattner         // Did I reach my breakpoint?  If so I'm done.
20330fdc8d8SChris Lattner         //
20430fdc8d8SChris Lattner         // I also check the stack depth, since if we've blown past the breakpoint for some
20530fdc8d8SChris Lattner         // reason and we're now stopping for some other reason altogether, then we're done
20630fdc8d8SChris Lattner         // with this step out operation.
20730fdc8d8SChris Lattner 
2082d4edfbcSGreg Clayton         LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_STEP));
20930fdc8d8SChris Lattner         if (log)
21030fdc8d8SChris Lattner             log->Printf("Completed step out plan.");
21130fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID (m_return_bp_id);
21230fdc8d8SChris Lattner         m_return_bp_id = LLDB_INVALID_BREAK_ID;
21330fdc8d8SChris Lattner         ThreadPlan::MischiefManaged ();
21430fdc8d8SChris Lattner         return true;
21530fdc8d8SChris Lattner     }
21630fdc8d8SChris Lattner     else
21730fdc8d8SChris Lattner     {
21830fdc8d8SChris Lattner         return false;
21930fdc8d8SChris Lattner     }
22030fdc8d8SChris Lattner }
22130fdc8d8SChris Lattner 
222