mirror of
https://github.com/google/sandboxed-api.git
synced 2024-03-22 13:11:30 +08:00
76 lines
2.3 KiB
C++
76 lines
2.3 KiB
C++
|
// Copyright 2019 Google LLC. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// 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 "sandboxed_api/transaction.h"
|
||
|
|
#include "sandboxed_api/util/canonical_errors.h"
|
||
|
|
#include "sandboxed_api/util/status_macros.h"
|
||
|
|
|
||
|
|
namespace sapi {
|
||
|
|
|
||
|
|
constexpr absl::Duration TransactionBase::kDefaultTimeLimit;
|
||
|
|
|
||
|
|
sapi::Status TransactionBase::RunTransactionFunctionInSandbox(
|
||
|
|
const std::function<sapi::Status()>& f) {
|
||
|
|
// Run Main(), invoking Init() if this hasn't been yet done.
|
||
|
|
SAPI_RETURN_IF_ERROR(GetSandbox()->Init());
|
||
|
|
|
||
|
|
// Set the wall-time limit for this transaction run, and clean it up
|
||
|
|
// afterwards, no matter what the result.
|
||
|
|
SAPI_RETURN_IF_ERROR(GetSandbox()->SetWallTimeLimit(GetTimeLimit()));
|
||
|
|
struct TimeCleanup {
|
||
|
|
~TimeCleanup() {
|
||
|
|
if (capture->GetSandbox()->IsActive()) {
|
||
|
|
capture->GetSandbox()->SetWallTimeLimit(0).IgnoreError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TransactionBase* capture;
|
||
|
|
} sandbox_cleanup{this};
|
||
|
|
|
||
|
|
if (!GetInited()) {
|
||
|
|
SAPI_RETURN_IF_ERROR(Init());
|
||
|
|
SetInited(true);
|
||
|
|
}
|
||
|
|
|
||
|
|
return f();
|
||
|
|
}
|
||
|
|
|
||
|
|
sapi::Status TransactionBase::RunTransactionLoop(
|
||
|
|
const std::function<sapi::Status()>& f) {
|
||
|
|
// Try to run Main() for a few times, return error if none of the tries
|
||
|
|
// succeeded.
|
||
|
|
sapi::Status status;
|
||
|
|
for (int i = 0; i <= GetRetryCnt(); i++) {
|
||
|
|
status = RunTransactionFunctionInSandbox(f);
|
||
|
|
if (status.ok()) {
|
||
|
|
return status;
|
||
|
|
}
|
||
|
|
GetSandbox()->Terminate();
|
||
|
|
SetInited(false);
|
||
|
|
}
|
||
|
|
|
||
|
|
LOG(ERROR) << "Tried " << (GetRetryCnt() + 1) << " times to run the "
|
||
|
|
<< "transaction, but it failed. SAPI error: '" << status
|
||
|
|
<< "'. Latest sandbox error: '"
|
||
|
|
<< GetSandbox()->AwaitResult().ToString() << "'";
|
||
|
|
return status;
|
||
|
|
}
|
||
|
|
|
||
|
|
TransactionBase::~TransactionBase() {
|
||
|
|
if (GetInited()) {
|
||
|
|
Finish().IgnoreError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace sapi
|