Changeset 2943 for pjproject/trunk/pjsip/src/pjsua-lib/pjsua_core.c

Timestamp:

Oct 14, 2009 1:58:04 AM (15 years ago)

Author:

bennylp

Message:

Ticket #970: More gracefull PJSUA-LIB shutdown sequence. Enhancements:

• wait for unregistration to complete (or a preconfigured delay expires)
• new account config field to set the maximum delay to wait for unregistration
• rejects incoming requests (INVITE, SUBSCRIBE, and OPTIONS) when shutdown is in progress

File:

• pjproject/trunk/pjsip/src/pjsua-lib/pjsua_core.c (modified) (7 diffs)

pjproject/trunk/pjsip/src/pjsua-lib/pjsua_core.c

@@ -162 +162 @@
 
     cfg->reg_timeout = PJSUA_REG_INTERVAL;
+    cfg->unreg_timeout = PJSUA_UNREG_TIMEOUT;
     pjsip_publishc_opt_default(&cfg->publish_opt);
     cfg->unpublish_max_wait_time_msec = PJSUA_UNPUBLISH_MAX_WAIT_TIME_MSEC;
@@ -304 +305 @@
     {
         return PJ_FALSE;
+    }
+
+    /* Don't want to handle if shutdown is in progress */
+    if (pjsua_var.thread_quit_flag) {
+        pjsip_endpt_respond_stateless(pjsua_var.endpt, rdata, 
+                                      PJSIP_SC_TEMPORARILY_UNAVAILABLE, NULL,
+                                      NULL, NULL);
+        return PJ_TRUE;
     }
 
@@ -1232 +1241 @@
         unsigned max_wait;
 
+        PJ_LOG(4,(THIS_FILE, "Shutting down..."));
+
         /* Terminate all calls. */
         pjsua_call_hangup_all();
@@ -1245 +1256 @@
         /* Terminate all presence subscriptions. */
         pjsua_pres_shutdown();
+
+        /* Destroy media (to shutdown media transports etc) */
+        pjsua_media_subsys_destroy();
 
         /* Wait for sometime until all publish client sessions are done
@@ -1291 +1305 @@
             }
         }
-    }
-
-    /* Destroy endpoint. */
-    if (pjsua_var.endpt) {
 
         /* Terminate any pending STUN resolution */
@@ -1306 +1316 @@
         }
 
+        /* Wait until all unregistrations are done (ticket #364) */
+        /* First stage, get the maximum wait time */
+        max_wait = 100;
+        for (i=0; i<PJ_ARRAY_SIZE(pjsua_var.acc); ++i) {
+            if (!pjsua_var.acc[i].valid)
+                continue;
+            if (pjsua_var.acc[i].cfg.unreg_timeout > max_wait)
+                max_wait = pjsua_var.acc[i].cfg.unreg_timeout;
+        }
+        
+        /* Second stage, wait for unregistrations to complete */
+        for (i=0; i<(int)(max_wait/50); ++i) {
+            unsigned j;
+            for (j=0; j<PJ_ARRAY_SIZE(pjsua_var.acc); ++j) {
+                if (!pjsua_var.acc[j].valid)
+                    continue;
+
+                if (pjsua_var.acc[j].regc)
+                    break;
+            }
+            if (j != PJ_ARRAY_SIZE(pjsua_var.acc))
+                busy_sleep(50);
+            else
+                break;
+        }
+        /* Note variable 'i' is used below */
+
         /* Wait for some time to allow unregistration and ICE/TURN
          * transports shutdown to complete: 
-        */
-        PJ_LOG(4,(THIS_FILE, "Shutting down..."));
-        busy_sleep(1000);
+         */
+        if (i < 20)
+            busy_sleep(1000 - i*50);
 
         PJ_LOG(4,(THIS_FILE, "Destroying..."));
-
-        /* Terminate all calls again, just in case there's new call
-         * picked up during busy_sleep()
-         */
-        pjsua_call_hangup_all();
-
-        /* Destroy media after all polling is done, as there may be
-         * incoming request that needs handling (e.g. OPTIONS)
-         */
-        pjsua_media_subsys_destroy();
 
         /* Must destroy endpoint first before destroying pools in
@@ -1347 +1374 @@
             }
         }
-    } else {
-        /* Destroy media */
-        pjsua_media_subsys_destroy();
     }