Index: webkit/browser/database/database_util.cc |
diff --git a/webkit/browser/database/database_util.cc b/webkit/browser/database/database_util.cc |
index f037638b2087e309dc6170ad60d45edbb4e70391..0fdf576b26f000949c08203a116185117670e475 100644 |
--- a/webkit/browser/database/database_util.cc |
+++ b/webkit/browser/database/database_util.cc |
@@ -8,9 +8,28 @@ |
#include "base/strings/utf_string_conversions.h" |
#include "webkit/browser/database/database_tracker.h" |
#include "webkit/browser/database/vfs_backend.h" |
+#include "webkit/common/database/database_identifier.h" |
namespace webkit_database { |
+namespace { |
+ |
+bool IsSafeSuffix(const base::string16& suffix) { |
+ base::char16 prev_c = 0; |
+ for (base::string16::const_iterator it = suffix.begin(); |
+ it < suffix.end(); ++it) { |
+ base::char16 c = *it; |
+ if (!(IsAsciiAlpha(c) || IsAsciiDigit(c) || c == '-' || c == '.')) |
+ return false; |
+ if (c == '.' && prev_c == '.') |
+ return false; |
+ prev_c = c; |
+ } |
+ return true; |
+} |
+ |
+} |
+ |
const char DatabaseUtil::kJournalFileSuffix[] = "-journal"; |
bool DatabaseUtil::CrackVfsFileName(const base::string16& vfs_file_name, |
@@ -31,18 +50,27 @@ bool DatabaseUtil::CrackVfsFileName(const base::string16& vfs_file_name, |
return false; |
} |
- if (origin_identifier) { |
- *origin_identifier = UTF16ToASCII( |
+ std::string origin_id = UTF16ToASCII( |
vfs_file_name.substr(0, first_slash_index)); |
- } |
+ if (!IsValidOriginIdentifier(origin_id)) |
+ return false; |
+ |
+ base::string16 suffix = vfs_file_name.substr( |
jschuh
2014/03/14 03:38:00
I can't tell how we verify host:port. Could you ch
michaeln
2014/03/14 03:55:56
Maybe DatabaseIdentifier::Parse should test for ':
|
+ last_pound_index + 1, vfs_file_name.length() - last_pound_index - 1); |
+ if (!IsSafeSuffix(suffix)) |
+ return false; |
+ |
+ if (origin_identifier) |
+ *origin_identifier = origin_id; |
+ |
if (database_name) { |
*database_name = vfs_file_name.substr( |
first_slash_index + 1, last_pound_index - first_slash_index - 1); |
} |
- if (sqlite_suffix) { |
- *sqlite_suffix = vfs_file_name.substr( |
- last_pound_index + 1, vfs_file_name.length() - last_pound_index - 1); |
- } |
+ |
+ if (sqlite_suffix) |
+ *sqlite_suffix = suffix; |
+ |
return true; |
} |
@@ -76,10 +104,14 @@ bool DatabaseUtil::IsValidOriginIdentifier( |
char forbidden[] = {'\\', '/', '\0'}; |
std::string::size_type pos = origin_identifier.find(dotdot); |
- if (pos == std::string::npos) |
- pos = origin_identifier.find_first_of(forbidden, 0, arraysize(forbidden)); |
+ if (origin_identifier.find(dotdot) != std::string::npos) |
+ return false; |
+ if (origin_identifier.find_first_of(forbidden, 0, arraysize(forbidden)) != |
+ std::string::npos) { |
+ return false; |
+ } |
- return pos == std::string::npos; |
+ return GetOriginFromIdentifier(origin_identifier).is_valid(); |
} |
} // namespace webkit_database |