add trailing newline to file
[p5sagit/p5-mst-13.2.git] / doop.c
CommitLineData
a0d0e21e 1/* doop.c
79072805 2 *
9607fc9c 3 * Copyright (c) 1991-1997, Larry Wall
79072805 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 *
a0d0e21e 8 */
9
10/*
11 * "'So that was the job I felt I had to do when I started,' thought Sam."
79072805 12 */
13
14#include "EXTERN.h"
15#include "perl.h"
16
17#if !defined(NSIG) || defined(M_UNIX) || defined(M_XENIX)
18#include <signal.h>
19#endif
20
79072805 21I32
8ac85365 22do_trans(SV *sv, OP *arg)
79072805 23{
11343788 24 dTHR;
1476d272 25 register U8 *s;
26 register U8 *send;
27 register U8 *d;
1476d272 28 register I32 matches = 0;
533c011a 29 register I32 squash = PL_op->op_private & OPpTRANS_SQUASH;
463ee0b2 30 STRLEN len;
79072805 31
533c011a 32 if (SvREADONLY(sv) && !(PL_op->op_private & OPpTRANS_COUNTONLY))
a0d0e21e 33 croak(no_modify);
a0ed51b3 34
35 if (PL_op->op_private & (OPpTRANS_FROM_UTF|OPpTRANS_TO_UTF)) {
36 SV* rv = (SV*)cSVOP->op_sv;
37 HV* hv = (HV*)SvRV(rv);
38 SV** svp = hv_fetch(hv, "NONE", 4, FALSE);
39 UV none = svp ? SvUV(*svp) : 0x7fffffff;
40 UV extra = none + 1;
41 I32 del = PL_op->op_private & OPpTRANS_DELETE;
42 UV final;
43 register UV uv;
44 UV puv;
45 char *dst;
46 register I32 from_utf = PL_op->op_private & OPpTRANS_FROM_UTF;
47 register I32 to_utf = PL_op->op_private & OPpTRANS_TO_UTF;
48
49 s = (U8*)SvPV(sv, len);
50 if (!len)
51 return 0;
52 if (!SvPOKp(sv))
53 s = (U8*)SvPV_force(sv, len);
54 (void)SvPOK_only(sv);
55 send = s + len;
56 DEBUG_t( deb("2.TBL\n"));
57 if (PL_op->op_private == (OPpTRANS_FROM_UTF|OPpTRANS_TO_UTF)) { /* no other flags */
58 svp = hv_fetch(hv, "FINAL", 5, FALSE);
59 if (svp)
60 final = SvUV(*svp);
61
62 d = s;
63 while (s < send) {
64 if ((uv = swash_fetch(rv, s)) < none) {
65 s += UTF8SKIP(s);
66 matches++;
67 d = uv_to_utf8(d, uv);
68 }
69 else if (uv == none) {
70 int i;
71 for (i = UTF8SKIP(s); i; i--)
72 *d++ = *s++;
73 }
74 else if (uv == extra) {
75 s += UTF8SKIP(s);
76 matches++;
77 d = uv_to_utf8(d, final);
78 }
79 else
80 s += UTF8SKIP(s);
79072805 81 }
a0ed51b3 82 *d = '\0';
83 SvCUR_set(sv, d - (U8*)SvPVX(sv));
84 SvSETMAGIC(sv);
79072805 85 }
a0ed51b3 86 else if (PL_op->op_private == OPpTRANS_FROM_UTF) { /* no other flags */
87 svp = hv_fetch(hv, "FINAL", 5, FALSE);
88 if (svp)
89 final = SvUV(*svp);
90
91 d = s;
92 while (s < send) {
93 if ((uv = swash_fetch(rv, s)) < none) {
94 s += UTF8SKIP(s);
95 matches++;
96 *d++ = (U8)uv;
97 }
98 else if (uv == none) {
99 I32 ulen;
100 uv = utf8_to_uv(s, &ulen);
101 s += ulen;
102 *d++ = (U8)uv;
103 }
104 else if (uv == extra) {
105 s += UTF8SKIP(s);
106 matches++;
107 *d++ = (U8)final;
108 }
109 else
110 s += UTF8SKIP(s);
111 }
112 *d = '\0';
113 SvCUR_set(sv, d - (U8*)SvPVX(sv));
114 SvSETMAGIC(sv);
115 }
116 else if (PL_op->op_private == OPpTRANS_TO_UTF) { /* no other flags */
117 svp = hv_fetch(hv, "FINAL", 5, FALSE);
118 if (svp)
119 final = SvUV(*svp);
120
121 d = s;
122 while (s < send) {
123 U8 tmpbuf[10];
124 uv_to_utf8(tmpbuf, *s); /* XXX suboptimal */
125 if ((uv = swash_fetch(rv, tmpbuf)) < none) {
126 s += UTF8SKIP(s);
127 matches++;
128 d = uv_to_utf8(d, uv);
129 }
130 else if (uv == none) {
131 I32 ulen;
132 uv = utf8_to_uv(s, &ulen);
133 s += ulen;
134 d = uv_to_utf8(d, uv);
135 }
136 else if (uv == extra) {
137 s += UTF8SKIP(s);
138 matches++;
139 d = uv_to_utf8(d, final);
140 }
141 else
142 s += UTF8SKIP(s);
143 }
144 *d = '\0';
145 SvCUR_set(sv, d - (U8*)SvPVX(sv));
146 SvSETMAGIC(sv);
5d06d08e 147 }
a0ed51b3 148 else if (PL_op->op_private & OPpTRANS_COUNTONLY) {
149 if (from_utf) {
150 while (s < send) {
151 if (swash_fetch(rv, s) < none)
152 matches++;
153 s += UTF8SKIP(s);
154 }
155 }
156 else {
157 while (s < send) {
158 char tmpbuf[10];
159 uv_to_utf8(tmpbuf, *s); /* XXX suboptimal */
160 if (swash_fetch(rv, tmpbuf) < none)
161 matches++;
162 s += UTF8SKIP(s);
163 }
164 }
165 }
166 else {
167 I32 bits = 16;
168 U8 *dst;
169
170 svp = hv_fetch(hv, "BITS", 4, FALSE);
171 if (svp)
172 bits = (I32)SvIV(*svp);
173
174 svp = hv_fetch(hv, "FINAL", 5, FALSE);
175 if (svp)
176 final = SvUV(*svp);
177
178 Newz(801, d, len * (bits >> 3) + 1, char);
179 dst = d;
180
181 puv = 0xfeedface;
182 if (squash) {
183 while (s < send) {
184 if (from_utf)
185 uv = swash_fetch(rv, s);
186 else {
187 char tmpbuf[10];
188 uv_to_utf8(tmpbuf, *s); /* XXX suboptimal */
189 uv = swash_fetch(rv, tmpbuf);
190 }
191 if (uv < none) {
192 matches++;
193 if (uv != puv) {
194 if (to_utf)
195 d = uv_to_utf8(d, uv);
196 else
197 *d++ = (U8)uv;
198 }
199 puv = uv;
200 s += UTF8SKIP(s);
201 continue;
202 }
203 else if (uv == none) { /* "none" is unmapped character */
204 int i;
205 if (to_utf) {
206 for (i = UTF8SKIP(s); i; --i)
207 *d++ = *s++;
208 }
209 else {
210 I32 ulen;
211 *d++ = (U8)utf8_to_uv(s, &ulen);
212 s += ulen;
213 }
214 puv = 0xfeedface;
215 continue;
216 }
217 else if (uv == extra && !del) {
218 matches++;
219 if (to_utf)
220 d = uv_to_utf8(d, final);
221 else
222 *d++ = (U8)final;
223 s += UTF8SKIP(s);
224 puv = 0xfeedface;
225 continue;
226 }
227 matches++; /* "none+1" is delete character */
228 s += UTF8SKIP(s);
229 }
230 }
231 else {
232 while (s < send) {
233 if (from_utf)
234 uv = swash_fetch(rv, s);
235 else {
236 char tmpbuf[10];
237 uv_to_utf8(tmpbuf, *s); /* XXX suboptimal */
238 uv = swash_fetch(rv, tmpbuf);
239 }
240 if (uv < none) {
241 if (to_utf)
242 d = uv_to_utf8(d, uv);
243 else
244 *d++ = (U8)uv;
245 matches++;
246 s += UTF8SKIP(s);
247 continue;
248 }
249 else if (uv == none) { /* "none" is unmapped character */
250 int i;
251 if (to_utf) {
252 for (i = UTF8SKIP(s); i; --i)
253 *d++ = *s++;
254 }
255 else {
256 I32 ulen;
257 *d++ = (U8)utf8_to_uv(s, &ulen);
258 s += ulen;
259 }
260 continue;
261 }
262 else if (uv == extra && !del) {
263 matches++;
264 if (to_utf)
265 d = uv_to_utf8(d, final);
266 else
267 *d++ = (U8)final;
268 s += UTF8SKIP(s);
269 continue;
270 }
271 matches++; /* "none+1" is delete character */
272 s += UTF8SKIP(s);
273 }
274 }
275 sv_usepvn_mg(sv, dst, d - dst);
276 }
277 return matches;
79072805 278 }
279 else {
a0ed51b3 280 register short *tbl;
281 register I32 ch;
282 register U8 *p;
283
284 tbl = (short*)cPVOP->op_pv;
285 s = (U8*)SvPV(sv, len);
286 if (!len)
287 return 0;
288 if (!SvPOKp(sv))
289 s = (U8*)SvPV_force(sv, len);
290 (void)SvPOK_only(sv);
291 send = s + len;
292 if (!tbl || !s)
293 croak("panic: do_trans");
294 DEBUG_t( deb("2.TBL\n"));
295 if (!PL_op->op_private) {
296 while (s < send) {
297 if ((ch = tbl[*s]) >= 0) {
298 matches++;
299 *s = ch;
300 }
301 s++;
302 }
303 SvSETMAGIC(sv);
304 }
305 else if (PL_op->op_private & OPpTRANS_COUNTONLY) {
306 while (s < send) {
307 if (tbl[*s] >= 0)
308 matches++;
309 s++;
310 }
311 }
312 else {
313 d = s;
314 p = send;
315 while (s < send) {
316 if ((ch = tbl[*s]) >= 0) {
317 *d = ch;
318 matches++;
319 if (squash) {
320 if (p == d - 1 && *p == *d)
321 matches--;
322 else
323 p = d++;
324 }
79072805 325 else
a0ed51b3 326 d++;
79072805 327 }
a0ed51b3 328 else if (ch == -1) /* -1 is unmapped character */
329 *d++ = *s; /* -2 is delete character */
330 s++;
79072805 331 }
a0ed51b3 332 matches += send - d; /* account for disappeared chars */
333 *d = '\0';
334 SvCUR_set(sv, d - (U8*)SvPVX(sv));
335 SvSETMAGIC(sv);
79072805 336 }
a0ed51b3 337 return matches;
79072805 338 }
79072805 339}
340
341void
8ac85365 342do_join(register SV *sv, SV *del, register SV **mark, register SV **sp)
79072805 343{
344 SV **oldmark = mark;
345 register I32 items = sp - mark;
79072805 346 register STRLEN len;
463ee0b2 347 STRLEN delimlen;
348 register char *delim = SvPV(del, delimlen);
349 STRLEN tmplen;
79072805 350
351 mark++;
352 len = (items > 0 ? (delimlen * (items - 1) ) : 0);
353 if (SvTYPE(sv) < SVt_PV)
354 sv_upgrade(sv, SVt_PV);
355 if (SvLEN(sv) < len + items) { /* current length is way too short */
356 while (items-- > 0) {
48c036b1 357 if (*mark && !SvGMAGICAL(*mark) && SvOK(*mark)) {
463ee0b2 358 SvPV(*mark, tmplen);
359 len += tmplen;
79072805 360 }
361 mark++;
362 }
363 SvGROW(sv, len + 1); /* so try to pre-extend */
364
365 mark = oldmark;
366 items = sp - mark;;
367 ++mark;
368 }
369
463ee0b2 370 if (items-- > 0) {
8990e307 371 char *s;
372
373 if (*mark) {
374 s = SvPV(*mark, tmplen);
375 sv_setpvn(sv, s, tmplen);
376 }
377 else
378 sv_setpv(sv, "");
463ee0b2 379 mark++;
380 }
79072805 381 else
382 sv_setpv(sv,"");
383 len = delimlen;
384 if (len) {
385 for (; items > 0; items--,mark++) {
386 sv_catpvn(sv,delim,len);
387 sv_catsv(sv,*mark);
388 }
389 }
390 else {
391 for (; items > 0; items--,mark++)
392 sv_catsv(sv,*mark);
393 }
394 SvSETMAGIC(sv);
395}
396
397void
8ac85365 398do_sprintf(SV *sv, I32 len, SV **sarg)
79072805 399{
46fc3d4c 400 STRLEN patlen;
401 char *pat = SvPV(*sarg, patlen);
402 bool do_taint = FALSE;
403
404 sv_vsetpvfn(sv, pat, patlen, Null(va_list*), sarg + 1, len - 1, &do_taint);
79072805 405 SvSETMAGIC(sv);
46fc3d4c 406 if (do_taint)
407 SvTAINTED_on(sv);
79072805 408}
409
410void
8ac85365 411do_vecset(SV *sv)
79072805 412{
413 SV *targ = LvTARG(sv);
414 register I32 offset;
415 register I32 size;
8990e307 416 register unsigned char *s;
417 register unsigned long lval;
79072805 418 I32 mask;
a0d0e21e 419 STRLEN targlen;
420 STRLEN len;
79072805 421
8990e307 422 if (!targ)
423 return;
a0d0e21e 424 s = (unsigned char*)SvPV_force(targ, targlen);
8990e307 425 lval = U_L(SvNV(sv));
79072805 426 offset = LvTARGOFF(sv);
427 size = LvTARGLEN(sv);
a0d0e21e 428
429 len = (offset + size + 7) / 8;
430 if (len > targlen) {
431 s = (unsigned char*)SvGROW(targ, len + 1);
432 (void)memzero(s + targlen, len - targlen + 1);
433 SvCUR_set(targ, len);
434 }
435
79072805 436 if (size < 8) {
437 mask = (1 << size) - 1;
438 size = offset & 7;
439 lval &= mask;
440 offset >>= 3;
441 s[offset] &= ~(mask << size);
442 s[offset] |= lval << size;
443 }
444 else {
a0d0e21e 445 offset >>= 3;
79072805 446 if (size == 8)
447 s[offset] = lval & 255;
448 else if (size == 16) {
449 s[offset] = (lval >> 8) & 255;
450 s[offset+1] = lval & 255;
451 }
452 else if (size == 32) {
453 s[offset] = (lval >> 24) & 255;
454 s[offset+1] = (lval >> 16) & 255;
455 s[offset+2] = (lval >> 8) & 255;
456 s[offset+3] = lval & 255;
457 }
458 }
459}
460
461void
8ac85365 462do_chop(register SV *astr, register SV *sv)
79072805 463{
463ee0b2 464 STRLEN len;
a0d0e21e 465 char *s;
466
79072805 467 if (SvTYPE(sv) == SVt_PVAV) {
a0d0e21e 468 register I32 i;
469 I32 max;
470 AV* av = (AV*)sv;
471 max = AvFILL(av);
472 for (i = 0; i <= max; i++) {
473 sv = (SV*)av_fetch(av, i, FALSE);
3280af22 474 if (sv && ((sv = *(SV**)sv), sv != &PL_sv_undef))
a0d0e21e 475 do_chop(astr, sv);
476 }
477 return;
79072805 478 }
479 if (SvTYPE(sv) == SVt_PVHV) {
a0d0e21e 480 HV* hv = (HV*)sv;
481 HE* entry;
482 (void)hv_iterinit(hv);
483 /*SUPPRESS 560*/
484 while (entry = hv_iternext(hv))
485 do_chop(astr,hv_iterval(hv,entry));
486 return;
79072805 487 }
a0d0e21e 488 s = SvPV(sv, len);
748a9306 489 if (len && !SvPOK(sv))
a0d0e21e 490 s = SvPV_force(sv, len);
a0ed51b3 491 if (IN_UTF8) {
492 if (s && len) {
493 char *send = s + len;
494 char *start = s;
495 s = send - 1;
496 while ((*s & 0xc0) == 0x80)
497 --s;
498 if (UTF8SKIP(s) != send - s)
499 warn("Malformed UTF-8 character");
500 sv_setpvn(astr, s, send - s);
501 *s = '\0';
502 SvCUR_set(sv, s - start);
503 SvNIOK_off(sv);
504 }
505 else
506 sv_setpvn(astr, "", 0);
507 }
508 else
a0d0e21e 509 if (s && len) {
510 s += --len;
511 sv_setpvn(astr, s, 1);
512 *s = '\0';
513 SvCUR_set(sv, len);
514 SvNIOK_off(sv);
79072805 515 }
516 else
a0d0e21e 517 sv_setpvn(astr, "", 0);
518 SvSETMAGIC(sv);
519}
520
521I32
8ac85365 522do_chomp(register SV *sv)
a0d0e21e 523{
aeea060c 524 dTHR;
c07a80fd 525 register I32 count;
a0d0e21e 526 STRLEN len;
527 char *s;
c07a80fd 528
3280af22 529 if (RsSNARF(PL_rs))
c07a80fd 530 return 0;
531 count = 0;
a0d0e21e 532 if (SvTYPE(sv) == SVt_PVAV) {
533 register I32 i;
534 I32 max;
535 AV* av = (AV*)sv;
536 max = AvFILL(av);
537 for (i = 0; i <= max; i++) {
538 sv = (SV*)av_fetch(av, i, FALSE);
3280af22 539 if (sv && ((sv = *(SV**)sv), sv != &PL_sv_undef))
a0d0e21e 540 count += do_chomp(sv);
541 }
542 return count;
543 }
544 if (SvTYPE(sv) == SVt_PVHV) {
545 HV* hv = (HV*)sv;
546 HE* entry;
547 (void)hv_iterinit(hv);
548 /*SUPPRESS 560*/
549 while (entry = hv_iternext(hv))
550 count += do_chomp(hv_iterval(hv,entry));
551 return count;
552 }
553 s = SvPV(sv, len);
554 if (len && !SvPOKp(sv))
555 s = SvPV_force(sv, len);
556 if (s && len) {
557 s += --len;
3280af22 558 if (RsPARA(PL_rs)) {
a0d0e21e 559 if (*s != '\n')
560 goto nope;
561 ++count;
562 while (len && s[-1] == '\n') {
563 --len;
564 --s;
565 ++count;
566 }
567 }
a0d0e21e 568 else {
c07a80fd 569 STRLEN rslen;
3280af22 570 char *rsptr = SvPV(PL_rs, rslen);
c07a80fd 571 if (rslen == 1) {
572 if (*s != *rsptr)
573 goto nope;
574 ++count;
575 }
576 else {
8c2cee6f 577 if (len < rslen - 1)
c07a80fd 578 goto nope;
579 len -= rslen - 1;
580 s -= rslen - 1;
36477c24 581 if (memNE(s, rsptr, rslen))
c07a80fd 582 goto nope;
583 count += rslen;
584 }
a0d0e21e 585 }
a0d0e21e 586 *s = '\0';
587 SvCUR_set(sv, len);
588 SvNIOK_off(sv);
589 }
590 nope:
591 SvSETMAGIC(sv);
592 return count;
593}
79072805 594
595void
8ac85365 596do_vop(I32 optype, SV *sv, SV *left, SV *right)
79072805 597{
aeea060c 598 dTHR; /* just for taint */
79072805 599#ifdef LIBERAL
600 register long *dl;
601 register long *ll;
602 register long *rl;
603#endif
604 register char *dc;
463ee0b2 605 STRLEN leftlen;
606 STRLEN rightlen;
7a4c00b4 607 register char *lc;
608 register char *rc;
79072805 609 register I32 len;
a0d0e21e 610 I32 lensave;
7a4c00b4 611 char *lsave;
612 char *rsave;
79072805 613
1fbd88dc 614 if (sv != left || (optype != OP_BIT_AND && !SvOK(sv) && !SvGMAGICAL(sv)))
615 sv_setpvn(sv, "", 0); /* avoid undef warning on |= and ^= */
7a4c00b4 616 lsave = lc = SvPV(left, leftlen);
617 rsave = rc = SvPV(right, rightlen);
93a17b20 618 len = leftlen < rightlen ? leftlen : rightlen;
a0d0e21e 619 lensave = len;
7a4c00b4 620 if (SvOK(sv) || SvTYPE(sv) > SVt_PVMG) {
3280af22 621 dc = SvPV_force(sv, PL_na);
ff68c719 622 if (SvCUR(sv) < len) {
623 dc = SvGROW(sv, len + 1);
624 (void)memzero(dc + SvCUR(sv), len - SvCUR(sv) + 1);
625 }
626 }
627 else {
628 I32 needlen = ((optype == OP_BIT_AND)
629 ? len : (leftlen > rightlen ? leftlen : rightlen));
630 Newz(801, dc, needlen + 1, char);
631 (void)sv_usepvn(sv, dc, needlen);
632 dc = SvPVX(sv); /* sv_usepvn() calls Renew() */
79072805 633 }
a0d0e21e 634 SvCUR_set(sv, len);
635 (void)SvPOK_only(sv);
79072805 636#ifdef LIBERAL
637 if (len >= sizeof(long)*4 &&
638 !((long)dc % sizeof(long)) &&
639 !((long)lc % sizeof(long)) &&
640 !((long)rc % sizeof(long))) /* It's almost always aligned... */
641 {
642 I32 remainder = len % (sizeof(long)*4);
643 len /= (sizeof(long)*4);
644
645 dl = (long*)dc;
646 ll = (long*)lc;
647 rl = (long*)rc;
648
649 switch (optype) {
650 case OP_BIT_AND:
651 while (len--) {
652 *dl++ = *ll++ & *rl++;
653 *dl++ = *ll++ & *rl++;
654 *dl++ = *ll++ & *rl++;
655 *dl++ = *ll++ & *rl++;
656 }
657 break;
a0d0e21e 658 case OP_BIT_XOR:
79072805 659 while (len--) {
660 *dl++ = *ll++ ^ *rl++;
661 *dl++ = *ll++ ^ *rl++;
662 *dl++ = *ll++ ^ *rl++;
663 *dl++ = *ll++ ^ *rl++;
664 }
665 break;
666 case OP_BIT_OR:
667 while (len--) {
668 *dl++ = *ll++ | *rl++;
669 *dl++ = *ll++ | *rl++;
670 *dl++ = *ll++ | *rl++;
671 *dl++ = *ll++ | *rl++;
672 }
673 }
674
675 dc = (char*)dl;
676 lc = (char*)ll;
677 rc = (char*)rl;
678
679 len = remainder;
680 }
681#endif
a0d0e21e 682 {
a0d0e21e 683 switch (optype) {
684 case OP_BIT_AND:
685 while (len--)
686 *dc++ = *lc++ & *rc++;
687 break;
688 case OP_BIT_XOR:
689 while (len--)
690 *dc++ = *lc++ ^ *rc++;
691 goto mop_up;
692 case OP_BIT_OR:
693 while (len--)
694 *dc++ = *lc++ | *rc++;
695 mop_up:
696 len = lensave;
697 if (rightlen > len)
698 sv_catpvn(sv, rsave + len, rightlen - len);
699 else if (leftlen > len)
700 sv_catpvn(sv, lsave + len, leftlen - len);
4633a7c4 701 else
702 *SvEND(sv) = '\0';
a0d0e21e 703 break;
704 }
79072805 705 }
fb73857a 706 SvTAINT(sv);
79072805 707}
463ee0b2 708
709OP *
8ac85365 710do_kv(ARGSproto)
463ee0b2 711{
4e35701f 712 djSP;
463ee0b2 713 HV *hv = (HV*)POPs;
800e9ae0 714 HV *keys;
463ee0b2 715 register HE *entry;
463ee0b2 716 SV *tmpstr;
54310121 717 I32 gimme = GIMME_V;
533c011a 718 I32 dokeys = (PL_op->op_type == OP_KEYS);
719 I32 dovalues = (PL_op->op_type == OP_VALUES);
c750a3ec 720 I32 realhv = (SvTYPE(hv) == SVt_PVHV);
721
533c011a 722 if (PL_op->op_type == OP_RV2HV || PL_op->op_type == OP_PADHV)
a0d0e21e 723 dokeys = dovalues = TRUE;
463ee0b2 724
85581909 725 if (!hv) {
533c011a 726 if (PL_op->op_flags & OPf_MOD) { /* lvalue */
85581909 727 dTARGET; /* make sure to clear its target here */
728 if (SvTYPE(TARG) == SVt_PVLV)
729 LvTARG(TARG) = Nullsv;
730 PUSHs(TARG);
731 }
463ee0b2 732 RETURN;
85581909 733 }
748a9306 734
800e9ae0 735 keys = realhv ? hv : avhv_keys((AV*)hv);
736 (void)hv_iterinit(keys); /* always reset iterator regardless */
748a9306 737
54310121 738 if (gimme == G_VOID)
aa689395 739 RETURN;
740
54310121 741 if (gimme == G_SCALAR) {
6ee623d5 742 IV i;
463ee0b2 743 dTARGET;
744
533c011a 745 if (PL_op->op_flags & OPf_MOD) { /* lvalue */
85581909 746 if (SvTYPE(TARG) < SVt_PVLV) {
747 sv_upgrade(TARG, SVt_PVLV);
748 sv_magic(TARG, Nullsv, 'k', Nullch, 0);
749 }
750 LvTYPE(TARG) = 'k';
800e9ae0 751 if (LvTARG(TARG) != (SV*)keys) {
6ff81951 752 if (LvTARG(TARG))
753 SvREFCNT_dec(LvTARG(TARG));
800e9ae0 754 LvTARG(TARG) = SvREFCNT_inc(keys);
6ff81951 755 }
85581909 756 PUSHs(TARG);
757 RETURN;
758 }
759
800e9ae0 760 if (!SvRMAGICAL(keys) || !mg_find((SV*)keys,'P'))
761 i = HvKEYS(keys);
463ee0b2 762 else {
763 i = 0;
463ee0b2 764 /*SUPPRESS 560*/
800e9ae0 765 while (hv_iternext(keys)) i++;
463ee0b2 766 }
767 PUSHi( i );
768 RETURN;
769 }
770
8ed4b672 771 EXTEND(SP, HvKEYS(keys) * (dokeys + dovalues));
463ee0b2 772
463ee0b2 773 PUTBACK; /* hv_iternext and hv_iterval might clobber stack_sp */
800e9ae0 774 while (entry = hv_iternext(keys)) {
463ee0b2 775 SPAGAIN;
8c2cee6f 776 if (dokeys)
777 XPUSHs(hv_iterkeysv(entry)); /* won't clobber stack_sp */
463ee0b2 778 if (dovalues) {
46fc3d4c 779 tmpstr = sv_newmortal();
463ee0b2 780 PUTBACK;
c750a3ec 781 sv_setsv(tmpstr,realhv ?
782 hv_iterval(hv,entry) : avhv_iterval((AV*)hv,entry));
46fc3d4c 783 DEBUG_H(sv_setpvf(tmpstr, "%lu%%%d=%lu",
784 (unsigned long)HeHASH(entry),
800e9ae0 785 HvMAX(keys)+1,
786 (unsigned long)(HeHASH(entry) & HvMAX(keys))));
463ee0b2 787 SPAGAIN;
46fc3d4c 788 XPUSHs(tmpstr);
463ee0b2 789 }
790 PUTBACK;
791 }
792 return NORMAL;
793}
4e35701f 794