blob: ef32af4b3f386c5269f59a7acb16d3582d19866a [file] [log] [blame]
Chris Lattner2c54a0d2002-07-23 17:56:53 +00001//===-- PluginLoader.cpp - Implement -load command line option ------------===//
Misha Brukman10468d82005-04-21 22:55:34 +00002//
John Criswell482202a2003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukman10468d82005-04-21 22:55:34 +00007//
John Criswell482202a2003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner2c54a0d2002-07-23 17:56:53 +00009//
Chris Lattner99dcad42004-07-11 01:04:33 +000010// This file implements the -load <plugin> command line option handler.
Chris Lattner2c54a0d2002-07-23 17:56:53 +000011//
12//===----------------------------------------------------------------------===//
13
Chris Lattner99dcad42004-07-11 01:04:33 +000014#define DONT_GET_PLUGIN_LOADER_OPTION
Julien Lerougee9de15b2008-10-22 16:30:41 +000015#include "llvm/Support/ManagedStatic.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000016#include "llvm/Support/PluginLoader.h"
Bill Wendling3750ae22006-11-26 10:52:51 +000017#include "llvm/Support/Streams.h"
Reid Spencer9ec27612004-11-29 14:07:46 +000018#include "llvm/System/DynamicLibrary.h"
Owen Andersonbee30f52009-06-23 00:02:39 +000019#include "llvm/System/Mutex.h"
Bill Wendling30c0f332006-12-07 23:41:45 +000020#include <ostream>
Andrew Lenharthbb4c9c02006-01-26 18:36:50 +000021#include <vector>
Chris Lattnerc9499b62003-12-14 21:35:53 +000022using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000023
Julien Lerougee9de15b2008-10-22 16:30:41 +000024static ManagedStatic<std::vector<std::string> > Plugins;
Owen Andersonbee30f52009-06-23 00:02:39 +000025static ManagedStatic<sys::SmartMutex<true> > PluginsLock;
Andrew Lenharthbb4c9c02006-01-26 18:36:50 +000026
Chris Lattner99dcad42004-07-11 01:04:33 +000027void PluginLoader::operator=(const std::string &Filename) {
Owen Andersonbee30f52009-06-23 00:02:39 +000028 sys::SmartScopedLock<true> Lock(&*PluginsLock);
Chris Lattnera41def52006-07-07 17:14:04 +000029 std::string Error;
30 if (sys::DynamicLibrary::LoadLibraryPermanently(Filename.c_str(), &Error)) {
Bill Wendlingf3baad32006-12-07 01:30:32 +000031 cerr << "Error opening '" << Filename << "': " << Error
32 << "\n -load request ignored.\n";
Chris Lattnera41def52006-07-07 17:14:04 +000033 } else {
34 Plugins->push_back(Filename);
35 }
Chris Lattner2c54a0d2002-07-23 17:56:53 +000036}
Andrew Lenharthbb4c9c02006-01-26 18:36:50 +000037
Chris Lattnera41def52006-07-07 17:14:04 +000038unsigned PluginLoader::getNumPlugins() {
Owen Andersonbee30f52009-06-23 00:02:39 +000039 sys::SmartScopedLock<true> Lock(&*PluginsLock);
Julien Lerougee9de15b2008-10-22 16:30:41 +000040 return Plugins.isConstructed() ? Plugins->size() : 0;
Andrew Lenharthbb4c9c02006-01-26 18:36:50 +000041}
42
Chris Lattnera41def52006-07-07 17:14:04 +000043std::string &PluginLoader::getPlugin(unsigned num) {
Owen Andersonbee30f52009-06-23 00:02:39 +000044 sys::SmartScopedLock<true> Lock(&*PluginsLock);
Julien Lerougee9de15b2008-10-22 16:30:41 +000045 assert(Plugins.isConstructed() && num < Plugins->size() &&
46 "Asking for an out of bounds plugin");
Chris Lattnera41def52006-07-07 17:14:04 +000047 return (*Plugins)[num];
Andrew Lenharthbb4c9c02006-01-26 18:36:50 +000048}