prestogres 0.1.0

data/pgpool2/watchdog/wd_ping.c

@@ -0,0 +1,330 @@
+/*
+ * $Header$
+ *
+ * Handles watchdog connection, and protocol communication with pgpool-II
+ *
+ * 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 <pthread.h>
+#include <stdio.h>
+#include <errno.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <netdb.h>
+#include <sys/wait.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include "pool.h"
+#include "pool_config.h"
+#include "watchdog.h"
+
+#define WD_MAX_PING_RESULT 256
+
+int wd_is_upper_ok(char * server_list);
+int wd_is_unused_ip(char * ip);
+
+static void * exec_ping(void * arg);
+static double get_result (char * ping_data);
+
+/**
+ * Try to connect to trusted servers.
+ */
+int
+wd_is_upper_ok(char * server_list)
+{
+	pthread_attr_t attr;
+	char * buf;
+	int rc = 0;
+	int i,cnt;
+	int len;
+	pthread_t thread[MAX_WATCHDOG_NUM];
+	WdInfo thread_arg[MAX_WATCHDOG_NUM];
+
+	char * bp, *ep;
+	int rtn = WD_NG;
+
+	if (server_list == NULL)
+	{
+		pool_error("wd_is_upper_ok: server_list is NULL");
+		return WD_NG;
+	}
+	len = strlen(server_list)+2;
+	buf = malloc(len);
+	if (buf == NULL)
+	{
+		pool_error("wd_is_upper_ok: malloc failed");
+		return WD_NG;
+	}
+
+	memset(buf,0,len);
+	strlcpy(buf,server_list,len);
+	/* thread init */
+	pthread_attr_init(&attr);
+	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
+	/* set hostname as a thread_arg */
+	bp = buf;
+	cnt = 0;
+	while (*bp != '\0')
+	{
+		ep = strchr(bp,',');
+		if (ep != NULL)
+		{
+			*ep = '\0';
+		}
+		strlcpy(thread_arg[cnt].hostname,bp,sizeof(thread_arg[cnt].hostname));
+		rc = pthread_create(&thread[cnt], &attr, exec_ping, (void*)&thread_arg[cnt]);
+
+		cnt ++;
+		if (ep != NULL)
+		{
+			bp = ep + 1;
+		}
+		else
+		{
+			break;
+		}
+		if (cnt >= MAX_WATCHDOG_NUM)
+		{
+			pool_debug("wd_is_upper_ok: trusted server num is out of range(%d)",cnt);
+			break;
+		}
+	}
+	pthread_attr_destroy(&attr);
+	for (i=0; i <cnt; )
+	{
+		int result;
+		rc = pthread_join(thread[i], (void **)&result);
+		if ((rc != 0) && (errno == EINTR))
+		{
+			usleep(100);
+			continue;
+		}
+		if (result == WD_OK)
+		{
+			rtn = WD_OK;
+		}
+		i++;
+	}
+	free(buf);
+	return rtn;
+}
+
+/**
+ * check if IP address is unused.
+ */
+int
+wd_is_unused_ip(char * ip)
+{
+	pthread_attr_t attr;
+	int rc = 0;
+	pthread_t thread;
+	WdInfo thread_arg;
+
+	int rtn = WD_NG;
+	int result;
+
+	if (ip == NULL)
+	{
+		return WD_NG;
+	}
+
+	/* thread init */
+	pthread_attr_init(&attr);
+	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
+
+	/* set hostname as a thread_arg */
+	strlcpy(thread_arg.hostname,ip,sizeof(thread_arg.hostname));
+
+	rc = pthread_create(&thread, &attr, exec_ping, (void*)&thread_arg);
+	pthread_attr_destroy(&attr);
+
+	rc = pthread_join(thread, (void **)&result);
+	if ((rc != 0) && (errno == EINTR))
+	{
+		return WD_NG;
+	}
+	if (result == WD_NG)
+	{
+		rtn = WD_OK;
+	}
+
+	return rtn;
+}
+
+
+/**
+ * Thread to execute ping against "trusted hosts" or delegate IP.
+ */
+static void *
+exec_ping(void * arg)
+{
+	WdInfo * thread_arg;
+	uintptr_t rtn = (uintptr_t)WD_NG;
+	int pfd[2];
+	int status;
+	char * args[8];
+	int pid, i = 0;
+	int r_size = 0;
+	char result[WD_MAX_PING_RESULT];
+	char ping_path[WD_MAX_PATH_LEN];
+
+	snprintf(ping_path,sizeof(ping_path),"%s/ping",pool_config->ping_path);
+	thread_arg = (WdInfo *)arg;
+	memset(result,0,sizeof(result));
+
+	if (pipe(pfd) == -1)
+	{
+		pool_error("exec_ping: pipe open error:%s", strerror(errno));
+		return WD_NG;
+	}
+
+	args[i++] = "ping";
+	args[i++] = "-q";
+	args[i++] = "-c3";
+	args[i++] = thread_arg->hostname;
+	args[i++] = NULL;
+
+	pid = fork();
+	if (pid == -1)
+	{
+		pool_error("exec_ping: fork() failed. reason: %s", strerror(errno));
+		exit(1);
+	}
+	if (pid == 0)
+	{
+		close(STDOUT_FILENO);
+		dup2(pfd[1], STDOUT_FILENO);
+		close(pfd[0]);
+		status = execv(ping_path,args);
+
+		if (status == -1)
+		{
+			pool_error("exec_ping: execv(%s) failed. reason: %s", ping_path, strerror(errno));
+			exit(1);
+		}
+		exit(0);
+	}
+	else
+	{
+		close(pfd[1]);
+		for (;;)
+		{
+			int r;
+			r = wait(&status);
+			if (r < 0)
+			{
+				if (errno == EINTR)
+					continue;
+
+				pool_error("exec_ping: wait() failed. reason: %s", strerror(errno));
+				close(pfd[0]);
+				return WD_NG;
+			}
+
+			if (WIFEXITED(status) == 0)
+			{
+				pool_error("exec_ping: %s exited abnormally", ping_path);
+				close(pfd[0]);
+				return WD_NG;
+			}
+			else if (WEXITSTATUS(status) != 0)
+			{
+				pool_debug("exec_ping: failed to ping %s", thread_arg->hostname);
+				return WD_NG;
+			}
+			else
+			{
+				pool_debug("exec_ping: succeed to ping %s", thread_arg->hostname);
+				break;
+			}
+		}
+
+		i = 0;
+		while  (( (r_size = read (pfd[0], &result[i], sizeof(result)-i-1)) > 0) && (errno == EINTR))
+		{
+			i += r_size;
+		}
+		result[sizeof(result)-1] = '\0';
+
+		close(pfd[0]);
+	}
+
+	/* Check whether average RTT >= 0 */
+	rtn = (get_result (result) >= 0) ? WD_OK : WD_NG;
+
+	pthread_exit((void *)rtn);
+}
+
+/**
+ * Get average round-trip time of ping result.
+ */
+static double
+get_result (char * ping_data)
+{
+	char * sp = NULL;
+	char * ep = NULL;
+	int i;
+	double msec = 0;
+
+	if (ping_data == NULL)
+	{
+		pool_error("get_result: no ping data");
+		return -1;
+	}
+
+	pool_debug("get_result: ping data: %s", ping_data);
+
+	/*
+	 skip result until average data
+	 typical result of ping is as follows,
+	 "rtt min/avg/max/mdev = 0.045/0.045/0.046/0.006 ms"
+	 we can find the average data beyond the 4th '/'.
+	 */
+	sp = ping_data;
+	for ( i = 0 ; i < 4 ; i ++)
+	{
+		sp = strchr(sp,'/');
+		if (sp == NULL)
+		{
+			return -1;
+		}
+		sp ++;
+	}
+
+	ep = strchr (sp,'/');
+	if (ep == NULL)
+	{
+		return -1;
+	}
+
+	*ep = '\0';
+	errno = 0;
+
+	/* convert to numeric data from text */
+	msec = strtod(sp,(char **)NULL);
+
+	if (errno != 0)
+	{
+		pool_error("get_result: strtod() failed. reason: %s", strerror(errno));
+		return -1;
+	}
+
+	return msec;
+}

data/pgpool2/ylwrap

@@ -0,0 +1,223 @@
+#! /bin/sh
+# ylwrap - wrapper for lex/yacc invocations.
+
+scriptversion=2005-05-14.22
+
+# Copyright (C) 1996, 1997, 1998, 1999, 2001, 2002, 2003, 2004, 2005
+#   Free Software Foundation, Inc.
+#
+# Written by Tom Tromey <tromey@cygnus.com>.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2, or (at your option)
+# any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+# 02110-1301, USA.
+
+# As a special exception to the GNU General Public License, if you
+# distribute this file as part of a program that contains a
+# configuration script generated by Autoconf, you may include it under
+# the same distribution terms that you use for the rest of that program.
+
+# This file is maintained in Automake, please report
+# bugs to <bug-automake@gnu.org> or send patches to
+# <automake-patches@gnu.org>.
+
+case "$1" in
+  '')
+    echo "$0: No files given.  Try \`$0 --help' for more information." 1>&2
+    exit 1
+    ;;
+  --basedir)
+    basedir=$2
+    shift 2
+    ;;
+  -h|--h*)
+    cat <<\EOF
+Usage: ylwrap [--help|--version] INPUT [OUTPUT DESIRED]... -- PROGRAM [ARGS]...
+
+Wrapper for lex/yacc invocations, renaming files as desired.
+
+  INPUT is the input file
+  OUTPUT is one file PROG generates
+  DESIRED is the file we actually want instead of OUTPUT
+  PROGRAM is program to run
+  ARGS are passed to PROG
+
+Any number of OUTPUT,DESIRED pairs may be used.
+
+Report bugs to <bug-automake@gnu.org>.
+EOF
+    exit $?
+    ;;
+  -v|--v*)
+    echo "ylwrap $scriptversion"
+    exit $?
+    ;;
+esac
+
+
+# The input.
+input="$1"
+shift
+case "$input" in
+  [\\/]* | ?:[\\/]*)
+    # Absolute path; do nothing.
+    ;;
+  *)
+    # Relative path.  Make it absolute.
+    input="`pwd`/$input"
+    ;;
+esac
+
+pairlist=
+while test "$#" -ne 0; do
+  if test "$1" = "--"; then
+    shift
+    break
+  fi
+  pairlist="$pairlist $1"
+  shift
+done
+
+# The program to run.
+prog="$1"
+shift
+# Make any relative path in $prog absolute.
+case "$prog" in
+  [\\/]* | ?:[\\/]*) ;;
+  *[\\/]*) prog="`pwd`/$prog" ;;
+esac
+
+# FIXME: add hostname here for parallel makes that run commands on
+# other machines.  But that might take us over the 14-char limit.
+dirname=ylwrap$$
+trap "cd `pwd`; rm -rf $dirname > /dev/null 2>&1" 1 2 3 15
+mkdir $dirname || exit 1
+
+cd $dirname
+
+case $# in
+  0) $prog "$input" ;;
+  *) $prog "$@" "$input" ;;
+esac
+ret=$?
+
+if test $ret -eq 0; then
+  set X $pairlist
+  shift
+  first=yes
+  # Since DOS filename conventions don't allow two dots,
+  # the DOS version of Bison writes out y_tab.c instead of y.tab.c
+  # and y_tab.h instead of y.tab.h. Test to see if this is the case.
+  y_tab_nodot="no"
+  if test -f y_tab.c || test -f y_tab.h; then
+    y_tab_nodot="yes"
+  fi
+
+  # The directory holding the input.
+  input_dir=`echo "$input" | sed -e 's,\([\\/]\)[^\\/]*$,\1,'`
+  # Quote $INPUT_DIR so we can use it in a regexp.
+  # FIXME: really we should care about more than `.' and `\'.
+  input_rx=`echo "$input_dir" | sed 's,\\\\,\\\\\\\\,g;s,\\.,\\\\.,g'`
+
+  while test "$#" -ne 0; do
+    from="$1"
+    # Handle y_tab.c and y_tab.h output by DOS
+    if test $y_tab_nodot = "yes"; then
+      if test $from = "y.tab.c"; then
+    	from="y_tab.c"
+      else
+    	if test $from = "y.tab.h"; then
+    	  from="y_tab.h"
+    	fi
+      fi
+    fi
+    if test -f "$from"; then
+      # If $2 is an absolute path name, then just use that,
+      # otherwise prepend `../'.
+      case "$2" in
+    	[\\/]* | ?:[\\/]*) target="$2";;
+    	*) target="../$2";;
+      esac
+
+      # We do not want to overwrite a header file if it hasn't
+      # changed.  This avoid useless recompilations.  However the
+      # parser itself (the first file) should always be updated,
+      # because it is the destination of the .y.c rule in the
+      # Makefile.  Divert the output of all other files to a temporary
+      # file so we can compare them to existing versions.
+      if test $first = no; then
+	realtarget="$target"
+	target="tmp-`echo $target | sed s/.*[\\/]//g`"
+      fi
+      # Edit out `#line' or `#' directives.
+      #
+      # We don't want the resulting debug information to point at
+      # an absolute srcdir; it is better for it to just mention the
+      # .y file with no path.
+      #
+      # We want to use the real output file name, not yy.lex.c for
+      # instance.
+      #
+      # We want the include guards to be adjusted too.
+      FROM=`echo "$from" | sed \
+            -e 'y/abcdefghijklmnopqrstuvwxyz/ABCDEFGHIJKLMNOPQRSTUVWXYZ/'\
+            -e 's/[^ABCDEFGHIJKLMNOPQRSTUVWXYZ]/_/g'`
+      TARGET=`echo "$2" | sed \
+            -e 'y/abcdefghijklmnopqrstuvwxyz/ABCDEFGHIJKLMNOPQRSTUVWXYZ/'\
+            -e 's/[^ABCDEFGHIJKLMNOPQRSTUVWXYZ]/_/g'`
+
+      sed -e "/^#/!b" -e "s,$input_rx,," -e "s,$from,$2," \
+          -e "s,$FROM,$TARGET," "$from" >"$target" || ret=$?
+
+      # Check whether header files must be updated.
+      if test $first = no; then
+	if test -f "$realtarget" && cmp -s "$realtarget" "$target"; then
+	  echo "$2" is unchanged
+	  rm -f "$target"
+	else
+          echo updating "$2"
+          mv -f "$target" "$realtarget"
+        fi
+      fi
+    else
+      # A missing file is only an error for the first file.  This
+      # is a blatant hack to let us support using "yacc -d".  If -d
+      # is not specified, we don't want an error when the header
+      # file is "missing".
+      if test $first = yes; then
+        ret=1
+      fi
+    fi
+    shift
+    shift
+    first=no
+  done
+else
+  ret=$?
+fi
+
+# Remove the directory.
+cd ..
+rm -rf $dirname
+
+exit $ret
+
+# Local Variables:
+# mode: shell-script
+# sh-indentation: 2
+# eval: (add-hook 'write-file-hooks 'time-stamp)
+# time-stamp-start: "scriptversion="
+# time-stamp-format: "%:y-%02m-%02d.%02H"
+# time-stamp-end: "$"
+# End: