1 # Copyright (C) 2010-2014 Free Software Foundation, Inc. 2 3 # This program is free software; you can redistribute it and/or modify 4 # it under the terms of the GNU General Public License as published by 5 # the Free Software Foundation; either version 3 of the License, or 6 # (at your option) any later version. 7 # 8 # This program is distributed in the hope that it will be useful, 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 11 # GNU General Public License for more details. 12 # 13 # You should have received a copy of the GNU General Public License 14 # along with this program. If not, see <http://www.gnu.org/licenses/>. 15 16 import traceback 17 import os 18 import sys 19 import _gdb 20 21 if sys.version_info[0] > 2: 22 # Python 3 moved "reload" 23 from imp import reload 24 25 from _gdb import * 26 27 class _GdbFile (object): 28 # These two are needed in Python 3 29 encoding = "UTF-8" 30 errors = "strict" 31 32 def close(self): 33 # Do nothing. 34 return None 35 36 def isatty(self): 37 return False 38 39 def writelines(self, iterable): 40 for line in iterable: 41 self.write(line) 42 43 def flush(self): 44 flush() 45 46 class GdbOutputFile (_GdbFile): 47 def write(self, s): 48 write(s, stream=STDOUT) 49 50 sys.stdout = GdbOutputFile() 51 52 class GdbOutputErrorFile (_GdbFile): 53 def write(self, s): 54 write(s, stream=STDERR) 55 56 sys.stderr = GdbOutputErrorFile() 57 58 # Default prompt hook does nothing. 59 prompt_hook = None 60 61 # Ensure that sys.argv is set to something. 62 # We do not use PySys_SetArgvEx because it did not appear until 2.6.6. 63 sys.argv = [''] 64 65 # Initial pretty printers. 66 pretty_printers = [] 67 68 # Initial type printers. 69 type_printers = [] 70 # Initial frame filters. 71 frame_filters = {} 72 73 # Convenience variable to GDB's python directory 74 PYTHONDIR = os.path.dirname(os.path.dirname(__file__)) 75 76 # Auto-load all functions/commands. 77 78 # Packages to auto-load. 79 80 packages = [ 81 'function', 82 'command' 83 ] 84 85 # pkgutil.iter_modules is not available prior to Python 2.6. Instead, 86 # manually iterate the list, collating the Python files in each module 87 # path. Construct the module name, and import. 88 89 def auto_load_packages(): 90 for package in packages: 91 location = os.path.join(os.path.dirname(__file__), package) 92 if os.path.exists(location): 93 py_files = filter(lambda x: x.endswith('.py') 94 and x != '__init__.py', 95 os.listdir(location)) 96 97 for py_file in py_files: 98 # Construct from foo.py, gdb.module.foo 99 modname = "%s.%s.%s" % ( __name__, package, py_file[:-3] ) 100 try: 101 if modname in sys.modules: 102 # reload modules with duplicate names 103 reload(__import__(modname)) 104 else: 105 __import__(modname) 106 except: 107 sys.stderr.write (traceback.format_exc() + "\n") 108 109 auto_load_packages() 110 111 def GdbSetPythonDirectory(dir): 112 """Update sys.path, reload gdb and auto-load packages.""" 113 global PYTHONDIR 114 115 try: 116 sys.path.remove(PYTHONDIR) 117 except ValueError: 118 pass 119 sys.path.insert(0, dir) 120 121 PYTHONDIR = dir 122 123 # note that reload overwrites the gdb module without deleting existing 124 # attributes 125 reload(__import__(__name__)) 126 auto_load_packages() 127