pg_jruby 0.14.1.rc2-java → 0.17.1-java

data/ext/extconf.rb

@@ -0,0 +1,97 @@
+require 'pp'
+require 'mkmf'
+
+
+if ENV['MAINTAINER_MODE']
+	$stderr.puts "Maintainer mode enabled."
+	$CFLAGS <<
+		' -Wall' <<
+		' -ggdb' <<
+		' -DDEBUG' <<
+		' -pedantic'
+end
+
+if pgdir = with_config( 'pg' )
+	ENV['PATH'] = "#{pgdir}/bin" + File::PATH_SEPARATOR + ENV['PATH']
+end
+
+if ENV['CROSS_COMPILING']
+	$LDFLAGS << " -L#{CONFIG['libdir']}"
+
+	# Link against all required libraries for static build, if they are available
+	have_library( 'crypt32', 'CertOpenStore' ) && append_library( $libs, 'crypt32' )
+	have_library( 'gdi32', 'CreateDC' ) && append_library( $libs, 'gdi32' )
+	have_library( 'secur32' ) && append_library( $libs, 'secur32' )
+	have_library( 'ws2_32', 'WSASocket') && append_library( $libs, 'ws2_32' )
+	have_library( 'crypto', 'BIO_new' ) && append_library( $libs, 'crypto' )
+	have_library( 'ssl', 'SSL_new' ) && append_library( $libs, 'ssl' )
+end
+
+if pgconfig = ( with_config('pg-config') || with_config('pg_config') || find_executable('pg_config') )
+	$stderr.puts "Using config values from %s" % [ pgconfig ]
+	incdir = `"#{pgconfig}" --includedir`.chomp
+	libdir = `"#{pgconfig}" --libdir`.chomp
+	dir_config 'pg', incdir, libdir
+
+	# Try to use runtime path linker option, even if RbConfig doesn't know about it.
+	# The rpath option is usually set implicit by dir_config(), but so far not
+	# on MacOS-X.
+	if RbConfig::CONFIG["RPATHFLAG"].to_s.empty? && try_link('int main() {return 0;}', " -Wl,-rpath,#{libdir}")
+		$LDFLAGS << " -Wl,-rpath,#{libdir}"
+	end
+else
+	$stderr.puts "No pg_config... trying anyway. If building fails, please try again with",
+		" --with-pg-config=/path/to/pg_config"
+	dir_config 'pg'
+end
+
+find_header( 'libpq-fe.h' ) or abort "Can't find the 'libpq-fe.h header"
+find_header( 'libpq/libpq-fs.h' ) or abort "Can't find the 'libpq/libpq-fs.h header"
+find_header( 'pg_config_manual.h' ) or abort "Can't find the 'pg_config_manual.h' header"
+
+abort "Can't find the PostgreSQL client library (libpq)" unless
+	have_library( 'pq', 'PQconnectdb', ['libpq-fe.h'] ) ||
+	have_library( 'libpq', 'PQconnectdb', ['libpq-fe.h'] ) ||
+	have_library( 'ms/libpq', 'PQconnectdb', ['libpq-fe.h'] )
+
+# optional headers/functions
+have_func 'PQconnectionUsedPassword' or
+	abort "Your PostgreSQL is too old. Either install an older version " +
+	      "of this gem or upgrade your database."
+have_func 'PQisthreadsafe'
+have_func 'PQprepare'
+have_func 'PQexecParams'
+have_func 'PQescapeString'
+have_func 'PQescapeStringConn'
+have_func 'PQescapeLiteral'
+have_func 'PQescapeIdentifier'
+have_func 'PQgetCancel'
+have_func 'lo_create'
+have_func 'pg_encoding_to_char'
+have_func 'pg_char_to_encoding'
+have_func 'PQsetClientEncoding'
+have_func 'PQlibVersion'
+have_func 'PQping'
+have_func 'PQsetSingleRowMode'
+
+have_func 'rb_encdb_alias'
+have_func 'rb_enc_alias'
+have_func 'rb_thread_call_without_gvl'
+have_func 'rb_thread_call_with_gvl'
+have_func 'rb_thread_fd_select'
+have_func 'rb_w32_wrap_io_handle'
+
+have_const 'PGRES_COPY_BOTH', 'libpq-fe.h'
+have_const 'PGRES_SINGLE_TUPLE', 'libpq-fe.h'
+have_const 'PG_DIAG_TABLE_NAME', 'libpq-fe.h'
+
+$defs.push( "-DHAVE_ST_NOTIFY_EXTRA" ) if
+	have_struct_member 'struct pgNotify', 'extra', 'libpq-fe.h'
+
+# unistd.h confilicts with ruby/win32.h when cross compiling for win32 and ruby 1.9.1
+have_header 'unistd.h'
+have_header 'ruby/st.h' or have_header 'st.h' or abort "pg currently requires the ruby/st.h header"
+
+create_header()
+create_makefile( "pg_ext" )
+

data/ext/gvl_wrappers.c

@@ -0,0 +1,13 @@
+/*
+ * gvl_wrappers.c - Wrapper functions for locking/unlocking the Ruby GVL
+ *
+ */
+
+#include "pg.h"
+
+FOR_EACH_BLOCKING_FUNCTION( DEFINE_GVL_WRAPPER_STRUCT );
+FOR_EACH_BLOCKING_FUNCTION( DEFINE_GVL_SKELETON );
+FOR_EACH_BLOCKING_FUNCTION( DEFINE_GVL_STUB );
+FOR_EACH_CALLBACK_FUNCTION( DEFINE_GVL_WRAPPER_STRUCT );
+FOR_EACH_CALLBACK_FUNCTION( DEFINE_GVLCB_SKELETON );
+FOR_EACH_CALLBACK_FUNCTION( DEFINE_GVLCB_STUB );

data/ext/gvl_wrappers.h

@@ -0,0 +1,253 @@
+/*
+ * gvl_wrappers.h - Wrapper functions for locking/unlocking the Ruby GVL
+ *
+ * These are some obscure preprocessor directives that allow to generate
+ * drop-in replacement wrapper functions in a declarative manner.
+ * These wrapper functions ensure that ruby's GVL is released on each
+ * function call and reacquired at the end of the call or in callbacks.
+ * This way blocking functions calls don't block concurrent ruby threads.
+ *
+ * The wrapper of each function is prefixed by "gvl_".
+ *
+ * Use "gcc -E" to retrieve the generated code.
+ */
+
+#ifndef __gvl_wrappers_h
+#define __gvl_wrappers_h
+
+#if defined(HAVE_RB_THREAD_CALL_WITH_GVL)
+extern void *rb_thread_call_with_gvl(void *(*func)(void *), void *data1);
+#endif
+
+#if defined(HAVE_RB_THREAD_CALL_WITHOUT_GVL)
+extern void *rb_thread_call_without_gvl(void *(*func)(void *), void *data1,
+				 rb_unblock_function_t *ubf, void *data2);
+#endif
+
+#define DEFINE_PARAM_LIST1(type, name) \
+	name,
+
+#define DEFINE_PARAM_LIST2(type, name) \
+	p->params.name,
+
+#define DEFINE_PARAM_LIST3(type, name) \
+	type name,
+
+#define DEFINE_PARAM_DECL(type, name) \
+	type name;
+
+#define DEFINE_GVL_WRAPPER_STRUCT(name, when_non_void, rettype, lastparamtype, lastparamname) \
+	struct gvl_wrapper_##name##_params { \
+		struct { \
+			FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_DECL) \
+			lastparamtype lastparamname; \
+		} params; \
+		when_non_void( rettype retval; ) \
+	};
+
+#define DEFINE_GVL_SKELETON(name, when_non_void, rettype, lastparamtype, lastparamname) \
+	static void * gvl_##name##_skeleton( void *data ){ \
+		struct gvl_wrapper_##name##_params *p = (struct gvl_wrapper_##name##_params*)data; \
+		when_non_void( p->retval = ) \
+			name( FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST2) p->params.lastparamname ); \
+		return NULL; \
+	}
+
+#if defined(HAVE_RB_THREAD_CALL_WITHOUT_GVL)
+	#define DEFINE_GVL_STUB(name, when_non_void, rettype, lastparamtype, lastparamname) \
+		rettype gvl_##name(FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST3) lastparamtype lastparamname){ \
+			struct gvl_wrapper_##name##_params params = { \
+				{FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST1) lastparamname}, when_non_void((rettype)0) \
+			}; \
+			rb_thread_call_without_gvl(gvl_##name##_skeleton, &params, RUBY_UBF_IO, 0); \
+			when_non_void( return params.retval; ) \
+		}
+#else
+	#define DEFINE_GVL_STUB(name, when_non_void, rettype, lastparamtype, lastparamname) \
+		rettype gvl_##name(FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST3) lastparamtype lastparamname){ \
+			return name( FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST1) lastparamname ); \
+		}
+#endif
+
+#define DEFINE_GVL_STUB_DECL(name, when_non_void, rettype, lastparamtype, lastparamname) \
+	rettype gvl_##name(FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST3) lastparamtype lastparamname);
+
+#define DEFINE_GVLCB_SKELETON(name, when_non_void, rettype, lastparamtype, lastparamname) \
+	static void * gvl_##name##_skeleton( void *data ){ \
+		struct gvl_wrapper_##name##_params *p = (struct gvl_wrapper_##name##_params*)data; \
+		when_non_void( p->retval = ) \
+			name( FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST2) p->params.lastparamname ); \
+		return NULL; \
+	}
+
+#if defined(HAVE_RB_THREAD_CALL_WITH_GVL)
+	#define DEFINE_GVLCB_STUB(name, when_non_void, rettype, lastparamtype, lastparamname) \
+		rettype gvl_##name(FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST3) lastparamtype lastparamname){ \
+			struct gvl_wrapper_##name##_params params = { \
+				{FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST1) lastparamname}, when_non_void((rettype)0) \
+			}; \
+			rb_thread_call_with_gvl(gvl_##name##_skeleton, &params); \
+			when_non_void( return params.retval; ) \
+		}
+#else
+	#define DEFINE_GVLCB_STUB(name, when_non_void, rettype, lastparamtype, lastparamname) \
+		rettype gvl_##name(FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST3) lastparamtype lastparamname){ \
+			return name( FOR_EACH_PARAM_OF_##name(DEFINE_PARAM_LIST1) lastparamname ); \
+		}
+#endif
+
+#define GVL_TYPE_VOID(string)
+#define GVL_TYPE_NONVOID(string) string
+
+
+/*
+ * Definitions of blocking functions and their parameters
+ */
+
+#define FOR_EACH_PARAM_OF_PQconnectdb(param)
+
+#define FOR_EACH_PARAM_OF_PQconnectStart(param)
+
+#define FOR_EACH_PARAM_OF_PQconnectPoll(param)
+
+#define FOR_EACH_PARAM_OF_PQreset(param)
+
+#define FOR_EACH_PARAM_OF_PQresetStart(param)
+
+#define FOR_EACH_PARAM_OF_PQresetPoll(param)
+
+#define FOR_EACH_PARAM_OF_PQexec(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQexecParams(param) \
+	param(PGconn *, conn) \
+	param(const char *, command) \
+	param(int, nParams) \
+	param(const Oid *, paramTypes) \
+	param(const char * const *, paramValues) \
+	param(const int *, paramLengths) \
+	param(const int *, paramFormats)
+
+#define FOR_EACH_PARAM_OF_PQexecPrepared(param) \
+	param(PGconn *, conn) \
+	param(const char *, stmtName) \
+	param(int, nParams) \
+	param(const char * const *, paramValues) \
+	param(const int *, paramLengths) \
+	param(const int *, paramFormats)
+
+#define FOR_EACH_PARAM_OF_PQprepare(param) \
+	param(PGconn *, conn) \
+	param(const char *, stmtName) \
+	param(const char *, query) \
+	param(int, nParams)
+
+#define FOR_EACH_PARAM_OF_PQdescribePrepared(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQdescribePortal(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQgetResult(param)
+
+#define FOR_EACH_PARAM_OF_PQputCopyData(param) \
+	param(PGconn *, conn) \
+	param(const char *, buffer)
+
+#define FOR_EACH_PARAM_OF_PQputCopyEnd(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQgetCopyData(param) \
+	param(PGconn *, conn) \
+	param(char **, buffer)
+
+#define FOR_EACH_PARAM_OF_PQnotifies(param)
+
+#define FOR_EACH_PARAM_OF_PQsendQuery(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQsendQueryParams(param) \
+	param(PGconn *, conn) \
+	param(const char *, command) \
+	param(int, nParams) \
+	param(const Oid *, paramTypes) \
+	param(const char *const *, paramValues) \
+	param(const int *, paramLengths) \
+	param(const int *, paramFormats)
+
+#define FOR_EACH_PARAM_OF_PQsendPrepare(param) \
+	param(PGconn *, conn) \
+	param(const char *, stmtName) \
+	param(const char *, query) \
+	param(int, nParams)
+
+#define FOR_EACH_PARAM_OF_PQsendQueryPrepared(param) \
+	param(PGconn *, conn) \
+	param(const char *, stmtName) \
+	param(int, nParams) \
+	param(const char *const *, paramValues) \
+	param(const int *, paramLengths) \
+	param(const int *, paramFormats)
+
+#define FOR_EACH_PARAM_OF_PQsendDescribePrepared(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQsendDescribePortal(param) \
+	param(PGconn *, conn)
+
+#define FOR_EACH_PARAM_OF_PQisBusy(param)
+
+#define FOR_EACH_PARAM_OF_PQcancel(param) \
+	param(PGcancel *, cancel) \
+	param(char *, errbuf)
+
+/* function( name, void_or_nonvoid, returntype, lastparamtype, lastparamname ) */
+#define FOR_EACH_BLOCKING_FUNCTION(function) \
+	function(PQconnectdb, GVL_TYPE_NONVOID, PGconn *, const char *, conninfo) \
+	function(PQconnectStart, GVL_TYPE_NONVOID, PGconn *, const char *, conninfo) \
+	function(PQconnectPoll, GVL_TYPE_NONVOID, PostgresPollingStatusType, PGconn *, conn) \
+	function(PQreset, GVL_TYPE_VOID, void, PGconn *, conn) \
+	function(PQresetStart, GVL_TYPE_NONVOID, int, PGconn *, conn) \
+	function(PQresetPoll, GVL_TYPE_NONVOID, PostgresPollingStatusType, PGconn *, conn) \
+	function(PQexec, GVL_TYPE_NONVOID, PGresult *, const char *, command) \
+	function(PQexecParams, GVL_TYPE_NONVOID, PGresult *, int, resultFormat) \
+	function(PQexecPrepared, GVL_TYPE_NONVOID, PGresult *, int, resultFormat) \
+	function(PQprepare, GVL_TYPE_NONVOID, PGresult *, const Oid *, paramTypes) \
+	function(PQdescribePrepared, GVL_TYPE_NONVOID, PGresult *, const char *, stmtName) \
+	function(PQdescribePortal, GVL_TYPE_NONVOID, PGresult *, const char *, portalName) \
+	function(PQgetResult, GVL_TYPE_NONVOID, PGresult *, PGconn *, conn) \
+	function(PQputCopyData, GVL_TYPE_NONVOID, int, int, nbytes) \
+	function(PQputCopyEnd, GVL_TYPE_NONVOID, int, const char *, errormsg) \
+	function(PQgetCopyData, GVL_TYPE_NONVOID, int, int, async) \
+	function(PQnotifies, GVL_TYPE_NONVOID, PGnotify *, PGconn *, conn) \
+	function(PQsendQuery, GVL_TYPE_NONVOID, int, const char *, query) \
+	function(PQsendQueryParams, GVL_TYPE_NONVOID, int, int, resultFormat) \
+	function(PQsendPrepare, GVL_TYPE_NONVOID, int, const Oid *, paramTypes) \
+	function(PQsendQueryPrepared, GVL_TYPE_NONVOID, int, int, resultFormat) \
+	function(PQsendDescribePrepared, GVL_TYPE_NONVOID, int, const char *, stmt) \
+	function(PQsendDescribePortal, GVL_TYPE_NONVOID, int, const char *, portal) \
+	function(PQisBusy, GVL_TYPE_NONVOID, int, PGconn *, conn) \
+	function(PQcancel, GVL_TYPE_NONVOID, int, int, errbufsize);
+
+
+FOR_EACH_BLOCKING_FUNCTION( DEFINE_GVL_STUB_DECL );
+
+
+/*
+ * Definitions of callback functions and their parameters
+ */
+
+#define FOR_EACH_PARAM_OF_notice_processor_proxy(param) \
+	param(void *, arg)
+
+#define FOR_EACH_PARAM_OF_notice_receiver_proxy(param) \
+	param(void *, arg)
+
+/* function( name, void_or_nonvoid, returntype, lastparamtype, lastparamname ) */
+#define FOR_EACH_CALLBACK_FUNCTION(function) \
+	function(notice_processor_proxy, GVL_TYPE_VOID, void, const char *, message) \
+	function(notice_receiver_proxy, GVL_TYPE_VOID, void, const PGresult *, result) \
+
+FOR_EACH_CALLBACK_FUNCTION( DEFINE_GVL_STUB_DECL );
+
+#endif /* end __gvl_wrappers_h */

data/ext/pg.c

@@ -0,0 +1,545 @@
+/*
+ * pg.c - Toplevel extension
+ * $Id$
+ *
+ * Author/s:
+ *
+ * - Jeff Davis <ruby-pg@j-davis.com>
+ * - Guy Decoux (ts) <decoux@moulon.inra.fr>
+ * - Michael Granger <ged@FaerieMUD.org>
+ * - Dave Lee
+ * - Eiji Matsumoto <usagi@ruby.club.or.jp>
+ * - Yukihiro Matsumoto <matz@ruby-lang.org>
+ * - Noboru Saitou <noborus@netlab.jp>
+ *
+ * See Contributors.rdoc for the many additional fine people that have contributed
+ * to this library over the years.
+ *
+ * Copyright (c) 1997-2012 by the authors.
+ *
+ * You may redistribute this software under the same terms as Ruby itself; see
+ * http://www.ruby-lang.org/en/LICENSE.txt or the LICENSE file in the source
+ * for details.
+ *
+ * Portions of the code are from the PostgreSQL project, and are distributed
+ * under the terms of the PostgreSQL license, included in the file "POSTGRES".
+ *
+ * Portions copyright LAIKA, Inc.
+ *
+ *
+ * The following functions are part of libpq, but not available from ruby-pg,
+ * because they are deprecated, obsolete, or generally not useful:
+ *
+ * - PQfreemem -- unnecessary: copied to ruby object, then freed. Ruby object's
+ *                memory is freed when it is garbage collected.
+ * - PQbinaryTuples -- better to use PQfformat
+ * - PQprint -- not very useful
+ * - PQsetdb -- not very useful
+ * - PQoidStatus -- deprecated, use PQoidValue
+ * - PQrequestCancel -- deprecated, use PQcancel
+ * - PQfn -- use a prepared statement instead
+ * - PQgetline -- deprecated, use PQgetCopyData
+ * - PQgetlineAsync -- deprecated, use PQgetCopyData
+ * - PQputline -- deprecated, use PQputCopyData
+ * - PQputnbytes -- deprecated, use PQputCopyData
+ * - PQendcopy -- deprecated, use PQputCopyEnd
+ */
+
+#include "pg.h"
+
+VALUE rb_mPG;
+VALUE rb_mPGconstants;
+
+
+/*
+ * Document-class: PG::Error
+ *
+ * This is the exception class raised when an error is returned from
+ * a libpq API call.
+ *
+ * The attributes +connection+ and +result+ are set to the connection
+ * object and result set object, respectively.
+ *
+ * If the connection object or result set object is not available from
+ * the context in which the error was encountered, it is +nil+.
+ */
+
+/*
+ * M17n functions
+ */
+
+#ifdef M17N_SUPPORTED
+/**
+ * The mapping from canonical encoding names in PostgreSQL to ones in Ruby.
+ */
+const char * const (pg_enc_pg2ruby_mapping[][2]) = {
+	{"BIG5",          "Big5"        },
+	{"EUC_CN",        "GB2312"      },
+	{"EUC_JP",        "EUC-JP"      },
+	{"EUC_JIS_2004",  "EUC-JP"      },
+	{"EUC_KR",        "EUC-KR"      },
+	{"EUC_TW",        "EUC-TW"      },
+	{"GB18030",       "GB18030"     },
+	{"GBK",           "GBK"         },
+	{"ISO_8859_5",    "ISO-8859-5"  },
+	{"ISO_8859_6",    "ISO-8859-6"  },
+	{"ISO_8859_7",    "ISO-8859-7"  },
+	{"ISO_8859_8",    "ISO-8859-8"  },
+	/* {"JOHAB",         "JOHAB"       }, dummy */
+	{"KOI8",          "KOI8-R"      },
+	{"KOI8R",         "KOI8-R"      },
+	{"KOI8U",         "KOI8-U"      },
+	{"LATIN1",        "ISO-8859-1"  },
+	{"LATIN2",        "ISO-8859-2"  },
+	{"LATIN3",        "ISO-8859-3"  },
+	{"LATIN4",        "ISO-8859-4"  },
+	{"LATIN5",        "ISO-8859-9"  },
+	{"LATIN6",        "ISO-8859-10" },
+	{"LATIN7",        "ISO-8859-13" },
+	{"LATIN8",        "ISO-8859-14" },
+	{"LATIN9",        "ISO-8859-15" },
+	{"LATIN10",       "ISO-8859-16" },
+	{"MULE_INTERNAL", "Emacs-Mule"  },
+	{"SJIS",          "Windows-31J" },
+	{"SHIFT_JIS_2004","Windows-31J" },
+	/* {"SQL_ASCII",     NULL          },  special case*/
+	{"UHC",           "CP949"       },
+	{"UTF8",          "UTF-8"       },
+	{"WIN866",        "IBM866"      },
+	{"WIN874",        "Windows-874" },
+	{"WIN1250",       "Windows-1250"},
+	{"WIN1251",       "Windows-1251"},
+	{"WIN1252",       "Windows-1252"},
+	{"WIN1253",       "Windows-1253"},
+	{"WIN1254",       "Windows-1254"},
+	{"WIN1255",       "Windows-1255"},
+	{"WIN1256",       "Windows-1256"},
+	{"WIN1257",       "Windows-1257"},
+	{"WIN1258",       "Windows-1258"}
+};
+
+
+/*
+ * A cache of mapping from PostgreSQL's encoding indices to Ruby's rb_encoding*s.
+ */
+static struct st_table *enc_pg2ruby;
+static ID s_id_index;
+
+
+/*
+ * Get the index of encoding +val+.
+ * :FIXME: Look into replacing this with rb_enc_get_index() since 1.9.1 isn't really
+ * used anymore.
+ */
+int
+pg_enc_get_index(VALUE val)
+{
+	int i = ENCODING_GET_INLINED(val);
+	if (i == ENCODING_INLINE_MAX) {
+		VALUE iv = rb_ivar_get(val, s_id_index);
+		i = NUM2INT(iv);
+	}
+	return i;
+}
+
+
+/*
+ * Look up the JOHAB encoding, creating it as a dummy encoding if it's not
+ * already defined.
+ */
+static rb_encoding *
+pg_find_or_create_johab(void)
+{
+	static const char * const aliases[] = { "JOHAB", "Windows-1361", "CP1361" };
+	int enc_index;
+	size_t i;
+
+	for (i = 0; i < sizeof(aliases)/sizeof(aliases[0]); ++i) {
+		enc_index = rb_enc_find_index(aliases[i]);
+		if (enc_index > 0) return rb_enc_from_index(enc_index);
+	}
+
+	enc_index = rb_define_dummy_encoding(aliases[0]);
+	for (i = 1; i < sizeof(aliases)/sizeof(aliases[0]); ++i) {
+		ENC_ALIAS(aliases[i], aliases[0]);
+	}
+	return rb_enc_from_index(enc_index);
+}
+
+/*
+ * Return the given PostgreSQL encoding ID as an rb_encoding.
+ *
+ * - returns NULL if the client encoding is 'SQL_ASCII'.
+ * - returns ASCII-8BIT if the client encoding is unknown.
+ */
+rb_encoding *
+pg_get_pg_encoding_as_rb_encoding( int enc_id )
+{
+	rb_encoding *enc;
+
+	/* Use the cached value if it exists */
+	if ( st_lookup(enc_pg2ruby, (st_data_t)enc_id, (st_data_t*)&enc) ) {
+		return enc;
+	}
+	else {
+		const char *name = pg_encoding_to_char( enc_id );
+
+		enc = pg_get_pg_encname_as_rb_encoding( name );
+		st_insert( enc_pg2ruby, (st_data_t)enc_id, (st_data_t)enc );
+
+		return enc;
+	}
+
+}
+
+/*
+ * Return the given PostgreSQL encoding name as an rb_encoding.
+ */
+rb_encoding *
+pg_get_pg_encname_as_rb_encoding( const char *pg_encname )
+{
+	size_t i;
+
+	/* Trying looking it up in the conversion table */
+	for ( i = 0; i < sizeof(pg_enc_pg2ruby_mapping)/sizeof(pg_enc_pg2ruby_mapping[0]); ++i ) {
+		if ( strcmp(pg_encname, pg_enc_pg2ruby_mapping[i][0]) == 0 )
+			return rb_enc_find( pg_enc_pg2ruby_mapping[i][1] );
+	}
+
+	/* JOHAB isn't a builtin encoding, so make up a dummy encoding if it's seen */
+	if ( strncmp(pg_encname, "JOHAB", 5) == 0 )
+		return pg_find_or_create_johab();
+
+	/* Fallthrough to ASCII-8BIT */
+	return rb_ascii8bit_encoding();
+}
+
+/*
+ * Get the client encoding of the specified connection handle and return it as a rb_encoding.
+ */
+rb_encoding *
+pg_conn_enc_get( PGconn *conn )
+{
+	int enc_id = PQclientEncoding( conn );
+	return pg_get_pg_encoding_as_rb_encoding( enc_id );
+}
+
+
+/*
+ * Returns the given rb_encoding as the equivalent PostgreSQL encoding string.
+ */
+const char *
+pg_get_rb_encoding_as_pg_encoding( rb_encoding *enc )
+{
+	const char *rb_encname = rb_enc_name( enc );
+	const char *encname = NULL;
+	size_t i;
+
+	for (i = 0; i < sizeof(pg_enc_pg2ruby_mapping)/sizeof(pg_enc_pg2ruby_mapping[0]); ++i) {
+		if (strcmp(rb_encname, pg_enc_pg2ruby_mapping[i][1]) == 0) {
+			encname = pg_enc_pg2ruby_mapping[i][0];
+		}
+	}
+
+	if ( !encname ) encname = "SQL_ASCII";
+
+	return encname;
+}
+
+#endif /* M17N_SUPPORTED */
+
+
+/**************************************************************************
+ * Module Methods
+ **************************************************************************/
+
+#ifdef HAVE_PQLIBVERSION
+/*
+ * call-seq:
+ *   PG.library_version -> Integer
+ *
+ * Get the version of the libpq library in use. The number is formed by
+ * converting the major, minor, and revision numbers into two-decimal-
+ * digit numbers and appending them together.
+ * For example, version 7.4.2 will be returned as 70402, and version
+ * 8.1 will be returned as 80100 (leading zeroes are not shown). Zero
+ * is returned if the connection is bad.
+ */
+static VALUE
+pg_s_library_version(VALUE self)
+{
+	UNUSED( self );
+	return INT2NUM(PQlibVersion());
+}
+#endif
+
+
+/*
+ * call-seq:
+ *    PG.isthreadsafe            -> Boolean
+ *    PG.is_threadsafe?          -> Boolean
+ *    PG.threadsafe?             -> Boolean
+ *
+ * Returns +true+ if libpq is thread-safe, +false+ otherwise.
+ */
+static VALUE
+pg_s_threadsafe_p(VALUE self)
+{
+	UNUSED( self );
+	return PQisthreadsafe() ? Qtrue : Qfalse;
+}
+
+
+
+/**************************************************************************
+ * Initializer
+ **************************************************************************/
+
+void
+Init_pg_ext()
+{
+	rb_mPG = rb_define_module( "PG" );
+	rb_mPGconstants = rb_define_module_under( rb_mPG, "Constants" );
+
+	/*************************
+	 *  PG module methods
+	 *************************/
+#ifdef HAVE_PQLIBVERSION
+	rb_define_singleton_method( rb_mPG, "library_version", pg_s_library_version, 0 );
+#endif
+	rb_define_singleton_method( rb_mPG, "isthreadsafe", pg_s_threadsafe_p, 0 );
+	SINGLETON_ALIAS( rb_mPG, "is_threadsafe?", "isthreadsafe" );
+	SINGLETON_ALIAS( rb_mPG, "threadsafe?", "isthreadsafe" );
+
+	/******     PG::Connection CLASS CONSTANTS: Connection Status     ******/
+
+	/* Connection succeeded */
+	rb_define_const(rb_mPGconstants, "CONNECTION_OK", INT2FIX(CONNECTION_OK));
+	/* Connection failed */
+	rb_define_const(rb_mPGconstants, "CONNECTION_BAD", INT2FIX(CONNECTION_BAD));
+
+	/******     PG::Connection CLASS CONSTANTS: Nonblocking connection status     ******/
+
+	/* Waiting for connection to be made. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_STARTED", INT2FIX(CONNECTION_STARTED));
+	/* Connection OK; waiting to send. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_MADE", INT2FIX(CONNECTION_MADE));
+	/* Waiting for a response from the server. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_AWAITING_RESPONSE", INT2FIX(CONNECTION_AWAITING_RESPONSE));
+	/* Received authentication; waiting for backend start-up to finish. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_AUTH_OK", INT2FIX(CONNECTION_AUTH_OK));
+	/* Negotiating SSL encryption. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_SSL_STARTUP", INT2FIX(CONNECTION_SSL_STARTUP));
+	/* Negotiating environment-driven parameter settings. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_SETENV", INT2FIX(CONNECTION_SETENV));
+	/* Internal state: connect() needed. */
+	rb_define_const(rb_mPGconstants, "CONNECTION_NEEDED", INT2FIX(CONNECTION_NEEDED));
+
+	/******     PG::Connection CLASS CONSTANTS: Nonblocking connection polling status     ******/
+
+	/* Async connection is waiting to read */
+	rb_define_const(rb_mPGconstants, "PGRES_POLLING_READING", INT2FIX(PGRES_POLLING_READING));
+	/* Async connection is waiting to write */
+	rb_define_const(rb_mPGconstants, "PGRES_POLLING_WRITING", INT2FIX(PGRES_POLLING_WRITING));
+	/* Async connection failed or was reset */
+	rb_define_const(rb_mPGconstants, "PGRES_POLLING_FAILED", INT2FIX(PGRES_POLLING_FAILED));
+	/* Async connection succeeded */
+	rb_define_const(rb_mPGconstants, "PGRES_POLLING_OK", INT2FIX(PGRES_POLLING_OK));
+
+	/******     PG::Connection CLASS CONSTANTS: Transaction Status     ******/
+
+	/* Transaction is currently idle (#transaction_status) */
+	rb_define_const(rb_mPGconstants, "PQTRANS_IDLE", INT2FIX(PQTRANS_IDLE));
+	/* Transaction is currently active; query has been sent to the server, but not yet completed. (#transaction_status) */
+	rb_define_const(rb_mPGconstants, "PQTRANS_ACTIVE", INT2FIX(PQTRANS_ACTIVE));
+	/* Transaction is currently idle, in a valid transaction block (#transaction_status) */
+	rb_define_const(rb_mPGconstants, "PQTRANS_INTRANS", INT2FIX(PQTRANS_INTRANS));
+	/* Transaction is currently idle, in a failed transaction block (#transaction_status) */
+	rb_define_const(rb_mPGconstants, "PQTRANS_INERROR", INT2FIX(PQTRANS_INERROR));
+	/* Transaction's connection is bad (#transaction_status) */
+	rb_define_const(rb_mPGconstants, "PQTRANS_UNKNOWN", INT2FIX(PQTRANS_UNKNOWN));
+
+	/******     PG::Connection CLASS CONSTANTS: Error Verbosity     ******/
+
+	/* Terse error verbosity level (#set_error_verbosity) */
+	rb_define_const(rb_mPGconstants, "PQERRORS_TERSE", INT2FIX(PQERRORS_TERSE));
+	/* Default error verbosity level (#set_error_verbosity) */
+	rb_define_const(rb_mPGconstants, "PQERRORS_DEFAULT", INT2FIX(PQERRORS_DEFAULT));
+	/* Verbose error verbosity level (#set_error_verbosity) */
+	rb_define_const(rb_mPGconstants, "PQERRORS_VERBOSE", INT2FIX(PQERRORS_VERBOSE));
+
+#ifdef HAVE_PQPING
+	/******     PG::Connection CLASS CONSTANTS: Check Server Status ******/
+
+	/* Server is accepting connections. */
+	rb_define_const(rb_mPGconstants, "PQPING_OK", INT2FIX(PQPING_OK));
+	/* Server is alive but rejecting connections. */
+	rb_define_const(rb_mPGconstants, "PQPING_REJECT", INT2FIX(PQPING_REJECT));
+	/* Could not establish connection. */
+	rb_define_const(rb_mPGconstants, "PQPING_NO_RESPONSE", INT2FIX(PQPING_NO_RESPONSE));
+	/* Connection not attempted (bad params). */
+	rb_define_const(rb_mPGconstants, "PQPING_NO_ATTEMPT", INT2FIX(PQPING_NO_ATTEMPT));
+#endif
+
+	/******     PG::Connection CLASS CONSTANTS: Large Objects     ******/
+
+	/* Flag for #lo_creat, #lo_open -- open for writing */
+	rb_define_const(rb_mPGconstants, "INV_WRITE", INT2FIX(INV_WRITE));
+	/* Flag for #lo_creat, #lo_open -- open for reading */
+	rb_define_const(rb_mPGconstants, "INV_READ", INT2FIX(INV_READ));
+	/* Flag for #lo_lseek -- seek from object start */
+	rb_define_const(rb_mPGconstants, "SEEK_SET", INT2FIX(SEEK_SET));
+	/* Flag for #lo_lseek -- seek from current position */
+	rb_define_const(rb_mPGconstants, "SEEK_CUR", INT2FIX(SEEK_CUR));
+	/* Flag for #lo_lseek -- seek from object end */
+	rb_define_const(rb_mPGconstants, "SEEK_END", INT2FIX(SEEK_END));
+
+	/******     PG::Result CONSTANTS: result status      ******/
+
+	/* #result_status constant: The string sent to the server was empty. */
+	rb_define_const(rb_mPGconstants, "PGRES_EMPTY_QUERY", INT2FIX(PGRES_EMPTY_QUERY));
+	/* #result_status constant: Successful completion of a command returning no data. */
+	rb_define_const(rb_mPGconstants, "PGRES_COMMAND_OK", INT2FIX(PGRES_COMMAND_OK));
+		/* #result_status constant: Successful completion of a command returning data
+	   (such as a SELECT or SHOW). */
+	rb_define_const(rb_mPGconstants, "PGRES_TUPLES_OK", INT2FIX(PGRES_TUPLES_OK));
+	/* #result_status constant: Copy Out (from server) data transfer started. */
+	rb_define_const(rb_mPGconstants, "PGRES_COPY_OUT", INT2FIX(PGRES_COPY_OUT));
+	/* #result_status constant: Copy In (to server) data transfer started. */
+	rb_define_const(rb_mPGconstants, "PGRES_COPY_IN", INT2FIX(PGRES_COPY_IN));
+	/* #result_status constant: The server’s response was not understood. */
+	rb_define_const(rb_mPGconstants, "PGRES_BAD_RESPONSE", INT2FIX(PGRES_BAD_RESPONSE));
+	/* #result_status constant: A nonfatal error (a notice or warning) occurred. */
+	rb_define_const(rb_mPGconstants, "PGRES_NONFATAL_ERROR",INT2FIX(PGRES_NONFATAL_ERROR));
+	/* #result_status constant: A fatal error occurred. */
+	rb_define_const(rb_mPGconstants, "PGRES_FATAL_ERROR", INT2FIX(PGRES_FATAL_ERROR));
+	/* #result_status constant: Copy In/Out data transfer in progress. */
+#ifdef HAVE_CONST_PGRES_COPY_BOTH
+	rb_define_const(rb_mPGconstants, "PGRES_COPY_BOTH", INT2FIX(PGRES_COPY_BOTH));
+#endif
+	/* #result_status constant: Single tuple from larger resultset. */
+#ifdef HAVE_CONST_PGRES_SINGLE_TUPLE
+	rb_define_const(rb_mPGconstants, "PGRES_SINGLE_TUPLE", INT2FIX(PGRES_SINGLE_TUPLE));
+#endif
+
+	/******     Result CONSTANTS: result error field codes      ******/
+
+	/* #result_error_field argument constant: The severity; the field contents
+	 * are ERROR, FATAL, or PANIC (in an error message), or WARNING, NOTICE,
+	 * DEBUG, INFO, or LOG (in a notice message), or a localized translation
+	 * of one of these. Always present.
+	 */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SEVERITY", INT2FIX(PG_DIAG_SEVERITY));
+
+	/* #result_error_field argument constant: The SQLSTATE code for the
+	 * error. The SQLSTATE code identies the type of error that has occurred;
+	 * it can be used by front-end applications to perform specic operations
+	 * (such as er- ror handling) in response to a particular database
+	 * error. For a list of the possible SQLSTATE codes, see Appendix A.
+	 * This eld is not localizable, and is always present.
+	 */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SQLSTATE", INT2FIX(PG_DIAG_SQLSTATE));
+
+	/* #result_error_field argument constant: The primary human-readable
+	 * error message (typically one line). Always present. */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_MESSAGE_PRIMARY", INT2FIX(PG_DIAG_MESSAGE_PRIMARY));
+
+	/* #result_error_field argument constant: Detail: an optional secondary
+	 * error message carrying more detail about the problem. Might run to
+	 * multiple lines.
+	 */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_MESSAGE_DETAIL", INT2FIX(PG_DIAG_MESSAGE_DETAIL));
+
+	/* #result_error_field argument constant: Hint: an optional suggestion
+	 * what to do about the problem. This is intended to differ from detail
+	 * in that it offers advice (potentially inappropriate) rather than
+	 * hard facts. Might run to multiple lines.
+	 */
+
+	rb_define_const(rb_mPGconstants, "PG_DIAG_MESSAGE_HINT", INT2FIX(PG_DIAG_MESSAGE_HINT));
+	/* #result_error_field argument constant: A string containing a decimal
+	 * integer indicating an error cursor position as an index into the
+	 * original statement string. The rst character has index 1, and
+	 * positions are measured in characters not bytes.
+	 */
+
+	rb_define_const(rb_mPGconstants, "PG_DIAG_STATEMENT_POSITION", INT2FIX(PG_DIAG_STATEMENT_POSITION));
+	/* #result_error_field argument constant: This is dened the same as
+	 * the PG_DIAG_STATEMENT_POSITION eld, but it is used when the cursor
+	 * position refers to an internally generated command rather than the
+	 * one submitted by the client. The PG_DIAG_INTERNAL_QUERY eld will
+	 * always appear when this eld appears.
+	 */
+
+	rb_define_const(rb_mPGconstants, "PG_DIAG_INTERNAL_POSITION", INT2FIX(PG_DIAG_INTERNAL_POSITION));
+	/* #result_error_field argument constant: The text of a failed
+	 * internally-generated command. This could be, for example, a SQL
+	 * query issued by a PL/pgSQL function.
+	 */
+
+	rb_define_const(rb_mPGconstants, "PG_DIAG_INTERNAL_QUERY", INT2FIX(PG_DIAG_INTERNAL_QUERY));
+	/* #result_error_field argument constant: An indication of the context
+	 * in which the error occurred. Presently this includes a call stack
+	 * traceback of active procedural language functions and internally-generated
+	 * queries. The trace is one entry per line, most recent rst.
+	 */
+
+	rb_define_const(rb_mPGconstants, "PG_DIAG_CONTEXT", INT2FIX(PG_DIAG_CONTEXT));
+	/* #result_error_field argument constant: The le name of the source-code
+	 * location where the error was reported. */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SOURCE_FILE", INT2FIX(PG_DIAG_SOURCE_FILE));
+
+	/* #result_error_field argument constant: The line number of the
+	 * source-code location where the error was reported. */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SOURCE_LINE", INT2FIX(PG_DIAG_SOURCE_LINE));
+
+	/* #result_error_field argument constant: The name of the source-code
+	 * function reporting the error. */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SOURCE_FUNCTION", INT2FIX(PG_DIAG_SOURCE_FUNCTION));
+
+#ifdef HAVE_CONST_PG_DIAG_TABLE_NAME
+	/* #result_error_field argument constant: If the error was associated with a
+	 * specific database object, the name of the schema containing that object, if any. */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_SCHEMA_NAME", INT2FIX(PG_DIAG_SCHEMA_NAME));
+
+	/* #result_error_field argument constant: If the error was associated with a
+	 *specific table, the name of the table. (When this field is present, the schema name
+	 * field provides the name of the table's schema.) */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_TABLE_NAME", INT2FIX(PG_DIAG_TABLE_NAME));
+
+	/* #result_error_field argument constant: If the error was associated with a
+	 * specific table column, the name of the column. (When this field is present, the
+	 * schema and table name fields identify the table.) */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_COLUMN_NAME", INT2FIX(PG_DIAG_COLUMN_NAME));
+
+	/* #result_error_field argument constant: If the error was associated with a
+	 * specific datatype, the name of the datatype. (When this field is present, the
+	 * schema name field provides the name of the datatype's schema.) */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_DATATYPE_NAME", INT2FIX(PG_DIAG_DATATYPE_NAME));
+
+	/* #result_error_field argument constant: If the error was associated with a
+	 * specific constraint, the name of the constraint. The table or domain that the
+	 * constraint belongs to is reported using the fields listed above. (For this
+	 * purpose, indexes are treated as constraints, even if they weren't created with
+	 * constraint syntax.) */
+	rb_define_const(rb_mPGconstants, "PG_DIAG_CONSTRAINT_NAME", INT2FIX(PG_DIAG_CONSTRAINT_NAME));
+#endif
+
+	/* Invalid OID constant */
+	rb_define_const(rb_mPGconstants, "INVALID_OID", INT2FIX(InvalidOid));
+	rb_define_const(rb_mPGconstants, "InvalidOid", INT2FIX(InvalidOid));
+
+	/* Add the constants to the toplevel namespace */
+	rb_include_module( rb_mPG, rb_mPGconstants );
+
+#ifdef M17N_SUPPORTED
+	enc_pg2ruby = st_init_numtable();
+	s_id_index = rb_intern("@encoding");
+#endif
+
+	/* Initialize the main extension classes */
+	init_pg_connection();
+	init_pg_result();
+	init_pg_errors();
+}
+