granicus.if.org Git - postgresql/blob - src/backend/utils/mb/conversion_procs/utf8_and_gbk/utf8_and_gbk.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  *        GBK <--> UTF8
   4  *
   5  * Portions Copyright (c) 1996-2017, PostgreSQL Global Development Group
   6  * Portions Copyright (c) 1994, Regents of the University of California
   7  *
   8  * IDENTIFICATION
   9  *        src/backend/utils/mb/conversion_procs/utf8_and_gbk/utf8_and_gbk.c
  10  *
  11  *-------------------------------------------------------------------------
  12  */
  13
  14 #include "postgres.h"
  15 #include "fmgr.h"
  16 #include "mb/pg_wchar.h"
  17 #include "../../Unicode/gbk_to_utf8.map"
  18 #include "../../Unicode/utf8_to_gbk.map"
  19
  20 PG_MODULE_MAGIC;
  21
  22 PG_FUNCTION_INFO_V1(gbk_to_utf8);
  23 PG_FUNCTION_INFO_V1(utf8_to_gbk);
  24
  25 /* ----------
  26  * conv_proc(
  27  *              INTEGER,        -- source encoding id
  28  *              INTEGER,        -- destination encoding id
  29  *              CSTRING,        -- source string (null terminated C string)
  30  *              CSTRING,        -- destination string (null terminated C string)
  31  *              INTEGER         -- source string length
  32  * ) returns VOID;
  33  * ----------
  34  */
  35 Datum
  36 gbk_to_utf8(PG_FUNCTION_ARGS)
  37 {
  38         unsigned char *src = (unsigned char *) PG_GETARG_CSTRING(2);
  39         unsigned char *dest = (unsigned char *) PG_GETARG_CSTRING(3);
  40         int                     len = PG_GETARG_INT32(4);
  41
  42         CHECK_ENCODING_CONVERSION_ARGS(PG_GBK, PG_UTF8);
  43
  44         LocalToUtf(src, len, dest,
  45                            LUmapGBK, lengthof(LUmapGBK),
  46                            NULL, 0,
  47                            NULL,
  48                            PG_GBK);
  49
  50         PG_RETURN_VOID();
  51 }
  52
  53 Datum
  54 utf8_to_gbk(PG_FUNCTION_ARGS)
  55 {
  56         unsigned char *src = (unsigned char *) PG_GETARG_CSTRING(2);
  57         unsigned char *dest = (unsigned char *) PG_GETARG_CSTRING(3);
  58         int                     len = PG_GETARG_INT32(4);
  59
  60         CHECK_ENCODING_CONVERSION_ARGS(PG_UTF8, PG_GBK);
  61
  62         UtfToLocal(src, len, dest,
  63                            ULmapGBK, lengthof(ULmapGBK),
  64                            NULL, 0,
  65                            NULL,
  66                            PG_GBK);
  67
  68         PG_RETURN_VOID();
  69 }