Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit9f652d4

Browse files
committed
Fix up several contrib modules that were using varlena datatypes in not-so-obvious
ways. I'm not totally sure that I caught everything, but at least now they passtheir regression tests with VARSIZE/SET_VARSIZE defined to reverse byte order.
1 parentd1ce4f7 commit9f652d4

24 files changed

+152
-147
lines changed

‎contrib/hstore/hstore.h‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
#define__HSTORE_H__
33

44
#include"postgres.h"
5+
56
#include"funcapi.h"
67
#include"access/gist.h"
78
#include"access/itup.h"
@@ -23,12 +24,12 @@ typedef struct
2324

2425
typedefstruct
2526
{
26-
int4len;
27+
int32vl_len_;/* varlena header (do not touch directly!) */
2728
int4size;
2829
chardata[1];
2930
}HStore;
3031

31-
#defineHSHRDSIZE(2*sizeof(int4))
32+
#defineHSHRDSIZE(VARHDRSZ +sizeof(int4))
3233
#defineCALCDATASIZE(x,lenstr) ( (x) * sizeof(HEntry) + HSHRDSIZE + (lenstr) )
3334
#defineARRPTR(x)( (HEntry*) ( (char*)(x) + HSHRDSIZE ) )
3435
#defineSTRPTR(x)( (char*)(x) + HSHRDSIZE + ( sizeof(HEntry) * ((HStore*)x)->size ) )

‎contrib/hstore/hstore_gist.c‎

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
#include"access/gist.h"
44
#include"access/itup.h"
5-
/*#include "access/rtree.h"*/
65
#include"crc32.h"
76

87
/* bigint defines */
@@ -38,7 +37,7 @@ typedef char *BITVECP;
3837

3938
typedefstruct
4039
{
41-
int4len;
40+
int32vl_len_;/* varlena header (do not touch directly!) */
4241
int4flag;
4342
chardata[1];
4443
}GISTTYPE;
@@ -47,7 +46,7 @@ typedef struct
4746

4847
#defineISALLTRUE(x)( ((GISTTYPE*)x)->flag & ALLISTRUE )
4948

50-
#defineGTHDRSIZE(sizeof(int4)*2)
49+
#defineGTHDRSIZE(VARHDRSZ +sizeof(int4))
5150
#defineCALCGTSIZE(flag) ( GTHDRSIZE+(((flag) & ALLISTRUE) ? 0 : SIGLEN) )
5251

5352
#defineGETSIGN(x)( (BITVECP)( (char*)x+GTHDRSIZE ) )
@@ -112,14 +111,13 @@ ghstore_compress(PG_FUNCTION_ARGS)
112111

113112
if (entry->leafkey)
114113
{
115-
GISTTYPE*res= (GISTTYPE*)palloc(CALCGTSIZE(0));
114+
GISTTYPE*res= (GISTTYPE*)palloc0(CALCGTSIZE(0));
116115
HStore*toastedval= (HStore*)DatumGetPointer(entry->key);
117116
HStore*val= (HStore*)DatumGetPointer(PG_DETOAST_DATUM(entry->key));
118117
HEntry*ptr=ARRPTR(val);
119118
char*words=STRPTR(val);
120119

121-
memset(res,0,CALCGTSIZE(0));
122-
res->len=CALCGTSIZE(0);
120+
SET_VARSIZE(res,CALCGTSIZE(0));
123121

124122
while (ptr-ARRPTR(val)<val->size)
125123
{
@@ -156,7 +154,7 @@ ghstore_compress(PG_FUNCTION_ARGS)
156154
);
157155

158156
res= (GISTTYPE*)palloc(CALCGTSIZE(ALLISTRUE));
159-
res->len=CALCGTSIZE(ALLISTRUE);
157+
SET_VARSIZE(res,CALCGTSIZE(ALLISTRUE));
160158
res->flag=ALLISTRUE;
161159

162160
retval= (GISTENTRY*)palloc(sizeof(GISTENTRY));
@@ -286,10 +284,11 @@ ghstore_union(PG_FUNCTION_ARGS)
286284

287285
len=CALCGTSIZE(flag);
288286
result= (GISTTYPE*)palloc(len);
289-
*size=result->len=len;
287+
SET_VARSIZE(result,len);
290288
result->flag=flag;
291289
if (!ISALLTRUE(result))
292290
memcpy((void*)GETSIGN(result), (void*)base,sizeof(BITVEC));
291+
*size=len;
293292

294293
PG_RETURN_POINTER(result);
295294
}
@@ -383,27 +382,27 @@ ghstore_picksplit(PG_FUNCTION_ARGS)
383382
if (ISALLTRUE(GETENTRY(entryvec,seed_1)))
384383
{
385384
datum_l= (GISTTYPE*)palloc(GTHDRSIZE);
386-
datum_l->len=GTHDRSIZE;
385+
SET_VARSIZE(datum_l,GTHDRSIZE);
387386
datum_l->flag=ALLISTRUE;
388387
}
389388
else
390389
{
391390
datum_l= (GISTTYPE*)palloc(GTHDRSIZE+SIGLEN);
392-
datum_l->len=GTHDRSIZE+SIGLEN;
391+
SET_VARSIZE(datum_l,GTHDRSIZE+SIGLEN);
393392
datum_l->flag=0;
394393
memcpy((void*)GETSIGN(datum_l), (void*)GETSIGN(GETENTRY(entryvec,seed_1)),sizeof(BITVEC))
395394
;
396395
}
397396
if (ISALLTRUE(GETENTRY(entryvec,seed_2)))
398397
{
399398
datum_r= (GISTTYPE*)palloc(GTHDRSIZE);
400-
datum_r->len=GTHDRSIZE;
399+
SET_VARSIZE(datum_r,GTHDRSIZE);
401400
datum_r->flag=ALLISTRUE;
402401
}
403402
else
404403
{
405404
datum_r= (GISTTYPE*)palloc(GTHDRSIZE+SIGLEN);
406-
datum_r->len=GTHDRSIZE+SIGLEN;
405+
SET_VARSIZE(datum_r,GTHDRSIZE+SIGLEN);
407406
datum_r->flag=0;
408407
memcpy((void*)GETSIGN(datum_r), (void*)GETSIGN(GETENTRY(entryvec,seed_2)),sizeof(BITVEC));
409408
}

‎contrib/hstore/hstore_io.c‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ hstore_in(PG_FUNCTION_ARGS)
363363
freeHSParse(&state);
364364
len=CALCDATASIZE(0,0);
365365
out=palloc(len);
366-
out->len=len;
366+
SET_VARSIZE(out,len);
367367
out->size=0;
368368
PG_RETURN_POINTER(out);
369369
}
@@ -372,7 +372,7 @@ hstore_in(PG_FUNCTION_ARGS)
372372

373373
len=CALCDATASIZE(state.pcur,buflen);
374374
out=palloc(len);
375-
out->len=len;
375+
SET_VARSIZE(out,len);
376376
out->size=state.pcur;
377377

378378
entries=ARRPTR(out);
@@ -436,7 +436,7 @@ hstore_out(PG_FUNCTION_ARGS)
436436
}
437437

438438
buflen= (4/* " */+2/* => */+2/* , */ )*in->size+
439-
2/* esc */* (in->len-CALCDATASIZE(in->size,0));
439+
2/* esc */* (VARSIZE(in)-CALCDATASIZE(in->size,0));
440440

441441
out=ptr=palloc(buflen);
442442
for (i=0;i<in->size;i++)

‎contrib/hstore/hstore_op.c‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,13 @@ delete(PG_FUNCTION_ARGS)
105105
{
106106
HStore*hs=PG_GETARG_HS(0);
107107
text*key=PG_GETARG_TEXT_P(1);
108-
HStore*out=palloc(hs->len);
108+
HStore*out=palloc(VARSIZE(hs));
109109
char*ptrs,
110110
*ptrd;
111111
HEntry*es,
112112
*ed;
113113

114-
out->len=hs->len;
114+
SET_VARSIZE(out,VARSIZE(hs));
115115
out->size=hs->size;/* temporary! */
116116

117117
ptrs=STRPTR(hs);
@@ -142,7 +142,7 @@ delete(PG_FUNCTION_ARGS)
142142
out->size=ed-ARRPTR(out);
143143

144144
memmove(STRPTR(out),ptrd,buflen);
145-
out->len=CALCDATASIZE(out->size,buflen);
145+
SET_VARSIZE(out,CALCDATASIZE(out->size,buflen));
146146
}
147147

148148

@@ -159,15 +159,15 @@ hs_concat(PG_FUNCTION_ARGS)
159159
{
160160
HStore*s1=PG_GETARG_HS(0);
161161
HStore*s2=PG_GETARG_HS(1);
162-
HStore*out=palloc(s1->len+s2->len);
162+
HStore*out=palloc(VARSIZE(s1)+VARSIZE(s2));
163163
char*ps1,
164164
*ps2,
165165
*pd;
166166
HEntry*es1,
167167
*es2,
168168
*ed;
169169

170-
out->len=s1->len+s2->len;
170+
SET_VARSIZE(out,VARSIZE(s1)+VARSIZE(s2));
171171
out->size=s1->size+s2->size;
172172

173173
ps1=STRPTR(s1);
@@ -256,7 +256,7 @@ hs_concat(PG_FUNCTION_ARGS)
256256
out->size=ed-ARRPTR(out);
257257

258258
memmove(STRPTR(out),pd,buflen);
259-
out->len=CALCDATASIZE(out->size,buflen);
259+
SET_VARSIZE(out,CALCDATASIZE(out->size,buflen));
260260
}
261261

262262
PG_FREE_IF_COPY(s1,0);
@@ -277,7 +277,7 @@ tconvert(PG_FUNCTION_ARGS)
277277

278278
len=CALCDATASIZE(1,VARSIZE(key)+VARSIZE(val)-2*VARHDRSZ);
279279
out=palloc(len);
280-
out->len=len;
280+
SET_VARSIZE(out,len);
281281
out->size=1;
282282

283283
ARRPTR(out)->keylen=VARSIZE(key)-VARHDRSZ;
@@ -399,8 +399,8 @@ setup_firstcall(FuncCallContext *funcctx, HStore * hs)
399399

400400
st= (AKStore*)palloc(sizeof(AKStore));
401401
st->i=0;
402-
st->hs= (HStore*)palloc(hs->len);
403-
memcpy(st->hs,hs,hs->len);
402+
st->hs= (HStore*)palloc(VARSIZE(hs));
403+
memcpy(st->hs,hs,VARSIZE(hs));
404404

405405
funcctx->user_fctx= (void*)st;
406406
MemoryContextSwitchTo(oldcontext);
@@ -568,8 +568,8 @@ each(PG_FUNCTION_ARGS)
568568
oldcontext=MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
569569
st= (AKStore*)palloc(sizeof(AKStore));
570570
st->i=0;
571-
st->hs= (HStore*)palloc(hs->len);
572-
memcpy(st->hs,hs,hs->len);
571+
st->hs= (HStore*)palloc(VARSIZE(hs));
572+
memcpy(st->hs,hs,VARSIZE(hs));
573573
funcctx->user_fctx= (void*)st;
574574

575575
tupdesc=RelationNameGetTupleDesc("hs_each");

‎contrib/intarray/_int.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ typedef char *BITVECP;
8585
*/
8686
typedefstruct
8787
{
88-
int4len;
88+
int32vl_len_;/* varlena header (do not touch directly!) */
8989
int4flag;
9090
chardata[1];
9191
}GISTTYPE;
@@ -94,7 +94,7 @@ typedef struct
9494

9595
#defineISALLTRUE(x)( ((GISTTYPE*)x)->flag & ALLISTRUE )
9696

97-
#defineGTHDRSIZE(sizeof(int4)*2)
97+
#defineGTHDRSIZE(VARHDRSZ +sizeof(int4))
9898
#defineCALCGTSIZE(flag) ( GTHDRSIZE+(((flag) & ALLISTRUE) ? 0 : SIGLEN) )
9999

100100
#defineGETSIGN(x)( (BITVECP)( (char*)x+GTHDRSIZE ) )
@@ -145,12 +145,12 @@ typedef struct ITEM
145145

146146
typedefstruct
147147
{
148-
int4len;
148+
int32vl_len_;/* varlena header (do not touch directly!) */
149149
int4size;
150150
chardata[1];
151151
}QUERYTYPE;
152152

153-
#defineHDRSIZEQT( 2*sizeof(int4))
153+
#defineHDRSIZEQT(VARHDRSZ +sizeof(int4))
154154
#defineCOMPUTESIZE(size)( HDRSIZEQT + size * sizeof(ITEM) )
155155
#defineGETQUERY(x) (ITEM*)( (char*)(x)+HDRSIZEQT )
156156

‎contrib/intarray/_int_bool.c‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ bqarr_in(PG_FUNCTION_ARGS)
465465

466466
commonlen=COMPUTESIZE(state.num);
467467
query= (QUERYTYPE*)palloc(commonlen);
468-
query->len=commonlen;
468+
SET_VARSIZE(query,commonlen);
469469
query->size=state.num;
470470
ptr=GETQUERY(query);
471471

‎contrib/intarray/_intbig_gist.c‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ g_intbig_compress(PG_FUNCTION_ARGS)
147147
ArrayType*in= (ArrayType*)PG_DETOAST_DATUM(entry->key);
148148
int4*ptr;
149149
intnum;
150-
GISTTYPE*res= (GISTTYPE*)palloc(CALCGTSIZE(0));
150+
GISTTYPE*res= (GISTTYPE*)palloc0(CALCGTSIZE(0));
151151

152152
CHECKARRVALID(in);
153153
if (ARRISVOID(in))
@@ -160,8 +160,7 @@ g_intbig_compress(PG_FUNCTION_ARGS)
160160
ptr=ARRPTR(in);
161161
num=ARRNELEMS(in);
162162
}
163-
memset(res,0,CALCGTSIZE(0));
164-
res->len=CALCGTSIZE(0);
163+
SET_VARSIZE(res,CALCGTSIZE(0));
165164

166165
while (num--)
167166
{
@@ -192,7 +191,7 @@ g_intbig_compress(PG_FUNCTION_ARGS)
192191
);
193192

194193
res= (GISTTYPE*)palloc(CALCGTSIZE(ALLISTRUE));
195-
res->len=CALCGTSIZE(ALLISTRUE);
194+
SET_VARSIZE(res,CALCGTSIZE(ALLISTRUE));
196195
res->flag=ALLISTRUE;
197196

198197
retval= (GISTENTRY*)palloc(sizeof(GISTENTRY));
@@ -292,10 +291,11 @@ g_intbig_union(PG_FUNCTION_ARGS)
292291

293292
len=CALCGTSIZE(flag);
294293
result= (GISTTYPE*)palloc(len);
295-
*size=result->len=len;
294+
SET_VARSIZE(result,len);
296295
result->flag=flag;
297296
if (!ISALLTRUE(result))
298297
memcpy((void*)GETSIGN(result), (void*)base,sizeof(BITVEC));
298+
*size=len;
299299

300300
PG_RETURN_POINTER(result);
301301
}
@@ -389,26 +389,26 @@ g_intbig_picksplit(PG_FUNCTION_ARGS)
389389
if (ISALLTRUE(GETENTRY(entryvec,seed_1)))
390390
{
391391
datum_l= (GISTTYPE*)palloc(GTHDRSIZE);
392-
datum_l->len=GTHDRSIZE;
392+
SET_VARSIZE(datum_l,GTHDRSIZE);
393393
datum_l->flag=ALLISTRUE;
394394
}
395395
else
396396
{
397397
datum_l= (GISTTYPE*)palloc(GTHDRSIZE+SIGLEN);
398-
datum_l->len=GTHDRSIZE+SIGLEN;
398+
SET_VARSIZE(datum_l,GTHDRSIZE+SIGLEN);
399399
datum_l->flag=0;
400400
memcpy((void*)GETSIGN(datum_l), (void*)GETSIGN(GETENTRY(entryvec,seed_1)),sizeof(BITVEC));
401401
}
402402
if (ISALLTRUE(GETENTRY(entryvec,seed_2)))
403403
{
404404
datum_r= (GISTTYPE*)palloc(GTHDRSIZE);
405-
datum_r->len=GTHDRSIZE;
405+
SET_VARSIZE(datum_r,GTHDRSIZE);
406406
datum_r->flag=ALLISTRUE;
407407
}
408408
else
409409
{
410410
datum_r= (GISTTYPE*)palloc(GTHDRSIZE+SIGLEN);
411-
datum_r->len=GTHDRSIZE+SIGLEN;
411+
SET_VARSIZE(datum_r,GTHDRSIZE+SIGLEN);
412412
datum_r->flag=0;
413413
memcpy((void*)GETSIGN(datum_r), (void*)GETSIGN(GETENTRY(entryvec,seed_2)),sizeof(BITVEC));
414414
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp