| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_NET_PROBE_MESSAGE_H_ | 5 #ifndef CHROME_BROWSER_NET_PROBE_MESSAGE_H_ |
| 6 #define CHROME_BROWSER_NET_PROBE_MESSAGE_H_ | 6 #define CHROME_BROWSER_NET_PROBE_MESSAGE_H_ |
| 7 | 7 |
| 8 #include <stdint.h> |
| 9 |
| 8 #include <string> | 10 #include <string> |
| 9 | 11 |
| 10 #include "base/basictypes.h" | |
| 11 #include "base/gtest_prod_util.h" | 12 #include "base/gtest_prod_util.h" |
| 13 #include "base/macros.h" |
| 12 #include "chrome/browser/net/probe_message.pb.h" | 14 #include "chrome/browser/net/probe_message.pb.h" |
| 13 | 15 |
| 14 namespace chrome_browser_net { | 16 namespace chrome_browser_net { |
| 15 | 17 |
| 16 // Packet format between client and server is defined in probe_message.proto. | 18 // Packet format between client and server is defined in probe_message.proto. |
| 17 class ProbeMessage { | 19 class ProbeMessage { |
| 18 public: | 20 public: |
| 19 ProbeMessage(); | 21 ProbeMessage(); |
| 20 | 22 |
| 21 // Generate a ProbeRequest packet. | 23 // Generate a ProbeRequest packet. |
| 22 void GenerateProbeRequest(const ProbePacket_Token& received_token, | 24 void GenerateProbeRequest(const ProbePacket_Token& received_token, |
| 23 uint32 group_id, | 25 uint32_t group_id, |
| 24 uint32 probe_size, | 26 uint32_t probe_size, |
| 25 uint32 pacing_interval_micros, | 27 uint32_t pacing_interval_micros, |
| 26 uint32 number_probe_packets, | 28 uint32_t number_probe_packets, |
| 27 ProbePacket* output); | 29 ProbePacket* output); |
| 28 // Make an encoded packet (string) from a ProbePacket object. | 30 // Make an encoded packet (string) from a ProbePacket object. |
| 29 std::string MakeEncodedPacket(const ProbePacket& packet) const; | 31 std::string MakeEncodedPacket(const ProbePacket& packet) const; |
| 30 // Fill some common fields in the packet header. | 32 // Fill some common fields in the packet header. |
| 31 void SetPacketHeader(ProbePacket_Type packet_type, | 33 void SetPacketHeader(ProbePacket_Type packet_type, |
| 32 ProbePacket* probe_packet) const; | 34 ProbePacket* probe_packet) const; |
| 33 // Parse the input string (which is an encoded packet) and save the result to | 35 // Parse the input string (which is an encoded packet) and save the result to |
| 34 // packet. Return true if there is no error and false otherwise. | 36 // packet. Return true if there is no error and false otherwise. |
| 35 bool ParseInput(const std::string& input, ProbePacket* packet) const; | 37 bool ParseInput(const std::string& input, ProbePacket* packet) const; |
| 36 | 38 |
| 37 static const uint32 kMaxProbePacketBytes; | 39 static const uint32_t kMaxProbePacketBytes; |
| 38 | 40 |
| 39 private: | 41 private: |
| 40 // For unittest. | 42 // For unittest. |
| 41 friend class ProbeMessageTest; | 43 friend class ProbeMessageTest; |
| 42 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestChecksum); | 44 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestChecksum); |
| 43 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestEncode); | 45 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestEncode); |
| 44 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestGenerateProbeRequest); | 46 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestGenerateProbeRequest); |
| 45 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestSetPacketHeader); | 47 FRIEND_TEST_ALL_PREFIXES(ProbeMessageTest, TestSetPacketHeader); |
| 46 | 48 |
| 47 // Compute the checksum of the padding string. | 49 // Compute the checksum of the padding string. |
| 48 uint32 Checksum(const std::string& str) const; | 50 uint32_t Checksum(const std::string& str) const; |
| 49 | 51 |
| 50 // Encode the packet with kEncodingString. This is also used for decoding. | 52 // Encode the packet with kEncodingString. This is also used for decoding. |
| 51 std::string Encode(const std::string& input) const; | 53 std::string Encode(const std::string& input) const; |
| 52 | 54 |
| 53 static const uint32 kVersion; | 55 static const uint32_t kVersion; |
| 54 static const uint32 kMaxNumberProbePackets; | 56 static const uint32_t kMaxNumberProbePackets; |
| 55 static const uint32 kMaxPacingIntervalMicros; | 57 static const uint32_t kMaxPacingIntervalMicros; |
| 56 static const char kEncodingString[]; | 58 static const char kEncodingString[]; |
| 57 | 59 |
| 58 DISALLOW_COPY_AND_ASSIGN(ProbeMessage); | 60 DISALLOW_COPY_AND_ASSIGN(ProbeMessage); |
| 59 }; | 61 }; |
| 60 } // namespace chrome_browser_net | 62 } // namespace chrome_browser_net |
| 61 #endif // CHROME_BROWSER_NET_PROBE_MESSAGE_H_ | 63 #endif // CHROME_BROWSER_NET_PROBE_MESSAGE_H_ |
| OLD | NEW |