blob: f067c157d06c1569815bc9113f55c1323429bb63 [file] [log] [blame]
#include "ipmi.hpp"
#include "manager_mock.hpp"
#include <cstring>
#include <gtest/gtest.h>
namespace blobs
{
using ::testing::_;
using ::testing::Invoke;
using ::testing::Matcher;
using ::testing::NotNull;
using ::testing::Return;
// ipmid.hpp isn't installed where we can grab it and this value is per BMC
// SoC.
#define MAX_IPMI_BUFFER 64
TEST(BlobSessionStatTest, RequestRejectedByManagerReturnsFailure)
{
// If the session ID is invalid, the request must fail.
ManagerMock mgr;
size_t dataLen;
uint8_t request[MAX_IPMI_BUFFER] = {0};
uint8_t reply[MAX_IPMI_BUFFER] = {0};
auto req = reinterpret_cast<struct BmcBlobSessionStatTx*>(request);
req->cmd = static_cast<std::uint8_t>(BlobOEMCommands::bmcBlobSessionStat);
req->crc = 0;
req->sessionId = 0x54;
dataLen = sizeof(struct BmcBlobSessionStatTx);
EXPECT_CALL(mgr,
stat(Matcher<uint16_t>(req->sessionId), Matcher<BlobMeta*>(_)))
.WillOnce(Return(false));
EXPECT_EQ(IPMI_CC_UNSPECIFIED_ERROR,
sessionStatBlob(&mgr, request, reply, &dataLen));
}
TEST(BlobSessionStatTest, RequestSucceedsNoMetadata)
{
// Stat request succeeeds but there were no metadata bytes.
ManagerMock mgr;
size_t dataLen;
uint8_t request[MAX_IPMI_BUFFER] = {0};
uint8_t reply[MAX_IPMI_BUFFER] = {0};
auto req = reinterpret_cast<struct BmcBlobSessionStatTx*>(request);
req->cmd = static_cast<std::uint8_t>(BlobOEMCommands::bmcBlobSessionStat);
req->crc = 0;
req->sessionId = 0x54;
dataLen = sizeof(struct BmcBlobSessionStatTx);
struct BmcBlobStatRx rep;
rep.crc = 0x00;
rep.blobState = 0x01;
rep.size = 0x100;
rep.metadataLen = 0x00;
uint16_t blobState = rep.blobState;
uint32_t size = rep.size;
EXPECT_CALL(mgr, stat(Matcher<uint16_t>(req->sessionId),
Matcher<BlobMeta*>(NotNull())))
.WillOnce(Invoke([&](uint16_t session, BlobMeta* meta) {
meta->blobState = blobState;
meta->size = size;
return true;
}));
EXPECT_EQ(IPMI_CC_OK, sessionStatBlob(&mgr, request, reply, &dataLen));
EXPECT_EQ(sizeof(rep), dataLen);
EXPECT_EQ(0, std::memcmp(reply, &rep, sizeof(rep)));
}
TEST(BlobSessionStatTest, RequestSucceedsWithMetadata)
{
// Stat request succeeds and there were metadata bytes.
ManagerMock mgr;
size_t dataLen;
uint8_t request[MAX_IPMI_BUFFER] = {0};
uint8_t reply[MAX_IPMI_BUFFER] = {0};
auto req = reinterpret_cast<struct BmcBlobSessionStatTx*>(request);
req->cmd = static_cast<std::uint8_t>(BlobOEMCommands::bmcBlobSessionStat);
req->crc = 0;
req->sessionId = 0x54;
dataLen = sizeof(struct BmcBlobSessionStatTx);
BlobMeta lmeta;
lmeta.blobState = 0x01;
lmeta.size = 0x100;
lmeta.metadata.push_back(0x01);
lmeta.metadata.push_back(0x02);
lmeta.metadata.push_back(0x03);
lmeta.metadata.push_back(0x04);
struct BmcBlobStatRx rep;
rep.crc = 0x00;
rep.blobState = lmeta.blobState;
rep.size = lmeta.size;
rep.metadataLen = lmeta.metadata.size();
EXPECT_CALL(mgr, stat(Matcher<uint16_t>(req->sessionId),
Matcher<BlobMeta*>(NotNull())))
.WillOnce(Invoke([&](uint16_t session, BlobMeta* meta) {
(*meta) = lmeta;
return true;
}));
EXPECT_EQ(IPMI_CC_OK, sessionStatBlob(&mgr, request, reply, &dataLen));
EXPECT_EQ(sizeof(rep) + lmeta.metadata.size(), dataLen);
EXPECT_EQ(0, std::memcmp(reply, &rep, sizeof(rep)));
EXPECT_EQ(0, std::memcmp(reply + sizeof(rep), lmeta.metadata.data(),
lmeta.metadata.size()));
}
} // namespace blobs