Logo Search packages:      
Sourcecode: pacemaker-mgmt version File versions  Download package

ha_mgmt_client.c

/*
 * ha_mgmt_client.c: functions called by JAVA 
 *
 * Author: Jia Ming Pan <jmltc@cn.ibm.com>
 * Copyright (c) 2006 International Business Machines
 *
 * 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 of the License, 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 */

#include <pygui_internal.h>
#ifdef HAVE_STDINT_H
#include <stdint.h>
#endif /* HAVE_STDINT_H */
#include <glib.h>
#include <string.h>
#include <pthread.h>
#include <signal.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <clplumbing/cl_log.h>
#include <clplumbing/GSource.h>
#include <clplumbing/cl_pidfile.h>
#include <clplumbing/cl_signal.h>
#include <clplumbing/coredumps.h>
#include <ha_msg.h>
#include <mgmt/mgmt.h>
#include "ha_mgmt_client.h"
#include "ha_tsa_common.h"


#define CMD_SUBSCRIBE_EVENT   "cmd_subscribe_event"
#define     IPC_PATH_ATTR     "path"            /* pathname attribute */
#define TSA_CMDPATH     HA_VARRUNDIR"/heartbeat/tsa_eventd"
#define TSA_HA_CLI      HA_LIBHBDIR"/tsa_hacli"
/*FIXME: remove hardcode pathname */
#define HB_SCRIPT "/etc/init.d/heartbeat" 

static IPC_Channel*     eventd_signon(void);
static int        eventd_signoff(IPC_Channel*);

static IPC_Channel*
eventd_signon ()
{
      GHashTable* eventd_chan_attrs;
      IPC_Channel * ipc_chan;
      char path[] = IPC_PATH_ATTR;
      char cmd_path[] = TSA_CMDPATH;

      /* create the command ipc channel to eventd*/
      eventd_chan_attrs = g_hash_table_new(g_str_hash, g_str_equal);
      g_hash_table_insert(eventd_chan_attrs, path, cmd_path);
      ipc_chan = ipc_channel_constructor(IPC_ANYTYPE, eventd_chan_attrs);
      g_hash_table_destroy(eventd_chan_attrs);

      if ( ipc_chan == NULL ){
            eventd_signoff(ipc_chan);
            cl_log(LOG_WARNING,
                  "eventd_signon: can not connect to eventd channel");
            return NULL;
      }

      if ( ipc_chan->ops->initiate_connection(ipc_chan) != IPC_OK) {
            eventd_signoff(ipc_chan);
            cl_log(LOG_WARNING,
                  "eventd_signon: can not initiate connection");
            return NULL;
      }
      return ipc_chan;
}

static int
eventd_signoff (IPC_Channel * ipc_chan)
{
      if (ipc_chan) {
            if (IPC_ISWCONN(ipc_chan)) {
                  ipc_chan->ops->destroy(ipc_chan);
            }
            ipc_chan = NULL;
      }
      return HA_OK;
}

/* wait_for_events: block until an event received,
      called by Java code */
char *      
wait_for_events(void)
{
      static char buf[1024];  /* to avoid memory leak */
      struct ha_msg *msg = NULL, *cmd = NULL;
      IPC_Channel * ipc_chan;
      
      while ( (ipc_chan = eventd_signon()) == NULL ) {
            sleep(3);
      }

      if ( ( cmd = ha_msg_new(1) ) == NULL ) {
            cl_log(LOG_ERR, "%s: alloc msg failed.", __FUNCTION__);
            goto exit;
      }
      
      ha_msg_add(cmd, "cmd", CMD_SUBSCRIBE_EVENT);
      if ( msg2ipcchan(cmd, ipc_chan) != HA_OK ) {
            cl_log(LOG_ERR, "%s: send msg to ipchan failed.",
                  __FUNCTION__);
      }

      msg = msgfromIPC(ipc_chan, MSG_ALLOWINTR);
      ha_msg_del(cmd);

exit:
      eventd_signoff(ipc_chan);
      if ( msg ) {
            strncpy(buf, ha_msg_value(msg, "event"), 1024);
            ha_msg_del(msg);
            return buf;
      }
      return NULL;
}


void
clLog(int priority, const char* logs)
{
      init_logger("TSA");
      cl_log(priority, "%s", logs);
}


void  
start_heartbeat(const char* node)
{
      char * ssh_path = NULL;
      int rc, len;

      ssh_path = run_shell_cmnd("which ssh", &rc, &len);
      if ( ssh_path == NULL ) {
            return;
      }
      ssh_path[strlen(ssh_path) - 1] = '\0';
      if ( fork() == 0 ) {
            char cmd[1024];
            snprintf(cmd, 1024, "%s %s \"%s start\"", ssh_path, node, HB_SCRIPT);
            if (system(cmd) != 0) {
                  cl_log(LOG_ERROR, "%s: system(%s) returned non-zero"
                  ,     __FUNCTION__, cmd);
            }
      }
      free(ssh_path);
}


void  
stop_heartbeat(const char* node)
{
      char * ssh_path = NULL;
      int rc, len;

      ssh_path = run_shell_cmnd("which ssh", &rc, &len);
      if ( ssh_path == NULL ) {
            return;
      }
      ssh_path[strlen(ssh_path) - 1] = '\0';
      if ( fork() == 0 ) {
            char cmd[1024];
            snprintf(cmd, 1024, "%s %s \"%s start\"", ssh_path, node, HB_SCRIPT);
            if (system(cmd) != 0) {
                  cl_log(LOG_ERRROR, "%s: system(%s) failed.", __FUNCTION__
                  ,     cmd);
            }
      }
      free(ssh_path);
}


/* native way: talk to library with JNI */
char*  
process_cmnd_native(const char* cmd)
{
      static char buf[1024];  /* to avoid memory leak */
      char *msg = NULL, *result = NULL, **cmd_args = NULL;
      int i, len;

      init_logger("TSA");
        
      init_mgmt_lib("tsa", ENABLE_LRM|ENABLE_CRM|ENABLE_HB|CACHE_CIB);
      cl_log(LOG_DEBUG, "%s: begin.", __FUNCTION__);
      /* cl_log(LOG_INFO, "%s: cmd: %s", __FUNCTION__, cmd); */
      cmd_args = split_string(cmd, &len, " ");
      if ( cmd_args == NULL ) {
            cl_log(LOG_INFO, "%s: cmd_args is NULL.", __FUNCTION__);
            final_mgmt_lib(); 
            return NULL;
      }

      msg = mgmt_new_msg(cmd_args[0], NULL);
      for(i = 1; i < len; i++ ) {
            msg = mgmt_msg_append(msg, cmd_args[i]);
      }

      cl_log(LOG_DEBUG, "process_command: [%s}", msg);
      result = process_msg(msg);
      cl_log(LOG_DEBUG, "process_command: [%s] => [%s]", msg, result); 
      mgmt_del_msg(msg);
      free_array((void**)cmd_args, len);
      if ( result == NULL ) {
            cl_log(LOG_ERR, "%s: end. result is NULL.", __FUNCTION__);
            final_mgmt_lib(); 
            return NULL;
      }
      strncpy(buf, result, 1024);
      mgmt_del_msg(result);
      cl_log(LOG_DEBUG, "%s: end.", __FUNCTION__);


        final_mgmt_lib(); 
      return buf;
}

/* external way: 
      invoke the external tool tsa_hacli to operate on linux-ha 
*/
char* 
process_cmnd_external(const char *cmd)
{
      static char result[4096];
      char cmd_buf[1024];
      char * buf = NULL;
      int rc, len;

      memset(result, 0, 4096);
      snprintf(cmd_buf, 1024, "%s %s", TSA_HA_CLI, cmd);
      buf = run_shell_cmnd(cmd_buf, &rc, &len); 
      if ( buf ) {
            strncpy(result, buf, 4096);   
            free(buf);
      }
      return result;
}

/* eventd way:
      talk to event daemon with IPC
*/
char*
process_cmnd_eventd(const char *cmd)
{
      static char buf[1024];  /* to avoid memory leak */
      struct ha_msg *cmd_msg = NULL, *msg = NULL;
      IPC_Channel * ipc_chan = NULL;
      const char * result = NULL;

      memset(buf, 0, 1024);
      while ( ( ipc_chan = eventd_signon()) == NULL ) {
            sleep(3);
      }

      if ( ( cmd_msg = ha_msg_new(1) ) == NULL ) {
            cl_log(LOG_ERR, "%s: alloc msg failed.", __FUNCTION__);
            goto exit;
      }
      
      ha_msg_add(cmd_msg, "cmd", cmd);
      if ( msg2ipcchan(cmd_msg, ipc_chan) != HA_OK ) {
            cl_log(LOG_ERR, "%s: send msg to ipchan failed.",
                  __FUNCTION__);
      }
      ha_msg_del(cmd_msg);
      msg = msgfromIPC(ipc_chan, MSG_ALLOWINTR);
      if ( msg && ( result = ha_msg_value(msg, "result"))) {
            strncpy(buf, result, 1024);
      }
exit:
      eventd_signoff(ipc_chan);
      return buf;
}

Generated by  Doxygen 1.6.0   Back to index