source/include/charset.h

   1 /*
   2    Unix SMB/CIFS implementation.
   3    charset defines
   4    Copyright (C) Andrew Tridgell 2001
   5    Copyright (C) Jelmer Vernooij 2002
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 /* this defines the charset types used in samba */
  22 typedef enum {CH_UTF16LE=0, CH_UTF16=0, CH_UNIX=1, CH_DISPLAY=2, CH_DOS=3, CH_UTF8=4, CH_UTF16BE=5} charset_t;
  23
  24 #define NUM_CHARSETS 6
  25
  26 /*
  27  *   for each charset we have a function that pushes from that charset to a ucs2
  28  *   buffer, and a function that pulls from ucs2 buffer to that  charset.
  29  *     */
  30
  31 struct charset_functions {
  32         const char *name;
  33         size_t (*pull)(void *, const char **inbuf, size_t *inbytesleft,
  34                                    char **outbuf, size_t *outbytesleft);
  35         size_t (*push)(void *, const char **inbuf, size_t *inbytesleft,
  36                                    char **outbuf, size_t *outbytesleft);
  37         struct charset_functions *prev, *next;
  38 };
  39
  40 /*
  41  * This is auxiliary struct used by source/script/gen-8-bit-gap.sh script
  42  * during generation of an encoding table for charset module
  43  *     */
  44
  45 struct charset_gap_table {
  46   uint16 start;
  47   uint16 end;
  48   int32 idx;
  49 };
  50
  51 /*
  52  *   Define stub for charset module which implements 8-bit encoding with gaps.
  53  *   Encoding tables for such module should be produced from glibc's CHARMAPs
  54  *   using script source/script/gen-8bit-gap.sh
  55  *   CHARSETNAME is CAPITALIZED charset name
  56  *
  57  *     */
  58 #define SMB_GENERATE_CHARSET_MODULE_8_BIT_GAP(CHARSETNAME)                                      \
  59 static size_t CHARSETNAME ## _push(void *cd, const char **inbuf, size_t *inbytesleft,                   \
  60                          char **outbuf, size_t *outbytesleft)                                   \
  61 {                                                                                               \
  62         while (*inbytesleft >= 2 && *outbytesleft >= 1) {                                       \
  63                 int i;                                                                          \
  64                 int done = 0;                                                                   \
  65                                                                                                 \
  66                 uint16 ch = SVAL(*inbuf,0);                                                     \
  67                                                                                                 \
  68                 for (i=0; from_idx[i].start != 0xffff; i++) {                                   \
  69                         if ((from_idx[i].start <= ch) && (from_idx[i].end >= ch)) {             \
  70                                 ((unsigned char*)(*outbuf))[0] = from_ucs2[from_idx[i].idx+ch]; \
  71                                 (*inbytesleft) -= 2;                                            \
  72                                 (*outbytesleft) -= 1;                                           \
  73                                 (*inbuf)  += 2;                                                 \
  74                                 (*outbuf) += 1;                                                 \
  75                                 done = 1;                                                       \
  76                                 break;                                                          \
  77                         }                                                                       \
  78                 }                                                                               \
  79                 if (!done) {                                                                    \
  80                         errno = EINVAL;                                                         \
  81                         return -1;                                                              \
  82                 }                                                                               \
  83                                                                                                 \
  84         }                                                                                       \
  85                                                                                                 \
  86         if (*inbytesleft == 1) {                                                                \
  87                 errno = EINVAL;                                                                 \
  88                 return -1;                                                                      \
  89         }                                                                                       \
  90                                                                                                 \
  91         if (*inbytesleft > 1) {                                                                 \
  92                 errno = E2BIG;                                                                  \
  93                 return -1;                                                                      \
  94         }                                                                                       \
  95                                                                                                 \
  96         return 0;                                                                               \
  97 }                                                                                               \
  98                                                                                                 \
  99 static size_t CHARSETNAME ## _pull(void *cd, const char **inbuf, size_t *inbytesleft,                           \
 100                          char **outbuf, size_t *outbytesleft)                                   \
 101 {                                                                                               \
 102         while (*inbytesleft >= 1 && *outbytesleft >= 2) {                                       \
 103                 *(uint16*)(*outbuf) = to_ucs2[((unsigned char*)(*inbuf))[0]];                   \
 104                 (*inbytesleft)  -= 1;                                                           \
 105                 (*outbytesleft) -= 2;                                                           \
 106                 (*inbuf)  += 1;                                                                 \
 107                 (*outbuf) += 2;                                                                 \
 108         }                                                                                       \
 109                                                                                                 \
 110         if (*inbytesleft > 0) {                                                                 \
 111                 errno = E2BIG;                                                                  \
 112                 return -1;                                                                      \
 113         }                                                                                       \
 114                                                                                                 \
 115         return 0;                                                                               \
 116 }                                                                                               \
 117                                                                                                 \
 118 struct charset_functions CHARSETNAME ## _functions =                                            \
 119                 {#CHARSETNAME, CHARSETNAME ## _pull, CHARSETNAME ## _push};                     \
 120                                                                                                 \
 121 NTSTATUS charset_ ## CHARSETNAME ## _init(void);                                                        \
 122 NTSTATUS charset_ ## CHARSETNAME ## _init(void)                                                 \
 123 {                                                                                               \
 124         return smb_register_charset(& CHARSETNAME ## _functions);                               \
 125 }                                                                                               \
 126