Branch data     Line data    Source code

       1                 :             : /*--------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * test_checksums.c
       4                 :             :  *      Test data checksums
       5                 :             :  *
       6                 :             :  * Copyright (c) 2026, PostgreSQL Global Development Group
       7                 :             :  *
       8                 :             :  * IDENTIFICATION
       9                 :             :  *      src/test/modules/test_checksums/test_checksums.c
      10                 :             :  *
      11                 :             :  * -------------------------------------------------------------------------
      12                 :             :  */
      13                 :             : #include "postgres.h"
      14                 :             : 
      15                 :             : #include "funcapi.h"
      16                 :             : #include "miscadmin.h"
      17                 :             : #include "postmaster/datachecksum_state.h"
      18                 :             : #include "storage/latch.h"
      19                 :             : #include "utils/injection_point.h"
      20                 :             : #include "utils/wait_event.h"
      21                 :             : 
      22                 :           2 : PG_MODULE_MAGIC;
      23                 :             : 
      24                 :             : extern PGDLLEXPORT void dc_delay_barrier(const char *name, const void *private_data, void *arg);
      25                 :             : 
      26                 :             : /*
      27                 :             :  * Test for delaying emission of procsignalbarriers.
      28                 :             :  */
      29                 :             : void
      30                 :           0 : dc_delay_barrier(const char *name, const void *private_data, void *arg)
      31                 :             : {
      32                 :             :     (void) name;
      33                 :             :     (void) private_data;
      34                 :             : 
      35                 :           0 :     (void) WaitLatch(MyLatch,
      36                 :             :                      WL_LATCH_SET | WL_TIMEOUT | WL_EXIT_ON_PM_DEATH,
      37                 :             :                      (3 * 1000),
      38                 :             :                      WAIT_EVENT_PG_SLEEP);
      39                 :           0 : }
      40                 :             : 
      41                 :           2 : PG_FUNCTION_INFO_V1(dcw_inject_delay_barrier);
      42                 :             : Datum
      43                 :           0 : dcw_inject_delay_barrier(PG_FUNCTION_ARGS)
      44                 :             : {
      45                 :             : #ifdef USE_INJECTION_POINTS
      46                 :           0 :     bool        attach = PG_GETARG_BOOL(0);
      47                 :             : 
      48         [ #  # ]:           0 :     if (attach)
      49                 :           0 :         InjectionPointAttach("datachecksums-enable-checksums-delay",
      50                 :             :                              "test_checksums",
      51                 :             :                              "dc_delay_barrier",
      52                 :             :                              NULL,
      53                 :             :                              0);
      54                 :             :     else
      55                 :           0 :         InjectionPointDetach("datachecksums-enable-checksums-delay");
      56                 :             : #else
      57                 :             :     elog(ERROR,
      58                 :             :          "test is not working as intended when injection points are disabled");
      59                 :             : #endif
      60                 :           0 :     PG_RETURN_VOID();
      61                 :             : }
      62                 :             : 
      63                 :           2 : PG_FUNCTION_INFO_V1(dcw_inject_launcher_delay);
      64                 :             : Datum
      65                 :           0 : dcw_inject_launcher_delay(PG_FUNCTION_ARGS)
      66                 :             : {
      67                 :             : #ifdef USE_INJECTION_POINTS
      68                 :           0 :     bool        attach = PG_GETARG_BOOL(0);
      69                 :             : 
      70         [ #  # ]:           0 :     if (attach)
      71                 :           0 :         InjectionPointAttach("datachecksumsworker-launcher-delay",
      72                 :             :                              "test_checksums",
      73                 :             :                              "dc_delay_barrier",
      74                 :             :                              NULL,
      75                 :             :                              0);
      76                 :             :     else
      77                 :           0 :         InjectionPointDetach("datachecksumsworker-launcher-delay");
      78                 :             : #else
      79                 :             :     elog(ERROR,
      80                 :             :          "test is not working as intended when injection points are disabled");
      81                 :             : #endif
      82                 :           0 :     PG_RETURN_VOID();
      83                 :             : }
      84                 :             : 
      85                 :           2 : PG_FUNCTION_INFO_V1(dcw_inject_startup_delay);
      86                 :             : Datum
      87                 :           0 : dcw_inject_startup_delay(PG_FUNCTION_ARGS)
      88                 :             : {
      89                 :             : #ifdef USE_INJECTION_POINTS
      90                 :           0 :     bool        attach = PG_GETARG_BOOL(0);
      91                 :             : 
      92         [ #  # ]:           0 :     if (attach)
      93                 :           0 :         InjectionPointAttach("datachecksumsworker-startup-delay",
      94                 :             :                              "test_checksums",
      95                 :             :                              "dc_delay_barrier",
      96                 :             :                              NULL,
      97                 :             :                              0);
      98                 :             :     else
      99                 :           0 :         InjectionPointDetach("datachecksumsworker-startup-delay");
     100                 :             : #else
     101                 :             :     elog(ERROR,
     102                 :             :          "test is not working as intended when injection points are disabled");
     103                 :             : #endif
     104                 :           0 :     PG_RETURN_VOID();
     105                 :             : }