Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * stringinfo_mb.c
       4                 :             :  *              Multibyte encoding-aware additional StringInfo facilities
       5                 :             :  *
       6                 :             :  * This is separate from common/stringinfo.c so that frontend users
       7                 :             :  * of that file need not pull in unnecessary multibyte-encoding support
       8                 :             :  * code.
       9                 :             :  *
      10                 :             :  *
      11                 :             :  * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
      12                 :             :  * Portions Copyright (c) 1994, Regents of the University of California
      13                 :             :  *
      14                 :             :  *
      15                 :             :  * IDENTIFICATION
      16                 :             :  *        src/backend/utils/mb/stringinfo_mb.c
      17                 :             :  *
      18                 :             :  *-------------------------------------------------------------------------
      19                 :             :  */
      20                 :             : #include "postgres.h"
      21                 :             : 
      22                 :             : #include "mb/pg_wchar.h"
      23                 :             : #include "mb/stringinfo_mb.h"
      24                 :             : 
      25                 :             : 
      26                 :             : /*
      27                 :             :  * appendStringInfoStringQuoted
      28                 :             :  *
      29                 :             :  * Append up to maxlen bytes from s to str, or the whole input string if
      30                 :             :  * maxlen < 0, adding single quotes around it and doubling all single quotes.
      31                 :             :  * Add an ellipsis if the copy is incomplete.
      32                 :             :  */
      33                 :             : void
      34                 :          78 : appendStringInfoStringQuoted(StringInfo str, const char *s, int maxlen)
      35                 :             : {
      36                 :          78 :         char       *copy = NULL;
      37                 :          78 :         const char *chunk_search_start,
      38                 :             :                            *chunk_copy_start,
      39                 :             :                            *chunk_end;
      40                 :          78 :         int                     slen;
      41                 :          78 :         bool            ellipsis;
      42                 :             : 
      43         [ +  - ]:          78 :         Assert(str != NULL);
      44                 :             : 
      45                 :          78 :         slen = strlen(s);
      46   [ -  +  #  # ]:          78 :         if (maxlen >= 0 && maxlen < slen)
      47                 :             :         {
      48                 :           0 :                 int                     finallen = pg_mbcliplen(s, slen, maxlen);
      49                 :             : 
      50                 :           0 :                 copy = pnstrdup(s, finallen);
      51                 :           0 :                 chunk_search_start = copy;
      52                 :           0 :                 chunk_copy_start = copy;
      53                 :             : 
      54                 :           0 :                 ellipsis = true;
      55                 :           0 :         }
      56                 :             :         else
      57                 :             :         {
      58                 :          78 :                 chunk_search_start = s;
      59                 :          78 :                 chunk_copy_start = s;
      60                 :             : 
      61                 :          78 :                 ellipsis = false;
      62                 :             :         }
      63                 :             : 
      64         [ -  + ]:          78 :         appendStringInfoCharMacro(str, '\'');
      65                 :             : 
      66         [ +  + ]:          82 :         while ((chunk_end = strchr(chunk_search_start, '\'')) != NULL)
      67                 :             :         {
      68                 :             :                 /* copy including the found delimiting ' */
      69                 :           8 :                 appendBinaryStringInfoNT(str,
      70                 :           4 :                                                                  chunk_copy_start,
      71                 :           4 :                                                                  chunk_end - chunk_copy_start + 1);
      72                 :             : 
      73                 :             :                 /* in order to double it, include this ' into the next chunk as well */
      74                 :           4 :                 chunk_copy_start = chunk_end;
      75                 :           4 :                 chunk_search_start = chunk_end + 1;
      76                 :             :         }
      77                 :             : 
      78                 :             :         /* copy the last chunk and terminate */
      79         [ -  + ]:          78 :         if (ellipsis)
      80                 :           0 :                 appendStringInfo(str, "%s...'", chunk_copy_start);
      81                 :             :         else
      82                 :          78 :                 appendStringInfo(str, "%s'", chunk_copy_start);
      83                 :             : 
      84         [ +  - ]:          78 :         if (copy)
      85                 :           0 :                 pfree(copy);
      86                 :          78 : }