This is a live mirror of the Perl 5 development currently hosted at https://github.com/perl/perl5
As 2/3rds (or 3/4s) of the SV head structure is rewritten, it doesn't
[perl5.git] / ext / Encode / Encode.xs
index 89f7a29..77d53af 100644 (file)
@@ -1,5 +1,5 @@
 /*
- $Id: Encode.xs,v 1.49 2002/10/21 19:47:47 dankogai Exp $
+ $Id: Encode.xs,v 2.0 2004/05/16 20:55:15 dankogai Exp $
  */
 
 #define PERL_NO_GET_CONTEXT
@@ -21,9 +21,9 @@
 #define ENCODE_XS_USEFP   1
 
 #define UNIMPLEMENTED(x,y) y x (SV *sv, char *encoding) {dTHX;   \
-                        Perl_croak(aTHX_ "panic_unimplemented"); \
+                         Perl_croak(aTHX_ "panic_unimplemented"); \
                         return (y)0; /* fool picky compilers */ \
-                        }
+                         }
 /**/
 
 UNIMPLEMENTED(_encoded_utf8_to_bytes, I32)
@@ -59,7 +59,7 @@ call_failure(SV * routine, U8 * done, U8 * dest, U8 * orig)
 
 static SV *
 encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
-             int check)
+             int check, STRLEN * offset, SV * term, int * retcode)
 {
     STRLEN slen;
     U8 *s = (U8 *) SvPV(src, slen);
@@ -72,25 +72,39 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
     SV *dst = sv_2mortal(newSV(slen+1));
     U8 *d = (U8 *)SvPVX(dst);
     STRLEN dlen = SvLEN(dst)-1;
-    int code;
+    int code = 0;
+    STRLEN trmlen = 0;
+    U8 *trm = term ? (U8*) SvPV(term, trmlen) : NULL;
+
+    if (offset) {
+      s += *offset;
+      if (slen > *offset){ /* safeguard against slen overflow */
+         slen -= *offset;
+      }else{
+         slen = 0;
+      }
+      tlen = slen;
+    }
 
-    if (!slen){
+    if (slen == 0){
        SvCUR_set(dst, 0);
        SvPOK_only(dst);
        goto ENCODE_END;
     }
 
-    while( (code = do_encode(dir, s, &slen, d, dlen, &dlen, !check)) )
+    while( (code = do_encode(dir, s, &slen, d, dlen, &dlen, !check,
+                            trm, trmlen)) ) 
     {
        SvCUR_set(dst, dlen+ddone);
        SvPOK_only(dst);
-
-       if (code == ENCODE_FALLBACK || code == ENCODE_PARTIAL){
+       
+       if (code == ENCODE_FALLBACK || code == ENCODE_PARTIAL ||
+           code == ENCODE_FOUND_TERM) {
            break;
        }
        switch (code) {
        case ENCODE_NOSPACE:
-       {
+       {       
            STRLEN more = 0; /* make sure you initialize! */
            STRLEN sleft;
            sdone += slen;
@@ -125,7 +139,7 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
            continue;
        }
        case ENCODE_NOREP:
-           /* encoding */
+           /* encoding */      
            if (dir == enc->f_utf8) {
                STRLEN clen;
                UV ch =
@@ -144,19 +158,19 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
                    goto ENCODE_SET_SRC;
                }
                if (check & ENCODE_PERLQQ){
-                   SV* perlqq =
+                   SV* perlqq = 
                        sv_2mortal(newSVpvf("\\x{%04"UVxf"}", (UV)ch));
                    sdone += slen + clen;
                    ddone += dlen + SvCUR(perlqq);
                    sv_catsv(dst, perlqq);
                }else if (check & ENCODE_HTMLCREF){
-                   SV* htmlcref =
+                   SV* htmlcref = 
                        sv_2mortal(newSVpvf("&#%" UVuf ";", (UV)ch));
                    sdone += slen + clen;
                    ddone += dlen + SvCUR(htmlcref);
                    sv_catsv(dst, htmlcref);
                }else if (check & ENCODE_XMLCREF){
-                   SV* xmlcref =
+                   SV* xmlcref = 
                        sv_2mortal(newSVpvf("&#x%" UVxf ";", (UV)ch));
                    sdone += slen + clen;
                    ddone += dlen + SvCUR(xmlcref);
@@ -172,21 +186,21 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
            else {
                if (check & ENCODE_DIE_ON_ERR){
                    Perl_croak(aTHX_ ERR_DECODE_NOMAP,
-                             enc->name[0], (UV)s[slen]);
+                              enc->name[0], (UV)s[slen]);
                    return &PL_sv_undef; /* never reaches but be safe */
                }
                if (check & ENCODE_WARN_ON_ERR){
                    Perl_warner(
                        aTHX_ packWARN(WARN_UTF8),
                        ERR_DECODE_NOMAP,
-                       enc->name[0], (UV)s[slen]);
+                               enc->name[0], (UV)s[slen]);
                }
                if (check & ENCODE_RETURN_ON_ERR){
                    goto ENCODE_SET_SRC;
                }
                if (check &
                    (ENCODE_PERLQQ|ENCODE_HTMLCREF|ENCODE_XMLCREF)){
-                   SV* perlqq =
+                   SV* perlqq = 
                        sv_2mortal(newSVpvf("\\x%02" UVXf, (UV)s[slen]));
                    sdone += slen + 1;
                    ddone += dlen + SvCUR(perlqq);
@@ -199,7 +213,7 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
            }
            /* settle variables when fallback */
            d    = (U8 *)SvEND(dst);
-           dlen = SvLEN(dst) - ddone - 1;
+            dlen = SvLEN(dst) - ddone - 1;
            s    = (U8*)SvPVX(src) + sdone;
            slen = tlen - sdone;
            break;
@@ -233,24 +247,38 @@ encode_method(pTHX_ encode_t * enc, encpage_t * dir, SV * src,
     }
 #endif
 
+    if (offset) 
+      *offset += sdone + slen;
+
  ENCODE_END:
     *SvEND(dst) = '\0';
+    if (retcode) *retcode = code;
     return dst;
 }
 
-MODULE = Encode         PACKAGE = Encode::utf8  PREFIX = Method_
+MODULE = Encode                PACKAGE = Encode::utf8  PREFIX = Method_
+
+PROTOTYPES: DISABLE
 
 void
-Method_decode(obj,src,check = 0)
-SV *    obj
-SV *    src
-int     check
+Method_renew(obj)
+SV *   obj
+CODE:
+{
+    XSRETURN(1);
+}
+
+void
+Method_decode_xs(obj,src,check = 0)
+SV *   obj
+SV *   src
+int    check
 CODE:
 {
     STRLEN slen;
     U8 *s = (U8 *) SvPV(src, slen);
     U8 *e = (U8 *) SvEND(src);
-    SV *dst = newSV(slen);
+    SV *dst = newSV(slen>0?slen:1); /* newSV() abhors 0 -- inaba */
     SvPOK_only(dst);
     SvCUR_set(dst,0);
     if (SvUTF8(src)) {
@@ -265,20 +293,20 @@ CODE:
        }
     }
     while (s < e) {
-       if (UTF8_IS_INVARIANT(*s) || UTF8_IS_START(*s)) {
+       if (UTF8_IS_INVARIANT(*s) || UTF8_IS_START(*s)) {
            U8 skip = UTF8SKIP(s);
            if ((s + skip) > e) {
-               /* Partial character - done */
-               break;
+               /* Partial character - done */
+               break;
            }
            else if (is_utf8_char(s)) {
-               /* Whole char is good */
+               /* Whole char is good */
                sv_catpvn(dst,(char *)s,skip);
                s += skip;
                continue;
            }
            else {
-               /* starts ok but isn't "good" */
+               /* starts ok but isn't "good" */
            }
        }
        else {
@@ -292,13 +320,13 @@ CODE:
        if (check & ENCODE_WARN_ON_ERR){
            Perl_warner(aTHX_ packWARN(WARN_UTF8),
                        ERR_DECODE_NOMAP, "utf8", (UV)*s);
-       }
-       if (check & ENCODE_RETURN_ON_ERR) {
+        }
+       if (check & ENCODE_RETURN_ON_ERR) {
                break;
-       }
-       if (check & (ENCODE_PERLQQ|ENCODE_HTMLCREF|ENCODE_XMLCREF)){
+       }
+        if (check & (ENCODE_PERLQQ|ENCODE_HTMLCREF|ENCODE_XMLCREF)){
            SV* perlqq = newSVpvf("\\x%02" UVXf, (UV)*s);
-           sv_catsv(dst, perlqq);
+           sv_catsv(dst, perlqq);
            SvREFCNT_dec(perlqq);
        } else {
            sv_catpv(dst, FBCHAR_UTF8);
@@ -321,35 +349,35 @@ CODE:
 }
 
 void
-Method_encode(obj,src,check = 0)
-SV *    obj
-SV *    src
-int     check
+Method_encode_xs(obj,src,check = 0)
+SV *   obj
+SV *   src
+int    check
 CODE:
 {
     STRLEN slen;
     U8 *s = (U8 *) SvPV(src, slen);
     U8 *e = (U8 *) SvEND(src);
-    SV *dst = newSV(slen);
+    SV *dst = newSV(slen>0?slen:1); /* newSV() abhors 0 -- inaba */
     if (SvUTF8(src)) {
-       /* Already encoded - trust it and just copy the octets */
-       sv_setpvn(dst,(char *)s,(e-s));
+        /* Already encoded - trust it and just copy the octets */
+       sv_setpvn(dst,(char *)s,(e-s));
        s = e;
     }
     else {
-       /* Native bytes - can always encode */
-       U8 *d = (U8 *) SvGROW(dst,2*slen);
-       while (s < e) {
-           UV uv = NATIVE_TO_UNI((UV) *s++);
-           if (UNI_IS_INVARIANT(uv))
-               *d++ = (U8)UTF_TO_NATIVE(uv);
-           else {
-               *d++ = (U8)UTF8_EIGHT_BIT_HI(uv);
-               *d++ = (U8)UTF8_EIGHT_BIT_LO(uv);
-           }
+       /* Native bytes - can always encode */
+       U8 *d = (U8 *) SvGROW(dst, 2*slen+1); /* +1 or assertion will botch */
+       while (s < e) {
+           UV uv = NATIVE_TO_UNI((UV) *s++);
+            if (UNI_IS_INVARIANT(uv))
+               *d++ = (U8)UTF_TO_NATIVE(uv);
+            else {
+               *d++ = (U8)UTF8_EIGHT_BIT_HI(uv);
+                *d++ = (U8)UTF8_EIGHT_BIT_LO(uv);
+            }
        }
-       SvCUR_set(dst, d- (U8 *)SvPVX(dst));
-       *SvEND(dst) = '\0';
+        SvCUR_set(dst, d- (U8 *)SvPVX(dst));
+       *SvEND(dst) = '\0';
     }
 
     /* Clear out translated part of source unless asked not to */
@@ -366,13 +394,21 @@ CODE:
     XSRETURN(1);
 }
 
-MODULE = Encode         PACKAGE = Encode::XS    PREFIX = Method_
+MODULE = Encode                PACKAGE = Encode::XS    PREFIX = Method_
 
 PROTOTYPES: ENABLE
 
 void
+Method_renew(obj)
+SV *   obj
+CODE:
+{
+    XSRETURN(1);
+}
+
+void
 Method_name(obj)
-SV *    obj
+SV *   obj
 CODE:
 {
     encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj)));
@@ -381,37 +417,66 @@ CODE:
 }
 
 void
+Method_cat_decode(obj, dst, src, off, term, check = 0)
+SV *   obj
+SV *   dst
+SV *   src
+SV *   off
+SV *   term
+int    check
+CODE:
+{
+    encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj)));
+    STRLEN offset = (STRLEN)SvIV(off);
+    int code = 0;
+    if (SvUTF8(src)) {
+       sv_utf8_downgrade(src, FALSE);
+    }
+    sv_catsv(dst, encode_method(aTHX_ enc, enc->t_utf8, src, check,
+                               &offset, term, &code));
+    SvIVX(off) = (IV)offset;
+    if (code == ENCODE_FOUND_TERM) {
+       ST(0) = &PL_sv_yes;
+    }else{
+       ST(0) = &PL_sv_no;
+    }
+    XSRETURN(1);
+}
+
+void
 Method_decode(obj,src,check = 0)
-SV *    obj
-SV *    src
-int     check
+SV *   obj
+SV *   src
+int    check
 CODE:
 {
     encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj)));
     if (SvUTF8(src)) {
        sv_utf8_downgrade(src, FALSE);
     }
-    ST(0) = encode_method(aTHX_ enc, enc->t_utf8, src, check);
+    ST(0) = encode_method(aTHX_ enc, enc->t_utf8, src, check,
+                         NULL, Nullsv, NULL);
     SvUTF8_on(ST(0));
     XSRETURN(1);
 }
 
 void
 Method_encode(obj,src,check = 0)
-SV *    obj
-SV *    src
-int     check
+SV *   obj
+SV *   src
+int    check
 CODE:
 {
     encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj)));
     sv_utf8_upgrade(src);
-    ST(0) = encode_method(aTHX_ enc, enc->f_utf8, src, check);
+    ST(0) = encode_method(aTHX_ enc, enc->f_utf8, src, check,
+                         NULL, Nullsv, NULL);
     XSRETURN(1);
 }
 
 void
 Method_needs_lines(obj)
-SV *    obj
+SV *   obj
 CODE:
 {
     /* encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj))); */
@@ -421,7 +486,7 @@ CODE:
 
 void
 Method_perlio_ok(obj)
-SV *    obj
+SV *   obj
 CODE:
 {
     /* encode_t *enc = INT2PTR(encode_t *, SvIV(SvRV(obj))); */
@@ -483,16 +548,16 @@ CODE:
        if (SvTRUE(check)) {
            /* Must do things the slow way */
            U8 *dest;
-           /* We need a copy to pass to check() */
+            /* We need a copy to pass to check() */
            U8 *src  = (U8*)savepv((char *)s);
            U8 *send = s + len;
 
            New(83, dest, len, U8); /* I think */
 
            while (s < send) {
-               if (*s < 0x80){
+                if (*s < 0x80){
                    *dest++ = *s++;
-               } else {
+                } else {
                    STRLEN ulen;
                    UV uv = *s++;
 
@@ -506,7 +571,7 @@ CODE:
                    else if (!(uv & 0x02)) { ulen = 6;  uv &= 0x01; }
                    else if (!(uv & 0x01)) { ulen = 7;  uv = 0; }
                    else                   { ulen = 13; uv = 0; }
-
+               
                    /* Note change to utf8.c variable naming, for variety */
                    while (ulen--) {
                        if ((*s & 0xc0) != 0x80){
@@ -533,8 +598,8 @@ OUTPUT:
 
 bool
 is_utf8(sv, check = 0)
-SV *    sv
-int     check
+SV *   sv
+int    check
 CODE:
 {
     if (SvGMAGICAL(sv)) /* it could be $1, for example */
@@ -556,7 +621,7 @@ OUTPUT:
 
 SV *
 _utf8_on(sv)
-SV *    sv
+SV *   sv
 CODE:
 {
     if (SvPOK(sv)) {
@@ -572,7 +637,7 @@ OUTPUT:
 
 SV *
 _utf8_off(sv)
-SV *    sv
+SV *   sv
 CODE:
 {
     if (SvPOK(sv)) {