44 lines
1.2 KiB
C++
44 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <glacier/container/hash_map.h>
|
|
#include <glacier/memory/shared_ptr.h>
|
|
#include <glacier/memory/unique_ptr.h>
|
|
#include <glacier/status/error_or.h>
|
|
#include <mammoth/sync/semaphore.h>
|
|
#include <victoriafalls/victoriafalls.yunq.client.h>
|
|
|
|
#include "hw/pcie.h"
|
|
#include "lib/yellowstone/yellowstone.yunq.server.h"
|
|
|
|
namespace yellowstone {
|
|
|
|
class YellowstoneServer : public YellowstoneServerBase {
|
|
public:
|
|
static glcr::ErrorOr<glcr::UniquePtr<YellowstoneServer>> Create();
|
|
|
|
glcr::Status HandleGetAhciInfo(AhciInfo&) override;
|
|
glcr::Status HandleGetFramebufferInfo(FramebufferInfo&) override;
|
|
glcr::Status HandleGetDenali(DenaliInfo&) override;
|
|
glcr::Status HandleRegisterEndpoint(const RegisterEndpointRequest&) override;
|
|
glcr::Status HandleGetEndpoint(const GetEndpointRequest&, Endpoint&) override;
|
|
|
|
void WaitDenaliRegistered();
|
|
void WaitVictoriaFallsRegistered();
|
|
|
|
private:
|
|
glcr::HashMap<glcr::String, z_cap_t> endpoint_map_;
|
|
|
|
uint64_t device_id_ = 0;
|
|
uint64_t lba_offset_ = 0;
|
|
glcr::SharedPtr<VFSClient> vfs_client_;
|
|
|
|
PciReader pci_reader_;
|
|
|
|
mmth::Semaphore has_denali_semaphore_;
|
|
mmth::Semaphore has_victoriafalls_semaphore_;
|
|
|
|
YellowstoneServer(z_cap_t endpoint_cap);
|
|
};
|
|
|
|
} // namespace yellowstone
|