1affc150dc44fab1911775a49636d0ce85333b634Zonr Chang//===- MsgHandling.cpp ----------------------------------------------------===// 2affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// 3affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// The MCLinker Project 4affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// 5affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// This file is distributed under the University of Illinois Open Source 6affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// License. See LICENSE.TXT for details. 7affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// 8affc150dc44fab1911775a49636d0ce85333b634Zonr Chang//===----------------------------------------------------------------------===// 937b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/LD/DiagnosticEngine.h" 1037b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/LD/DiagnosticLineInfo.h" 1137b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/LD/DiagnosticPrinter.h" 1237b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/LD/MsgHandler.h" 1337b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/LD/TextDiagnosticPrinter.h" 1437b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/Support/MsgHandling.h" 1537b74a387bb3993387029859c2d9d051c41c724eStephen Hines#include "mcld/Support/raw_ostream.h" 1622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 17affc150dc44fab1911775a49636d0ce85333b634Zonr Chang#include <llvm/Support/ManagedStatic.h> 18affc150dc44fab1911775a49636d0ce85333b634Zonr Chang#include <llvm/Support/raw_ostream.h> 1922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <llvm/Support/Signals.h> 2022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 2122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <cstdlib> 22affc150dc44fab1911775a49636d0ce85333b634Zonr Chang 2337b74a387bb3993387029859c2d9d051c41c724eStephen Hinesnamespace mcld { 24affc150dc44fab1911775a49636d0ce85333b634Zonr Chang 25affc150dc44fab1911775a49636d0ce85333b634Zonr Chang//===----------------------------------------------------------------------===// 2667e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao// static variables 2767e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao//===----------------------------------------------------------------------===// 2867e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liaostatic llvm::ManagedStatic<DiagnosticEngine> g_pEngine; 29affc150dc44fab1911775a49636d0ce85333b634Zonr Chang 3037b74a387bb3993387029859c2d9d051c41c724eStephen Hinesvoid InitializeDiagnosticEngine(const LinkerConfig& pConfig, 3137b74a387bb3993387029859c2d9d051c41c724eStephen Hines DiagnosticPrinter* pPrinter) { 3222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao g_pEngine->reset(pConfig); 3337b74a387bb3993387029859c2d9d051c41c724eStephen Hines if (pPrinter != NULL) 3467e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao g_pEngine->setPrinter(*pPrinter, false); 3567e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao else { 3637b74a387bb3993387029859c2d9d051c41c724eStephen Hines DiagnosticPrinter* printer = 3737b74a387bb3993387029859c2d9d051c41c724eStephen Hines new TextDiagnosticPrinter(errs(), pConfig); 3867e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao g_pEngine->setPrinter(*printer, true); 39affc150dc44fab1911775a49636d0ce85333b634Zonr Chang } 40affc150dc44fab1911775a49636d0ce85333b634Zonr Chang} 41affc150dc44fab1911775a49636d0ce85333b634Zonr Chang 4237b74a387bb3993387029859c2d9d051c41c724eStephen HinesDiagnosticEngine& getDiagnosticEngine() { 4367e37f1be98c926645219cfb47fab9e90d8c725cShih-wei Liao return *g_pEngine; 44affc150dc44fab1911775a49636d0ce85333b634Zonr Chang} 45affc150dc44fab1911775a49636d0ce85333b634Zonr Chang 4637b74a387bb3993387029859c2d9d051c41c724eStephen Hinesbool Diagnose() { 4722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao if (g_pEngine->getPrinter()->getNumErrors() > 0) { 4837b74a387bb3993387029859c2d9d051c41c724eStephen Hines // If we reached here, we are failing ungracefully. Run the interrupt 4937b74a387bb3993387029859c2d9d051c41c724eStephen Hines // handlers 5022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao // to make sure any special cleanups get done, in particular that we remove 5122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao // files registered with RemoveFileOnSignal. 5222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao llvm::sys::RunInterruptHandlers(); 5322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao g_pEngine->getPrinter()->finish(); 5422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao return false; 5522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao } 5622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao return true; 5722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao} 5822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 5937b74a387bb3993387029859c2d9d051c41c724eStephen Hinesvoid FinalizeDiagnosticEngine() { 6022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao g_pEngine->getPrinter()->finish(); 6122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao} 6222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 6337b74a387bb3993387029859c2d9d051c41c724eStephen Hines} // namespace mcld 64