PostgreSQL Source Code  git master
regress.c File Reference
#include "postgres.h"
#include <math.h>
#include <signal.h>
#include "access/detoast.h"
#include "access/htup_details.h"
#include "access/transam.h"
#include "access/xact.h"
#include "catalog/namespace.h"
#include "catalog/pg_operator.h"
#include "catalog/pg_type.h"
#include "commands/sequence.h"
#include "commands/trigger.h"
#include "executor/executor.h"
#include "executor/spi.h"
#include "funcapi.h"
#include "mb/pg_wchar.h"
#include "miscadmin.h"
#include "nodes/supportnodes.h"
#include "optimizer/optimizer.h"
#include "optimizer/plancat.h"
#include "parser/parse_coerce.h"
#include "port/atomics.h"
#include "storage/spin.h"
#include "utils/array.h"
#include "utils/builtins.h"
#include "utils/geo_decls.h"
#include "utils/memutils.h"
#include "utils/rel.h"
#include "utils/typcache.h"
Include dependency graph for regress.c:

Go to the source code of this file.

Data Structures

struct  WIDGET
 

Macros

#define EXPECT_TRUE(expr)
 
#define EXPECT_EQ_U32(result_expr, expected_expr)
 
#define EXPECT_EQ_U64(result_expr, expected_expr)
 
#define LDELIM   '('
 
#define RDELIM   ')'
 
#define DELIM   ','
 
#define NARGS   3
 
#define TTDUMMY_INFINITY   999999
 
#define NUM_TEST_ATOMICS   (NUM_SPINLOCK_SEMAPHORES + NUM_ATOMICS_SEMAPHORES + 27)
 

Functions

static void regress_lseg_construct (LSEG *lseg, Point *pt1, Point *pt2)
 
 PG_FUNCTION_INFO_V1 (interpt_pp)
 
Datum interpt_pp (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (overpaid)
 
Datum overpaid (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (widget_in)
 
 PG_FUNCTION_INFO_V1 (widget_out)
 
Datum widget_in (PG_FUNCTION_ARGS)
 
Datum widget_out (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (pt_in_widget)
 
Datum pt_in_widget (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (reverse_name)
 
Datum reverse_name (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (trigger_return_old)
 
Datum trigger_return_old (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (ttdummy)
 
Datum ttdummy (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (set_ttdummy)
 
Datum set_ttdummy (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (int44in)
 
Datum int44in (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (int44out)
 
Datum int44out (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (test_canonicalize_path)
 
Datum test_canonicalize_path (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (make_tuple_indirect)
 
Datum make_tuple_indirect (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (regress_setenv)
 
Datum regress_setenv (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (wait_pid)
 
Datum wait_pid (PG_FUNCTION_ARGS)
 
static void test_atomic_flag (void)
 
static void test_atomic_uint32 (void)
 
static void test_atomic_uint64 (void)
 
static void test_spinlock (void)
 
static void test_atomic_spin_nest (void)
 
 PG_FUNCTION_INFO_V1 (test_atomic_ops)
 
Datum test_atomic_ops (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (test_fdw_handler)
 
Datum test_fdw_handler (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (test_support_func)
 
Datum test_support_func (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (test_opclass_options_func)
 
Datum test_opclass_options_func (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (test_enc_conversion)
 
Datum test_enc_conversion (PG_FUNCTION_ARGS)
 
 PG_FUNCTION_INFO_V1 (binary_coercible)
 
Datum binary_coercible (PG_FUNCTION_ARGS)
 

Variables

 PG_MODULE_MAGIC
 
static SPIPlanPtr splan = NULL
 
static bool ttoff = false
 

Macro Definition Documentation

◆ DELIM

#define DELIM   ','

Definition at line 79 of file regress.c.

◆ EXPECT_EQ_U32

#define EXPECT_EQ_U32 (   result_expr,
  expected_expr 
)
Value:
do { \
uint32 actual_result = (result_expr); \
uint32 expected_result = (expected_expr); \
if (actual_result != expected_result) \
elog(ERROR, \
"%s yielded %u, expected %s in file \"%s\" line %u", \
#result_expr, actual_result, #expected_expr, __FILE__, __LINE__); \
} while (0)
#define ERROR
Definition: elog.h:39

Definition at line 57 of file regress.c.

◆ EXPECT_EQ_U64

#define EXPECT_EQ_U64 (   result_expr,
  expected_expr 
)
Value:
do { \
uint64 actual_result = (result_expr); \
uint64 expected_result = (expected_expr); \
if (actual_result != expected_result) \
elog(ERROR, \
"%s yielded " UINT64_FORMAT ", expected %s in file \"%s\" line %u", \
#result_expr, actual_result, #expected_expr, __FILE__, __LINE__); \
} while (0)
#define UINT64_FORMAT
Definition: c.h:549

Definition at line 67 of file regress.c.

◆ EXPECT_TRUE

#define EXPECT_TRUE (   expr)
Value:
do { \
if (!(expr)) \
elog(ERROR, \
"%s was unexpectedly false in file \"%s\" line %u", \
#expr, __FILE__, __LINE__); \
} while (0)

Definition at line 49 of file regress.c.

◆ LDELIM

#define LDELIM   '('

Definition at line 77 of file regress.c.

◆ NARGS

#define NARGS   3

Definition at line 168 of file regress.c.

◆ NUM_TEST_ATOMICS

#define NUM_TEST_ATOMICS   (NUM_SPINLOCK_SEMAPHORES + NUM_ATOMICS_SEMAPHORES + 27)

◆ RDELIM

#define RDELIM   ')'

Definition at line 78 of file regress.c.

◆ TTDUMMY_INFINITY

#define TTDUMMY_INFINITY   999999

Definition at line 267 of file regress.c.

Function Documentation

◆ binary_coercible()

Datum binary_coercible ( PG_FUNCTION_ARGS  )

Definition at line 1217 of file regress.c.

1218 {
1219  Oid srctype = PG_GETARG_OID(0);
1220  Oid targettype = PG_GETARG_OID(1);
1221 
1222  PG_RETURN_BOOL(IsBinaryCoercible(srctype, targettype));
1223 }
#define PG_GETARG_OID(n)
Definition: fmgr.h:275
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359
bool IsBinaryCoercible(Oid srctype, Oid targettype)
unsigned int Oid
Definition: postgres_ext.h:31

References IsBinaryCoercible(), PG_GETARG_OID, and PG_RETURN_BOOL.

◆ int44in()

Datum int44in ( PG_FUNCTION_ARGS  )

Definition at line 502 of file regress.c.

503 {
504  char *input_string = PG_GETARG_CSTRING(0);
505  int32 *result = (int32 *) palloc(4 * sizeof(int32));
506  int i;
507 
508  i = sscanf(input_string,
509  "%d, %d, %d, %d",
510  &result[0],
511  &result[1],
512  &result[2],
513  &result[3]);
514  while (i < 4)
515  result[i++] = 0;
516 
517  PG_RETURN_POINTER(result);
518 }
signed int int32
Definition: c.h:494
#define PG_GETARG_CSTRING(n)
Definition: fmgr.h:277
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:361
int i
Definition: isn.c:73
void * palloc(Size size)
Definition: mcxt.c:1316

References i, palloc(), PG_GETARG_CSTRING, and PG_RETURN_POINTER.

◆ int44out()

Datum int44out ( PG_FUNCTION_ARGS  )

Definition at line 526 of file regress.c.

527 {
528  int32 *an_array = (int32 *) PG_GETARG_POINTER(0);
529  char *result = (char *) palloc(16 * 4);
530 
531  snprintf(result, 16 * 4, "%d,%d,%d,%d",
532  an_array[0],
533  an_array[1],
534  an_array[2],
535  an_array[3]);
536 
537  PG_RETURN_CSTRING(result);
538 }
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_RETURN_CSTRING(x)
Definition: fmgr.h:362
#define snprintf
Definition: port.h:238

References palloc(), PG_GETARG_POINTER, PG_RETURN_CSTRING, and snprintf.

◆ interpt_pp()

Datum interpt_pp ( PG_FUNCTION_ARGS  )

Definition at line 90 of file regress.c.

91 {
92  PATH *p1 = PG_GETARG_PATH_P(0);
93  PATH *p2 = PG_GETARG_PATH_P(1);
94  int i,
95  j;
96  LSEG seg1,
97  seg2;
98  bool found; /* We've found the intersection */
99 
100  found = false; /* Haven't found it yet */
101 
102  for (i = 0; i < p1->npts - 1 && !found; i++)
103  {
104  regress_lseg_construct(&seg1, &p1->p[i], &p1->p[i + 1]);
105  for (j = 0; j < p2->npts - 1 && !found; j++)
106  {
107  regress_lseg_construct(&seg2, &p2->p[j], &p2->p[j + 1]);
109  LsegPGetDatum(&seg1),
110  LsegPGetDatum(&seg2))))
111  found = true;
112  }
113  }
114 
115  if (!found)
116  PG_RETURN_NULL();
117 
118  /*
119  * Note: DirectFunctionCall2 will kick out an error if lseg_interpt()
120  * returns NULL, but that should be impossible since we know the two
121  * segments intersect.
122  */
124  LsegPGetDatum(&seg1),
125  LsegPGetDatum(&seg2)));
126 }
#define DirectFunctionCall2(func, arg1, arg2)
Definition: fmgr.h:644
#define PG_RETURN_NULL()
Definition: fmgr.h:345
#define PG_RETURN_DATUM(x)
Definition: fmgr.h:353
static Datum LsegPGetDatum(const LSEG *X)
Definition: geo_decls.h:194
#define PG_GETARG_PATH_P(n)
Definition: geo_decls.h:216
Datum lseg_intersect(PG_FUNCTION_ARGS)
Definition: geo_ops.c:2188
Datum lseg_interpt(PG_FUNCTION_ARGS)
Definition: geo_ops.c:2361
int j
Definition: isn.c:74
static bool DatumGetBool(Datum X)
Definition: postgres.h:90
static void regress_lseg_construct(LSEG *lseg, Point *pt1, Point *pt2)
Definition: regress.c:131
Point p[FLEXIBLE_ARRAY_MEMBER]
Definition: geo_decls.h:121
int32 npts
Definition: geo_decls.h:118

References DatumGetBool(), DirectFunctionCall2, i, j, lseg_interpt(), lseg_intersect(), LsegPGetDatum(), PATH::npts, PATH::p, p2, PG_GETARG_PATH_P, PG_RETURN_DATUM, PG_RETURN_NULL, and regress_lseg_construct().

◆ make_tuple_indirect()

Datum make_tuple_indirect ( PG_FUNCTION_ARGS  )

Definition at line 552 of file regress.c.

553 {
555  HeapTupleData tuple;
556  int ncolumns;
557  Datum *values;
558  bool *nulls;
559 
560  Oid tupType;
561  int32 tupTypmod;
562  TupleDesc tupdesc;
563 
564  HeapTuple newtup;
565 
566  int i;
567 
568  MemoryContext old_context;
569 
570  /* Extract type info from the tuple itself */
571  tupType = HeapTupleHeaderGetTypeId(rec);
572  tupTypmod = HeapTupleHeaderGetTypMod(rec);
573  tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
574  ncolumns = tupdesc->natts;
575 
576  /* Build a temporary HeapTuple control structure */
578  ItemPointerSetInvalid(&(tuple.t_self));
579  tuple.t_tableOid = InvalidOid;
580  tuple.t_data = rec;
581 
582  values = (Datum *) palloc(ncolumns * sizeof(Datum));
583  nulls = (bool *) palloc(ncolumns * sizeof(bool));
584 
585  heap_deform_tuple(&tuple, tupdesc, values, nulls);
586 
588 
589  for (i = 0; i < ncolumns; i++)
590  {
591  struct varlena *attr;
592  struct varlena *new_attr;
593  struct varatt_indirect redirect_pointer;
594 
595  /* only work on existing, not-null varlenas */
596  if (TupleDescAttr(tupdesc, i)->attisdropped ||
597  nulls[i] ||
598  TupleDescAttr(tupdesc, i)->attlen != -1)
599  continue;
600 
601  attr = (struct varlena *) DatumGetPointer(values[i]);
602 
603  /* don't recursively indirect */
604  if (VARATT_IS_EXTERNAL_INDIRECT(attr))
605  continue;
606 
607  /* copy datum, so it still lives later */
608  if (VARATT_IS_EXTERNAL_ONDISK(attr))
609  attr = detoast_external_attr(attr);
610  else
611  {
612  struct varlena *oldattr = attr;
613 
614  attr = palloc0(VARSIZE_ANY(oldattr));
615  memcpy(attr, oldattr, VARSIZE_ANY(oldattr));
616  }
617 
618  /* build indirection Datum */
619  new_attr = (struct varlena *) palloc0(INDIRECT_POINTER_SIZE);
620  redirect_pointer.pointer = attr;
622  memcpy(VARDATA_EXTERNAL(new_attr), &redirect_pointer,
623  sizeof(redirect_pointer));
624 
625  values[i] = PointerGetDatum(new_attr);
626  }
627 
628  newtup = heap_form_tuple(tupdesc, values, nulls);
629  pfree(values);
630  pfree(nulls);
631  ReleaseTupleDesc(tupdesc);
632 
633  MemoryContextSwitchTo(old_context);
634 
635  /*
636  * We intentionally don't use PG_RETURN_HEAPTUPLEHEADER here, because that
637  * would cause the indirect toast pointers to be flattened out of the
638  * tuple immediately, rendering subsequent testing irrelevant. So just
639  * return the HeapTupleHeader pointer as-is. This violates the general
640  * rule that composite Datums shouldn't contain toast pointers, but so
641  * long as the regression test scripts don't insert the result of this
642  * function into a container type (record, array, etc) it should be OK.
643  */
644  PG_RETURN_POINTER(newtup->t_data);
645 }
static Datum values[MAXATTR]
Definition: bootstrap.c:152
struct varlena * detoast_external_attr(struct varlena *attr)
Definition: detoast.c:45
#define INDIRECT_POINTER_SIZE
Definition: detoast.h:34
#define PG_GETARG_HEAPTUPLEHEADER(n)
Definition: fmgr.h:312
HeapTuple heap_form_tuple(TupleDesc tupleDescriptor, const Datum *values, const bool *isnull)
Definition: heaptuple.c:1116
void heap_deform_tuple(HeapTuple tuple, TupleDesc tupleDesc, Datum *values, bool *isnull)
Definition: heaptuple.c:1345
#define HeapTupleHeaderGetTypMod(tup)
Definition: htup_details.h:466
#define HeapTupleHeaderGetTypeId(tup)
Definition: htup_details.h:456
#define HeapTupleHeaderGetDatumLength(tup)
Definition: htup_details.h:450
static void ItemPointerSetInvalid(ItemPointerData *pointer)
Definition: itemptr.h:184
MemoryContext TopTransactionContext
Definition: mcxt.c:154
void pfree(void *pointer)
Definition: mcxt.c:1520
void * palloc0(Size size)
Definition: mcxt.c:1346
int16 attlen
Definition: pg_attribute.h:59
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:322
uintptr_t Datum
Definition: postgres.h:64
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:312
#define InvalidOid
Definition: postgres_ext.h:36
MemoryContextSwitchTo(old_ctx)
ItemPointerData t_self
Definition: htup.h:65
uint32 t_len
Definition: htup.h:64
HeapTupleHeader t_data
Definition: htup.h:68
Oid t_tableOid
Definition: htup.h:66
Definition: c.h:687
#define ReleaseTupleDesc(tupdesc)
Definition: tupdesc.h:122
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
TupleDesc lookup_rowtype_tupdesc(Oid type_id, int32 typmod)
Definition: typcache.c:1833
#define VARATT_IS_EXTERNAL_ONDISK(PTR)
Definition: varatt.h:290
#define VARATT_IS_EXTERNAL_INDIRECT(PTR)
Definition: varatt.h:292
#define VARSIZE_ANY(PTR)
Definition: varatt.h:311
#define SET_VARTAG_EXTERNAL(PTR, tag)
Definition: varatt.h:309
#define VARDATA_EXTERNAL(PTR)
Definition: varatt.h:286
@ VARTAG_INDIRECT
Definition: varatt.h:86

References attlen, DatumGetPointer(), detoast_external_attr(), heap_deform_tuple(), heap_form_tuple(), HeapTupleHeaderGetDatumLength, HeapTupleHeaderGetTypeId, HeapTupleHeaderGetTypMod, i, INDIRECT_POINTER_SIZE, InvalidOid, ItemPointerSetInvalid(), lookup_rowtype_tupdesc(), MemoryContextSwitchTo(), TupleDescData::natts, palloc(), palloc0(), pfree(), PG_GETARG_HEAPTUPLEHEADER, PG_RETURN_POINTER, varatt_indirect::pointer, PointerGetDatum(), ReleaseTupleDesc, SET_VARTAG_EXTERNAL, HeapTupleData::t_data, HeapTupleData::t_len, HeapTupleData::t_self, HeapTupleData::t_tableOid, TopTransactionContext, TupleDescAttr, values, VARATT_IS_EXTERNAL_INDIRECT, VARATT_IS_EXTERNAL_ONDISK, VARDATA_EXTERNAL, VARSIZE_ANY, and VARTAG_INDIRECT.

◆ overpaid()

Datum overpaid ( PG_FUNCTION_ARGS  )

Definition at line 142 of file regress.c.

143 {
145  bool isnull;
146  int32 salary;
147 
148  salary = DatumGetInt32(GetAttributeByName(tuple, "salary", &isnull));
149  if (isnull)
150  PG_RETURN_NULL();
151  PG_RETURN_BOOL(salary > 699);
152 }
Datum GetAttributeByName(HeapTupleHeader tuple, const char *attname, bool *isNull)
Definition: execUtils.c:995
static int32 DatumGetInt32(Datum X)
Definition: postgres.h:202

References DatumGetInt32(), GetAttributeByName(), PG_GETARG_HEAPTUPLEHEADER, PG_RETURN_BOOL, and PG_RETURN_NULL.

◆ PG_FUNCTION_INFO_V1() [1/21]

PG_FUNCTION_INFO_V1 ( binary_coercible  )

◆ PG_FUNCTION_INFO_V1() [2/21]

PG_FUNCTION_INFO_V1 ( int44in  )

◆ PG_FUNCTION_INFO_V1() [3/21]

PG_FUNCTION_INFO_V1 ( int44out  )

◆ PG_FUNCTION_INFO_V1() [4/21]

PG_FUNCTION_INFO_V1 ( interpt_pp  )

◆ PG_FUNCTION_INFO_V1() [5/21]

PG_FUNCTION_INFO_V1 ( make_tuple_indirect  )

◆ PG_FUNCTION_INFO_V1() [6/21]

PG_FUNCTION_INFO_V1 ( overpaid  )

◆ PG_FUNCTION_INFO_V1() [7/21]

PG_FUNCTION_INFO_V1 ( pt_in_widget  )

◆ PG_FUNCTION_INFO_V1() [8/21]

PG_FUNCTION_INFO_V1 ( regress_setenv  )

◆ PG_FUNCTION_INFO_V1() [9/21]

PG_FUNCTION_INFO_V1 ( reverse_name  )

◆ PG_FUNCTION_INFO_V1() [10/21]

PG_FUNCTION_INFO_V1 ( set_ttdummy  )

◆ PG_FUNCTION_INFO_V1() [11/21]

PG_FUNCTION_INFO_V1 ( test_atomic_ops  )

◆ PG_FUNCTION_INFO_V1() [12/21]

PG_FUNCTION_INFO_V1 ( test_canonicalize_path  )

◆ PG_FUNCTION_INFO_V1() [13/21]

PG_FUNCTION_INFO_V1 ( test_enc_conversion  )

◆ PG_FUNCTION_INFO_V1() [14/21]

PG_FUNCTION_INFO_V1 ( test_fdw_handler  )

◆ PG_FUNCTION_INFO_V1() [15/21]

PG_FUNCTION_INFO_V1 ( test_opclass_options_func  )

◆ PG_FUNCTION_INFO_V1() [16/21]

PG_FUNCTION_INFO_V1 ( test_support_func  )

◆ PG_FUNCTION_INFO_V1() [17/21]

PG_FUNCTION_INFO_V1 ( trigger_return_old  )

◆ PG_FUNCTION_INFO_V1() [18/21]

PG_FUNCTION_INFO_V1 ( ttdummy  )

◆ PG_FUNCTION_INFO_V1() [19/21]

PG_FUNCTION_INFO_V1 ( wait_pid  )

◆ PG_FUNCTION_INFO_V1() [20/21]

PG_FUNCTION_INFO_V1 ( widget_in  )

◆ PG_FUNCTION_INFO_V1() [21/21]

PG_FUNCTION_INFO_V1 ( widget_out  )

◆ pt_in_widget()

Datum pt_in_widget ( PG_FUNCTION_ARGS  )

Definition at line 217 of file regress.c.

218 {
219  Point *point = PG_GETARG_POINT_P(0);
220  WIDGET *widget = (WIDGET *) PG_GETARG_POINTER(1);
221  float8 distance;
222 
224  PointPGetDatum(point),
225  PointPGetDatum(&widget->center)));
226 
227  PG_RETURN_BOOL(distance < widget->radius);
228 }
double float8
Definition: c.h:630
#define PG_GETARG_POINT_P(n)
Definition: geo_decls.h:185
static Datum PointPGetDatum(const Point *X)
Definition: geo_decls.h:181
Datum point_distance(PG_FUNCTION_ARGS)
Definition: geo_ops.c:1993
static float8 DatumGetFloat8(Datum X)
Definition: postgres.h:494
Point center
Definition: regress.c:161

References WIDGET::center, DatumGetFloat8(), DirectFunctionCall2, PG_GETARG_POINT_P, PG_GETARG_POINTER, PG_RETURN_BOOL, point_distance(), and PointPGetDatum().

◆ regress_lseg_construct()

static void regress_lseg_construct ( LSEG lseg,
Point pt1,
Point pt2 
)
static

Definition at line 131 of file regress.c.

132 {
133  lseg->p[0].x = pt1->x;
134  lseg->p[0].y = pt1->y;
135  lseg->p[1].x = pt2->x;
136  lseg->p[1].y = pt2->y;
137 }
Point p[2]
Definition: geo_decls.h:108
float8 y
Definition: geo_decls.h:99
float8 x
Definition: geo_decls.h:98

References LSEG::p, Point::x, and Point::y.

Referenced by interpt_pp().

◆ regress_setenv()

Datum regress_setenv ( PG_FUNCTION_ARGS  )

Definition at line 650 of file regress.c.

651 {
652  char *envvar = text_to_cstring(PG_GETARG_TEXT_PP(0));
653  char *envval = text_to_cstring(PG_GETARG_TEXT_PP(1));
654 
655  if (!superuser())
656  elog(ERROR, "must be superuser to change environment variables");
657 
658  if (setenv(envvar, envval, 1) != 0)
659  elog(ERROR, "could not set environment variable: %m");
660 
661  PG_RETURN_VOID();
662 }
#define elog(elevel,...)
Definition: elog.h:224
#define PG_RETURN_VOID()
Definition: fmgr.h:349
#define PG_GETARG_TEXT_PP(n)
Definition: fmgr.h:309
bool superuser(void)
Definition: superuser.c:46
char * text_to_cstring(const text *t)
Definition: varlena.c:217
#define setenv(x, y, z)
Definition: win32_port.h:537

References elog, ERROR, PG_GETARG_TEXT_PP, PG_RETURN_VOID, setenv, superuser(), and text_to_cstring().

◆ reverse_name()

Datum reverse_name ( PG_FUNCTION_ARGS  )

Definition at line 233 of file regress.c.

234 {
235  char *string = PG_GETARG_CSTRING(0);
236  int i;
237  int len;
238  char *new_string;
239 
240  new_string = palloc0(NAMEDATALEN);
241  for (i = 0; i < NAMEDATALEN && string[i]; ++i)
242  ;
243  if (i == NAMEDATALEN || !string[i])
244  --i;
245  len = i;
246  for (; i >= 0; --i)
247  new_string[len - i] = string[i];
248  PG_RETURN_CSTRING(new_string);
249 }
#define NAMEDATALEN
const void size_t len

References i, len, NAMEDATALEN, palloc0(), PG_GETARG_CSTRING, and PG_RETURN_CSTRING.

◆ set_ttdummy()

Datum set_ttdummy ( PG_FUNCTION_ARGS  )

Definition at line 464 of file regress.c.

465 {
466  int32 on = PG_GETARG_INT32(0);
467 
468  if (ttoff) /* OFF currently */
469  {
470  if (on == 0)
471  PG_RETURN_INT32(0);
472 
473  /* turn ON */
474  ttoff = false;
475  PG_RETURN_INT32(0);
476  }
477 
478  /* ON currently */
479  if (on != 0)
480  PG_RETURN_INT32(1);
481 
482  /* turn OFF */
483  ttoff = true;
484 
485  PG_RETURN_INT32(1);
486 }
#define PG_RETURN_INT32(x)
Definition: fmgr.h:354
#define PG_GETARG_INT32(n)
Definition: fmgr.h:269
static bool ttoff
Definition: regress.c:270

References PG_GETARG_INT32, PG_RETURN_INT32, and ttoff.

◆ test_atomic_flag()

static void test_atomic_flag ( void  )
static

Definition at line 688 of file regress.c.

689 {
691 
701 }
static void pg_atomic_clear_flag(volatile pg_atomic_flag *ptr)
Definition: atomics.h:202
static bool pg_atomic_test_set_flag(volatile pg_atomic_flag *ptr)
Definition: atomics.h:178
static bool pg_atomic_unlocked_test_flag(volatile pg_atomic_flag *ptr)
Definition: atomics.h:191
static void pg_atomic_init_flag(volatile pg_atomic_flag *ptr)
Definition: atomics.h:165
#define EXPECT_TRUE(expr)
Definition: regress.c:49
char * flag(int b)
Definition: test-ctype.c:33

References EXPECT_TRUE, flag(), pg_atomic_clear_flag(), pg_atomic_init_flag(), pg_atomic_test_set_flag(), and pg_atomic_unlocked_test_flag().

Referenced by test_atomic_ops().

◆ test_atomic_ops()

Datum test_atomic_ops ( PG_FUNCTION_ARGS  )

Definition at line 977 of file regress.c.

978 {
980 
982 
984 
985  /*
986  * Arguably this shouldn't be tested as part of this function, but it's
987  * closely enough related that that seems ok for now.
988  */
989  test_spinlock();
990 
992 
993  PG_RETURN_BOOL(true);
994 }
static void test_spinlock(void)
Definition: regress.c:829
static void test_atomic_spin_nest(void)
Definition: regress.c:940
static void test_atomic_uint32(void)
Definition: regress.c:704
static void test_atomic_flag(void)
Definition: regress.c:688
static void test_atomic_uint64(void)
Definition: regress.c:776

References PG_RETURN_BOOL, test_atomic_flag(), test_atomic_spin_nest(), test_atomic_uint32(), test_atomic_uint64(), and test_spinlock().

◆ test_atomic_spin_nest()

static void test_atomic_spin_nest ( void  )
static

Definition at line 940 of file regress.c.

941 {
942  slock_t lock;
943 #define NUM_TEST_ATOMICS (NUM_SPINLOCK_SEMAPHORES + NUM_ATOMICS_SEMAPHORES + 27)
946 
947  SpinLockInit(&lock);
948 
949  for (int i = 0; i < NUM_TEST_ATOMICS; i++)
950  {
951  pg_atomic_init_u32(&atomics32[i], 0);
952  pg_atomic_init_u64(&atomics64[i], 0);
953  }
954 
955  /* just so it's not all zeroes */
956  for (int i = 0; i < NUM_TEST_ATOMICS; i++)
957  {
958  EXPECT_EQ_U32(pg_atomic_fetch_add_u32(&atomics32[i], i), 0);
959  EXPECT_EQ_U64(pg_atomic_fetch_add_u64(&atomics64[i], i), 0);
960  }
961 
962  /* test whether we can do atomic op with lock held */
963  SpinLockAcquire(&lock);
964  for (int i = 0; i < NUM_TEST_ATOMICS; i++)
965  {
966  EXPECT_EQ_U32(pg_atomic_fetch_sub_u32(&atomics32[i], i), i);
967  EXPECT_EQ_U32(pg_atomic_read_u32(&atomics32[i]), 0);
968  EXPECT_EQ_U64(pg_atomic_fetch_sub_u64(&atomics64[i], i), i);
969  EXPECT_EQ_U64(pg_atomic_read_u64(&atomics64[i]), 0);
970  }
971  SpinLockRelease(&lock);
972 }
static uint32 pg_atomic_fetch_sub_u32(volatile pg_atomic_uint32 *ptr, int32 sub_)
Definition: atomics.h:376
static void pg_atomic_init_u32(volatile pg_atomic_uint32 *ptr, uint32 val)
Definition: atomics.h:216
static uint32 pg_atomic_fetch_add_u32(volatile pg_atomic_uint32 *ptr, int32 add_)
Definition: atomics.h:361
static uint64 pg_atomic_fetch_add_u64(volatile pg_atomic_uint64 *ptr, int64 add_)
Definition: atomics.h:518
static uint32 pg_atomic_read_u32(volatile pg_atomic_uint32 *ptr)
Definition: atomics.h:234
static void pg_atomic_init_u64(volatile pg_atomic_uint64 *ptr, uint64 val)
Definition: atomics.h:448
static uint64 pg_atomic_read_u64(volatile pg_atomic_uint64 *ptr)
Definition: atomics.h:462
static uint64 pg_atomic_fetch_sub_u64(volatile pg_atomic_uint64 *ptr, int64 sub_)
Definition: atomics.h:527
#define EXPECT_EQ_U32(result_expr, expected_expr)
Definition: regress.c:57
#define NUM_TEST_ATOMICS
#define EXPECT_EQ_U64(result_expr, expected_expr)
Definition: regress.c:67
int slock_t
Definition: s_lock.h:735
#define SpinLockInit(lock)
Definition: spin.h:60
#define SpinLockRelease(lock)
Definition: spin.h:64
#define SpinLockAcquire(lock)
Definition: spin.h:62

References EXPECT_EQ_U32, EXPECT_EQ_U64, i, NUM_TEST_ATOMICS, pg_atomic_fetch_add_u32(), pg_atomic_fetch_add_u64(), pg_atomic_fetch_sub_u32(), pg_atomic_fetch_sub_u64(), pg_atomic_init_u32(), pg_atomic_init_u64(), pg_atomic_read_u32(), pg_atomic_read_u64(), SpinLockAcquire, SpinLockInit, and SpinLockRelease.

Referenced by test_atomic_ops().

◆ test_atomic_uint32()

static void test_atomic_uint32 ( void  )
static

Definition at line 704 of file regress.c.

705 {
706  pg_atomic_uint32 var;
707  uint32 expected;
708  int i;
709 
710  pg_atomic_init_u32(&var, 0);
712  pg_atomic_write_u32(&var, 3);
715  3);
718  EXPECT_EQ_U32(pg_atomic_add_fetch_u32(&var, 10), 10);
721 
722  /* test around numerical limits */
723  EXPECT_EQ_U32(pg_atomic_fetch_add_u32(&var, INT_MAX), 0);
724  EXPECT_EQ_U32(pg_atomic_fetch_add_u32(&var, INT_MAX), INT_MAX);
725  pg_atomic_fetch_add_u32(&var, 2); /* wrap to 0 */
728  PG_INT16_MAX);
730  2 * PG_INT16_MAX + 1);
732  PG_INT16_MAX);
733  pg_atomic_fetch_add_u32(&var, 1); /* top up to UINT_MAX */
734  EXPECT_EQ_U32(pg_atomic_read_u32(&var), UINT_MAX);
735  EXPECT_EQ_U32(pg_atomic_fetch_sub_u32(&var, INT_MAX), UINT_MAX);
736  EXPECT_EQ_U32(pg_atomic_read_u32(&var), (uint32) INT_MAX + 1);
737  EXPECT_EQ_U32(pg_atomic_sub_fetch_u32(&var, INT_MAX), 1);
738  pg_atomic_sub_fetch_u32(&var, 1);
739  expected = PG_INT16_MAX;
740  EXPECT_TRUE(!pg_atomic_compare_exchange_u32(&var, &expected, 1));
741  expected = PG_INT16_MAX + 1;
742  EXPECT_TRUE(!pg_atomic_compare_exchange_u32(&var, &expected, 1));
743  expected = PG_INT16_MIN;
744  EXPECT_TRUE(!pg_atomic_compare_exchange_u32(&var, &expected, 1));
745  expected = PG_INT16_MIN - 1;
746  EXPECT_TRUE(!pg_atomic_compare_exchange_u32(&var, &expected, 1));
747 
748  /* fail exchange because of old expected */
749  expected = 10;
750  EXPECT_TRUE(!pg_atomic_compare_exchange_u32(&var, &expected, 1));
751 
752  /* CAS is allowed to fail due to interrupts, try a couple of times */
753  for (i = 0; i < 1000; i++)
754  {
755  expected = 0;
756  if (!pg_atomic_compare_exchange_u32(&var, &expected, 1))
757  break;
758  }
759  if (i == 1000)
760  elog(ERROR, "atomic_compare_exchange_u32() never succeeded");
762  pg_atomic_write_u32(&var, 0);
763 
764  /* try setting flagbits */
765  EXPECT_TRUE(!(pg_atomic_fetch_or_u32(&var, 1) & 1));
766  EXPECT_TRUE(pg_atomic_fetch_or_u32(&var, 2) & 1);
768  /* try clearing flagbits */
769  EXPECT_EQ_U32(pg_atomic_fetch_and_u32(&var, ~2) & 3, 3);
771  /* no bits set anymore */
773 }
static uint32 pg_atomic_fetch_and_u32(volatile pg_atomic_uint32 *ptr, uint32 and_)
Definition: atomics.h:391
static bool pg_atomic_compare_exchange_u32(volatile pg_atomic_uint32 *ptr, uint32 *expected, uint32 newval)
Definition: atomics.h:344
static uint32 pg_atomic_fetch_or_u32(volatile pg_atomic_uint32 *ptr, uint32 or_)
Definition: atomics.h:405
static uint32 pg_atomic_sub_fetch_u32(volatile pg_atomic_uint32 *ptr, int32 sub_)
Definition: atomics.h:434
static uint32 pg_atomic_add_fetch_u32(volatile pg_atomic_uint32 *ptr, int32 add_)
Definition: atomics.h:419
static void pg_atomic_write_u32(volatile pg_atomic_uint32 *ptr, uint32 val)
Definition: atomics.h:271
static uint32 pg_atomic_exchange_u32(volatile pg_atomic_uint32 *ptr, uint32 newval)
Definition: atomics.h:325
unsigned int uint32
Definition: c.h:506
#define PG_INT16_MIN
Definition: c.h:585
#define PG_INT16_MAX
Definition: c.h:586

References elog, ERROR, EXPECT_EQ_U32, EXPECT_TRUE, i, pg_atomic_add_fetch_u32(), pg_atomic_compare_exchange_u32(), pg_atomic_exchange_u32(), pg_atomic_fetch_add_u32(), pg_atomic_fetch_and_u32(), pg_atomic_fetch_or_u32(), pg_atomic_fetch_sub_u32(), pg_atomic_init_u32(), pg_atomic_read_u32(), pg_atomic_sub_fetch_u32(), pg_atomic_write_u32(), PG_INT16_MAX, and PG_INT16_MIN.

Referenced by test_atomic_ops().

◆ test_atomic_uint64()

static void test_atomic_uint64 ( void  )
static

Definition at line 776 of file regress.c.

777 {
778  pg_atomic_uint64 var;
779  uint64 expected;
780  int i;
781 
782  pg_atomic_init_u64(&var, 0);
784  pg_atomic_write_u64(&var, 3);
787  3);
790  EXPECT_EQ_U64(pg_atomic_add_fetch_u64(&var, 10), 10);
793 
794  /* fail exchange because of old expected */
795  expected = 10;
796  EXPECT_TRUE(!pg_atomic_compare_exchange_u64(&var, &expected, 1));
797 
798  /* CAS is allowed to fail due to interrupts, try a couple of times */
799  for (i = 0; i < 100; i++)
800  {
801  expected = 0;
802  if (!pg_atomic_compare_exchange_u64(&var, &expected, 1))
803  break;
804  }
805  if (i == 100)
806  elog(ERROR, "atomic_compare_exchange_u64() never succeeded");
808 
809  pg_atomic_write_u64(&var, 0);
810 
811  /* try setting flagbits */
812  EXPECT_TRUE(!(pg_atomic_fetch_or_u64(&var, 1) & 1));
813  EXPECT_TRUE(pg_atomic_fetch_or_u64(&var, 2) & 1);
815  /* try clearing flagbits */
816  EXPECT_EQ_U64((pg_atomic_fetch_and_u64(&var, ~2) & 3), 3);
818  /* no bits set anymore */
820 }
static void pg_atomic_write_u64(volatile pg_atomic_uint64 *ptr, uint64 val)
Definition: atomics.h:480
static bool pg_atomic_compare_exchange_u64(volatile pg_atomic_uint64 *ptr, uint64 *expected, uint64 newval)
Definition: atomics.h:507
static uint64 pg_atomic_sub_fetch_u64(volatile pg_atomic_uint64 *ptr, int64 sub_)
Definition: atomics.h:564
static uint64 pg_atomic_fetch_and_u64(volatile pg_atomic_uint64 *ptr, uint64 and_)
Definition: atomics.h:537
static uint64 pg_atomic_fetch_or_u64(volatile pg_atomic_uint64 *ptr, uint64 or_)
Definition: atomics.h:546
static uint64 pg_atomic_add_fetch_u64(volatile pg_atomic_uint64 *ptr, int64 add_)
Definition: atomics.h:555
static uint64 pg_atomic_exchange_u64(volatile pg_atomic_uint64 *ptr, uint64 newval)
Definition: atomics.h:498

References elog, ERROR, EXPECT_EQ_U64, EXPECT_TRUE, i, pg_atomic_add_fetch_u64(), pg_atomic_compare_exchange_u64(), pg_atomic_exchange_u64(), pg_atomic_fetch_add_u64(), pg_atomic_fetch_and_u64(), pg_atomic_fetch_or_u64(), pg_atomic_fetch_sub_u64(), pg_atomic_init_u64(), pg_atomic_read_u64(), pg_atomic_sub_fetch_u64(), and pg_atomic_write_u64().

Referenced by test_atomic_ops().

◆ test_canonicalize_path()

Datum test_canonicalize_path ( PG_FUNCTION_ARGS  )

Definition at line 542 of file regress.c.

543 {
544  char *path = text_to_cstring(PG_GETARG_TEXT_PP(0));
545 
546  canonicalize_path(path);
548 }
#define PG_RETURN_TEXT_P(x)
Definition: fmgr.h:372
void canonicalize_path(char *path)
Definition: path.c:264
text * cstring_to_text(const char *s)
Definition: varlena.c:184

References canonicalize_path(), cstring_to_text(), PG_GETARG_TEXT_PP, PG_RETURN_TEXT_P, and text_to_cstring().

◆ test_enc_conversion()

Datum test_enc_conversion ( PG_FUNCTION_ARGS  )

Definition at line 1100 of file regress.c.

1101 {
1102  bytea *string = PG_GETARG_BYTEA_PP(0);
1103  char *src_encoding_name = NameStr(*PG_GETARG_NAME(1));
1104  int src_encoding = pg_char_to_encoding(src_encoding_name);
1105  char *dest_encoding_name = NameStr(*PG_GETARG_NAME(2));
1106  int dest_encoding = pg_char_to_encoding(dest_encoding_name);
1107  bool noError = PG_GETARG_BOOL(3);
1108  TupleDesc tupdesc;
1109  char *src;
1110  char *dst;
1111  bytea *retval;
1112  Size srclen;
1113  Size dstsize;
1114  Oid proc;
1115  int convertedbytes;
1116  int dstlen;
1117  Datum values[2];
1118  bool nulls[2] = {0};
1119  HeapTuple tuple;
1120 
1121  if (src_encoding < 0)
1122  ereport(ERROR,
1123  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1124  errmsg("invalid source encoding name \"%s\"",
1125  src_encoding_name)));
1126  if (dest_encoding < 0)
1127  ereport(ERROR,
1128  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1129  errmsg("invalid destination encoding name \"%s\"",
1130  dest_encoding_name)));
1131 
1132  /* Build a tuple descriptor for our result type */
1133  if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
1134  elog(ERROR, "return type must be a row type");
1135  tupdesc = BlessTupleDesc(tupdesc);
1136 
1137  srclen = VARSIZE_ANY_EXHDR(string);
1138  src = VARDATA_ANY(string);
1139 
1140  if (src_encoding == dest_encoding)
1141  {
1142  /* just check that the source string is valid */
1143  int oklen;
1144 
1145  oklen = pg_encoding_verifymbstr(src_encoding, src, srclen);
1146 
1147  if (oklen == srclen)
1148  {
1149  convertedbytes = oklen;
1150  retval = string;
1151  }
1152  else if (!noError)
1153  {
1154  report_invalid_encoding(src_encoding, src + oklen, srclen - oklen);
1155  }
1156  else
1157  {
1158  /*
1159  * build bytea data type structure.
1160  */
1161  Assert(oklen < srclen);
1162  convertedbytes = oklen;
1163  retval = (bytea *) palloc(oklen + VARHDRSZ);
1164  SET_VARSIZE(retval, oklen + VARHDRSZ);
1165  memcpy(VARDATA(retval), src, oklen);
1166  }
1167  }
1168  else
1169  {
1170  proc = FindDefaultConversionProc(src_encoding, dest_encoding);
1171  if (!OidIsValid(proc))
1172  ereport(ERROR,
1173  (errcode(ERRCODE_UNDEFINED_FUNCTION),
1174  errmsg("default conversion function for encoding \"%s\" to \"%s\" does not exist",
1175  pg_encoding_to_char(src_encoding),
1176  pg_encoding_to_char(dest_encoding))));
1177 
1178  if (srclen >= (MaxAllocSize / (Size) MAX_CONVERSION_GROWTH))
1179  ereport(ERROR,
1180  (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
1181  errmsg("out of memory"),
1182  errdetail("String of %d bytes is too long for encoding conversion.",
1183  (int) srclen)));
1184 
1185  dstsize = (Size) srclen * MAX_CONVERSION_GROWTH + 1;
1186  dst = MemoryContextAlloc(CurrentMemoryContext, dstsize);
1187 
1188  /* perform conversion */
1189  convertedbytes = pg_do_encoding_conversion_buf(proc,
1190  src_encoding,
1191  dest_encoding,
1192  (unsigned char *) src, srclen,
1193  (unsigned char *) dst, dstsize,
1194  noError);
1195  dstlen = strlen(dst);
1196 
1197  /*
1198  * build bytea data type structure.
1199  */
1200  retval = (bytea *) palloc(dstlen + VARHDRSZ);
1201  SET_VARSIZE(retval, dstlen + VARHDRSZ);
1202  memcpy(VARDATA(retval), dst, dstlen);
1203 
1204  pfree(dst);
1205  }
1206 
1207  values[0] = Int32GetDatum(convertedbytes);
1208  values[1] = PointerGetDatum(retval);
1209  tuple = heap_form_tuple(tupdesc, values, nulls);
1210 
1212 }
#define NameStr(name)
Definition: c.h:746
#define VARHDRSZ
Definition: c.h:692
#define Assert(condition)
Definition: c.h:858
#define OidIsValid(objectId)
Definition: c.h:775
size_t Size
Definition: c.h:605
int errdetail(const char *fmt,...)
Definition: elog.c:1203
int errcode(int sqlerrcode)
Definition: elog.c:857
int errmsg(const char *fmt,...)
Definition: elog.c:1070
#define ereport(elevel,...)
Definition: elog.h:149
TupleDesc BlessTupleDesc(TupleDesc tupdesc)
Definition: execTuples.c:2158
#define PG_GETARG_BYTEA_PP(n)
Definition: fmgr.h:308
#define PG_GETARG_NAME(n)
Definition: fmgr.h:278
#define PG_GETARG_BOOL(n)
Definition: fmgr.h:274
TypeFuncClass get_call_result_type(FunctionCallInfo fcinfo, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:276
@ TYPEFUNC_COMPOSITE
Definition: funcapi.h:149
static Datum HeapTupleGetDatum(const HeapTupleData *tuple)
Definition: funcapi.h:230
int pg_do_encoding_conversion_buf(Oid proc, int src_encoding, int dest_encoding, unsigned char *src, int srclen, unsigned char *dest, int destlen, bool noError)
Definition: mbutils.c:469
void report_invalid_encoding(int encoding, const char *mbstr, int len)
Definition: mbutils.c:1698
MemoryContext CurrentMemoryContext
Definition: mcxt.c:143
void * MemoryContextAlloc(MemoryContext context, Size size)
Definition: mcxt.c:1180
#define MaxAllocSize
Definition: memutils.h:40
Oid FindDefaultConversionProc(int32 for_encoding, int32 to_encoding)
Definition: namespace.c:4065
#define MAX_CONVERSION_GROWTH
Definition: pg_wchar.h:302
#define pg_encoding_to_char
Definition: pg_wchar.h:630
#define pg_char_to_encoding
Definition: pg_wchar.h:629
static Datum Int32GetDatum(int32 X)
Definition: postgres.h:212
char string[11]
Definition: preproc-type.c:52
#define VARDATA(PTR)
Definition: varatt.h:278
#define VARDATA_ANY(PTR)
Definition: varatt.h:324
#define SET_VARSIZE(PTR, len)
Definition: varatt.h:305
#define VARSIZE_ANY_EXHDR(PTR)
Definition: varatt.h:317
int pg_encoding_verifymbstr(int encoding, const char *mbstr, int len)
Definition: wchar.c:2116

References Assert, BlessTupleDesc(), CurrentMemoryContext, elog, ereport, errcode(), errdetail(), errmsg(), ERROR, FindDefaultConversionProc(), get_call_result_type(), heap_form_tuple(), HeapTupleGetDatum(), Int32GetDatum(), MAX_CONVERSION_GROWTH, MaxAllocSize, MemoryContextAlloc(), NameStr, OidIsValid, palloc(), pfree(), pg_char_to_encoding, pg_do_encoding_conversion_buf(), pg_encoding_to_char, pg_encoding_verifymbstr(), PG_GETARG_BOOL, PG_GETARG_BYTEA_PP, PG_GETARG_NAME, PG_RETURN_DATUM, PointerGetDatum(), report_invalid_encoding(), SET_VARSIZE, TYPEFUNC_COMPOSITE, values, VARDATA, VARDATA_ANY, VARHDRSZ, and VARSIZE_ANY_EXHDR.

◆ test_fdw_handler()

Datum test_fdw_handler ( PG_FUNCTION_ARGS  )

Definition at line 998 of file regress.c.

999 {
1000  elog(ERROR, "test_fdw_handler is not implemented");
1001  PG_RETURN_NULL();
1002 }

References elog, ERROR, and PG_RETURN_NULL.

◆ test_opclass_options_func()

Datum test_opclass_options_func ( PG_FUNCTION_ARGS  )

Definition at line 1079 of file regress.c.

1080 {
1081  PG_RETURN_NULL();
1082 }

References PG_RETURN_NULL.

◆ test_spinlock()

static void test_spinlock ( void  )
static

Definition at line 829 of file regress.c.

830 {
831  /*
832  * Basic tests for spinlocks, as well as the underlying operations.
833  *
834  * We embed the spinlock in a struct with other members to test that the
835  * spinlock operations don't perform too wide writes.
836  */
837  {
838  struct test_lock_struct
839  {
840  char data_before[4];
841  slock_t lock;
842  char data_after[4];
843  } struct_w_lock;
844 
845  memcpy(struct_w_lock.data_before, "abcd", 4);
846  memcpy(struct_w_lock.data_after, "ef12", 4);
847 
848  /* test basic operations via the SpinLock* API */
849  SpinLockInit(&struct_w_lock.lock);
850  SpinLockAcquire(&struct_w_lock.lock);
851  SpinLockRelease(&struct_w_lock.lock);
852 
853  /* test basic operations via underlying S_* API */
854  S_INIT_LOCK(&struct_w_lock.lock);
855  S_LOCK(&struct_w_lock.lock);
856  S_UNLOCK(&struct_w_lock.lock);
857 
858  /* and that "contended" acquisition works */
859  s_lock(&struct_w_lock.lock, "testfile", 17, "testfunc");
860  S_UNLOCK(&struct_w_lock.lock);
861 
862  /*
863  * Check, using TAS directly, that a single spin cycle doesn't block
864  * when acquiring an already acquired lock.
865  */
866 #ifdef TAS
867  S_LOCK(&struct_w_lock.lock);
868 
869  if (!TAS(&struct_w_lock.lock))
870  elog(ERROR, "acquired already held spinlock");
871 
872 #ifdef TAS_SPIN
873  if (!TAS_SPIN(&struct_w_lock.lock))
874  elog(ERROR, "acquired already held spinlock");
875 #endif /* defined(TAS_SPIN) */
876 
877  S_UNLOCK(&struct_w_lock.lock);
878 #endif /* defined(TAS) */
879 
880  /*
881  * Verify that after all of this the non-lock contents are still
882  * correct.
883  */
884  if (memcmp(struct_w_lock.data_before, "abcd", 4) != 0)
885  elog(ERROR, "padding before spinlock modified");
886  if (memcmp(struct_w_lock.data_after, "ef12", 4) != 0)
887  elog(ERROR, "padding after spinlock modified");
888  }
889 
890  /*
891  * Ensure that allocating more than INT32_MAX emulated spinlocks works.
892  * That's interesting because the spinlock emulation uses a 32bit integer
893  * to map spinlocks onto semaphores. There've been bugs...
894  */
895 #ifndef HAVE_SPINLOCKS
896  {
897  /*
898  * Initialize enough spinlocks to advance counter close to wraparound.
899  * It's too expensive to perform acquire/release for each, as those
900  * may be syscalls when the spinlock emulation is used (and even just
901  * atomic TAS would be expensive).
902  */
903  for (uint32 i = 0; i < INT32_MAX - 100000; i++)
904  {
905  slock_t lock;
906 
907  SpinLockInit(&lock);
908  }
909 
910  for (uint32 i = 0; i < 200000; i++)
911  {
912  slock_t lock;
913 
914  SpinLockInit(&lock);
915 
916  SpinLockAcquire(&lock);
917  SpinLockRelease(&lock);
918  SpinLockAcquire(&lock);
919  SpinLockRelease(&lock);
920  }
921  }
922 #endif
923 }
int s_lock(volatile slock_t *lock, const char *file, int line, const char *func)
Definition: s_lock.c:99
#define TAS(lock)
Definition: s_lock.h:745
#define S_UNLOCK(lock)
Definition: s_lock.h:743
#define TAS_SPIN(lock)
Definition: s_lock.h:802
#define S_INIT_LOCK(lock)
Definition: s_lock.h:744
#define S_LOCK(lock)
Definition: s_lock.h:756

References elog, ERROR, i, S_INIT_LOCK, S_LOCK, s_lock(), S_UNLOCK, SpinLockAcquire, SpinLockInit, SpinLockRelease, TAS, and TAS_SPIN.

Referenced by test_atomic_ops().

◆ test_support_func()

Datum test_support_func ( PG_FUNCTION_ARGS  )

Definition at line 1006 of file regress.c.

1007 {
1008  Node *rawreq = (Node *) PG_GETARG_POINTER(0);
1009  Node *ret = NULL;
1010 
1011  if (IsA(rawreq, SupportRequestSelectivity))
1012  {
1013  /*
1014  * Assume that the target is int4eq; that's safe as long as we don't
1015  * attach this to any other boolean-returning function.
1016  */
1018  Selectivity s1;
1019 
1020  if (req->is_join)
1021  s1 = join_selectivity(req->root, Int4EqualOperator,
1022  req->args,
1023  req->inputcollid,
1024  req->jointype,
1025  req->sjinfo);
1026  else
1027  s1 = restriction_selectivity(req->root, Int4EqualOperator,
1028  req->args,
1029  req->inputcollid,
1030  req->varRelid);
1031 
1032  req->selectivity = s1;
1033  ret = (Node *) req;
1034  }
1035 
1036  if (IsA(rawreq, SupportRequestCost))
1037  {
1038  /* Provide some generic estimate */
1039  SupportRequestCost *req = (SupportRequestCost *) rawreq;
1040 
1041  req->startup = 0;
1042  req->per_tuple = 2 * cpu_operator_cost;
1043  ret = (Node *) req;
1044  }
1045 
1046  if (IsA(rawreq, SupportRequestRows))
1047  {
1048  /*
1049  * Assume that the target is generate_series_int4; that's safe as long
1050  * as we don't attach this to any other set-returning function.
1051  */
1052  SupportRequestRows *req = (SupportRequestRows *) rawreq;
1053 
1054  if (req->node && IsA(req->node, FuncExpr)) /* be paranoid */
1055  {
1056  List *args = ((FuncExpr *) req->node)->args;
1057  Node *arg1 = linitial(args);
1058  Node *arg2 = lsecond(args);
1059 
1060  if (IsA(arg1, Const) &&
1061  !((Const *) arg1)->constisnull &&
1062  IsA(arg2, Const) &&
1063  !((Const *) arg2)->constisnull)
1064  {
1065  int32 val1 = DatumGetInt32(((Const *) arg1)->constvalue);
1066  int32 val2 = DatumGetInt32(((Const *) arg2)->constvalue);
1067 
1068  req->rows = val2 - val1 + 1;
1069  ret = (Node *) req;
1070  }
1071  }
1072  }
1073 
1074  PG_RETURN_POINTER(ret);
1075 }
double cpu_operator_cost
Definition: costsize.c:123
#define IsA(nodeptr, _type_)
Definition: nodes.h:158
double Selectivity
Definition: nodes.h:250
#define linitial(l)
Definition: pg_list.h:178
#define lsecond(l)
Definition: pg_list.h:183
Selectivity restriction_selectivity(PlannerInfo *root, Oid operatorid, List *args, Oid inputcollid, int varRelid)
Definition: plancat.c:1935
Selectivity join_selectivity(PlannerInfo *root, Oid operatorid, List *args, Oid inputcollid, JoinType jointype, SpecialJoinInfo *sjinfo)
Definition: plancat.c:1974
char * s1
Definition: pg_list.h:54
Definition: nodes.h:129
struct PlannerInfo * root
Definition: supportnodes.h:96
struct SpecialJoinInfo * sjinfo
Definition: supportnodes.h:103

References generate_unaccent_rules::args, SupportRequestSelectivity::args, cpu_operator_cost, DatumGetInt32(), SupportRequestSelectivity::inputcollid, SupportRequestSelectivity::is_join, IsA, join_selectivity(), SupportRequestSelectivity::jointype, linitial, lsecond, SupportRequestRows::node, SupportRequestCost::per_tuple, PG_GETARG_POINTER, PG_RETURN_POINTER, restriction_selectivity(), SupportRequestSelectivity::root, SupportRequestRows::rows, s1, SupportRequestSelectivity::selectivity, SupportRequestSelectivity::sjinfo, SupportRequestCost::startup, and SupportRequestSelectivity::varRelid.

◆ trigger_return_old()

Datum trigger_return_old ( PG_FUNCTION_ARGS  )

Definition at line 254 of file regress.c.

255 {
256  TriggerData *trigdata = (TriggerData *) fcinfo->context;
257  HeapTuple tuple;
258 
259  if (!CALLED_AS_TRIGGER(fcinfo))
260  elog(ERROR, "trigger_return_old: not fired by trigger manager");
261 
262  tuple = trigdata->tg_trigtuple;
263 
264  return PointerGetDatum(tuple);
265 }
if(TABLE==NULL||TABLE_index==NULL)
Definition: isn.c:77
HeapTuple tg_trigtuple
Definition: trigger.h:36
#define CALLED_AS_TRIGGER(fcinfo)
Definition: trigger.h:26

References CALLED_AS_TRIGGER, elog, ERROR, if(), PointerGetDatum(), and TriggerData::tg_trigtuple.

◆ ttdummy()

Datum ttdummy ( PG_FUNCTION_ARGS  )

Definition at line 275 of file regress.c.

276 {
277  TriggerData *trigdata = (TriggerData *) fcinfo->context;
278  Trigger *trigger; /* to get trigger name */
279  char **args; /* arguments */
280  int attnum[2]; /* fnumbers of start/stop columns */
281  Datum oldon,
282  oldoff;
283  Datum newon,
284  newoff;
285  Datum *cvals; /* column values */
286  char *cnulls; /* column nulls */
287  char *relname; /* triggered relation name */
288  Relation rel; /* triggered relation */
289  HeapTuple trigtuple;
290  HeapTuple newtuple = NULL;
291  HeapTuple rettuple;
292  TupleDesc tupdesc; /* tuple description */
293  int natts; /* # of attributes */
294  bool isnull; /* to know is some column NULL or not */
295  int ret;
296  int i;
297 
298  if (!CALLED_AS_TRIGGER(fcinfo))
299  elog(ERROR, "ttdummy: not fired by trigger manager");
300  if (!TRIGGER_FIRED_FOR_ROW(trigdata->tg_event))
301  elog(ERROR, "ttdummy: must be fired for row");
302  if (!TRIGGER_FIRED_BEFORE(trigdata->tg_event))
303  elog(ERROR, "ttdummy: must be fired before event");
304  if (TRIGGER_FIRED_BY_INSERT(trigdata->tg_event))
305  elog(ERROR, "ttdummy: cannot process INSERT event");
306  if (TRIGGER_FIRED_BY_UPDATE(trigdata->tg_event))
307  newtuple = trigdata->tg_newtuple;
308 
309  trigtuple = trigdata->tg_trigtuple;
310 
311  rel = trigdata->tg_relation;
312  relname = SPI_getrelname(rel);
313 
314  /* check if TT is OFF for this relation */
315  if (ttoff) /* OFF - nothing to do */
316  {
317  pfree(relname);
318  return PointerGetDatum((newtuple != NULL) ? newtuple : trigtuple);
319  }
320 
321  trigger = trigdata->tg_trigger;
322 
323  if (trigger->tgnargs != 2)
324  elog(ERROR, "ttdummy (%s): invalid (!= 2) number of arguments %d",
325  relname, trigger->tgnargs);
326 
327  args = trigger->tgargs;
328  tupdesc = rel->rd_att;
329  natts = tupdesc->natts;
330 
331  for (i = 0; i < 2; i++)
332  {
333  attnum[i] = SPI_fnumber(tupdesc, args[i]);
334  if (attnum[i] <= 0)
335  elog(ERROR, "ttdummy (%s): there is no attribute %s",
336  relname, args[i]);
337  if (SPI_gettypeid(tupdesc, attnum[i]) != INT4OID)
338  elog(ERROR, "ttdummy (%s): attribute %s must be of integer type",
339  relname, args[i]);
340  }
341 
342  oldon = SPI_getbinval(trigtuple, tupdesc, attnum[0], &isnull);
343  if (isnull)
344  elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[0]);
345 
346  oldoff = SPI_getbinval(trigtuple, tupdesc, attnum[1], &isnull);
347  if (isnull)
348  elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[1]);
349 
350  if (newtuple != NULL) /* UPDATE */
351  {
352  newon = SPI_getbinval(newtuple, tupdesc, attnum[0], &isnull);
353  if (isnull)
354  elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[0]);
355  newoff = SPI_getbinval(newtuple, tupdesc, attnum[1], &isnull);
356  if (isnull)
357  elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[1]);
358 
359  if (oldon != newon || oldoff != newoff)
360  ereport(ERROR,
361  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
362  errmsg("ttdummy (%s): you cannot change %s and/or %s columns (use set_ttdummy)",
363  relname, args[0], args[1])));
364 
365  if (newoff != TTDUMMY_INFINITY)
366  {
367  pfree(relname); /* allocated in upper executor context */
368  return PointerGetDatum(NULL);
369  }
370  }
371  else if (oldoff != TTDUMMY_INFINITY) /* DELETE */
372  {
373  pfree(relname);
374  return PointerGetDatum(NULL);
375  }
376 
377  newoff = DirectFunctionCall1(nextval, CStringGetTextDatum("ttdummy_seq"));
378  /* nextval now returns int64; coerce down to int32 */
379  newoff = Int32GetDatum((int32) DatumGetInt64(newoff));
380 
381  /* Connect to SPI manager */
382  if ((ret = SPI_connect()) < 0)
383  elog(ERROR, "ttdummy (%s): SPI_connect returned %d", relname, ret);
384 
385  /* Fetch tuple values and nulls */
386  cvals = (Datum *) palloc(natts * sizeof(Datum));
387  cnulls = (char *) palloc(natts * sizeof(char));
388  for (i = 0; i < natts; i++)
389  {
390  cvals[i] = SPI_getbinval((newtuple != NULL) ? newtuple : trigtuple,
391  tupdesc, i + 1, &isnull);
392  cnulls[i] = (isnull) ? 'n' : ' ';
393  }
394 
395  /* change date column(s) */
396  if (newtuple) /* UPDATE */
397  {
398  cvals[attnum[0] - 1] = newoff; /* start_date eq current date */
399  cnulls[attnum[0] - 1] = ' ';
400  cvals[attnum[1] - 1] = TTDUMMY_INFINITY; /* stop_date eq INFINITY */
401  cnulls[attnum[1] - 1] = ' ';
402  }
403  else
404  /* DELETE */
405  {
406  cvals[attnum[1] - 1] = newoff; /* stop_date eq current date */
407  cnulls[attnum[1] - 1] = ' ';
408  }
409 
410  /* if there is no plan ... */
411  if (splan == NULL)
412  {
413  SPIPlanPtr pplan;
414  Oid *ctypes;
415  char *query;
416 
417  /* allocate space in preparation */
418  ctypes = (Oid *) palloc(natts * sizeof(Oid));
419  query = (char *) palloc(100 + 16 * natts);
420 
421  /*
422  * Construct query: INSERT INTO _relation_ VALUES ($1, ...)
423  */
424  sprintf(query, "INSERT INTO %s VALUES (", relname);
425  for (i = 1; i <= natts; i++)
426  {
427  sprintf(query + strlen(query), "$%d%s",
428  i, (i < natts) ? ", " : ")");
429  ctypes[i - 1] = SPI_gettypeid(tupdesc, i);
430  }
431 
432  /* Prepare plan for query */
433  pplan = SPI_prepare(query, natts, ctypes);
434  if (pplan == NULL)
435  elog(ERROR, "ttdummy (%s): SPI_prepare returned %s", relname, SPI_result_code_string(SPI_result));
436 
437  if (SPI_keepplan(pplan))
438  elog(ERROR, "ttdummy (%s): SPI_keepplan failed", relname);
439 
440  splan = pplan;
441  }
442 
443  ret = SPI_execp(splan, cvals, cnulls, 0);
444 
445  if (ret < 0)
446  elog(ERROR, "ttdummy (%s): SPI_execp returned %d", relname, ret);
447 
448  /* Tuple to return to upper Executor ... */
449  if (newtuple) /* UPDATE */
450  rettuple = SPI_modifytuple(rel, trigtuple, 1, &(attnum[1]), &newoff, NULL);
451  else /* DELETE */
452  rettuple = trigtuple;
453 
454  SPI_finish(); /* don't forget say Bye to SPI mgr */
455 
456  pfree(relname);
457 
458  return PointerGetDatum(rettuple);
459 }
#define CStringGetTextDatum(s)
Definition: builtins.h:97
Datum nextval(PG_FUNCTION_ARGS)
Definition: sequence.c:586
#define DirectFunctionCall1(func, arg1)
Definition: fmgr.h:642
int16 attnum
Definition: pg_attribute.h:74
NameData relname
Definition: pg_class.h:38
#define sprintf
Definition: port.h:240
static int64 DatumGetInt64(Datum X)
Definition: postgres.h:385
static SPIPlanPtr splan
Definition: regress.c:269
#define TTDUMMY_INFINITY
Definition: regress.c:267
char * SPI_getrelname(Relation rel)
Definition: spi.c:1323
int SPI_fnumber(TupleDesc tupdesc, const char *fname)
Definition: spi.c:1172
Oid SPI_gettypeid(TupleDesc tupdesc, int fnumber)
Definition: spi.c:1305
int SPI_connect(void)
Definition: spi.c:94
int SPI_result
Definition: spi.c:46
const char * SPI_result_code_string(int code)
Definition: spi.c:1969
int SPI_finish(void)
Definition: spi.c:182
HeapTuple SPI_modifytuple(Relation rel, HeapTuple tuple, int natts, int *attnum, Datum *Values, const char *Nulls)
Definition: spi.c:1103
SPIPlanPtr SPI_prepare(const char *src, int nargs, Oid *argtypes)
Definition: spi.c:857
int SPI_keepplan(SPIPlanPtr plan)
Definition: spi.c:973
int SPI_execp(SPIPlanPtr plan, Datum *Values, const char *Nulls, long tcount)
Definition: spi.c:701
Datum SPI_getbinval(HeapTuple tuple, TupleDesc tupdesc, int fnumber, bool *isnull)
Definition: spi.c:1249
Relation tg_relation
Definition: trigger.h:35
TriggerEvent tg_event
Definition: trigger.h:34
HeapTuple tg_newtuple
Definition: trigger.h:37
Trigger * tg_trigger
Definition: trigger.h:38
#define TRIGGER_FIRED_BEFORE(event)
Definition: trigger.h:128
#define TRIGGER_FIRED_FOR_ROW(event)
Definition: trigger.h:122
#define TRIGGER_FIRED_BY_INSERT(event)
Definition: trigger.h:110
#define TRIGGER_FIRED_BY_UPDATE(event)
Definition: trigger.h:116

References generate_unaccent_rules::args, attnum, CALLED_AS_TRIGGER, CStringGetTextDatum, DatumGetInt64(), DirectFunctionCall1, elog, ereport, errcode(), errmsg(), ERROR, i, if(), Int32GetDatum(), nextval(), palloc(), pfree(), PointerGetDatum(), relname, SPI_connect(), SPI_execp(), SPI_finish(), SPI_fnumber(), SPI_getbinval(), SPI_getrelname(), SPI_gettypeid(), SPI_keepplan(), SPI_modifytuple(), SPI_prepare(), SPI_result, SPI_result_code_string(), splan, sprintf, TriggerData::tg_event, TriggerData::tg_newtuple, TriggerData::tg_relation, TriggerData::tg_trigger, TriggerData::tg_trigtuple, TRIGGER_FIRED_BEFORE, TRIGGER_FIRED_BY_INSERT, TRIGGER_FIRED_BY_UPDATE, TRIGGER_FIRED_FOR_ROW, TTDUMMY_INFINITY, and ttoff.

◆ wait_pid()

Datum wait_pid ( PG_FUNCTION_ARGS  )

Definition at line 668 of file regress.c.

669 {
670  int pid = PG_GETARG_INT32(0);
671 
672  if (!superuser())
673  elog(ERROR, "must be superuser to check PID liveness");
674 
675  while (kill(pid, 0) == 0)
676  {
678  pg_usleep(50000);
679  }
680 
681  if (errno != ESRCH)
682  elog(ERROR, "could not check PID %d liveness: %m", pid);
683 
684  PG_RETURN_VOID();
685 }
#define CHECK_FOR_INTERRUPTS()
Definition: miscadmin.h:122
void pg_usleep(long microsec)
Definition: signal.c:53
#define kill(pid, sig)
Definition: win32_port.h:485

References CHECK_FOR_INTERRUPTS, elog, ERROR, kill, PG_GETARG_INT32, PG_RETURN_VOID, pg_usleep(), and superuser().

◆ widget_in()

Datum widget_in ( PG_FUNCTION_ARGS  )

Definition at line 171 of file regress.c.

172 {
173  char *str = PG_GETARG_CSTRING(0);
174  char *p,
175  *coord[NARGS];
176  int i;
177  WIDGET *result;
178 
179  for (i = 0, p = str; *p && i < NARGS && *p != RDELIM; p++)
180  {
181  if (*p == DELIM || (*p == LDELIM && i == 0))
182  coord[i++] = p + 1;
183  }
184 
185  /*
186  * Note: DON'T convert this error to "soft" style (errsave/ereturn). We
187  * want this data type to stay permanently in the hard-error world so that
188  * it can be used for testing that such cases still work reasonably.
189  */
190  if (i < NARGS)
191  ereport(ERROR,
192  (errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),
193  errmsg("invalid input syntax for type %s: \"%s\"",
194  "widget", str)));
195 
196  result = (WIDGET *) palloc(sizeof(WIDGET));
197  result->center.x = atof(coord[0]);
198  result->center.y = atof(coord[1]);
199  result->radius = atof(coord[2]);
200 
201  PG_RETURN_POINTER(result);
202 }
const char * str
#define DELIM
Definition: regress.c:79
#define RDELIM
Definition: regress.c:78
#define LDELIM
Definition: regress.c:77
#define NARGS
Definition: regress.c:168
double radius
Definition: regress.c:162

References WIDGET::center, DELIM, ereport, errcode(), errmsg(), ERROR, i, LDELIM, NARGS, palloc(), PG_GETARG_CSTRING, PG_RETURN_POINTER, WIDGET::radius, RDELIM, str, Point::x, and Point::y.

◆ widget_out()

Datum widget_out ( PG_FUNCTION_ARGS  )

Definition at line 205 of file regress.c.

206 {
207  WIDGET *widget = (WIDGET *) PG_GETARG_POINTER(0);
208  char *str = psprintf("(%g,%g,%g)",
209  widget->center.x, widget->center.y, widget->radius);
210 
212 }
char * psprintf(const char *fmt,...)
Definition: psprintf.c:46

References WIDGET::center, PG_GETARG_POINTER, PG_RETURN_CSTRING, psprintf(), WIDGET::radius, str, Point::x, and Point::y.

Variable Documentation

◆ PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 83 of file regress.c.

◆ splan

◆ ttoff

bool ttoff = false
static

Definition at line 270 of file regress.c.

Referenced by set_ttdummy(), and ttdummy().