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