From 500c79cf55bbdb263dc9ae65a026638e09c4e892 Mon Sep 17 00:00:00 2001 From: Daniel Lehman Date: Thu, 29 Dec 2011 16:08:39 -0800 Subject: [PATCH] msvcp90: handle npos as a len in basic_string<>::replace passing npos as the len causes off+len to rollover which in turn causes later EOS to be misplaced --- dlls/msvcp90/string.c | 8 +++--- dlls/msvcp90/tests/string.c | 54 +++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 58 insertions(+), 4 deletions(-) diff --git a/dlls/msvcp90/string.c b/dlls/msvcp90/string.c index d8bc6f3..795a819 100644 --- a/dlls/msvcp90/string.c +++ b/dlls/msvcp90/string.c @@ -1949,7 +1949,7 @@ basic_string_char* __thiscall basic_string_char_replace_cstr_len(basic_string_ch if(this->size < off) MSVCP__String_base_Xran(); - if(off+len > this->size) + if(off+len < off || off+len > this->size) len = this->size-off; if(MSVCP_basic_string_char_npos-str_len <= this->size-len) @@ -2040,7 +2040,7 @@ basic_string_char* __thiscall basic_string_char_replace_ch(basic_string_char *th if(this->size < off) MSVCP__String_base_Xran(); - if(off+len > this->size) + if(off+len < off || off+len > this->size) len = this->size-off; if(MSVCP_basic_string_char_npos-count <= this->size-len) @@ -3901,7 +3901,7 @@ basic_string_wchar* __thiscall basic_string_wchar_replace_cstr_len(basic_string_ if(this->size < off) MSVCP__String_base_Xran(); - if(off+len > this->size) + if(off+len < off || off+len > this->size) len = this->size-off; if(MSVCP_basic_string_wchar_npos-str_len <= this->size-len) @@ -4000,7 +4000,7 @@ basic_string_wchar* __thiscall basic_string_wchar_replace_ch(basic_string_wchar if(this->size < off) MSVCP__String_base_Xran(); - if(off+len > this->size) + if(off+len < off || off+len > this->size) len = this->size-off; if(MSVCP_basic_string_wchar_npos-count <= this->size-len) diff --git a/dlls/msvcp90/tests/string.c b/dlls/msvcp90/tests/string.c index 8bc5f18..1e3ccec 100644 --- a/dlls/msvcp90/tests/string.c +++ b/dlls/msvcp90/tests/string.c @@ -76,6 +76,7 @@ static int (__thiscall *p_basic_string_char_compare_substr_substr)(basic_string_ static int (__thiscall *p_basic_string_char_compare_substr_cstr_len)(basic_string_char*, size_t, size_t, const char*, size_t); static size_t (__thiscall *p_basic_string_char_find_cstr_substr)(basic_string_char*, const char*, size_t, size_t); static size_t (__thiscall *p_basic_string_char_rfind_cstr_substr)(basic_string_char*, const char*, size_t, size_t); +static basic_string_char* (__thiscall *p_basic_string_char_replace_cstr)(basic_string_char*, size_t, size_t, const char*); static size_t *p_basic_string_char_npos; @@ -225,6 +226,8 @@ static BOOL init(void) "?find@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEBA_KPEBD_K1@Z"); SET(p_basic_string_char_rfind_cstr_substr, "?rfind@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEBA_KPEBD_K1@Z"); + SET(p_basic_string_char_replace_cstr, + "?replace@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEAAAEAV12@_K0PEBD@Z"); SET(p_basic_string_char_npos, "?npos@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@2_KB"); @@ -289,6 +292,8 @@ static BOOL init(void) "?find@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QBEIPBDII@Z"); SET(p_basic_string_char_rfind_cstr_substr, "?rfind@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QBEIPBDII@Z"); + SET(p_basic_string_char_replace_cstr, + "?replace@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QAEAAV12@IIPBD@Z"); SET(p_basic_string_char_npos, "?npos@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@2IB"); @@ -549,6 +554,54 @@ static void test_basic_string_char_rfind(void) { } } +static void test_basic_string_char_replace(void) { + struct replace_char_test { + const char *str; + size_t off; + size_t len; + const char *replace; + const char *ret; + }; + + int i; + basic_string_char str; + basic_string_char *ret; + struct replace_char_test tests[] = { + { "", 0, 0, "", "" }, /* empty string */ + { "", 0, 10, "", "" }, /* empty string with invalid len */ + + { "ABCDEF", 0, 0, "", "ABCDEF" }, /* replace with empty string */ + { "ABCDEF", 0, 0, "-", "-ABCDEF"}, /* replace with 0 len */ + { "ABCDEF", 0, 1, "-", "-BCDEF" }, /* replace 1 at beginning */ + { "ABCDEF", 0, 3, "-", "-DEF" }, /* replace 3 at beginning */ + { "ABCDEF", 0, 42, "-", "-" }, /* replace whole string with invalid long len */ + { "ABCDEF", 0, *p_basic_string_char_npos, "-", "-" }, /* replace whole string with npos */ + + { "ABCDEF", 5, 0, "", "ABCDEF" }, /* replace at end with empty string */ + { "ABCDEF", 5, 0, "-", "ABCDE-F"}, /* replace at end with 0 len */ + { "ABCDEF", 5, 1, "-", "ABCDE-" }, /* replace 1 at end */ + { "ABCDEF", 5, 42, "-", "ABCDE-" }, /* replace end with invalid long len */ + { "ABCDEF", 5, *p_basic_string_char_npos, "-", "ABCDE-" }, /* replace end with npos */ + + { "ABCDEF", 6, 0, "", "ABCDEF" }, /* replace after end with empty string */ + { "ABCDEF", 6, 0, "-", "ABCDEF-"}, /* replace after end with 0 len */ + { "ABCDEF", 6, 1, "-", "ABCDEF-" }, /* replace 1 after end */ + { "ABCDEF", 6, 42, "-", "ABCDEF-" }, /* replace after end with invalid long len */ + { "ABCDEF", 6, *p_basic_string_char_npos, "-", "ABCDEF-" }, /* replace after end with npos */ + }; + + for(i=0; i