gdb/python/lib/gdb/__init__.py

   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()