1 //===-- SBSourceManager.cpp -------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "lldb/API/SBSourceManager.h"
11 #include "lldb/API/SBDebugger.h"
12 #include "lldb/API/SBStream.h"
13 #include "lldb/API/SBTarget.h"
14 
15 #include "lldb/API/SBFileSpec.h"
16 #include "lldb/Core/Debugger.h"
17 #include "lldb/Core/SourceManager.h"
18 #include "lldb/Core/Stream.h"
19 #include "lldb/Core/StreamFile.h"
20 
21 #include "lldb/Target/Target.h"
22 
23 namespace lldb_private {
24 class SourceManagerImpl {
25 public:
26   SourceManagerImpl(const lldb::DebuggerSP &debugger_sp)
27       : m_debugger_wp(debugger_sp), m_target_wp() {}
28 
29   SourceManagerImpl(const lldb::TargetSP &target_sp)
30       : m_debugger_wp(), m_target_wp(target_sp) {}
31 
32   SourceManagerImpl(const SourceManagerImpl &rhs) {
33     if (&rhs == this)
34       return;
35     m_debugger_wp = rhs.m_debugger_wp;
36     m_target_wp = rhs.m_target_wp;
37   }
38 
39   size_t DisplaySourceLinesWithLineNumbers(const lldb_private::FileSpec &file,
40                                            uint32_t line,
41                                            uint32_t context_before,
42                                            uint32_t context_after,
43                                            const char *current_line_cstr,
44                                            lldb_private::Stream *s) {
45     if (!file)
46       return 0;
47 
48     lldb::TargetSP target_sp(m_target_wp.lock());
49     if (target_sp) {
50       return target_sp->GetSourceManager().DisplaySourceLinesWithLineNumbers(
51           file, line, context_before, context_after, current_line_cstr, s);
52     } else {
53       lldb::DebuggerSP debugger_sp(m_debugger_wp.lock());
54       if (debugger_sp) {
55         return debugger_sp->GetSourceManager()
56             .DisplaySourceLinesWithLineNumbers(file, line, context_before,
57                                                context_after, current_line_cstr,
58                                                s);
59       }
60     }
61     return 0;
62   }
63 
64 private:
65   lldb::DebuggerWP m_debugger_wp;
66   lldb::TargetWP m_target_wp;
67 };
68 }
69 
70 using namespace lldb;
71 using namespace lldb_private;
72 
73 SBSourceManager::SBSourceManager(const SBDebugger &debugger) {
74   m_opaque_ap.reset(new SourceManagerImpl(debugger.get_sp()));
75 }
76 
77 SBSourceManager::SBSourceManager(const SBTarget &target) {
78   m_opaque_ap.reset(new SourceManagerImpl(target.GetSP()));
79 }
80 
81 SBSourceManager::SBSourceManager(const SBSourceManager &rhs) {
82   if (&rhs == this)
83     return;
84 
85   m_opaque_ap.reset(new SourceManagerImpl(*(rhs.m_opaque_ap.get())));
86 }
87 
88 const lldb::SBSourceManager &SBSourceManager::
89 operator=(const lldb::SBSourceManager &rhs) {
90   m_opaque_ap.reset(new SourceManagerImpl(*(rhs.m_opaque_ap.get())));
91   return *this;
92 }
93 
94 SBSourceManager::~SBSourceManager() {}
95 
96 size_t SBSourceManager::DisplaySourceLinesWithLineNumbers(
97     const SBFileSpec &file, uint32_t line, uint32_t context_before,
98     uint32_t context_after, const char *current_line_cstr, SBStream &s) {
99   if (m_opaque_ap.get() == NULL)
100     return 0;
101 
102   return m_opaque_ap->DisplaySourceLinesWithLineNumbers(
103       file.ref(), line, context_before, context_after, current_line_cstr,
104       s.get());
105 }
106