File: library_loader.cpp

package info (click to toggle)
mysql-connector-c%2B%2B 1.1.0~r814-1
  • links: PTS
  • area: main
  • in suites: squeeze
  • size: 4,388 kB
  • ctags: 3,838
  • sloc: cpp: 39,820; ansic: 2,114; perl: 936; php: 528; xml: 399; sql: 399; sh: 85; makefile: 47
file content (131 lines) | stat: -rw-r--r-- 2,954 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/*
   Copyright 2009 - 2010 Sun Microsystems, Inc.  All rights reserved.

   The MySQL Connector/C++ is licensed under the terms of the GPL
   <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
   MySQL Connectors. There are special exceptions to the terms and
   conditions of the GPL as it is applied to this software, see the
   FLOSS License Exception
   <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
*/

#include "binding_config.h"
#ifndef MYSQLCLIENT_STATIC_BINDING

#include <string>
#include <stdexcept>


// Needed to make SetDllDirectory available
#define _WIN32_WINNT	0x0502
#include "library_loader.h"

/* TODO consider using of dlopen, dlsym and dlclose definitions in my_global.h
 * sort of doesn't like that.
 */
#ifndef _WIN32
#define LoadLibrary(p)			::dlopen(p, RTLD_LAZY)
#define FreeLibrary(p)			::dlclose(p)
#define GetProcAddress(p1,p2)	::dlsym(p1,p2)
#endif


namespace sql {
namespace mysql {
namespace util {

std::string ErrorMessage()
{
#ifdef _WIN32
	TCHAR buffer[255];
	DWORD _errcode = GetLastError();
	::FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
					NULL, _errcode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
					buffer, sizeof(buffer), NULL);

	return buffer;
#else
	return dlerror();
#endif
}

/* {{{ LibraryLoader::LibraryLoader() */
LibraryLoader::LibraryLoader(const std::string & path2libFile)
	: loadedLibHandle (NULL)
{
	if ((loadedLibHandle = LoadLibrary(path2libFile.c_str())) == NULL) {
		throw std::runtime_error("Couldn't load library " + path2libFile + ": " + ErrorMessage());
	}
}
/* }}} */


/* {{{ LibraryLoader::LibraryLoader() */
LibraryLoader::LibraryLoader(const std::string & dir2look, const std::string & libFileName)
	: loadedLibHandle(NULL)
{

#if _WIN32
	SetDllDirectory(dir2look.c_str());
	if ((loadedLibHandle= LoadLibrary(libFileName.c_str())) == NULL) {
#else
	std::string fullname(dir2look);
	fullname += "/";
	fullname += libFileName;
	if ((loadedLibHandle = LoadLibrary(fullname.c_str())) == NULL) {
#endif
		throw std::runtime_error("Couldn't load library " + libFileName + ": " + ErrorMessage());
	}
}
/* }}} */


/* {{{ LibraryLoader::~LibraryLoader() */
LibraryLoader::~LibraryLoader()
{
	FreeLibrary(loadedLibHandle);
}
/* }}} */


/* {{{ LibraryLoader::GetProcAddr() */
SymbolHandle
LibraryLoader::GetProcAddr(const std::string & name)
{
	if (loadedLibHandle == NULL) {
		return NULL;
	}

	ProcCache::const_iterator cit = functions.find(name);

	if (cit != functions.end()) {
		return cit->second;
	}

	SymbolHandle proc = GetProcAddress(loadedLibHandle, name.c_str());

	if (proc == NULL) {
		throw std::runtime_error("Couldn't find symbol " + name);
	}

	functions.insert(std::make_pair(name, proc));

	return proc;
}
/* }}} */


} /* namespace util */
} /* namespace mysql */
} /* namespace sql */

#endif

/*
 * Local variables:
 * tab-width: 4
 * c-basic-offset: 4
 * End:
 * vim600: noet sw=4 ts=4 fdm=marker
 * vim<600: noet sw=4 ts=4
 */