radiant/server.cpp

   1 /*
   2    Copyright (C) 2001-2006, William Joseph.
   3    All Rights Reserved.
   4
   5    This file is part of GtkRadiant.
   6
   7    GtkRadiant is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    GtkRadiant is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with GtkRadiant; if not, write to the Free Software
  19    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22 #include "server.h"
  23
  24 #include "debugging/debugging.h"
  25 #include "warnings.h"
  26
  27 #include <vector>
  28 #include <map>
  29 #include "os/path.h"
  30
  31 #include "modulesystem.h"
  32
  33 class RadiantModuleServer : public ModuleServer
  34 {
  35 typedef std::pair<CopiedString, int> ModuleType;
  36 typedef std::pair<ModuleType, CopiedString> ModuleKey;
  37 typedef std::map<ModuleKey, Module*> Modules_;
  38 Modules_ m_modules;
  39 bool m_error;
  40
  41 public:
  42 RadiantModuleServer() : m_error( false ){
  43 }
  44
  45 void setError( bool error ){
  46         m_error = error;
  47 }
  48 bool getError() const {
  49         return m_error;
  50 }
  51
  52 TextOutputStream& getOutputStream(){
  53         return globalOutputStream();
  54 }
  55 TextOutputStream& getErrorStream(){
  56         return globalErrorStream();
  57 }
  58 DebugMessageHandler& getDebugMessageHandler(){
  59         return globalDebugMessageHandler();
  60 }
  61
  62 void registerModule( const char* type, int version, const char* name, Module& module ){
  63         ASSERT_NOTNULL( &module );
  64         if ( !m_modules.insert( Modules_::value_type( ModuleKey( ModuleType( type, version ), name ), &module ) ).second ) {
  65                 globalErrorStream() << "module already registered: type=" << makeQuoted( type ) << " name=" << makeQuoted( name ) << "\n";
  66         }
  67         else
  68         {
  69                 globalOutputStream() << "Module Registered: type=" << makeQuoted( type ) << " version=" << makeQuoted( version ) << " name=" << makeQuoted( name ) << "\n";
  70         }
  71 }
  72
  73 Module* findModule( const char* type, int version, const char* name ) const {
  74         Modules_::const_iterator i = m_modules.find( ModuleKey( ModuleType( type, version ), name ) );
  75         if ( i != m_modules.end() ) {
  76                 return ( *i ).second;
  77         }
  78         return 0;
  79 }
  80
  81 void foreachModule( const char* type, int version, const Visitor& visitor ){
  82         for ( Modules_::const_iterator i = m_modules.begin(); i != m_modules.end(); ++i )
  83         {
  84                 if ( string_equal( ( *i ).first.first.first.c_str(), type ) ) {
  85                         visitor.visit( ( *i ).first.second.c_str(), *( *i ).second );
  86                 }
  87         }
  88 }
  89 };
  90
  91
  92 #if defined( WIN32 )
  93
  94 #include <windows.h>
  95
  96 #define FORMAT_BUFSIZE 2048
  97 const char* FormatGetLastError(){
  98         static char buf[FORMAT_BUFSIZE];
  99         FormatMessage(
 100                 FORMAT_MESSAGE_FROM_SYSTEM |
 101                 FORMAT_MESSAGE_IGNORE_INSERTS,
 102                 NULL,
 103                 GetLastError(),
 104                 MAKELANGID( LANG_NEUTRAL, SUBLANG_DEFAULT ), // Default language
 105                 buf,
 106                 FORMAT_BUFSIZE,
 107                 NULL
 108                 );
 109         return buf;
 110 }
 111
 112 class DynamicLibrary
 113 {
 114 HMODULE m_library;
 115 public:
 116 typedef int ( __stdcall * FunctionPointer )();
 117
 118 DynamicLibrary( const char* filename ){
 119         m_library = LoadLibrary( filename );
 120         if ( m_library == 0 ) {
 121                 globalErrorStream() << "LoadLibrary failed: '" << filename << "'\n";
 122                 globalErrorStream() << "GetLastError: " << FormatGetLastError();
 123         }
 124 }
 125 ~DynamicLibrary(){
 126         if ( !failed() ) {
 127                 FreeLibrary( m_library );
 128         }
 129 }
 130 bool failed(){
 131         return m_library == 0;
 132 }
 133 FunctionPointer findSymbol( const char* symbol ){
 134         FunctionPointer address = (FunctionPointer) GetProcAddress( m_library, symbol );
 135         if ( address == 0 ) {
 136                 globalErrorStream() << "GetProcAddress failed: '" << symbol << "'\n";
 137                 globalErrorStream() << "GetLastError: " << FormatGetLastError();
 138         }
 139         return address;
 140 }
 141 };
 142
 143 #elif defined( POSIX )
 144
 145 #include <dlfcn.h>
 146
 147 #ifndef RTLD_DEEPBIND
 148 #define RTLD_DEEPBIND 0
 149 #endif
 150
 151 class DynamicLibrary
 152 {
 153 void* m_library;
 154 public:
 155 typedef int ( *FunctionPointer )();
 156
 157 DynamicLibrary( const char* filename ){
 158         m_library = dlopen( filename, RTLD_NOW | (RTLD_DEEPBIND + 0) );
 159 }
 160 ~DynamicLibrary(){
 161         if ( !failed() ) {
 162                 dlclose( m_library );
 163         }
 164 }
 165 bool failed(){
 166         return m_library == 0;
 167 }
 168 FunctionPointer findSymbol( const char* symbol ){
 169         FunctionPointer p = (FunctionPointer)dlsym( m_library, symbol );
 170         if ( p == 0 ) {
 171                 const char* error = reinterpret_cast<const char*>( dlerror() );
 172                 if ( error != 0 ) {
 173                         globalErrorStream() << error;
 174                 }
 175         }
 176         return p;
 177 }
 178 };
 179
 180 #else
 181 #error "unsupported platform"
 182 #endif
 183
 184 class DynamicLibraryModule
 185 {
 186 typedef void ( RADIANT_DLLIMPORT * RegisterModulesFunc )( ModuleServer& server );
 187 DynamicLibrary m_library;
 188 RegisterModulesFunc m_registerModule;
 189 public:
 190 DynamicLibraryModule( const char* filename )
 191         : m_library( filename ), m_registerModule( 0 ){
 192         if ( !m_library.failed() ) {
 193                 m_registerModule = reinterpret_cast<RegisterModulesFunc>( m_library.findSymbol( "Radiant_RegisterModules" ) );
 194 #if 0
 195                 if ( !m_registerModule ) {
 196                         m_registerModule = reinterpret_cast<RegisterModulesFunc>( m_library.findSymbol( "Radiant_RegisterModules@4" ) );
 197                 }
 198 #endif
 199         }
 200 }
 201 bool failed(){
 202         return m_registerModule == 0;
 203 }
 204 void registerModules( ModuleServer& server ){
 205         m_registerModule( server );
 206 }
 207 };
 208
 209
 210 class Libraries
 211 {
 212 typedef std::vector<DynamicLibraryModule*> libraries_t;
 213 libraries_t m_libraries;
 214
 215 public:
 216 ~Libraries(){
 217         release();
 218 }
 219 void registerLibrary( const char* filename, ModuleServer& server ){
 220         DynamicLibraryModule* library = new DynamicLibraryModule( filename );
 221
 222         if ( library->failed() ) {
 223                 delete library;
 224         }
 225         else
 226         {
 227                 m_libraries.push_back( library );
 228                 library->registerModules( server );
 229         }
 230 }
 231 void release(){
 232         for ( libraries_t::iterator i = m_libraries.begin(); i != m_libraries.end(); ++i )
 233         {
 234                 delete *i;
 235         }
 236 }
 237 void clear(){
 238         m_libraries.clear();
 239 }
 240 };
 241
 242
 243 Libraries g_libraries;
 244 RadiantModuleServer g_server;
 245
 246 ModuleServer& GlobalModuleServer_get(){
 247         return g_server;
 248 }
 249
 250 void GlobalModuleServer_loadModule( const char* filename ){
 251         g_libraries.registerLibrary( filename, g_server );
 252 }
 253
 254 void GlobalModuleServer_Initialise(){
 255 }
 256
 257 void GlobalModuleServer_Shutdown(){
 258 }