Fail hard if strip_names_and_args fails.
[p5sagit/Devel-Declare.git] / Declare.xs
CommitLineData
94caac6e 1#include "EXTERN.h"
2#include "perl.h"
3#include "XSUB.h"
b15aa864 4#include "hook_op_check.h"
94caac6e 5#undef printf
e807ee50 6#include "stolen_chunk_of_toke.c"
94caac6e 7#include <stdio.h>
8#include <string.h>
9
9673d7ca 10#ifndef Newx
11# define Newx(v,n,t) New(0,v,n,t)
12#endif /* !Newx */
13
e81bee92 14static int dd_debug = 0;
c630715a 15
94caac6e 16#define LEX_NORMAL 10
17#define LEX_INTERPNORMAL 9
18
94caac6e 19/* flag to trigger removal of temporary declaree sub */
20
21static int in_declare = 0;
22
96f12726 23/* in 5.10, PL_parser will be NULL if we aren't parsing, and PL_lex_stuff
24 is a lookup into it - so if anything else we can use to tell, so we
25 need to be a bit more careful if PL_parser exists */
26
27#define DD_AM_LEXING_CHECK (PL_lex_state == LEX_NORMAL || PL_lex_state == LEX_INTERPNORMAL)
28
f19b3507 29#if defined(PL_parser) || defined(PERL_5_9_PLUS)
96f12726 30#define DD_HAVE_PARSER PL_parser
31#define DD_HAVE_LEX_STUFF (PL_parser && PL_lex_stuff)
32#define DD_AM_LEXING (PL_parser && DD_AM_LEXING_CHECK)
33#else
34#define DD_HAVE_PARSER 1
35#define DD_HAVE_LEX_STUFF PL_lex_stuff
36#define DD_AM_LEXING DD_AM_LEXING_CHECK
37#endif
38
72f20f69 39/* thing that decides whether we're dealing with a declarator */
40
41int dd_is_declarator(pTHX_ char* name) {
42 HV* is_declarator;
43 SV** is_declarator_pack_ref;
44 HV* is_declarator_pack_hash;
45 SV** is_declarator_flag_ref;
46 int dd_flags;
47
48 is_declarator = get_hv("Devel::Declare::declarators", FALSE);
49
50 if (!is_declarator)
51 return -1;
52
569ac469 53 /* $declarators{$current_package_name} */
54
9603b8dc 55 if (!HvNAME(PL_curstash))
f2a41aa1 56 return -1;
9603b8dc 57
72f20f69 58 is_declarator_pack_ref = hv_fetch(is_declarator, HvNAME(PL_curstash),
59 strlen(HvNAME(PL_curstash)), FALSE);
60
61 if (!is_declarator_pack_ref || !SvROK(*is_declarator_pack_ref))
62 return -1; /* not a hashref */
63
64 is_declarator_pack_hash = (HV*) SvRV(*is_declarator_pack_ref);
65
569ac469 66 /* $declarators{$current_package_name}{$name} */
67
72f20f69 68 is_declarator_flag_ref = hv_fetch(
69 is_declarator_pack_hash, name,
70 strlen(name), FALSE
71 );
72
73 /* requires SvIOK as well as TRUE since flags not being an int is useless */
74
75 if (!is_declarator_flag_ref
48ee5c99 76 || !SvIOK(*is_declarator_flag_ref)
72f20f69 77 || !SvTRUE(*is_declarator_flag_ref))
78 return -1;
79
80 dd_flags = SvIVX(*is_declarator_flag_ref);
81
82 return dd_flags;
83}
84
569ac469 85/* callback thingy */
86
a9bd9b5e 87void dd_linestr_callback (pTHX_ char* type, char* name) {
569ac469 88
89 char* linestr = SvPVX(PL_linestr);
a9bd9b5e 90 int offset = PL_bufptr - linestr;
569ac469 91
569ac469 92 dSP;
93
94 ENTER;
95 SAVETMPS;
96
97 PUSHMARK(SP);
98 XPUSHs(sv_2mortal(newSVpv(type, 0)));
99 XPUSHs(sv_2mortal(newSVpv(name, 0)));
100 XPUSHs(sv_2mortal(newSViv(offset)));
101 PUTBACK;
102
04a8a223 103 call_pv("Devel::Declare::linestr_callback", G_VOID|G_DISCARD);
569ac469 104
569ac469 105 FREETMPS;
106 LEAVE;
107}
108
109char* dd_get_linestr(pTHX) {
5f0b59d5 110 if (!DD_HAVE_PARSER) {
111 return NULL;
112 }
569ac469 113 return SvPVX(PL_linestr);
114}
115
116void dd_set_linestr(pTHX_ char* new_value) {
6f5220b7 117 unsigned int new_len = strlen(new_value);
569ac469 118
ce9a252b 119 if (SvLEN(PL_linestr) < new_len) {
f9eb22eb 120 croak("forced to realloc PL_linestr for line %s, bailing out before we crash harder", SvPVX(PL_linestr));
ce9a252b 121 }
122
123 SvGROW(PL_linestr, new_len);
569ac469 124
125 memcpy(SvPVX(PL_linestr), new_value, new_len+1);
126
127 SvCUR_set(PL_linestr, new_len);
128
129 PL_bufend = SvPVX(PL_linestr) + new_len;
130}
131
04a8a223 132char* dd_get_lex_stuff(pTHX) {
96f12726 133 return (DD_HAVE_LEX_STUFF ? SvPVX(PL_lex_stuff) : "");
04a8a223 134}
135
bd85a06e 136void dd_clear_lex_stuff(pTHX) {
96f12726 137 if (DD_HAVE_PARSER)
f05cbc90 138 PL_lex_stuff = (SV*)NULL;
04a8a223 139}
140
141char* dd_get_curstash_name(pTHX) {
142 return HvNAME(PL_curstash);
143}
144
022eb0cc 145int dd_get_linestr_offset(pTHX) {
0da63271 146 char* linestr;
147 if (!DD_HAVE_PARSER) {
148 return -1;
149 }
150 linestr = SvPVX(PL_linestr);
022eb0cc 151 return PL_bufptr - linestr;
152}
153
840ebcbb 154char* dd_move_past_token (pTHX_ char* s) {
923c07a8 155
156 /*
157 * buffer will be at the beginning of the declarator, -unless- the
158 * declarator is at EOL in which case it'll be the next useful line
159 * so we don't short-circuit out if we don't find the declarator
160 */
161
162 while (s < PL_bufend && isSPACE(*s)) s++;
163 if (memEQ(s, PL_tokenbuf, strlen(PL_tokenbuf)))
164 s += strlen(PL_tokenbuf);
165 return s;
166}
167
840ebcbb 168int dd_toke_move_past_token (pTHX_ int offset) {
04a8a223 169 char* base_s = SvPVX(PL_linestr) + offset;
170 char* s = dd_move_past_token(aTHX_ base_s);
171 return s - base_s;
172}
173
923c07a8 174int dd_toke_scan_word(pTHX_ int offset, int handle_package) {
175 char tmpbuf[sizeof PL_tokenbuf];
176 char* base_s = SvPVX(PL_linestr) + offset;
177 STRLEN len;
178 char* s = scan_word(base_s, tmpbuf, sizeof tmpbuf, handle_package, &len);
179 return s - base_s;
180}
181
f11d21b2 182int dd_toke_scan_ident(pTHX_ int offset) {
183 char tmpbuf[sizeof PL_tokenbuf];
184 char* base_s = SvPVX(PL_linestr) + offset;
f11d21b2 185 char* s = scan_ident(base_s, PL_bufend, tmpbuf, sizeof tmpbuf, 0);
186 return s - base_s;
187}
188
923c07a8 189int dd_toke_scan_str(pTHX_ int offset) {
19b277fc 190 STRLEN remaining = sv_len(PL_linestr) - offset;
86964fb3 191 SV* line_copy = newSVsv(PL_linestr);
923c07a8 192 char* base_s = SvPVX(PL_linestr) + offset;
193 char* s = scan_str(base_s, FALSE, FALSE);
86964fb3 194 if (s != base_s && sv_len(PL_lex_stuff) > remaining) {
195 int ret = (s - SvPVX(PL_linestr)) + remaining;
196 sv_catsv(line_copy, PL_linestr);
197 dd_set_linestr(aTHX_ SvPV_nolen(line_copy));
198 SvREFCNT_dec(line_copy);
199 return ret;
200 }
923c07a8 201 return s - base_s;
202}
203
204int dd_toke_skipspace(pTHX_ int offset) {
205 char* base_s = SvPVX(PL_linestr) + offset;
206 char* s = skipspace(base_s);
207 return s - base_s;
208}
209
94caac6e 210/* replacement PL_check rv2cv entry */
211
6e67754a 212STATIC OP *dd_ck_rv2cv(pTHX_ OP *o, void *user_data) {
19b7ec0f 213 dSP;
94caac6e 214 OP* kid;
0ba8c7aa 215 int dd_flags;
94caac6e 216
6f5220b7 217 PERL_UNUSED_VAR(user_data);
218
94caac6e 219 if (in_declare) {
e81bee92 220 if (dd_debug) {
221 printf("Deconstructing declare\n");
222 printf("PL_bufptr: %s\n", PL_bufptr);
223 printf("bufend at: %i\n", PL_bufend - PL_bufptr);
224 printf("linestr: %s\n", SvPVX(PL_linestr));
225 printf("linestr len: %i\n", PL_bufend - SvPVX(PL_linestr));
226 }
001d53d0 227
001d53d0 228 ENTER;
229 SAVETMPS;
19b7ec0f 230
001d53d0 231 PUSHMARK(SP);
19b7ec0f 232
001d53d0 233 call_pv("Devel::Declare::done_declare", G_VOID|G_DISCARD);
234
235 FREETMPS;
236 LEAVE;
237
e81bee92 238 if (dd_debug) {
239 printf("PL_bufptr: %s\n", PL_bufptr);
240 printf("bufend at: %i\n", PL_bufend - PL_bufptr);
241 printf("linestr: %s\n", SvPVX(PL_linestr));
242 printf("linestr len: %i\n", PL_bufend - SvPVX(PL_linestr));
243 printf("actual len: %i\n", strlen(PL_bufptr));
244 }
94caac6e 245 return o;
246 }
247
248 kid = cUNOPo->op_first;
249
250 if (kid->op_type != OP_GV) /* not a GV so ignore */
251 return o;
252
96f12726 253 if (!DD_AM_LEXING)
94caac6e 254 return o; /* not lexing? */
255
e81bee92 256 if (dd_debug) {
257 printf("Checking GV %s -> %s\n", HvNAME(GvSTASH(kGVOP_gv)), GvNAME(kGVOP_gv));
258 }
840ebcbb 259
72f20f69 260 dd_flags = dd_is_declarator(aTHX_ GvNAME(kGVOP_gv));
94caac6e 261
72f20f69 262 if (dd_flags == -1)
94caac6e 263 return o;
264
e81bee92 265 if (dd_debug) {
266 printf("dd_flags are: %i\n", dd_flags);
267 printf("PL_tokenbuf: %s\n", PL_tokenbuf);
268 }
840ebcbb 269
a9bd9b5e 270 dd_linestr_callback(aTHX_ "rv2cv", GvNAME(kGVOP_gv));
94caac6e 271
53e3ab32 272 return o;
273}
274
b7505981 275OP* dd_pp_entereval(pTHX) {
276 dSP;
b7505981 277 STRLEN len;
278 const char* s;
c93bc23e 279 SV *sv;
280#ifdef PERL_5_9_PLUS
281 SV *saved_hh;
282 if (PL_op->op_private & OPpEVAL_HAS_HH) {
283 saved_hh = POPs;
284 }
285#endif
286 sv = POPs;
b7505981 287 if (SvPOK(sv)) {
e81bee92 288 if (dd_debug) {
289 printf("mangling eval sv\n");
290 }
b7505981 291 if (SvREADONLY(sv))
292 sv = sv_2mortal(newSVsv(sv));
293 s = SvPVX(sv);
294 len = SvCUR(sv);
295 if (!len || s[len-1] != ';') {
296 if (!(SvFLAGS(sv) & SVs_TEMP))
297 sv = sv_2mortal(newSVsv(sv));
298 sv_catpvn(sv, "\n;", 2);
53e3ab32 299 }
b7505981 300 SvGROW(sv, 8192);
53e3ab32 301 }
b7505981 302 PUSHs(sv);
c93bc23e 303#ifdef PERL_5_9_PLUS
304 if (PL_op->op_private & OPpEVAL_HAS_HH) {
305 PUSHs(saved_hh);
306 }
307#endif
b7505981 308 return PL_ppaddr[OP_ENTEREVAL](aTHX);
309}
53e3ab32 310
6e67754a 311STATIC OP *dd_ck_entereval(pTHX_ OP *o, void *user_data) {
6f5220b7 312 PERL_UNUSED_VAR(user_data);
313
d8e65fc8 314 if (o->op_ppaddr == PL_ppaddr[OP_ENTEREVAL])
315 o->op_ppaddr = dd_pp_entereval;
94caac6e 316 return o;
317}
318
6a0bcf35 319static I32 dd_filter_realloc(pTHX_ int idx, SV *sv, int maxlen)
320{
321 const I32 count = FILTER_READ(idx+1, sv, maxlen);
322 SvGROW(sv, 8192); /* please try not to have a line longer than this :) */
323 /* filter_del(dd_filter_realloc); */
324 return count;
325}
326
6e67754a 327STATIC OP *dd_ck_const(pTHX_ OP *o, void *user_data) {
bedac9ff 328 int dd_flags;
04a8a223 329 char* name;
bedac9ff 330
6f5220b7 331 PERL_UNUSED_VAR(user_data);
332
e21e91fe 333 if (DD_HAVE_PARSER && PL_expect == XOPERATOR) {
334 return o;
335 }
336
34335b63 337 /* if this is set, we just grabbed a delimited string or something,
338 not a bareword, so NO TOUCHY */
339
79abf18a 340 if (DD_HAVE_LEX_STUFF)
34335b63 341 return o;
342
bedac9ff 343 /* don't try and look this up if it's not a string const */
344 if (!SvPOK(cSVOPo->op_sv))
345 return o;
346
04a8a223 347 name = SvPVX(cSVOPo->op_sv);
348
349 dd_flags = dd_is_declarator(aTHX_ name);
bedac9ff 350
a9fb5fb1 351 if (dd_flags == -1)
bedac9ff 352 return o;
353
c48ac26f 354 switch (PL_lex_inwhat) {
355 case OP_QR:
356 case OP_MATCH:
357 case OP_SUBST:
358 case OP_TRANS:
f2a41aa1 359 case OP_BACKTICK:
360 case OP_STRINGIFY:
c48ac26f 361 return o;
362 break;
363 default:
364 break;
8d96afb7 365 }
48ee5c99 366
a9bd9b5e 367 dd_linestr_callback(aTHX_ "const", name);
bedac9ff 368
3ea50944 369 return o;
370}
371
94caac6e 372static int initialized = 0;
373
374MODULE = Devel::Declare PACKAGE = Devel::Declare
375
376PROTOTYPES: DISABLE
377
378void
379setup()
380 CODE:
381 if (!initialized++) {
6e67754a 382 hook_op_check(OP_RV2CV, dd_ck_rv2cv, NULL);
383 hook_op_check(OP_ENTEREVAL, dd_ck_entereval, NULL);
384 hook_op_check(OP_CONST, dd_ck_const, NULL);
94caac6e 385 }
6a0bcf35 386 filter_add(dd_filter_realloc, NULL);
569ac469 387
388char*
389get_linestr()
390 CODE:
391 RETVAL = dd_get_linestr(aTHX);
392 OUTPUT:
393 RETVAL
923c07a8 394
395void
396set_linestr(char* new_value)
397 CODE:
398 dd_set_linestr(aTHX_ new_value);
399
04a8a223 400char*
401get_lex_stuff()
402 CODE:
403 RETVAL = dd_get_lex_stuff(aTHX);
404 OUTPUT:
405 RETVAL
406
407void
408clear_lex_stuff()
409 CODE:
410 dd_clear_lex_stuff(aTHX);
411
412char*
413get_curstash_name()
414 CODE:
415 RETVAL = dd_get_curstash_name(aTHX);
416 OUTPUT:
417 RETVAL
418
923c07a8 419int
022eb0cc 420get_linestr_offset()
421 CODE:
422 RETVAL = dd_get_linestr_offset(aTHX);
423 OUTPUT:
424 RETVAL
425
426int
923c07a8 427toke_scan_word(int offset, int handle_package)
428 CODE:
429 RETVAL = dd_toke_scan_word(aTHX_ offset, handle_package);
430 OUTPUT:
431 RETVAL
432
433int
04a8a223 434toke_move_past_token(int offset);
435 CODE:
436 RETVAL = dd_toke_move_past_token(aTHX_ offset);
437 OUTPUT:
438 RETVAL
439
440int
923c07a8 441toke_scan_str(int offset);
442 CODE:
443 RETVAL = dd_toke_scan_str(aTHX_ offset);
444 OUTPUT:
445 RETVAL
446
447int
f11d21b2 448toke_scan_ident(int offset)
449 CODE:
450 RETVAL = dd_toke_scan_ident(aTHX_ offset);
451 OUTPUT:
452 RETVAL
453
454int
923c07a8 455toke_skipspace(int offset)
456 CODE:
457 RETVAL = dd_toke_skipspace(aTHX_ offset);
458 OUTPUT:
459 RETVAL
04a8a223 460
461int
462get_in_declare()
463 CODE:
464 RETVAL = in_declare;
465 OUTPUT:
466 RETVAL
467
468void
469set_in_declare(int value)
470 CODE:
471 in_declare = value;
e81bee92 472
473BOOT:
474 if (getenv ("DD_DEBUG")) {
475 dd_debug = 1;
476 }