DNAMP3/PAK: Make use of unique_ptr in getBuffer()

Same motivation as the previous change: seating the allocated memory
into a type that will always do the right thing immediately, rather than
later.
This commit is contained in:
Lioncash 2020-04-05 10:20:57 -04:00
parent 9e503edaf1
commit fa2f37a14c
1 changed files with 7 additions and 7 deletions

View File

@ -142,7 +142,7 @@ std::unique_ptr<atUint8[]> PAK::Entry::getBuffer(const nod::Node& pak, atUint64&
strm->read(&head, 8);
if (head.magic != CMPD) {
Log.report(logvisor::Error, fmt("invalid CMPD block"));
return std::unique_ptr<atUint8[]>();
return nullptr;
}
head.blockCount = hecl::SBig(head.blockCount);
@ -165,8 +165,8 @@ std::unique_ptr<atUint8[]> PAK::Entry::getBuffer(const nod::Node& pak, atUint64&
}
std::unique_ptr<atUint8[]> compBuf(new atUint8[maxBlockSz]);
atUint8* buf = new atUint8[totalDecompSz];
atUint8* bufCur = buf;
std::unique_ptr<atUint8[]> buf{new atUint8[totalDecompSz]};
atUint8* bufCur = buf.get();
for (atUint32 b = 0; b < head.blockCount; ++b) {
Block& block = blocks[b];
atUint8* compBufCur = compBuf.get();
@ -189,12 +189,12 @@ std::unique_ptr<atUint8[]> PAK::Entry::getBuffer(const nod::Node& pak, atUint64&
}
szOut = totalDecompSz;
return std::unique_ptr<atUint8[]>(buf);
return buf;
} else {
atUint8* buf = new atUint8[size];
pak.beginReadStream(offset)->read(buf, size);
std::unique_ptr<atUint8[]> buf{new atUint8[size]};
pak.beginReadStream(offset)->read(buf.get(), size);
szOut = size;
return std::unique_ptr<atUint8[]>(buf);
return buf;
}
}