|
38 | 38 |
|
39 | 39 | class PeerManager; |
40 | 40 |
|
| 41 | +class FuzzedSock : public Sock |
| 42 | +{ |
| 43 | + FuzzedDataProvider& m_fuzzed_data_provider; |
| 44 | + |
| 45 | + /** |
| 46 | + * Data to return when `MSG_PEEK` is used as a `Recv()` flag. |
| 47 | + * If `MSG_PEEK` is used, then our `Recv()` returns some random data as usual, but on the next |
| 48 | + * `Recv()` call we must return the same data, thus we remember it here. |
| 49 | + */ |
| 50 | + mutable std::optional<uint8_t> m_peek_data; |
| 51 | + |
| 52 | +public: |
| 53 | + explicit FuzzedSock(FuzzedDataProvider& fuzzed_data_provider); |
| 54 | + |
| 55 | + ~FuzzedSock() override; |
| 56 | + |
| 57 | + FuzzedSock& operator=(Sock&& other) override; |
| 58 | + |
| 59 | + void Reset() override; |
| 60 | + |
| 61 | + ssize_t Send(const void* data, size_t len, int flags) const override; |
| 62 | + |
| 63 | + ssize_t Recv(void* buf, size_t len, int flags) const override; |
| 64 | + |
| 65 | + int Connect(const sockaddr*, socklen_t) const override; |
| 66 | + |
| 67 | + std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override; |
| 68 | + |
| 69 | + int GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* opt_len) const override; |
| 70 | + |
| 71 | + bool Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred = nullptr) const override; |
| 72 | + |
| 73 | + bool IsConnected(std::string& errmsg) const override; |
| 74 | +}; |
| 75 | + |
| 76 | +[[nodiscard]] inline FuzzedSock ConsumeSock(FuzzedDataProvider& fuzzed_data_provider) |
| 77 | +{ |
| 78 | + return FuzzedSock{fuzzed_data_provider}; |
| 79 | +} |
| 80 | + |
41 | 81 | template <typename... Callables> |
42 | 82 | size_t CallOneOf(FuzzedDataProvider& fuzzed_data_provider, Callables... callables) |
43 | 83 | { |
@@ -375,44 +415,4 @@ void ReadFromStream(FuzzedDataProvider& fuzzed_data_provider, Stream& stream) no |
375 | 415 | } |
376 | 416 | } |
377 | 417 |
|
378 | | -class FuzzedSock : public Sock |
379 | | -{ |
380 | | - FuzzedDataProvider& m_fuzzed_data_provider; |
381 | | - |
382 | | - /** |
383 | | - * Data to return when `MSG_PEEK` is used as a `Recv()` flag. |
384 | | - * If `MSG_PEEK` is used, then our `Recv()` returns some random data as usual, but on the next |
385 | | - * `Recv()` call we must return the same data, thus we remember it here. |
386 | | - */ |
387 | | - mutable std::optional<uint8_t> m_peek_data; |
388 | | - |
389 | | -public: |
390 | | - explicit FuzzedSock(FuzzedDataProvider& fuzzed_data_provider); |
391 | | - |
392 | | - ~FuzzedSock() override; |
393 | | - |
394 | | - FuzzedSock& operator=(Sock&& other) override; |
395 | | - |
396 | | - void Reset() override; |
397 | | - |
398 | | - ssize_t Send(const void* data, size_t len, int flags) const override; |
399 | | - |
400 | | - ssize_t Recv(void* buf, size_t len, int flags) const override; |
401 | | - |
402 | | - int Connect(const sockaddr*, socklen_t) const override; |
403 | | - |
404 | | - std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override; |
405 | | - |
406 | | - int GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* opt_len) const override; |
407 | | - |
408 | | - bool Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred = nullptr) const override; |
409 | | - |
410 | | - bool IsConnected(std::string& errmsg) const override; |
411 | | -}; |
412 | | - |
413 | | -[[nodiscard]] inline FuzzedSock ConsumeSock(FuzzedDataProvider& fuzzed_data_provider) |
414 | | -{ |
415 | | - return FuzzedSock{fuzzed_data_provider}; |
416 | | -} |
417 | | - |
418 | 418 | #endif // BITCOIN_TEST_FUZZ_UTIL_H |
0 commit comments