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,
3830fdc8d8SChris Lattner     Vote run_vote
3930fdc8d8SChris Lattner ) :
40b01e742aSJim Ingham     ThreadPlan (ThreadPlan::eKindStepOut, "Step out", thread, stop_vote, run_vote),
4130fdc8d8SChris Lattner     m_step_from_context (context),
4230fdc8d8SChris Lattner     m_step_from_insn (LLDB_INVALID_ADDRESS),
431ee0d4f7SBenjamin Kramer     m_return_bp_id (LLDB_INVALID_BREAK_ID),
4430fdc8d8SChris Lattner     m_return_addr (LLDB_INVALID_ADDRESS),
4530fdc8d8SChris Lattner     m_first_insn (first_insn),
4630fdc8d8SChris Lattner     m_stop_others (stop_others)
4730fdc8d8SChris Lattner {
4830fdc8d8SChris Lattner     m_step_from_insn = m_thread.GetRegisterContext()->GetPC(0);
4930fdc8d8SChris Lattner 
5030fdc8d8SChris Lattner     // Find the return address and set a breakpoint there:
5130fdc8d8SChris Lattner     // FIXME - can we do this more securely if we know first_insn?
5230fdc8d8SChris Lattner 
5330fdc8d8SChris Lattner     StackFrame *return_frame = m_thread.GetStackFrameAtIndex(1).get();
5430fdc8d8SChris Lattner     if (return_frame)
5530fdc8d8SChris Lattner     {
569da7bd07SGreg Clayton         // TODO: check for inlined frames and do the right thing...
579da7bd07SGreg Clayton         m_return_addr = return_frame->GetRegisterContext()->GetPC();
5830fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().CreateBreakpoint (m_return_addr, true).get();
5930fdc8d8SChris Lattner         if (return_bp != NULL)
6030fdc8d8SChris Lattner         {
6130fdc8d8SChris Lattner             return_bp->SetThreadID(m_thread.GetID());
6230fdc8d8SChris Lattner             m_return_bp_id = return_bp->GetID();
6330fdc8d8SChris Lattner         }
6430fdc8d8SChris Lattner         else
6530fdc8d8SChris Lattner         {
6630fdc8d8SChris Lattner             m_return_bp_id = LLDB_INVALID_BREAK_ID;
6730fdc8d8SChris Lattner         }
6830fdc8d8SChris Lattner     }
6930fdc8d8SChris Lattner 
7030fdc8d8SChris Lattner     m_stack_depth = m_thread.GetStackFrameCount();
7130fdc8d8SChris Lattner }
7230fdc8d8SChris Lattner 
7330fdc8d8SChris Lattner ThreadPlanStepOut::~ThreadPlanStepOut ()
7430fdc8d8SChris Lattner {
7530fdc8d8SChris Lattner     if (m_return_bp_id != LLDB_INVALID_BREAK_ID)
7630fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID(m_return_bp_id);
7730fdc8d8SChris Lattner }
7830fdc8d8SChris Lattner 
7930fdc8d8SChris Lattner void
8030fdc8d8SChris Lattner ThreadPlanStepOut::GetDescription (Stream *s, lldb::DescriptionLevel level)
8130fdc8d8SChris Lattner {
8230fdc8d8SChris Lattner     if (level == lldb::eDescriptionLevelBrief)
8330fdc8d8SChris Lattner         s->Printf ("step out");
8430fdc8d8SChris Lattner     else
8530fdc8d8SChris Lattner     {
8630fdc8d8SChris Lattner         s->Printf ("Stepping out from address 0x%llx to return address 0x%llx using breakpoint site %d",
8730fdc8d8SChris Lattner                    (uint64_t)m_step_from_insn,
8830fdc8d8SChris Lattner                    (uint64_t)m_return_addr,
8930fdc8d8SChris Lattner                    m_return_bp_id);
9030fdc8d8SChris Lattner     }
9130fdc8d8SChris Lattner }
9230fdc8d8SChris Lattner 
9330fdc8d8SChris Lattner bool
9430fdc8d8SChris Lattner ThreadPlanStepOut::ValidatePlan (Stream *error)
9530fdc8d8SChris Lattner {
9630fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
9730fdc8d8SChris Lattner         return false;
9830fdc8d8SChris Lattner     else
9930fdc8d8SChris Lattner         return true;
10030fdc8d8SChris Lattner }
10130fdc8d8SChris Lattner 
10230fdc8d8SChris Lattner bool
10330fdc8d8SChris Lattner ThreadPlanStepOut::PlanExplainsStop ()
10430fdc8d8SChris Lattner {
10530fdc8d8SChris Lattner     // We don't explain signals or breakpoints (breakpoints that handle stepping in or
10630fdc8d8SChris Lattner     // out will be handled by a child plan.
107b15bfc75SJim Ingham     StopInfoSP stop_info_sp = GetPrivateStopReason();
108b15bfc75SJim Ingham     if (stop_info_sp)
10930fdc8d8SChris Lattner     {
110b15bfc75SJim Ingham         StopReason reason = stop_info_sp->GetStopReason();
11130fdc8d8SChris Lattner         switch (reason)
11230fdc8d8SChris Lattner         {
11330fdc8d8SChris Lattner         case eStopReasonBreakpoint:
11430fdc8d8SChris Lattner         {
11530fdc8d8SChris Lattner             // If this is OUR breakpoint, we're fine, otherwise we don't know why this happened...
116b15bfc75SJim Ingham             BreakpointSiteSP site_sp (m_thread.GetProcess().GetBreakpointSiteList().FindByID (stop_info_sp->GetValue()));
117f4b47e15SGreg Clayton             if (site_sp && site_sp->IsBreakpointAtThisSite (m_return_bp_id))
11830fdc8d8SChris Lattner             {
11930fdc8d8SChris Lattner                 // If there was only one owner, then we're done.  But if we also hit some
12030fdc8d8SChris Lattner                 // user breakpoint on our way out, we should mark ourselves as done, but
12130fdc8d8SChris Lattner                 // also not claim to explain the stop, since it is more important to report
12230fdc8d8SChris Lattner                 // the user breakpoint than the step out completion.
12330fdc8d8SChris Lattner 
124f4b47e15SGreg Clayton                 if (site_sp->GetNumberOfOwners() == 1)
12530fdc8d8SChris Lattner                     return true;
126f4b47e15SGreg Clayton 
12730fdc8d8SChris Lattner                 SetPlanComplete();
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 {
14630fdc8d8SChris Lattner     if (IsPlanComplete()
14730fdc8d8SChris Lattner         || m_thread.GetRegisterContext()->GetPC() == m_return_addr
14830fdc8d8SChris Lattner         || m_stack_depth > m_thread.GetStackFrameCount())
14930fdc8d8SChris Lattner     {
15030fdc8d8SChris Lattner         SetPlanComplete();
15130fdc8d8SChris Lattner         return true;
15230fdc8d8SChris Lattner     }
15330fdc8d8SChris Lattner     else
15430fdc8d8SChris Lattner         return false;
15530fdc8d8SChris Lattner }
15630fdc8d8SChris Lattner 
15730fdc8d8SChris Lattner bool
15830fdc8d8SChris Lattner ThreadPlanStepOut::StopOthers ()
15930fdc8d8SChris Lattner {
16030fdc8d8SChris Lattner     return m_stop_others;
16130fdc8d8SChris Lattner }
16230fdc8d8SChris Lattner 
16330fdc8d8SChris Lattner StateType
16430fdc8d8SChris Lattner ThreadPlanStepOut::RunState ()
16530fdc8d8SChris Lattner {
16630fdc8d8SChris Lattner     return eStateRunning;
16730fdc8d8SChris Lattner }
16830fdc8d8SChris Lattner 
16930fdc8d8SChris Lattner bool
17030fdc8d8SChris Lattner ThreadPlanStepOut::WillResume (StateType resume_state, bool current_plan)
17130fdc8d8SChris Lattner {
17230fdc8d8SChris Lattner     ThreadPlan::WillResume (resume_state, current_plan);
17330fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
17430fdc8d8SChris Lattner         return false;
17530fdc8d8SChris Lattner 
17630fdc8d8SChris Lattner     if (current_plan)
17730fdc8d8SChris Lattner     {
17830fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
17930fdc8d8SChris Lattner         if (return_bp != NULL)
18030fdc8d8SChris Lattner             return_bp->SetEnabled (true);
18130fdc8d8SChris Lattner     }
18230fdc8d8SChris Lattner     return true;
18330fdc8d8SChris Lattner }
18430fdc8d8SChris Lattner 
18530fdc8d8SChris Lattner bool
18630fdc8d8SChris Lattner ThreadPlanStepOut::WillStop ()
18730fdc8d8SChris Lattner {
18830fdc8d8SChris Lattner     Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
18930fdc8d8SChris Lattner     if (return_bp != NULL)
19030fdc8d8SChris Lattner         return_bp->SetEnabled (false);
19130fdc8d8SChris Lattner     return true;
19230fdc8d8SChris Lattner }
19330fdc8d8SChris Lattner 
19430fdc8d8SChris Lattner bool
19530fdc8d8SChris Lattner ThreadPlanStepOut::MischiefManaged ()
19630fdc8d8SChris Lattner {
19730fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
19830fdc8d8SChris Lattner     {
19930fdc8d8SChris Lattner         // If I couldn't set this breakpoint, then I'm just going to jettison myself.
20030fdc8d8SChris Lattner         return true;
20130fdc8d8SChris Lattner     }
20230fdc8d8SChris Lattner     else if (IsPlanComplete())
20330fdc8d8SChris Lattner     {
20430fdc8d8SChris Lattner         // Did I reach my breakpoint?  If so I'm done.
20530fdc8d8SChris Lattner         //
20630fdc8d8SChris Lattner         // I also check the stack depth, since if we've blown past the breakpoint for some
20730fdc8d8SChris Lattner         // reason and we're now stopping for some other reason altogether, then we're done
20830fdc8d8SChris Lattner         // with this step out operation.
20930fdc8d8SChris Lattner 
210*2d4edfbcSGreg Clayton         LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_STEP));
21130fdc8d8SChris Lattner         if (log)
21230fdc8d8SChris Lattner             log->Printf("Completed step out plan.");
21330fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID (m_return_bp_id);
21430fdc8d8SChris Lattner         m_return_bp_id = LLDB_INVALID_BREAK_ID;
21530fdc8d8SChris Lattner         ThreadPlan::MischiefManaged ();
21630fdc8d8SChris Lattner         return true;
21730fdc8d8SChris Lattner     }
21830fdc8d8SChris Lattner     else
21930fdc8d8SChris Lattner     {
22030fdc8d8SChris Lattner         return false;
22130fdc8d8SChris Lattner     }
22230fdc8d8SChris Lattner }
22330fdc8d8SChris Lattner 
224