1 #include <hwtrust/hwtrust.h>
2 #include <hwtrust/lib.rs.h>
3
4 using android::base::Error;
5 using android::base::Result;
6
7 namespace hwtrust {
8
9 struct BoxedDiceChain {
10 ::rust::Box<rust::DiceChain> chain;
11 };
12
13 // Define with a full definition of BoxedDiceChain to satisfy unique_ptr.
~DiceChain()14 DiceChain::~DiceChain() {}
15
DiceChain(std::unique_ptr<BoxedDiceChain> chain,size_t size)16 DiceChain::DiceChain(std::unique_ptr<BoxedDiceChain> chain, size_t size) noexcept
17 : chain_(std::move(chain)), size_(size) {}
18
Verify(const std::vector<uint8_t> & chain,DiceChain::Kind kind)19 Result<DiceChain> DiceChain::Verify(const std::vector<uint8_t>& chain, DiceChain::Kind kind) noexcept {
20 rust::DiceChainKind chainKind;
21 switch (kind) {
22 case DiceChain::Kind::kVsr13:
23 chainKind = rust::DiceChainKind::Vsr13;
24 break;
25 case DiceChain::Kind::kVsr14:
26 chainKind = rust::DiceChainKind::Vsr14;
27 break;
28 }
29 auto res = rust::VerifyDiceChain({chain.data(), chain.size()}, chainKind);
30 if (!res.error.empty()) {
31 return Error() << static_cast<std::string>(res.error);
32 }
33 BoxedDiceChain boxedChain = { std::move(res.chain) };
34 auto diceChain = std::make_unique<BoxedDiceChain>(std::move(boxedChain));
35 return DiceChain(std::move(diceChain), res.len);
36 }
37
CosePublicKeys() const38 Result<std::vector<std::vector<uint8_t>>> DiceChain::CosePublicKeys() const noexcept {
39 std::vector<std::vector<uint8_t>> result;
40 for (auto i = 0; i < size_; ++i) {
41 auto key = rust::GetDiceChainPublicKey(*chain_->chain, i);
42 if (key.empty()) {
43 return Error() << "Failed to get public key from chain entry " << i;
44 }
45 result.emplace_back(key.begin(), key.end());
46 }
47 return result;
48 }
49
50 } // namespace hwtrust
51