text: #! /usr/bin/env python
  
  # Copyright (C) 2011-2014 Free Software Foundation, Inc.
  #
  # This file is part of GDB.
  #
  # This program is free software; you can redistribute it and/or modify
  # it under the terms of the GNU General Public License as published by
  # the Free Software Foundation; either version 3 of the License, or
  # (at your option) any later version.
  #
  # This program is distributed in the hope that it will be useful,
  # but WITHOUT ANY WARRANTY; without even the implied warranty of
  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  # GNU General Public License for more details.
  #
  # You should have received a copy of the GNU General Public License
  # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  
  # This program requires readelf, gdb and objcopy.  The default values are gdb
  # from the build tree and objcopy and readelf from $PATH.  They may be
  # overridden by setting environment variables GDB, READELF and OBJCOPY
  # respectively.  We assume the current directory is either $obj/gdb or
  # $obj/gdb/testsuite.
  #
  # Example usage:
  #
  # bash$ cd $objdir/gdb/testsuite
  # bash$ python test_pubnames_and_indexes.py <binary_name>
  
  """test_pubnames_and_indexes.py
  
  Test that the gdb_index produced by gold is identical to the gdb_index
  produced by gdb itself.
  
  Further check that the pubnames and pubtypes produced by gcc are identical
  to those that gdb produces.
  
  Finally, check that all strings are canonicalized identically.
  """
  
  __author__ = 'saugustine@google.com (Sterling Augustine)'
  
  import os
  import subprocess
  import sys
  
  OBJCOPY = None
  READELF = None
  GDB = None
  
  def get_pub_info(filename, readelf_option):
    """Parse and return all the pubnames or pubtypes produced by readelf with the
    given option.
    """
    readelf = subprocess.Popen([READELF, '--debug-dump=' + readelf_option,
                                         filename], stdout=subprocess.PIPE)
    pubnames = []
  
    in_list = False;
    for line in readelf.stdout:
      fields = line.split(None, 1)
      if (len(fields) == 2 and fields[0] == 'Offset'
          and fields[1].strip() == 'Name'):
        in_list = True
      # Either a blank-line or a new Length field terminates the current section.
      elif (len(fields) == 0 or fields[0] == 'Length:'):
        in_list = False;
      elif (in_list):
        pubnames.append(fields[1].strip())
  
    readelf.wait()
    return pubnames
  
  
  def get_gdb_index(filename):
    """Use readelf to dump the gdb index and collect the types and names"""
    readelf = subprocess.Popen([READELF, '--debug-dump=gdb_index',
                                filename], stdout=subprocess.PIPE)
    index_symbols = []
    symbol_table_started = False
    for line in readelf.stdout:
      if (line == 'Symbol table:\n'):
        symbol_table_started = True;
      elif (symbol_table_started):
        # Readelf prints gdb-index lines formatted like so:
        # [  4] two::c2<double>::c2: 0
        # So take the string between the first close bracket and the last colon.
        index_symbols.append(line[line.find(']') + 2: line.rfind(':')])
  
    readelf.wait()
    return index_symbols
  
  
  def CheckSets(list0, list1, name0, name1):
    """Report any setwise differences between the two lists"""
  
    if len(list0) == 0 or len(list1) == 0:
      return False
  
    difference0 = set(list0) - set(list1)
    if len(difference0) != 0:
      print "Elements in " + name0 + " but not " + name1 + ": (",
      print len(difference0),
      print ")"
      for element in difference0:
        print "  " + element
  
    difference1 = set(list1) - set(list0)
    if len(difference1) != 0:
      print "Elements in " + name1 + " but not " + name0 + ": (",
      print len(difference1),
      print ")"
      for element in difference1:
        print "  " + element
  
    if (len(difference0) != 0 or len(difference1) != 0):
      return True
  
    print name0 + " and " + name1 + " are identical."
    return False
  
  
  def find_executables():
    """Find the copies of readelf, objcopy and gdb to use."""
    # Executable finding logic follows cc-with-index.sh
    global READELF
    READELF = os.getenv('READELF')
    if READELF is None:
      READELF = 'readelf'
    global OBJCOPY
    OBJCOPY = os.getenv('OBJCOPY')
    if OBJCOPY is None:
      OBJCOPY = 'objcopy'
  
    global GDB
    GDB = os.getenv('GDB')
    if (GDB is None):
      if os.path.isfile('./gdb') and os.access('./gdb', os.X_OK):
        GDB = './gdb'
      elif os.path.isfile('../gdb') and os.access('../gdb', os.X_OK):
        GDB = '../gdb'
      elif os.path.isfile('../../gdb') and os.access('../../gdb', os.X_OK):
        GDB = '../../gdb'
      else:
        # Punt and use the gdb in the path.
        GDB = 'gdb'
  
  
  def main(argv):
    """The main subprogram."""
    if len(argv) != 2:
      print "Usage: test_pubnames_and_indexes.py <filename>"
      sys.exit(2)
  
    find_executables();
  
    # Get the index produced by Gold--It should have been built into the binary.
    gold_index = get_gdb_index(argv[1])
  
    # Collect the pubnames and types list
    pubs_list = get_pub_info(argv[1], "pubnames")
    pubs_list = pubs_list + get_pub_info(argv[1], "pubtypes")
  
    # Generate a .gdb_index with gdb
    gdb_index_file = argv[1] + '.gdb-generated-index'
    subprocess.check_call([OBJCOPY, '--remove-section', '.gdb_index',
                           argv[1], gdb_index_file])
    subprocess.check_call([GDB, '-batch', '-nx', gdb_index_file,
                           '-ex', 'save gdb-index ' + os.path.dirname(argv[1]),
                           '-ex', 'quit'])
    subprocess.check_call([OBJCOPY, '--add-section',
                           '.gdb_index=' + gdb_index_file + '.gdb-index',
                           gdb_index_file])
    gdb_index = get_gdb_index(gdb_index_file)
    os.remove(gdb_index_file)
    os.remove(gdb_index_file + '.gdb-index')
  
    failed = False
    gdb_index.sort()
    gold_index.sort()
    pubs_list.sort()
  
    # Find the differences between the various indices.
    if len(gold_index) == 0:
      print "Gold index is empty"
      failed |= True
  
    if len(gdb_index) == 0:
      print "Gdb index is empty"
      failed |= True
  
    if len(pubs_list) == 0:
      print "Pubs list is empty"
      failed |= True
  
    failed |= CheckSets(gdb_index, gold_index, "gdb index", "gold index")
    failed |= CheckSets(pubs_list, gold_index, "pubs list", "gold index")
    failed |= CheckSets(pubs_list, gdb_index, "pubs list", "gdb index")
  
    if failed:
      print "Test failed"
      sys.exit(1)
  
  
  if __name__ == '__main__':
    main(sys.argv)
  