DLM: save / restore all socket callbacks

index dc9ae6d..f9eb366 100644
@@ -124,7 +124,10 @@ struct connection {
 	struct connection *othercon;
 	struct work_struct rwork; /* Receive workqueue */
 	struct work_struct swork; /* Send workqueue */
-	void (*orig_error_report)(struct sock *sk);
+	void (*orig_error_report)(struct sock *);
+	void (*orig_data_ready)(struct sock *);
+	void (*orig_state_change)(struct sock *);
+	void (*orig_write_space)(struct sock *);
 };
 #define sock2con(x) ((struct connection *)(x)->sk_user_data)
 

@@ -465,12 +468,43 @@ int dlm_lowcomms_connect_node(int nodeid)
 	return 0;
 }
 
-static void lowcomms_error_report(struct sock *sk)
+static void save_old_callbacks(struct sock *sk)
+{
+	struct connection *con = sock2con(sk);
+
+	if (sk->sk_error_report) {
+		con->orig_data_ready = sk->sk_data_ready;
+		con->orig_state_change = sk->sk_state_change;
+		con->orig_write_space = sk->sk_write_space;
+		con->orig_error_report = sk->sk_error_report;
+	}
+}
+
+static void restore_old_callbacks(struct sock *sk)
 {
 	struct connection *con = sock2con(sk);
+
+	if (con->orig_error_report) {
+		sk->sk_data_ready = con->orig_data_ready;
+		sk->sk_state_change = con->orig_state_change;
+		sk->sk_write_space = con->orig_write_space;
+		sk->sk_error_report = con->orig_error_report;
+	}
+}
+
+static void lowcomms_error_report(struct sock *sk)
+{
+	struct connection *con;
 	struct sockaddr_storage saddr;
 	int buflen;
+	void (*orig_report)(struct sock *) = NULL;
+
+	read_lock_bh(&sk->sk_callback_lock);
+	con = sock2con(sk);
+	if (con == NULL)
+		goto out;
 
+	orig_report = con->orig_error_report;
 	if (con->sock == NULL ||
 	    kernel_getpeername(con->sock, (struct sockaddr *)&saddr, &buflen)) {
 		printk_ratelimited(KERN_ERR "dlm: node %d: socket error "

@@ -478,7 +512,6 @@ static void lowcomms_error_report(struct sock *sk)
 				   "sk_err=%d/%d\n", dlm_our_nodeid(),
 				   con->nodeid, dlm_config.ci_tcp_port,
 				   sk->sk_err, sk->sk_err_soft);
-		return;
 	} else if (saddr.ss_family == AF_INET) {
 		struct sockaddr_in *sin4 = (struct sockaddr_in *)&saddr;
 

@@ -501,22 +534,31 @@ static void lowcomms_error_report(struct sock *sk)
 				   dlm_config.ci_tcp_port, sk->sk_err,
 				   sk->sk_err_soft);
 	}
-	con->orig_error_report(sk);
+out:
+	read_unlock_bh(&sk->sk_callback_lock);
+	if (orig_report)
+		orig_report(sk);
 }
 
 /* Make a socket active */
 static void add_sock(struct socket *sock, struct connection *con)
 {
+	struct sock *sk = sock->sk;
+
+	write_lock_bh(&sk->sk_callback_lock);
 	con->sock = sock;
 
-	/* Install a data_ready callback */
-	con->sock->sk->sk_data_ready = lowcomms_data_ready;
-	con->sock->sk->sk_write_space = lowcomms_write_space;
-	con->sock->sk->sk_state_change = lowcomms_state_change;
-	con->sock->sk->sk_user_data = con;
-	con->sock->sk->sk_allocation = GFP_NOFS;
-	con->orig_error_report = con->sock->sk->sk_error_report;
-	con->sock->sk->sk_error_report = lowcomms_error_report;
+	if (sk->sk_error_report != lowcomms_error_report) {
+		sk->sk_user_data = con;
+		save_old_callbacks(sk);
+		/* Install a data_ready callback */
+		sk->sk_data_ready = lowcomms_data_ready;
+		sk->sk_write_space = lowcomms_write_space;
+		sk->sk_state_change = lowcomms_state_change;
+		sk->sk_allocation = GFP_NOFS;
+		sk->sk_error_report = lowcomms_error_report;
+	}
+	write_unlock_bh(&sk->sk_callback_lock);
 }
 
 /* Add the port number to an IPv6 or 4 sockaddr and return the address

@@ -551,6 +593,13 @@ static void close_connection(struct connection *con, bool and_other,
 
 	mutex_lock(&con->sock_mutex);
 	if (con->sock) {
+		struct sock *sk = con->sock->sk;
+
+		if (sk) {
+			write_lock_bh(&sk->sk_callback_lock);
+			restore_old_callbacks(sk);
+			write_unlock_bh(&sk->sk_callback_lock);
+		}
 		sock_release(con->sock);
 		con->sock = NULL;
 	}

@@ -1131,6 +1180,10 @@ static void tcp_connect_to_sock(struct connection *con)
 	if (result == 0)
 		goto out;
 
+	write_lock_bh(&sock->sk->sk_callback_lock);
+	restore_old_callbacks(sock->sk);
+	write_unlock_bh(&sock->sk->sk_callback_lock);
+
 out_err:
 	if (con->sock) {
 		sock_release(con->sock);

@@ -1273,13 +1326,19 @@ static int sctp_listen_for_all(void)
 	if (result < 0)
 		log_print("Could not set SCTP NODELAY error %d\n", result);
 
+	write_lock_bh(&sock->sk->sk_callback_lock);
 	/* Init con struct */
 	sock->sk->sk_user_data = con;
+
+	save_old_callbacks(sock->sk);
+
 	con->sock = sock;
 	con->sock->sk->sk_data_ready = lowcomms_data_ready;
 	con->rx_action = sctp_accept_from_sock;
 	con->connect_action = sctp_connect_to_sock;
 
+	write_unlock_bh(&sock->sk->sk_callback_lock);
+
 	/* Bind to all addresses. */
 	if (sctp_bind_addrs(con, dlm_config.ci_tcp_port))
 		goto create_delsock;

@@ -1293,6 +1352,9 @@ static int sctp_listen_for_all(void)
 	return 0;
 
 create_delsock:
+	write_lock_bh(&sock->sk->sk_callback_lock);
+	restore_old_callbacks(sock->sk);
+	write_unlock_bh(&sock->sk->sk_callback_lock);
 	sock_release(sock);
 	con->sock = NULL;
 out: