mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-03 16:39:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

SF bug #422088: [OSF1 alpha] string.replace().

Browse source 
Platform blew up on "123".replace("123", "").  Michael Hudson pinned the
blame on platform malloc(0) returning NULL.
This is a candidate for all bugfix releases.
This commit is contained in:
Tim Peters 2001-05-09 23:00:26 +00:00
parent 0f8b494df6
commit 1a7b3eee94
2 changed files with 41 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

63
Modules/stropmodule.c
View file

@ -1061,36 +1061,43 @@ mymemreplace(char *str, int len,
nfound = nfound > count ? count : nfound;
if (nfound == 0)
goto return_same;
new_len = len + nfound*(sub_len - pat_len);
new_s = (char *)PyMem_MALLOC(new_len);
if (new_s == NULL) return NULL;
*out_len = new_len;
out_s = new_s;
while (len > 0) {
/* find index of next instance of pattern */
offset = mymemfind(str, len, pat, pat_len);
/* if not found, break out of loop */
if (offset == -1) break;
/* copy non matching part of input string */
memcpy(new_s, str, offset); /* copy part of str before pat */
str += offset + pat_len; /* move str past pattern */
len -= offset + pat_len; /* reduce length of str remaining */
/* copy substitute into the output string */
new_s += offset; /* move new_s to dest for sub string */
memcpy(new_s, sub, sub_len); /* copy substring into new_s */
new_s += sub_len; /* offset new_s past sub string */
/* break when we've done count replacements */
if (--count == 0) break;
if (new_len == 0) {
out_s = "";
}
/* copy any remaining values into output string */
if (len > 0)
memcpy(new_s, str, len);
else {
assert(new_len > 0);
new_s = (char *)PyMem_MALLOC(new_len);
if (new_s == NULL)
return NULL;
out_s = new_s;
while (len > 0) {
/* find index of next instance of pattern */
offset = mymemfind(str, len, pat, pat_len);
if (offset == -1)
break;
/* copy non matching part of input string */
memcpy(new_s, str, offset);
str += offset + pat_len;
len -= offset + pat_len;
/* copy substitute into the output string */
new_s += offset;
memcpy(new_s, sub, sub_len);
new_s += sub_len;
/* note count==0 is effectively infinity */
if (--count == 0)
break;
}
/* copy any remaining values into output string */
if (len > 0)
memcpy(new_s, str, len);
}
*out_len = new_len;
return out_s;
return_same: