Add new shill client for VPN

chromeos/dbus/shill_third_party_vpn_driver_client.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chromeos/dbus/shill_third_party_vpn_driver_client.h"
+
+#include "base/bind.h"
+#include "base/macros.h"
+#include "chromeos/dbus/shill_third_party_vpn_observer.h"
+#include "dbus/bus.h"
+#include "dbus/message.h"
+#include "dbus/object_proxy.h"
+#include "third_party/cros_system_api/dbus/service_constants.h"
+
+namespace shill {
+
+// TODO(kaliamoorthi): Move these constants to dbus/service_constants.cc

[stevenjb, 2014/10/31 16:35:41]

The individual constants will be in service_constants.h, but there is no
service_constants.cc, this should be in an anonymous namespace in this file. (We
will need to maintain this array in Chrome, i.e. if an entry gets added to
service_constants.h, we will need to modify this array. Not ideal, but in
practice we will need to update any extension api / documentation anyway, so not
really a big deal).

[kaliamoorthi, 2014/11/03 10:55:38]

Done.

+const char* kSetParametersKeyList[] = {kAddressParameterThirdPartyVpn,
+                                       kBroadcastAddressParameterThirdPartyVpn,
+                                       kGatewayParameterThirdPartyVpn,
+                                       kBypassTunnelForIpParameterThirdPartyVpn,
+                                       kSubnetPrefixParameterThirdPartyVpn,
+                                       kMtuParameterThirdPartyVpn,
+                                       kDomainSearchParameterThirdPartyVpn,
+                                       kDnsServersParameterThirdPartyVpn};
+
+}  // namespace shill
+
+namespace chromeos {
+
+namespace {
+
+// The ShillThirdPartyVpnDriverClient implementation.
+class ShillThirdPartyVpnDriverClientImpl
+    : public ShillThirdPartyVpnDriverClient {
+ public:
+  ShillThirdPartyVpnDriverClientImpl();
+  ~ShillThirdPartyVpnDriverClientImpl() override;
+
+  // ShillThirdPartyVpnDriverClient overrides
+  void AddShillThirdPartyVpnObserver(
+      const dbus::ObjectPath& object_path,
+      ShillThirdPartyVpnObserver* observer) override;
+
+  void RemoveShillThirdPartyVpnObserver(
+      const dbus::ObjectPath& object_path) override;
+
+  void SetParameters(const dbus::ObjectPath& object_path,
+                     const base::DictionaryValue& parameters,
+                     const VoidDBusMethodCallback& callback) override;
+
+  void UpdateConnectionState(const dbus::ObjectPath& object_path,
+                             const uint32 connection_state,
+                             const VoidDBusMethodCallback& callback) override;
+
+  void SendPacket(const dbus::ObjectPath& object_path,
+                  const std::vector<uint8>& ip_packet,
+                  const VoidDBusMethodCallback& callback) override;
+
+ protected:
+  void Init(dbus::Bus* bus) override { bus_ = bus; }
+
+  ShillThirdPartyVpnDriverClient::TestInterface* GetTestInterface() override {
+    return NULL;
+  }
+
+ private:
+  class HelperInfo {
+   public:
+    explicit HelperInfo(dbus::ObjectProxy* object_proxy);
+    ShillClientHelper* helper() { return &helper_; }
+    ShillThirdPartyVpnObserver* observer() { return observer_; }
+
+    void set_observer(ShillThirdPartyVpnObserver* observer) {
+      observer_ = observer;
+    }
+
+    base::WeakPtr<HelperInfo> GetWeakPtr() {
+      return weak_ptr_factory_.GetWeakPtr();
+    }
+
+   private:
+    ShillClientHelper helper_;
+    ShillThirdPartyVpnObserver* observer_;
+
+    base::WeakPtrFactory<HelperInfo> weak_ptr_factory_;
+  };
+  typedef std::map<std::string, HelperInfo*> HelperMap;
+
+  void ReleaseHelper(const dbus::ObjectPath& object_path);
+
+  static void OnPacketReceived(base::WeakPtr<HelperInfo> helper_info,
+                               dbus::Signal* signal);
+  static void OnPlatformMessage(base::WeakPtr<HelperInfo> helper_info,
+                                dbus::Signal* signal);
+  static void OnSignalConnected(const std::string& interface,
+                                const std::string& signal,
+                                bool success);
+
+  // Returns the corresponding ShillClientHelper for the object_path.
+  ShillClientHelper* GetHelper(const dbus::ObjectPath& object_path) {
+    return GetHelperInfo(object_path)->helper();
+  }
+
+  HelperInfo* FindHelperInfo(const dbus::ObjectPath& object_path) {
+    HelperMap::iterator it = helpers_.find(object_path.value());
+    return (it != helpers_.end()) ? it->second : NULL;
+  }
+
+  // Returns the corresponding HelperInfo for the object_path.
+  HelperInfo* GetHelperInfo(const dbus::ObjectPath& object_path) {
+    HelperInfo* helper_info = FindHelperInfo(object_path);
+    if (helper_info)
+      return helper_info;
+
+    // There is no helper for the profile, create it.
+    dbus::ObjectProxy* object_proxy =
+        bus_->GetObjectProxy(shill::kFlimflamServiceName, object_path);
+    helper_info = new HelperInfo(object_proxy);
+    helpers_.insert(HelperMap::value_type(object_path.value(), helper_info));
+    return helper_info;
+  }
+
+  dbus::Bus* bus_;
+  HelperMap helpers_;
+  std::set<std::string> valid_keys_;
+
+  DISALLOW_COPY_AND_ASSIGN(ShillThirdPartyVpnDriverClientImpl);
+};
+
+ShillThirdPartyVpnDriverClientImpl::HelperInfo::HelperInfo(
+    dbus::ObjectProxy* object_proxy)
+    : helper_(object_proxy), observer_(NULL), weak_ptr_factory_(this) {
+}
+
+ShillThirdPartyVpnDriverClientImpl::ShillThirdPartyVpnDriverClientImpl()
+    : bus_(NULL) {
+  for (uint32 i = 0; i < arraysize(shill::kSetParametersKeyList); ++i) {
+    valid_keys_.insert(shill::kSetParametersKeyList[i]);
+  }
+}
+
+ShillThirdPartyVpnDriverClientImpl::~ShillThirdPartyVpnDriverClientImpl() {
+  for (HelperMap::iterator iter = helpers_.begin(); iter != helpers_.end();
+       ++iter) {

[stevenjb, 2014/10/31 16:35:41]

nit: Use c++11 style iterators here and below:
for (auto& iter : helpers_)

[kaliamoorthi, 2014/11/03 10:55:38]

Done.

+    HelperInfo* helper_info = iter->second;
+    bus_->RemoveObjectProxy(
+        shill::kFlimflamServiceName,
+        helper_info->helper()->object_proxy()->object_path(),
+        base::Bind(&base::DoNothing));
+    delete helper_info;
+  }
+}
+
+void ShillThirdPartyVpnDriverClientImpl::AddShillThirdPartyVpnObserver(
+    const dbus::ObjectPath& object_path,
+    ShillThirdPartyVpnObserver* observer) {
+  HelperInfo* helper_info = GetHelperInfo(object_path);
+  if (helper_info->observer()) {
+    LOG(ERROR) << "Observer exists for " << object_path.value();
+    return;
+  }
+
+  helper_info->set_observer(observer);
+  dbus::ObjectProxy* proxy =
+      const_cast<dbus::ObjectProxy*>(helper_info->helper()->object_proxy());
+
+  proxy->ConnectToSignal(
+      shill::kFlimflamThirdPartyVpnInterface, shill::kOnPlatformMessageFunction,
+      base::Bind(&ShillThirdPartyVpnDriverClientImpl::OnPlatformMessage,
+                 helper_info->GetWeakPtr()),
+      base::Bind(&ShillThirdPartyVpnDriverClientImpl::OnSignalConnected));
+
+  proxy->ConnectToSignal(
+      shill::kFlimflamThirdPartyVpnInterface, shill::kOnPacketReceivedFunction,
+      base::Bind(&ShillThirdPartyVpnDriverClientImpl::OnPacketReceived,
+                 helper_info->GetWeakPtr()),
+      base::Bind(&ShillThirdPartyVpnDriverClientImpl::OnSignalConnected));
+}
+
+void ShillThirdPartyVpnDriverClientImpl::RemoveShillThirdPartyVpnObserver(
+    const dbus::ObjectPath& object_path) {
+  HelperInfo* helper_info = FindHelperInfo(object_path);
+  if (!helper_info) {
+    LOG(ERROR) << "Unknown object_path " << object_path.value();
+    return;
+  }
+
+  CHECK(helper_info->observer());
+  helper_info->set_observer(NULL);
+  ReleaseHelper(object_path);
+}
+
+void ShillThirdPartyVpnDriverClientImpl::ReleaseHelper(
+    const dbus::ObjectPath& object_path) {
+  HelperInfo* helper_info = FindHelperInfo(object_path);
+  if (!helper_info) {
+    LOG(ERROR) << "Unknown object_path " << object_path.value();
+    return;
+  }
+
+  bus_->RemoveObjectProxy(shill::kFlimflamServiceName, object_path,
+                          base::Bind(&base::DoNothing));
+  helpers_.erase(helpers_.find(object_path.value()));
+  delete helper_info;
+}
+
+void ShillThirdPartyVpnDriverClientImpl::SetParameters(
+    const dbus::ObjectPath& object_path,
+    const base::DictionaryValue& parameters,
+    const VoidDBusMethodCallback& callback) {
+  dbus::MethodCall method_call(shill::kFlimflamThirdPartyVpnInterface,
+                               shill::kSetParametersFunction);
+  dbus::MessageWriter writer(&method_call);
+  dbus::MessageWriter array_writer(NULL);
+  writer.OpenArray("{ss}", &array_writer);
+  for (base::DictionaryValue::Iterator it(parameters); !it.IsAtEnd();
+       it.Advance()) {
+    std::string value;
+    if (valid_keys_.find(it.key()) == valid_keys_.end() ||
+        !it.value().GetAsString(&value)) {
+      LOG(WARNING) << "Unknown key " << it.key() << " or non string value "
+                   << it.value() << ".";

[stevenjb, 2014/10/31 16:35:41]

nit: Maybe have two ifs with separate warnings.

[kaliamoorthi, 2014/11/03 10:55:38]

Done.

+      continue;
+    }
+    dbus::MessageWriter entry_writer(NULL);
+    array_writer.OpenDictEntry(&entry_writer);
+    entry_writer.AppendString(it.key());
+    entry_writer.AppendString(value);
+    array_writer.CloseContainer(&entry_writer);
+  }
+  writer.CloseContainer(&array_writer);
+  GetHelper(object_path)->CallVoidMethod(&method_call, callback);
+}
+
+void ShillThirdPartyVpnDriverClientImpl::UpdateConnectionState(
+    const dbus::ObjectPath& object_path,
+    const uint32 connection_state,
+    const VoidDBusMethodCallback& callback) {
+  dbus::MethodCall method_call(shill::kFlimflamThirdPartyVpnInterface,
+                               shill::kUpdateConnectionStateFunction);
+  dbus::MessageWriter writer(&method_call);
+  writer.AppendUint32(connection_state);
+  GetHelper(object_path)->CallVoidMethod(&method_call, callback);
+}
+
+void ShillThirdPartyVpnDriverClientImpl::SendPacket(
+    const dbus::ObjectPath& object_path,
+    const std::vector<uint8>& ip_packet,
+    const VoidDBusMethodCallback& callback) {
+  dbus::MethodCall method_call(shill::kFlimflamThirdPartyVpnInterface,
+                               shill::kSendPacketFunction);
+  dbus::MessageWriter writer(&method_call);
+  writer.AppendArrayOfBytes(ip_packet.data(), ip_packet.size());
+  GetHelper(object_path)->CallVoidMethod(&method_call, callback);
+}
+
+// static
+void ShillThirdPartyVpnDriverClientImpl::OnPacketReceived(
+    base::WeakPtr<HelperInfo> helper_info,
+    dbus::Signal* signal) {
+  if (!helper_info || !helper_info->observer())
+    return;
+
+  dbus::MessageReader reader(signal);
+  const uint8* data;
+  size_t length;
+  if (reader.PopArrayOfBytes(&data, &length))
+    helper_info->observer()->OnPacketReceived(data, length);
+}
+
+// static
+void ShillThirdPartyVpnDriverClientImpl::OnPlatformMessage(
+    base::WeakPtr<HelperInfo> helper_info,
+    dbus::Signal* signal) {
+  if (!helper_info || !helper_info->observer())
+    return;
+
+  dbus::MessageReader reader(signal);
+  uint32 platform_message;
+  if (reader.PopUint32(&platform_message))
+    helper_info->observer()->OnPlatformMessage(platform_message);
+}
+
+// static
+void ShillThirdPartyVpnDriverClientImpl::OnSignalConnected(
+    const std::string& interface,
+    const std::string& signal,
+    bool success) {
+  LOG_IF(ERROR, !success) << "Connect to " << interface << " " << signal
+                          << " failed.";
+}
+
+}  // namespace
+
+ShillThirdPartyVpnDriverClient::ShillThirdPartyVpnDriverClient() {
+}
+
+ShillThirdPartyVpnDriverClient::~ShillThirdPartyVpnDriverClient() {
+}
+
+// static
+ShillThirdPartyVpnDriverClient* ShillThirdPartyVpnDriverClient::Create() {
+  return new ShillThirdPartyVpnDriverClientImpl();
+}
+
+}  // namespace chromeos