mirror of
https://github.com/akuker/RASCSI.git
synced 2024-12-22 15:30:09 +00:00
621cc7d5a2
* Unit test updates * Lambda syntax cleanup * Use new-style casts * Use std::none_of when saving the cache * Use to_integer instead of casts * Use accessors for getting CDB data * Made ctrl_t private * Improved encapsulation * Replaced pointers by references * Removed all remaining occurrences of DWORD and BYTE, making os.h obsolete
81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
//---------------------------------------------------------------------------
|
|
//
|
|
// SCSI Target Emulator RaSCSI Reloaded
|
|
// for Raspberry Pi
|
|
//
|
|
// Copyright (C) 2022 Uwe Seimet
|
|
//
|
|
//---------------------------------------------------------------------------
|
|
|
|
#include "rascsi_interface.pb.h"
|
|
#include "protobuf_serializer.h"
|
|
#include "rascsi_exceptions.h"
|
|
#include <unistd.h>
|
|
|
|
using namespace std;
|
|
using namespace rascsi_interface;
|
|
|
|
//---------------------------------------------------------------------------
|
|
//
|
|
// Serialize/Deserialize protobuf message: Length followed by the actual data.
|
|
// A little endian platform is assumed.
|
|
//
|
|
//---------------------------------------------------------------------------
|
|
|
|
void ProtobufSerializer::SerializeMessage(int fd, const google::protobuf::Message& message) const
|
|
{
|
|
string data;
|
|
message.SerializeToString(&data);
|
|
|
|
// Write the size of the protobuf data as a header
|
|
auto size = static_cast<int32_t>(data.length());
|
|
if (write(fd, &size, sizeof(size)) != sizeof(size)) {
|
|
throw io_exception("Can't write protobuf message header");
|
|
}
|
|
|
|
// Write the actual protobuf data
|
|
if (write(fd, data.data(), size) != size) {
|
|
throw io_exception("Can't write protobuf message data");
|
|
}
|
|
}
|
|
|
|
void ProtobufSerializer::DeserializeMessage(int fd, google::protobuf::Message& message) const
|
|
{
|
|
// Read the header with the size of the protobuf data
|
|
vector<byte> header_buf(4);
|
|
if (ReadBytes(fd, header_buf) < header_buf.size()) {
|
|
throw io_exception("Invalid protobuf message header");
|
|
}
|
|
|
|
const int size = (static_cast<int>(header_buf[3]) << 24) + (static_cast<int>(header_buf[2]) << 16)
|
|
+ (static_cast<int>(header_buf[1]) << 8) + static_cast<int>(header_buf[0]);
|
|
if (size < 0) {
|
|
throw io_exception("Invalid protobuf message header");
|
|
}
|
|
|
|
// Read the binary protobuf data
|
|
vector<byte> data_buf(size);
|
|
if (ReadBytes(fd, data_buf) < data_buf.size()) {
|
|
throw io_exception("Missing protobuf message data");
|
|
}
|
|
|
|
// Create protobuf message
|
|
string data((const char *)data_buf.data(), size);
|
|
message.ParseFromString(data);
|
|
}
|
|
|
|
size_t ProtobufSerializer::ReadBytes(int fd, vector<byte>& buf) const
|
|
{
|
|
size_t offset = 0;
|
|
while (offset < buf.size()) {
|
|
const ssize_t len = read(fd, &buf.data()[offset], buf.size() - offset);
|
|
if (len <= 0) {
|
|
return len;
|
|
}
|
|
|
|
offset += len;
|
|
}
|
|
|
|
return offset;
|
|
}
|