1// SPDX-License-Identifier: GPL-2.0-or-later
   2/*
   3 * dlmdomain.c
   4 *
   5 * defines domain join / leave apis
   6 *
   7 * Copyright (C) 2004 Oracle.  All rights reserved.
   8 */
   9
  10#include <linux/module.h>
  11#include <linux/types.h>
  12#include <linux/slab.h>
  13#include <linux/highmem.h>
  14#include <linux/init.h>
  15#include <linux/spinlock.h>
  16#include <linux/delay.h>
  17#include <linux/err.h>
  18#include <linux/debugfs.h>
  19#include <linux/sched/signal.h>
  20
  21#include "../cluster/heartbeat.h"
  22#include "../cluster/nodemanager.h"
  23#include "../cluster/tcp.h"
  24
  25#include "dlmapi.h"
  26#include "dlmcommon.h"
  27#include "dlmdomain.h"
  28#include "dlmdebug.h"
  29
  30#define MLOG_MASK_PREFIX (ML_DLM|ML_DLM_DOMAIN)
  31#include "../cluster/masklog.h"
  32
  33/*
  34 * ocfs2 node maps are array of long int, which limits to send them freely
  35 * across the wire due to endianness issues. To workaround this, we convert
  36 * long ints to byte arrays. Following 3 routines are helper functions to
  37 * set/test/copy bits within those array of bytes
  38 */
  39static inline void byte_set_bit(u8 nr, u8 map[])
  40{
  41	map[nr >> 3] |= (1UL << (nr & 7));
  42}
  43
  44static inline int byte_test_bit(u8 nr, u8 map[])
  45{
  46	return ((1UL << (nr & 7)) & (map[nr >> 3])) != 0;
  47}
  48
  49static inline void byte_copymap(u8 dmap[], unsigned long smap[],
  50			unsigned int sz)
  51{
  52	unsigned int nn;
  53
  54	if (!sz)
  55		return;
  56
  57	memset(dmap, 0, ((sz + 7) >> 3));
  58	for (nn = 0 ; nn < sz; nn++)
  59		if (test_bit(nn, smap))
  60			byte_set_bit(nn, dmap);
  61}
  62
  63static void dlm_free_pagevec(void **vec, int pages)
  64{
  65	while (pages--)
  66		free_page((unsigned long)vec[pages]);
  67	kfree(vec);
  68}
  69
  70static void **dlm_alloc_pagevec(int pages)
  71{
  72	void **vec = kmalloc_array(pages, sizeof(void *), GFP_KERNEL);
  73	int i;
  74
  75	if (!vec)
  76		return NULL;
  77
  78	for (i = 0; i < pages; i++)
  79		if (!(vec[i] = (void *)__get_free_page(GFP_KERNEL)))
  80			goto out_free;
  81
  82	mlog(0, "Allocated DLM hash pagevec; %d pages (%lu expected), %lu buckets per page\n",
  83	     pages, (unsigned long)DLM_HASH_PAGES,
  84	     (unsigned long)DLM_BUCKETS_PER_PAGE);
  85	return vec;
  86out_free:
  87	dlm_free_pagevec(vec, i);
  88	return NULL;
  89}
  90
  91/*
  92 *
  93 * spinlock lock ordering: if multiple locks are needed, obey this ordering:
  94 *    dlm_domain_lock
  95 *    struct dlm_ctxt->spinlock
  96 *    struct dlm_lock_resource->spinlock
  97 *    struct dlm_ctxt->master_lock
  98 *    struct dlm_ctxt->ast_lock
  99 *    dlm_master_list_entry->spinlock
 100 *    dlm_lock->spinlock
 101 *
 102 */
 103
 104DEFINE_SPINLOCK(dlm_domain_lock);
 105LIST_HEAD(dlm_domains);
 106static DECLARE_WAIT_QUEUE_HEAD(dlm_domain_events);
 107
 108/*
 109 * The supported protocol version for DLM communication.  Running domains
 110 * will have a negotiated version with the same major number and a minor
 111 * number equal or smaller.  The dlm_ctxt->dlm_locking_proto field should
 112 * be used to determine what a running domain is actually using.
 113 *
 114 * New in version 1.1:
 115 *	- Message DLM_QUERY_REGION added to support global heartbeat
 116 *	- Message DLM_QUERY_NODEINFO added to allow online node removes
 117 * New in version 1.2:
 118 * 	- Message DLM_BEGIN_EXIT_DOMAIN_MSG added to mark start of exit domain
 119 * New in version 1.3:
 120 *	- Message DLM_DEREF_LOCKRES_DONE added to inform non-master that the
 121 *	  refmap is cleared
 122 */
 123static const struct dlm_protocol_version dlm_protocol = {
 124	.pv_major = 1,
 125	.pv_minor = 3,
 126};
 127
 128#define DLM_DOMAIN_BACKOFF_MS 200
 129
 130static int dlm_query_join_handler(struct o2net_msg *msg, u32 len, void *data,
 131				  void **ret_data);
 132static int dlm_assert_joined_handler(struct o2net_msg *msg, u32 len, void *data,
 133				     void **ret_data);
 134static int dlm_cancel_join_handler(struct o2net_msg *msg, u32 len, void *data,
 135				   void **ret_data);
 136static int dlm_query_region_handler(struct o2net_msg *msg, u32 len,
 137				    void *data, void **ret_data);
 138static int dlm_exit_domain_handler(struct o2net_msg *msg, u32 len, void *data,
 139				   void **ret_data);
 140static int dlm_protocol_compare(struct dlm_protocol_version *existing,
 141				struct dlm_protocol_version *request);
 142
 143static void dlm_unregister_domain_handlers(struct dlm_ctxt *dlm);
 144
 145void __dlm_unhash_lockres(struct dlm_ctxt *dlm, struct dlm_lock_resource *res)
 146{
 147	if (hlist_unhashed(&res->hash_node))
 148		return;
 149
 150	mlog(0, "%s: Unhash res %.*s\n", dlm->name, res->lockname.len,
 151	     res->lockname.name);
 152	hlist_del_init(&res->hash_node);
 153	dlm_lockres_put(res);
 154}
 155
 156void __dlm_insert_lockres(struct dlm_ctxt *dlm, struct dlm_lock_resource *res)
 157{
 158	struct hlist_head *bucket;
 159
 160	assert_spin_locked(&dlm->spinlock);
 161
 162	bucket = dlm_lockres_hash(dlm, res->lockname.hash);
 163
 164	/* get a reference for our hashtable */
 165	dlm_lockres_get(res);
 166
 167	hlist_add_head(&res->hash_node, bucket);
 168
 169	mlog(0, "%s: Hash res %.*s\n", dlm->name, res->lockname.len,
 170	     res->lockname.name);
 171}
 172
 173struct dlm_lock_resource * __dlm_lookup_lockres_full(struct dlm_ctxt *dlm,
 174						     const char *name,
 175						     unsigned int len,
 176						     unsigned int hash)
 177{
 178	struct hlist_head *bucket;
 179	struct dlm_lock_resource *res;
 180
 181	mlog(0, "%.*s\n", len, name);
 182
 183	assert_spin_locked(&dlm->spinlock);
 184
 185	bucket = dlm_lockres_hash(dlm, hash);
 186
 187	hlist_for_each_entry(res, bucket, hash_node) {
 188		if (res->lockname.name[0] != name[0])
 189			continue;
 190		if (unlikely(res->lockname.len != len))
 191			continue;
 192		if (memcmp(res->lockname.name + 1, name + 1, len - 1))
 193			continue;
 194		dlm_lockres_get(res);
 195		return res;
 196	}
 197	return NULL;
 198}
 199
 200/* intended to be called by functions which do not care about lock
 201 * resources which are being purged (most net _handler functions).
 202 * this will return NULL for any lock resource which is found but
 203 * currently in the process of dropping its mastery reference.
 204 * use __dlm_lookup_lockres_full when you need the lock resource
 205 * regardless (e.g. dlm_get_lock_resource) */
 206struct dlm_lock_resource * __dlm_lookup_lockres(struct dlm_ctxt *dlm,
 207						const char *name,
 208						unsigned int len,
 209						unsigned int hash)
 210{
 211	struct dlm_lock_resource *res = NULL;
 212
 213	mlog(0, "%.*s\n", len, name);
 214
 215	assert_spin_locked(&dlm->spinlock);
 216
 217	res = __dlm_lookup_lockres_full(dlm, name, len, hash);
 218	if (res) {
 219		spin_lock(&res->spinlock);
 220		if (res->state & DLM_LOCK_RES_DROPPING_REF) {
 221			spin_unlock(&res->spinlock);
 222			dlm_lockres_put(res);
 223			return NULL;
 224		}
 225		spin_unlock(&res->spinlock);
 226	}
 227
 228	return res;
 229}
 230
 231struct dlm_lock_resource * dlm_lookup_lockres(struct dlm_ctxt *dlm,
 232				    const char *name,
 233				    unsigned int len)
 234{
 235	struct dlm_lock_resource *res;
 236	unsigned int hash = dlm_lockid_hash(name, len);
 237
 238	spin_lock(&dlm->spinlock);
 239	res = __dlm_lookup_lockres(dlm, name, len, hash);
 240	spin_unlock(&dlm->spinlock);
 241	return res;
 242}
 243
 244static struct dlm_ctxt * __dlm_lookup_domain_full(const char *domain, int len)
 245{
 246	struct dlm_ctxt *tmp;
 247
 248	assert_spin_locked(&dlm_domain_lock);
 249
 250	/* tmp->name here is always NULL terminated,
 251	 * but domain may not be! */
 252	list_for_each_entry(tmp, &dlm_domains, list) {
 253		if (strlen(tmp->name) == len &&
 254		    memcmp(tmp->name, domain, len)==0)
 255			return tmp;
 256	}
 257
 258	return NULL;
 259}
 260
 261/* For null terminated domain strings ONLY */
 262static struct dlm_ctxt * __dlm_lookup_domain(const char *domain)
 263{
 264	assert_spin_locked(&dlm_domain_lock);
 265
 266	return __dlm_lookup_domain_full(domain, strlen(domain));
 267}
 268
 269
 270/* returns true on one of two conditions:
 271 * 1) the domain does not exist
 272 * 2) the domain exists and it's state is "joined" */
 273static int dlm_wait_on_domain_helper(const char *domain)
 274{
 275	int ret = 0;
 276	struct dlm_ctxt *tmp = NULL;
 277
 278	spin_lock(&dlm_domain_lock);
 279
 280	tmp = __dlm_lookup_domain(domain);
 281	if (!tmp)
 282		ret = 1;
 283	else if (tmp->dlm_state == DLM_CTXT_JOINED)
 284		ret = 1;
 285
 286	spin_unlock(&dlm_domain_lock);
 287	return ret;
 288}
 289
 290static void dlm_free_ctxt_mem(struct dlm_ctxt *dlm)
 291{
 292	dlm_destroy_debugfs_subroot(dlm);
 293
 294	if (dlm->lockres_hash)
 295		dlm_free_pagevec((void **)dlm->lockres_hash, DLM_HASH_PAGES);
 296
 297	if (dlm->master_hash)
 298		dlm_free_pagevec((void **)dlm->master_hash, DLM_HASH_PAGES);
 299
 300	kfree(dlm->name);
 301	kfree(dlm);
 302}
 303
 304/* A little strange - this function will be called while holding
 305 * dlm_domain_lock and is expected to be holding it on the way out. We
 306 * will however drop and reacquire it multiple times */
 307static void dlm_ctxt_release(struct kref *kref)
 308{
 309	struct dlm_ctxt *dlm;
 310
 311	dlm = container_of(kref, struct dlm_ctxt, dlm_refs);
 312
 313	BUG_ON(dlm->num_joins);
 314	BUG_ON(dlm->dlm_state == DLM_CTXT_JOINED);
 315
 316	/* we may still be in the list if we hit an error during join. */
 317	list_del_init(&dlm->list);
 318
 319	spin_unlock(&dlm_domain_lock);
 320
 321	mlog(0, "freeing memory from domain %s\n", dlm->name);
 322
 323	wake_up(&dlm_domain_events);
 324
 325	dlm_free_ctxt_mem(dlm);
 326
 327	spin_lock(&dlm_domain_lock);
 328}
 329
 330void dlm_put(struct dlm_ctxt *dlm)
 331{
 332	spin_lock(&dlm_domain_lock);
 333	kref_put(&dlm->dlm_refs, dlm_ctxt_release);
 334	spin_unlock(&dlm_domain_lock);
 335}
 336
 337static void __dlm_get(struct dlm_ctxt *dlm)
 338{
 339	kref_get(&dlm->dlm_refs);
 340}
 341
 342/* given a questionable reference to a dlm object, gets a reference if
 343 * it can find it in the list, otherwise returns NULL in which case
 344 * you shouldn't trust your pointer. */
 345struct dlm_ctxt *dlm_grab(struct dlm_ctxt *dlm)
 346{
 347	struct dlm_ctxt *target;
 348	struct dlm_ctxt *ret = NULL;
 349
 350	spin_lock(&dlm_domain_lock);
 351
 352	list_for_each_entry(target, &dlm_domains, list) {
 353		if (target == dlm) {
 354			__dlm_get(target);
 355			ret = target;
 356			break;
 357		}
 358	}
 359
 360	spin_unlock(&dlm_domain_lock);
 361
 362	return ret;
 363}
 364
 365int dlm_domain_fully_joined(struct dlm_ctxt *dlm)
 366{
 367	int ret;
 368
 369	spin_lock(&dlm_domain_lock);
 370	ret = (dlm->dlm_state == DLM_CTXT_JOINED) ||
 371		(dlm->dlm_state == DLM_CTXT_IN_SHUTDOWN);
 372	spin_unlock(&dlm_domain_lock);
 373
 374	return ret;
 375}
 376
 377static void dlm_destroy_dlm_worker(struct dlm_ctxt *dlm)
 378{
 379	if (dlm->dlm_worker) {
 380		destroy_workqueue(dlm->dlm_worker);
 381		dlm->dlm_worker = NULL;
 382	}
 383}
 384
 385static void dlm_complete_dlm_shutdown(struct dlm_ctxt *dlm)
 386{
 387	dlm_unregister_domain_handlers(dlm);
 388	dlm_complete_thread(dlm);
 389	dlm_complete_recovery_thread(dlm);
 390	dlm_destroy_dlm_worker(dlm);
 391
 392	/* We've left the domain. Now we can take ourselves out of the
 393	 * list and allow the kref stuff to help us free the
 394	 * memory. */
 395	spin_lock(&dlm_domain_lock);
 396	list_del_init(&dlm->list);
 397	spin_unlock(&dlm_domain_lock);
 398
 399	/* Wake up anyone waiting for us to remove this domain */
 400	wake_up(&dlm_domain_events);
 401}
 402
 403static int dlm_migrate_all_locks(struct dlm_ctxt *dlm)
 404{
 405	int i, num, n, ret = 0;
 406	struct dlm_lock_resource *res;
 407	struct hlist_node *iter;
 408	struct hlist_head *bucket;
 409	int dropped;
 410
 411	mlog(0, "Migrating locks from domain %s\n", dlm->name);
 412
 413	num = 0;
 414	spin_lock(&dlm->spinlock);
 415	for (i = 0; i < DLM_HASH_BUCKETS; i++) {
 416redo_bucket:
 417		n = 0;
 418		bucket = dlm_lockres_hash(dlm, i);
 419		iter = bucket->first;
 420		while (iter) {
 421			n++;
 422			res = hlist_entry(iter, struct dlm_lock_resource,
 423					  hash_node);
 424			dlm_lockres_get(res);
 425			/* migrate, if necessary.  this will drop the dlm
 426			 * spinlock and retake it if it does migration. */
 427			dropped = dlm_empty_lockres(dlm, res);
 428
 429			spin_lock(&res->spinlock);
 430			if (dropped)
 431				__dlm_lockres_calc_usage(dlm, res);
 432			else
 433				iter = res->hash_node.next;
 434			spin_unlock(&res->spinlock);
 435
 436			dlm_lockres_put(res);
 437
 438			if (dropped) {
 439				cond_resched_lock(&dlm->spinlock);
 440				goto redo_bucket;
 441			}
 442		}
 443		cond_resched_lock(&dlm->spinlock);
 444		num += n;
 445	}
 446
 447	if (!num) {
 448		if (dlm->reco.state & DLM_RECO_STATE_ACTIVE) {
 449			mlog(0, "%s: perhaps there are more lock resources "
 450			     "need to be migrated after dlm recovery\n", dlm->name);
 451			ret = -EAGAIN;
 452		} else {
 453			mlog(0, "%s: we won't do dlm recovery after migrating "
 454			     "all lock resources\n", dlm->name);
 455			dlm->migrate_done = 1;
 456		}
 457	}
 458
 459	spin_unlock(&dlm->spinlock);
 460	wake_up(&dlm->dlm_thread_wq);
 461
 462	/* let the dlm thread take care of purging, keep scanning until
 463	 * nothing remains in the hash */
 464	if (num) {
 465		mlog(0, "%s: %d lock resources in hash last pass\n",
 466		     dlm->name, num);
 467		ret = -EAGAIN;
 468	}
 469	mlog(0, "DONE Migrating locks from domain %s\n", dlm->name);
 470	return ret;
 471}
 472
 473static int dlm_no_joining_node(struct dlm_ctxt *dlm)
 474{
 475	int ret;
 476
 477	spin_lock(&dlm->spinlock);
 478	ret = dlm->joining_node == DLM_LOCK_RES_OWNER_UNKNOWN;
 479	spin_unlock(&dlm->spinlock);
 480
 481	return ret;
 482}
 483
 484static int dlm_begin_exit_domain_handler(struct o2net_msg *msg, u32 len,
 485					 void *data, void **ret_data)
 486{
 487	struct dlm_ctxt *dlm = data;
 488	unsigned int node;
 489	struct dlm_exit_domain *exit_msg = (struct dlm_exit_domain *) msg->buf;
 490
 491	if (!dlm_grab(dlm))
 492		return 0;
 493
 494	node = exit_msg->node_idx;
 495	mlog(0, "%s: Node %u sent a begin exit domain message\n", dlm->name, node);
 496
 497	spin_lock(&dlm->spinlock);
 498	set_bit(node, dlm->exit_domain_map);
 499	spin_unlock(&dlm->spinlock);
 500
 501	dlm_put(dlm);
 502
 503	return 0;
 504}
 505
 506static void dlm_mark_domain_leaving(struct dlm_ctxt *dlm)
 507{
 508	/* Yikes, a double spinlock! I need domain_lock for the dlm
 509	 * state and the dlm spinlock for join state... Sorry! */
 510again:
 511	spin_lock(&dlm_domain_lock);
 512	spin_lock(&dlm->spinlock);
 513
 514	if (dlm->joining_node != DLM_LOCK_RES_OWNER_UNKNOWN) {
 515		mlog(0, "Node %d is joining, we wait on it.\n",
 516			  dlm->joining_node);
 517		spin_unlock(&dlm->spinlock);
 518		spin_unlock(&dlm_domain_lock);
 519
 520		wait_event(dlm->dlm_join_events, dlm_no_joining_node(dlm));
 521		goto again;
 522	}
 523
 524	dlm->dlm_state = DLM_CTXT_LEAVING;
 525	spin_unlock(&dlm->spinlock);
 526	spin_unlock(&dlm_domain_lock);
 527}
 528
 529static void __dlm_print_nodes(struct dlm_ctxt *dlm)
 530{
 531	int node = -1, num = 0;
 532
 533	assert_spin_locked(&dlm->spinlock);
 534
 535	printk("( ");
 536	while ((node = find_next_bit(dlm->domain_map, O2NM_MAX_NODES,
 537				     node + 1)) < O2NM_MAX_NODES) {
 538		printk("%d ", node);
 539		++num;
 540	}
 541	printk(") %u nodes\n", num);
 542}
 543
 544static int dlm_exit_domain_handler(struct o2net_msg *msg, u32 len, void *data,
 545				   void **ret_data)
 546{
 547	struct dlm_ctxt *dlm = data;
 548	unsigned int node;
 549	struct dlm_exit_domain *exit_msg = (struct dlm_exit_domain *) msg->buf;
 550
 551	mlog(0, "%p %u %p", msg, len, data);
 552
 553	if (!dlm_grab(dlm))
 554		return 0;
 555
 556	node = exit_msg->node_idx;
 557
 558	spin_lock(&dlm->spinlock);
 559	clear_bit(node, dlm->domain_map);
 560	clear_bit(node, dlm->exit_domain_map);
 561	printk(KERN_NOTICE "o2dlm: Node %u leaves domain %s ", node, dlm->name);
 562	__dlm_print_nodes(dlm);
 563
 564	/* notify anything attached to the heartbeat events */
 565	dlm_hb_event_notify_attached(dlm, node, 0);
 566
 567	spin_unlock(&dlm->spinlock);
 568
 569	dlm_put(dlm);
 570
 571	return 0;
 572}
 573
 574static int dlm_send_one_domain_exit(struct dlm_ctxt *dlm, u32 msg_type,
 575				    unsigned int node)
 576{
 577	int status;
 578	struct dlm_exit_domain leave_msg;
 579
 580	mlog(0, "%s: Sending domain exit message %u to node %u\n", dlm->name,
 581	     msg_type, node);
 582
 583	memset(&leave_msg, 0, sizeof(leave_msg));
 584	leave_msg.node_idx = dlm->node_num;
 585
 586	status = o2net_send_message(msg_type, dlm->key, &leave_msg,
 587				    sizeof(leave_msg), node, NULL);
 588	if (status < 0)
 589		mlog(ML_ERROR, "Error %d sending domain exit message %u "
 590		     "to node %u on domain %s\n", status, msg_type, node,
 591		     dlm->name);
 592
 593	return status;
 594}
 595
 596static void dlm_begin_exit_domain(struct dlm_ctxt *dlm)
 597{
 598	int node = -1;
 599
 600	/* Support for begin exit domain was added in 1.2 */
 601	if (dlm->dlm_locking_proto.pv_major == 1 &&
 602	    dlm->dlm_locking_proto.pv_minor < 2)
 603		return;
 604
 605	/*
 606	 * Unlike DLM_EXIT_DOMAIN_MSG, DLM_BEGIN_EXIT_DOMAIN_MSG is purely
 607	 * informational. Meaning if a node does not receive the message,
 608	 * so be it.
 609	 */
 610	spin_lock(&dlm->spinlock);
 611	while (1) {
 612		node = find_next_bit(dlm->domain_map, O2NM_MAX_NODES, node + 1);
 613		if (node >= O2NM_MAX_NODES)
 614			break;
 615		if (node == dlm->node_num)
 616			continue;
 617
 618		spin_unlock(&dlm->spinlock);
 619		dlm_send_one_domain_exit(dlm, DLM_BEGIN_EXIT_DOMAIN_MSG, node);
 620		spin_lock(&dlm->spinlock);
 621	}
 622	spin_unlock(&dlm->spinlock);
 623}
 624
 625static void dlm_leave_domain(struct dlm_ctxt *dlm)
 626{
 627	int node, clear_node, status;
 628
 629	/* At this point we've migrated away all our locks and won't
 630	 * accept mastership of new ones. The dlm is responsible for
 631	 * almost nothing now. We make sure not to confuse any joining
 632	 * nodes and then commence shutdown procedure. */
 633
 634	spin_lock(&dlm->spinlock);
 635	/* Clear ourselves from the domain map */
 636	clear_bit(dlm->node_num, dlm->domain_map);
 637	while ((node = find_next_bit(dlm->domain_map, O2NM_MAX_NODES,
 638				     0)) < O2NM_MAX_NODES) {
 639		/* Drop the dlm spinlock. This is safe wrt the domain_map.
 640		 * -nodes cannot be added now as the
 641		 *   query_join_handlers knows to respond with OK_NO_MAP
 642		 * -we catch the right network errors if a node is
 643		 *   removed from the map while we're sending him the
 644		 *   exit message. */
 645		spin_unlock(&dlm->spinlock);
 646
 647		clear_node = 1;
 648
 649		status = dlm_send_one_domain_exit(dlm, DLM_EXIT_DOMAIN_MSG,
 650						  node);
 651		if (status < 0 &&
 652		    status != -ENOPROTOOPT &&
 653		    status != -ENOTCONN) {
 654			mlog(ML_NOTICE, "Error %d sending domain exit message "
 655			     "to node %d\n", status, node);
 656
 657			/* Not sure what to do here but lets sleep for
 658			 * a bit in case this was a transient
 659			 * error... */
 660			msleep(DLM_DOMAIN_BACKOFF_MS);
 661			clear_node = 0;
 662		}
 663
 664		spin_lock(&dlm->spinlock);
 665		/* If we're not clearing the node bit then we intend
 666		 * to loop back around to try again. */
 667		if (clear_node)
 668			clear_bit(node, dlm->domain_map);
 669	}
 670	spin_unlock(&dlm->spinlock);
 671}
 672
 673void dlm_unregister_domain(struct dlm_ctxt *dlm)
 674{
 675	int leave = 0;
 676	struct dlm_lock_resource *res;
 677
 678	spin_lock(&dlm_domain_lock);
 679	BUG_ON(dlm->dlm_state != DLM_CTXT_JOINED);
 680	BUG_ON(!dlm->num_joins);
 681
 682	dlm->num_joins--;
 683	if (!dlm->num_joins) {
 684		/* We mark it "in shutdown" now so new register
 685		 * requests wait until we've completely left the
 686		 * domain. Don't use DLM_CTXT_LEAVING yet as we still
 687		 * want new domain joins to communicate with us at
 688		 * least until we've completed migration of our
 689		 * resources. */
 690		dlm->dlm_state = DLM_CTXT_IN_SHUTDOWN;
 691		leave = 1;
 692	}
 693	spin_unlock(&dlm_domain_lock);
 694
 695	if (leave) {
 696		mlog(0, "shutting down domain %s\n", dlm->name);
 697		dlm_begin_exit_domain(dlm);
 698
 699		/* We changed dlm state, notify the thread */
 700		dlm_kick_thread(dlm, NULL);
 701
 702		while (dlm_migrate_all_locks(dlm)) {
 703			/* Give dlm_thread time to purge the lockres' */
 704			msleep(500);
 705			mlog(0, "%s: more migration to do\n", dlm->name);
 706		}
 707
 708		/* This list should be empty. If not, print remaining lockres */
 709		if (!list_empty(&dlm->tracking_list)) {
 710			mlog(ML_ERROR, "Following lockres' are still on the "
 711			     "tracking list:\n");
 712			list_for_each_entry(res, &dlm->tracking_list, tracking)
 713				dlm_print_one_lock_resource(res);
 714		}
 715
 716		dlm_mark_domain_leaving(dlm);
 717		dlm_leave_domain(dlm);
 718		printk(KERN_NOTICE "o2dlm: Leaving domain %s\n", dlm->name);
 719		dlm_force_free_mles(dlm);
 720		dlm_complete_dlm_shutdown(dlm);
 721	}
 722	dlm_put(dlm);
 723}
 724EXPORT_SYMBOL_GPL(dlm_unregister_domain);
 725
 726static int dlm_query_join_proto_check(char *proto_type, int node,
 727				      struct dlm_protocol_version *ours,
 728				      struct dlm_protocol_version *request)
 729{
 730	int rc;
 731	struct dlm_protocol_version proto = *request;
 732
 733	if (!dlm_protocol_compare(ours, &proto)) {
 734		mlog(0,
 735		     "node %u wanted to join with %s locking protocol "
 736		     "%u.%u, we respond with %u.%u\n",
 737		     node, proto_type,
 738		     request->pv_major,
 739		     request->pv_minor,
 740		     proto.pv_major, proto.pv_minor);
 741		request->pv_minor = proto.pv_minor;
 742		rc = 0;
 743	} else {
 744		mlog(ML_NOTICE,
 745		     "Node %u wanted to join with %s locking "
 746		     "protocol %u.%u, but we have %u.%u, disallowing\n",
 747		     node, proto_type,
 748		     request->pv_major,
 749		     request->pv_minor,
 750		     ours->pv_major,
 751		     ours->pv_minor);
 752		rc = 1;
 753	}
 754
 755	return rc;
 756}
 757
 758/*
 759 * struct dlm_query_join_packet is made up of four one-byte fields.  They
 760 * are effectively in big-endian order already.  However, little-endian
 761 * machines swap them before putting the packet on the wire (because
 762 * query_join's response is a status, and that status is treated as a u32
 763 * on the wire).  Thus, a big-endian and little-endian machines will treat
 764 * this structure differently.
 765 *
 766 * The solution is to have little-endian machines swap the structure when
 767 * converting from the structure to the u32 representation.  This will
 768 * result in the structure having the correct format on the wire no matter
 769 * the host endian format.
 770 */
 771static void dlm_query_join_packet_to_wire(struct dlm_query_join_packet *packet,
 772					  u32 *wire)
 773{
 774	union dlm_query_join_response response;
 775
 776	response.packet = *packet;
 777	*wire = be32_to_cpu(response.intval);
 778}
 779
 780static void dlm_query_join_wire_to_packet(u32 wire,
 781					  struct dlm_query_join_packet *packet)
 782{
 783	union dlm_query_join_response response;
 784
 785	response.intval = cpu_to_be32(wire);
 786	*packet = response.packet;
 787}
 788
 789static int dlm_query_join_handler(struct o2net_msg *msg, u32 len, void *data,
 790				  void **ret_data)
 791{
 792	struct dlm_query_join_request *query;
 793	struct dlm_query_join_packet packet = {
 794		.code = JOIN_DISALLOW,
 795	};
 796	struct dlm_ctxt *dlm = NULL;
 797	u32 response;
 798	u8 nodenum;
 799
 800	query = (struct dlm_query_join_request *) msg->buf;
 801
 802	mlog(0, "node %u wants to join domain %s\n", query->node_idx,
 803		  query->domain);
 804
 805	/*
 806	 * If heartbeat doesn't consider the node live, tell it
 807	 * to back off and try again.  This gives heartbeat a chance
 808	 * to catch up.
 809	 */
 810	if (!o2hb_check_node_heartbeating_no_sem(query->node_idx)) {
 811		mlog(0, "node %u is not in our live map yet\n",
 812		     query->node_idx);
 813
 814		packet.code = JOIN_DISALLOW;
 815		goto respond;
 816	}
 817
 818	packet.code = JOIN_OK_NO_MAP;
 819
 820	spin_lock(&dlm_domain_lock);
 821	dlm = __dlm_lookup_domain_full(query->domain, query->name_len);
 822	if (!dlm)
 823		goto unlock_respond;
 824
 825	/*
 826	 * There is a small window where the joining node may not see the
 827	 * node(s) that just left but still part of the cluster. DISALLOW
 828	 * join request if joining node has different node map.
 829	 */
 830	nodenum=0;
 831	while (nodenum < O2NM_MAX_NODES) {
 832		if (test_bit(nodenum, dlm->domain_map)) {
 833			if (!byte_test_bit(nodenum, query->node_map)) {
 834				mlog(0, "disallow join as node %u does not "
 835				     "have node %u in its nodemap\n",
 836				     query->node_idx, nodenum);
 837				packet.code = JOIN_DISALLOW;
 838				goto unlock_respond;
 839			}
 840		}
 841		nodenum++;
 842	}
 843
 844	/* Once the dlm ctxt is marked as leaving then we don't want
 845	 * to be put in someone's domain map.
 846	 * Also, explicitly disallow joining at certain troublesome
 847	 * times (ie. during recovery). */
 848	if (dlm->dlm_state != DLM_CTXT_LEAVING) {
 849		int bit = query->node_idx;
 850		spin_lock(&dlm->spinlock);
 851
 852		if (dlm->dlm_state == DLM_CTXT_NEW &&
 853		    dlm->joining_node == DLM_LOCK_RES_OWNER_UNKNOWN) {
 854			/*If this is a brand new context and we
 855			 * haven't started our join process yet, then
 856			 * the other node won the race. */
 857			packet.code = JOIN_OK_NO_MAP;
 858		} else if (dlm->joining_node != DLM_LOCK_RES_OWNER_UNKNOWN) {
 859			/* Disallow parallel joins. */
 860			packet.code = JOIN_DISALLOW;
 861		} else if (dlm->reco.state & DLM_RECO_STATE_ACTIVE) {
 862			mlog(0, "node %u trying to join, but recovery "
 863			     "is ongoing.\n", bit);
 864			packet.code = JOIN_DISALLOW;
 865		} else if (test_bit(bit, dlm->recovery_map)) {
 866			mlog(0, "node %u trying to join, but it "
 867			     "still needs recovery.\n", bit);
 868			packet.code = JOIN_DISALLOW;
 869		} else if (test_bit(bit, dlm->domain_map)) {
 870			mlog(0, "node %u trying to join, but it "
 871			     "is still in the domain! needs recovery?\n",
 872			     bit);
 873			packet.code = JOIN_DISALLOW;
 874		} else {
 875			/* Alright we're fully a part of this domain
 876			 * so we keep some state as to who's joining
 877			 * and indicate to him that needs to be fixed
 878			 * up. */
 879
 880			/* Make sure we speak compatible locking protocols.  */
 881			if (dlm_query_join_proto_check("DLM", bit,
 882						       &dlm->dlm_locking_proto,
 883						       &query->dlm_proto)) {
 884				packet.code = JOIN_PROTOCOL_MISMATCH;
 885			} else if (dlm_query_join_proto_check("fs", bit,
 886							      &dlm->fs_locking_proto,
 887							      &query->fs_proto)) {
 888				packet.code = JOIN_PROTOCOL_MISMATCH;
 889			} else {
 890				packet.dlm_minor = query->dlm_proto.pv_minor;
 891				packet.fs_minor = query->fs_proto.pv_minor;
 892				packet.code = JOIN_OK;
 893				__dlm_set_joining_node(dlm, query->node_idx);
 894			}
 895		}
 896
 897		spin_unlock(&dlm->spinlock);
 898	}
 899unlock_respond:
 900	spin_unlock(&dlm_domain_lock);
 901
 902respond:
 903	mlog(0, "We respond with %u\n", packet.code);
 904
 905	dlm_query_join_packet_to_wire(&packet, &response);
 906	return response;
 907}
 908
 909static int dlm_assert_joined_handler(struct o2net_msg *msg, u32 len, void *data,
 910				     void **ret_data)
 911{
 912	struct dlm_assert_joined *assert;
 913	struct dlm_ctxt *dlm = NULL;
 914
 915	assert = (struct dlm_assert_joined *) msg->buf;
 916
 917	mlog(0, "node %u asserts join on domain %s\n", assert->node_idx,
 918		  assert->domain);
 919
 920	spin_lock(&dlm_domain_lock);
 921	dlm = __dlm_lookup_domain_full(assert->domain, assert->name_len);
 922	/* XXX should we consider no dlm ctxt an error? */
 923	if (dlm) {
 924		spin_lock(&dlm->spinlock);
 925
 926		/* Alright, this node has officially joined our
 927		 * domain. Set him in the map and clean up our
 928		 * leftover join state. */
 929		BUG_ON(dlm->joining_node != assert->node_idx);
 930
 931		if (dlm->reco.state & DLM_RECO_STATE_ACTIVE) {
 932			mlog(0, "dlm recovery is ongoing, disallow join\n");
 933			spin_unlock(&dlm->spinlock);
 934			spin_unlock(&dlm_domain_lock);
 935			return -EAGAIN;
 936		}
 937
 938		set_bit(assert->node_idx, dlm->domain_map);
 939		clear_bit(assert->node_idx, dlm->exit_domain_map);
 940		__dlm_set_joining_node(dlm, DLM_LOCK_RES_OWNER_UNKNOWN);
 941
 942		printk(KERN_NOTICE "o2dlm: Node %u joins domain %s ",
 943		       assert->node_idx, dlm->name);
 944		__dlm_print_nodes(dlm);
 945
 946		/* notify anything attached to the heartbeat events */
 947		dlm_hb_event_notify_attached(dlm, assert->node_idx, 1);
 948
 949		spin_unlock(&dlm->spinlock);
 950	}
 951	spin_unlock(&dlm_domain_lock);
 952
 953	return 0;
 954}
 955
 956static int dlm_match_regions(struct dlm_ctxt *dlm,
 957			     struct dlm_query_region *qr,
 958			     char *local, int locallen)
 959{
 960	char *remote = qr->qr_regions;
 961	char *l, *r;
 962	int localnr, i, j, foundit;
 963	int status = 0;
 964
 965	if (!o2hb_global_heartbeat_active()) {
 966		if (qr->qr_numregions) {
 967			mlog(ML_ERROR, "Domain %s: Joining node %d has global "
 968			     "heartbeat enabled but local node %d does not\n",
 969			     qr->qr_domain, qr->qr_node, dlm->node_num);
 970			status = -EINVAL;
 971		}
 972		goto bail;
 973	}
 974
 975	if (o2hb_global_heartbeat_active() && !qr->qr_numregions) {
 976		mlog(ML_ERROR, "Domain %s: Local node %d has global "
 977		     "heartbeat enabled but joining node %d does not\n",
 978		     qr->qr_domain, dlm->node_num, qr->qr_node);
 979		status = -EINVAL;
 980		goto bail;
 981	}
 982
 983	r = remote;
 984	for (i = 0; i < qr->qr_numregions; ++i) {
 985		mlog(0, "Region %.*s\n", O2HB_MAX_REGION_NAME_LEN, r);
 986		r += O2HB_MAX_REGION_NAME_LEN;
 987	}
 988
 989	localnr = min(O2NM_MAX_REGIONS, locallen/O2HB_MAX_REGION_NAME_LEN);
 990	localnr = o2hb_get_all_regions(local, (u8)localnr);
 991
 992	/* compare local regions with remote */
 993	l = local;
 994	for (i = 0; i < localnr; ++i) {
 995		foundit = 0;
 996		r = remote;
 997		for (j = 0; j <= qr->qr_numregions; ++j) {
 998			if (!memcmp(l, r, O2HB_MAX_REGION_NAME_LEN)) {
 999				foundit = 1;
1000				break;
1001			}
1002			r += O2HB_MAX_REGION_NAME_LEN;
1003		}
1004		if (!foundit) {
1005			status = -EINVAL;
1006			mlog(ML_ERROR, "Domain %s: Region '%.*s' registered "
1007			     "in local node %d but not in joining node %d\n",
1008			     qr->qr_domain, O2HB_MAX_REGION_NAME_LEN, l,
1009			     dlm->node_num, qr->qr_node);
1010			goto bail;
1011		}
1012		l += O2HB_MAX_REGION_NAME_LEN;
1013	}
1014
1015	/* compare remote with local regions */
1016	r = remote;
1017	for (i = 0; i < qr->qr_numregions; ++i) {
1018		foundit = 0;
1019		l = local;
1020		for (j = 0; j < localnr; ++j) {
1021			if (!memcmp(r, l, O2HB_MAX_REGION_NAME_LEN)) {
1022				foundit = 1;
1023				break;
1024			}
1025			l += O2HB_MAX_REGION_NAME_LEN;
1026		}
1027		if (!foundit) {
1028			status = -EINVAL;
1029			mlog(ML_ERROR, "Domain %s: Region '%.*s' registered "
1030			     "in joining node %d but not in local node %d\n",
1031			     qr->qr_domain, O2HB_MAX_REGION_NAME_LEN, r,
1032			     qr->qr_node, dlm->node_num);
1033			goto bail;
1034		}
1035		r += O2HB_MAX_REGION_NAME_LEN;
1036	}
1037
1038bail:
1039	return status;
1040}
1041
1042static int dlm_send_regions(struct dlm_ctxt *dlm, unsigned long *node_map)
1043{
1044	struct dlm_query_region *qr = NULL;
1045	int status, ret = 0, i;
1046	char *p;
1047
1048	if (find_first_bit(node_map, O2NM_MAX_NODES) >= O2NM_MAX_NODES)
1049		goto bail;
1050
1051	qr = kzalloc(sizeof(struct dlm_query_region), GFP_KERNEL);
1052	if (!qr) {
1053		ret = -ENOMEM;
1054		mlog_errno(ret);
1055		goto bail;
1056	}
1057
1058	qr->qr_node = dlm->node_num;
1059	qr->qr_namelen = strlen(dlm->name);
1060	memcpy(qr->qr_domain, dlm->name, qr->qr_namelen);
1061	/* if local hb, the numregions will be zero */
1062	if (o2hb_global_heartbeat_active())
1063		qr->qr_numregions = o2hb_get_all_regions(qr->qr_regions,
1064							 O2NM_MAX_REGIONS);
1065
1066	p = qr->qr_regions;
1067	for (i = 0; i < qr->qr_numregions; ++i, p += O2HB_MAX_REGION_NAME_LEN)
1068		mlog(0, "Region %.*s\n", O2HB_MAX_REGION_NAME_LEN, p);
1069
1070	i = -1;
1071	while ((i = find_next_bit(node_map, O2NM_MAX_NODES,
1072				  i + 1)) < O2NM_MAX_NODES) {
1073		if (i == dlm->node_num)
1074			continue;
1075
1076		mlog(0, "Sending regions to node %d\n", i);
1077
1078		ret = o2net_send_message(DLM_QUERY_REGION, DLM_MOD_KEY, qr,
1079					 sizeof(struct dlm_query_region),
1080					 i, &status);
1081		if (ret >= 0)
1082			ret = status;
1083		if (ret) {
1084			mlog(ML_ERROR, "Region mismatch %d, node %d\n",
1085			     ret, i);
1086			break;
1087		}
1088	}
1089
1090bail:
1091	kfree(qr);
1092	return ret;
1093}
1094
1095static int dlm_query_region_handler(struct o2net_msg *msg, u32 len,
1096				    void *data, void **ret_data)
1097{
1098	struct dlm_query_region *qr;
1099	struct dlm_ctxt *dlm = NULL;
1100	char *local = NULL;
1101	int status = 0;
1102
1103	qr = (struct dlm_query_region *) msg->buf;
1104
1105	mlog(0, "Node %u queries hb regions on domain %s\n", qr->qr_node,
1106	     qr->qr_domain);
1107
1108	/* buffer used in dlm_mast_regions() */
1109	local = kmalloc(sizeof(qr->qr_regions), GFP_KERNEL);
1110	if (!local)
1111		return -ENOMEM;
1112
1113	status = -EINVAL;
1114
1115	spin_lock(&dlm_domain_lock);
1116	dlm = __dlm_lookup_domain_full(qr->qr_domain, qr->qr_namelen);
1117	if (!dlm) {
1118		mlog(ML_ERROR, "Node %d queried hb regions on domain %s "
1119		     "before join domain\n", qr->qr_node, qr->qr_domain);
1120		goto out_domain_lock;
1121	}
1122
1123	spin_lock(&dlm->spinlock);
1124	if (dlm->joining_node != qr->qr_node) {
1125		mlog(ML_ERROR, "Node %d queried hb regions on domain %s "
1126		     "but joining node is %d\n", qr->qr_node, qr->qr_domain,
1127		     dlm->joining_node);
1128		goto out_dlm_lock;
1129	}
1130
1131	/* Support for global heartbeat was added in 1.1 */
1132	if (dlm->dlm_locking_proto.pv_major == 1 &&
1133	    dlm->dlm_locking_proto.pv_minor == 0) {
1134		mlog(ML_ERROR, "Node %d queried hb regions on domain %s "
1135		     "but active dlm protocol is %d.%d\n", qr->qr_node,
1136		     qr->qr_domain, dlm->dlm_locking_proto.pv_major,
1137		     dlm->dlm_locking_proto.pv_minor);
1138		goto out_dlm_lock;
1139	}
1140
1141	status = dlm_match_regions(dlm, qr, local, sizeof(qr->qr_regions));
1142
1143out_dlm_lock:
1144	spin_unlock(&dlm->spinlock);
1145
1146out_domain_lock:
1147	spin_unlock(&dlm_domain_lock);
1148
1149	kfree(local);
1150
1151	return status;
1152}
1153
1154static int dlm_match_nodes(struct dlm_ctxt *dlm, struct dlm_query_nodeinfo *qn)
1155{
1156	struct o2nm_node *local;
1157	struct dlm_node_info *remote;
1158	int i, j;
1159	int status = 0;
1160
1161	for (j = 0; j < qn->qn_numnodes; ++j)
1162		mlog(0, "Node %3d, %pI4:%u\n", qn->qn_nodes[j].ni_nodenum,
1163		     &(qn->qn_nodes[j].ni_ipv4_address),
1164		     ntohs(qn->qn_nodes[j].ni_ipv4_port));
1165
1166	for (i = 0; i < O2NM_MAX_NODES && !status; ++i) {
1167		local = o2nm_get_node_by_num(i);
1168		remote = NULL;
1169		for (j = 0; j < qn->qn_numnodes; ++j) {
1170			if (qn->qn_nodes[j].ni_nodenum == i) {
1171				remote = &(qn->qn_nodes[j]);
1172				break;
1173			}
1174		}
1175
1176		if (!local && !remote)
1177			continue;
1178
1179		if ((local && !remote) || (!local && remote))
1180			status = -EINVAL;
1181
1182		if (!status &&
1183		    ((remote->ni_nodenum != local->nd_num) ||
1184		     (remote->ni_ipv4_port != local->nd_ipv4_port) ||
1185		     (remote->ni_ipv4_address != local->nd_ipv4_address)))
1186			status = -EINVAL;
1187
1188		if (status) {
1189			if (remote && !local)
1190				mlog(ML_ERROR, "Domain %s: Node %d (%pI4:%u) "
1191				     "registered in joining node %d but not in "
1192				     "local node %d\n", qn->qn_domain,
1193				     remote->ni_nodenum,
1194				     &(remote->ni_ipv4_address),
1195				     ntohs(remote->ni_ipv4_port),
1196				     qn->qn_nodenum, dlm->node_num);
1197			if (local && !remote)
1198				mlog(ML_ERROR, "Domain %s: Node %d (%pI4:%u) "
1199				     "registered in local node %d but not in "
1200				     "joining node %d\n", qn->qn_domain,
1201				     local->nd_num, &(local->nd_ipv4_address),
1202				     ntohs(local->nd_ipv4_port),
1203				     dlm->node_num, qn->qn_nodenum);
1204			BUG_ON((!local && !remote));
1205		}
1206
1207		if (local)
1208			o2nm_node_put(local);
1209	}
1210
1211	return status;
1212}
1213
1214static int dlm_send_nodeinfo(struct dlm_ctxt *dlm, unsigned long *node_map)
1215{
1216	struct dlm_query_nodeinfo *qn = NULL;
1217	struct o2nm_node *node;
1218	int ret = 0, status, count, i;
1219
1220	if (find_first_bit(node_map, O2NM_MAX_NODES) >= O2NM_MAX_NODES)
1221		goto bail;
1222
1223	qn = kzalloc(sizeof(struct dlm_query_nodeinfo), GFP_KERNEL);
1224	if (!qn) {
1225		ret = -ENOMEM;
1226		mlog_errno(ret);
1227		goto bail;
1228	}
1229
1230	for (i = 0, count = 0; i < O2NM_MAX_NODES; ++i) {
1231		node = o2nm_get_node_by_num(i);
1232		if (!node)
1233			continue;
1234		qn->qn_nodes[count].ni_nodenum = node->nd_num;
1235		qn->qn_nodes[count].ni_ipv4_port = node->nd_ipv4_port;
1236		qn->qn_nodes[count].ni_ipv4_address = node->nd_ipv4_address;
1237		mlog(0, "Node %3d, %pI4:%u\n", node->nd_num,
1238		     &(node->nd_ipv4_address), ntohs(node->nd_ipv4_port));
1239		++count;
1240		o2nm_node_put(node);
1241	}
1242
1243	qn->qn_nodenum = dlm->node_num;
1244	qn->qn_numnodes = count;
1245	qn->qn_namelen = strlen(dlm->name);
1246	memcpy(qn->qn_domain, dlm->name, qn->qn_namelen);
1247
1248	i = -1;
1249	while ((i = find_next_bit(node_map, O2NM_MAX_NODES,
1250				  i + 1)) < O2NM_MAX_NODES) {
1251		if (i == dlm->node_num)
1252			continue;
1253
1254		mlog(0, "Sending nodeinfo to node %d\n", i);
1255
1256		ret = o2net_send_message(DLM_QUERY_NODEINFO, DLM_MOD_KEY,
1257					 qn, sizeof(struct dlm_query_nodeinfo),
1258					 i, &status);
1259		if (ret >= 0)
1260			ret = status;
1261		if (ret) {
1262			mlog(ML_ERROR, "node mismatch %d, node %d\n", ret, i);
1263			break;
1264		}
1265	}
1266
1267bail:
1268	kfree(qn);
1269	return ret;
1270}
1271
1272static int dlm_query_nodeinfo_handler(struct o2net_msg *msg, u32 len,
1273				      void *data, void **ret_data)
1274{
1275	struct dlm_query_nodeinfo *qn;
1276	struct dlm_ctxt *dlm = NULL;
1277	int locked = 0, status = -EINVAL;
1278
1279	qn = (struct dlm_query_nodeinfo *) msg->buf;
1280
1281	mlog(0, "Node %u queries nodes on domain %s\n", qn->qn_nodenum,
1282	     qn->qn_domain);
1283
1284	spin_lock(&dlm_domain_lock);
1285	dlm = __dlm_lookup_domain_full(qn->qn_domain, qn->qn_namelen);
1286	if (!dlm) {
1287		mlog(ML_ERROR, "Node %d queried nodes on domain %s before "
1288		     "join domain\n", qn->qn_nodenum, qn->qn_domain);
1289		goto bail;
1290	}
1291
1292	spin_lock(&dlm->spinlock);
1293	locked = 1;
1294	if (dlm->joining_node != qn->qn_nodenum) {
1295		mlog(ML_ERROR, "Node %d queried nodes on domain %s but "
1296		     "joining node is %d\n", qn->qn_nodenum, qn->qn_domain,
1297		     dlm->joining_node);
1298		goto bail;
1299	}
1300
1301	/* Support for node query was added in 1.1 */
1302	if (dlm->dlm_locking_proto.pv_major == 1 &&
1303	    dlm->dlm_locking_proto.pv_minor == 0) {
1304		mlog(ML_ERROR, "Node %d queried nodes on domain %s "
1305		     "but active dlm protocol is %d.%d\n", qn->qn_nodenum,
1306		     qn->qn_domain, dlm->dlm_locking_proto.pv_major,
1307		     dlm->dlm_locking_proto.pv_minor);
1308		goto bail;
1309	}
1310
1311	status = dlm_match_nodes(dlm, qn);
1312
1313bail:
1314	if (locked)
1315		spin_unlock(&dlm->spinlock);
1316	spin_unlock(&dlm_domain_lock);
1317
1318	return status;
1319}
1320
1321static int dlm_cancel_join_handler(struct o2net_msg *msg, u32 len, void *data,
1322				   void **ret_data)
1323{
1324	struct dlm_cancel_join *cancel;
1325	struct dlm_ctxt *dlm = NULL;
1326
1327	cancel = (struct dlm_cancel_join *) msg->buf;
1328
1329	mlog(0, "node %u cancels join on domain %s\n", cancel->node_idx,
1330		  cancel->domain);
1331
1332	spin_lock(&dlm_domain_lock);
1333	dlm = __dlm_lookup_domain_full(cancel->domain, cancel->name_len);
1334
1335	if (dlm) {
1336		spin_lock(&dlm->spinlock);
1337
1338		/* Yikes, this guy wants to cancel his join. No
1339		 * problem, we simply cleanup our join state. */
1340		BUG_ON(dlm->joining_node != cancel->node_idx);
1341		__dlm_set_joining_node(dlm, DLM_LOCK_RES_OWNER_UNKNOWN);
1342
1343		spin_unlock(&dlm->spinlock);
1344	}
1345	spin_unlock(&dlm_domain_lock);
1346
1347	return 0;
1348}
1349
1350static int dlm_send_one_join_cancel(struct dlm_ctxt *dlm,
1351				    unsigned int node)
1352{
1353	int status;
1354	struct dlm_cancel_join cancel_msg;
1355
1356	memset(&cancel_msg, 0, sizeof(cancel_msg));
1357	cancel_msg.node_idx = dlm->node_num;
1358	cancel_msg.name_len = strlen(dlm->name);
1359	memcpy(cancel_msg.domain, dlm->name, cancel_msg.name_len);
1360
1361	status = o2net_send_message(DLM_CANCEL_JOIN_MSG, DLM_MOD_KEY,
1362				    &cancel_msg, sizeof(cancel_msg), node,
1363				    NULL);
1364	if (status < 0) {
1365		mlog(ML_ERROR, "Error %d when sending message %u (key 0x%x) to "
1366		     "node %u\n", status, DLM_CANCEL_JOIN_MSG, DLM_MOD_KEY,
1367		     node);
1368		goto bail;
1369	}
1370
1371bail:
1372	return status;
1373}
1374
1375/* map_size should be in bytes. */
1376static int dlm_send_join_cancels(struct dlm_ctxt *dlm,
1377				 unsigned long *node_map,
1378				 unsigned int map_size)
1379{
1380	int status, tmpstat;
1381	int node;
1382
1383	if (map_size != (BITS_TO_LONGS(O2NM_MAX_NODES) *
1384			 sizeof(unsigned long))) {
1385		mlog(ML_ERROR,
1386		     "map_size %u != BITS_TO_LONGS(O2NM_MAX_NODES) %u\n",
1387		     map_size, (unsigned)BITS_TO_LONGS(O2NM_MAX_NODES));
1388		return -EINVAL;
1389	}
1390
1391	status = 0;
1392	node = -1;
1393	while ((node = find_next_bit(node_map, O2NM_MAX_NODES,
1394				     node + 1)) < O2NM_MAX_NODES) {
1395		if (node == dlm->node_num)
1396			continue;
1397
1398		tmpstat = dlm_send_one_join_cancel(dlm, node);
1399		if (tmpstat) {
1400			mlog(ML_ERROR, "Error return %d cancelling join on "
1401			     "node %d\n", tmpstat, node);
1402			if (!status)
1403				status = tmpstat;
1404		}
1405	}
1406
1407	if (status)
1408		mlog_errno(status);
1409	return status;
1410}
1411
1412static int dlm_request_join(struct dlm_ctxt *dlm,
1413			    int node,
1414			    enum dlm_query_join_response_code *response)
1415{
1416	int status;
1417	struct dlm_query_join_request join_msg;
1418	struct dlm_query_join_packet packet;
1419	u32 join_resp;
1420
1421	mlog(0, "querying node %d\n", node);
1422
1423	memset(&join_msg, 0, sizeof(join_msg));
1424	join_msg.node_idx = dlm->node_num;
1425	join_msg.name_len = strlen(dlm->name);
1426	memcpy(join_msg.domain, dlm->name, join_msg.name_len);
1427	join_msg.dlm_proto = dlm->dlm_locking_proto;
1428	join_msg.fs_proto = dlm->fs_locking_proto;
1429
1430	/* copy live node map to join message */
1431	byte_copymap(join_msg.node_map, dlm->live_nodes_map, O2NM_MAX_NODES);
1432
1433	status = o2net_send_message(DLM_QUERY_JOIN_MSG, DLM_MOD_KEY, &join_msg,
1434				    sizeof(join_msg), node, &join_resp);
1435	if (status < 0 && status != -ENOPROTOOPT) {
1436		mlog(ML_ERROR, "Error %d when sending message %u (key 0x%x) to "
1437		     "node %u\n", status, DLM_QUERY_JOIN_MSG, DLM_MOD_KEY,
1438		     node);
1439		goto bail;
1440	}
1441	dlm_query_join_wire_to_packet(join_resp, &packet);
1442
1443	/* -ENOPROTOOPT from the net code means the other side isn't
1444	    listening for our message type -- that's fine, it means
1445	    his dlm isn't up, so we can consider him a 'yes' but not
1446	    joined into the domain.  */
1447	if (status == -ENOPROTOOPT) {
1448		status = 0;
1449		*response = JOIN_OK_NO_MAP;
1450	} else {
1451		*response = packet.code;
1452		switch (packet.code) {
1453		case JOIN_DISALLOW:
1454		case JOIN_OK_NO_MAP:
1455			break;
1456		case JOIN_PROTOCOL_MISMATCH:
1457			mlog(ML_NOTICE,
1458			     "This node requested DLM locking protocol %u.%u and "
1459			     "filesystem locking protocol %u.%u.  At least one of "
1460			     "the protocol versions on node %d is not compatible, "
1461			     "disconnecting\n",
1462			     dlm->dlm_locking_proto.pv_major,
1463			     dlm->dlm_locking_proto.pv_minor,
1464			     dlm->fs_locking_proto.pv_major,
1465			     dlm->fs_locking_proto.pv_minor,
1466			     node);
1467			status = -EPROTO;
1468			break;
1469		case JOIN_OK:
1470			/* Use the same locking protocol as the remote node */
1471			dlm->dlm_locking_proto.pv_minor = packet.dlm_minor;
1472			dlm->fs_locking_proto.pv_minor = packet.fs_minor;
1473			mlog(0,
1474			     "Node %d responds JOIN_OK with DLM locking protocol "
1475			     "%u.%u and fs locking protocol %u.%u\n",
1476			     node,
1477			     dlm->dlm_locking_proto.pv_major,
1478			     dlm->dlm_locking_proto.pv_minor,
1479			     dlm->fs_locking_proto.pv_major,
1480			     dlm->fs_locking_proto.pv_minor);
1481			break;
1482		default:
1483			status = -EINVAL;
1484			mlog(ML_ERROR, "invalid response %d from node %u\n",
1485			     packet.code, node);
1486			/* Reset response to JOIN_DISALLOW */
1487			*response = JOIN_DISALLOW;
1488			break;
1489		}
1490	}
1491
1492	mlog(0, "status %d, node %d response is %d\n", status, node,
1493	     *response);
1494
1495bail:
1496	return status;
1497}
1498
1499static int dlm_send_one_join_assert(struct dlm_ctxt *dlm,
1500				    unsigned int node)
1501{
1502	int status;
1503	int ret;
1504	struct dlm_assert_joined assert_msg;
1505
1506	mlog(0, "Sending join assert to node %u\n", node);
1507
1508	memset(&assert_msg, 0, sizeof(assert_msg));
1509	assert_msg.node_idx = dlm->node_num;
1510	assert_msg.name_len = strlen(dlm->name);
1511	memcpy(assert_msg.domain, dlm->name, assert_msg.name_len);
1512
1513	status = o2net_send_message(DLM_ASSERT_JOINED_MSG, DLM_MOD_KEY,
1514				    &assert_msg, sizeof(assert_msg), node,
1515				    &ret);
1516	if (status < 0)
1517		mlog(ML_ERROR, "Error %d when sending message %u (key 0x%x) to "
1518		     "node %u\n", status, DLM_ASSERT_JOINED_MSG, DLM_MOD_KEY,
1519		     node);
1520	else
1521		status = ret;
1522
1523	return status;
1524}
1525
1526static void dlm_send_join_asserts(struct dlm_ctxt *dlm,
1527				  unsigned long *node_map)
1528{
1529	int status, node, live;
1530
1531	status = 0;
1532	node = -1;
1533	while ((node = find_next_bit(node_map, O2NM_MAX_NODES,
1534				     node + 1)) < O2NM_MAX_NODES) {
1535		if (node == dlm->node_num)
1536			continue;
1537
1538		do {
1539			/* It is very important that this message be
1540			 * received so we spin until either the node
1541			 * has died or it gets the message. */
1542			status = dlm_send_one_join_assert(dlm, node);
1543
1544			spin_lock(&dlm->spinlock);
1545			live = test_bit(node, dlm->live_nodes_map);
1546			spin_unlock(&dlm->spinlock);
1547
1548			if (status) {
1549				mlog(ML_ERROR, "Error return %d asserting "
1550				     "join on node %d\n", status, node);
1551
1552				/* give us some time between errors... */
1553				if (live)
1554					msleep(DLM_DOMAIN_BACKOFF_MS);
1555			}
1556		} while (status && live);
1557	}
1558}
1559
1560struct domain_join_ctxt {
1561	unsigned long live_map[BITS_TO_LONGS(O2NM_MAX_NODES)];
1562	unsigned long yes_resp_map[BITS_TO_LONGS(O2NM_MAX_NODES)];
1563};
1564
1565static int dlm_should_restart_join(struct dlm_ctxt *dlm,
1566				   struct domain_join_ctxt *ctxt,
1567				   enum dlm_query_join_response_code response)
1568{
1569	int ret;
1570
1571	if (response == JOIN_DISALLOW) {
1572		mlog(0, "Latest response of disallow -- should restart\n");
1573		return 1;
1574	}
1575
1576	spin_lock(&dlm->spinlock);
1577	/* For now, we restart the process if the node maps have
1578	 * changed at all */
1579	ret = !bitmap_equal(ctxt->live_map, dlm->live_nodes_map,
1580			    O2NM_MAX_NODES);
1581	spin_unlock(&dlm->spinlock);
1582
1583	if (ret)
1584		mlog(0, "Node maps changed -- should restart\n");
1585
1586	return ret;
1587}
1588
1589static int dlm_try_to_join_domain(struct dlm_ctxt *dlm)
1590{
1591	int status = 0, tmpstat, node;
1592	struct domain_join_ctxt *ctxt;
1593	enum dlm_query_join_response_code response = JOIN_DISALLOW;
1594
1595	mlog(0, "%p", dlm);
1596
1597	ctxt = kzalloc(sizeof(*ctxt), GFP_KERNEL);
1598	if (!ctxt) {
1599		status = -ENOMEM;
1600		mlog_errno(status);
1601		goto bail;
1602	}
1603
1604	/* group sem locking should work for us here -- we're already
1605	 * registered for heartbeat events so filling this should be
1606	 * atomic wrt getting those handlers called. */
1607	o2hb_fill_node_map(dlm->live_nodes_map, O2NM_MAX_NODES);
1608
1609	spin_lock(&dlm->spinlock);
1610	bitmap_copy(ctxt->live_map, dlm->live_nodes_map, O2NM_MAX_NODES);
1611	__dlm_set_joining_node(dlm, dlm->node_num);
1612	spin_unlock(&dlm->spinlock);
1613
1614	node = -1;
1615	while ((node = find_next_bit(ctxt->live_map, O2NM_MAX_NODES,
1616				     node + 1)) < O2NM_MAX_NODES) {
1617		if (node == dlm->node_num)
1618			continue;
1619
1620		status = dlm_request_join(dlm, node, &response);
1621		if (status < 0) {
1622			mlog_errno(status);
1623			goto bail;
1624		}
1625
1626		/* Ok, either we got a response or the node doesn't have a
1627		 * dlm up. */
1628		if (response == JOIN_OK)
1629			set_bit(node, ctxt->yes_resp_map);
1630
1631		if (dlm_should_restart_join(dlm, ctxt, response)) {
1632			status = -EAGAIN;
1633			goto bail;
1634		}
1635	}
1636
1637	mlog(0, "Yay, done querying nodes!\n");
1638
1639	/* Yay, everyone agree's we can join the domain. My domain is
1640	 * comprised of all nodes who were put in the
1641	 * yes_resp_map. Copy that into our domain map and send a join
1642	 * assert message to clean up everyone elses state. */
1643	spin_lock(&dlm->spinlock);
1644	bitmap_copy(dlm->domain_map, ctxt->yes_resp_map, O2NM_MAX_NODES);
1645	set_bit(dlm->node_num, dlm->domain_map);
1646	spin_unlock(&dlm->spinlock);
1647
1648	/* Support for global heartbeat and node info was added in 1.1 */
1649	if (dlm->dlm_locking_proto.pv_major > 1 ||
1650	    dlm->dlm_locking_proto.pv_minor > 0) {
1651		status = dlm_send_nodeinfo(dlm, ctxt->yes_resp_map);
1652		if (status) {
1653			mlog_errno(status);
1654			goto bail;
1655		}
1656		status = dlm_send_regions(dlm, ctxt->yes_resp_map);
1657		if (status) {
1658			mlog_errno(status);
1659			goto bail;
1660		}
1661	}
1662
1663	dlm_send_join_asserts(dlm, ctxt->yes_resp_map);
1664
1665	/* Joined state *must* be set before the joining node
1666	 * information, otherwise the query_join handler may read no
1667	 * current joiner but a state of NEW and tell joining nodes
1668	 * we're not in the domain. */
1669	spin_lock(&dlm_domain_lock);
1670	dlm->dlm_state = DLM_CTXT_JOINED;
1671	dlm->num_joins++;
1672	spin_unlock(&dlm_domain_lock);
1673
1674bail:
1675	spin_lock(&dlm->spinlock);
1676	__dlm_set_joining_node(dlm, DLM_LOCK_RES_OWNER_UNKNOWN);
1677	if (!status) {
1678		printk(KERN_NOTICE "o2dlm: Joining domain %s ", dlm->name);
1679		__dlm_print_nodes(dlm);
1680	}
1681	spin_unlock(&dlm->spinlock);
1682
1683	if (ctxt) {
1684		/* Do we need to send a cancel message to any nodes? */
1685		if (status < 0) {
1686			tmpstat = dlm_send_join_cancels(dlm,
1687							ctxt->yes_resp_map,
1688							sizeof(ctxt->yes_resp_map));
1689			if (tmpstat < 0)
1690				mlog_errno(tmpstat);
1691		}
1692		kfree(ctxt);
1693	}
1694
1695	mlog(0, "returning %d\n", status);
1696	return status;
1697}
1698
1699static void dlm_unregister_domain_handlers(struct dlm_ctxt *dlm)
1700{
1701	o2hb_unregister_callback(dlm->name, &dlm->dlm_hb_up);
1702	o2hb_unregister_callback(dlm->name, &dlm->dlm_hb_down);
1703	o2net_unregister_handler_list(&dlm->dlm_domain_handlers);
1704}
1705
1706static int dlm_register_domain_handlers(struct dlm_ctxt *dlm)
1707{
1708	int status;
1709
1710	mlog(0, "registering handlers.\n");
1711
1712	o2hb_setup_callback(&dlm->dlm_hb_down, O2HB_NODE_DOWN_CB,
1713			    dlm_hb_node_down_cb, dlm, DLM_HB_NODE_DOWN_PRI);
1714	o2hb_setup_callback(&dlm->dlm_hb_up, O2HB_NODE_UP_CB,
1715			    dlm_hb_node_up_cb, dlm, DLM_HB_NODE_UP_PRI);
1716
1717	status = o2hb_register_callback(dlm->name, &dlm->dlm_hb_down);
1718	if (status)
1719		goto bail;
1720
1721	status = o2hb_register_callback(dlm->name, &dlm->dlm_hb_up);
1722	if (status)
1723		goto bail;
1724
1725	status = o2net_register_handler(DLM_MASTER_REQUEST_MSG, dlm->key,
1726					sizeof(struct dlm_master_request),
1727					dlm_master_request_handler,
1728					dlm, NULL, &dlm->dlm_domain_handlers);
1729	if (status)
1730		goto bail;
1731
1732	status = o2net_register_handler(DLM_ASSERT_MASTER_MSG, dlm->key,
1733					sizeof(struct dlm_assert_master),
1734					dlm_assert_master_handler,
1735					dlm, dlm_assert_master_post_handler,
1736					&dlm->dlm_domain_handlers);
1737	if (status)
1738		goto bail;
1739
1740	status = o2net_register_handler(DLM_CREATE_LOCK_MSG, dlm->key,
1741					sizeof(struct dlm_create_lock),
1742					dlm_create_lock_handler,
1743					dlm, NULL, &dlm->dlm_domain_handlers);
1744	if (status)
1745		goto bail;
1746
1747	status = o2net_register_handler(DLM_CONVERT_LOCK_MSG, dlm->key,
1748					DLM_CONVERT_LOCK_MAX_LEN,
1749					dlm_convert_lock_handler,
1750					dlm, NULL, &dlm->dlm_domain_handlers);
1751	if (status)
1752		goto bail;
1753
1754	status = o2net_register_handler(DLM_UNLOCK_LOCK_MSG, dlm->key,
1755					DLM_UNLOCK_LOCK_MAX_LEN,
1756					dlm_unlock_lock_handler,
1757					dlm, NULL, &dlm->dlm_domain_handlers);
1758	if (status)
1759		goto bail;
1760
1761	status = o2net_register_handler(DLM_PROXY_AST_MSG, dlm->key,
1762					DLM_PROXY_AST_MAX_LEN,
1763					dlm_proxy_ast_handler,
1764					dlm, NULL, &dlm->dlm_domain_handlers);
1765	if (status)
1766		goto bail;
1767
1768	status = o2net_register_handler(DLM_EXIT_DOMAIN_MSG, dlm->key,
1769					sizeof(struct dlm_exit_domain),
1770					dlm_exit_domain_handler,
1771					dlm, NULL, &dlm->dlm_domain_handlers);
1772	if (status)
1773		goto bail;
1774
1775	status = o2net_register_handler(DLM_DEREF_LOCKRES_MSG, dlm->key,
1776					sizeof(struct dlm_deref_lockres),
1777					dlm_deref_lockres_handler,
1778					dlm, NULL, &dlm->dlm_domain_handlers);
1779	if (status)
1780		goto bail;
1781
1782	status = o2net_register_handler(DLM_MIGRATE_REQUEST_MSG, dlm->key,
1783					sizeof(struct dlm_migrate_request),
1784					dlm_migrate_request_handler,
1785					dlm, NULL, &dlm->dlm_domain_handlers);
1786	if (status)
1787		goto bail;
1788
1789	status = o2net_register_handler(DLM_MIG_LOCKRES_MSG, dlm->key,
1790					DLM_MIG_LOCKRES_MAX_LEN,
1791					dlm_mig_lockres_handler,
1792					dlm, NULL, &dlm->dlm_domain_handlers);
1793	if (status)
1794		goto bail;
1795
1796	status = o2net_register_handler(DLM_MASTER_REQUERY_MSG, dlm->key,
1797					sizeof(struct dlm_master_requery),
1798					dlm_master_requery_handler,
1799					dlm, NULL, &dlm->dlm_domain_handlers);
1800	if (status)
1801		goto bail;
1802
1803	status = o2net_register_handler(DLM_LOCK_REQUEST_MSG, dlm->key,
1804					sizeof(struct dlm_lock_request),
1805					dlm_request_all_locks_handler,
1806					dlm, NULL, &dlm->dlm_domain_handlers);
1807	if (status)
1808		goto bail;
1809
1810	status = o2net_register_handler(DLM_RECO_DATA_DONE_MSG, dlm->key,
1811					sizeof(struct dlm_reco_data_done),
1812					dlm_reco_data_done_handler,
1813					dlm, NULL, &dlm->dlm_domain_handlers);
1814	if (status)
1815		goto bail;
1816
1817	status = o2net_register_handler(DLM_BEGIN_RECO_MSG, dlm->key,
1818					sizeof(struct dlm_begin_reco),
1819					dlm_begin_reco_handler,
1820					dlm, NULL, &dlm->dlm_domain_handlers);
1821	if (status)
1822		goto bail;
1823
1824	status = o2net_register_handler(DLM_FINALIZE_RECO_MSG, dlm->key,
1825					sizeof(struct dlm_finalize_reco),
1826					dlm_finalize_reco_handler,
1827					dlm, NULL, &dlm->dlm_domain_handlers);
1828	if (status)
1829		goto bail;
1830
1831	status = o2net_register_handler(DLM_BEGIN_EXIT_DOMAIN_MSG, dlm->key,
1832					sizeof(struct dlm_exit_domain),
1833					dlm_begin_exit_domain_handler,
1834					dlm, NULL, &dlm->dlm_domain_handlers);
1835	if (status)
1836		goto bail;
1837
1838	status = o2net_register_handler(DLM_DEREF_LOCKRES_DONE, dlm->key,
1839					sizeof(struct dlm_deref_lockres_done),
1840					dlm_deref_lockres_done_handler,
1841					dlm, NULL, &dlm->dlm_domain_handlers);
1842bail:
1843	if (status)
1844		dlm_unregister_domain_handlers(dlm);
1845
1846	return status;
1847}
1848
1849static int dlm_join_domain(struct dlm_ctxt *dlm)
1850{
1851	int status;
1852	unsigned int backoff;
1853	unsigned int total_backoff = 0;
1854	char wq_name[O2NM_MAX_NAME_LEN];
1855
1856	BUG_ON(!dlm);
1857
1858	mlog(0, "Join domain %s\n", dlm->name);
1859
1860	status = dlm_register_domain_handlers(dlm);
1861	if (status) {
1862		mlog_errno(status);
1863		goto bail;
1864	}
1865
1866	status = dlm_launch_thread(dlm);
1867	if (status < 0) {
1868		mlog_errno(status);
1869		goto bail;
1870	}
1871
1872	status = dlm_launch_recovery_thread(dlm);
1873	if (status < 0) {
1874		mlog_errno(status);
1875		goto bail;
1876	}
1877
1878	dlm_debug_init(dlm);
1879
1880	snprintf(wq_name, O2NM_MAX_NAME_LEN, "dlm_wq-%s", dlm->name);
1881	dlm->dlm_worker = alloc_workqueue(wq_name, WQ_MEM_RECLAIM, 0);
1882	if (!dlm->dlm_worker) {
1883		status = -ENOMEM;
1884		mlog_errno(status);
1885		goto bail;
1886	}
1887
1888	do {
1889		status = dlm_try_to_join_domain(dlm);
1890
1891		/* If we're racing another node to the join, then we
1892		 * need to back off temporarily and let them
1893		 * complete. */
1894#define	DLM_JOIN_TIMEOUT_MSECS	90000
1895		if (status == -EAGAIN) {
1896			if (signal_pending(current)) {
1897				status = -ERESTARTSYS;
1898				goto bail;
1899			}
1900
1901			if (total_backoff > DLM_JOIN_TIMEOUT_MSECS) {
1902				status = -ERESTARTSYS;
1903				mlog(ML_NOTICE, "Timed out joining dlm domain "
1904				     "%s after %u msecs\n", dlm->name,
1905				     total_backoff);
1906				goto bail;
1907			}
1908
1909			/*
1910			 * <chip> After you!
1911			 * <dale> No, after you!
1912			 * <chip> I insist!
1913			 * <dale> But you first!
1914			 * ...
1915			 */
1916			backoff = (unsigned int)(jiffies & 0x3);
1917			backoff *= DLM_DOMAIN_BACKOFF_MS;
1918			total_backoff += backoff;
1919			mlog(0, "backoff %d\n", backoff);
1920			msleep(backoff);
1921		}
1922	} while (status == -EAGAIN);
1923
1924	if (status < 0) {
1925		mlog_errno(status);
1926		goto bail;
1927	}
1928
1929	status = 0;
1930bail:
1931	wake_up(&dlm_domain_events);
1932
1933	if (status) {
1934		dlm_unregister_domain_handlers(dlm);
1935		dlm_complete_thread(dlm);
1936		dlm_complete_recovery_thread(dlm);
1937		dlm_destroy_dlm_worker(dlm);
1938	}
1939
1940	return status;
1941}
1942
1943static struct dlm_ctxt *dlm_alloc_ctxt(const char *domain,
1944				u32 key)
1945{
1946	int i;
1947	int ret;
1948	struct dlm_ctxt *dlm = NULL;
1949
1950	dlm = kzalloc(sizeof(*dlm), GFP_KERNEL);
1951	if (!dlm) {
1952		ret = -ENOMEM;
1953		mlog_errno(ret);
1954		goto leave;
1955	}
1956
1957	dlm->name = kstrdup(domain, GFP_KERNEL);
1958	if (dlm->name == NULL) {
1959		ret = -ENOMEM;
1960		mlog_errno(ret);
1961		goto leave;
1962	}
1963
1964	dlm->lockres_hash = (struct hlist_head **)dlm_alloc_pagevec(DLM_HASH_PAGES);
1965	if (!dlm->lockres_hash) {
1966		ret = -ENOMEM;
1967		mlog_errno(ret);
1968		goto leave;
1969	}
1970
1971	for (i = 0; i < DLM_HASH_BUCKETS; i++)
1972		INIT_HLIST_HEAD(dlm_lockres_hash(dlm, i));
1973
1974	dlm->master_hash = (struct hlist_head **)
1975				dlm_alloc_pagevec(DLM_HASH_PAGES);
1976	if (!dlm->master_hash) {
1977		ret = -ENOMEM;
1978		mlog_errno(ret);
1979		goto leave;
1980	}
1981
1982	for (i = 0; i < DLM_HASH_BUCKETS; i++)
1983		INIT_HLIST_HEAD(dlm_master_hash(dlm, i));
1984
1985	dlm->key = key;
1986	dlm->node_num = o2nm_this_node();
1987
1988	dlm_create_debugfs_subroot(dlm);
1989
1990	spin_lock_init(&dlm->spinlock);
1991	spin_lock_init(&dlm->master_lock);
1992	spin_lock_init(&dlm->ast_lock);
1993	spin_lock_init(&dlm->track_lock);
1994	INIT_LIST_HEAD(&dlm->list);
1995	INIT_LIST_HEAD(&dlm->dirty_list);
1996	INIT_LIST_HEAD(&dlm->reco.resources);
1997	INIT_LIST_HEAD(&dlm->reco.node_data);
1998	INIT_LIST_HEAD(&dlm->purge_list);
1999	INIT_LIST_HEAD(&dlm->dlm_domain_handlers);
2000	INIT_LIST_HEAD(&dlm->tracking_list);
2001	dlm->reco.state = 0;
2002
2003	INIT_LIST_HEAD(&dlm->pending_asts);
2004	INIT_LIST_HEAD(&dlm->pending_basts);
2005
2006	mlog(0, "dlm->recovery_map=%p, &(dlm->recovery_map[0])=%p\n",
2007		  dlm->recovery_map, &(dlm->recovery_map[0]));
2008
2009	bitmap_zero(dlm->recovery_map, O2NM_MAX_NODES);
2010	bitmap_zero(dlm->live_nodes_map, O2NM_MAX_NODES);
2011	bitmap_zero(dlm->domain_map, O2NM_MAX_NODES);
2012
2013	dlm->dlm_thread_task = NULL;
2014	dlm->dlm_reco_thread_task = NULL;
2015	dlm->dlm_worker = NULL;
2016	init_waitqueue_head(&dlm->dlm_thread_wq);
2017	init_waitqueue_head(&dlm->dlm_reco_thread_wq);
2018	init_waitqueue_head(&dlm->reco.event);
2019	init_waitqueue_head(&dlm->ast_wq);
2020	init_waitqueue_head(&dlm->migration_wq);
2021	INIT_LIST_HEAD(&dlm->mle_hb_events);
2022
2023	dlm->joining_node = DLM_LOCK_RES_OWNER_UNKNOWN;
2024	init_waitqueue_head(&dlm->dlm_join_events);
2025
2026	dlm->migrate_done = 0;
2027
2028	dlm->reco.new_master = O2NM_INVALID_NODE_NUM;
2029	dlm->reco.dead_node = O2NM_INVALID_NODE_NUM;
2030
2031	atomic_set(&dlm->res_tot_count, 0);
2032	atomic_set(&dlm->res_cur_count, 0);
2033	for (i = 0; i < DLM_MLE_NUM_TYPES; ++i) {
2034		atomic_set(&dlm->mle_tot_count[i], 0);
2035		atomic_set(&dlm->mle_cur_count[i], 0);
2036	}
2037
2038	spin_lock_init(&dlm->work_lock);
2039	INIT_LIST_HEAD(&dlm->work_list);
2040	INIT_WORK(&dlm->dispatched_work, dlm_dispatch_work);
2041
2042	kref_init(&dlm->dlm_refs);
2043	dlm->dlm_state = DLM_CTXT_NEW;
2044
2045	INIT_LIST_HEAD(&dlm->dlm_eviction_callbacks);
2046
2047	mlog(0, "context init: refcount %u\n",
2048		  kref_read(&dlm->dlm_refs));
2049
2050	ret = 0;
2051leave:
2052	if (ret < 0 && dlm) {
2053		if (dlm->master_hash)
2054			dlm_free_pagevec((void **)dlm->master_hash,
2055					DLM_HASH_PAGES);
2056
2057		if (dlm->lockres_hash)
2058			dlm_free_pagevec((void **)dlm->lockres_hash,
2059					DLM_HASH_PAGES);
2060
2061		kfree(dlm->name);
2062		kfree(dlm);
2063		dlm = NULL;
2064	}
2065	return dlm;
2066}
2067
2068/*
2069 * Compare a requested locking protocol version against the current one.
2070 *
2071 * If the major numbers are different, they are incompatible.
2072 * If the current minor is greater than the request, they are incompatible.
2073 * If the current minor is less than or equal to the request, they are
2074 * compatible, and the requester should run at the current minor version.
2075 */
2076static int dlm_protocol_compare(struct dlm_protocol_version *existing,
2077				struct dlm_protocol_version *request)
2078{
2079	if (existing->pv_major != request->pv_major)
2080		return 1;
2081
2082	if (existing->pv_minor > request->pv_minor)
2083		return 1;
2084
2085	if (existing->pv_minor < request->pv_minor)
2086		request->pv_minor = existing->pv_minor;
2087
2088	return 0;
2089}
2090
2091/*
2092 * dlm_register_domain: one-time setup per "domain".
2093 *
2094 * The filesystem passes in the requested locking version via proto.
2095 * If registration was successful, proto will contain the negotiated
2096 * locking protocol.
2097 */
2098struct dlm_ctxt * dlm_register_domain(const char *domain,
2099			       u32 key,
2100			       struct dlm_protocol_version *fs_proto)
2101{
2102	int ret;
2103	struct dlm_ctxt *dlm = NULL;
2104	struct dlm_ctxt *new_ctxt = NULL;
2105
2106	if (strlen(domain) >= O2NM_MAX_NAME_LEN) {
2107		ret = -ENAMETOOLONG;
2108		mlog(ML_ERROR, "domain name length too long\n");
2109		goto leave;
2110	}
2111
2112	mlog(0, "register called for domain \"%s\"\n", domain);
2113
2114retry:
2115	dlm = NULL;
2116	if (signal_pending(current)) {
2117		ret = -ERESTARTSYS;
2118		mlog_errno(ret);
2119		goto leave;
2120	}
2121
2122	spin_lock(&dlm_domain_lock);
2123
2124	dlm = __dlm_lookup_domain(domain);
2125	if (dlm) {
2126		if (dlm->dlm_state != DLM_CTXT_JOINED) {
2127			spin_unlock(&dlm_domain_lock);
2128
2129			mlog(0, "This ctxt is not joined yet!\n");
2130			wait_event_interruptible(dlm_domain_events,
2131						 dlm_wait_on_domain_helper(
2132							 domain));
2133			goto retry;
2134		}
2135
2136		if (dlm_protocol_compare(&dlm->fs_locking_proto, fs_proto)) {
2137			spin_unlock(&dlm_domain_lock);
2138			mlog(ML_ERROR,
2139			     "Requested locking protocol version is not "
2140			     "compatible with already registered domain "
2141			     "\"%s\"\n", domain);
2142			ret = -EPROTO;
2143			goto leave;
2144		}
2145
2146		__dlm_get(dlm);
2147		dlm->num_joins++;
2148
2149		spin_unlock(&dlm_domain_lock);
2150
2151		ret = 0;
2152		goto leave;
2153	}
2154
2155	/* doesn't exist */
2156	if (!new_ctxt) {
2157		spin_unlock(&dlm_domain_lock);
2158
2159		new_ctxt = dlm_alloc_ctxt(domain, key);
2160		if (new_ctxt)
2161			goto retry;
2162
2163		ret = -ENOMEM;
2164		mlog_errno(ret);
2165		goto leave;
2166	}
2167
2168	/* a little variable switch-a-roo here... */
2169	dlm = new_ctxt;
2170	new_ctxt = NULL;
2171
2172	/* add the new domain */
2173	list_add_tail(&dlm->list, &dlm_domains);
2174	spin_unlock(&dlm_domain_lock);
2175
2176	/*
2177	 * Pass the locking protocol version into the join.  If the join
2178	 * succeeds, it will have the negotiated protocol set.
2179	 */
2180	dlm->dlm_locking_proto = dlm_protocol;
2181	dlm->fs_locking_proto = *fs_proto;
2182
2183	ret = dlm_join_domain(dlm);
2184	if (ret) {
2185		mlog_errno(ret);
2186		dlm_put(dlm);
2187		goto leave;
2188	}
2189
2190	/* Tell the caller what locking protocol we negotiated */
2191	*fs_proto = dlm->fs_locking_proto;
2192
2193	ret = 0;
2194leave:
2195	if (new_ctxt)
2196		dlm_free_ctxt_mem(new_ctxt);
2197
2198	if (ret < 0)
2199		dlm = ERR_PTR(ret);
2200
2201	return dlm;
2202}
2203EXPORT_SYMBOL_GPL(dlm_register_domain);
2204
2205static LIST_HEAD(dlm_join_handlers);
2206
2207static void dlm_unregister_net_handlers(void)
2208{
2209	o2net_unregister_handler_list(&dlm_join_handlers);
2210}
2211
2212static int dlm_register_net_handlers(void)
2213{
2214	int status = 0;
2215
2216	status = o2net_register_handler(DLM_QUERY_JOIN_MSG, DLM_MOD_KEY,
2217					sizeof(struct dlm_query_join_request),
2218					dlm_query_join_handler,
2219					NULL, NULL, &dlm_join_handlers);
2220	if (status)
2221		goto bail;
2222
2223	status = o2net_register_handler(DLM_ASSERT_JOINED_MSG, DLM_MOD_KEY,
2224					sizeof(struct dlm_assert_joined),
2225					dlm_assert_joined_handler,
2226					NULL, NULL, &dlm_join_handlers);
2227	if (status)
2228		goto bail;
2229
2230	status = o2net_register_handler(DLM_CANCEL_JOIN_MSG, DLM_MOD_KEY,
2231					sizeof(struct dlm_cancel_join),
2232					dlm_cancel_join_handler,
2233					NULL, NULL, &dlm_join_handlers);
2234	if (status)
2235		goto bail;
2236
2237	status = o2net_register_handler(DLM_QUERY_REGION, DLM_MOD_KEY,
2238					sizeof(struct dlm_query_region),
2239					dlm_query_region_handler,
2240					NULL, NULL, &dlm_join_handlers);
2241
2242	if (status)
2243		goto bail;
2244
2245	status = o2net_register_handler(DLM_QUERY_NODEINFO, DLM_MOD_KEY,
2246					sizeof(struct dlm_query_nodeinfo),
2247					dlm_query_nodeinfo_handler,
2248					NULL, NULL, &dlm_join_handlers);
2249bail:
2250	if (status < 0)
2251		dlm_unregister_net_handlers();
2252
2253	return status;
2254}
2255
2256/* Domain eviction callback handling.
2257 *
2258 * The file system requires notification of node death *before* the
2259 * dlm completes it's recovery work, otherwise it may be able to
2260 * acquire locks on resources requiring recovery. Since the dlm can
2261 * evict a node from it's domain *before* heartbeat fires, a similar
2262 * mechanism is required. */
2263
2264/* Eviction is not expected to happen often, so a per-domain lock is
2265 * not necessary. Eviction callbacks are allowed to sleep for short
2266 * periods of time. */
2267static DECLARE_RWSEM(dlm_callback_sem);
2268
2269void dlm_fire_domain_eviction_callbacks(struct dlm_ctxt *dlm,
2270					int node_num)
2271{
2272	struct dlm_eviction_cb *cb;
2273
2274	down_read(&dlm_callback_sem);
2275	list_for_each_entry(cb, &dlm->dlm_eviction_callbacks, ec_item) {
2276		cb->ec_func(node_num, cb->ec_data);
2277	}
2278	up_read(&dlm_callback_sem);
2279}
2280
2281void dlm_setup_eviction_cb(struct dlm_eviction_cb *cb,
2282			   dlm_eviction_func *f,
2283			   void *data)
2284{
2285	INIT_LIST_HEAD(&cb->ec_item);
2286	cb->ec_func = f;
2287	cb->ec_data = data;
2288}
2289EXPORT_SYMBOL_GPL(dlm_setup_eviction_cb);
2290
2291void dlm_register_eviction_cb(struct dlm_ctxt *dlm,
2292			      struct dlm_eviction_cb *cb)
2293{
2294	down_write(&dlm_callback_sem);
2295	list_add_tail(&cb->ec_item, &dlm->dlm_eviction_callbacks);
2296	up_write(&dlm_callback_sem);
2297}
2298EXPORT_SYMBOL_GPL(dlm_register_eviction_cb);
2299
2300void dlm_unregister_eviction_cb(struct dlm_eviction_cb *cb)
2301{
2302	down_write(&dlm_callback_sem);
2303	list_del_init(&cb->ec_item);
2304	up_write(&dlm_callback_sem);
2305}
2306EXPORT_SYMBOL_GPL(dlm_unregister_eviction_cb);
2307
2308static int __init dlm_init(void)
2309{
2310	int status;
2311
2312	status = dlm_init_mle_cache();
2313	if (status) {
2314		mlog(ML_ERROR, "Could not create o2dlm_mle slabcache\n");
2315		goto error;
2316	}
2317
2318	status = dlm_init_master_caches();
2319	if (status) {
2320		mlog(ML_ERROR, "Could not create o2dlm_lockres and "
2321		     "o2dlm_lockname slabcaches\n");
2322		goto error;
2323	}
2324
2325	status = dlm_init_lock_cache();
2326	if (status) {
2327		mlog(ML_ERROR, "Count not create o2dlm_lock slabcache\n");
2328		goto error;
2329	}
2330
2331	status = dlm_register_net_handlers();
2332	if (status) {
2333		mlog(ML_ERROR, "Unable to register network handlers\n");
2334		goto error;
2335	}
2336
2337	dlm_create_debugfs_root();
2338
2339	return 0;
2340error:
2341	dlm_unregister_net_handlers();
2342	dlm_destroy_lock_cache();
2343	dlm_destroy_master_caches();
2344	dlm_destroy_mle_cache();
2345	return -1;
2346}
2347
2348static void __exit dlm_exit (void)
2349{
2350	dlm_destroy_debugfs_root();
2351	dlm_unregister_net_handlers();
2352	dlm_destroy_lock_cache();
2353	dlm_destroy_master_caches();
2354	dlm_destroy_mle_cache();
2355}
2356
2357MODULE_AUTHOR("Oracle");
2358MODULE_LICENSE("GPL");
2359MODULE_DESCRIPTION("OCFS2 Distributed Lock Management");
2360
2361module_init(dlm_init);
2362module_exit(dlm_exit);