1 #include "postgres.h"
2
3 #include "fmgr.h"
4 #include "plpython.h"
5 #include "plpy_typeio.h"
6 #include "hstore/hstore.h"
7
8 PG_MODULE_MAGIC;
9
10 extern void _PG_init(void);
11
12 /* Linkage to functions in plpython module */
13 typedef char *(*PLyObject_AsString_t) (PyObject *plrv);
14 static PLyObject_AsString_t PLyObject_AsString_p;
15 #if PY_MAJOR_VERSION >= 3
16 typedef PyObject *(*PLyUnicode_FromStringAndSize_t) (const char *s, Py_ssize_t size);
17 static PLyUnicode_FromStringAndSize_t PLyUnicode_FromStringAndSize_p;
18 #endif
git_blob_rawcontent(const git_blob * blob)19
20 /* Linkage to functions in hstore module */
21 typedef HStore *(*hstoreUpgrade_t) (Datum orig);
22 static hstoreUpgrade_t hstoreUpgrade_p;
23 typedef int (*hstoreUniquePairs_t) (Pairs *a, int32 l, int32 *buflen);
24 static hstoreUniquePairs_t hstoreUniquePairs_p;
25 typedef HStore *(*hstorePairs_t) (Pairs *pairs, int32 pcount, int32 buflen);
26 static hstorePairs_t hstorePairs_p;
27 typedef size_t (*hstoreCheckKeyLen_t) (size_t len);
28 static hstoreCheckKeyLen_t hstoreCheckKeyLen_p;
git_blob_rawsize(const git_blob * blob)29 typedef size_t (*hstoreCheckValLen_t) (size_t len);
30 static hstoreCheckValLen_t hstoreCheckValLen_p;
31
32
33 /*
34 * Module initialize function: fetch function pointers for cross-module calls.
35 */
36 void
37 _PG_init(void)
38 {
git_blob__getbuf(git_buf * buffer,git_blob * blob)39 /* Asserts verify that typedefs above match original declarations */
40 AssertVariableIsOfType(&PLyObject_AsString, PLyObject_AsString_t);
41 PLyObject_AsString_p = (PLyObject_AsString_t)
42 load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyObject_AsString",
43 true, NULL);
44 #if PY_MAJOR_VERSION >= 3
45 AssertVariableIsOfType(&PLyUnicode_FromStringAndSize, PLyUnicode_FromStringAndSize_t);
46 PLyUnicode_FromStringAndSize_p = (PLyUnicode_FromStringAndSize_t)
git_blob__free(void * _blob)47 load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyUnicode_FromStringAndSize",
48 true, NULL);
49 #endif
50 AssertVariableIsOfType(&hstoreUpgrade, hstoreUpgrade_t);
51 hstoreUpgrade_p = (hstoreUpgrade_t)
52 load_external_function("$libdir/hstore", "hstoreUpgrade",
53 true, NULL);
54 AssertVariableIsOfType(&hstoreUniquePairs, hstoreUniquePairs_t);
git_blob__parse_raw(void * _blob,const char * data,size_t size)55 hstoreUniquePairs_p = (hstoreUniquePairs_t)
56 load_external_function("$libdir/hstore", "hstoreUniquePairs",
57 true, NULL);
58 AssertVariableIsOfType(&hstorePairs, hstorePairs_t);
59 hstorePairs_p = (hstorePairs_t)
60 load_external_function("$libdir/hstore", "hstorePairs",
61 true, NULL);
62 AssertVariableIsOfType(&hstoreCheckKeyLen, hstoreCheckKeyLen_t);
63 hstoreCheckKeyLen_p = (hstoreCheckKeyLen_t)
64 load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
65 true, NULL);
66 AssertVariableIsOfType(&hstoreCheckValLen, hstoreCheckValLen_t);
git_blob__parse(void * _blob,git_odb_object * odb_obj)67 hstoreCheckValLen_p = (hstoreCheckValLen_t)
68 load_external_function("$libdir/hstore", "hstoreCheckValLen",
69 true, NULL);
70 }
71
72
73 /* These defines must be after the module init function */
74 #define PLyObject_AsString PLyObject_AsString_p
75 #define PLyUnicode_FromStringAndSize PLyUnicode_FromStringAndSize_p
76 #define hstoreUpgrade hstoreUpgrade_p
77 #define hstoreUniquePairs hstoreUniquePairs_p
78 #define hstorePairs hstorePairs_p
git_blob_create_from_buffer(git_oid * id,git_repository * repo,const void * buffer,size_t len)79 #define hstoreCheckKeyLen hstoreCheckKeyLen_p
80 #define hstoreCheckValLen hstoreCheckValLen_p
81
82
83 PG_FUNCTION_INFO_V1(hstore_to_plpython);
84
85 Datum
86 hstore_to_plpython(PG_FUNCTION_ARGS)
87 {
88 HStore *in = PG_GETARG_HSTORE_P(0);
89 int i;
90 int count = HS_COUNT(in);
91 char *base = STRPTR(in);
92 HEntry *entries = ARRPTR(in);
93 PyObject *dict;
94
95 dict = PyDict_New();
96 if (!dict)
97 ereport(ERROR,
98 (errcode(ERRCODE_OUT_OF_MEMORY),
99 errmsg("out of memory")));
write_file_stream(git_oid * id,git_odb * odb,const char * path,git_object_size_t file_size)100
101 for (i = 0; i < count; i++)
102 {
103 PyObject *key;
104
105 key = PyString_FromStringAndSize(HSTORE_KEY(entries, base, i),
106 HSTORE_KEYLEN(entries, i));
107 if (HSTORE_VALISNULL(entries, i))
108 PyDict_SetItem(dict, key, Py_None);
109 else
110 {
111 PyObject *value;
112
113 value = PyString_FromStringAndSize(HSTORE_VAL(entries, base, i),
114 HSTORE_VALLEN(entries, i));
115 PyDict_SetItem(dict, key, value);
116 Py_XDECREF(value);
117 }
118 Py_XDECREF(key);
119 }
120
121 return PointerGetDatum(dict);
122 }
123
124
125 PG_FUNCTION_INFO_V1(plpython_to_hstore);
126
127 Datum
128 plpython_to_hstore(PG_FUNCTION_ARGS)
129 {
130 PyObject *dict;
131 PyObject *volatile items;
132 Py_ssize_t pcount;
133 HStore *volatile out;
134
135 dict = (PyObject *) PG_GETARG_POINTER(0);
136 if (!PyMapping_Check(dict))
write_file_filtered(git_oid * id,git_object_size_t * size,git_odb * odb,const char * full_path,git_filter_list * fl)137 ereport(ERROR,
138 (errcode(ERRCODE_WRONG_OBJECT_TYPE),
139 errmsg("not a Python mapping")));
140
141 pcount = PyMapping_Size(dict);
142 items = PyMapping_Items(dict);
143
144 PG_TRY();
145 {
146 int32 buflen;
147 Py_ssize_t i;
148 Pairs *pairs;
149
150 pairs = palloc(pcount * sizeof(*pairs));
151
152 for (i = 0; i < pcount; i++)
153 {
154 PyObject *tuple;
155 PyObject *key;
156 PyObject *value;
157
158 tuple = PyList_GetItem(items, i);
159 key = PyTuple_GetItem(tuple, 0);
write_symlink(git_oid * id,git_odb * odb,const char * path,size_t link_size)160 value = PyTuple_GetItem(tuple, 1);
161
162 pairs[i].key = PLyObject_AsString(key);
163 pairs[i].keylen = hstoreCheckKeyLen(strlen(pairs[i].key));
164 pairs[i].needfree = true;
165
166 if (value == Py_None)
167 {
168 pairs[i].val = NULL;
169 pairs[i].vallen = 0;
170 pairs[i].isnull = true;
171 }
172 else
173 {
174 pairs[i].val = PLyObject_AsString(value);
175 pairs[i].vallen = hstoreCheckValLen(strlen(pairs[i].val));
176 pairs[i].isnull = false;
177 }
178 }
179
180 pcount = hstoreUniquePairs(pairs, pcount, &buflen);
181 out = hstorePairs(pairs, pcount, buflen);
git_blob__create_from_paths(git_oid * id,struct stat * out_st,git_repository * repo,const char * content_path,const char * hint_path,mode_t hint_mode,bool try_load_filters)182 }
183 PG_CATCH();
184 {
185 Py_DECREF(items);
186 PG_RE_THROW();
187 }
188 PG_END_TRY();
189
190 Py_DECREF(items);
191
192 PG_RETURN_POINTER(out);
193 }
194