| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048 |
- /*
- * Copyright (c) 2006 Red Hat, Inc.
- *
- * All rights reserved.
- *
- * Author: Patrick Caulfield (pcaulfie@redhat.com)
- *
- * 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 CONTIBUTORS "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 <sys/types.h>
- #include <sys/socket.h>
- #include <sys/un.h>
- #include <sys/ioctl.h>
- #include <netinet/in.h>
- #include <sys/uio.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include <errno.h>
- #include <signal.h>
- #include <time.h>
- #include <unistd.h>
- #include <netinet/in.h>
- #include <arpa/inet.h>
- #include "../include/saAis.h"
- #include "../include/saClm.h"
- #include "../include/ipc_gen.h"
- #include "../include/ipc_cpg.h"
- #include "../include/mar_cpg.h"
- #include "../include/list.h"
- #include "../include/queue.h"
- #include "../lcr/lcr_comp.h"
- #include "totempg.h"
- #include "totemip.h"
- #include "main.h"
- #include "ipc.h"
- #include "mempool.h"
- #include "service.h"
- #include "jhash.h"
- #include "swab.h"
- #include "ipc.h"
- #include "print.h"
- #define GROUP_HASH_SIZE 32
- #define PI_FLAG_MEMBER 1
- enum cpg_message_req_types {
- MESSAGE_REQ_EXEC_CPG_PROCJOIN = 0,
- MESSAGE_REQ_EXEC_CPG_PROCLEAVE = 1,
- MESSAGE_REQ_EXEC_CPG_JOINLIST = 2,
- MESSAGE_REQ_EXEC_CPG_MCAST = 3
- };
- struct removed_group
- {
- struct group_info *gi;
- struct list_head list; /* on removed_list */
- int left_list_entries;
- mar_cpg_address_t left_list[PROCESSOR_COUNT_MAX];
- int left_list_size;
- };
- struct group_info {
- mar_cpg_name_t group_name;
- struct list_head members;
- struct list_head list; /* on hash list */
- struct removed_group *rg; /* when a node goes down */
- };
- struct process_info {
- unsigned int nodeid;
- uint32_t pid;
- uint32_t flags;
- void *conn;
- void *trackerconn;
- struct group_info *group;
- struct list_head list; /* on the group_info members list */
- };
- struct join_list_entry {
- uint32_t pid;
- mar_cpg_name_t group_name;
- };
- static struct list_head group_lists[GROUP_HASH_SIZE];
- /*
- * Service Interfaces required by service_message_handler struct
- */
- static void cpg_confchg_fn (
- enum totem_configuration_type configuration_type,
- unsigned int *member_list, int member_list_entries,
- unsigned int *left_list, int left_list_entries,
- unsigned int *joined_list, int joined_list_entries,
- struct memb_ring_id *ring_id);
- static int cpg_exec_init_fn (struct objdb_iface_ver0 *objdb);
- static int cpg_lib_init_fn (void *conn);
- static int cpg_lib_exit_fn (void *conn);
- static void message_handler_req_exec_cpg_procjoin (
- void *message,
- unsigned int nodeid);
- static void message_handler_req_exec_cpg_procleave (
- void *message,
- unsigned int nodeid);
- static void message_handler_req_exec_cpg_joinlist (
- void *message,
- unsigned int nodeid);
- static void message_handler_req_exec_cpg_mcast (
- void *message,
- unsigned int nodeid);
- static void exec_cpg_procjoin_endian_convert (void *msg);
- static void exec_cpg_joinlist_endian_convert (void *msg);
- static void exec_cpg_mcast_endian_convert (void *msg);
- static void message_handler_req_lib_cpg_join (void *conn, void *message);
- static void message_handler_req_lib_cpg_leave (void *conn, void *message);
- static void message_handler_req_lib_cpg_mcast (void *conn, void *message);
- static void message_handler_req_lib_cpg_membership (void *conn, void *message);
- static void message_handler_req_lib_cpg_trackstart (void *conn, void *message);
- static void message_handler_req_lib_cpg_trackstop (void *conn, void *message);
- static int cpg_node_joinleave_send (struct group_info *gi, struct process_info *pi, int fn, int reason);
- static void cpg_exec_send_joinlist(void);
- /*
- * Library Handler Definition
- */
- static struct openais_lib_handler cpg_lib_service[] =
- {
- { /* 0 */
- .lib_handler_fn = message_handler_req_lib_cpg_join,
- .response_size = sizeof (struct res_lib_cpg_join),
- .response_id = MESSAGE_RES_CPG_JOIN,
- .flow_control = OPENAIS_FLOW_CONTROL_REQUIRED
- },
- { /* 1 */
- .lib_handler_fn = message_handler_req_lib_cpg_leave,
- .response_size = sizeof (struct res_lib_cpg_leave),
- .response_id = MESSAGE_RES_CPG_LEAVE,
- .flow_control = OPENAIS_FLOW_CONTROL_REQUIRED
- },
- { /* 2 */
- .lib_handler_fn = message_handler_req_lib_cpg_mcast,
- .response_size = sizeof (mar_res_header_t),
- .response_id = MESSAGE_RES_CPG_MCAST,
- .flow_control = OPENAIS_FLOW_CONTROL_REQUIRED
- },
- { /* 3 */
- .lib_handler_fn = message_handler_req_lib_cpg_membership,
- .response_size = sizeof (mar_res_header_t),
- .response_id = MESSAGE_RES_CPG_MEMBERSHIP,
- .flow_control = OPENAIS_FLOW_CONTROL_NOT_REQUIRED
- },
- { /* 4 */
- .lib_handler_fn = message_handler_req_lib_cpg_trackstart,
- .response_size = sizeof (struct res_lib_cpg_trackstart),
- .response_id = MESSAGE_RES_CPG_TRACKSTART,
- .flow_control = OPENAIS_FLOW_CONTROL_NOT_REQUIRED
- },
- { /* 5 */
- .lib_handler_fn = message_handler_req_lib_cpg_trackstop,
- .response_size = sizeof (struct res_lib_cpg_trackstart),
- .response_id = MESSAGE_RES_CPG_TRACKSTOP,
- .flow_control = OPENAIS_FLOW_CONTROL_NOT_REQUIRED
- }
- };
- static struct openais_exec_handler cpg_exec_service[] =
- {
- { /* 0 */
- .exec_handler_fn = message_handler_req_exec_cpg_procjoin,
- .exec_endian_convert_fn = exec_cpg_procjoin_endian_convert
- },
- { /* 1 */
- .exec_handler_fn = message_handler_req_exec_cpg_procleave,
- .exec_endian_convert_fn = exec_cpg_procjoin_endian_convert
- },
- { /* 2 */
- .exec_handler_fn = message_handler_req_exec_cpg_joinlist,
- .exec_endian_convert_fn = exec_cpg_joinlist_endian_convert
- },
- { /* 3 */
- .exec_handler_fn = message_handler_req_exec_cpg_mcast,
- .exec_endian_convert_fn = exec_cpg_mcast_endian_convert
- },
- };
- struct openais_service_handler cpg_service_handler = {
- .name = (unsigned char*)"openais cluster closed process group service v1.01",
- .id = CPG_SERVICE,
- .private_data_size = sizeof (struct process_info),
- .lib_init_fn = cpg_lib_init_fn,
- .lib_exit_fn = cpg_lib_exit_fn,
- .lib_service = cpg_lib_service,
- .lib_service_count = sizeof (cpg_lib_service) / sizeof (struct openais_lib_handler),
- .exec_init_fn = cpg_exec_init_fn,
- .exec_dump_fn = NULL,
- .exec_service = cpg_exec_service,
- .exec_service_count = sizeof (cpg_exec_service) / sizeof (struct openais_exec_handler),
- .confchg_fn = cpg_confchg_fn,
- };
- /*
- * Dynamic loader definition
- */
- static struct openais_service_handler *cpg_get_service_handler_ver0 (void);
- static struct openais_service_handler_iface_ver0 cpg_service_handler_iface = {
- .openais_get_service_handler_ver0 = cpg_get_service_handler_ver0
- };
- static struct lcr_iface openais_cpg_ver0[1] = {
- {
- .name = "openais_cpg",
- .version = 0,
- .versions_replace = 0,
- .versions_replace_count = 0,
- .dependencies = 0,
- .dependency_count = 0,
- .constructor = NULL,
- .destructor = NULL,
- .interfaces = NULL
- }
- };
- static struct lcr_comp cpg_comp_ver0 = {
- .iface_count = 1,
- .ifaces = openais_cpg_ver0
- };
- static struct openais_service_handler *cpg_get_service_handler_ver0 (void)
- {
- return (&cpg_service_handler);
- }
- __attribute__ ((constructor)) static void cpg_comp_register (void) {
- lcr_interfaces_set (&openais_cpg_ver0[0], &cpg_service_handler_iface);
- lcr_component_register (&cpg_comp_ver0);
- }
- struct req_exec_cpg_procjoin {
- mar_req_header_t header __attribute__((aligned(8)));
- mar_cpg_name_t group_name __attribute__((aligned(8)));
- mar_uint32_t pid __attribute__((aligned(8)));
- mar_uint32_t reason __attribute__((aligned(8)));
- };
- struct req_exec_cpg_mcast {
- mar_req_header_t header __attribute__((aligned(8)));
- mar_cpg_name_t group_name __attribute__((aligned(8)));
- mar_uint32_t msglen __attribute__((aligned(8)));
- mar_uint32_t pid __attribute__((aligned(8)));
- mar_uint8_t message[] __attribute__((aligned(8)));
- };
- static int notify_lib_joinlist(
- struct group_info *gi,
- void *conn,
- int joined_list_entries,
- mar_cpg_address_t *joined_list,
- int left_list_entries,
- mar_cpg_address_t *left_list,
- int id)
- {
- int count = 0;
- char *buf;
- struct res_lib_cpg_confchg_callback *res;
- struct list_head *iter;
- struct list_head *tmp;
- mar_cpg_address_t *retgi;
- int size;
- /* First, we need to know how many nodes are in the list. While we're
- traversing this list, look for the 'us' entry so we know which
- connection to send back down */
- for (iter = gi->members.next; iter != &gi->members; iter = iter->next) {
- struct process_info *pi = list_entry(iter, struct process_info, list);
- if (pi->pid)
- count++;
- }
- log_printf(LOG_LEVEL_DEBUG, "Sending new joinlist (%d elements) to clients\n", count);
- size = sizeof(struct res_lib_cpg_confchg_callback) +
- sizeof(mar_cpg_address_t) * (count + left_list_entries + joined_list_entries);
- buf = alloca(size);
- if (!buf)
- return SA_AIS_ERR_NO_SPACE;
- res = (struct res_lib_cpg_confchg_callback *)buf;
- res->joined_list_entries = joined_list_entries;
- res->left_list_entries = left_list_entries;
- retgi = res->member_list;
- res->header.size = size;
- res->header.id = id;
- memcpy(&res->group_name, &gi->group_name, sizeof(mar_cpg_name_t));
- /* Build up the message */
- count = 0;
- for (iter = gi->members.next; iter != &gi->members; iter = iter->next) {
- struct process_info *pi = list_entry(iter, struct process_info, list);
- if (pi->pid) {
- /* Processes leaving will be removed AFTER this is done (so that they get their
- own leave notifications), so exclude them from the members list here */
- int i;
- for (i=0; i<left_list_entries; i++) {
- if (left_list[i].pid == pi->pid && left_list[i].nodeid == pi->nodeid)
- goto next_member;
- }
- retgi->nodeid = pi->nodeid;
- retgi->pid = pi->pid;
- retgi++;
- count++;
- next_member: ;
- }
- }
- res->member_list_entries = count;
- if (left_list_entries) {
- memcpy(retgi, left_list, left_list_entries * sizeof(mar_cpg_address_t));
- retgi += left_list_entries;
- }
- if (joined_list_entries) {
- memcpy(retgi, joined_list, joined_list_entries * sizeof(mar_cpg_address_t));
- retgi += joined_list_entries;
- }
- if (conn) {
- openais_conn_send_response(conn, buf, size);
- }
- else {
- /* Send it to all listeners */
- for (iter = gi->members.next, tmp=iter->next; iter != &gi->members; iter = tmp, tmp=iter->next) {
- struct process_info *pi = list_entry(iter, struct process_info, list);
- if (pi->trackerconn && (pi->flags & PI_FLAG_MEMBER)) {
- if (openais_conn_send_response(pi->trackerconn, buf, size) == -1) {
- // Error ??
- }
- }
- }
- }
- return SA_AIS_OK;
- }
- static void remove_group(struct group_info *gi)
- {
- list_del(&gi->list);
- free(gi);
- }
- static int cpg_exec_init_fn (struct objdb_iface_ver0 *objdb)
- {
- int i;
- log_init ("CPG");
- for (i=0; i<GROUP_HASH_SIZE; i++) {
- list_init(&group_lists[i]);
- }
- return (0);
- }
- static int cpg_lib_exit_fn (void *conn)
- {
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- struct group_info *gi = pi->group;
- mar_cpg_address_t notify_info;
- log_printf(LOG_LEVEL_DEBUG, "exit_fn for conn=%p\n", conn);
- if (gi) {
- notify_info.pid = pi->pid;
- notify_info.nodeid = this_ip->nodeid;
- notify_info.reason = CONFCHG_CPG_REASON_PROCDOWN;
- cpg_node_joinleave_send(gi, pi, MESSAGE_REQ_EXEC_CPG_PROCLEAVE, CONFCHG_CPG_REASON_PROCDOWN);
- list_del(&pi->list);
- }
- return (0);
- }
- static struct group_info *get_group(mar_cpg_name_t *name)
- {
- struct list_head *iter;
- struct group_info *gi = NULL;
- uint32_t hash = jhash(name->value, name->length, 0) % GROUP_HASH_SIZE;
- for (iter = group_lists[hash].next; iter != &group_lists[hash]; iter = iter->next) {
- gi = list_entry(iter, struct group_info, list);
- if (memcmp(gi->group_name.value, name->value, name->length) == 0)
- break;
- }
- if (!gi) {
- gi = malloc(sizeof(struct group_info));
- if (!gi) {
- log_printf(LOG_LEVEL_WARNING, "Unable to allocate group_info struct");
- return NULL;
- }
- memcpy(&gi->group_name, name, sizeof(mar_cpg_name_t));
- gi->rg = NULL;
- list_init(&gi->members);
- list_add(&gi->list, &group_lists[hash]);
- }
- return gi;
- }
- static int cpg_node_joinleave_send (struct group_info *gi, struct process_info *pi, int fn, int reason)
- {
- struct req_exec_cpg_procjoin req_exec_cpg_procjoin;
- struct iovec req_exec_cpg_iovec;
- int result;
- memcpy(&req_exec_cpg_procjoin.group_name, &gi->group_name, sizeof(mar_cpg_name_t));
- req_exec_cpg_procjoin.pid = pi->pid;
- req_exec_cpg_procjoin.reason = reason;
- req_exec_cpg_procjoin.header.size = sizeof(req_exec_cpg_procjoin);
- req_exec_cpg_procjoin.header.id = SERVICE_ID_MAKE(CPG_SERVICE, fn);
- req_exec_cpg_iovec.iov_base = &req_exec_cpg_procjoin;
- req_exec_cpg_iovec.iov_len = sizeof(req_exec_cpg_procjoin);
- result = totempg_groups_mcast_joined (openais_group_handle, &req_exec_cpg_iovec, 1, TOTEMPG_AGREED);
- return (result);
- }
- static void remove_node_from_groups(
- unsigned int nodeid,
- struct list_head *remlist)
- {
- int i;
- struct list_head *iter, *iter2, *tmp;
- struct process_info *pi;
- struct group_info *gi;
- for (i=0; i < GROUP_HASH_SIZE; i++) {
- for (iter = group_lists[i].next; iter != &group_lists[i]; iter = iter->next) {
- gi = list_entry(iter, struct group_info, list);
- for (iter2 = gi->members.next, tmp = iter2->next; iter2 != &gi->members; iter2 = tmp, tmp = iter2->next) {
- pi = list_entry(iter2, struct process_info, list);
- if (pi->nodeid == nodeid) {
- /* Add it to the list of nodes to send notifications for */
- if (!gi->rg) {
- gi->rg = malloc(sizeof(struct removed_group));
- if (gi->rg) {
- list_add(&gi->rg->list, remlist);
- gi->rg->gi = gi;
- gi->rg->left_list_entries = 0;
- gi->rg->left_list_size = PROCESSOR_COUNT_MAX;
- }
- else {
- log_printf(LOG_LEVEL_CRIT, "Unable to allocate removed group struct. CPG callbacks will be junk.");
- return;
- }
- }
- /* Do we need to increase the size ?
- * Yes, I increase this exponentially. Generally, if you've got a lot of groups,
- * you'll have a /lot/ of groups, and cgp_groupinfo is pretty small anyway
- */
- if (gi->rg->left_list_size == gi->rg->left_list_entries) {
- int newsize;
- struct removed_group *newrg;
- list_del(&gi->rg->list);
- newsize = gi->rg->left_list_size * 2;
- newrg = realloc(gi->rg, sizeof(struct removed_group) + newsize*sizeof(mar_cpg_address_t));
- if (!newrg) {
- log_printf(LOG_LEVEL_CRIT, "Unable to realloc removed group struct. CPG callbacks will be junk.");
- return;
- }
- newrg->left_list_size = newsize+PROCESSOR_COUNT_MAX;
- gi->rg = newrg;
- list_add(&gi->rg->list, remlist);
- }
- gi->rg->left_list[gi->rg->left_list_entries].pid = pi->pid;
- gi->rg->left_list[gi->rg->left_list_entries].nodeid = pi->nodeid;
- gi->rg->left_list[gi->rg->left_list_entries].reason = CONFCHG_CPG_REASON_NODEDOWN;
- gi->rg->left_list_entries++;
- /* Remove node info for dead node */
- list_del(&pi->list);
- free(pi);
- }
- }
- }
- }
- }
- static void cpg_confchg_fn (
- enum totem_configuration_type configuration_type,
- unsigned int *member_list, int member_list_entries,
- unsigned int *left_list, int left_list_entries,
- unsigned int *joined_list, int joined_list_entries,
- struct memb_ring_id *ring_id)
- {
- int i;
- struct list_head removed_list;
- log_printf(LOG_LEVEL_DEBUG, "confchg. joined_list: %d, left_list: %d\n", joined_list_entries, left_list_entries);
- list_init(&removed_list);
- /* Tell any newly joined nodes our list of joined groups */
- if (configuration_type == TOTEM_CONFIGURATION_REGULAR) {
- cpg_exec_send_joinlist();
- }
- /* Remove nodes from joined groups and add removed groups to the list */
- for (i = 0; i < left_list_entries; i++) {
- remove_node_from_groups(left_list[i], &removed_list);
- }
- if (!list_empty(&removed_list)) {
- struct list_head *iter, *tmp;
- for (iter = removed_list.next, tmp=iter->next; iter != &removed_list; iter = tmp, tmp = iter->next) {
- struct removed_group *rg = list_entry(iter, struct removed_group, list);
- notify_lib_joinlist(rg->gi, NULL,
- 0, NULL,
- rg->left_list_entries, rg->left_list,
- MESSAGE_RES_CPG_CONFCHG_CALLBACK);
- rg->gi->rg = NULL;
- free(rg);
- }
- }
- }
- /* Can byteswap join & leave messages */
- static void exec_cpg_procjoin_endian_convert (void *msg)
- {
- struct req_exec_cpg_procjoin *req_exec_cpg_procjoin = (struct req_exec_cpg_procjoin *)msg;
- req_exec_cpg_procjoin->pid = swab32(req_exec_cpg_procjoin->pid);
- req_exec_cpg_procjoin->group_name.length = swab32(req_exec_cpg_procjoin->group_name.length);
- req_exec_cpg_procjoin->reason = swab32(req_exec_cpg_procjoin->reason);
- }
- static void exec_cpg_joinlist_endian_convert (void *msg)
- {
- mar_res_header_t *res = (mar_res_header_t *)msg;
- struct join_list_entry *jle = (struct join_list_entry *)(msg + sizeof(mar_res_header_t));
- while ((void*)jle < msg + res->size) {
- jle->pid = swab32(jle->pid);
- jle->group_name.length = swab32(jle->group_name.length);
- jle++;
- }
- }
- static void exec_cpg_mcast_endian_convert (void *msg)
- {
- struct req_exec_cpg_mcast *req_exec_cpg_mcast = (struct req_exec_cpg_mcast *)msg;
- req_exec_cpg_mcast->pid = swab32(req_exec_cpg_mcast->pid);
- req_exec_cpg_mcast->msglen = swab32(req_exec_cpg_mcast->msglen);
- req_exec_cpg_mcast->group_name.length = swab32(req_exec_cpg_mcast->group_name.length);
- }
- static void do_proc_join(
- mar_cpg_name_t *name,
- uint32_t pid,
- unsigned int nodeid,
- int reason)
- {
- struct group_info *gi;
- struct process_info *pi;
- struct list_head *iter;
- mar_cpg_address_t notify_info;
- gi = get_group(name); /* this will always succeed ! */
- assert(gi);
- /* See if it already exists in this group */
- for (iter = gi->members.next; iter != &gi->members; iter = iter->next) {
- pi = list_entry(iter, struct process_info, list);
- if (pi->pid == pid && pi->nodeid == nodeid) {
- /* It could be a local join message */
- if ((nodeid == this_ip->nodeid) &&
- (!pi->flags & PI_FLAG_MEMBER)) {
- goto local_join;
- } else {
- return;
- }
- }
- }
- pi = malloc(sizeof(struct process_info));
- if (!pi) {
- log_printf(LOG_LEVEL_WARNING, "Unable to allocate process_info struct");
- return;
- }
- pi->nodeid = nodeid;
- pi->pid = pid;
- pi->group = gi;
- pi->conn = NULL;
- pi->trackerconn = NULL;
- list_add_tail(&pi->list, &gi->members);
- local_join:
- pi->flags = PI_FLAG_MEMBER;
- notify_info.pid = pi->pid;
- notify_info.nodeid = nodeid;
- notify_info.reason = reason;
- notify_lib_joinlist(gi, NULL,
- 1, ¬ify_info,
- 0, NULL,
- MESSAGE_RES_CPG_CONFCHG_CALLBACK);
- }
- static void message_handler_req_exec_cpg_procjoin (
- void *message,
- unsigned int nodeid)
- {
- struct req_exec_cpg_procjoin *req_exec_cpg_procjoin = (struct req_exec_cpg_procjoin *)message;
- log_printf(LOG_LEVEL_DEBUG, "got procjoin message from cluster\n");
- do_proc_join(&req_exec_cpg_procjoin->group_name,
- req_exec_cpg_procjoin->pid, nodeid,
- CONFCHG_CPG_REASON_JOIN);
- }
- static void message_handler_req_exec_cpg_procleave (
- void *message,
- unsigned int nodeid)
- {
- struct req_exec_cpg_procjoin *req_exec_cpg_procjoin = (struct req_exec_cpg_procjoin *)message;
- struct group_info *gi;
- struct process_info *pi;
- struct list_head *iter;
- mar_cpg_address_t notify_info;
- log_printf(LOG_LEVEL_DEBUG, "got procleave message from cluster\n");
- gi = get_group(&req_exec_cpg_procjoin->group_name); /* this will always succeed ! */
- assert(gi);
- notify_info.pid = req_exec_cpg_procjoin->pid;
- notify_info.nodeid = nodeid;
- notify_info.reason = req_exec_cpg_procjoin->reason;
- notify_lib_joinlist(gi, NULL,
- 0, NULL,
- 1, ¬ify_info,
- MESSAGE_RES_CPG_CONFCHG_CALLBACK);
- /* Find the node/PID to remove */
- for (iter = gi->members.next; iter != &gi->members; iter = iter->next) {
- pi = list_entry(iter, struct process_info, list);
- if (pi->pid == req_exec_cpg_procjoin->pid &&
- pi->nodeid == nodeid) {
- list_del(&pi->list);
- if (!pi->conn)
- free(pi);
- if (list_empty(&gi->members)) {
- remove_group(gi);
- }
- break;
- }
- }
- }
- /* Got a proclist from another node */
- static void message_handler_req_exec_cpg_joinlist (
- void *message,
- unsigned int nodeid)
- {
- mar_res_header_t *res = (mar_res_header_t *)message;
- struct join_list_entry *jle = (struct join_list_entry *)(message + sizeof(mar_res_header_t));
- log_printf(LOG_LEVEL_NOTICE, "got joinlist message from node %x\n",
- nodeid);
- /* Ignore our own messages */
- if (nodeid == this_ip->nodeid) {
- return;
- }
- while ((void*)jle < message + res->size) {
- do_proc_join(&jle->group_name, jle->pid, nodeid,
- CONFCHG_CPG_REASON_NODEUP);
- jle++;
- }
- }
- static void message_handler_req_exec_cpg_mcast (
- void *message,
- unsigned int nodeid)
- {
- struct req_exec_cpg_mcast *req_exec_cpg_mcast = (struct req_exec_cpg_mcast *)message;
- struct res_lib_cpg_deliver_callback *res_lib_cpg_mcast;
- int msglen = req_exec_cpg_mcast->msglen;
- char buf[sizeof(*res_lib_cpg_mcast) + msglen];
- struct group_info *gi;
- struct list_head *iter;
- gi = get_group(&req_exec_cpg_mcast->group_name); /* this will always succeed ! */
- assert(gi);
- res_lib_cpg_mcast = (struct res_lib_cpg_deliver_callback *)buf;
- res_lib_cpg_mcast->header.id = MESSAGE_RES_CPG_DELIVER_CALLBACK;
- res_lib_cpg_mcast->header.size = sizeof(*res_lib_cpg_mcast) + msglen;
- res_lib_cpg_mcast->msglen = msglen;
- res_lib_cpg_mcast->pid = req_exec_cpg_mcast->pid;
- res_lib_cpg_mcast->nodeid = nodeid;
- memcpy(&res_lib_cpg_mcast->group_name, &gi->group_name,
- sizeof(mar_cpg_name_t));
- memcpy(&res_lib_cpg_mcast->message, (char*)message+sizeof(*req_exec_cpg_mcast),
- msglen);
- /* Send to all interested members */
- for (iter = gi->members.next; iter != &gi->members; iter = iter->next) {
- struct process_info *pi = list_entry(iter, struct process_info, list);
- if (pi->trackerconn) {
- openais_conn_send_response(
- pi->trackerconn,
- buf,
- res_lib_cpg_mcast->header.size);
- }
- }
- }
- static void cpg_exec_send_joinlist(void)
- {
- int count = 0;
- char *buf;
- int i;
- struct list_head *iter;
- struct list_head *iter2;
- struct group_info *gi;
- mar_res_header_t *res;
- struct join_list_entry *jle;
- struct iovec req_exec_cpg_iovec;
- log_printf(LOG_LEVEL_DEBUG, "sending joinlist to cluster\n");
- /* Count the number of groups we are a member of */
- for (i=0; i<GROUP_HASH_SIZE; i++) {
- for (iter = group_lists[i].next; iter != &group_lists[i]; iter = iter->next) {
- gi = list_entry(iter, struct group_info, list);
- for (iter2 = gi->members.next; iter2 != &gi->members; iter2 = iter2->next) {
- struct process_info *pi = list_entry(iter2, struct process_info, list);
- if (pi->pid && pi->nodeid == this_ip->nodeid) {
- count++;
- }
- }
- }
- }
- /* Nothing to send */
- if (!count)
- return;
- buf = alloca(sizeof(mar_res_header_t) + sizeof(struct join_list_entry) * count);
- if (!buf) {
- log_printf(LOG_LEVEL_WARNING, "Unable to allocate joinlist buffer");
- return;
- }
- jle = (struct join_list_entry *)(buf + sizeof(mar_res_header_t));
- res = (mar_res_header_t *)buf;
- for (i=0; i<GROUP_HASH_SIZE; i++) {
- for (iter = group_lists[i].next; iter != &group_lists[i]; iter = iter->next) {
- gi = list_entry(iter, struct group_info, list);
- for (iter2 = gi->members.next; iter2 != &gi->members; iter2 = iter2->next) {
- struct process_info *pi = list_entry(iter2, struct process_info, list);
- if (pi->pid && pi->nodeid == this_ip->nodeid) {
- memcpy(&jle->group_name, &gi->group_name, sizeof(mar_cpg_name_t));
- jle->pid = pi->pid;
- jle++;
- }
- }
- }
- }
- res->id = SERVICE_ID_MAKE(CPG_SERVICE, MESSAGE_REQ_EXEC_CPG_JOINLIST);
- res->size = sizeof(mar_res_header_t)+sizeof(struct join_list_entry) * count;
- req_exec_cpg_iovec.iov_base = buf;
- req_exec_cpg_iovec.iov_len = res->size;
- totempg_groups_mcast_joined (openais_group_handle, &req_exec_cpg_iovec, 1, TOTEMPG_AGREED);
- }
- static int cpg_lib_init_fn (void *conn)
- {
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- pi->conn = conn;
- log_printf(LOG_LEVEL_DEBUG, "lib_init_fn: conn=%p, pi=%p\n", conn, pi);
- return (0);
- }
- /* Join message from the library */
- static void message_handler_req_lib_cpg_join (void *conn, void *message)
- {
- struct req_lib_cpg_join *req_lib_cpg_join = (struct req_lib_cpg_join *)message;
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- struct res_lib_cpg_join res_lib_cpg_join;
- struct group_info *gi;
- SaAisErrorT error = SA_AIS_OK;
- log_printf(LOG_LEVEL_DEBUG, "got join request on %p, pi=%p, pi->pid=%d\n", conn, pi, pi->pid);
- /* Already joined on this conn */
- if (pi->pid) {
- error = SA_AIS_ERR_INVALID_PARAM;
- goto join_err;
- }
- gi = get_group(&req_lib_cpg_join->group_name);
- if (!gi) {
- error = SA_AIS_ERR_NO_SPACE;
- goto join_err;
- }
- /* Add a node entry for us */
- pi->nodeid = this_ip->nodeid;
- pi->pid = req_lib_cpg_join->pid;
- pi->group = gi;
- list_add(&pi->list, &gi->members);
- /* Tell the rest of the cluster */
- cpg_node_joinleave_send(gi, pi, MESSAGE_REQ_EXEC_CPG_PROCJOIN, CONFCHG_CPG_REASON_JOIN);
- join_err:
- res_lib_cpg_join.header.size = sizeof(res_lib_cpg_join);
- res_lib_cpg_join.header.id = MESSAGE_RES_CPG_JOIN;
- res_lib_cpg_join.header.error = error;
- openais_conn_send_response(conn, &res_lib_cpg_join, sizeof(res_lib_cpg_join));
- }
- /* Leave message from the library */
- static void message_handler_req_lib_cpg_leave (void *conn, void *message)
- {
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- struct res_lib_cpg_leave res_lib_cpg_leave;
- struct group_info *gi;
- SaAisErrorT error = SA_AIS_OK;
- log_printf(LOG_LEVEL_DEBUG, "got leave request on %p\n", conn);
- if (!pi || !pi->pid || !pi->group) {
- error = SA_AIS_ERR_INVALID_PARAM;
- goto leave_ret;
- }
- gi = pi->group;
- /* Tell other nodes we are leaving.
- When we get this message back we will leave too */
- cpg_node_joinleave_send(gi, pi, MESSAGE_REQ_EXEC_CPG_PROCLEAVE, CONFCHG_CPG_REASON_LEAVE);
- pi->group = NULL;
- leave_ret:
- /* send return */
- res_lib_cpg_leave.header.size = sizeof(res_lib_cpg_leave);
- res_lib_cpg_leave.header.id = MESSAGE_RES_CPG_LEAVE;
- res_lib_cpg_leave.header.error = error;
- openais_conn_send_response(conn, &res_lib_cpg_leave, sizeof(res_lib_cpg_leave));
- }
- /* Mcast message from the library */
- static void message_handler_req_lib_cpg_mcast (void *conn, void *message)
- {
- struct req_lib_cpg_mcast *req_lib_cpg_mcast = (struct req_lib_cpg_mcast *)message;
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- struct group_info *gi = pi->group;
- struct iovec req_exec_cpg_iovec[2];
- struct req_exec_cpg_mcast req_exec_cpg_mcast;
- mar_res_header_t res;
- int msglen = req_lib_cpg_mcast->msglen;
- int result;
- log_printf(LOG_LEVEL_DEBUG, "got mcast request on %p\n", conn);
- /* Can't send if we're not joined */
- if (!gi) {
- res.size = sizeof(res);
- res.id = MESSAGE_RES_CPG_MCAST;
- res.error = SA_AIS_ERR_ACCESS; /* TODO Better error code ?? */
- openais_conn_send_response(conn, &res, sizeof(res));
- return;
- }
- req_exec_cpg_mcast.header.size = sizeof(req_exec_cpg_mcast) + msglen;
- req_exec_cpg_mcast.header.id = SERVICE_ID_MAKE(CPG_SERVICE,
- MESSAGE_REQ_EXEC_CPG_MCAST);
- req_exec_cpg_mcast.pid = pi->pid;
- req_exec_cpg_mcast.msglen = msglen;
- memcpy(&req_exec_cpg_mcast.group_name, &gi->group_name,
- sizeof(mar_cpg_name_t));
- req_exec_cpg_iovec[0].iov_base = &req_exec_cpg_mcast;
- req_exec_cpg_iovec[0].iov_len = sizeof(req_exec_cpg_mcast);
- req_exec_cpg_iovec[1].iov_base = &req_lib_cpg_mcast->message;
- req_exec_cpg_iovec[1].iov_len = msglen;
- // TODO: guarantee type...
- result = totempg_groups_mcast_joined (openais_group_handle, req_exec_cpg_iovec, 2, TOTEMPG_AGREED);
- res.size = sizeof(res);
- res.id = MESSAGE_RES_CPG_MCAST;
- res.error = SA_AIS_OK;
- openais_conn_send_response(conn, &res, sizeof(res));
- }
- static void message_handler_req_lib_cpg_membership (void *conn, void *message)
- {
- struct process_info *pi = (struct process_info *)openais_conn_private_data_get (conn);
- log_printf(LOG_LEVEL_DEBUG, "got membership request on %p\n", conn);
- if (!pi->group) {
- mar_res_header_t res;
- res.size = sizeof(res);
- res.id = MESSAGE_RES_CPG_MEMBERSHIP;
- res.error = SA_AIS_ERR_ACCESS; /* TODO Better error code */
- openais_conn_send_response(conn, &res, sizeof(res));
- return;
- }
- notify_lib_joinlist(pi->group, conn, 0, NULL, 0, NULL, MESSAGE_RES_CPG_MEMBERSHIP);
- }
- static void message_handler_req_lib_cpg_trackstart (void *conn, void *message)
- {
- struct req_lib_cpg_trackstart *req_lib_cpg_trackstart = (struct req_lib_cpg_trackstart *)message;
- struct res_lib_cpg_trackstart res_lib_cpg_trackstart;
- struct group_info *gi;
- struct process_info *otherpi;
- void *otherconn;
- SaAisErrorT error = SA_AIS_OK;
- log_printf(LOG_LEVEL_DEBUG, "got trackstart request on %p\n", conn);
- gi = get_group(&req_lib_cpg_trackstart->group_name);
- if (!gi) {
- error = SA_AIS_ERR_NO_SPACE;
- goto tstart_ret;
- }
- /* Find the partner connection and add us to it's process_info struct */
- otherconn = openais_conn_partner_get (conn);
- otherpi = (struct process_info *)openais_conn_private_data_get (conn);
- otherpi->trackerconn = conn;
- tstart_ret:
- res_lib_cpg_trackstart.header.size = sizeof(res_lib_cpg_trackstart);
- res_lib_cpg_trackstart.header.id = MESSAGE_RES_CPG_TRACKSTART;
- res_lib_cpg_trackstart.header.error = SA_AIS_OK;
- openais_conn_send_response(conn, &res_lib_cpg_trackstart, sizeof(res_lib_cpg_trackstart));
- }
- static void message_handler_req_lib_cpg_trackstop (void *conn, void *message)
- {
- struct req_lib_cpg_trackstop *req_lib_cpg_trackstop = (struct req_lib_cpg_trackstop *)message;
- struct res_lib_cpg_trackstop res_lib_cpg_trackstop;
- struct process_info *otherpi;
- void *otherconn;
- struct group_info *gi;
- SaAisErrorT error = SA_AIS_OK;
- log_printf(LOG_LEVEL_DEBUG, "got trackstop request on %p\n", conn);
- gi = get_group(&req_lib_cpg_trackstop->group_name);
- if (!gi) {
- error = SA_AIS_ERR_NO_SPACE;
- goto tstop_ret;
- }
- /* Find the partner connection and add us to it's process_info struct */
- otherconn = openais_conn_partner_get (conn);
- otherpi = (struct process_info *)openais_conn_private_data_get (conn);
- otherpi->trackerconn = NULL;
- tstop_ret:
- res_lib_cpg_trackstop.header.size = sizeof(res_lib_cpg_trackstop);
- res_lib_cpg_trackstop.header.id = MESSAGE_RES_CPG_TRACKSTOP;
- res_lib_cpg_trackstop.header.error = SA_AIS_OK;
- openais_conn_send_response(conn, &res_lib_cpg_trackstop.header, sizeof(res_lib_cpg_trackstop));
- }
|