X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/5fd8cc8409d0efadc07dfe8d6774ad9ff477663d..7312c5fd3b74fd9cad485121bc31fc501accd355:/handle.cc

diff --git a/handle.cc b/handle.cc
index a233d5b..ff38a56 100644
--- a/handle.cc
+++ b/handle.cc
@@ -1,114 +1,110 @@
 #include "handle.h"
 #include <stdio.h>
 #include "tprintf.h"
+#include "lock.h"
 
 handle_mgr mgr;
 
 handle::handle(std::string m) 
 {
-  h = mgr.get_handle(m);
+    h = mgr.get_handle(m);
 }
 
 rpcc *
 handle::safebind()
 {
-  if (!h)
-    return NULL;
-  ScopedLock ml(&h->cl_mutex);
-  if (h->del)
-    return NULL;
-  if (h->cl)
+    if (!h)
+        return NULL;
+    lock ml(h->cl_mutex);
+    if (h->del)
+        return NULL;
+    if (h->cl)
+        return h->cl;
+    sockaddr_in dstsock;
+    make_sockaddr(h->m.c_str(), &dstsock);
+    rpcc *cl = new rpcc(dstsock);
+    tprintf("handler_mgr::get_handle trying to bind...%s\n", h->m.c_str());
+    int ret;
+    // The test script assumes that the failure can be detected by paxos and
+    // rsm layer within few seconds. We have to set the timeout with a small
+    // value to support the assumption.
+    // 
+    // With RPC_LOSSY=5, tests may fail due to delays and time outs.
+    ret = cl->bind(rpcc::to(1000));
+    if (ret < 0) {
+        tprintf("handle_mgr::get_handle bind failure! %s %d\n", h->m.c_str(), ret);
+        delete cl;
+        h->del = true;
+    } else {
+        tprintf("handle_mgr::get_handle bind succeeded %s\n", h->m.c_str());
+        h->cl = cl;
+    }
     return h->cl;
-  sockaddr_in dstsock;
-  make_sockaddr(h->m.c_str(), &dstsock);
-  rpcc *cl = new rpcc(dstsock);
-  tprintf("handler_mgr::get_handle trying to bind...%s\n", h->m.c_str());
-  int ret;
-  // Starting with lab 6, our test script assumes that the failure
-  // can be detected by paxos and rsm layer within few seconds. We have
-  // to set the timeout with a small value to support the assumption.
-  // 
-  // Note: with RPC_LOSSY=5, your lab would failed to pass the tests of
-  // lab 6 and lab 7 because the rpc layer may delay your RPC request, 
-  // and cause a time out failure. Please make sure RPC_LOSSY is set to 0.
-  ret = cl->bind(rpcc::to(1000));
-  if (ret < 0) {
-    tprintf("handle_mgr::get_handle bind failure! %s %d\n", h->m.c_str(), ret);
-    delete cl;
-    h->del = true;
-  } else {
-    tprintf("handle_mgr::get_handle bind succeeded %s\n", h->m.c_str());
-    h->cl = cl;
-  }
-  return h->cl;
 }
 
 handle::~handle() 
 {
-  if (h) mgr.done_handle(h);
+    if (h) mgr.done_handle(h);
 }
 
 handle_mgr::handle_mgr()
 {
-  VERIFY (pthread_mutex_init(&handle_mutex, NULL) == 0);
 }
 
 struct hinfo *
 handle_mgr::get_handle(std::string m)
 {
-  ScopedLock ml(&handle_mutex);
-  struct hinfo *h = 0;
-  if (hmap.find(m) == hmap.end()) {
-    h = new hinfo;
-    h->cl = NULL;
-    h->del = false;
-    h->refcnt = 1;
-    h->m = m;
-    pthread_mutex_init(&h->cl_mutex, NULL);
-    hmap[m] = h;
-  } else if (!hmap[m]->del) {
-    h = hmap[m];
-    h->refcnt ++;
-  }
-  return h;
+    lock ml(handle_mutex);
+    struct hinfo *h = 0;
+    if (hmap.find(m) == hmap.end()) {
+        h = new hinfo;
+        h->cl = NULL;
+        h->del = false;
+        h->refcnt = 1;
+        h->m = m;
+        hmap[m] = h;
+    } else if (!hmap[m]->del) {
+        h = hmap[m];
+        h->refcnt ++;
+    }
+    return h;
 }
 
 void 
 handle_mgr::done_handle(struct hinfo *h)
 {
-  ScopedLock ml(&handle_mutex);
-  h->refcnt--;
-  if (h->refcnt == 0 && h->del)
-    delete_handle_wo(h->m);
+    lock ml(handle_mutex);
+    h->refcnt--;
+    if (h->refcnt == 0 && h->del)
+        delete_handle_wo(h->m);
 }
 
 void
 handle_mgr::delete_handle(std::string m)
 {
-  ScopedLock ml(&handle_mutex);
-  delete_handle_wo(m);
+    lock ml(handle_mutex);
+    delete_handle_wo(m);
 }
 
 // Must be called with handle_mutex locked.
 void
 handle_mgr::delete_handle_wo(std::string m)
 {
-  if (hmap.find(m) == hmap.end()) {
-    tprintf("handle_mgr::delete_handle_wo: cl %s isn't in cl list\n", m.c_str());
-  } else {
-    tprintf("handle_mgr::delete_handle_wo: cl %s refcnt %d\n", m.c_str(),
-	   hmap[m]->refcnt);
-    struct hinfo *h = hmap[m];
-    if (h->refcnt == 0) {
-      if (h->cl) {
-        h->cl->cancel();
-        delete h->cl;
-      }
-      pthread_mutex_destroy(&h->cl_mutex);
-      hmap.erase(m);
-      delete h;
+    if (hmap.find(m) == hmap.end()) {
+        tprintf("handle_mgr::delete_handle_wo: cl %s isn't in cl list\n", m.c_str());
     } else {
-      h->del = true;
+        tprintf("handle_mgr::delete_handle_wo: cl %s refcnt %d\n", m.c_str(),
+                hmap[m]->refcnt);
+        struct hinfo *h = hmap[m];
+        if (h->refcnt == 0) {
+            if (h->cl) {
+                h->cl->cancel();
+                delete h->cl;
+            }
+            hmap.erase(m);
+            delete h;
+        } else {
+            h->del = true;
+        }
     }
-  }
 }