QDianaDebug.cc

Go to the documentation of this file.

#include "QDianaDebug.hh"
#include "QDemangle.hh"
#include "QError.hh"
#include <sstream>
#include <sys/ioctl.h>
#include <stdlib.h>
#include <execinfo.h>
#include <cxxabi.h>
 
#ifdef __BACKTRACE_ON__
#include <TSystem.h>
#endif
 
namespace Diana {
 
    bool DianaDebug = true;
 
    void SetDianaDebug(bool value) { DianaDebug = value; }
 
    bool GetDianaDebug() { return DianaDebug; }
 
 
    void Abort(const QError& err)
    {
        std::stringstream errstr;
        errstr << "Abort: "<<std::endl
            << "Q> Exception: QError" << std::endl
            << "Q> Code     : " << (int)err.GetCode() 
            << " (" << err.ToString() << ")";
        if("" != err.GetDescription()) {
            errstr << std::endl<<"Q> Descr    : " << err.GetDescription();
        }
        std::cerr<<errstr.str()<<std::endl;
        diana_stacktrace(std::cerr);
        std::cerr << std::flush;
        abort();
    }
 
    void Abort(std::exception& ex) 
    {
        std::stringstream errstr;
        errstr << "Abort: "<<std::endl
            <<"Q> Exception: " << ex.what();
        std::cerr<<errstr.str()<<std::endl;
        diana_stacktrace(std::cerr);
        std::cerr << std::flush;
        abort();
    }
 
    void Abort(const std::string& str) 
    {
        std::stringstream errstr;
        errstr << "Abort: "<<std::endl
            <<"Q> Exception: " << str;
        std::cerr<<errstr.str()<<std::endl;
        diana_stacktrace(std::cerr);
        std::cerr << std::flush;
        abort();
    }
 
    void Abort(...)
    {
        std::stringstream errstr;
        errstr << "Error: "<<std::endl
            << "Q> Exception: Unknown type";
        std::cerr<<errstr.str()<<std::endl;
        diana_stacktrace(std::cerr);
        std::cerr << std::flush;
        abort();
    }
 
};
 
void diana_stacktrace(std::ostream &o,size_t MaxLines){
#ifdef __BACKTRACE_ON__
  FILE *tmpFile=tmpfile();  // Create temp file
  int save_stderr=dup(fileno(stderr));   // Store the fileno for stderr
  dup2(fileno(tmpFile),fileno(stderr));  // redirect stderr -> tmpFile
  gSystem->StackTrace();  // Run the stack trace
  dup2(save_stderr,STDERR_FILENO);  // Redirect stderr to where it was
  rewind(tmpFile);   // Rewind the tmp file
 
  // output the tmpFile to the stream
  // Get the window width (silly)
  struct winsize w;
  ioctl(0, TIOCGWINSZ, &w);
  int NCol=w.ws_col;
  if (NCol < 18) NCol = 18;
  o << "\n" << std::string((NCol-18)/2,'-') << " DIANA Backtrace: " 
    << std::string((NCol-18)/2,'-') << std::endl;
  while(!feof(tmpFile)){
    char c=fgetc(tmpFile);
    o << c;
  }
  o << "\b";
  o << std::string((NCol-18)/2,'-') << " Finish Backtrace " 
    << std::string((NCol-18)/2,'-') << std::endl;
 
  fclose(tmpFile);  // close the tmpFile
#endif
}
 
  /*
  // Get the window width (silly)
  struct winsize w;
  ioctl(0, TIOCGWINSZ, &w);
  int NCol=w.ws_col;
 
  o << "DIANA Backtrace: " << std::endl;
  o << std::string(NCol,'+') << std::endl;
 
  MaxLines+=1;
 
  void *AddrList[MaxLines];
  
  int AddrLen = backtrace(AddrList, MaxLines);
 
  if (AddrLen == 0) {
    o << "  <empty, possibly corrupt>" << std::endl;
    o << std::string(NCol,'+') << std::endl;
    return;
  }
 
  char **SymbolList = backtrace_symbols(AddrList,AddrLen);
  
  size_t maxNameSize = 256;
  char *FuncName=(char*) malloc(maxNameSize*sizeof(char));
  
  for(int i=1;i<AddrLen;i++){
    
    char *begin_name = 0, *begin_offset = 0, *end_offset = 0;
 
    for (char *p = SymbolList[i]; *p; p++){
      if (*p == '(')
        begin_name = p;
      else if (*p == '+')
        begin_offset = p;
      else if (*p == ')' && begin_offset) {
        end_offset = p;
        break;
      }
    }
    
    if (begin_name && begin_offset && end_offset && begin_name < begin_offset){
      *begin_name++ = '\0';
      *begin_offset++ = '\0';
      *end_offset = '\0';
 
      int status;
      // Diana::CAXDemangle doesnt handle spaces in the name properly
      char* ret = abi::__cxa_demangle(begin_name,
                                      FuncName,&maxNameSize,&status);
 
      if (status == 0) {
        FuncName = ret; 
        o << "  " << SymbolList[i] << " : " << FuncName << "+" 
          << begin_offset << std::endl;
      }
      else {
        o << "  " << SymbolList[i] << " : " << begin_name 
          << begin_offset << std::endl;
      }
    }
    else {
      o << SymbolList[i] << std::endl;
    }
    o << std::string(NCol,'+') << std::endl;
  }
 
  free(FuncName);
  free(SymbolList);
}
*/