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"
2130fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2230fdc8d8SChris Lattner 
2330fdc8d8SChris Lattner using namespace lldb;
2430fdc8d8SChris Lattner using namespace lldb_private;
2530fdc8d8SChris Lattner 
2630fdc8d8SChris Lattner //----------------------------------------------------------------------
2730fdc8d8SChris Lattner // ThreadPlanStepOut: Step out of the current frame
2830fdc8d8SChris Lattner //----------------------------------------------------------------------
2930fdc8d8SChris Lattner 
3030fdc8d8SChris Lattner ThreadPlanStepOut::ThreadPlanStepOut
3130fdc8d8SChris Lattner (
3230fdc8d8SChris Lattner     Thread &thread,
3330fdc8d8SChris Lattner     SymbolContext *context,
3430fdc8d8SChris Lattner     bool first_insn,
3530fdc8d8SChris Lattner     bool stop_others,
3630fdc8d8SChris Lattner     Vote stop_vote,
3730fdc8d8SChris Lattner     Vote run_vote
3830fdc8d8SChris Lattner ) :
39*b01e742aSJim Ingham     ThreadPlan (ThreadPlan::eKindStepOut, "Step out", thread, stop_vote, run_vote),
4030fdc8d8SChris Lattner     m_step_from_context (context),
4130fdc8d8SChris Lattner     m_step_from_insn (LLDB_INVALID_ADDRESS),
4230fdc8d8SChris Lattner     m_return_addr (LLDB_INVALID_ADDRESS),
4330fdc8d8SChris Lattner     m_first_insn (first_insn),
4430fdc8d8SChris Lattner     m_return_bp_id(LLDB_INVALID_BREAK_ID),
4530fdc8d8SChris Lattner     m_stop_others (stop_others)
4630fdc8d8SChris Lattner {
4730fdc8d8SChris Lattner     m_step_from_insn = m_thread.GetRegisterContext()->GetPC(0);
4830fdc8d8SChris Lattner 
4930fdc8d8SChris Lattner     // Find the return address and set a breakpoint there:
5030fdc8d8SChris Lattner     // FIXME - can we do this more securely if we know first_insn?
5130fdc8d8SChris Lattner 
5230fdc8d8SChris Lattner     StackFrame *return_frame = m_thread.GetStackFrameAtIndex(1).get();
5330fdc8d8SChris Lattner     if (return_frame)
5430fdc8d8SChris Lattner     {
5530fdc8d8SChris Lattner         m_return_addr = return_frame->GetPC().GetLoadAddress(&m_thread.GetProcess());
5630fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().CreateBreakpoint (m_return_addr, true).get();
5730fdc8d8SChris Lattner         if (return_bp != NULL)
5830fdc8d8SChris Lattner         {
5930fdc8d8SChris Lattner             return_bp->SetThreadID(m_thread.GetID());
6030fdc8d8SChris Lattner             m_return_bp_id = return_bp->GetID();
6130fdc8d8SChris Lattner         }
6230fdc8d8SChris Lattner         else
6330fdc8d8SChris Lattner         {
6430fdc8d8SChris Lattner             m_return_bp_id = LLDB_INVALID_BREAK_ID;
6530fdc8d8SChris Lattner         }
6630fdc8d8SChris Lattner     }
6730fdc8d8SChris Lattner 
6830fdc8d8SChris Lattner     m_stack_depth = m_thread.GetStackFrameCount();
6930fdc8d8SChris Lattner }
7030fdc8d8SChris Lattner 
7130fdc8d8SChris Lattner ThreadPlanStepOut::~ThreadPlanStepOut ()
7230fdc8d8SChris Lattner {
7330fdc8d8SChris Lattner     if (m_return_bp_id != LLDB_INVALID_BREAK_ID)
7430fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID(m_return_bp_id);
7530fdc8d8SChris Lattner }
7630fdc8d8SChris Lattner 
7730fdc8d8SChris Lattner void
7830fdc8d8SChris Lattner ThreadPlanStepOut::GetDescription (Stream *s, lldb::DescriptionLevel level)
7930fdc8d8SChris Lattner {
8030fdc8d8SChris Lattner     if (level == lldb::eDescriptionLevelBrief)
8130fdc8d8SChris Lattner         s->Printf ("step out");
8230fdc8d8SChris Lattner     else
8330fdc8d8SChris Lattner     {
8430fdc8d8SChris Lattner         s->Printf ("Stepping out from address 0x%llx to return address 0x%llx using breakpoint site %d",
8530fdc8d8SChris Lattner                    (uint64_t)m_step_from_insn,
8630fdc8d8SChris Lattner                    (uint64_t)m_return_addr,
8730fdc8d8SChris Lattner                    m_return_bp_id);
8830fdc8d8SChris Lattner     }
8930fdc8d8SChris Lattner }
9030fdc8d8SChris Lattner 
9130fdc8d8SChris Lattner bool
9230fdc8d8SChris Lattner ThreadPlanStepOut::ValidatePlan (Stream *error)
9330fdc8d8SChris Lattner {
9430fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
9530fdc8d8SChris Lattner         return false;
9630fdc8d8SChris Lattner     else
9730fdc8d8SChris Lattner         return true;
9830fdc8d8SChris Lattner }
9930fdc8d8SChris Lattner 
10030fdc8d8SChris Lattner bool
10130fdc8d8SChris Lattner ThreadPlanStepOut::PlanExplainsStop ()
10230fdc8d8SChris Lattner {
10330fdc8d8SChris Lattner     // We don't explain signals or breakpoints (breakpoints that handle stepping in or
10430fdc8d8SChris Lattner     // out will be handled by a child plan.
10530fdc8d8SChris Lattner     Thread::StopInfo info;
10630fdc8d8SChris Lattner     if (m_thread.GetStopInfo (&info))
10730fdc8d8SChris Lattner     {
10830fdc8d8SChris Lattner         StopReason reason = info.GetStopReason();
10930fdc8d8SChris Lattner 
11030fdc8d8SChris Lattner         switch (reason)
11130fdc8d8SChris Lattner         {
11230fdc8d8SChris Lattner             case eStopReasonBreakpoint:
11330fdc8d8SChris Lattner             {
11430fdc8d8SChris Lattner                 // If this is OUR breakpoint, we're fine, otherwise we don't know why this happened...
11530fdc8d8SChris Lattner                 BreakpointSiteSP this_site = m_thread.GetProcess().GetBreakpointSiteList().FindByID (info.GetBreakpointSiteID());
11630fdc8d8SChris Lattner                 if (!this_site)
11730fdc8d8SChris Lattner                     return false;
11830fdc8d8SChris Lattner 
11930fdc8d8SChris Lattner                 if (this_site->IsBreakpointAtThisSite (m_return_bp_id))
12030fdc8d8SChris Lattner                 {
12130fdc8d8SChris Lattner                     // If there was only one owner, then we're done.  But if we also hit some
12230fdc8d8SChris Lattner                     // user breakpoint on our way out, we should mark ourselves as done, but
12330fdc8d8SChris Lattner                     // also not claim to explain the stop, since it is more important to report
12430fdc8d8SChris Lattner                     // the user breakpoint than the step out completion.
12530fdc8d8SChris Lattner 
12630fdc8d8SChris Lattner                     if (this_site->GetNumberOfOwners() == 1)
12730fdc8d8SChris Lattner                         return true;
12830fdc8d8SChris Lattner                     else
12930fdc8d8SChris Lattner                     {
13030fdc8d8SChris Lattner                         SetPlanComplete();
13130fdc8d8SChris Lattner                         return false;
13230fdc8d8SChris Lattner                     }
13330fdc8d8SChris Lattner                 }
13430fdc8d8SChris Lattner                 else
13530fdc8d8SChris Lattner                     return false;
13630fdc8d8SChris Lattner             }
13730fdc8d8SChris Lattner             case eStopReasonWatchpoint:
13830fdc8d8SChris Lattner             case eStopReasonSignal:
13930fdc8d8SChris Lattner             case eStopReasonException:
14030fdc8d8SChris Lattner                 return false;
14130fdc8d8SChris Lattner             default:
14230fdc8d8SChris Lattner                 return true;
14330fdc8d8SChris Lattner         }
14430fdc8d8SChris Lattner     }
14530fdc8d8SChris Lattner     return true;
14630fdc8d8SChris Lattner }
14730fdc8d8SChris Lattner 
14830fdc8d8SChris Lattner bool
14930fdc8d8SChris Lattner ThreadPlanStepOut::ShouldStop (Event *event_ptr)
15030fdc8d8SChris Lattner {
15130fdc8d8SChris Lattner     if (IsPlanComplete()
15230fdc8d8SChris Lattner         || m_thread.GetRegisterContext()->GetPC() == m_return_addr
15330fdc8d8SChris Lattner         || m_stack_depth > m_thread.GetStackFrameCount())
15430fdc8d8SChris Lattner     {
15530fdc8d8SChris Lattner         SetPlanComplete();
15630fdc8d8SChris Lattner         return true;
15730fdc8d8SChris Lattner     }
15830fdc8d8SChris Lattner     else
15930fdc8d8SChris Lattner         return false;
16030fdc8d8SChris Lattner }
16130fdc8d8SChris Lattner 
16230fdc8d8SChris Lattner bool
16330fdc8d8SChris Lattner ThreadPlanStepOut::StopOthers ()
16430fdc8d8SChris Lattner {
16530fdc8d8SChris Lattner     return m_stop_others;
16630fdc8d8SChris Lattner }
16730fdc8d8SChris Lattner 
16830fdc8d8SChris Lattner StateType
16930fdc8d8SChris Lattner ThreadPlanStepOut::RunState ()
17030fdc8d8SChris Lattner {
17130fdc8d8SChris Lattner     return eStateRunning;
17230fdc8d8SChris Lattner }
17330fdc8d8SChris Lattner 
17430fdc8d8SChris Lattner bool
17530fdc8d8SChris Lattner ThreadPlanStepOut::WillResume (StateType resume_state, bool current_plan)
17630fdc8d8SChris Lattner {
17730fdc8d8SChris Lattner     ThreadPlan::WillResume (resume_state, current_plan);
17830fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
17930fdc8d8SChris Lattner         return false;
18030fdc8d8SChris Lattner 
18130fdc8d8SChris Lattner     if (current_plan)
18230fdc8d8SChris Lattner     {
18330fdc8d8SChris Lattner         Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
18430fdc8d8SChris Lattner         if (return_bp != NULL)
18530fdc8d8SChris Lattner             return_bp->SetEnabled (true);
18630fdc8d8SChris Lattner     }
18730fdc8d8SChris Lattner     return true;
18830fdc8d8SChris Lattner }
18930fdc8d8SChris Lattner 
19030fdc8d8SChris Lattner bool
19130fdc8d8SChris Lattner ThreadPlanStepOut::WillStop ()
19230fdc8d8SChris Lattner {
19330fdc8d8SChris Lattner     Breakpoint *return_bp = m_thread.GetProcess().GetTarget().GetBreakpointByID(m_return_bp_id).get();
19430fdc8d8SChris Lattner     if (return_bp != NULL)
19530fdc8d8SChris Lattner         return_bp->SetEnabled (false);
19630fdc8d8SChris Lattner     return true;
19730fdc8d8SChris Lattner }
19830fdc8d8SChris Lattner 
19930fdc8d8SChris Lattner bool
20030fdc8d8SChris Lattner ThreadPlanStepOut::MischiefManaged ()
20130fdc8d8SChris Lattner {
20230fdc8d8SChris Lattner     if (m_return_bp_id == LLDB_INVALID_BREAK_ID)
20330fdc8d8SChris Lattner     {
20430fdc8d8SChris Lattner         // If I couldn't set this breakpoint, then I'm just going to jettison myself.
20530fdc8d8SChris Lattner         return true;
20630fdc8d8SChris Lattner     }
20730fdc8d8SChris Lattner     else if (IsPlanComplete())
20830fdc8d8SChris Lattner     {
20930fdc8d8SChris Lattner         // Did I reach my breakpoint?  If so I'm done.
21030fdc8d8SChris Lattner         //
21130fdc8d8SChris Lattner         // I also check the stack depth, since if we've blown past the breakpoint for some
21230fdc8d8SChris Lattner         // reason and we're now stopping for some other reason altogether, then we're done
21330fdc8d8SChris Lattner         // with this step out operation.
21430fdc8d8SChris Lattner 
21530fdc8d8SChris Lattner         Log *log = lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_STEP);
21630fdc8d8SChris Lattner         if (log)
21730fdc8d8SChris Lattner             log->Printf("Completed step out plan.");
21830fdc8d8SChris Lattner         m_thread.GetProcess().GetTarget().RemoveBreakpointByID (m_return_bp_id);
21930fdc8d8SChris Lattner         m_return_bp_id = LLDB_INVALID_BREAK_ID;
22030fdc8d8SChris Lattner         ThreadPlan::MischiefManaged ();
22130fdc8d8SChris Lattner         return true;
22230fdc8d8SChris Lattner     }
22330fdc8d8SChris Lattner     else
22430fdc8d8SChris Lattner     {
22530fdc8d8SChris Lattner         return false;
22630fdc8d8SChris Lattner     }
22730fdc8d8SChris Lattner }
22830fdc8d8SChris Lattner 
229