149 lines
3.1 KiB
C
149 lines
3.1 KiB
C
/* src/interfaces/ecpg/pgtypeslib/common.c */
|
|
|
|
#include "postgres_fe.h"
|
|
|
|
#include "pgtypes.h"
|
|
#include "pgtypeslib_extern.h"
|
|
|
|
/* Return value is zero-filled. */
|
|
char *
|
|
pgtypes_alloc(long size)
|
|
{
|
|
char *new = (char *) calloc(1L, size);
|
|
|
|
if (!new)
|
|
errno = ENOMEM;
|
|
return new;
|
|
}
|
|
|
|
char *
|
|
pgtypes_strdup(const char *str)
|
|
{
|
|
char *new = (char *) strdup(str);
|
|
|
|
if (!new)
|
|
errno = ENOMEM;
|
|
return new;
|
|
}
|
|
|
|
int
|
|
pgtypes_fmt_replace(union un_fmt_comb replace_val, int replace_type, char **output, int *pstr_len)
|
|
{
|
|
/*
|
|
* general purpose variable, set to 0 in order to fix compiler warning
|
|
*/
|
|
int i = 0;
|
|
|
|
switch (replace_type)
|
|
{
|
|
case PGTYPES_TYPE_NOTHING:
|
|
break;
|
|
case PGTYPES_TYPE_STRING_CONSTANT:
|
|
case PGTYPES_TYPE_STRING_MALLOCED:
|
|
i = strlen(replace_val.str_val);
|
|
if (i + 1 <= *pstr_len)
|
|
{
|
|
/* include trailing terminator in what we copy */
|
|
memcpy(*output, replace_val.str_val, i + 1);
|
|
*pstr_len -= i;
|
|
*output += i;
|
|
if (replace_type == PGTYPES_TYPE_STRING_MALLOCED)
|
|
free(replace_val.str_val);
|
|
return 0;
|
|
}
|
|
else
|
|
return -1;
|
|
break;
|
|
case PGTYPES_TYPE_CHAR:
|
|
if (*pstr_len >= 2)
|
|
{
|
|
(*output)[0] = replace_val.char_val;
|
|
(*output)[1] = '\0';
|
|
(*pstr_len)--;
|
|
(*output)++;
|
|
return 0;
|
|
}
|
|
else
|
|
return -1;
|
|
break;
|
|
case PGTYPES_TYPE_DOUBLE_NF:
|
|
case PGTYPES_TYPE_INT64:
|
|
case PGTYPES_TYPE_UINT:
|
|
case PGTYPES_TYPE_UINT_2_LZ:
|
|
case PGTYPES_TYPE_UINT_2_LS:
|
|
case PGTYPES_TYPE_UINT_3_LZ:
|
|
case PGTYPES_TYPE_UINT_4_LZ:
|
|
{
|
|
char *t = pgtypes_alloc(PGTYPES_FMT_NUM_MAX_DIGITS);
|
|
|
|
if (!t)
|
|
return ENOMEM;
|
|
switch (replace_type)
|
|
{
|
|
case PGTYPES_TYPE_DOUBLE_NF:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%0.0g", replace_val.double_val);
|
|
break;
|
|
case PGTYPES_TYPE_INT64:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
INT64_FORMAT, replace_val.int64_val);
|
|
break;
|
|
case PGTYPES_TYPE_UINT:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%u", replace_val.uint_val);
|
|
break;
|
|
case PGTYPES_TYPE_UINT_2_LZ:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%02u", replace_val.uint_val);
|
|
break;
|
|
case PGTYPES_TYPE_UINT_2_LS:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%2u", replace_val.uint_val);
|
|
break;
|
|
case PGTYPES_TYPE_UINT_3_LZ:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%03u", replace_val.uint_val);
|
|
break;
|
|
case PGTYPES_TYPE_UINT_4_LZ:
|
|
i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
|
|
"%04u", replace_val.uint_val);
|
|
break;
|
|
}
|
|
|
|
if (i < 0 || i >= PGTYPES_FMT_NUM_MAX_DIGITS)
|
|
{
|
|
free(t);
|
|
return -1;
|
|
}
|
|
i = strlen(t);
|
|
*pstr_len -= i;
|
|
|
|
/*
|
|
* if *pstr_len == 0, we don't have enough space for the
|
|
* terminator and the conversion fails
|
|
*/
|
|
if (*pstr_len <= 0)
|
|
{
|
|
free(t);
|
|
return -1;
|
|
}
|
|
strcpy(*output, t);
|
|
*output += i;
|
|
free(t);
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/* Functions declared in pgtypes.h. */
|
|
|
|
/* Just frees memory (mostly needed for Windows) */
|
|
void
|
|
PGTYPESchar_free(char *ptr)
|
|
{
|
|
free(ptr);
|
|
}
|