15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef NET_BASE_TEST_DATA_STREAM_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define NET_BASE_TEST_DATA_STREAM_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string.h> // for memcpy(). 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <algorithm> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_export.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This is a class for generating an infinite stream of data which can be 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// verified independently to be the correct stream of data. 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class NET_EXPORT TestDataStream { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestDataStream(); 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Fill |buffer| with |length| bytes of data from the stream. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void GetBytes(char* buffer, int length); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify that |buffer| contains the expected next |length| bytes from the 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // stream. Returns true if correct, false otherwise. 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool VerifyBytes(const char *buffer, int length); 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Resets all the data. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Reset(); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If there is no data spilled over from the previous index, advance the 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // index and fill the buffer. 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AdvanceIndex(); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Consume data from the spill buffer. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Consume(int bytes); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int index_; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_remaining_; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char buffer_[16]; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char* buffer_ptr_; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // NET_BASE_TEST_DATA_STREAM_H_ 48