blob: fd7388026353c4ffcb9b4770a6a8e41d570372e4 [file] [log] [blame]
// Copyright (C) 2018 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.
#pragma once
#include "aemu/base/files/Stream.h"
#include "aemu/base/files/StreamSerializing.h"
#include "goldfish_vk_private_defs.h"
#include "VulkanHandleMapping.h"
#include "IOStream.h"
#include "ResourceTracker.h"
#include "aemu/base/BumpPool.h"
#include "aemu/base/Tracing.h"
#include <vector>
#include <memory>
#include <log/log.h>
#include <inttypes.h>
class IOStream;
namespace goldfish_vk {
class VulkanStreamGuest : public android::base::Stream {
public:
VulkanStreamGuest(IOStream* stream);
~VulkanStreamGuest();
// Returns whether the connection is valid.
bool valid();
// General allocation function
void alloc(void** ptrAddr, size_t bytes);
// Utility functions to load strings or
// string arrays in place with allocation.
void loadStringInPlace(char** forOutput);
void loadStringArrayInPlace(char*** forOutput);
// When we load a string and are using a reserved pointer.
void loadStringInPlaceWithStreamPtr(char** forOutput, uint8_t** streamPtr);
void loadStringArrayInPlaceWithStreamPtr(char*** forOutput, uint8_t** streamPtr);
ssize_t read(void *buffer, size_t size) override;
ssize_t write(const void *buffer, size_t size) override;
void writeLarge(const void* buffer, size_t size);
// Frees everything that got alloc'ed.
void clearPool();
void setHandleMapping(VulkanHandleMapping* mapping);
void unsetHandleMapping();
VulkanHandleMapping* handleMapping() const;
void flush();
uint32_t getFeatureBits() const;
void incStreamRef();
bool decStreamRef();
uint8_t* reserve(size_t size);
private:
android::base::BumpPool mPool;
std::vector<uint8_t> mWriteBuffer;
IOStream* mStream = nullptr;
DefaultHandleMapping mDefaultHandleMapping;
VulkanHandleMapping* mCurrentHandleMapping;
uint32_t mFeatureBits = 0;
};
class VulkanCountingStream : public VulkanStreamGuest {
public:
VulkanCountingStream();
~VulkanCountingStream();
ssize_t read(void *buffer, size_t size) override;
ssize_t write(const void *buffer, size_t size) override;
size_t bytesWritten() const { return m_written; }
size_t bytesRead() const { return m_read; }
void rewind();
private:
size_t m_written = 0;
size_t m_read = 0;
};
} // namespace goldfish_vk