LCOV - code coverage report
Current view: top level - src/include/mb - pg_wchar.h (source / functions) Hit Total Coverage
Test: PostgreSQL 16beta1 Lines: 19 19 100.0 %
Date: 2023-06-01 12:12:28 Functions: 5 5 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*-------------------------------------------------------------------------
       2             :  *
       3             :  * pg_wchar.h
       4             :  *    multibyte-character support
       5             :  *
       6             :  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
       7             :  * Portions Copyright (c) 1994, Regents of the University of California
       8             :  *
       9             :  * src/include/mb/pg_wchar.h
      10             :  *
      11             :  *  NOTES
      12             :  *      This is used both by the backend and by frontends, but should not be
      13             :  *      included by libpq client programs.  In particular, a libpq client
      14             :  *      should not assume that the encoding IDs used by the version of libpq
      15             :  *      it's linked to match up with the IDs declared here.
      16             :  *
      17             :  *-------------------------------------------------------------------------
      18             :  */
      19             : #ifndef PG_WCHAR_H
      20             : #define PG_WCHAR_H
      21             : 
      22             : #include "port/simd.h"
      23             : 
      24             : /*
      25             :  * The pg_wchar type
      26             :  */
      27             : typedef unsigned int pg_wchar;
      28             : 
      29             : /*
      30             :  * Maximum byte length of multibyte characters in any backend encoding
      31             :  */
      32             : #define MAX_MULTIBYTE_CHAR_LEN  4
      33             : 
      34             : /*
      35             :  * various definitions for EUC
      36             :  */
      37             : #define SS2 0x8e                /* single shift 2 (JIS0201) */
      38             : #define SS3 0x8f                /* single shift 3 (JIS0212) */
      39             : 
      40             : /*
      41             :  * SJIS validation macros
      42             :  */
      43             : #define ISSJISHEAD(c) (((c) >= 0x81 && (c) <= 0x9f) || ((c) >= 0xe0 && (c) <= 0xfc))
      44             : #define ISSJISTAIL(c) (((c) >= 0x40 && (c) <= 0x7e) || ((c) >= 0x80 && (c) <= 0xfc))
      45             : 
      46             : /*----------------------------------------------------
      47             :  * MULE Internal Encoding (MIC)
      48             :  *
      49             :  * This encoding follows the design used within XEmacs; it is meant to
      50             :  * subsume many externally-defined character sets.  Each character includes
      51             :  * identification of the character set it belongs to, so the encoding is
      52             :  * general but somewhat bulky.
      53             :  *
      54             :  * Currently PostgreSQL supports 5 types of MULE character sets:
      55             :  *
      56             :  * 1) 1-byte ASCII characters.  Each byte is below 0x80.
      57             :  *
      58             :  * 2) "Official" single byte charsets such as ISO-8859-1 (Latin1).
      59             :  *    Each MULE character consists of 2 bytes: LC1 + C1, where LC1 is
      60             :  *    an identifier for the charset (in the range 0x81 to 0x8d) and C1
      61             :  *    is the character code (in the range 0xa0 to 0xff).
      62             :  *
      63             :  * 3) "Private" single byte charsets such as SISHENG.  Each MULE
      64             :  *    character consists of 3 bytes: LCPRV1 + LC12 + C1, where LCPRV1
      65             :  *    is a private-charset flag, LC12 is an identifier for the charset,
      66             :  *    and C1 is the character code (in the range 0xa0 to 0xff).
      67             :  *    LCPRV1 is either 0x9a (if LC12 is in the range 0xa0 to 0xdf)
      68             :  *    or 0x9b (if LC12 is in the range 0xe0 to 0xef).
      69             :  *
      70             :  * 4) "Official" multibyte charsets such as JIS X0208.  Each MULE
      71             :  *    character consists of 3 bytes: LC2 + C1 + C2, where LC2 is
      72             :  *    an identifier for the charset (in the range 0x90 to 0x99) and C1
      73             :  *    and C2 form the character code (each in the range 0xa0 to 0xff).
      74             :  *
      75             :  * 5) "Private" multibyte charsets such as CNS 11643-1992 Plane 3.
      76             :  *    Each MULE character consists of 4 bytes: LCPRV2 + LC22 + C1 + C2,
      77             :  *    where LCPRV2 is a private-charset flag, LC22 is an identifier for
      78             :  *    the charset, and C1 and C2 form the character code (each in the range
      79             :  *    0xa0 to 0xff).  LCPRV2 is either 0x9c (if LC22 is in the range 0xf0
      80             :  *    to 0xf4) or 0x9d (if LC22 is in the range 0xf5 to 0xfe).
      81             :  *
      82             :  * "Official" encodings are those that have been assigned code numbers by
      83             :  * the XEmacs project; "private" encodings have Postgres-specific charset
      84             :  * identifiers.
      85             :  *
      86             :  * See the "XEmacs Internals Manual", available at http://www.xemacs.org,
      87             :  * for more details.  Note that for historical reasons, Postgres'
      88             :  * private-charset flag values do not match what XEmacs says they should be,
      89             :  * so this isn't really exactly MULE (not that private charsets would be
      90             :  * interoperable anyway).
      91             :  *
      92             :  * Note that XEmacs's implementation is different from what emacs does.
      93             :  * We follow emacs's implementation, rather than XEmacs's.
      94             :  *----------------------------------------------------
      95             :  */
      96             : 
      97             : /*
      98             :  * Charset identifiers (also called "leading bytes" in the MULE documentation)
      99             :  */
     100             : 
     101             : /*
     102             :  * Charset IDs for official single byte encodings (0x81-0x8e)
     103             :  */
     104             : #define LC_ISO8859_1        0x81    /* ISO8859 Latin 1 */
     105             : #define LC_ISO8859_2        0x82    /* ISO8859 Latin 2 */
     106             : #define LC_ISO8859_3        0x83    /* ISO8859 Latin 3 */
     107             : #define LC_ISO8859_4        0x84    /* ISO8859 Latin 4 */
     108             : #define LC_TIS620           0x85    /* Thai (not supported yet) */
     109             : #define LC_ISO8859_7        0x86    /* Greek (not supported yet) */
     110             : #define LC_ISO8859_6        0x87    /* Arabic (not supported yet) */
     111             : #define LC_ISO8859_8        0x88    /* Hebrew (not supported yet) */
     112             : #define LC_JISX0201K        0x89    /* Japanese 1 byte kana */
     113             : #define LC_JISX0201R        0x8a    /* Japanese 1 byte Roman */
     114             : /* Note that 0x8b seems to be unused as of Emacs 20.7.
     115             :  * However, there might be a chance that 0x8b could be used
     116             :  * in later versions of Emacs.
     117             :  */
     118             : #define LC_KOI8_R           0x8b    /* Cyrillic KOI8-R */
     119             : #define LC_ISO8859_5        0x8c    /* ISO8859 Cyrillic */
     120             : #define LC_ISO8859_9        0x8d    /* ISO8859 Latin 5 (not supported yet) */
     121             : #define LC_ISO8859_15       0x8e    /* ISO8859 Latin 15 (not supported yet) */
     122             : /* #define CONTROL_1        0x8f    control characters (unused) */
     123             : 
     124             : /* Is a leading byte for "official" single byte encodings? */
     125             : #define IS_LC1(c)   ((unsigned char)(c) >= 0x81 && (unsigned char)(c) <= 0x8d)
     126             : 
     127             : /*
     128             :  * Charset IDs for official multibyte encodings (0x90-0x99)
     129             :  * 0x9a-0x9d are free. 0x9e and 0x9f are reserved.
     130             :  */
     131             : #define LC_JISX0208_1978    0x90    /* Japanese Kanji, old JIS (not supported) */
     132             : #define LC_GB2312_80        0x91    /* Chinese */
     133             : #define LC_JISX0208         0x92    /* Japanese Kanji (JIS X 0208) */
     134             : #define LC_KS5601           0x93    /* Korean */
     135             : #define LC_JISX0212         0x94    /* Japanese Kanji (JIS X 0212) */
     136             : #define LC_CNS11643_1       0x95    /* CNS 11643-1992 Plane 1 */
     137             : #define LC_CNS11643_2       0x96    /* CNS 11643-1992 Plane 2 */
     138             : #define LC_JISX0213_1       0x97    /* Japanese Kanji (JIS X 0213 Plane 1)
     139             :                                      * (not supported) */
     140             : #define LC_BIG5_1           0x98    /* Plane 1 Chinese traditional (not
     141             :                                      * supported) */
     142             : #define LC_BIG5_2           0x99    /* Plane 1 Chinese traditional (not
     143             :                                      * supported) */
     144             : 
     145             : /* Is a leading byte for "official" multibyte encodings? */
     146             : #define IS_LC2(c)   ((unsigned char)(c) >= 0x90 && (unsigned char)(c) <= 0x99)
     147             : 
     148             : /*
     149             :  * Postgres-specific prefix bytes for "private" single byte encodings
     150             :  * (According to the MULE docs, we should be using 0x9e for this)
     151             :  */
     152             : #define LCPRV1_A        0x9a
     153             : #define LCPRV1_B        0x9b
     154             : #define IS_LCPRV1(c)    ((unsigned char)(c) == LCPRV1_A || (unsigned char)(c) == LCPRV1_B)
     155             : #define IS_LCPRV1_A_RANGE(c)    \
     156             :     ((unsigned char)(c) >= 0xa0 && (unsigned char)(c) <= 0xdf)
     157             : #define IS_LCPRV1_B_RANGE(c)    \
     158             :     ((unsigned char)(c) >= 0xe0 && (unsigned char)(c) <= 0xef)
     159             : 
     160             : /*
     161             :  * Postgres-specific prefix bytes for "private" multibyte encodings
     162             :  * (According to the MULE docs, we should be using 0x9f for this)
     163             :  */
     164             : #define LCPRV2_A        0x9c
     165             : #define LCPRV2_B        0x9d
     166             : #define IS_LCPRV2(c)    ((unsigned char)(c) == LCPRV2_A || (unsigned char)(c) == LCPRV2_B)
     167             : #define IS_LCPRV2_A_RANGE(c)    \
     168             :     ((unsigned char)(c) >= 0xf0 && (unsigned char)(c) <= 0xf4)
     169             : #define IS_LCPRV2_B_RANGE(c)    \
     170             :     ((unsigned char)(c) >= 0xf5 && (unsigned char)(c) <= 0xfe)
     171             : 
     172             : /*
     173             :  * Charset IDs for private single byte encodings (0xa0-0xef)
     174             :  */
     175             : #define LC_SISHENG          0xa0    /* Chinese SiSheng characters for
     176             :                                      * PinYin/ZhuYin (not supported) */
     177             : #define LC_IPA              0xa1    /* IPA (International Phonetic
     178             :                                      * Association) (not supported) */
     179             : #define LC_VISCII_LOWER     0xa2    /* Vietnamese VISCII1.1 lower-case (not
     180             :                                      * supported) */
     181             : #define LC_VISCII_UPPER     0xa3    /* Vietnamese VISCII1.1 upper-case (not
     182             :                                      * supported) */
     183             : #define LC_ARABIC_DIGIT     0xa4    /* Arabic digit (not supported) */
     184             : #define LC_ARABIC_1_COLUMN  0xa5    /* Arabic 1-column (not supported) */
     185             : #define LC_ASCII_RIGHT_TO_LEFT  0xa6    /* ASCII (left half of ISO8859-1) with
     186             :                                          * right-to-left direction (not
     187             :                                          * supported) */
     188             : #define LC_LAO              0xa7    /* Lao characters (ISO10646 0E80..0EDF)
     189             :                                      * (not supported) */
     190             : #define LC_ARABIC_2_COLUMN  0xa8    /* Arabic 1-column (not supported) */
     191             : 
     192             : /*
     193             :  * Charset IDs for private multibyte encodings (0xf0-0xff)
     194             :  */
     195             : #define LC_INDIAN_1_COLUMN  0xf0    /* Indian charset for 1-column width
     196             :                                      * glyphs (not supported) */
     197             : #define LC_TIBETAN_1_COLUMN 0xf1    /* Tibetan 1-column width glyphs (not
     198             :                                      * supported) */
     199             : #define LC_UNICODE_SUBSET_2 0xf2    /* Unicode characters of the range
     200             :                                      * U+2500..U+33FF. (not supported) */
     201             : #define LC_UNICODE_SUBSET_3 0xf3    /* Unicode characters of the range
     202             :                                      * U+E000..U+FFFF. (not supported) */
     203             : #define LC_UNICODE_SUBSET   0xf4    /* Unicode characters of the range
     204             :                                      * U+0100..U+24FF. (not supported) */
     205             : #define LC_ETHIOPIC         0xf5    /* Ethiopic characters (not supported) */
     206             : #define LC_CNS11643_3       0xf6    /* CNS 11643-1992 Plane 3 */
     207             : #define LC_CNS11643_4       0xf7    /* CNS 11643-1992 Plane 4 */
     208             : #define LC_CNS11643_5       0xf8    /* CNS 11643-1992 Plane 5 */
     209             : #define LC_CNS11643_6       0xf9    /* CNS 11643-1992 Plane 6 */
     210             : #define LC_CNS11643_7       0xfa    /* CNS 11643-1992 Plane 7 */
     211             : #define LC_INDIAN_2_COLUMN  0xfb    /* Indian charset for 2-column width
     212             :                                      * glyphs (not supported) */
     213             : #define LC_TIBETAN          0xfc    /* Tibetan (not supported) */
     214             : /* #define FREE             0xfd    free (unused) */
     215             : /* #define FREE             0xfe    free (unused) */
     216             : /* #define FREE             0xff    free (unused) */
     217             : 
     218             : /*----------------------------------------------------
     219             :  * end of MULE stuff
     220             :  *----------------------------------------------------
     221             :  */
     222             : 
     223             : /*
     224             :  * PostgreSQL encoding identifiers
     225             :  *
     226             :  * WARNING: the order of this enum must be same as order of entries
     227             :  *          in the pg_enc2name_tbl[] array (in src/common/encnames.c), and
     228             :  *          in the pg_wchar_table[] array (in src/common/wchar.c)!
     229             :  *
     230             :  *          If you add some encoding don't forget to check
     231             :  *          PG_ENCODING_BE_LAST macro.
     232             :  *
     233             :  * PG_SQL_ASCII is default encoding and must be = 0.
     234             :  *
     235             :  * XXX  We must avoid renumbering any backend encoding until libpq's major
     236             :  * version number is increased beyond 5; it turns out that the backend
     237             :  * encoding IDs are effectively part of libpq's ABI as far as 8.2 initdb and
     238             :  * psql are concerned.
     239             :  */
     240             : typedef enum pg_enc
     241             : {
     242             :     PG_SQL_ASCII = 0,           /* SQL/ASCII */
     243             :     PG_EUC_JP,                  /* EUC for Japanese */
     244             :     PG_EUC_CN,                  /* EUC for Chinese */
     245             :     PG_EUC_KR,                  /* EUC for Korean */
     246             :     PG_EUC_TW,                  /* EUC for Taiwan */
     247             :     PG_EUC_JIS_2004,            /* EUC-JIS-2004 */
     248             :     PG_UTF8,                    /* Unicode UTF8 */
     249             :     PG_MULE_INTERNAL,           /* Mule internal code */
     250             :     PG_LATIN1,                  /* ISO-8859-1 Latin 1 */
     251             :     PG_LATIN2,                  /* ISO-8859-2 Latin 2 */
     252             :     PG_LATIN3,                  /* ISO-8859-3 Latin 3 */
     253             :     PG_LATIN4,                  /* ISO-8859-4 Latin 4 */
     254             :     PG_LATIN5,                  /* ISO-8859-9 Latin 5 */
     255             :     PG_LATIN6,                  /* ISO-8859-10 Latin6 */
     256             :     PG_LATIN7,                  /* ISO-8859-13 Latin7 */
     257             :     PG_LATIN8,                  /* ISO-8859-14 Latin8 */
     258             :     PG_LATIN9,                  /* ISO-8859-15 Latin9 */
     259             :     PG_LATIN10,                 /* ISO-8859-16 Latin10 */
     260             :     PG_WIN1256,                 /* windows-1256 */
     261             :     PG_WIN1258,                 /* Windows-1258 */
     262             :     PG_WIN866,                  /* (MS-DOS CP866) */
     263             :     PG_WIN874,                  /* windows-874 */
     264             :     PG_KOI8R,                   /* KOI8-R */
     265             :     PG_WIN1251,                 /* windows-1251 */
     266             :     PG_WIN1252,                 /* windows-1252 */
     267             :     PG_ISO_8859_5,              /* ISO-8859-5 */
     268             :     PG_ISO_8859_6,              /* ISO-8859-6 */
     269             :     PG_ISO_8859_7,              /* ISO-8859-7 */
     270             :     PG_ISO_8859_8,              /* ISO-8859-8 */
     271             :     PG_WIN1250,                 /* windows-1250 */
     272             :     PG_WIN1253,                 /* windows-1253 */
     273             :     PG_WIN1254,                 /* windows-1254 */
     274             :     PG_WIN1255,                 /* windows-1255 */
     275             :     PG_WIN1257,                 /* windows-1257 */
     276             :     PG_KOI8U,                   /* KOI8-U */
     277             :     /* PG_ENCODING_BE_LAST points to the above entry */
     278             : 
     279             :     /* followings are for client encoding only */
     280             :     PG_SJIS,                    /* Shift JIS (Windows-932) */
     281             :     PG_BIG5,                    /* Big5 (Windows-950) */
     282             :     PG_GBK,                     /* GBK (Windows-936) */
     283             :     PG_UHC,                     /* UHC (Windows-949) */
     284             :     PG_GB18030,                 /* GB18030 */
     285             :     PG_JOHAB,                   /* EUC for Korean JOHAB */
     286             :     PG_SHIFT_JIS_2004,          /* Shift-JIS-2004 */
     287             :     _PG_LAST_ENCODING_          /* mark only */
     288             : 
     289             : } pg_enc;
     290             : 
     291             : #define PG_ENCODING_BE_LAST PG_KOI8U
     292             : 
     293             : /*
     294             :  * Please use these tests before access to pg_enc2name_tbl[]
     295             :  * or to other places...
     296             :  */
     297             : #define PG_VALID_BE_ENCODING(_enc) \
     298             :         ((_enc) >= 0 && (_enc) <= PG_ENCODING_BE_LAST)
     299             : 
     300             : #define PG_ENCODING_IS_CLIENT_ONLY(_enc) \
     301             :         ((_enc) > PG_ENCODING_BE_LAST && (_enc) < _PG_LAST_ENCODING_)
     302             : 
     303             : #define PG_VALID_ENCODING(_enc) \
     304             :         ((_enc) >= 0 && (_enc) < _PG_LAST_ENCODING_)
     305             : 
     306             : /* On FE are possible all encodings */
     307             : #define PG_VALID_FE_ENCODING(_enc)  PG_VALID_ENCODING(_enc)
     308             : 
     309             : /*
     310             :  * When converting strings between different encodings, we assume that space
     311             :  * for converted result is 4-to-1 growth in the worst case.  The rate for
     312             :  * currently supported encoding pairs are within 3 (SJIS JIS X0201 half width
     313             :  * kana -> UTF8 is the worst case).  So "4" should be enough for the moment.
     314             :  *
     315             :  * Note that this is not the same as the maximum character width in any
     316             :  * particular encoding.
     317             :  */
     318             : #define MAX_CONVERSION_GROWTH  4
     319             : 
     320             : /*
     321             :  * Maximum byte length of a string that's required in any encoding to convert
     322             :  * at least one character to any other encoding.  In other words, if you feed
     323             :  * MAX_CONVERSION_INPUT_LENGTH bytes to any encoding conversion function, it
     324             :  * is guaranteed to be able to convert something without needing more input
     325             :  * (assuming the input is valid).
     326             :  *
     327             :  * Currently, the maximum case is the conversion UTF8 -> SJIS JIS X0201 half
     328             :  * width kana, where a pair of UTF-8 characters is converted into a single
     329             :  * SHIFT_JIS_2004 character (the reverse of the worst case for
     330             :  * MAX_CONVERSION_GROWTH).  It needs 6 bytes of input.  In theory, a
     331             :  * user-defined conversion function might have more complicated cases, although
     332             :  * for the reverse mapping you would probably also need to bump up
     333             :  * MAX_CONVERSION_GROWTH.  But there is no need to be stingy here, so make it
     334             :  * generous.
     335             :  */
     336             : #define MAX_CONVERSION_INPUT_LENGTH 16
     337             : 
     338             : /*
     339             :  * Maximum byte length of the string equivalent to any one Unicode code point,
     340             :  * in any backend encoding.  The current value assumes that a 4-byte UTF-8
     341             :  * character might expand by MAX_CONVERSION_GROWTH, which is a huge
     342             :  * overestimate.  But in current usage we don't allocate large multiples of
     343             :  * this, so there's little point in being stingy.
     344             :  */
     345             : #define MAX_UNICODE_EQUIVALENT_STRING   16
     346             : 
     347             : /*
     348             :  * Table for mapping an encoding number to official encoding name and
     349             :  * possibly other subsidiary data.  Be careful to check encoding number
     350             :  * before accessing a table entry!
     351             :  *
     352             :  * if (PG_VALID_ENCODING(encoding))
     353             :  *      pg_enc2name_tbl[ encoding ];
     354             :  */
     355             : typedef struct pg_enc2name
     356             : {
     357             :     const char *name;
     358             :     pg_enc      encoding;
     359             : #ifdef WIN32
     360             :     unsigned    codepage;       /* codepage for WIN32 */
     361             : #endif
     362             : } pg_enc2name;
     363             : 
     364             : extern PGDLLIMPORT const pg_enc2name pg_enc2name_tbl[];
     365             : 
     366             : /*
     367             :  * Encoding names for gettext
     368             :  */
     369             : typedef struct pg_enc2gettext
     370             : {
     371             :     pg_enc      encoding;
     372             :     const char *name;
     373             : } pg_enc2gettext;
     374             : 
     375             : extern PGDLLIMPORT const pg_enc2gettext pg_enc2gettext_tbl[];
     376             : 
     377             : /*
     378             :  * pg_wchar stuff
     379             :  */
     380             : typedef int (*mb2wchar_with_len_converter) (const unsigned char *from,
     381             :                                             pg_wchar *to,
     382             :                                             int len);
     383             : 
     384             : typedef int (*wchar2mb_with_len_converter) (const pg_wchar *from,
     385             :                                             unsigned char *to,
     386             :                                             int len);
     387             : 
     388             : typedef int (*mblen_converter) (const unsigned char *mbstr);
     389             : 
     390             : typedef int (*mbdisplaylen_converter) (const unsigned char *mbstr);
     391             : 
     392             : typedef bool (*mbcharacter_incrementer) (unsigned char *mbstr, int len);
     393             : 
     394             : typedef int (*mbchar_verifier) (const unsigned char *mbstr, int len);
     395             : 
     396             : typedef int (*mbstr_verifier) (const unsigned char *mbstr, int len);
     397             : 
     398             : typedef struct
     399             : {
     400             :     mb2wchar_with_len_converter mb2wchar_with_len;  /* convert a multibyte
     401             :                                                      * string to a wchar */
     402             :     wchar2mb_with_len_converter wchar2mb_with_len;  /* convert a wchar string
     403             :                                                      * to a multibyte */
     404             :     mblen_converter mblen;      /* get byte length of a char */
     405             :     mbdisplaylen_converter dsplen;  /* get display width of a char */
     406             :     mbchar_verifier mbverifychar;   /* verify multibyte character */
     407             :     mbstr_verifier mbverifystr; /* verify multibyte string */
     408             :     int         maxmblen;       /* max bytes for a char in this encoding */
     409             : } pg_wchar_tbl;
     410             : 
     411             : extern PGDLLIMPORT const pg_wchar_tbl pg_wchar_table[];
     412             : 
     413             : /*
     414             :  * Data structures for conversions between UTF-8 and other encodings
     415             :  * (UtfToLocal() and LocalToUtf()).  In these data structures, characters of
     416             :  * either encoding are represented by uint32 words; hence we can only support
     417             :  * characters up to 4 bytes long.  For example, the byte sequence 0xC2 0x89
     418             :  * would be represented by 0x0000C289, and 0xE8 0xA2 0xB4 by 0x00E8A2B4.
     419             :  *
     420             :  * There are three possible ways a character can be mapped:
     421             :  *
     422             :  * 1. Using a radix tree, from source to destination code.
     423             :  * 2. Using a sorted array of source -> destination code pairs. This
     424             :  *    method is used for "combining" characters. There are so few of
     425             :  *    them that building a radix tree would be wasteful.
     426             :  * 3. Using a conversion function.
     427             :  */
     428             : 
     429             : /*
     430             :  * Radix tree for character conversion.
     431             :  *
     432             :  * Logically, this is actually four different radix trees, for 1-byte,
     433             :  * 2-byte, 3-byte and 4-byte inputs. The 1-byte tree is a simple lookup
     434             :  * table from source to target code. The 2-byte tree consists of two levels:
     435             :  * one lookup table for the first byte, where the value in the lookup table
     436             :  * points to a lookup table for the second byte. And so on.
     437             :  *
     438             :  * Physically, all the trees are stored in one big array, in 'chars16' or
     439             :  * 'chars32', depending on the maximum value that needs to be represented. For
     440             :  * each level in each tree, we also store lower and upper bound of allowed
     441             :  * values - values outside those bounds are considered invalid, and are left
     442             :  * out of the tables.
     443             :  *
     444             :  * In the intermediate levels of the trees, the values stored are offsets
     445             :  * into the chars[16|32] array.
     446             :  *
     447             :  * In the beginning of the chars[16|32] array, there is always a number of
     448             :  * zeros, so that you safely follow an index from an intermediate table
     449             :  * without explicitly checking for a zero. Following a zero any number of
     450             :  * times will always bring you to the dummy, all-zeros table in the
     451             :  * beginning. This helps to shave some cycles when looking up values.
     452             :  */
     453             : typedef struct
     454             : {
     455             :     /*
     456             :      * Array containing all the values. Only one of chars16 or chars32 is
     457             :      * used, depending on how wide the values we need to represent are.
     458             :      */
     459             :     const uint16 *chars16;
     460             :     const uint32 *chars32;
     461             : 
     462             :     /* Radix tree for 1-byte inputs */
     463             :     uint32      b1root;         /* offset of table in the chars[16|32] array */
     464             :     uint8       b1_lower;       /* min allowed value for a single byte input */
     465             :     uint8       b1_upper;       /* max allowed value for a single byte input */
     466             : 
     467             :     /* Radix tree for 2-byte inputs */
     468             :     uint32      b2root;         /* offset of 1st byte's table */
     469             :     uint8       b2_1_lower;     /* min/max allowed value for 1st input byte */
     470             :     uint8       b2_1_upper;
     471             :     uint8       b2_2_lower;     /* min/max allowed value for 2nd input byte */
     472             :     uint8       b2_2_upper;
     473             : 
     474             :     /* Radix tree for 3-byte inputs */
     475             :     uint32      b3root;         /* offset of 1st byte's table */
     476             :     uint8       b3_1_lower;     /* min/max allowed value for 1st input byte */
     477             :     uint8       b3_1_upper;
     478             :     uint8       b3_2_lower;     /* min/max allowed value for 2nd input byte */
     479             :     uint8       b3_2_upper;
     480             :     uint8       b3_3_lower;     /* min/max allowed value for 3rd input byte */
     481             :     uint8       b3_3_upper;
     482             : 
     483             :     /* Radix tree for 4-byte inputs */
     484             :     uint32      b4root;         /* offset of 1st byte's table */
     485             :     uint8       b4_1_lower;     /* min/max allowed value for 1st input byte */
     486             :     uint8       b4_1_upper;
     487             :     uint8       b4_2_lower;     /* min/max allowed value for 2nd input byte */
     488             :     uint8       b4_2_upper;
     489             :     uint8       b4_3_lower;     /* min/max allowed value for 3rd input byte */
     490             :     uint8       b4_3_upper;
     491             :     uint8       b4_4_lower;     /* min/max allowed value for 4th input byte */
     492             :     uint8       b4_4_upper;
     493             : 
     494             : } pg_mb_radix_tree;
     495             : 
     496             : /*
     497             :  * UTF-8 to local code conversion map (for combined characters)
     498             :  */
     499             : typedef struct
     500             : {
     501             :     uint32      utf1;           /* UTF-8 code 1 */
     502             :     uint32      utf2;           /* UTF-8 code 2 */
     503             :     uint32      code;           /* local code */
     504             : } pg_utf_to_local_combined;
     505             : 
     506             : /*
     507             :  * local code to UTF-8 conversion map (for combined characters)
     508             :  */
     509             : typedef struct
     510             : {
     511             :     uint32      code;           /* local code */
     512             :     uint32      utf1;           /* UTF-8 code 1 */
     513             :     uint32      utf2;           /* UTF-8 code 2 */
     514             : } pg_local_to_utf_combined;
     515             : 
     516             : /*
     517             :  * callback function for algorithmic encoding conversions (in either direction)
     518             :  *
     519             :  * if function returns zero, it does not know how to convert the code
     520             :  */
     521             : typedef uint32 (*utf_local_conversion_func) (uint32 code);
     522             : 
     523             : /*
     524             :  * Support macro for encoding conversion functions to validate their
     525             :  * arguments.  (This could be made more compact if we included fmgr.h
     526             :  * here, but we don't want to do that because this header file is also
     527             :  * used by frontends.)
     528             :  */
     529             : #define CHECK_ENCODING_CONVERSION_ARGS(srcencoding,destencoding) \
     530             :     check_encoding_conversion_args(PG_GETARG_INT32(0), \
     531             :                                    PG_GETARG_INT32(1), \
     532             :                                    PG_GETARG_INT32(4), \
     533             :                                    (srcencoding), \
     534             :                                    (destencoding))
     535             : 
     536             : 
     537             : /*
     538             :  * Some handy functions for Unicode-specific tests.
     539             :  */
     540             : static inline bool
     541        1284 : is_valid_unicode_codepoint(pg_wchar c)
     542             : {
     543        1284 :     return (c > 0 && c <= 0x10FFFF);
     544             : }
     545             : 
     546             : static inline bool
     547         988 : is_utf16_surrogate_first(pg_wchar c)
     548             : {
     549         988 :     return (c >= 0xD800 && c <= 0xDBFF);
     550             : }
     551             : 
     552             : static inline bool
     553         874 : is_utf16_surrogate_second(pg_wchar c)
     554             : {
     555         874 :     return (c >= 0xDC00 && c <= 0xDFFF);
     556             : }
     557             : 
     558             : static inline pg_wchar
     559          60 : surrogate_pair_to_codepoint(pg_wchar first, pg_wchar second)
     560             : {
     561          60 :     return ((first & 0x3FF) << 10) + 0x10000 + (second & 0x3FF);
     562             : }
     563             : 
     564             : 
     565             : /*
     566             :  * These functions are considered part of libpq's exported API and
     567             :  * are also declared in libpq-fe.h.
     568             :  */
     569             : extern int  pg_char_to_encoding(const char *name);
     570             : extern const char *pg_encoding_to_char(int encoding);
     571             : extern int  pg_valid_server_encoding_id(int encoding);
     572             : 
     573             : /*
     574             :  * These functions are available to frontend code that links with libpgcommon
     575             :  * (in addition to the ones just above).  The constant tables declared
     576             :  * earlier in this file are also available from libpgcommon.
     577             :  */
     578             : extern int  pg_encoding_mblen(int encoding, const char *mbstr);
     579             : extern int  pg_encoding_mblen_bounded(int encoding, const char *mbstr);
     580             : extern int  pg_encoding_dsplen(int encoding, const char *mbstr);
     581             : extern int  pg_encoding_verifymbchar(int encoding, const char *mbstr, int len);
     582             : extern int  pg_encoding_verifymbstr(int encoding, const char *mbstr, int len);
     583             : extern int  pg_encoding_max_length(int encoding);
     584             : extern int  pg_valid_client_encoding(const char *name);
     585             : extern int  pg_valid_server_encoding(const char *name);
     586             : extern bool is_encoding_supported_by_icu(int encoding);
     587             : extern const char *get_encoding_name_for_icu(int encoding);
     588             : 
     589             : extern unsigned char *unicode_to_utf8(pg_wchar c, unsigned char *utf8string);
     590             : extern pg_wchar utf8_to_unicode(const unsigned char *c);
     591             : extern bool pg_utf8_islegal(const unsigned char *source, int length);
     592             : extern int  pg_utf_mblen(const unsigned char *s);
     593             : extern int  pg_mule_mblen(const unsigned char *s);
     594             : 
     595             : /*
     596             :  * The remaining functions are backend-only.
     597             :  */
     598             : extern int  pg_mb2wchar(const char *from, pg_wchar *to);
     599             : extern int  pg_mb2wchar_with_len(const char *from, pg_wchar *to, int len);
     600             : extern int  pg_encoding_mb2wchar_with_len(int encoding,
     601             :                                           const char *from, pg_wchar *to, int len);
     602             : extern int  pg_wchar2mb(const pg_wchar *from, char *to);
     603             : extern int  pg_wchar2mb_with_len(const pg_wchar *from, char *to, int len);
     604             : extern int  pg_encoding_wchar2mb_with_len(int encoding,
     605             :                                           const pg_wchar *from, char *to, int len);
     606             : extern int  pg_char_and_wchar_strcmp(const char *s1, const pg_wchar *s2);
     607             : extern int  pg_wchar_strncmp(const pg_wchar *s1, const pg_wchar *s2, size_t n);
     608             : extern int  pg_char_and_wchar_strncmp(const char *s1, const pg_wchar *s2, size_t n);
     609             : extern size_t pg_wchar_strlen(const pg_wchar *str);
     610             : extern int  pg_mblen(const char *mbstr);
     611             : extern int  pg_dsplen(const char *mbstr);
     612             : extern int  pg_mbstrlen(const char *mbstr);
     613             : extern int  pg_mbstrlen_with_len(const char *mbstr, int limit);
     614             : extern int  pg_mbcliplen(const char *mbstr, int len, int limit);
     615             : extern int  pg_encoding_mbcliplen(int encoding, const char *mbstr,
     616             :                                   int len, int limit);
     617             : extern int  pg_mbcharcliplen(const char *mbstr, int len, int limit);
     618             : extern int  pg_database_encoding_max_length(void);
     619             : extern mbcharacter_incrementer pg_database_encoding_character_incrementer(void);
     620             : 
     621             : extern int  PrepareClientEncoding(int encoding);
     622             : extern int  SetClientEncoding(int encoding);
     623             : extern void InitializeClientEncoding(void);
     624             : extern int  pg_get_client_encoding(void);
     625             : extern const char *pg_get_client_encoding_name(void);
     626             : 
     627             : extern void SetDatabaseEncoding(int encoding);
     628             : extern int  GetDatabaseEncoding(void);
     629             : extern const char *GetDatabaseEncodingName(void);
     630             : extern void SetMessageEncoding(int encoding);
     631             : extern int  GetMessageEncoding(void);
     632             : 
     633             : #ifdef ENABLE_NLS
     634             : extern int  pg_bind_textdomain_codeset(const char *domainname);
     635             : #endif
     636             : 
     637             : extern unsigned char *pg_do_encoding_conversion(unsigned char *src, int len,
     638             :                                                 int src_encoding,
     639             :                                                 int dest_encoding);
     640             : extern int  pg_do_encoding_conversion_buf(Oid proc,
     641             :                                           int src_encoding,
     642             :                                           int dest_encoding,
     643             :                                           unsigned char *src, int srclen,
     644             :                                           unsigned char *dest, int destlen,
     645             :                                           bool noError);
     646             : 
     647             : extern char *pg_client_to_server(const char *s, int len);
     648             : extern char *pg_server_to_client(const char *s, int len);
     649             : extern char *pg_any_to_server(const char *s, int len, int encoding);
     650             : extern char *pg_server_to_any(const char *s, int len, int encoding);
     651             : 
     652             : extern void pg_unicode_to_server(pg_wchar c, unsigned char *s);
     653             : extern bool pg_unicode_to_server_noerror(pg_wchar c, unsigned char *s);
     654             : 
     655             : extern unsigned short BIG5toCNS(unsigned short big5, unsigned char *lc);
     656             : extern unsigned short CNStoBIG5(unsigned short cns, unsigned char lc);
     657             : 
     658             : extern int  UtfToLocal(const unsigned char *utf, int len,
     659             :                        unsigned char *iso,
     660             :                        const pg_mb_radix_tree *map,
     661             :                        const pg_utf_to_local_combined *cmap, int cmapsize,
     662             :                        utf_local_conversion_func conv_func,
     663             :                        int encoding, bool noError);
     664             : extern int  LocalToUtf(const unsigned char *iso, int len,
     665             :                        unsigned char *utf,
     666             :                        const pg_mb_radix_tree *map,
     667             :                        const pg_local_to_utf_combined *cmap, int cmapsize,
     668             :                        utf_local_conversion_func conv_func,
     669             :                        int encoding, bool noError);
     670             : 
     671             : extern bool pg_verifymbstr(const char *mbstr, int len, bool noError);
     672             : extern bool pg_verify_mbstr(int encoding, const char *mbstr, int len,
     673             :                             bool noError);
     674             : extern int  pg_verify_mbstr_len(int encoding, const char *mbstr, int len,
     675             :                                 bool noError);
     676             : 
     677             : extern void check_encoding_conversion_args(int src_encoding,
     678             :                                            int dest_encoding,
     679             :                                            int len,
     680             :                                            int expected_src_encoding,
     681             :                                            int expected_dest_encoding);
     682             : 
     683             : extern void report_invalid_encoding(int encoding, const char *mbstr, int len) pg_attribute_noreturn();
     684             : extern void report_untranslatable_char(int src_encoding, int dest_encoding,
     685             :                                        const char *mbstr, int len) pg_attribute_noreturn();
     686             : 
     687             : extern int  local2local(const unsigned char *l, unsigned char *p, int len,
     688             :                         int src_encoding, int dest_encoding,
     689             :                         const unsigned char *tab, bool noError);
     690             : extern int  latin2mic(const unsigned char *l, unsigned char *p, int len,
     691             :                       int lc, int encoding, bool noError);
     692             : extern int  mic2latin(const unsigned char *mic, unsigned char *p, int len,
     693             :                       int lc, int encoding, bool noError);
     694             : extern int  latin2mic_with_table(const unsigned char *l, unsigned char *p,
     695             :                                  int len, int lc, int encoding,
     696             :                                  const unsigned char *tab, bool noError);
     697             : extern int  mic2latin_with_table(const unsigned char *mic, unsigned char *p,
     698             :                                  int len, int lc, int encoding,
     699             :                                  const unsigned char *tab, bool noError);
     700             : 
     701             : #ifdef WIN32
     702             : extern WCHAR *pgwin32_message_to_UTF16(const char *str, int len, int *utf16len);
     703             : #endif
     704             : 
     705             : 
     706             : /*
     707             :  * Verify a chunk of bytes for valid ASCII.
     708             :  *
     709             :  * Returns false if the input contains any zero bytes or bytes with the
     710             :  * high-bit set. Input len must be a multiple of the chunk size (8 or 16).
     711             :  */
     712             : static inline bool
     713     7203408 : is_valid_ascii(const unsigned char *s, int len)
     714             : {
     715     7203408 :     const unsigned char *const s_end = s + len;
     716             :     Vector8     chunk;
     717     7203408 :     Vector8     highbit_cum = vector8_broadcast(0);
     718             : #ifdef USE_NO_SIMD
     719             :     Vector8     zero_cum = vector8_broadcast(0x80);
     720             : #endif
     721             : 
     722             :     Assert(len % sizeof(chunk) == 0);
     723             : 
     724    21610224 :     while (s < s_end)
     725             :     {
     726    14406816 :         vector8_load(&chunk, s);
     727             : 
     728             :         /* Capture any zero bytes in this chunk. */
     729             : #ifdef USE_NO_SIMD
     730             : 
     731             :         /*
     732             :          * First, add 0x7f to each byte. This sets the high bit in each byte,
     733             :          * unless it was a zero. If any resulting high bits are zero, the
     734             :          * corresponding high bits in the zero accumulator will be cleared.
     735             :          *
     736             :          * If none of the bytes in the chunk had the high bit set, the max
     737             :          * value each byte can have after the addition is 0x7f + 0x7f = 0xfe,
     738             :          * and we don't need to worry about carrying over to the next byte. If
     739             :          * any input bytes did have the high bit set, it doesn't matter
     740             :          * because we check for those separately.
     741             :          */
     742             :         zero_cum &= (chunk + vector8_broadcast(0x7F));
     743             : #else
     744             : 
     745             :         /*
     746             :          * Set all bits in each lane of the highbit accumulator where input
     747             :          * bytes are zero.
     748             :          */
     749    14406816 :         highbit_cum = vector8_or(highbit_cum,
     750             :                                  vector8_eq(chunk, vector8_broadcast(0)));
     751             : #endif
     752             : 
     753             :         /* Capture all set bits in this chunk. */
     754    14406816 :         highbit_cum = vector8_or(highbit_cum, chunk);
     755             : 
     756    14406816 :         s += sizeof(chunk);
     757             :     }
     758             : 
     759             :     /* Check if any high bits in the high bit accumulator got set. */
     760     7203408 :     if (vector8_is_highbit_set(highbit_cum))
     761         866 :         return false;
     762             : 
     763             : #ifdef USE_NO_SIMD
     764             :     /* Check if any high bits in the zero accumulator got cleared. */
     765             :     if (zero_cum != vector8_broadcast(0x80))
     766             :         return false;
     767             : #endif
     768             : 
     769     7202542 :     return true;
     770             : }
     771             : 
     772             : #endif                          /* PG_WCHAR_H */

Generated by: LCOV version 1.14