From 828c6786452e18fff15ff223ae235cebd1a0badb Mon Sep 17 00:00:00 2001
From: Stephen Gallagher <sgallagh@redhat.com>
Date: Thu, 26 Mar 2009 10:55:05 -0400
Subject: [PATCH] Enable autoreconnection to the Data Provider in PAM

---
 server/responder/common/responder_common.c |    6 ++-
 server/responder/common/responder_common.h |    3 +-
 server/responder/pam/pamsrv.c              |   62 +++++++++++++++++++++++++++-
 3 files changed, 67 insertions(+), 4 deletions(-)

diff --git a/server/responder/common/responder_common.c b/server/responder/common/responder_common.c
index 3780184..3b566b9 100644
--- a/server/responder/common/responder_common.c
+++ b/server/responder/common/responder_common.c
@@ -498,7 +498,8 @@ int sss_process_init(TALLOC_CTX *mem_ctx,
                      const char *sss_pipe_name,
                      const char *sss_priv_pipe_name,
                      const char *confdb_socket_path,
-                     struct sbus_method dp_methods[])
+                     struct sbus_method dp_methods[],
+                     struct resp_ctx **responder_ctx)
 {
     struct resp_ctx *rctx;
     int ret;
@@ -550,8 +551,9 @@ int sss_process_init(TALLOC_CTX *mem_ctx,
 
     rctx->cache_timeout = 600; /* FIXME: read from conf */
 
-    DEBUG(1, ("NSS Initialization complete\n"));
+    DEBUG(1, ("Responder Initialization complete\n"));
 
+    *responder_ctx = rctx;
     return EOK;
 }
 
diff --git a/server/responder/common/responder_common.h b/server/responder/common/responder_common.h
index 0a5b627..66e01f9 100644
--- a/server/responder/common/responder_common.h
+++ b/server/responder/common/responder_common.h
@@ -17,7 +17,8 @@ int sss_process_init(TALLOC_CTX *mem_ctx,
                      const char *sss_pipe_name,
                      const char *sss_priv_pipe_name,
                      const char *confdb_socket_path,
-                     struct sbus_method dp_methods[]);
+                     struct sbus_method dp_methods[],
+                     struct resp_ctx **responder_ctx);
 
 int sss_parse_name(TALLOC_CTX *memctx,
                    const char *fullname,
diff --git a/server/responder/pam/pamsrv.c b/server/responder/pam/pamsrv.c
index 9952289..4abc0a2 100644
--- a/server/responder/pam/pamsrv.c
+++ b/server/responder/pam/pamsrv.c
@@ -109,6 +109,8 @@ static int service_pong(DBusMessage *message, struct sbus_conn_ctx *sconn)
     return EOK;
 }
 
+static void pam_shutdown(struct resp_ctx *ctx);
+
 static int service_reload(DBusMessage *message, struct sbus_conn_ctx *sconn) {
     /* Monitor calls this function when we need to reload
      * our configuration information. Perform whatever steps
@@ -119,6 +121,59 @@ static int service_reload(DBusMessage *message, struct sbus_conn_ctx *sconn) {
     return service_pong(message, sconn);
 }
 
+static void pam_dp_reconnect_init(struct sbus_conn_ctx *sconn, int status, void *pvt)
+{
+    int ret;
+    struct resp_ctx *rctx = talloc_get_type(pvt, struct resp_ctx);
+
+    /* Did we reconnect successfully? */
+    if (status == SBUS_RECONNECT_SUCCESS) {
+        /* Add the methods back to the new connection */
+        ret = sbus_conn_add_method_ctx(rctx->dp_ctx->scon_ctx,
+                                       rctx->dp_ctx->sm_ctx);
+        if (ret != EOK) {
+            DEBUG(0, ("Could not re-add methods on reconnection.\n"));
+            pam_shutdown(rctx);
+        }
+
+        DEBUG(1, ("Reconnected to the Data Provider.\n"));
+        return;
+    }
+
+    /* Handle failure */
+    DEBUG(0, ("Could not reconnect to data provider.\n"));
+    /* Kill the backend and let the monitor restart it */
+    pam_shutdown(rctx);
+}
+
+static void pam_shutdown(struct resp_ctx *rctx)
+{
+    /* TODO: Do clean-up here */
+
+    /* Nothing left to do but exit() */
+    exit(0);
+}
+
+
+static int pam_process_init(struct main_context *main_ctx,
+                            struct resp_ctx *rctx)
+{
+    int ret, max_retries;
+
+    /* Enable automatic reconnection to the Data Provider */
+    ret = confdb_get_int(rctx->cdb, rctx, rctx->confdb_socket_path,
+                         "retries", 3, &max_retries);
+    if (ret != EOK) {
+        DEBUG(0, ("Failed to set up automatic reconnection\n"));
+        return ret;
+    }
+
+    sbus_reconnect_init(rctx->dp_ctx->scon_ctx, max_retries,
+                        pam_dp_reconnect_init, rctx);
+
+    return EOK;
+}
+
 int main(int argc, const char *argv[])
 {
     int opt;
@@ -127,6 +182,7 @@ int main(int argc, const char *argv[])
     int ret;
     struct sbus_method *pam_dp_methods;
     struct sss_cmd_table *sss_cmds;
+    struct resp_ctx *rctx;
 
 	struct poptOption long_options[] = {
 		POPT_AUTOHELP
@@ -161,9 +217,13 @@ int main(int argc, const char *argv[])
                            SSS_PAM_SOCKET_NAME,
                            SSS_PAM_PRIV_SOCKET_NAME,
                            CONFDB_SOCKET_PATH,
-                           pam_dp_methods);
+                           pam_dp_methods,
+                           &rctx);
     if (ret != EOK) return 3;
 
+    ret = pam_process_init(main_ctx, rctx);
+    if (ret != EOK) return 4;
+
     /* loop on main */
     server_loop(main_ctx);
 
-- 
1.6.0.6