diff options
author | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2022-04-28 20:09:16 +0000 |
---|---|---|
committer | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2022-04-28 20:09:16 +0000 |
commit | e829cd3dd5c92dc0699347e66d0be8eb4c79fd44 (patch) | |
tree | fced7922414aa2c66a73af722daa5d4c2f04c0b4 | |
parent | d018d2669427948bc9a1d4110d4e34d223314d33 (diff) | |
parent | b1f22fd744d0c18e00fc8343b54ceebbbcf6ce8f (diff) | |
download | libfmq-e829cd3dd5c92dc0699347e66d0be8eb4c79fd44.tar.gz |
Snap for 8505378 from b1f22fd744d0c18e00fc8343b54ceebbbcf6ce8f to mainline-go-adservices-releaseaml_go_ads_330915100aml_go_ads_330915000aml_go_ads_330913000android13-mainline-go-adservices-release
Change-Id: Iefb1a67ef31672efa31e6cda450e7d1fec641790
-rw-r--r-- | fuzzer/fmq_fuzzer.cpp | 68 |
1 files changed, 43 insertions, 25 deletions
diff --git a/fuzzer/fmq_fuzzer.cpp b/fuzzer/fmq_fuzzer.cpp index 79ce5fc..8c8a78e 100644 --- a/fuzzer/fmq_fuzzer.cpp +++ b/fuzzer/fmq_fuzzer.cpp @@ -21,6 +21,7 @@ #include <thread> #include <android-base/logging.h> +#include <android-base/scopeguard.h> #include <fmq/AidlMessageQueue.h> #include <fmq/ConvertMQDescriptors.h> #include <fmq/EventFlag.h> @@ -58,7 +59,7 @@ static constexpr int kWriteCounterOffsetBytes = 8; static constexpr int kMaxNumSyncReaders = 1; static constexpr int kMaxNumUnsyncReaders = 5; -static constexpr int kMaxDataPerReader = 5; +static constexpr int kMaxDataPerReader = 1000; typedef android::AidlMessageQueue<payload_t, SynchronizedReadWrite> AidlMessageQueueSync; typedef android::AidlMessageQueue<payload_t, UnsynchronizedWrite> AidlMessageQueueUnsync; @@ -111,30 +112,35 @@ void reader(const Desc& desc, std::vector<uint8_t> readerData, bool userFd) { } template <typename Queue, typename Desc> -void readerBlocking(const Desc& desc, std::vector<uint8_t> readerData) { +void readerBlocking(const Desc& desc, std::vector<uint8_t>& readerData, + std::atomic<size_t>& readersNotFinished, + std::atomic<size_t>& writersNotFinished) { + android::base::ScopeGuard guard([&readersNotFinished]() { readersNotFinished--; }); Queue readMq(desc); if (!readMq.isValid()) { LOG(ERROR) << "read mq invalid"; return; } FuzzedDataProvider fdp(&readerData[0], readerData.size()); - bool success; do { size_t count = fdp.remaining_bytes() ? fdp.ConsumeIntegralInRange<size_t>(1, readMq.getQuantumCount()) : 1; std::vector<payload_t> data; data.resize(count); - success = readMq.readBlocking(data.data(), count, kBlockingTimeoutNs); - } while (success == true || fdp.remaining_bytes() > sizeof(size_t)); + readMq.readBlocking(data.data(), count, kBlockingTimeoutNs); + } while (fdp.remaining_bytes() > sizeof(size_t) && writersNotFinished > 0); } // Can't use blocking calls with Unsync queues(there is a static_assert) template <> void readerBlocking<AidlMessageQueueUnsync, AidlMQDescUnsync>(const AidlMQDescUnsync&, - std::vector<uint8_t>) {} + std::vector<uint8_t>&, + std::atomic<size_t>&, + std::atomic<size_t>&) {} template <> -void readerBlocking<MessageQueueUnsync, MQDescUnsync>(const MQDescUnsync&, std::vector<uint8_t>) {} +void readerBlocking<MessageQueueUnsync, MQDescUnsync>(const MQDescUnsync&, std::vector<uint8_t>&, + std::atomic<size_t>&, std::atomic<size_t>&) {} template <typename Queue> void writer(Queue& writeMq, FuzzedDataProvider& fdp, bool userFd) { @@ -168,8 +174,11 @@ void writer(Queue& writeMq, FuzzedDataProvider& fdp, bool userFd) { } template <typename Queue> -void writerBlocking(Queue& writeMq, FuzzedDataProvider& fdp) { - while (fdp.remaining_bytes() > sizeof(size_t)) { +void writerBlocking(Queue& writeMq, FuzzedDataProvider& fdp, + std::atomic<size_t>& writersNotFinished, + std::atomic<size_t>& readersNotFinished) { + android::base::ScopeGuard guard([&writersNotFinished]() { writersNotFinished--; }); + while (fdp.remaining_bytes() > sizeof(size_t) && readersNotFinished > 0) { size_t count = fdp.ConsumeIntegralInRange<size_t>(1, writeMq.getQuantumCount()); std::vector<payload_t> data; for (int i = 0; i < count; i++) { @@ -181,9 +190,11 @@ void writerBlocking(Queue& writeMq, FuzzedDataProvider& fdp) { // Can't use blocking calls with Unsync queues(there is a static_assert) template <> -void writerBlocking<AidlMessageQueueUnsync>(AidlMessageQueueUnsync&, FuzzedDataProvider&) {} +void writerBlocking<AidlMessageQueueUnsync>(AidlMessageQueueUnsync&, FuzzedDataProvider&, + std::atomic<size_t>&, std::atomic<size_t>&) {} template <> -void writerBlocking<MessageQueueUnsync>(MessageQueueUnsync&, FuzzedDataProvider&) {} +void writerBlocking<MessageQueueUnsync>(MessageQueueUnsync&, FuzzedDataProvider&, + std::atomic<size_t>&, std::atomic<size_t>&) {} template <typename Queue, typename Desc> void fuzzAidlWithReaders(std::vector<uint8_t>& writerData, @@ -207,25 +218,29 @@ void fuzzAidlWithReaders(std::vector<uint8_t>& writerData, const auto desc = writeMq.dupeDesc(); CHECK(desc.handle.fds[0].get() != -1); - std::vector<std::thread> clients; + std::atomic<size_t> readersNotFinished = readerData.size(); + std::atomic<size_t> writersNotFinished = 1; + std::vector<std::thread> readers; for (int i = 0; i < readerData.size(); i++) { if (blocking) { - clients.emplace_back(readerBlocking<Queue, Desc>, std::ref(desc), - std::ref(readerData[i])); + readers.emplace_back(readerBlocking<Queue, Desc>, std::ref(desc), + std::ref(readerData[i]), std::ref(readersNotFinished), + std::ref(writersNotFinished)); + } else { - clients.emplace_back(reader<Queue, Desc>, std::ref(desc), std::ref(readerData[i]), + readers.emplace_back(reader<Queue, Desc>, std::ref(desc), std::ref(readerData[i]), userFd); } } if (blocking) { - writerBlocking<Queue>(writeMq, fdp); + writerBlocking<Queue>(writeMq, fdp, writersNotFinished, readersNotFinished); } else { writer<Queue>(writeMq, fdp, userFd); } - for (auto& client : clients) { - client.join(); + for (auto& reader : readers) { + reader.join(); } } @@ -251,25 +266,28 @@ void fuzzHidlWithReaders(std::vector<uint8_t>& writerData, const auto desc = writeMq.getDesc(); CHECK(desc->isHandleValid()); - std::vector<std::thread> clients; + std::atomic<size_t> readersNotFinished = readerData.size(); + std::atomic<size_t> writersNotFinished = 1; + std::vector<std::thread> readers; for (int i = 0; i < readerData.size(); i++) { if (blocking) { - clients.emplace_back(readerBlocking<Queue, Desc>, std::ref(*desc), - std::ref(readerData[i])); + readers.emplace_back(readerBlocking<Queue, Desc>, std::ref(*desc), + std::ref(readerData[i]), std::ref(readersNotFinished), + std::ref(writersNotFinished)); } else { - clients.emplace_back(reader<Queue, Desc>, std::ref(*desc), std::ref(readerData[i]), + readers.emplace_back(reader<Queue, Desc>, std::ref(*desc), std::ref(readerData[i]), userFd); } } if (blocking) { - writerBlocking<Queue>(writeMq, fdp); + writerBlocking<Queue>(writeMq, fdp, writersNotFinished, readersNotFinished); } else { writer<Queue>(writeMq, fdp, userFd); } - for (auto& client : clients) { - client.join(); + for (auto& reader : readers) { + reader.join(); } } |