chiark - git - mdw - disorder/blob - lib/base64.c

   1 /*
   2  * This file is part of DisOrder
   3  * Copyright (C) 2005, 2007, 2008 Richard Kettlewell
   4  *
   5  * This program is free software: you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation, either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18 /** @file lib/base64.c
  19  * @brief Support for MIME base64
  20  */
  21
  22 #include "common.h"
  23
  24 #include "mem.h"
  25 #include "base64.h"
  26 #include "vector.h"
  27
  28 static const char mime_base64_table[] =
  29   "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
  30
  31 /** @brief Convert MIME base64
  32  * @param s base64 data
  33  * @param nsp Where to store length of converted data
  34  * @return Decoded data
  35  *
  36  * See <a href="http://tools.ietf.org/html/rfc2045#section-6.8">RFC
  37  * 2045 s6.8</a>.
  38  */
  39 char *mime_base64(const char *s, size_t *nsp) {
  40   return generic_base64(s, nsp, mime_base64_table);
  41 }
  42
  43 /** @brief Convert base64
  44  * @param s base64 data
  45  * @param nsp Where to store length of converted data
  46  * @param table Table of characters to use
  47  * @return Decoded data
  48  *
  49  * @p table should consist of 65 characters.  The first 64 will be used to
  50  * represents the 64 digits and the 65th will be used as padding at the end
  51  * (i.e. the role of '=' in RFC2045 base64).
  52  */
  53 char *generic_base64(const char *s, size_t *nsp, const char *table) {
  54   struct dynstr d;
  55   const char *t;
  56   int b[4], n, c;
  57
  58   dynstr_init(&d);
  59   n = 0;
  60   while((c = (unsigned char)*s++)) {
  61     if(c == table[64]) {
  62       if(n >= 2) {
  63         dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
  64         if(n == 3)
  65           dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
  66       }
  67       break;
  68     } else if((t = strchr(table, c))) {
  69       b[n++] = t - table;
  70       if(n == 4) {
  71         dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
  72         dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
  73         dynstr_append(&d, (b[2] << 6) + b[3]);
  74         n = 0;
  75       }
  76     }
  77   }
  78   if(nsp)
  79     *nsp = d.nvec;
  80   dynstr_terminate(&d);
  81   return d.vec;
  82 }
  83
  84 /** @brief Convert a binary string to MIME base64
  85  * @param s Bytes to convert
  86  * @param ns Number of bytes to convert
  87  * @return Encoded data
  88  *
  89  * This function does not attempt to split up lines.
  90  *
  91  * See <a href="http://tools.ietf.org/html/rfc2045#section-6.8">RFC
  92  * 2045 s6.8</a>.
  93  */
  94 char *mime_to_base64(const uint8_t *s, size_t ns) {
  95   return generic_to_base64(s, ns, mime_base64_table);
  96 }
  97
  98 /** @brief Convert a binary string to base64
  99  * @param s Bytes to convert
 100  * @param ns Number of bytes to convert
 101  * @param table Table of characters to use
 102  * @return Encoded data
 103  *
 104  * This function does not attempt to split up lines.
 105  *
 106  * @p table should consist of 65 characters.  The first 64 will be used to
 107  * represents the 64 digits and the 65th will be used as padding at the end
 108  * (i.e. the role of '=' in RFC2045 base64).
 109  */
 110 char *generic_to_base64(const uint8_t *s, size_t ns, const char *table) {
 111   struct dynstr d[1];
 112
 113   dynstr_init(d);
 114   while(ns >= 3) {
 115     /* Input bytes with output bits: AAAAAABB BBBBCCCC CCDDDDDD */
 116     /* Output bytes with input bits: 000000 001111 111122 222222 */
 117     dynstr_append(d, table[s[0] >> 2]);
 118     dynstr_append(d, table[((s[0] & 3) << 4)
 119                                        + (s[1] >> 4)]);
 120     dynstr_append(d, table[((s[1] & 15) << 2)
 121                                        + (s[2] >> 6)]);
 122     dynstr_append(d, table[s[2] & 63]);
 123     ns -= 3;
 124     s += 3;
 125   }
 126   if(ns > 0) {
 127     dynstr_append(d, table[s[0] >> 2]);
 128     switch(ns) {
 129     case 1:
 130       dynstr_append(d, table[(s[0] & 3) << 4]);
 131       dynstr_append(d, table[64]);
 132       dynstr_append(d, table[64]);
 133       break;
 134     case 2:
 135       dynstr_append(d, table[((s[0] & 3) << 4)
 136                                          + (s[1] >> 4)]);
 137       dynstr_append(d, table[(s[1] & 15) << 2]);
 138       dynstr_append(d, table[64]);
 139       break;
 140     }
 141   }
 142   dynstr_terminate(d);
 143   return d->vec;
 144 }
 145
 146 /*
 147 Local Variables:
 148 c-basic-offset:2
 149 comment-column:40
 150 fill-column:79
 151 End:
 152 */