check pg_pathman's Pl/PgSQL frontend for compatibility with pg_pathman's shared library

funbringer · funbringer · commit e31e4b5e3845 · 2016-12-01T17:25:53.000+03:00
diff --git a/init.sql b/init.sql
@@ -759,6 +759,10 @@ CREATE OR REPLACE FUNCTION @extschema@.debug_capture()
 RETURNS VOID AS 'pg_pathman', 'debug_capture'
 LANGUAGE C STRICT;
 
+CREATE OR REPLACE FUNCTION @extschema@.get_pathman_lib_version()
+RETURNS CSTRING AS 'pg_pathman', 'get_pathman_lib_version'
+LANGUAGE C STRICT;
+
 
 /*
  * Invoke init_callback on RANGE partition.
diff --git a/pg_pathman--1.1--1.2.sql b/pg_pathman--1.1--1.2.sql
@@ -1281,6 +1281,11 @@ BEFORE INSERT OR UPDATE OR DELETE ON @extschema@.pathman_config_params
 FOR EACH ROW EXECUTE PROCEDURE @extschema@.pathman_config_params_trigger_func();
 
 
+CREATE OR REPLACE FUNCTION @extschema@.get_pathman_lib_version()
+RETURNS CSTRING AS 'pg_pathman', 'get_pathman_lib_version'
+LANGUAGE C STRICT;
+
+
 /* ------------------------------------------------------------------------
  * Alter tables
  * ----------------------------------------------------------------------*/
diff --git a/src/init.c b/src/init.c
@@ -22,6 +22,7 @@
 #include "access/sysattr.h"
 #include "catalog/indexing.h"
 #include "catalog/pg_constraint.h"
+#include "catalog/pg_extension.h"
 #include "catalog/pg_inherits.h"
 #include "catalog/pg_inherits_fn.h"
 #include "catalog/pg_type.h"
@@ -62,6 +63,7 @@ PathmanInitState 	pg_pathman_init_state;
 /* Shall we install new relcache callback? */
 static bool			relcache_callback_needed = true;
 
+
 /* Functions for various local caches */
 static bool init_pathman_relation_oids(void);
 static void fini_pathman_relation_oids(void);
@@ -89,6 +91,11 @@ static bool read_opexpr_const(const OpExpr *opexpr,
 static int oid_cmp(const void *p1, const void *p2);
 
 
+/* Validate SQL facade */
+static uint32 build_sql_facade_version(char *version_cstr);
+static uint32 get_sql_facade_version(void);
+static void validate_sql_facade_version(uint32 ver);
+
 /*
  * Save and restore main init state.
  */
@@ -167,6 +174,9 @@ load_config(void)
 	if (!init_pathman_relation_oids())
 		return false; /* remain 'uninitialized', exit before creating main caches */
 
+	/* Validate pg_pathman's Pl/PgSQL facade (might be outdated) */
+	validate_sql_facade_version(get_sql_facade_version());
+
 	init_local_cache();		/* create 'partitioned_rels' hash table */
 	read_pathman_config();	/* read PATHMAN_CONFIG table & fill cache */
 
@@ -1098,3 +1108,89 @@ oid_cmp(const void *p1, const void *p2)
 		return 1;
 	return 0;
 }
+
+
+/* Parse cstring and build uint32 representing the version */
+static uint32
+build_sql_facade_version(char *version_cstr)
+{
+	uint32	version;
+
+	/* expect to see x+.y+.z+ */
+	version = strtol(version_cstr, &version_cstr, 10) & 0xFF;
+
+	version <<= 8;
+	if (strlen(version_cstr) > 1)
+		version |= (strtol(version_cstr + 1, &version_cstr, 10) & 0xFF);
+
+	version <<= 8;
+	if (strlen(version_cstr) > 1)
+		version |= (strtol(version_cstr + 1, &version_cstr, 10) & 0xFF);
+
+	return version;
+}
+
+/* Get version of pg_pathman's facade written in Pl/PgSQL */
+static uint32
+get_sql_facade_version(void)
+{
+	Relation		pg_extension_rel;
+	ScanKeyData		skey;
+	SysScanDesc		scan;
+	HeapTuple		htup;
+
+	Datum			datum;
+	bool			isnull;
+	char		   *version_cstr;
+
+	/* Look up the extension */
+	pg_extension_rel = heap_open(ExtensionRelationId, AccessShareLock);
+
+	ScanKeyInit(&skey,
+				Anum_pg_extension_extname,
+				BTEqualStrategyNumber, F_NAMEEQ,
+				CStringGetDatum("pg_pathman"));
+
+	scan = systable_beginscan(pg_extension_rel,
+							  ExtensionNameIndexId,
+							  true, NULL, 1, &skey);
+
+	htup = systable_getnext(scan);
+
+	/* Exit if pg_pathman's missing */
+	if (!HeapTupleIsValid(htup))
+		return 0;
+
+	datum = heap_getattr(htup, Anum_pg_extension_extversion,
+						 RelationGetDescr(pg_extension_rel), &isnull);
+	Assert(isnull == false); /* extversion should not be NULL */
+
+	/* Extract pg_pathman's version as cstring */
+	version_cstr = text_to_cstring(DatumGetTextPP(datum));
+
+	systable_endscan(scan);
+	heap_close(pg_extension_rel, AccessShareLock);
+
+	return build_sql_facade_version(version_cstr);
+}
+
+/* Check that current Pl/PgSQL facade is compatible with internals */
+static void
+validate_sql_facade_version(uint32 ver)
+{
+	Assert(ver > 0);
+
+	/* Compare ver to 'lowest compatible frontend' version */
+	if (ver < LOWEST_COMPATIBLE_FRONT)
+	{
+		elog(DEBUG1, "current version: %x, lowest compatible: %x",
+					 ver, LOWEST_COMPATIBLE_FRONT);
+
+		DisablePathman(); /* disable pg_pathman since config is broken */
+		ereport(ERROR,
+				(errmsg("pg_pathman's Pl/PgSQL frontend is incompatible with "
+						"its shared library"),
+				 errdetail("consider performing an update procedure"),
+				 errhint(INIT_ERROR_HINT)));
+	}
+}
diff --git a/src/init.h b/src/init.h
@@ -95,6 +95,11 @@ extern PathmanInitState 	pg_pathman_init_state;
 #define DEFAULT_SPAWN_USING_BGW		false
 
 
+/* Lowest version of Pl/PgSQL frontend compatible with internals (0xAA_BB_CC) */
+#define LOWEST_COMPATIBLE_FRONT		0x010200
+#define CURRENT_LIB_VERSION			0x010200
+
+
 /*
  * Save and restore PathmanInitState.
  */
diff --git a/src/pl_funcs.c b/src/pl_funcs.c
@@ -67,6 +67,7 @@ PG_FUNCTION_INFO_V1( invoke_on_partition_created_callback );
 PG_FUNCTION_INFO_V1( check_security_policy );
 
 PG_FUNCTION_INFO_V1( debug_capture );
+PG_FUNCTION_INFO_V1( get_pathman_lib_version );
 
 
 /*
@@ -913,3 +914,12 @@ debug_capture(PG_FUNCTION_ARGS)
 
 	PG_RETURN_VOID();
 }
+
+/*
+ * NOTE: just in case.
+ */
+Datum
+get_pathman_lib_version(PG_FUNCTION_ARGS)
+{
+	PG_RETURN_CSTRING(psprintf("%x", CURRENT_LIB_VERSION));
+}
