aboutsummaryrefslogtreecommitdiffstats
path: root/string.c
diff options
context:
space:
mode:
Diffstat (limited to 'string.c')
-rw-r--r--string.c24
1 files changed, 18 insertions, 6 deletions
diff --git a/string.c b/string.c
index 405500b1d2..b714864474 100644
--- a/string.c
+++ b/string.c
@@ -5380,19 +5380,31 @@ rb_str_count(int argc, VALUE *argv, VALUE str)
rb_raise(rb_eArgError, "wrong number of arguments");
}
for (i=0; i<argc; i++) {
- VALUE s = argv[i];
+ VALUE tstr = argv[i];
+ unsigned char c;
- StringValue(s);
- enc = rb_enc_check(str, s);
- tr_setup_table(s, table, i==0, &del, &nodel, enc);
+ StringValue(tstr);
+ enc = rb_enc_check(str, tstr);
+ if (argc == 1 && RSTRING_LEN(tstr) == 1 && rb_enc_asciicompat(enc) &&
+ (c = RSTRING_PTR(tstr)[0]) < 0x80 && !is_broken_string(str)) {
+ int n = 0;
+
+ s = RSTRING_PTR(str);
+ if (!s || RSTRING_LEN(str) == 0) return INT2FIX(0);
+ send = RSTRING_END(str);
+ while (s < send) {
+ if (*(unsigned char*)s++ == c) n++;
+ }
+ return INT2NUM(n);
+ }
+ tr_setup_table(tstr, table, i==0, &del, &nodel, enc);
}
s = RSTRING_PTR(str);
if (!s || RSTRING_LEN(str) == 0) return INT2FIX(0);
send = RSTRING_END(str);
- i = 0;
ascompat = rb_enc_asciicompat(enc);
-
+ i = 0;
while (s < send) {
unsigned int c;
int clen;