This is a live mirror of the Perl 5 development currently hosted at https://github.com/perl/perl5
Merge maint-5.004 branch (5.004_04) with mainline.
[perl5.git] / doop.c
CommitLineData
a0d0e21e 1/* doop.c
79072805 2 *
9607fc9c 3 * Copyright (c) 1991-1997, Larry Wall
79072805
LW
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
LW
8 */
9
10/*
11 * "'So that was the job I felt I had to do when I started,' thought Sam."
79072805
LW
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
LW
21I32
22do_trans(sv,arg)
23SV *sv;
24OP *arg;
25{
11343788 26 dTHR;
79072805 27 register short *tbl;
1476d272 28 register U8 *s;
29 register U8 *send;
30 register U8 *d;
79072805 31 register I32 ch;
1476d272 32 register I32 matches = 0;
79072805 33 register I32 squash = op->op_private & OPpTRANS_SQUASH;
463ee0b2 34 STRLEN len;
79072805 35
a0d0e21e
LW
36 if (SvREADONLY(sv))
37 croak(no_modify);
1476d272 38 tbl = (short*)cPVOP->op_pv;
39 s = (U8*)SvPV(sv, len);
a0d0e21e
LW
40 if (!len)
41 return 0;
42 if (!SvPOKp(sv))
1476d272 43 s = (U8*)SvPV_force(sv, len);
a0d0e21e 44 (void)SvPOK_only(sv);
463ee0b2 45 send = s + len;
79072805 46 if (!tbl || !s)
463ee0b2 47 croak("panic: do_trans");
79072805
LW
48 DEBUG_t( deb("2.TBL\n"));
49 if (!op->op_private) {
50 while (s < send) {
1476d272 51 if ((ch = tbl[*s]) >= 0) {
79072805
LW
52 matches++;
53 *s = ch;
54 }
55 s++;
56 }
57 }
58 else {
59 d = s;
60 while (s < send) {
1476d272 61 if ((ch = tbl[*s]) >= 0) {
79072805
LW
62 *d = ch;
63 if (matches++ && squash) {
64 if (d[-1] == *d)
65 matches--;
66 else
67 d++;
68 }
69 else
70 d++;
71 }
72 else if (ch == -1) /* -1 is unmapped character */
73 *d++ = *s; /* -2 is delete character */
74 s++;
75 }
76 matches += send - d; /* account for disappeared chars */
77 *d = '\0';
1476d272 78 SvCUR_set(sv, d - (U8*)SvPVX(sv));
79072805
LW
79 }
80 SvSETMAGIC(sv);
81 return matches;
82}
83
84void
85do_join(sv,del,mark,sp)
86register SV *sv;
87SV *del;
88register SV **mark;
89register SV **sp;
90{
91 SV **oldmark = mark;
92 register I32 items = sp - mark;
79072805 93 register STRLEN len;
463ee0b2
LW
94 STRLEN delimlen;
95 register char *delim = SvPV(del, delimlen);
96 STRLEN tmplen;
79072805
LW
97
98 mark++;
99 len = (items > 0 ? (delimlen * (items - 1) ) : 0);
100 if (SvTYPE(sv) < SVt_PV)
101 sv_upgrade(sv, SVt_PV);
102 if (SvLEN(sv) < len + items) { /* current length is way too short */
103 while (items-- > 0) {
104 if (*mark) {
463ee0b2
LW
105 SvPV(*mark, tmplen);
106 len += tmplen;
79072805
LW
107 }
108 mark++;
109 }
110 SvGROW(sv, len + 1); /* so try to pre-extend */
111
112 mark = oldmark;
113 items = sp - mark;;
114 ++mark;
115 }
116
463ee0b2 117 if (items-- > 0) {
8990e307
LW
118 char *s;
119
120 if (*mark) {
121 s = SvPV(*mark, tmplen);
122 sv_setpvn(sv, s, tmplen);
123 }
124 else
125 sv_setpv(sv, "");
463ee0b2
LW
126 mark++;
127 }
79072805
LW
128 else
129 sv_setpv(sv,"");
130 len = delimlen;
131 if (len) {
132 for (; items > 0; items--,mark++) {
133 sv_catpvn(sv,delim,len);
134 sv_catsv(sv,*mark);
135 }
136 }
137 else {
138 for (; items > 0; items--,mark++)
139 sv_catsv(sv,*mark);
140 }
141 SvSETMAGIC(sv);
142}
143
144void
145do_sprintf(sv,len,sarg)
46fc3d4c 146SV *sv;
147I32 len;
148SV **sarg;
79072805 149{
46fc3d4c 150 STRLEN patlen;
151 char *pat = SvPV(*sarg, patlen);
152 bool do_taint = FALSE;
153
154 sv_vsetpvfn(sv, pat, patlen, Null(va_list*), sarg + 1, len - 1, &do_taint);
79072805 155 SvSETMAGIC(sv);
46fc3d4c 156 if (do_taint)
157 SvTAINTED_on(sv);
79072805
LW
158}
159
160void
161do_vecset(sv)
162SV *sv;
163{
164 SV *targ = LvTARG(sv);
165 register I32 offset;
166 register I32 size;
8990e307
LW
167 register unsigned char *s;
168 register unsigned long lval;
79072805 169 I32 mask;
a0d0e21e
LW
170 STRLEN targlen;
171 STRLEN len;
79072805 172
8990e307
LW
173 if (!targ)
174 return;
a0d0e21e 175 s = (unsigned char*)SvPV_force(targ, targlen);
8990e307 176 lval = U_L(SvNV(sv));
79072805
LW
177 offset = LvTARGOFF(sv);
178 size = LvTARGLEN(sv);
a0d0e21e
LW
179
180 len = (offset + size + 7) / 8;
181 if (len > targlen) {
182 s = (unsigned char*)SvGROW(targ, len + 1);
183 (void)memzero(s + targlen, len - targlen + 1);
184 SvCUR_set(targ, len);
185 }
186
79072805
LW
187 if (size < 8) {
188 mask = (1 << size) - 1;
189 size = offset & 7;
190 lval &= mask;
191 offset >>= 3;
192 s[offset] &= ~(mask << size);
193 s[offset] |= lval << size;
194 }
195 else {
a0d0e21e 196 offset >>= 3;
79072805
LW
197 if (size == 8)
198 s[offset] = lval & 255;
199 else if (size == 16) {
200 s[offset] = (lval >> 8) & 255;
201 s[offset+1] = lval & 255;
202 }
203 else if (size == 32) {
204 s[offset] = (lval >> 24) & 255;
205 s[offset+1] = (lval >> 16) & 255;
206 s[offset+2] = (lval >> 8) & 255;
207 s[offset+3] = lval & 255;
208 }
209 }
210}
211
212void
213do_chop(astr,sv)
214register SV *astr;
215register SV *sv;
216{
463ee0b2 217 STRLEN len;
a0d0e21e
LW
218 char *s;
219
79072805 220 if (SvTYPE(sv) == SVt_PVAV) {
a0d0e21e
LW
221 register I32 i;
222 I32 max;
223 AV* av = (AV*)sv;
224 max = AvFILL(av);
225 for (i = 0; i <= max; i++) {
226 sv = (SV*)av_fetch(av, i, FALSE);
227 if (sv && ((sv = *(SV**)sv), sv != &sv_undef))
228 do_chop(astr, sv);
229 }
230 return;
79072805
LW
231 }
232 if (SvTYPE(sv) == SVt_PVHV) {
a0d0e21e
LW
233 HV* hv = (HV*)sv;
234 HE* entry;
235 (void)hv_iterinit(hv);
236 /*SUPPRESS 560*/
237 while (entry = hv_iternext(hv))
238 do_chop(astr,hv_iterval(hv,entry));
239 return;
79072805 240 }
a0d0e21e 241 s = SvPV(sv, len);
748a9306 242 if (len && !SvPOK(sv))
a0d0e21e
LW
243 s = SvPV_force(sv, len);
244 if (s && len) {
245 s += --len;
246 sv_setpvn(astr, s, 1);
247 *s = '\0';
248 SvCUR_set(sv, len);
249 SvNIOK_off(sv);
79072805
LW
250 }
251 else
a0d0e21e
LW
252 sv_setpvn(astr, "", 0);
253 SvSETMAGIC(sv);
254}
255
256I32
257do_chomp(sv)
258register SV *sv;
259{
c07a80fd 260 register I32 count;
a0d0e21e
LW
261 STRLEN len;
262 char *s;
c07a80fd 263
264 if (RsSNARF(rs))
265 return 0;
266 count = 0;
a0d0e21e
LW
267 if (SvTYPE(sv) == SVt_PVAV) {
268 register I32 i;
269 I32 max;
270 AV* av = (AV*)sv;
271 max = AvFILL(av);
272 for (i = 0; i <= max; i++) {
273 sv = (SV*)av_fetch(av, i, FALSE);
274 if (sv && ((sv = *(SV**)sv), sv != &sv_undef))
275 count += do_chomp(sv);
276 }
277 return count;
278 }
279 if (SvTYPE(sv) == SVt_PVHV) {
280 HV* hv = (HV*)sv;
281 HE* entry;
282 (void)hv_iterinit(hv);
283 /*SUPPRESS 560*/
284 while (entry = hv_iternext(hv))
285 count += do_chomp(hv_iterval(hv,entry));
286 return count;
287 }
288 s = SvPV(sv, len);
289 if (len && !SvPOKp(sv))
290 s = SvPV_force(sv, len);
291 if (s && len) {
292 s += --len;
c07a80fd 293 if (RsPARA(rs)) {
a0d0e21e
LW
294 if (*s != '\n')
295 goto nope;
296 ++count;
297 while (len && s[-1] == '\n') {
298 --len;
299 --s;
300 ++count;
301 }
302 }
a0d0e21e 303 else {
c07a80fd 304 STRLEN rslen;
305 char *rsptr = SvPV(rs, rslen);
306 if (rslen == 1) {
307 if (*s != *rsptr)
308 goto nope;
309 ++count;
310 }
311 else {
8c2cee6f 312 if (len < rslen - 1)
c07a80fd 313 goto nope;
314 len -= rslen - 1;
315 s -= rslen - 1;
36477c24 316 if (memNE(s, rsptr, rslen))
c07a80fd 317 goto nope;
318 count += rslen;
319 }
a0d0e21e 320 }
a0d0e21e
LW
321 *s = '\0';
322 SvCUR_set(sv, len);
323 SvNIOK_off(sv);
324 }
325 nope:
326 SvSETMAGIC(sv);
327 return count;
328}
79072805
LW
329
330void
331do_vop(optype,sv,left,right)
332I32 optype;
333SV *sv;
334SV *left;
335SV *right;
336{
337#ifdef LIBERAL
338 register long *dl;
339 register long *ll;
340 register long *rl;
341#endif
342 register char *dc;
463ee0b2
LW
343 STRLEN leftlen;
344 STRLEN rightlen;
7a4c00b4 345 register char *lc;
346 register char *rc;
79072805 347 register I32 len;
a0d0e21e 348 I32 lensave;
7a4c00b4 349 char *lsave;
350 char *rsave;
79072805 351
1fbd88dc
CS
352 if (sv != left || (optype != OP_BIT_AND && !SvOK(sv) && !SvGMAGICAL(sv)))
353 sv_setpvn(sv, "", 0); /* avoid undef warning on |= and ^= */
7a4c00b4 354 lsave = lc = SvPV(left, leftlen);
355 rsave = rc = SvPV(right, rightlen);
93a17b20 356 len = leftlen < rightlen ? leftlen : rightlen;
a0d0e21e 357 lensave = len;
7a4c00b4 358 if (SvOK(sv) || SvTYPE(sv) > SVt_PVMG) {
ff68c719 359 dc = SvPV_force(sv, na);
360 if (SvCUR(sv) < len) {
361 dc = SvGROW(sv, len + 1);
362 (void)memzero(dc + SvCUR(sv), len - SvCUR(sv) + 1);
363 }
364 }
365 else {
366 I32 needlen = ((optype == OP_BIT_AND)
367 ? len : (leftlen > rightlen ? leftlen : rightlen));
368 Newz(801, dc, needlen + 1, char);
369 (void)sv_usepvn(sv, dc, needlen);
370 dc = SvPVX(sv); /* sv_usepvn() calls Renew() */
79072805 371 }
a0d0e21e
LW
372 SvCUR_set(sv, len);
373 (void)SvPOK_only(sv);
79072805
LW
374#ifdef LIBERAL
375 if (len >= sizeof(long)*4 &&
376 !((long)dc % sizeof(long)) &&
377 !((long)lc % sizeof(long)) &&
378 !((long)rc % sizeof(long))) /* It's almost always aligned... */
379 {
380 I32 remainder = len % (sizeof(long)*4);
381 len /= (sizeof(long)*4);
382
383 dl = (long*)dc;
384 ll = (long*)lc;
385 rl = (long*)rc;
386
387 switch (optype) {
388 case OP_BIT_AND:
389 while (len--) {
390 *dl++ = *ll++ & *rl++;
391 *dl++ = *ll++ & *rl++;
392 *dl++ = *ll++ & *rl++;
393 *dl++ = *ll++ & *rl++;
394 }
395 break;
a0d0e21e 396 case OP_BIT_XOR:
79072805
LW
397 while (len--) {
398 *dl++ = *ll++ ^ *rl++;
399 *dl++ = *ll++ ^ *rl++;
400 *dl++ = *ll++ ^ *rl++;
401 *dl++ = *ll++ ^ *rl++;
402 }
403 break;
404 case OP_BIT_OR:
405 while (len--) {
406 *dl++ = *ll++ | *rl++;
407 *dl++ = *ll++ | *rl++;
408 *dl++ = *ll++ | *rl++;
409 *dl++ = *ll++ | *rl++;
410 }
411 }
412
413 dc = (char*)dl;
414 lc = (char*)ll;
415 rc = (char*)rl;
416
417 len = remainder;
418 }
419#endif
a0d0e21e 420 {
a0d0e21e
LW
421 switch (optype) {
422 case OP_BIT_AND:
423 while (len--)
424 *dc++ = *lc++ & *rc++;
425 break;
426 case OP_BIT_XOR:
427 while (len--)
428 *dc++ = *lc++ ^ *rc++;
429 goto mop_up;
430 case OP_BIT_OR:
431 while (len--)
432 *dc++ = *lc++ | *rc++;
433 mop_up:
434 len = lensave;
435 if (rightlen > len)
436 sv_catpvn(sv, rsave + len, rightlen - len);
437 else if (leftlen > len)
438 sv_catpvn(sv, lsave + len, leftlen - len);
4633a7c4
LW
439 else
440 *SvEND(sv) = '\0';
a0d0e21e
LW
441 break;
442 }
79072805 443 }
fb73857a 444 SvTAINT(sv);
79072805 445}
463ee0b2
LW
446
447OP *
448do_kv(ARGS)
449dARGS
450{
451 dSP;
452 HV *hv = (HV*)POPs;
463ee0b2 453 register HE *entry;
463ee0b2 454 SV *tmpstr;
54310121 455 I32 gimme = GIMME_V;
a0d0e21e
LW
456 I32 dokeys = (op->op_type == OP_KEYS);
457 I32 dovalues = (op->op_type == OP_VALUES);
c750a3ec
MB
458 I32 realhv = (SvTYPE(hv) == SVt_PVHV);
459
a0d0e21e
LW
460 if (op->op_type == OP_RV2HV || op->op_type == OP_PADHV)
461 dokeys = dovalues = TRUE;
463ee0b2 462
85581909
SB
463 if (!hv) {
464 if (op->op_flags & OPf_MOD) { /* lvalue */
465 dTARGET; /* make sure to clear its target here */
466 if (SvTYPE(TARG) == SVt_PVLV)
467 LvTARG(TARG) = Nullsv;
468 PUSHs(TARG);
469 }
463ee0b2 470 RETURN;
85581909 471 }
748a9306 472
c750a3ec
MB
473 if (realhv)
474 (void)hv_iterinit(hv); /* always reset iterator regardless */
475 else
476 (void)avhv_iterinit((AV*)hv);
748a9306 477
54310121 478 if (gimme == G_VOID)
aa689395 479 RETURN;
480
54310121 481 if (gimme == G_SCALAR) {
8c2cee6f 482 I32 i;
463ee0b2
LW
483 dTARGET;
484
85581909
SB
485 if (op->op_flags & OPf_MOD) { /* lvalue */
486 if (SvTYPE(TARG) < SVt_PVLV) {
487 sv_upgrade(TARG, SVt_PVLV);
488 sv_magic(TARG, Nullsv, 'k', Nullch, 0);
489 }
490 LvTYPE(TARG) = 'k';
491 LvTARG(TARG) = (SV*)hv;
492 PUSHs(TARG);
493 RETURN;
494 }
495
8990e307 496 if (!SvRMAGICAL(hv) || !mg_find((SV*)hv,'P'))
463ee0b2
LW
497 i = HvKEYS(hv);
498 else {
499 i = 0;
463ee0b2 500 /*SUPPRESS 560*/
c750a3ec 501 while (entry = realhv ? hv_iternext(hv) : avhv_iternext((AV*)hv)) {
463ee0b2
LW
502 i++;
503 }
504 }
505 PUSHi( i );
506 RETURN;
507 }
508
509 /* Guess how much room we need. hv_max may be a few too many. Oh well. */
510 EXTEND(sp, HvMAX(hv) * (dokeys + dovalues));
511
463ee0b2 512 PUTBACK; /* hv_iternext and hv_iterval might clobber stack_sp */
c750a3ec 513 while (entry = realhv ? hv_iternext(hv) : avhv_iternext((AV*)hv)) {
463ee0b2 514 SPAGAIN;
8c2cee6f 515 if (dokeys)
516 XPUSHs(hv_iterkeysv(entry)); /* won't clobber stack_sp */
463ee0b2 517 if (dovalues) {
46fc3d4c 518 tmpstr = sv_newmortal();
463ee0b2 519 PUTBACK;
c750a3ec
MB
520 sv_setsv(tmpstr,realhv ?
521 hv_iterval(hv,entry) : avhv_iterval((AV*)hv,entry));
46fc3d4c 522 DEBUG_H(sv_setpvf(tmpstr, "%lu%%%d=%lu",
523 (unsigned long)HeHASH(entry),
524 HvMAX(hv)+1,
525 (unsigned long)(HeHASH(entry) & HvMAX(hv))));
463ee0b2 526 SPAGAIN;
46fc3d4c 527 XPUSHs(tmpstr);
463ee0b2
LW
528 }
529 PUTBACK;
530 }
531 return NORMAL;
532}