Bug 919426 - Part 2: Add new file WifiNetUtil.jsm. r=vchang

WifiNetUtil.jsm provides network related facilities.

dom/wifi/WifiNetUtil.jsm

@@ -0,0 +1,211 @@
+/* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set shiftwidth=2 tabstop=2 autoindent cindent expandtab: */
+/* 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/. */
+
+"use strict";
+
+const {classes: Cc, interfaces: Ci, utils: Cu, results: Cr} = Components;
+
+Cu.import("resource://gre/modules/XPCOMUtils.jsm");
+Cu.import("resource://gre/modules/systemlibs.js");
+
+XPCOMUtils.defineLazyServiceGetter(this, "gNetworkManager",
+                                   "@mozilla.org/network/manager;1",
+                                   "nsINetworkManager");
+
+this.EXPORTED_SYMBOLS = ["WifiNetUtil"];
+
+const DHCP_PROP = "init.svc.dhcpcd";
+const DHCP      = "dhcpcd";
+
+this.WifiNetUtil = function(controlMessage) {
+  var util = {};
+
+  util.configureInterface = function(cfg, callback) {
+    let message = { cmd:     "ifc_configure",
+                    ifname:  cfg.ifname,
+                    ipaddr:  cfg.ipaddr,
+                    mask:    cfg.mask,
+                    gateway: cfg.gateway,
+                    dns1:    cfg.dns1,
+                    dns2:    cfg.dns2 };
+
+    controlMessage(message, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.runDhcp = function (ifname, callback) {
+    controlMessage({ cmd: "dhcp_do_request", ifname: ifname }, function(data) {
+      var dhcpInfo = data.status ? null : data;
+      util.runIpConfig(ifname, dhcpInfo, callback);
+    });
+  };
+
+  util.stopDhcp = function (ifname, callback) {
+    // This function does exactly what dhcp_stop does. Unforunately, if we call
+    // this function twice before the previous callback is returned. We may block
+    // our self waiting for the callback. It slows down the wifi startup procedure.
+    // Therefore, we have to roll our own version here.
+    let dhcpService = DHCP_PROP + "_" + ifname;
+    let suffix = (ifname.substr(0, 3) === "p2p") ? "p2p" : ifname;
+    let processName = DHCP + "_" + suffix;
+    stopProcess(dhcpService, processName, callback);
+  };
+
+  util.enableInterface = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_enable", ifname: ifname }, function (data) {
+      callback(!data.status);
+    });
+  };
+
+  util.disableInterface = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_disable", ifname: ifname }, function (data) {
+      callback(!data.status);
+    });
+  };
+
+  util.startDhcpServer = function (range, callback) {
+    gNetworkManager.setDhcpServer(true, range, function (error) {
+      callback(!error);
+    });
+  };
+
+  util.stopDhcpServer = function (callback) {
+    gNetworkManager.setDhcpServer(false, null, function (error) {
+      callback(!error);
+    });
+  };
+
+  util.addHostRoute = function (ifname, route, callback) {
+    controlMessage({ cmd: "ifc_add_host_route", ifname: ifname, route: route }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.removeHostRoutes = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_remove_host_routes", ifname: ifname }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.setDefaultRoute = function (ifname, route, callback) {
+    controlMessage({ cmd: "ifc_set_default_route", ifname: ifname, route: route }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.getDefaultRoute = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_get_default_route", ifname: ifname }, function(data) {
+      callback(!data.route);
+    });
+  };
+
+  util.removeDefaultRoute = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_remove_default_route", ifname: ifname }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.resetConnections = function (ifname, callback) {
+    controlMessage({ cmd: "ifc_reset_connections", ifname: ifname }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.releaseDhcpLease = function (ifname, callback) {
+    controlMessage({ cmd: "dhcp_release_lease", ifname: ifname }, function(data) {
+      callback(!data.status);
+    });
+  };
+
+  util.getDhcpError = function (callback) {
+    controlMessage({ cmd: "dhcp_get_errmsg" }, function(data) {
+      callback(data.error);
+    });
+  };
+
+  util.runDhcpRenew = function (ifname, callback) {
+    controlMessage({ cmd: "dhcp_do_request", ifname: ifname }, function(data) {
+      callback(data.status ? null : data);
+    });
+  };
+
+  util.runIpConfig = function (name, data, callback) {
+    if (!data) {
+      callback({ info: data });
+      return;
+    }
+
+    setProperty("net." + name + ".dns1", ipToString(data.dns1),
+                function(ok) {
+      if (!ok) {
+        return;
+      }
+      setProperty("net." + name + ".dns2", ipToString(data.dns2),
+                  function(ok) {
+        if (!ok) {
+          return;
+        }
+        setProperty("net." + name + ".gw", ipToString(data.gateway),
+                    function(ok) {
+          if (!ok) {
+            return;
+          }
+          callback({ info: data });
+        });
+      });
+    });
+  };
+
+  //--------------------------------------------------
+  // Helper functions.
+  //--------------------------------------------------
+
+  function stopProcess(service, process, callback) {
+    var count = 0;
+    var timer = Cc["@mozilla.org/timer;1"].createInstance(Ci.nsITimer);
+    function tick() {
+      let result = libcutils.property_get(service);
+      if (result === null) {
+        callback();
+        return;
+      }
+      if (result === "stopped" || ++count >= 5) {
+        // Either we succeeded or ran out of time.
+        timer = null;
+        callback();
+        return;
+      }
+
+      // Else it's still running, continue waiting.
+      timer.initWithCallback(tick, 1000, Ci.nsITimer.TYPE_ONE_SHOT);
+    }
+
+    setProperty("ctl.stop", process, tick);
+  }
+
+  // Wrapper around libcutils.property_set that returns true if setting the
+  // value was successful.
+  // Note that the callback is not called asynchronously.
+  function setProperty(key, value, callback) {
+    let ok = true;
+    try {
+      libcutils.property_set(key, value);
+    } catch(e) {
+      ok = false;
+    }
+    callback(ok);
+  }
+
+  function ipToString(n) {
+    return String((n >>  0) & 0xFF) + "." +
+                 ((n >>  8) & 0xFF) + "." +
+                 ((n >> 16) & 0xFF) + "." +
+                 ((n >> 24) & 0xFF);
+  }
+
+  return util;
+};