sandboxed-api/sandboxed_api/var_lenval.h

85 lines
2.8 KiB
C++

// Copyright 2019 Google LLC
//
// 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
//
// https://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.
#ifndef SANDBOXED_API_VAR_LENVAL_H_
#define SANDBOXED_API_VAR_LENVAL_H_
#include <sys/types.h>
#include <sys/uio.h>
#include <cstring>
#include <memory>
#include <string>
#include <vector>
#include "absl/base/macros.h"
#include "absl/status/status.h"
#include "sandboxed_api/lenval_core.h"
#include "sandboxed_api/var_abstract.h"
#include "sandboxed_api/var_array.h"
#include "sandboxed_api/var_ptr.h"
#include "sandboxed_api/var_struct.h"
namespace sapi::v {
template <class T>
class Proto;
// Length + value container. Represents a pointer to a LenValStruct inside the
// sandboxee which allows the bidirectional synchronization data structures with
// changing lengths (e.g. protobuf structures). You probably want to directly
// use protobufs as they are easier to handle.
class LenVal : public Var {
public:
explicit LenVal(const char* data, uint64_t size)
: array_(const_cast<uint8_t*>(reinterpret_cast<const uint8_t*>(data)),
size),
struct_(size, nullptr) {}
explicit LenVal(const std::vector<uint8_t>& data)
: array_(data.size()), struct_(data.size(), nullptr) {
memcpy(array_.GetData(), data.data(), data.size());
}
explicit LenVal(size_t size) : array_(size), struct_(size, nullptr) {}
Type GetType() const final { return Type::kLenVal; }
std::string GetTypeString() const final { return "LengthValue"; }
std::string ToString() const final { return "LenVal"; }
absl::Status ResizeData(RPCChannel* rpc_channel, size_t size);
size_t GetDataSize() const { return struct_.data().size; }
uint8_t* GetData() const { return array_.GetData(); }
void* GetRemote() const final { return struct_.GetRemote(); }
protected:
size_t GetSize() const final { return 0; }
absl::Status Allocate(RPCChannel* rpc_channel, bool automatic_free) override;
absl::Status Free(RPCChannel* rpc_channel) override;
absl::Status TransferToSandboxee(RPCChannel* rpc_channel, pid_t pid) override;
absl::Status TransferFromSandboxee(RPCChannel* rpc_channel,
pid_t pid) override;
Array<uint8_t> array_;
Struct<LenValStruct> struct_;
template <class T>
friend class Proto;
};
} // namespace sapi::v
#endif // SANDBOXED_API_VAR_LENVAL_H_