1 files changed, 68 insertions, 0 deletions
diff --git a/dom/system/gonk/NetIdManager.cpp b/dom/system/gonk/NetIdManager.cpp
new file mode 100644
index 000000000..510ec8b22
--- /dev/null
+++ b/dom/system/gonk/NetIdManager.cpp
@@ -0,0 +1,68 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "NetIdManager.h"
+
+NetIdManager::NetIdManager()
+  : mNextNetId(MIN_NET_ID)
+{
+}
+
+int NetIdManager::getNextNetId()
+{
+  // Modified from
+  // http://androidxref.com/5.0.0_r2/xref/frameworks/base/services/
+  //        core/java/com/android/server/ConnectivityService.java#764
+
+  int netId = mNextNetId;
+  if (++mNextNetId > MAX_NET_ID) {
+    mNextNetId = MIN_NET_ID;
+  }
+
+  return netId;
+}
+
+void NetIdManager::acquire(const nsString& aInterfaceName,
+                           NetIdInfo* aNetIdInfo)
+{
+  // Lookup or create one.
+  if (!mInterfaceToNetIdHash.Get(aInterfaceName, aNetIdInfo)) {
+    aNetIdInfo->mNetId = getNextNetId();
+    aNetIdInfo->mRefCnt = 1;
+  } else {
+    aNetIdInfo->mRefCnt++;
+  }
+
+  // Update hash and return.
+  mInterfaceToNetIdHash.Put(aInterfaceName, *aNetIdInfo);
+
+  return;
+}
+
+bool NetIdManager::lookup(const nsString& aInterfaceName,
+                          NetIdInfo* aNetIdInfo)
+{
+  return mInterfaceToNetIdHash.Get(aInterfaceName, aNetIdInfo);
+}
+
+bool NetIdManager::release(const nsString& aInterfaceName,
+                           NetIdInfo* aNetIdInfo)
+{
+  if (!mInterfaceToNetIdHash.Get(aInterfaceName, aNetIdInfo)) {
+    return false; // No such key.
+  }
+
+  aNetIdInfo->mRefCnt--;
+
+  // Update the hash if still be referenced.
+  if (aNetIdInfo->mRefCnt > 0) {
+    mInterfaceToNetIdHash.Put(aInterfaceName, *aNetIdInfo);
+    return true;
+  }
+
+  // No longer be referenced. Remove the entry.
+  mInterfaceToNetIdHash.Remove(aInterfaceName);
+
+  return true;
+}
+\ No newline at end of file