lib/Support/PluginLoader.cpp

   1 //===-- PluginLoader.cpp - Implement -load command line option ------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the -load <plugin> command line option handler.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #define DONT_GET_PLUGIN_LOADER_OPTION
  15 #include "llvm/Support/PluginLoader.h"
  16 #include "llvm/System/DynamicLibrary.h"
  17 #include <iostream>
  18 #include <vector>
  19
  20 using namespace llvm;
  21
  22 std::vector<std::string> plugins;
  23
  24 void PluginLoader::operator=(const std::string &Filename) {
  25   std::string ErrorMessage;
  26   try {
  27     sys::DynamicLibrary::LoadLibraryPermanently(Filename.c_str());
  28     plugins.push_back(Filename);
  29   } catch (const std::string& errmsg) {
  30     if (errmsg.empty()) {
  31       ErrorMessage = "Unknown";
  32     } else {
  33       ErrorMessage = errmsg;
  34     }
  35   }
  36   if (!ErrorMessage.empty())
  37     std::cerr << "Error opening '" << Filename << "': " << ErrorMessage
  38               << "\n  -load request ignored.\n";
  39 }
  40
  41 unsigned PluginLoader::getNumPlugins()
  42 {
  43   return plugins.size();
  44 }
  45
  46 std::string& PluginLoader::getPlugin(unsigned num)
  47 {
  48   assert(num < plugins.size() && "Asking for an out of bounds plugin");
  49   return plugins[num];
  50 }