This is a live mirror of the Perl 5 development currently hosted at https://github.com/perl/perl5
ext/ + -Wall
[perl5.git] / ext / MIME / Base64 / Base64.xs
1 /*
2
3 Copyright 1997-1999,2001 Gisle Aas
4
5 This library is free software; you can redistribute it and/or
6 modify it under the same terms as Perl itself.
7
8
9 The tables and some of the code that used to be here was borrowed from
10 metamail, which comes with this message:
11
12   Copyright (c) 1991 Bell Communications Research, Inc. (Bellcore)
13
14   Permission to use, copy, modify, and distribute this material
15   for any purpose and without fee is hereby granted, provided
16   that the above copyright notice and this permission notice
17   appear in all copies, and that the name of Bellcore not be
18   used in advertising or publicity pertaining to this
19   material without the specific, prior written permission
20   of an authorized representative of Bellcore.  BELLCORE
21   MAKES NO REPRESENTATIONS ABOUT THE ACCURACY OR SUITABILITY
22   OF THIS MATERIAL FOR ANY PURPOSE.  IT IS PROVIDED "AS IS",
23   WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES.
24
25 */
26
27
28 #ifdef __cplusplus
29 extern "C" {
30 #endif
31 #include "EXTERN.h"
32 #include "perl.h"
33 #include "XSUB.h"
34 #ifdef __cplusplus
35 }
36 #endif
37
38
39 #define MAX_LINE  76 /* size of encoded lines */
40
41 static char basis_64[] =
42    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
43
44 #define XX      255     /* illegal base64 char */
45 #define EQ      254     /* padding */
46 #define INVALID XX
47
48 static unsigned char index_64[256] = {
49     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
50     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
51     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,62, XX,XX,XX,63,
52     52,53,54,55, 56,57,58,59, 60,61,XX,XX, XX,EQ,XX,XX,
53     XX, 0, 1, 2,  3, 4, 5, 6,  7, 8, 9,10, 11,12,13,14,
54     15,16,17,18, 19,20,21,22, 23,24,25,XX, XX,XX,XX,XX,
55     XX,26,27,28, 29,30,31,32, 33,34,35,36, 37,38,39,40,
56     41,42,43,44, 45,46,47,48, 49,50,51,XX, XX,XX,XX,XX,
57
58     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
59     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
60     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
61     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
62     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
63     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
64     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
65     XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX, XX,XX,XX,XX,
66 };
67
68
69
70 MODULE = MIME::Base64           PACKAGE = MIME::Base64
71
72 SV*
73 encode_base64(sv,...)
74         SV* sv
75         PROTOTYPE: $;$
76
77         PREINIT:
78         char *str;     /* string to encode */
79         SSize_t len;   /* length of the string */
80         char *eol;     /* the end-of-line sequence to use */
81         STRLEN eollen; /* length of the EOL sequence */
82         char *r;       /* result string */
83         STRLEN rlen;   /* length of result string */
84         unsigned char c1, c2, c3;
85         int chunk;
86
87         CODE:
88         sv_utf8_downgrade(sv, FALSE);
89         str = SvPV(sv, rlen); /* SvPV(sv, len) gives warning for signed len */
90         len = (SSize_t)rlen;
91
92         /* set up EOL from the second argument if present, default to "\n" */
93         if (items > 1 && SvOK(ST(1))) {
94             eol = SvPV(ST(1), eollen);
95         } else {
96             eol = "\n";
97             eollen = 1;
98         }
99
100         /* calculate the length of the result */
101         rlen = (len+2) / 3 * 4;  /* encoded bytes */
102         if (rlen) {
103             /* add space for EOL */
104             rlen += ((rlen-1) / MAX_LINE + 1) * eollen;
105         }
106
107         /* allocate a result buffer */
108         RETVAL = newSV(rlen ? rlen : 1);
109         SvPOK_on(RETVAL);       
110         SvCUR_set(RETVAL, rlen);
111         r = SvPVX(RETVAL);
112
113         /* encode */
114         for (chunk=0; len > 0; len -= 3, chunk++) {
115             if (chunk == (MAX_LINE/4)) {
116                 char *c = eol;
117                 char *e = eol + eollen;
118                 while (c < e)
119                     *r++ = *c++;
120                 chunk = 0;
121             }
122             c1 = *str++;
123             c2 = *str++;
124             *r++ = basis_64[c1>>2];
125             *r++ = basis_64[((c1 & 0x3)<< 4) | ((c2 & 0xF0) >> 4)];
126             if (len > 2) {
127                 c3 = *str++;
128                 *r++ = basis_64[((c2 & 0xF) << 2) | ((c3 & 0xC0) >>6)];
129                 *r++ = basis_64[c3 & 0x3F];
130             } else if (len == 2) {
131                 *r++ = basis_64[(c2 & 0xF) << 2];
132                 *r++ = '=';
133             } else { /* len == 1 */
134                 *r++ = '=';
135                 *r++ = '=';
136             }
137         }
138         if (rlen) {
139             /* append eol to the result string */
140             char *c = eol;
141             char *e = eol + eollen;
142             while (c < e)
143                 *r++ = *c++;
144         }
145         *r = '\0';  /* every SV in perl should be NUL-terminated */
146
147         OUTPUT:
148         RETVAL
149
150 SV*
151 decode_base64(sv)
152         SV* sv
153         PROTOTYPE: $
154
155         PREINIT:
156         STRLEN len;
157         register unsigned char *str = (unsigned char*)SvPVbyte(sv, len);
158         unsigned char const* end = str + len;
159         char *r;
160         unsigned char c[4];
161
162         CODE:
163         {
164             /* always enough, but might be too much */
165             STRLEN rlen = len * 3 / 4;
166             RETVAL = newSV(rlen ? rlen : 1);
167         }
168         SvPOK_on(RETVAL);
169         r = SvPVX(RETVAL);
170
171         while (str < end) {
172             int i = 0;
173             do {
174                 unsigned char uc = index_64[NATIVE_TO_ASCII(*str++)];
175                 if (uc != INVALID)
176                     c[i++] = uc;
177
178                 if (str == end) {
179                     if (i < 4) {
180                         if (i && PL_dowarn)
181                             warn("Premature end of base64 data");
182                         if (i < 2) goto thats_it;
183                         if (i == 2) c[2] = EQ;
184                         c[3] = EQ;
185                     }
186                     break;
187                 }
188             } while (i < 4);
189         
190             if (c[0] == EQ || c[1] == EQ) {
191                 if (PL_dowarn) warn("Premature padding of base64 data");
192                 break;
193             }
194             /* printf("c0=%d,c1=%d,c2=%d,c3=%d\n", c[0],c[1],c[2],c[3]);*/
195
196             *r++ = (c[0] << 2) | ((c[1] & 0x30) >> 4);
197
198             if (c[2] == EQ)
199                 break;
200             *r++ = ((c[1] & 0x0F) << 4) | ((c[2] & 0x3C) >> 2);
201
202             if (c[3] == EQ)
203                 break;
204             *r++ = ((c[2] & 0x03) << 6) | c[3];
205         }
206
207       thats_it:
208         SvCUR_set(RETVAL, r - SvPVX(RETVAL));
209         *r = '\0';
210
211         OUTPUT:
212         RETVAL