Fix for ID 20010306.008, UTF-8 and \w without 'use utf8' coredump.
[p5sagit/p5-mst-13.2.git] / doop.c
1 /*    doop.c
2  *
3  *    Copyright (c) 1991-2001, Larry Wall
4  *
5  *    You may distribute under the terms of either the GNU General Public
6  *    License or the Artistic License, as specified in the README file.
7  *
8  */
9
10 /*
11  * "'So that was the job I felt I had to do when I started,' thought Sam."
12  */
13
14 #include "EXTERN.h"
15 #define PERL_IN_DOOP_C
16 #include "perl.h"
17
18 #ifndef PERL_MICRO
19 #if !defined(NSIG) || defined(M_UNIX) || defined(M_XENIX)
20 #include <signal.h>
21 #endif
22 #endif
23
24 STATIC I32
25 S_do_trans_simple(pTHX_ SV *sv)
26 {
27     U8 *s;
28     U8 *d;
29     U8 *send;
30     U8 *dstart;
31     I32 matches = 0;
32     I32 grows = PL_op->op_private & OPpTRANS_GROWS;
33     STRLEN len;
34     short *tbl;
35     I32 ch;
36
37     tbl = (short*)cPVOP->op_pv;
38     if (!tbl)
39         Perl_croak(aTHX_ "panic: do_trans_simple");
40
41     s = (U8*)SvPV(sv, len);
42     send = s + len;
43
44     /* First, take care of non-UTF8 input strings, because they're easy */
45     if (!SvUTF8(sv)) {
46         while (s < send) {
47             if ((ch = tbl[*s]) >= 0) {
48                 matches++;
49                 *s++ = ch;
50             }
51             else
52                 s++;
53         }
54         SvSETMAGIC(sv);
55         return matches;
56     }
57
58     /* Allow for expansion: $_="a".chr(400); tr/a/\xFE/, FE needs encoding */
59     if (grows)
60         New(0, d, len*2+1, U8);
61     else
62         d = s;
63     dstart = d;
64     while (s < send) {
65         STRLEN ulen;
66         UV c;
67
68         /* Need to check this, otherwise 128..255 won't match */
69         c = utf8n_to_uvchr(s, send - s, &ulen, 0);
70         if (c < 0x100 && (ch = tbl[c]) >= 0) {
71             matches++;
72             d = uvchr_to_utf8(d, ch);
73             s += ulen;
74         }
75         else { /* No match -> copy */
76             Copy(s, d, ulen, U8);
77             d += ulen;
78             s += ulen;
79         }
80     }
81     if (grows) {
82         sv_setpvn(sv, (char*)dstart, d - dstart);
83         Safefree(dstart);
84     }
85     else {
86         *d = '\0';
87         SvCUR_set(sv, d - dstart);
88     }
89     SvUTF8_on(sv);
90     SvSETMAGIC(sv);
91     return matches;
92 }
93
94 STATIC I32
95 S_do_trans_count(pTHX_ SV *sv)/* SPC - OK */
96 {
97     U8 *s;
98     U8 *send;
99     I32 matches = 0;
100     STRLEN len;
101     short *tbl;
102     I32 complement = PL_op->op_private & OPpTRANS_COMPLEMENT;
103
104     tbl = (short*)cPVOP->op_pv;
105     if (!tbl)
106         Perl_croak(aTHX_ "panic: do_trans_count");
107
108     s = (U8*)SvPV(sv, len);
109     send = s + len;
110
111     if (!SvUTF8(sv))
112         while (s < send) {
113             if (tbl[*s++] >= 0)
114                 matches++;
115         }
116     else
117         while (s < send) {
118             UV c;
119             STRLEN ulen;
120             c = utf8n_to_uvchr(s, send - s, &ulen, 0);
121             if (c < 0x100) {
122                 if (tbl[c] >= 0)
123                     matches++;
124             } else if (complement)
125                 matches++;
126             s += ulen;
127         }
128
129     return matches;
130 }
131
132 STATIC I32
133 S_do_trans_complex(pTHX_ SV *sv)/* SPC - NOT OK */
134 {
135     U8 *s;
136     U8 *send;
137     U8 *d;
138     U8 *dstart;
139     I32 isutf8;
140     I32 matches = 0;
141     I32 grows = PL_op->op_private & OPpTRANS_GROWS;
142     I32 complement = PL_op->op_private & OPpTRANS_COMPLEMENT;
143     I32 del = PL_op->op_private & OPpTRANS_DELETE;
144     STRLEN len, rlen;
145     short *tbl;
146     I32 ch;
147
148     tbl = (short*)cPVOP->op_pv;
149     if (!tbl)
150         Perl_croak(aTHX_ "panic: do_trans_complex");
151
152     s = (U8*)SvPV(sv, len);
153     isutf8 = SvUTF8(sv);
154     send = s + len;
155
156     if (!isutf8) {
157         dstart = d = s;
158         if (PL_op->op_private & OPpTRANS_SQUASH) {
159             U8* p = send;
160             while (s < send) {
161                 if ((ch = tbl[*s]) >= 0) {
162                     *d = ch;
163                     matches++;
164                     if (p != d - 1 || *p != *d)
165                         p = d++;
166                 }
167                 else if (ch == -1)      /* -1 is unmapped character */
168                     *d++ = *s;  
169                 else if (ch == -2)      /* -2 is delete character */
170                     matches++;
171                 s++;
172             }
173         }
174         else {
175             while (s < send) {
176                 if ((ch = tbl[*s]) >= 0) {
177                     matches++;
178                     *d++ = ch;
179                 }
180                 else if (ch == -1)      /* -1 is unmapped character */
181                     *d++ = *s;
182                 else if (ch == -2)      /* -2 is delete character */
183                     matches++;
184                 s++;
185             }
186         }
187         SvCUR_set(sv, d - dstart);
188     }
189     else { /* isutf8 */
190         if (grows)
191             New(0, d, len*2+1, U8);
192         else
193             d = s;
194         dstart = d;
195         if (complement && !del)
196             rlen = tbl[0x100];
197
198 #ifdef MACOS_TRADITIONAL
199 #define comp CoMP   /* "comp" is a keyword in some compilers ... */
200 #endif
201
202         if (PL_op->op_private & OPpTRANS_SQUASH) {
203             UV pch = 0xfeedface;
204             while (s < send) {
205                 STRLEN len;
206                 UV comp = utf8_to_uvchr(s, &len);
207
208                 if (comp > 0xff) {
209                     if (!complement) {
210                         Copy(s, d, len, U8);
211                         d += len;
212                     }
213                     else {
214                         matches++;
215                         if (!del) {
216                             ch = (comp - 0x100 < rlen) ?
217                                 tbl[comp+1] : tbl[0x100+rlen];
218                             if (ch != pch) {
219                                 d = uvchr_to_utf8(d, ch);
220                                 pch = ch;
221                             }
222                             s += len;
223                             continue;
224                         }
225                     }
226                 }
227                 else if ((ch = tbl[comp]) >= 0) {
228                     matches++;
229                     if (ch != pch) {
230                         d = uvchr_to_utf8(d, ch);
231                         pch = ch;
232                     }
233                     s += len;
234                     continue;
235                 }
236                 else if (ch == -1) {    /* -1 is unmapped character */
237                     Copy(s, d, len, U8);
238                     d += len;
239                 }
240                 else if (ch == -2)      /* -2 is delete character */
241                     matches++;
242                 s += len;
243                 pch = 0xfeedface;
244             }
245         }
246         else {
247             while (s < send) {
248                 STRLEN len;
249                 UV comp = utf8_to_uvchr(s, &len);
250                 if (comp > 0xff) {
251                     if (!complement) {
252                         Copy(s, d, len, U8);
253                         d += len;
254                     }
255                     else {
256                         matches++;
257                         if (!del) {
258                             if (comp - 0x100 < rlen)
259                                 d = uvchr_to_utf8(d, tbl[comp+1]);
260                             else
261                                 d = uvchr_to_utf8(d, tbl[0x100+rlen]);
262                         }
263                     }
264                 }
265                 else if ((ch = tbl[comp]) >= 0) {
266                     d = uvchr_to_utf8(d, ch);
267                     matches++;
268                 }
269                 else if (ch == -1) {    /* -1 is unmapped character */
270                     Copy(s, d, len, U8);
271                     d += len;
272                 }
273                 else if (ch == -2)      /* -2 is delete character */
274                     matches++;
275                 s += len;
276             }
277         }
278         if (grows) {
279             sv_setpvn(sv, (char*)dstart, d - dstart);
280             Safefree(dstart);
281         }
282         else {
283             *d = '\0';
284             SvCUR_set(sv, d - dstart);
285         }
286         SvUTF8_on(sv);
287     }
288     SvSETMAGIC(sv);
289     return matches;
290 }
291
292 STATIC I32
293 S_do_trans_simple_utf8(pTHX_ SV *sv)/* SPC - OK */
294 {
295     U8 *s;
296     U8 *send;
297     U8 *d;
298     U8 *start;
299     U8 *dstart, *dend;
300     I32 matches = 0;
301     I32 grows = PL_op->op_private & OPpTRANS_GROWS;
302     STRLEN len;
303
304     SV* rv = (SV*)cSVOP->op_sv;
305     HV* hv = (HV*)SvRV(rv);
306     SV** svp = hv_fetch(hv, "NONE", 4, FALSE);
307     UV none = svp ? SvUV(*svp) : 0x7fffffff;
308     UV extra = none + 1;
309     UV final;
310     UV uv;
311     I32 isutf8;
312     U8 hibit = 0;
313
314     s = (U8*)SvPV(sv, len);
315     isutf8 = SvUTF8(sv);
316     if (!isutf8) {
317         U8 *t = s, *e = s + len;
318         while (t < e)
319             if ((hibit = UTF8_IS_CONTINUED(*t++)))
320                 break;
321         if (hibit)
322             s = bytes_to_utf8(s, &len);
323     }
324     send = s + len;
325     start = s;
326
327     svp = hv_fetch(hv, "FINAL", 5, FALSE);
328     if (svp)
329         final = SvUV(*svp);
330
331     if (grows) {
332         /* d needs to be bigger than s, in case e.g. upgrading is required */
333         New(0, d, len*3+UTF8_MAXLEN, U8);
334         dend = d + len * 3;
335         dstart = d;
336     }
337     else {
338         dstart = d = s;
339         dend = d + len;
340     }
341
342     while (s < send) {
343         if ((uv = swash_fetch(rv, s)) < none) {
344             s += UTF8SKIP(s);
345             matches++;
346             d = uvchr_to_utf8(d, uv);
347         }
348         else if (uv == none) {
349             int i = UTF8SKIP(s);
350             Copy(s, d, i, U8);
351             d += i;
352             s += i;
353         }
354         else if (uv == extra) {
355             int i = UTF8SKIP(s);
356             s += i;
357             matches++;
358             d = uvchr_to_utf8(d, final);
359         }
360         else
361             s += UTF8SKIP(s);
362
363         if (d > dend) {
364             STRLEN clen = d - dstart;
365             STRLEN nlen = dend - dstart + len + UTF8_MAXLEN;
366             if (!grows)
367                 Perl_croak(aTHX_ "panic: do_trans_complex_utf8");
368             Renew(dstart, nlen+UTF8_MAXLEN, U8);
369             d = dstart + clen;
370             dend = dstart + nlen;
371         }
372     }
373     if (grows || hibit) {
374         sv_setpvn(sv, (char*)dstart, d - dstart);
375         Safefree(dstart);
376         if (grows && hibit)
377             Safefree(start);
378     }
379     else {
380         *d = '\0';
381         SvCUR_set(sv, d - dstart);
382     }
383     SvSETMAGIC(sv);
384     SvUTF8_on(sv);
385     /* Downgrading just 'cos it will is suspect - NI-S */
386     if (!isutf8 && !(PL_hints & HINT_UTF8))
387         sv_utf8_downgrade(sv, TRUE);
388
389     return matches;
390 }
391
392 STATIC I32
393 S_do_trans_count_utf8(pTHX_ SV *sv)/* SPC - OK */
394 {
395     U8 *s;
396     U8 *start, *send;
397     I32 matches = 0;
398     STRLEN len;
399
400     SV* rv = (SV*)cSVOP->op_sv;
401     HV* hv = (HV*)SvRV(rv);
402     SV** svp = hv_fetch(hv, "NONE", 4, FALSE);
403     UV none = svp ? SvUV(*svp) : 0x7fffffff;
404     UV uv;
405     U8 hibit = 0;
406
407     s = (U8*)SvPV(sv, len);
408     if (!SvUTF8(sv)) {
409         U8 *t = s, *e = s + len;
410         while (t < e)
411             if ((hibit = !UTF8_IS_ASCII(*t++)))
412                 break;
413         if (hibit)
414             start = s = bytes_to_utf8(s, &len);
415     }
416     send = s + len;
417
418     while (s < send) {
419         if ((uv = swash_fetch(rv, s)) < none)
420             matches++;
421         s += UTF8SKIP(s);
422     }
423     if (hibit)
424         Safefree(start);
425
426     return matches;
427 }
428
429 STATIC I32
430 S_do_trans_complex_utf8(pTHX_ SV *sv) /* SPC - NOT OK */
431 {
432     U8 *s;
433     U8 *start, *send;
434     U8 *d;
435     I32 matches = 0;
436     I32 squash   = PL_op->op_private & OPpTRANS_SQUASH;
437     I32 del      = PL_op->op_private & OPpTRANS_DELETE;
438     I32 grows    = PL_op->op_private & OPpTRANS_GROWS;
439     SV* rv = (SV*)cSVOP->op_sv;
440     HV* hv = (HV*)SvRV(rv);
441     SV** svp = hv_fetch(hv, "NONE", 4, FALSE);
442     UV none = svp ? SvUV(*svp) : 0x7fffffff;
443     UV extra = none + 1;
444     UV final;
445     UV uv;
446     STRLEN len;
447     U8 *dstart, *dend;
448     I32 isutf8;
449     U8 hibit = 0;
450
451     s = (U8*)SvPV(sv, len);
452     isutf8 = SvUTF8(sv);
453     if (!isutf8) {
454         U8 *t = s, *e = s + len;
455         while (t < e)
456             if ((hibit = !UTF8_IS_ASCII(*t++)))
457                 break;
458         if (hibit)
459             s = bytes_to_utf8(s, &len);
460     }
461     send = s + len;
462     start = s;
463
464     svp = hv_fetch(hv, "FINAL", 5, FALSE);
465     if (svp)
466         final = SvUV(*svp);
467
468     if (grows) {
469         /* d needs to be bigger than s, in case e.g. upgrading is required */
470         New(0, d, len*3+UTF8_MAXLEN, U8);
471         dend = d + len * 3;
472         dstart = d;
473     }
474     else {
475         dstart = d = s;
476         dend = d + len;
477     }
478
479     if (squash) {
480         UV puv = 0xfeedface;
481         while (s < send) {
482             uv = swash_fetch(rv, s);
483         
484             if (d > dend) {
485                 STRLEN clen = d - dstart;
486                 STRLEN nlen = dend - dstart + len + UTF8_MAXLEN;
487                 if (!grows)
488                     Perl_croak(aTHX_ "panic: do_trans_complex_utf8");
489                 Renew(dstart, nlen+UTF8_MAXLEN, U8);
490                 d = dstart + clen;
491                 dend = dstart + nlen;
492             }
493             if (uv < none) {
494                 matches++;
495                 s += UTF8SKIP(s);
496                 if (uv != puv) {
497                     d = uvchr_to_utf8(d, uv);
498                     puv = uv;
499                 }
500                 continue;
501             }
502             else if (uv == none) {      /* "none" is unmapped character */
503                 int i = UTF8SKIP(s);
504                 Copy(s, d, i, U8);
505                 d += i;
506                 s += i;
507                 puv = 0xfeedface;
508                 continue;
509             }
510             else if (uv == extra && !del) {
511                 matches++;
512                 s += UTF8SKIP(s);
513                 if (uv != puv) {
514                     d = uvchr_to_utf8(d, final);
515                     puv = final;
516                 }
517                 continue;
518             }
519             matches++;                  /* "none+1" is delete character */
520             s += UTF8SKIP(s);
521         }
522     }
523     else {
524         while (s < send) {
525             uv = swash_fetch(rv, s);
526             if (d > dend) {
527                 STRLEN clen = d - dstart;
528                 STRLEN nlen = dend - dstart + len + UTF8_MAXLEN;
529                 if (!grows)
530                     Perl_croak(aTHX_ "panic: do_trans_complex_utf8");
531                 Renew(dstart, nlen+UTF8_MAXLEN, U8);
532                 d = dstart + clen;
533                 dend = dstart + nlen;
534             }
535             if (uv < none) {
536                 matches++;
537                 s += UTF8SKIP(s);
538                 d = uvchr_to_utf8(d, uv);
539                 continue;
540             }
541             else if (uv == none) {      /* "none" is unmapped character */
542                 int i = UTF8SKIP(s);
543                 Copy(s, d, i, U8);
544                 d += i;
545                 s += i;
546                 continue;
547             }
548             else if (uv == extra && !del) {
549                 matches++;
550                 s += UTF8SKIP(s);
551                 d = uvchr_to_utf8(d, final);
552                 continue;
553             }
554             matches++;                  /* "none+1" is delete character */
555             s += UTF8SKIP(s);
556         }
557     }
558     if (grows || hibit) {
559         sv_setpvn(sv, (char*)dstart, d - dstart);
560         Safefree(dstart);
561         if (grows && hibit)
562             Safefree(start);
563     }
564     else {
565         *d = '\0';
566         SvCUR_set(sv, d - dstart);
567     }
568     SvUTF8_on(sv);
569     if (!isutf8 && !(PL_hints & HINT_UTF8))
570         sv_utf8_downgrade(sv, TRUE);
571     SvSETMAGIC(sv);
572
573     return matches;
574 }
575
576 I32
577 Perl_do_trans(pTHX_ SV *sv)
578 {
579     STRLEN len;
580     I32 hasutf = (PL_op->op_private &
581                     (OPpTRANS_FROM_UTF|OPpTRANS_TO_UTF));
582
583     if (SvREADONLY(sv) && !(PL_op->op_private & OPpTRANS_IDENTICAL))
584         Perl_croak(aTHX_ PL_no_modify);
585
586     (void)SvPV(sv, len);
587     if (!len)
588         return 0;
589     if (!SvPOKp(sv))
590         (void)SvPV_force(sv, len);
591     if (!(PL_op->op_private & OPpTRANS_IDENTICAL))
592         (void)SvPOK_only_UTF8(sv);
593
594     DEBUG_t( Perl_deb(aTHX_ "2.TBL\n"));
595
596     switch (PL_op->op_private & ~hasutf & 63) {
597     case 0:
598         if (hasutf)
599             return do_trans_simple_utf8(sv);
600         else
601             return do_trans_simple(sv);
602
603     case OPpTRANS_IDENTICAL:
604         if (hasutf)
605             return do_trans_count_utf8(sv);
606         else
607             return do_trans_count(sv);
608
609     default:
610         if (hasutf)
611             return do_trans_complex_utf8(sv);
612         else
613             return do_trans_complex(sv);
614     }
615 }
616
617 void
618 Perl_do_join(pTHX_ register SV *sv, SV *del, register SV **mark, register SV **sp)
619 {
620     SV **oldmark = mark;
621     register I32 items = sp - mark;
622     register STRLEN len;
623     STRLEN delimlen;
624     register char *delim = SvPV(del, delimlen);
625     STRLEN tmplen;
626
627     mark++;
628     len = (items > 0 ? (delimlen * (items - 1) ) : 0);
629     (void)SvUPGRADE(sv, SVt_PV);
630     if (SvLEN(sv) < len + items) {      /* current length is way too short */
631         while (items-- > 0) {
632             if (*mark && !SvGAMAGIC(*mark) && SvOK(*mark)) {
633                 SvPV(*mark, tmplen);
634                 len += tmplen;
635             }
636             mark++;
637         }
638         SvGROW(sv, len + 1);            /* so try to pre-extend */
639
640         mark = oldmark;
641         items = sp - mark;
642         ++mark;
643     }
644
645     if (items-- > 0) {
646         sv_setpv(sv, "");
647         if (*mark)
648             sv_catsv(sv, *mark);
649         mark++;
650     }
651     else
652         sv_setpv(sv,"");
653     if (delimlen) {
654         for (; items > 0; items--,mark++) {
655             sv_catsv(sv,del);
656             sv_catsv(sv,*mark);
657         }
658     }
659     else {
660         for (; items > 0; items--,mark++)
661             sv_catsv(sv,*mark);
662     }
663     SvSETMAGIC(sv);
664 }
665
666 void
667 Perl_do_sprintf(pTHX_ SV *sv, I32 len, SV **sarg)
668 {
669     STRLEN patlen;
670     char *pat = SvPV(*sarg, patlen);
671     bool do_taint = FALSE;
672
673     sv_vsetpvfn(sv, pat, patlen, Null(va_list*), sarg + 1, len - 1, &do_taint);
674     SvSETMAGIC(sv);
675     if (do_taint)
676         SvTAINTED_on(sv);
677 }
678
679 /* currently converts input to bytes if possible, but doesn't sweat failure */
680 UV
681 Perl_do_vecget(pTHX_ SV *sv, I32 offset, I32 size)
682 {
683     STRLEN srclen, len;
684     unsigned char *s = (unsigned char *) SvPV(sv, srclen);
685     UV retnum = 0;
686
687     if (offset < 0)
688         return retnum;
689     if (size < 1 || (size & (size-1))) /* size < 1 or not a power of two */
690         Perl_croak(aTHX_ "Illegal number of bits in vec");
691
692     if (SvUTF8(sv))
693         (void) Perl_sv_utf8_downgrade(aTHX_ sv, TRUE);
694
695     offset *= size;     /* turn into bit offset */
696     len = (offset + size + 7) / 8;      /* required number of bytes */
697     if (len > srclen) {
698         if (size <= 8)
699             retnum = 0;
700         else {
701             offset >>= 3;       /* turn into byte offset */
702             if (size == 16) {
703                 if (offset >= srclen)
704                     retnum = 0;
705                 else
706                     retnum = (UV) s[offset] <<  8;
707             }
708             else if (size == 32) {
709                 if (offset >= srclen)
710                     retnum = 0;
711                 else if (offset + 1 >= srclen)
712                     retnum =
713                         ((UV) s[offset    ] << 24);
714                 else if (offset + 2 >= srclen)
715                     retnum =
716                         ((UV) s[offset    ] << 24) +
717                         ((UV) s[offset + 1] << 16);
718                 else
719                     retnum =
720                         ((UV) s[offset    ] << 24) +
721                         ((UV) s[offset + 1] << 16) +
722                         (     s[offset + 2] <<  8);
723             }
724 #ifdef UV_IS_QUAD
725             else if (size == 64) {
726                 if (ckWARN(WARN_PORTABLE))
727                     Perl_warner(aTHX_ WARN_PORTABLE,
728                                 "Bit vector size > 32 non-portable");
729                 if (offset >= srclen)
730                     retnum = 0;
731                 else if (offset + 1 >= srclen)
732                     retnum =
733                         (UV) s[offset     ] << 56;
734                 else if (offset + 2 >= srclen)
735                     retnum =
736                         ((UV) s[offset    ] << 56) +
737                         ((UV) s[offset + 1] << 48);
738                 else if (offset + 3 >= srclen)
739                     retnum =
740                         ((UV) s[offset    ] << 56) +
741                         ((UV) s[offset + 1] << 48) +
742                         ((UV) s[offset + 2] << 40);
743                 else if (offset + 4 >= srclen)
744                     retnum =
745                         ((UV) s[offset    ] << 56) +
746                         ((UV) s[offset + 1] << 48) +
747                         ((UV) s[offset + 2] << 40) +
748                         ((UV) s[offset + 3] << 32);
749                 else if (offset + 5 >= srclen)
750                     retnum =
751                         ((UV) s[offset    ] << 56) +
752                         ((UV) s[offset + 1] << 48) +
753                         ((UV) s[offset + 2] << 40) +
754                         ((UV) s[offset + 3] << 32) +
755                         (     s[offset + 4] << 24);
756                 else if (offset + 6 >= srclen)
757                     retnum =
758                         ((UV) s[offset    ] << 56) +
759                         ((UV) s[offset + 1] << 48) +
760                         ((UV) s[offset + 2] << 40) +
761                         ((UV) s[offset + 3] << 32) +
762                         ((UV) s[offset + 4] << 24) +
763                         ((UV) s[offset + 5] << 16);
764                 else
765                     retnum =
766                         ((UV) s[offset    ] << 56) +
767                         ((UV) s[offset + 1] << 48) +
768                         ((UV) s[offset + 2] << 40) +
769                         ((UV) s[offset + 3] << 32) +
770                         ((UV) s[offset + 4] << 24) +
771                         ((UV) s[offset + 5] << 16) +
772                         (     s[offset + 6] <<  8);
773             }
774 #endif
775         }
776     }
777     else if (size < 8)
778         retnum = (s[offset >> 3] >> (offset & 7)) & ((1 << size) - 1);
779     else {
780         offset >>= 3;   /* turn into byte offset */
781         if (size == 8)
782             retnum = s[offset];
783         else if (size == 16)
784             retnum =
785                 ((UV) s[offset] <<      8) +
786                       s[offset + 1];
787         else if (size == 32)
788             retnum =
789                 ((UV) s[offset    ] << 24) +
790                 ((UV) s[offset + 1] << 16) +
791                 (     s[offset + 2] <<  8) +
792                       s[offset + 3];
793 #ifdef UV_IS_QUAD
794         else if (size == 64) {
795             if (ckWARN(WARN_PORTABLE))
796                 Perl_warner(aTHX_ WARN_PORTABLE,
797                             "Bit vector size > 32 non-portable");
798             retnum =
799                 ((UV) s[offset    ] << 56) +
800                 ((UV) s[offset + 1] << 48) +
801                 ((UV) s[offset + 2] << 40) +
802                 ((UV) s[offset + 3] << 32) +
803                 ((UV) s[offset + 4] << 24) +
804                 ((UV) s[offset + 5] << 16) +
805                 (     s[offset + 6] <<  8) +
806                       s[offset + 7];
807         }
808 #endif
809     }
810
811     return retnum;
812 }
813
814 /* currently converts input to bytes if possible but doesn't sweat failures,
815  * although it does ensure that the string it clobbers is not marked as
816  * utf8-valid any more
817  */
818 void
819 Perl_do_vecset(pTHX_ SV *sv)
820 {
821     SV *targ = LvTARG(sv);
822     register I32 offset;
823     register I32 size;
824     register unsigned char *s;
825     register UV lval;
826     I32 mask;
827     STRLEN targlen;
828     STRLEN len;
829
830     if (!targ)
831         return;
832     s = (unsigned char*)SvPV_force(targ, targlen);
833     if (SvUTF8(targ)) {
834         /* This is handled by the SvPOK_only below...
835         if (!Perl_sv_utf8_downgrade(aTHX_ targ, TRUE))
836             SvUTF8_off(targ);
837          */
838         (void) Perl_sv_utf8_downgrade(aTHX_ targ, TRUE);
839     }
840
841     (void)SvPOK_only(targ);
842     lval = SvUV(sv);
843     offset = LvTARGOFF(sv);
844     if (offset < 0)
845         Perl_croak(aTHX_ "Assigning to negative offset in vec");
846     size = LvTARGLEN(sv);
847     if (size < 1 || (size & (size-1))) /* size < 1 or not a power of two */
848         Perl_croak(aTHX_ "Illegal number of bits in vec");
849
850     offset *= size;                     /* turn into bit offset */
851     len = (offset + size + 7) / 8;      /* required number of bytes */
852     if (len > targlen) {
853         s = (unsigned char*)SvGROW(targ, len + 1);
854         (void)memzero((char *)(s + targlen), len - targlen + 1);
855         SvCUR_set(targ, len);
856     }
857
858     if (size < 8) {
859         mask = (1 << size) - 1;
860         size = offset & 7;
861         lval &= mask;
862         offset >>= 3;                   /* turn into byte offset */
863         s[offset] &= ~(mask << size);
864         s[offset] |= lval << size;
865     }
866     else {
867         offset >>= 3;                   /* turn into byte offset */
868         if (size == 8)
869             s[offset  ] = lval         & 0xff;
870         else if (size == 16) {
871             s[offset  ] = (lval >>  8) & 0xff;
872             s[offset+1] = lval         & 0xff;
873         }
874         else if (size == 32) {
875             s[offset  ] = (lval >> 24) & 0xff;
876             s[offset+1] = (lval >> 16) & 0xff;
877             s[offset+2] = (lval >>  8) & 0xff;
878             s[offset+3] =  lval        & 0xff;
879         }
880 #ifdef UV_IS_QUAD
881         else if (size == 64) {
882             if (ckWARN(WARN_PORTABLE))
883                 Perl_warner(aTHX_ WARN_PORTABLE,
884                             "Bit vector size > 32 non-portable");
885             s[offset  ] = (lval >> 56) & 0xff;
886             s[offset+1] = (lval >> 48) & 0xff;
887             s[offset+2] = (lval >> 40) & 0xff;
888             s[offset+3] = (lval >> 32) & 0xff;
889             s[offset+4] = (lval >> 24) & 0xff;
890             s[offset+5] = (lval >> 16) & 0xff;
891             s[offset+6] = (lval >>  8) & 0xff;
892             s[offset+7] =  lval        & 0xff;
893         }
894 #endif
895     }
896     SvSETMAGIC(targ);
897 }
898
899 void
900 Perl_do_chop(pTHX_ register SV *astr, register SV *sv)
901 {
902     STRLEN len;
903     char *s;
904
905     if (SvTYPE(sv) == SVt_PVAV) {
906         register I32 i;
907         I32 max;
908         AV* av = (AV*)sv;
909         max = AvFILL(av);
910         for (i = 0; i <= max; i++) {
911             sv = (SV*)av_fetch(av, i, FALSE);
912             if (sv && ((sv = *(SV**)sv), sv != &PL_sv_undef))
913                 do_chop(astr, sv);
914         }
915         return;
916     }
917     else if (SvTYPE(sv) == SVt_PVHV) {
918         HV* hv = (HV*)sv;
919         HE* entry;
920         (void)hv_iterinit(hv);
921         /*SUPPRESS 560*/
922         while ((entry = hv_iternext(hv)))
923             do_chop(astr,hv_iterval(hv,entry));
924         return;
925     }
926     else if (SvREADONLY(sv))
927         Perl_croak(aTHX_ PL_no_modify);
928     s = SvPV(sv, len);
929     if (len && !SvPOK(sv))
930         s = SvPV_force(sv, len);
931     if (DO_UTF8(sv)) {
932         if (s && len) {
933             char *send = s + len;
934             char *start = s;
935             s = send - 1;
936             while (s > start && UTF8_IS_CONTINUATION(*s))
937                 s--;
938             if (utf8_to_uvchr((U8*)s, 0)) {
939                 sv_setpvn(astr, s, send - s);
940                 *s = '\0';
941                 SvCUR_set(sv, s - start);
942                 SvNIOK_off(sv);
943                 SvUTF8_on(astr);
944             }
945         }
946         else
947             sv_setpvn(astr, "", 0);
948     }
949     else if (s && len) {
950         s += --len;
951         sv_setpvn(astr, s, 1);
952         *s = '\0';
953         SvCUR_set(sv, len);
954         SvUTF8_off(sv);
955         SvNIOK_off(sv);
956     }
957     else
958         sv_setpvn(astr, "", 0);
959     SvSETMAGIC(sv);
960 }
961
962 I32
963 Perl_do_chomp(pTHX_ register SV *sv)
964 {
965     register I32 count;
966     STRLEN len;
967     char *s;
968
969     if (RsSNARF(PL_rs))
970         return 0;
971     if (RsRECORD(PL_rs))
972       return 0;
973     count = 0;
974     if (SvTYPE(sv) == SVt_PVAV) {
975         register I32 i;
976         I32 max;
977         AV* av = (AV*)sv;
978         max = AvFILL(av);
979         for (i = 0; i <= max; i++) {
980             sv = (SV*)av_fetch(av, i, FALSE);
981             if (sv && ((sv = *(SV**)sv), sv != &PL_sv_undef))
982                 count += do_chomp(sv);
983         }
984         return count;
985     }
986     else if (SvTYPE(sv) == SVt_PVHV) {
987         HV* hv = (HV*)sv;
988         HE* entry;
989         (void)hv_iterinit(hv);
990         /*SUPPRESS 560*/
991         while ((entry = hv_iternext(hv)))
992             count += do_chomp(hv_iterval(hv,entry));
993         return count;
994     }
995     else if (SvREADONLY(sv))
996         Perl_croak(aTHX_ PL_no_modify);
997     s = SvPV(sv, len);
998     if (len && !SvPOKp(sv))
999         s = SvPV_force(sv, len);
1000     if (s && len) {
1001         s += --len;
1002         if (RsPARA(PL_rs)) {
1003             if (*s != '\n')
1004                 goto nope;
1005             ++count;
1006             while (len && s[-1] == '\n') {
1007                 --len;
1008                 --s;
1009                 ++count;
1010             }
1011         }
1012         else {
1013             STRLEN rslen;
1014             char *rsptr = SvPV(PL_rs, rslen);
1015             if (rslen == 1) {
1016                 if (*s != *rsptr)
1017                     goto nope;
1018                 ++count;
1019             }
1020             else {
1021                 if (len < rslen - 1)
1022                     goto nope;
1023                 len -= rslen - 1;
1024                 s -= rslen - 1;
1025                 if (memNE(s, rsptr, rslen))
1026                     goto nope;
1027                 count += rslen;
1028             }
1029         }
1030         *s = '\0';
1031         SvCUR_set(sv, len);
1032         SvNIOK_off(sv);
1033     }
1034   nope:
1035     SvSETMAGIC(sv);
1036     return count;
1037 }
1038
1039 void
1040 Perl_do_vop(pTHX_ I32 optype, SV *sv, SV *left, SV *right)
1041 {
1042 #ifdef LIBERAL
1043     register long *dl;
1044     register long *ll;
1045     register long *rl;
1046 #endif
1047     register char *dc;
1048     STRLEN leftlen;
1049     STRLEN rightlen;
1050     register char *lc;
1051     register char *rc;
1052     register I32 len;
1053     I32 lensave;
1054     char *lsave;
1055     char *rsave;
1056     bool left_utf = DO_UTF8(left);
1057     bool right_utf = DO_UTF8(right);
1058     I32 needlen;
1059
1060     if (left_utf && !right_utf)
1061         sv_utf8_upgrade(right);
1062     else if (!left_utf && right_utf)
1063         sv_utf8_upgrade(left);
1064
1065     if (sv != left || (optype != OP_BIT_AND && !SvOK(sv) && !SvGMAGICAL(sv)))
1066         sv_setpvn(sv, "", 0);   /* avoid undef warning on |= and ^= */
1067     lsave = lc = SvPV(left, leftlen);
1068     rsave = rc = SvPV(right, rightlen);
1069     len = leftlen < rightlen ? leftlen : rightlen;
1070     lensave = len;
1071     if ((left_utf || right_utf) && (sv == left || sv == right)) {
1072         needlen = optype == OP_BIT_AND ? len : leftlen + rightlen;
1073         Newz(801, dc, needlen + 1, char);
1074     }
1075     else if (SvOK(sv) || SvTYPE(sv) > SVt_PVMG) {
1076         STRLEN n_a;
1077         dc = SvPV_force(sv, n_a);
1078         if (SvCUR(sv) < len) {
1079             dc = SvGROW(sv, len + 1);
1080             (void)memzero(dc + SvCUR(sv), len - SvCUR(sv) + 1);
1081         }
1082         if (optype != OP_BIT_AND && (left_utf || right_utf))
1083             dc = SvGROW(sv, leftlen + rightlen + 1);
1084     }
1085     else {
1086         needlen = ((optype == OP_BIT_AND)
1087                     ? len : (leftlen > rightlen ? leftlen : rightlen));
1088         Newz(801, dc, needlen + 1, char);
1089         (void)sv_usepvn(sv, dc, needlen);
1090         dc = SvPVX(sv);         /* sv_usepvn() calls Renew() */
1091     }
1092     SvCUR_set(sv, len);
1093     (void)SvPOK_only(sv);
1094     if (left_utf || right_utf) {
1095         UV duc, luc, ruc;
1096         char *dcsave = dc;
1097         STRLEN lulen = leftlen;
1098         STRLEN rulen = rightlen;
1099         STRLEN ulen;
1100
1101         switch (optype) {
1102         case OP_BIT_AND:
1103             while (lulen && rulen) {
1104                 luc = utf8n_to_uvchr((U8*)lc, lulen, &ulen, UTF8_ALLOW_ANYUV);
1105                 lc += ulen;
1106                 lulen -= ulen;
1107                 ruc = utf8n_to_uvchr((U8*)rc, rulen, &ulen, UTF8_ALLOW_ANYUV);
1108                 rc += ulen;
1109                 rulen -= ulen;
1110                 duc = luc & ruc;
1111                 dc = (char*)uvchr_to_utf8((U8*)dc, duc);
1112             }
1113             if (sv == left || sv == right)
1114                 (void)sv_usepvn(sv, dcsave, needlen);
1115             SvCUR_set(sv, dc - dcsave);
1116             break;
1117         case OP_BIT_XOR:
1118             while (lulen && rulen) {
1119                 luc = utf8n_to_uvchr((U8*)lc, lulen, &ulen, UTF8_ALLOW_ANYUV);
1120                 lc += ulen;
1121                 lulen -= ulen;
1122                 ruc = utf8n_to_uvchr((U8*)rc, rulen, &ulen, UTF8_ALLOW_ANYUV);
1123                 rc += ulen;
1124                 rulen -= ulen;
1125                 duc = luc ^ ruc;
1126                 dc = (char*)uvchr_to_utf8((U8*)dc, duc);
1127             }
1128             goto mop_up_utf;
1129         case OP_BIT_OR:
1130             while (lulen && rulen) {
1131                 luc = utf8n_to_uvchr((U8*)lc, lulen, &ulen, UTF8_ALLOW_ANYUV);
1132                 lc += ulen;
1133                 lulen -= ulen;
1134                 ruc = utf8n_to_uvchr((U8*)rc, rulen, &ulen, UTF8_ALLOW_ANYUV);
1135                 rc += ulen;
1136                 rulen -= ulen;
1137                 duc = luc | ruc;
1138                 dc = (char*)uvchr_to_utf8((U8*)dc, duc);
1139             }
1140           mop_up_utf:
1141             if (sv == left || sv == right)
1142                 (void)sv_usepvn(sv, dcsave, needlen);
1143             SvCUR_set(sv, dc - dcsave);
1144             if (rulen)
1145                 sv_catpvn(sv, rc, rulen);
1146             else if (lulen)
1147                 sv_catpvn(sv, lc, lulen);
1148             else
1149                 *SvEND(sv) = '\0';
1150             break;
1151         }
1152         SvUTF8_on(sv);
1153         goto finish;
1154     }
1155     else
1156 #ifdef LIBERAL
1157     if (len >= sizeof(long)*4 &&
1158         !((long)dc % sizeof(long)) &&
1159         !((long)lc % sizeof(long)) &&
1160         !((long)rc % sizeof(long)))     /* It's almost always aligned... */
1161     {
1162         I32 remainder = len % (sizeof(long)*4);
1163         len /= (sizeof(long)*4);
1164
1165         dl = (long*)dc;
1166         ll = (long*)lc;
1167         rl = (long*)rc;
1168
1169         switch (optype) {
1170         case OP_BIT_AND:
1171             while (len--) {
1172                 *dl++ = *ll++ & *rl++;
1173                 *dl++ = *ll++ & *rl++;
1174                 *dl++ = *ll++ & *rl++;
1175                 *dl++ = *ll++ & *rl++;
1176             }
1177             break;
1178         case OP_BIT_XOR:
1179             while (len--) {
1180                 *dl++ = *ll++ ^ *rl++;
1181                 *dl++ = *ll++ ^ *rl++;
1182                 *dl++ = *ll++ ^ *rl++;
1183                 *dl++ = *ll++ ^ *rl++;
1184             }
1185             break;
1186         case OP_BIT_OR:
1187             while (len--) {
1188                 *dl++ = *ll++ | *rl++;
1189                 *dl++ = *ll++ | *rl++;
1190                 *dl++ = *ll++ | *rl++;
1191                 *dl++ = *ll++ | *rl++;
1192             }
1193         }
1194
1195         dc = (char*)dl;
1196         lc = (char*)ll;
1197         rc = (char*)rl;
1198
1199         len = remainder;
1200     }
1201 #endif
1202     {
1203         switch (optype) {
1204         case OP_BIT_AND:
1205             while (len--)
1206                 *dc++ = *lc++ & *rc++;
1207             break;
1208         case OP_BIT_XOR:
1209             while (len--)
1210                 *dc++ = *lc++ ^ *rc++;
1211             goto mop_up;
1212         case OP_BIT_OR:
1213             while (len--)
1214                 *dc++ = *lc++ | *rc++;
1215           mop_up:
1216             len = lensave;
1217             if (rightlen > len)
1218                 sv_catpvn(sv, rsave + len, rightlen - len);
1219             else if (leftlen > len)
1220                 sv_catpvn(sv, lsave + len, leftlen - len);
1221             else
1222                 *SvEND(sv) = '\0';
1223             break;
1224         }
1225     }
1226 finish:
1227     SvTAINT(sv);
1228 }
1229
1230 OP *
1231 Perl_do_kv(pTHX)
1232 {
1233     dSP;
1234     HV *hv = (HV*)POPs;
1235     HV *keys;
1236     register HE *entry;
1237     SV *tmpstr;
1238     I32 gimme = GIMME_V;
1239     I32 dokeys =   (PL_op->op_type == OP_KEYS);
1240     I32 dovalues = (PL_op->op_type == OP_VALUES);
1241     I32 realhv = (SvTYPE(hv) == SVt_PVHV);
1242
1243     if (PL_op->op_type == OP_RV2HV || PL_op->op_type == OP_PADHV)
1244         dokeys = dovalues = TRUE;
1245
1246     if (!hv) {
1247         if (PL_op->op_flags & OPf_MOD || LVRET) {       /* lvalue */
1248             dTARGET;            /* make sure to clear its target here */
1249             if (SvTYPE(TARG) == SVt_PVLV)
1250                 LvTARG(TARG) = Nullsv;
1251             PUSHs(TARG);
1252         }
1253         RETURN;
1254     }
1255
1256     keys = realhv ? hv : avhv_keys((AV*)hv);
1257     (void)hv_iterinit(keys);    /* always reset iterator regardless */
1258
1259     if (gimme == G_VOID)
1260         RETURN;
1261
1262     if (gimme == G_SCALAR) {
1263         IV i;
1264         dTARGET;
1265
1266         if (PL_op->op_flags & OPf_MOD || LVRET) {       /* lvalue */
1267             if (SvTYPE(TARG) < SVt_PVLV) {
1268                 sv_upgrade(TARG, SVt_PVLV);
1269                 sv_magic(TARG, Nullsv, 'k', Nullch, 0);
1270             }
1271             LvTYPE(TARG) = 'k';
1272             if (LvTARG(TARG) != (SV*)keys) {
1273                 if (LvTARG(TARG))
1274                     SvREFCNT_dec(LvTARG(TARG));
1275                 LvTARG(TARG) = SvREFCNT_inc(keys);
1276             }
1277             PUSHs(TARG);
1278             RETURN;
1279         }
1280
1281         if (! SvTIED_mg((SV*)keys, 'P'))
1282             i = HvKEYS(keys);
1283         else {
1284             i = 0;
1285             /*SUPPRESS 560*/
1286             while (hv_iternext(keys)) i++;
1287         }
1288         PUSHi( i );
1289         RETURN;
1290     }
1291
1292     EXTEND(SP, HvKEYS(keys) * (dokeys + dovalues));
1293
1294     PUTBACK;    /* hv_iternext and hv_iterval might clobber stack_sp */
1295     while ((entry = hv_iternext(keys))) {
1296         SPAGAIN;
1297         if (dokeys)
1298             XPUSHs(hv_iterkeysv(entry));        /* won't clobber stack_sp */
1299         if (dovalues) {
1300             PUTBACK;
1301             tmpstr = realhv ?
1302                      hv_iterval(hv,entry) : avhv_iterval((AV*)hv,entry);
1303             DEBUG_H(Perl_sv_setpvf(aTHX_ tmpstr, "%lu%%%d=%lu",
1304                             (unsigned long)HeHASH(entry),
1305                             HvMAX(keys)+1,
1306                             (unsigned long)(HeHASH(entry) & HvMAX(keys))));
1307             SPAGAIN;
1308             XPUSHs(tmpstr);
1309         }
1310         PUTBACK;
1311     }
1312     return NORMAL;
1313 }
1314