mirror_corosync/test/testamf1.c
2007-06-23 09:14:15 +00:00

628 lines
17 KiB
C

/*
* Copyright (c) 2002-2003 MontaVista Software, Inc.
* Copyright (c) 2006 Ericsson AB.
* Copyright (c) 2006 Sun Microsystems, Inc.
*
* All rights reserved.
*
* Author: Steven Dake (sdake@mvista.com)
* Hans Feldt
*
* This software licensed under BSD license, the text of which follows:
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* - Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* - Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* - Neither the name of the MontaVista Software, Inc. nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <signal.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sched.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdarg.h>
#include "saAis.h"
#include "saAmf.h"
SaAmfHandleT handle;
SaAmfHealthcheckKeyT keyAmfInvoked = {
.key = "amfInvoked",
.keyLen = 10
};
SaAmfHealthcheckKeyT keyCompInvoked = {
.key = "compInvoked",
.keyLen = 11
};
SaNameT compNameGlobal;
int good_health = 0;
int good_health_limit = 0;
enum {
FINALIZE = 0,
UNREGISTER,
ERROR_REPORT
};
#define die(format, args...) _die (__FILE__, __LINE__, format, ##args)
static void _die (char *file, int line, char *format, ...) __attribute__((format(printf, 3, 4)));
static void _die (char *file, int line, char *format, ...)
{
char buf[1024];
va_list ap;
sprintf (buf, "%d - %s:#%d - Error: '%s', exiting...\n",
(int)getpid(), file, line, format);
va_start (ap, format);
vfprintf (stderr, buf, ap);
va_end(ap);
exit (-1);
}
static void response (
SaAmfHandleT handle, SaInvocationT invocation, SaAisErrorT error)
{
SaAisErrorT result;
do {
result = saAmfResponse (handle, invocation, error);
if (result == SA_AIS_ERR_TRY_AGAIN) {
fprintf(stderr, "%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfResponse failed %d", result);
}
}
void printSaNameT (SaNameT *name)
{
int i;
for (i = 0; i < name->length; i++) {
printf ("%c", name->value[i]);
}
}
void setSanameT (SaNameT *name, char *str) {
name->length = strlen (str);
memcpy (name->value, str, name->length);
}
static unsigned int healthcheck_no = 0;
int stop = 0;
void HealthcheckCallback (SaInvocationT invocation,
const SaNameT *compName,
SaAmfHealthcheckKeyT *healthcheckKey)
{
SaAisErrorT res;
if( !good_health && healthcheck_no++);
if (healthcheck_no == good_health_limit ) {
response (handle, invocation, SA_AIS_OK);
res = saAmfHealthcheckStop (handle,
&compNameGlobal,
&keyAmfInvoked);
printf ("healthcheck stop result %d (should be %d)\n", res, SA_AIS_OK);
printf ("COMPONENT REPORTING ERROR %s\n", compNameGlobal.value);
saAmfComponentErrorReport (handle, compName, 0, SA_AMF_COMPONENT_RESTART, 0);
printf ("COMPONENT DONE REPORTING ERROR\n");
} else {
response (handle, invocation, SA_AIS_OK);
}
}
void ComponentTerminateCallback (
SaInvocationT invocation,
const SaNameT *compName)
{
printf ("ComponentTerminateCallback\n");
response (handle, invocation, SA_AIS_OK);
exit (0);
}
#if 0
#include <sys/time.h>
#define TRU "%d"
#define TRS "%s"
#define TR(format,x) do { \
struct timeval t;\
gettimeofday(&t,NULL); \
printf("%s:%d: %s : %d : %u: %u :%s : " format "\n",\
__FILE__,__LINE__,__FUNCTION__, \
(int)getpid(),(int)t.tv_sec, (int)t.tv_usec,#x,x); \
}while(0)
#else
#define TRU "%d"
#define TRS "%s"
#define TR(format,x)
#endif
void CSISetCallback (
SaInvocationT invocation,
const SaNameT *compName,
SaAmfHAStateT haState,
SaAmfCSIDescriptorT *csiDescriptor)
{
SaAmfHAStateT state;
int res;
int i;
switch (haState) {
case SA_AMF_HA_ACTIVE:
printf ("PID %d: Component '%s' requested to enter hastate SA_AMF_ACTIVE"
" for \n\tCSI '%s'\n",
(int)getpid(), compName->value, csiDescriptor->csiName.value);
response (handle, invocation, SA_AIS_OK);
res = saAmfHAStateGet (handle, compName, &csiDescriptor->csiName, &state);
if (res != SA_AIS_OK || haState != state) {
fprintf (stderr, "saAmfHAStateGet failed: %d\n", res);
exit (-1);
}
TR(TRU, csiDescriptor->csiAttr.number);
for(i=0; i<csiDescriptor->csiAttr.number; i++) {
if( strcmp((char*)csiDescriptor->csiAttr.attr[i].attrName, "good_health_limit") == 0){
good_health = strcmp((char*)csiDescriptor->csiAttr.attr[i].attrValue, "0") ? 0 : 1;
good_health_limit = atoi((char*)csiDescriptor->csiAttr.attr[i].attrValue);
}
#if 0
TR(TRS,csiDescriptor->csiAttr.attr[i].attrName);
TR(TRS, csiDescriptor->csiAttr.attr[i].attrValue);
#endif
}
TR(TRU, csiDescriptor->csiFlags);
printSaNameT((SaNameT*) &csiDescriptor->csiStateDescriptor.activeDescriptor.activeCompName);
TR(TRU, csiDescriptor->csiStateDescriptor.activeDescriptor.transitionDescriptor);
break;
case SA_AMF_HA_STANDBY:
printf ("PID %d: Component '%s' requested to enter hastate SA_AMF_STANDBY "
"for \n\tCSI '%s'\n",
(int)getpid(), compName->value, csiDescriptor->csiName.value);
response (handle, invocation, SA_AIS_OK);
TR(TRU,csiDescriptor->csiAttr.number);
for(i=0; i<csiDescriptor->csiAttr.number; i++) {
if(!strcmp((char*)csiDescriptor->csiAttr.attr[i].attrName, "good_health") &&
!strcmp((char*)csiDescriptor->csiAttr.attr[i].attrValue, "true")){
good_health = 1;
}
TR(TRS,csiDescriptor->csiAttr.attr[i].attrName);
TR(TRS,csiDescriptor->csiAttr.attr[i].attrValue);
}
TR(TRU,csiDescriptor->csiFlags);
printSaNameT((SaNameT*) &csiDescriptor->csiStateDescriptor.standbyDescriptor.activeCompName);
TR(TRU,csiDescriptor->csiStateDescriptor.standbyDescriptor.standbyRank);
break;
case SA_AMF_HA_QUIESCED:
printf ("%d: Component '%s' requested to enter hastate SA_AMF_HA_QUIESCED "
"for \n\tCSI '%s'\n",
(int)getpid(), compName->value, csiDescriptor->csiName.value);
response (handle, invocation, SA_AIS_OK);
break;
case SA_AMF_HA_QUIESCING:
break;
default:
break;
}
}
void CSIRemoveCallback (
SaInvocationT invocation,
const SaNameT *compName,
const SaNameT *csiName,
SaAmfCSIFlagsT csiFlags)
{
printf ("CSIRemoveCallback for component '");
printSaNameT ((SaNameT *)compName);
printf ("' in CSI '");
printSaNameT ((SaNameT *)csiName);
printf ("'\n");
response (handle, invocation, SA_AIS_OK);
}
#ifdef COMPILE_OUT
void ProtectionGroupTrackCallback (
const SaNameT *csiName,
SaAmfProtectionGroupNotificationT *notificationBuffer,
SaUint32T numberOfItems,
SaUint32T numberOfMembers,
SaAisErrorT error)
{
int i;
printf ("ProtectionGroupTrackCallback items %d members %d\n", (int)numberOfItems, (int)numberOfMembers);
printf ("buffer is %p\n", notificationBuffer);
for (i = 0; i < numberOfItems; i++) {
printf ("component name");
printSaNameT (&notificationBuffer[i].member.compName);
printf ("\n");
printf ("\treadiness state is %d\n", notificationBuffer[i].member.readinessState);
printf ("\thastate %d\n", notificationBuffer[i].member.haState);
printf ("\tchange is %d\n", notificationBuffer[i].change);
}
}
#endif
SaAmfCallbacksT amfCallbacks = {
.saAmfHealthcheckCallback = HealthcheckCallback,
.saAmfComponentTerminateCallback = ComponentTerminateCallback,
.saAmfCSISetCallback = CSISetCallback,
.saAmfCSIRemoveCallback = CSIRemoveCallback,
};
SaAmfCallbacksT amfCallbacks;
SaVersionT version = { 'B', 1, 1 };
#if ! defined(TS_CLASS) && (defined(OPENAIS_BSD) || defined(OPENAIS_LINUX) || defined(OPENAIS_SOLARIS))
static struct sched_param sched_param;
#endif
void sigintr_handler (int signum) {
stop = FINALIZE;
}
void sigusr1_handler (int signum) {
stop = UNREGISTER;
}
void sigusr2_handler (int signum) {
stop = ERROR_REPORT;
}
void write_pid (void) {
char pid[256];
char filename[256];
int fd;
int res;
sprintf (filename, "/var/run/openais_cleanup_%s", compNameGlobal.value);
fd = open (filename, O_CREAT | O_TRUNC | O_RDWR, S_IRWXU);
if (fd == -1) {
printf("%d: Failed using /var/run for pid file, using /tmp\n", (int)getpid());
sprintf (filename, "/tmp/openais_cleanup_%s", compNameGlobal.value);
fd = open (filename, O_CREAT | O_TRUNC | O_RDWR, S_IRWXU);
}
sprintf (pid, "%d", (int)getpid());
res = write (fd, pid, strlen (pid));
close (fd);
}
static SaSelectionObjectT comp_init ()
{
char *name;
char *env;
int result;
SaSelectionObjectT select_fd;
SaAmfPmErrorsT pmErrors = (SA_AMF_PM_ZERO_EXIT |
SA_AMF_PM_NON_ZERO_EXIT |
SA_AMF_PM_ABNORMAL_END);
name = getenv ("SA_AMF_COMPONENT_NAME");
if (name == NULL) {
die ("SA_AMF_COMPONENT_NAME missing");
}
if (strstr (name, "safComp=") == NULL ||
strstr (name, "safSu=") == NULL ||
strstr (name, "safSg=") == NULL ||
strstr (name, "safApp=") == NULL) {
die ("SA_AMF_COMPONENT_NAME value wrong");
}
printf("%d: Hello world from %s\n", (int)getpid(), name);
env = getenv ("var1");
if (env == NULL) {
die ("var1 missing");
}
if (strcmp (env, "val1") != 0) {
die ("var1 value wrong");
}
env = getenv ("var2");
if (env == NULL) {
die ("var2 wrong");
}
if (strcmp (env, "val2") != 0) {
die ("var2 value wrong");
}
signal (SIGINT, sigintr_handler);
signal (SIGUSR1, sigusr1_handler);
signal (SIGUSR2, sigusr2_handler);
#if ! defined(TS_CLASS) && (defined(OPENAIS_BSD) || defined(OPENAIS_LINUX) || defined(OPENAIS_SOLARIS))
sched_param.sched_priority = sched_get_priority_max(SCHED_RR);
if (sched_param.sched_priority == -1) {
fprintf (stderr, "%d: couldn't retrieve the maximum scheduling " \
"priority supported by the Round-Robin class (%s)\n",
(int)getpid(), strerror(errno));
} else {
result = sched_setscheduler (0, SCHED_RR, &sched_param);
if (result == -1) {
fprintf (stderr, "%d: couldn't set sched priority (%s)\n",
(int)getpid(), strerror(errno));
}
}
#endif
do {
result = saAmfInitialize (&handle, &amfCallbacks, &version);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfInitialize result is %d", result);
}
do {
result = saAmfSelectionObjectGet (handle, &select_fd);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfSelectionObjectGet failed %d", result);
}
do {
result = saAmfComponentNameGet (handle, &compNameGlobal);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfComponentNameGet failed %d", result);
}
write_pid ();
do {
result = saAmfHealthcheckStart (handle,
&compNameGlobal,
&keyAmfInvoked,
SA_AMF_HEALTHCHECK_AMF_INVOKED,
SA_AMF_COMPONENT_FAILOVER);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfHealthcheckStart failed %d", result);
}
do {
result = saAmfHealthcheckStart (handle,
&compNameGlobal,
&keyCompInvoked,
SA_AMF_HEALTHCHECK_COMPONENT_INVOKED,
SA_AMF_COMPONENT_FAILOVER);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfHealthcheckStart failed %d", result);
}
{
SaNameT badname;
strcpy ((char*)badname.value, "badname");
badname.length = 7;
do {
result = saAmfComponentRegister (handle, &badname, NULL);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_ERR_INVALID_PARAM) {
die ("saAmfComponentRegister failed %d", result);
}
}
do {
result = saAmfComponentRegister (handle, &compNameGlobal, NULL);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfComponentRegister failed %d", result);
}
/*
* startup passive monitoring
*/
do {
result = saAmfPmStart (handle,
&compNameGlobal, getpid(), 1,
pmErrors,
SA_AMF_COMPONENT_FAILOVER);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
/*
* Test already started healthcheck
*/
do {
result = saAmfHealthcheckStart (handle,
&compNameGlobal,
&keyAmfInvoked,
SA_AMF_HEALTHCHECK_AMF_INVOKED,
SA_AMF_COMPONENT_FAILOVER);
if (result == SA_AIS_ERR_TRY_AGAIN) {
printf("%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_ERR_EXIST) {
die ("saAmfHealthcheckStart failed %d", result);
}
return select_fd;
}
static void handle_intr (void)
{
SaAisErrorT result;
switch (stop) {
case FINALIZE:
result = saAmfFinalize (handle);
if (result != SA_AIS_OK) {
die ("saAmfFinalize failed %d", result);
}
fprintf(stderr, "%d: %s exiting\n",
(int)getpid(), compNameGlobal.value);
exit (EXIT_SUCCESS);
break;
case UNREGISTER:
fprintf(stderr, "%d: %s unregistering\n",
(int)getpid(), compNameGlobal.value);
result = saAmfComponentUnregister (
handle, &compNameGlobal, NULL);
if (result != SA_AIS_OK) {
die ("saAmfComponentUnregister failed %d", result);
}
fprintf(stderr, "%d: waiting after unregister\n", (int)getpid());
while (1) {
sleep (100000000);
}
break;
case ERROR_REPORT:
fprintf(stderr, "%d: %s error reporting\n",
(int)getpid(), compNameGlobal.value);
result = saAmfComponentErrorReport (
handle, &compNameGlobal, 0, SA_AMF_COMPONENT_RESTART, 0);
if (result != SA_AIS_OK) {
die ("saAmfComponentErrorReport failed %d", result);
}
fprintf(stderr, "%d: waiting after error report\n", (int)getpid());
while (1) {
sleep (100000000);
}
break;
default:
die ("unknown %d", stop);
break;
}
}
int main (int argc, char **argv)
{
int result;
SaSelectionObjectT select_fd;
fd_set read_fds;
struct timeval tv;
select_fd = comp_init();
FD_ZERO (&read_fds);
do {
tv.tv_sec = 2; /* related to value in amf.conf! */
tv.tv_usec = 0;
FD_SET (select_fd, &read_fds);
result = select (select_fd + 1, &read_fds, 0, 0, &tv);
if (result == -1) {
if (errno == EINTR) {
handle_intr ();
} else {
die ("select failed - %s", strerror (errno));
}
} else if (result > 0) {
do {
result = saAmfDispatch (handle, SA_DISPATCH_ALL);
if (result == SA_AIS_ERR_TRY_AGAIN) {
fprintf(stderr, "%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfDispatch failed %d", result);
}
} else {
/* timeout */
do {
result = saAmfHealthcheckConfirm (handle, &compNameGlobal,
&keyCompInvoked, SA_AIS_OK);
if (result == SA_AIS_ERR_TRY_AGAIN) {
fprintf(stderr, "%d: TRY_AGAIN received\n", (int)getpid());
usleep (100000);
}
} while (result == SA_AIS_ERR_TRY_AGAIN);
if (result != SA_AIS_OK) {
die ("saAmfHealthcheckConfirm failed %d", result);
}
}
} while (stop == 0);
fprintf(stderr, "%d: exiting...\n", (int)getpid());
exit (EXIT_SUCCESS);
}