1 #include "helper.hpp"
2 #include "ipmi.hpp"
3 #include "manager_mock.hpp"
4 
5 #include <cstring>
6 
7 #include <gtest/gtest.h>
8 
9 namespace blobs
10 {
11 
12 using ::testing::_;
13 using ::testing::Invoke;
14 using ::testing::Matcher;
15 using ::testing::NotNull;
16 using ::testing::Return;
17 
18 // ipmid.hpp isn't installed where we can grab it and this value is per BMC
19 // SoC.
20 #define MAX_IPMI_BUFFER 64
21 
22 TEST(BlobSessionStatTest, RequestRejectedByManagerReturnsFailure)
23 {
24     // If the session ID is invalid, the request must fail.
25 
26     ManagerMock mgr;
27     std::vector<uint8_t> request;
28     struct BmcBlobSessionStatTx req;
29     req.crc = 0;
30     req.sessionId = 0x54;
31 
32     request.resize(sizeof(struct BmcBlobSessionStatTx));
33     std::memcpy(request.data(), &req, sizeof(struct BmcBlobSessionStatTx));
34 
35     EXPECT_CALL(mgr,
36                 stat(Matcher<uint16_t>(req.sessionId), Matcher<BlobMeta*>(_)))
37         .WillOnce(Return(false));
38 
39     EXPECT_EQ(ipmi::responseUnspecifiedError(), sessionStatBlob(&mgr, request));
40 }
41 
42 TEST(BlobSessionStatTest, RequestSucceedsNoMetadata)
43 {
44     // Stat request succeeeds but there were no metadata bytes.
45 
46     ManagerMock mgr;
47     std::vector<uint8_t> request;
48     struct BmcBlobSessionStatTx req;
49     req.crc = 0;
50     req.sessionId = 0x54;
51 
52     request.resize(sizeof(struct BmcBlobSessionStatTx));
53     std::memcpy(request.data(), &req, sizeof(struct BmcBlobSessionStatTx));
54 
55     struct BmcBlobStatRx rep;
56     rep.crc = 0x00;
57     rep.blobState = 0x01;
58     rep.size = 0x100;
59     rep.metadataLen = 0x00;
60 
61     uint16_t blobState = rep.blobState;
62     uint32_t size = rep.size;
63 
64     EXPECT_CALL(mgr, stat(Matcher<uint16_t>(req.sessionId),
65                           Matcher<BlobMeta*>(NotNull())))
66         .WillOnce(Invoke([&](uint16_t, BlobMeta* meta) {
67             meta->blobState = blobState;
68             meta->size = size;
69             return true;
70         }));
71 
72     auto result = validateReply(sessionStatBlob(&mgr, request));
73 
74     EXPECT_EQ(sizeof(rep), result.size());
75     EXPECT_EQ(0, std::memcmp(result.data(), &rep, sizeof(rep)));
76 }
77 
78 TEST(BlobSessionStatTest, RequestSucceedsWithMetadata)
79 {
80     // Stat request succeeds and there were metadata bytes.
81 
82     ManagerMock mgr;
83     std::vector<uint8_t> request;
84     struct BmcBlobSessionStatTx req;
85     req.crc = 0;
86     req.sessionId = 0x54;
87 
88     request.resize(sizeof(struct BmcBlobSessionStatTx));
89     std::memcpy(request.data(), &req, sizeof(struct BmcBlobSessionStatTx));
90 
91     BlobMeta lmeta;
92     lmeta.blobState = 0x01;
93     lmeta.size = 0x100;
94     lmeta.metadata.push_back(0x01);
95     lmeta.metadata.push_back(0x02);
96     lmeta.metadata.push_back(0x03);
97     lmeta.metadata.push_back(0x04);
98 
99     struct BmcBlobStatRx rep;
100     rep.crc = 0x00;
101     rep.blobState = lmeta.blobState;
102     rep.size = lmeta.size;
103     rep.metadataLen = lmeta.metadata.size();
104 
105     EXPECT_CALL(mgr, stat(Matcher<uint16_t>(req.sessionId),
106                           Matcher<BlobMeta*>(NotNull())))
107         .WillOnce(Invoke([&](uint16_t, BlobMeta* meta) {
108             (*meta) = lmeta;
109             return true;
110         }));
111 
112     auto result = validateReply(sessionStatBlob(&mgr, request));
113 
114     EXPECT_EQ(sizeof(rep) + lmeta.metadata.size(), result.size());
115     EXPECT_EQ(0, std::memcmp(result.data(), &rep, sizeof(rep)));
116     EXPECT_EQ(0, std::memcmp(result.data() + sizeof(rep), lmeta.metadata.data(),
117                              lmeta.metadata.size()));
118 }
119 } // namespace blobs
120