prestogres 0.1.0

data/pgpool2/pool_passwd.h

@@ -0,0 +1,41 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL 
+ * written by Tatsuo Ishii
+ *
+ * Copyright (c) 2003-2010	PgPool Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ * pool_passwd.h: pool_passwd related definitions.
+ *
+ */
+
+#ifndef POOL_PASSWD_H
+#define POOL_PASSWD_H
+
+#include "pool.h"
+
+#define POOL_PASSWD_FILENAME "pool_passwd"
+#define POOL_PASSWD_LEN 35
+
+extern void pool_init_pool_passwd(char *pool_passwd_filename);
+extern int pool_create_passwdent(char *username, char *passwd);
+extern char *pool_get_passwd(char *username);
+extern void pool_delete_passwdent(char *username);
+extern void pool_finish_pool_passwd(void);
+extern void pool_reopen_passwd_file(void);
+
+#endif /* POOL_PASSWD_H */

data/pgpool2/pool_path.c

@@ -0,0 +1,193 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL
+ * written by Tatsuo Ishii
+ *
+ * Portions Copyright (c) 2003-2008,	PgPool Global Development Group
+ * Portions Copyright (c) 2004, PostgreSQL Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ * pool_path.c.: small functions to manipulate paths
+ *
+ */
+
+#include "pool_type.h"
+#include "pool_path.h"
+#include <stdio.h>
+#include <string.h>
+
+static void trim_directory(char *path);
+static void trim_trailing_separator(char *path);
+
+/*
+ * get_parent_directory
+ *
+ * Modify the given string in-place to name the parent directory of the
+ * named file.
+ */
+void get_parent_directory(char *path)
+{
+	trim_directory(path);
+}
+
+
+/*
+ *  trim_directory
+ *
+ *  Trim trailing directory from path, that is, remove any trailing slashes,
+ *  the last pathname component, and the slash just ahead of it --- but never
+ *  remove a leading slash.
+ */
+static void trim_directory(char *path)
+{
+	char *p;
+
+	if (path[0] == '\0')
+		return;
+
+	/* back up over trailing slash(es) */
+	for (p = path + strlen(path) - 1; IS_DIR_SEP(*p) && p > path; p--);
+
+	/* back up over directory name */
+	for (; !IS_DIR_SEP(*p) && p > path; p--);
+
+	/* if multiple slashes before directory name, remove 'em all */
+	for (; p > path && IS_DIR_SEP(*(p - 1)); p--);
+
+	/* don't erase a leading slash */
+	if (p == path && IS_DIR_SEP(*p))
+		p++;
+
+	*p = '\0';
+}
+
+
+/*
+ * join_path_components - join two path components, inserting a slash
+ *
+ * ret_path is the output area (must be of size MAXPGPATH)
+ *
+ * ret_path can be the same as head, but not the same as tail.
+ */
+void join_path_components(char *ret_path, const char *head, const char *tail)
+{
+	if (ret_path != head)
+		StrNCpy(ret_path, head, MAXPGPATH);
+
+	/*
+	 * Remove any leading "." and ".." in the tail component,
+	 * adjusting head as needed.
+	 */
+	for (;;)
+	{
+		if (tail[0] == '.' && IS_DIR_SEP(tail[1]))
+		{
+			tail += 2;
+		}
+		else if (tail[0] == '.' && tail[1] == '\0')
+		{
+			tail += 1;
+			break;
+		}
+		else if (tail[0] == '.' && tail[1] == '.' && IS_DIR_SEP(tail[2]))
+		{
+			trim_directory(ret_path);
+			tail += 3;
+		}
+		else if (tail[0] == '.' && tail[1] == '.' && tail[2] == '\0')
+		{
+			trim_directory(ret_path);
+			tail += 2;
+			break;
+		}
+		else
+			break;
+	}
+
+	if (*tail)
+		snprintf(ret_path + strlen(ret_path), MAXPGPATH - strlen(ret_path), "/%s", tail);
+}
+
+
+/*
+ *  Clean up path by:
+ *      o  remove trailing slash
+ *      o  remove duplicate adjacent separators
+ *      o  remove trailing '.'
+ *      o  process trailing '..' ourselves
+ */
+void canonicalize_path(char *path)
+{
+	char *p, *to_p;
+	bool was_sep = false;
+
+	/*
+	 * Removing the trailing slash on a path means we never get ugly
+	 * double trailing slashes.
+	 */
+	trim_trailing_separator(path);
+
+	/*
+	 *  Remove duplicate adjacent separators
+	 */
+	p = path;
+	to_p = p;
+	for (; *p; p++, to_p++)
+	{
+		/* Handle many adjacent slashes, like "/a///b" */
+		while (*p == '/' && was_sep)
+			p++;
+		if (to_p != p)
+			*to_p = *p;
+		was_sep = (*p == '/');
+	}
+	*to_p = '\0';
+
+	/*
+	 * Remove any trailing uses of "." and process ".." ourselves
+	 */
+	for (;;)
+	{
+		int len = strlen(path);
+
+		if (len > 2 && strcmp(path + len - 2, "/.") == 0)
+			trim_directory(path);
+		else if (len > 3 && strcmp(path + len - 3, "/..") == 0)
+		{
+			trim_directory(path);
+			trim_directory(path);
+			/* remove directory above */
+		}
+		else
+			break;
+	}
+}
+
+
+/*
+ *  trim_trailing_separator
+ *
+ * trim off trailing slashes, but not a leading slash
+ */
+static void trim_trailing_separator(char *path)
+{
+	char *p;
+
+	p = path + strlen(path);
+	if (p > path)
+		for (p--; p > path && IS_DIR_SEP(*p); p--)
+			*p = '\0';
+}

data/pgpool2/pool_path.h

@@ -0,0 +1,81 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL 
+ * written by Tatsuo Ishii
+ *
+ * Portions Copyright (c) 2003-2008,	PgPool Global Development Group
+ * Portions Copyright (c) 2004, PostgreSQL Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ * pool_path.h.: interface to pool_path.c
+ *
+ */
+
+#ifndef POOL_PATH_H
+#define POOL_PATH_H
+
+/*
+ * MAXPGPATH: standard size of a pathname buffer in PostgreSQL (hence,
+ * maximum usable pathname length is one less).
+ *
+ * We'd use a standard system header symbol for this, if there weren't
+ * so many to choose from: MAXPATHLEN, MAX_PATH, PATH_MAX are all
+ * defined by different "standards", and often have different values
+ * on the same platform!  So we just punt and use a reasonably
+ * generous setting here.
+ */
+#define MAXPGPATH       1024
+
+#define IS_DIR_SEP(ch)  ((ch) == '/')
+#define is_absolute_path(filename) \
+( \
+    ((filename)[0] == '/') \
+)
+
+/*
+ * StrNCpy
+ *  Like standard library function strncpy(), except that result string
+ *  is guaranteed to be null-terminated --- that is, at most N-1 bytes
+ *  of the source string will be kept.
+ *  Also, the macro returns no result (too hard to do that without
+ *  evaluating the arguments multiple times, which seems worse).
+ *
+ *  BTW: when you need to copy a non-null-terminated string (like a text
+ *  datum) and add a null, do not do it with StrNCpy(..., len+1).  That
+ *  might seem to work, but it fetches one byte more than there is in the
+ *  text object.  One fine day you'll have a SIGSEGV because there isn't
+ *  another byte before the end of memory.  Don't laugh, we've had real
+ *  live bug reports from real live users over exactly this mistake.
+ *  Do it honestly with "memcpy(dst,src,len); dst[len] = '\0';", instead.
+ */
+#define StrNCpy(dst,src,len) \
+    do \
+    { \
+        char * _dst = (dst); \
+        size_t _len = (len); \
+\
+        if (_len > 0) \
+        { \
+            strncpy(_dst, (src), _len); \
+            _dst[_len-1] = '\0'; \
+        } \
+    } while (0)
+
+extern void get_parent_directory(char *path);
+extern void join_path_components(char *ret_path, const char *head, const char *tail);
+extern void canonicalize_path(char *path);
+	
+#endif /* POOL_PATH_H */

data/pgpool2/pool_process_context.c

@@ -0,0 +1,247 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL 
+ * written by Tatsuo Ishii
+ *
+ * Copyright (c) 2003-2013	PgPool Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ */
+
+#include "pool.h"
+#include "pool_process_context.h"
+#include "pool_config.h"		/* remove me afterwards */
+
+static POOL_PROCESS_CONTEXT process_context_d;
+static POOL_PROCESS_CONTEXT *process_context;
+
+/*
+ * Initialize per process context
+ */
+void pool_init_process_context(void)
+{
+	process_context = &process_context_d;
+
+	if (!process_info)
+	{
+		pool_error("pool_init_process_context: process_info is not set");
+		child_exit(1);
+	}
+	process_context->process_info = process_info;
+
+	if (!pool_config->backend_desc)
+	{
+		pool_error("pool_init_process_context: backend_desc is not set");
+		child_exit(1);
+	}
+	process_context->backend_desc = pool_config->backend_desc;
+
+	process_context->proc_id = my_proc_id;
+
+	process_context->local_session_id = 0;		/* initialize local session counter */
+}
+
+/*
+ * Return process context
+ */
+POOL_PROCESS_CONTEXT *pool_get_process_context(void)
+{
+	if (!process_context)
+	{
+		pool_error("pool_get_process_context: process context is not initialized");
+	}
+	return process_context;
+}
+
+/*
+ * Return my process info
+ */
+ProcessInfo *pool_get_my_process_info(void)
+{
+	if (!process_context)
+	{
+		pool_error("pool_get_my_process_info: process context is not initialized");
+		return NULL;
+	}
+
+	return &process_context->process_info[process_context->proc_id];
+}
+
+/*
+ * Increment local session id
+ */
+void pool_incremnet_local_session_id(void)
+{
+	POOL_PROCESS_CONTEXT *p = pool_get_process_context();
+
+	if (!p)
+	{
+		pool_error("pool_incremnet_local_session_id: cannot get process context");
+		return;
+	}
+	p->local_session_id++;
+}
+
+/*
+ * Return byte size of connection info(ConnectionInfo) on shmem.
+ */
+int pool_coninfo_size(void)
+{
+	int size;
+	size = pool_config->num_init_children *
+		pool_config->max_pool *
+		MAX_NUM_BACKENDS *
+		sizeof(ConnectionInfo);
+
+	return size;
+}
+
+/*
+ * Return number of elements of connection info(ConnectionInfo) on shmem.
+ */
+int pool_coninfo_num(void)
+{
+	int nelm;
+	nelm = pool_config->num_init_children *
+		pool_config->max_pool *
+		MAX_NUM_BACKENDS;
+
+	return nelm;
+}
+
+/*
+ * Return pointer to i th child, j th connection pool and k th backend
+ * of connection info on shmem.
+ */
+ConnectionInfo *pool_coninfo(int child, int connection_pool, int backend)
+{
+	if (child < 0 || child >= pool_config->num_init_children)
+	{
+		pool_error("pool_coninfo: invalid child number: %d", child);
+		return NULL;
+	}
+
+	if (connection_pool < 0 || connection_pool >= pool_config->max_pool)
+	{
+		pool_error("pool_coninfo: invalid connection_pool number: %d", connection_pool);
+		return NULL;
+	}
+
+	if (backend < 0 || backend >= MAX_NUM_BACKENDS)
+	{
+		pool_error("pool_coninfo: invalid backend number: %d", backend);
+		return NULL;
+	}
+
+	return &con_info[child*pool_config->max_pool*MAX_NUM_BACKENDS+
+					 connection_pool*MAX_NUM_BACKENDS+
+					 backend];
+}
+
+/*
+ * Return pointer to child which has OS process id pid, j th connection
+ * pool and k th backend of connection info on shmem.
+ */
+ConnectionInfo *pool_coninfo_pid(int pid, int connection_pool, int backend)
+{
+	int child = -1;
+	int		i;
+
+	for (i = 0; i < pool_config->num_init_children; i++)
+	{
+		if (process_info[i].pid == pid)
+		{
+			child = i;
+			break;
+		}
+	}
+
+	if (child < 0)
+	{
+		pool_error("pool_coninfo_pid: invalid child pid: %d", pid);
+		return NULL;
+	}
+
+	if (child < 0 || child >= pool_config->num_init_children)
+	{
+		pool_error("pool_coninfo_pid: invalid child number: %d", child);
+		return NULL;
+	}
+
+	if (connection_pool < 0 || connection_pool >= pool_config->max_pool)
+	{
+		pool_error("pool_coninfo_pid: invalid connection_pool number: %d", connection_pool);
+		return NULL;
+	}
+
+	if (backend < 0 || backend >= MAX_NUM_BACKENDS)
+	{
+		pool_error("pool_coninfo: invalid backend number: %d", backend);
+		return NULL;
+	}
+
+	return &con_info[child*pool_config->max_pool*MAX_NUM_BACKENDS+
+					 connection_pool*MAX_NUM_BACKENDS+
+					 backend];
+}
+
+/*
+ * Set frontend connected flag
+ */
+void pool_coninfo_set_frontend_connected(int proc_id, int pool_index)
+{
+	ConnectionInfo *con;
+	int i;
+
+	for (i=0;i<NUM_BACKENDS;i++)
+	{
+		if (!VALID_BACKEND(i))
+			continue;
+
+		con = pool_coninfo(proc_id, pool_index, i);
+
+		if (con == NULL)
+		{
+			pool_error("pool_coninfo_set_frontend_connected: cannot get ConnectionInfo");
+			return;
+		}
+		con->connected = true;
+	}
+}
+
+/*
+ * Unset frontend connected flag
+ */
+void pool_coninfo_unset_frontend_connected(int proc_id, int pool_index)
+{
+	ConnectionInfo *con;
+	int i;
+
+	for (i=0;i<NUM_BACKENDS;i++)
+	{
+		if (!VALID_BACKEND(i))
+			continue;
+
+		con = pool_coninfo(proc_id, pool_index, i);
+
+		if (con == NULL)
+		{
+			pool_error("pool_coninfo_unset_frontend_connected: cannot get ConnectionInfo");
+			return;
+		}
+		con->connected = false;
+	}
+}