133 lines
4.3 KiB
C++
133 lines
4.3 KiB
C++
//
|
|
// Copyright (C) 2020 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
#include "update_engine/aosp/binder_service_stable_android.h"
|
|
|
|
#include <memory>
|
|
|
|
#include <base/bind.h>
|
|
#include <base/logging.h>
|
|
#include <binderwrapper/binder_wrapper.h>
|
|
#include <brillo/errors/error.h>
|
|
#include <utils/String8.h>
|
|
|
|
#include "update_engine/aosp/binder_service_android_common.h"
|
|
|
|
using android::binder::Status;
|
|
using android::os::IUpdateEngineStableCallback;
|
|
using android::os::ParcelFileDescriptor;
|
|
using std::string;
|
|
using std::vector;
|
|
using update_engine::UpdateEngineStatus;
|
|
|
|
namespace chromeos_update_engine {
|
|
|
|
BinderUpdateEngineAndroidStableService::BinderUpdateEngineAndroidStableService(
|
|
ServiceDelegateAndroidInterface* service_delegate)
|
|
: service_delegate_(service_delegate) {}
|
|
|
|
void BinderUpdateEngineAndroidStableService::SendStatusUpdate(
|
|
const UpdateEngineStatus& update_engine_status) {
|
|
last_status_ = static_cast<int>(update_engine_status.status);
|
|
last_progress_ = update_engine_status.progress;
|
|
if (callback_) {
|
|
callback_->onStatusUpdate(last_status_, last_progress_);
|
|
}
|
|
}
|
|
|
|
void BinderUpdateEngineAndroidStableService::SendPayloadApplicationComplete(
|
|
ErrorCode error_code) {
|
|
if (callback_) {
|
|
callback_->onPayloadApplicationComplete(static_cast<int>(error_code));
|
|
}
|
|
}
|
|
|
|
Status BinderUpdateEngineAndroidStableService::bind(
|
|
const android::sp<IUpdateEngineStableCallback>& callback,
|
|
bool* return_value) {
|
|
// Reject binding if another callback is already bound.
|
|
if (callback_ != nullptr) {
|
|
LOG(ERROR) << "Another callback is already bound. Can't bind new callback.";
|
|
*return_value = false;
|
|
return Status::ok();
|
|
}
|
|
|
|
// See BinderUpdateEngineAndroidService::bind.
|
|
if (last_status_ != -1) {
|
|
auto status = callback->onStatusUpdate(last_status_, last_progress_);
|
|
if (!status.isOk()) {
|
|
LOG(ERROR) << "Failed to call onStatusUpdate() from callback: "
|
|
<< status.toString8();
|
|
*return_value = false;
|
|
return Status::ok();
|
|
}
|
|
}
|
|
|
|
callback_ = callback;
|
|
|
|
const android::sp<IBinder>& callback_binder =
|
|
IUpdateEngineStableCallback::asBinder(callback);
|
|
auto binder_wrapper = android::BinderWrapper::Get();
|
|
binder_wrapper->RegisterForDeathNotifications(
|
|
callback_binder,
|
|
base::Bind(base::IgnoreResult(
|
|
&BinderUpdateEngineAndroidStableService::UnbindCallback),
|
|
base::Unretained(this),
|
|
base::Unretained(callback_binder.get())));
|
|
|
|
*return_value = true;
|
|
return Status::ok();
|
|
}
|
|
|
|
Status BinderUpdateEngineAndroidStableService::unbind(
|
|
const android::sp<IUpdateEngineStableCallback>& callback,
|
|
bool* return_value) {
|
|
const android::sp<IBinder>& callback_binder =
|
|
IUpdateEngineStableCallback::asBinder(callback);
|
|
auto binder_wrapper = android::BinderWrapper::Get();
|
|
binder_wrapper->UnregisterForDeathNotifications(callback_binder);
|
|
|
|
*return_value = UnbindCallback(callback_binder.get());
|
|
return Status::ok();
|
|
}
|
|
|
|
Status BinderUpdateEngineAndroidStableService::applyPayloadFd(
|
|
const ParcelFileDescriptor& pfd,
|
|
int64_t payload_offset,
|
|
int64_t payload_size,
|
|
const vector<android::String16>& header_kv_pairs) {
|
|
vector<string> str_headers = ToVecString(header_kv_pairs);
|
|
|
|
brillo::ErrorPtr error;
|
|
if (!service_delegate_->ApplyPayload(
|
|
pfd.get(), payload_offset, payload_size, str_headers, &error)) {
|
|
return ErrorPtrToStatus(error);
|
|
}
|
|
return Status::ok();
|
|
}
|
|
|
|
bool BinderUpdateEngineAndroidStableService::UnbindCallback(
|
|
const IBinder* callback) {
|
|
if (IUpdateEngineStableCallback::asBinder(callback_).get() != callback) {
|
|
LOG(ERROR) << "Unable to unbind unknown callback.";
|
|
return false;
|
|
}
|
|
callback_ = nullptr;
|
|
return true;
|
|
}
|
|
|
|
} // namespace chromeos_update_engine
|