| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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_SAFE_BROWSING_PROTOCOL_PARSER_H_ | 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_PROTOCOL_PARSER_H_ |
| 6 #define CHROME_BROWSER_SAFE_BROWSING_PROTOCOL_PARSER_H_ | 6 #define CHROME_BROWSER_SAFE_BROWSING_PROTOCOL_PARSER_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 // Parse the data returned from the chunk response. | 9 // Parse the data returned from the chunk response. |
| 10 // | 10 // |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 49 #include "chrome/browser/safe_browsing/safe_browsing_util.h" | 49 #include "chrome/browser/safe_browsing/safe_browsing_util.h" |
| 50 | 50 |
| 51 | 51 |
| 52 class SafeBrowsingProtocolParser { | 52 class SafeBrowsingProtocolParser { |
| 53 public: | 53 public: |
| 54 SafeBrowsingProtocolParser(); | 54 SafeBrowsingProtocolParser(); |
| 55 | 55 |
| 56 // Parse the response of an update request. Results for chunk deletions (both | 56 // Parse the response of an update request. Results for chunk deletions (both |
| 57 // add-del and sub-del are returned in 'chunk_deletes', and new chunk URLs to | 57 // add-del and sub-del are returned in 'chunk_deletes', and new chunk URLs to |
| 58 // download are contained in 'chunk_urls'. The next time the client is allowed | 58 // download are contained in 'chunk_urls'. The next time the client is allowed |
| 59 // to request another update is returned in 'next_update_sec'. If the service | 59 // to request another update is returned in 'next_update_sec'. 'reset' will |
| 60 // wants us to retrieve new MAC keys, 're_key' will be set to true. If we are | 60 // be set to true if the SafeBrowsing service wants us to dump our database. |
| 61 // using MACs to verify responses, the 'key' must be set to the private key | |
| 62 // returned from the SafeBrowsing servers. 'reset' will be set to true if the | |
| 63 // SafeBrowsing service wants us to dump our database. | |
| 64 // Returns 'true'if it was able to decode the chunk properly, 'false' if not | 61 // Returns 'true'if it was able to decode the chunk properly, 'false' if not |
| 65 // decoded properly and the results should be ignored. | 62 // decoded properly and the results should be ignored. |
| 66 bool ParseUpdate(const char* chunk_data, | 63 bool ParseUpdate(const char* chunk_data, |
| 67 int chunk_len, | 64 int chunk_len, |
| 68 const std::string& key, | |
| 69 int* next_update_sec, | 65 int* next_update_sec, |
| 70 bool* re_key, | |
| 71 bool* reset, | 66 bool* reset, |
| 72 std::vector<SBChunkDelete>* chunk_deletes, | 67 std::vector<SBChunkDelete>* chunk_deletes, |
| 73 std::vector<ChunkUrl>* chunk_urls); | 68 std::vector<ChunkUrl>* chunk_urls); |
| 74 | 69 |
| 75 // Parse the response from a chunk URL request and returns the hosts/prefixes | 70 // Parse the response from a chunk URL request and returns the hosts/prefixes |
| 76 // for adds and subs in "chunks". Returns 'true' on successful parsing, | 71 // for adds and subs in "chunks". Returns 'true' on successful parsing, |
| 77 // 'false' otherwise. Any result should be ignored when a parse has failed. | 72 // 'false' otherwise. Any result should be ignored when a parse has failed. |
| 78 bool ParseChunk(const std::string& list_name, | 73 bool ParseChunk(const std::string& list_name, |
| 79 const char* chunk_data, | 74 const char* chunk_data, |
| 80 int chunk_len, | 75 int chunk_len, |
| 81 const std::string& key, | |
| 82 const std::string& mac, | |
| 83 bool* re_key, | |
| 84 SBChunkList* chunks); | 76 SBChunkList* chunks); |
| 85 | 77 |
| 86 // Parse the result of a GetHash request, returning the list of full hashes. | 78 // Parse the result of a GetHash request, returning the list of full hashes. |
| 87 // If we are checking for valid MACs, the caller should populate 'key'. | |
| 88 bool ParseGetHash(const char* chunk_data, | 79 bool ParseGetHash(const char* chunk_data, |
| 89 int chunk_len, | 80 int chunk_len, |
| 90 const std::string& key, | |
| 91 bool* re_key, | |
| 92 std::vector<SBFullHashResult>* full_hashes); | 81 std::vector<SBFullHashResult>* full_hashes); |
| 93 | 82 |
| 94 // Convert a list of partial hashes into a proper GetHash request. | 83 // Convert a list of partial hashes into a proper GetHash request. |
| 95 void FormatGetHash(const std::vector<SBPrefix>& prefixes, | 84 void FormatGetHash(const std::vector<SBPrefix>& prefixes, |
| 96 std::string* request); | 85 std::string* request); |
| 97 | 86 |
| 98 // Parse the keys used for subsequent communications with the SafeBrowsing | |
| 99 // servers. Returns true on successful parse, false on parse error. | |
| 100 bool ParseNewKey(const char* chunk_data, | |
| 101 int chunk_length, | |
| 102 std::string* client_key, | |
| 103 std::string* wrapped_key); | |
| 104 | |
| 105 private: | 87 private: |
| 106 bool ParseAddChunk(const std::string& list_name, | 88 bool ParseAddChunk(const std::string& list_name, |
| 107 const char* data, | 89 const char* data, |
| 108 int data_len, | 90 int data_len, |
| 109 int hash_len, | 91 int hash_len, |
| 110 std::deque<SBChunkHost>* hosts); | 92 std::deque<SBChunkHost>* hosts); |
| 111 bool ParseSubChunk(const std::string& list_name, | 93 bool ParseSubChunk(const std::string& list_name, |
| 112 const char* data, | 94 const char* data, |
| 113 int data_len, | 95 int data_len, |
| 114 int hash_len, | 96 int hash_len, |
| 115 std::deque<SBChunkHost>* hosts); | 97 std::deque<SBChunkHost>* hosts); |
| 116 | 98 |
| 117 // Helper functions used by ParseAddChunk and ParseSubChunk. | 99 // Helper functions used by ParseAddChunk and ParseSubChunk. |
| 118 static bool ReadHostAndPrefixCount(const char** data, | 100 static bool ReadHostAndPrefixCount(const char** data, |
| 119 int* remaining, | 101 int* remaining, |
| 120 SBPrefix* host, | 102 SBPrefix* host, |
| 121 int* count); | 103 int* count); |
| 122 static bool ReadChunkId(const char** data, int* remaining, int* chunk_id); | 104 static bool ReadChunkId(const char** data, int* remaining, int* chunk_id); |
| 123 static bool ReadPrefixes( | 105 static bool ReadPrefixes( |
| 124 const char** data, int* remaining, SBEntry* entry, int count); | 106 const char** data, int* remaining, SBEntry* entry, int count); |
| 125 | 107 |
| 126 // The name of the current list | 108 // The name of the current list |
| 127 std::string list_name_; | 109 std::string list_name_; |
| 128 | 110 |
| 129 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingProtocolParser); | 111 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingProtocolParser); |
| 130 }; | 112 }; |
| 131 | 113 |
| 132 | 114 |
| 133 #endif // CHROME_BROWSER_SAFE_BROWSING_PROTOCOL_PARSER_H_ | 115 #endif // CHROME_BROWSER_SAFE_BROWSING_PROTOCOL_PARSER_H_ |
| OLD | NEW |