← Back to branch summary

~vcs-imports/mammoth-replicator/trunk

« back to all changes in this revision

Viewing changes to src/interfaces/ecpg/pgtypeslib/common.c

  • Committer: alvherre
  • Date: 2005-12-16 21:24:52 UTC
  • Revision ID: svn-v4:db760fc0-0f08-0410-9d63-cc6633f64896:trunk:1
Initial import of the REL8_0_3 sources from the Pgsql CVS repository.

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/interfaces/ecpg/pgtypeslib/common.c
 
1
#include "postgres_fe.h"
 
2
 
 
3
#include "extern.h"
 
4
 
 
5
char *
 
6
pgtypes_alloc(long size)
 
7
{
 
8
        char       *new = (char *) calloc(1L, size);
 
9
 
 
10
        if (!new)
 
11
        {
 
12
                errno = ENOMEM;
 
13
                return NULL;
 
14
        }
 
15
 
 
16
        memset(new, '\0', size);
 
17
        return (new);
 
18
}
 
19
 
 
20
char *
 
21
pgtypes_strdup(char *str)
 
22
{
 
23
        char       *new = (char *) strdup(str);
 
24
 
 
25
        if (!new)
 
26
                errno = ENOMEM;
 
27
        return (new);
 
28
}
 
29
 
 
30
int
 
31
pgtypes_fmt_replace(union un_fmt_comb replace_val, int replace_type, char **output, int *pstr_len)
 
32
{
 
33
        /*
 
34
         * general purpose variable, set to 0 in order to fix compiler warning
 
35
         */
 
36
        int                     i = 0;
 
37
 
 
38
        switch (replace_type)
 
39
        {
 
40
                case PGTYPES_TYPE_NOTHING:
 
41
                        break;
 
42
                case PGTYPES_TYPE_STRING_CONSTANT:
 
43
                case PGTYPES_TYPE_STRING_MALLOCED:
 
44
                        i = strlen(replace_val.str_val);
 
45
                        if (i + 1 <= *pstr_len)
 
46
                        {
 
47
                                /*
 
48
                                 * copy over i + 1 bytes, that includes the tailing
 
49
                                 * terminator
 
50
                                 */
 
51
                                strncpy(*output, replace_val.str_val, i + 1);
 
52
                                *pstr_len -= i;
 
53
                                *output += i;
 
54
                                if (replace_type == PGTYPES_TYPE_STRING_MALLOCED)
 
55
                                        free(replace_val.str_val);
 
56
                                return 0;
 
57
                        }
 
58
                        else
 
59
                                return -1;
 
60
                        break;
 
61
                case PGTYPES_TYPE_CHAR:
 
62
                        if (*pstr_len >= 2)
 
63
                        {
 
64
                                (*output)[0] = replace_val.char_val;
 
65
                                (*output)[1] = '\0';
 
66
                                (*pstr_len)--;
 
67
                                (*output)++;
 
68
                                return 0;
 
69
                        }
 
70
                        else
 
71
                                return -1;
 
72
                        break;
 
73
                case PGTYPES_TYPE_DOUBLE_NF:
 
74
                case PGTYPES_TYPE_INT64:
 
75
                case PGTYPES_TYPE_UINT:
 
76
                case PGTYPES_TYPE_UINT_2_LZ:
 
77
                case PGTYPES_TYPE_UINT_2_LS:
 
78
                case PGTYPES_TYPE_UINT_3_LZ:
 
79
                case PGTYPES_TYPE_UINT_4_LZ:
 
80
                        {
 
81
                                char       *t = pgtypes_alloc(PGTYPES_FMT_NUM_MAX_DIGITS);
 
82
 
 
83
                                if (!t)
 
84
                                        return ENOMEM;
 
85
                                switch (replace_type)
 
86
                                {
 
87
                                        case PGTYPES_TYPE_DOUBLE_NF:
 
88
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
89
                                                                         "%0.0g", replace_val.double_val);
 
90
                                                break;
 
91
                                        case PGTYPES_TYPE_INT64:
 
92
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
93
                                                                         INT64_FORMAT, replace_val.int64_val);
 
94
                                                break;
 
95
                                        case PGTYPES_TYPE_UINT:
 
96
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
97
                                                                         "%u", replace_val.uint_val);
 
98
                                                break;
 
99
                                        case PGTYPES_TYPE_UINT_2_LZ:
 
100
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
101
                                                                         "%02u", replace_val.uint_val);
 
102
                                                break;
 
103
                                        case PGTYPES_TYPE_UINT_2_LS:
 
104
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
105
                                                                         "%2u", replace_val.uint_val);
 
106
                                                break;
 
107
                                        case PGTYPES_TYPE_UINT_3_LZ:
 
108
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
109
                                                                         "%03u", replace_val.uint_val);
 
110
                                                break;
 
111
                                        case PGTYPES_TYPE_UINT_4_LZ:
 
112
                                                i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
 
113
                                                                         "%04u", replace_val.uint_val);
 
114
                                                break;
 
115
                                }
 
116
 
 
117
                                if (i < 0)
 
118
                                {
 
119
                                        free(t);
 
120
                                        return -1;
 
121
                                }
 
122
                                i = strlen(t);
 
123
                                *pstr_len -= i;
 
124
 
 
125
                                /*
 
126
                                 * if *pstr_len == 0, we don't have enough space for the
 
127
                                 * terminator and the conversion fails
 
128
                                 */
 
129
                                if (*pstr_len <= 0)
 
130
                                {
 
131
                                        free(t);
 
132
                                        return -1;
 
133
                                }
 
134
                                strcpy(*output, t);
 
135
                                *output += i;
 
136
                                free(t);
 
137
                        }
 
138
                        break;
 
139
                default:
 
140
                        break;
 
141
        }
 
142
        return 0;
 
143
}