1 //
2 // Copyright (C) 2015 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 //      http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16 
17 #include "tpm_manager/client/tpm_ownership_dbus_proxy.h"
18 
19 #include <brillo/bind_lambda.h>
20 #include <brillo/dbus/dbus_method_invoker.h>
21 
22 #include "tpm_manager/common/tpm_manager_constants.h"
23 #include "tpm_manager/common/tpm_ownership_dbus_interface.h"
24 
25 namespace {
26 
27 // Use a two minute timeout because TPM operations can take a long time.
28 const int kDBusTimeoutMS = 2 * 60 * 1000;
29 
30 }  // namespace
31 
32 namespace tpm_manager {
33 
~TpmOwnershipDBusProxy()34 TpmOwnershipDBusProxy::~TpmOwnershipDBusProxy() {
35   if (bus_) {
36     bus_->ShutdownAndBlock();
37   }
38 }
39 
Initialize()40 bool TpmOwnershipDBusProxy::Initialize() {
41   dbus::Bus::Options options;
42   options.bus_type = dbus::Bus::SYSTEM;
43   bus_ = new dbus::Bus(options);
44   object_proxy_ = bus_->GetObjectProxy(
45       tpm_manager::kTpmManagerServiceName,
46       dbus::ObjectPath(tpm_manager::kTpmManagerServicePath));
47   return (object_proxy_ != nullptr);
48 }
49 
GetTpmStatus(const GetTpmStatusRequest & request,const GetTpmStatusCallback & callback)50 void TpmOwnershipDBusProxy::GetTpmStatus(
51     const GetTpmStatusRequest& request,
52     const GetTpmStatusCallback& callback) {
53   CallMethod<GetTpmStatusReply>(tpm_manager::kGetTpmStatus, request, callback);
54 }
55 
TakeOwnership(const TakeOwnershipRequest & request,const TakeOwnershipCallback & callback)56 void TpmOwnershipDBusProxy::TakeOwnership(
57     const TakeOwnershipRequest& request,
58     const TakeOwnershipCallback& callback) {
59   CallMethod<TakeOwnershipReply>(
60       tpm_manager::kTakeOwnership, request, callback);
61 }
62 
RemoveOwnerDependency(const RemoveOwnerDependencyRequest & request,const RemoveOwnerDependencyCallback & callback)63 void TpmOwnershipDBusProxy::RemoveOwnerDependency(
64     const RemoveOwnerDependencyRequest& request,
65     const RemoveOwnerDependencyCallback& callback) {
66   CallMethod<RemoveOwnerDependencyReply>(
67       tpm_manager::kRemoveOwnerDependency, request, callback);
68 }
69 
70 template<typename ReplyProtobufType,
71          typename RequestProtobufType,
72          typename CallbackType>
CallMethod(const std::string & method_name,const RequestProtobufType & request,const CallbackType & callback)73 void TpmOwnershipDBusProxy::CallMethod(const std::string& method_name,
74                                        const RequestProtobufType& request,
75                                        const CallbackType& callback) {
76   auto on_error = [callback](brillo::Error* error) {
77     ReplyProtobufType reply;
78     reply.set_status(STATUS_NOT_AVAILABLE);
79     callback.Run(reply);
80   };
81   brillo::dbus_utils::CallMethodWithTimeout(
82       kDBusTimeoutMS,
83       object_proxy_,
84       tpm_manager::kTpmOwnershipInterface,
85       method_name,
86       callback,
87       base::Bind(on_error),
88       request);
89 }
90 
91 }  // namespace tpm_manager
92