svn rev #25350: trunk/src/kdc/

[ghudson@MIT.EDU]

http://src.mit.edu/fisheye/changelog/krb5/?cs=25350
Commit By: ghudson
Log Message:
Drop retransmits while processing requests.

Supporting asynchronous preauth modules means that the KDC can receive
a retransmitted request before it finishes processing the initial
request.  Ignore those retransmits instead of processing them.


Changed Files:
U   trunk/src/kdc/dispatch.c
U   trunk/src/kdc/kdc_util.h
U   trunk/src/kdc/replay.c
Modified: trunk/src/kdc/dispatch.c
===================================================================
--- trunk/src/kdc/dispatch.c	2011-10-15 15:08:02 UTC (rev 25349)
+++ trunk/src/kdc/dispatch.c	2011-10-15 15:35:46 UTC (rev 25350)
@@ -54,6 +54,11 @@
     oldrespond = state->respond;
     oldarg = state->arg;
 
+#ifndef NOCACHE
+    /* Remove our NULL cache entry to indicate request completion. */
+    kdc_remove_lookaside(kdc_context, state->request);
+#endif
+
     if (state->is_tcp == 0 && response &&
         response->length > max_dgram_reply_size) {
         krb5_free_data(kdc_context, response);
@@ -67,7 +72,7 @@
 
 #ifndef NOCACHE
     /* put the response into the lookaside buffer */
-    else if (!code)
+    else if (!code && response)
         kdc_insert_lookaside(state->request, response);
 #endif
 
@@ -104,20 +109,30 @@
         const char *name = 0;
         char buf[46];
 
-        if (is_tcp != 0 || response->length <= max_dgram_reply_size) {
+        if (!response || is_tcp != 0 ||
+            response->length <= max_dgram_reply_size) {
             name = inet_ntop (ADDRTYPE2FAMILY (from->address->addrtype),
                               from->address->contents, buf, sizeof (buf));
             if (name == 0)
                 name = "[unknown address type]";
-            krb5_klog_syslog(LOG_INFO,
-                             "DISPATCH: repeated (retransmitted?) request "
-                             "from %s, resending previous response",
-                             name);
+            if (response)
+                krb5_klog_syslog(LOG_INFO,
+                                 "DISPATCH: repeated (retransmitted?) request "
+                                 "from %s, resending previous response", name);
+            else
+                krb5_klog_syslog(LOG_INFO,
+                                 "DISPATCH: repeated (retransmitted?) request "
+                                 "from %s during request processing, dropping "
+                                 "repeated request", name);
         }
 
-        finish_dispatch(state, 0, response);
+        finish_dispatch(state, response ? 0 : KRB5KDC_ERR_DISCARD, response);
         return;
     }
+
+    /* Insert a NULL entry into the lookaside to indicate that this request
+     * is currently being processed. */
+    kdc_insert_lookaside(pkt, NULL);
 #endif
 
     retval = krb5_crypto_us_timeofday(&now, &now_usec);

Modified: trunk/src/kdc/kdc_util.h
===================================================================
--- trunk/src/kdc/kdc_util.h	2011-10-15 15:08:02 UTC (rev 25349)
+++ trunk/src/kdc/kdc_util.h	2011-10-15 15:35:46 UTC (rev 25350)
@@ -236,6 +236,7 @@
 /* replay.c */
 krb5_boolean kdc_check_lookaside (krb5_data *, krb5_data **);
 void kdc_insert_lookaside (krb5_data *, krb5_data *);
+void kdc_remove_lookaside (krb5_context kcontext, krb5_data *);
 void kdc_free_lookaside(krb5_context);
 
 /* kdc_util.c */

Modified: trunk/src/kdc/replay.c
===================================================================
--- trunk/src/kdc/replay.c	2011-10-15 15:08:02 UTC (rev 25349)
+++ trunk/src/kdc/replay.c	2011-10-15 15:35:46 UTC (rev 25350)
@@ -55,6 +55,29 @@
    Todo:  quench the size of the queue...
 */
 
+/* Removes the most recent cache entry for a given packet. */
+void
+kdc_remove_lookaside(krb5_context kcontext, krb5_data *inpkt)
+{
+    register krb5_kdc_replay_ent *eptr, *last;
+
+    if (!root_ptr.next)
+        return;
+
+    for (last = &root_ptr, eptr = root_ptr.next;
+         eptr;
+         last = eptr, eptr = eptr->next) {
+        if (!MATCH(eptr))
+            continue;
+
+        last->next = eptr->next;
+        krb5_free_data(kcontext, eptr->req_packet);
+        krb5_free_data(kcontext, eptr->reply_packet);
+        free(eptr);
+        return;
+    }
+}
+
 /* return TRUE if outpkt is filled in with a packet to reply with,
    FALSE if the caller should do the work */