# Copyright (c) 2021-2025, PostgreSQL Global Development Group

# Check that pg_check_visible() and pg_check_frozen() report correct TIDs for
# corruption.
use strict;
use warnings FATAL => 'all';
use File::Copy;
use PostgreSQL::Test::Cluster;
use PostgreSQL::Test::Utils;
use Test::More;

my $node = PostgreSQL::Test::Cluster->new('main');
$node->init;
# Anything holding a snapshot, including auto-analyze of pg_proc, could stop
# VACUUM from updating the visibility map.
$node->append_conf('postgresql.conf', 'autovacuum=off');
$node->start;

my $blck_size = $node->safe_psql("postgres", "SHOW block_size;");

# Create a sample table with at least 10 pages and then run VACUUM. 10 is
# selected manually as it is big enough to select 5 random tuples from the
# relation.
$node->safe_psql(
 'postgres', qq(
  CREATE EXTENSION pg_visibility;
  CREATE TABLE corruption_test
   WITH (autovacuum_enabled = false) AS
   SELECT
    i,
    repeat('a', 10) AS data
   FROM
    generate_series(1, $blck_size) i;
  VACUUM (FREEZE, DISABLE_PAGE_SKIPPING) corruption_test;
));

# VACUUM is run, it is safe to get the number of pages.
my $npages = $node->safe_psql(
 "postgres",
 "SELECT relpages FROM pg_class
  WHERE relname = 'corruption_test';"
);
ok($npages >= 10, 'table has at least 10 pages');

my $file = $node->safe_psql("postgres",
 "SELECT pg_relation_filepath('corruption_test');");

# Delete the first block to make sure that it will be skipped as it is
# not visible nor frozen.
$node->safe_psql(
 "postgres",
 "DELETE FROM corruption_test
  WHERE (ctid::text::point)[0] = 0;"
);

# Copy visibility map.
$node->stop;
my $vm_file = $node->data_dir . '/' . $file . '_vm';
copy("$vm_file", "${vm_file}_temp");
$node->start;

# Select 5 random tuples that are starting from the second block of the
# relation. The first block is skipped because it is deleted above.
my $tuples = $node->safe_psql(
 "postgres",
 "SELECT ctid FROM (
  SELECT ctid FROM corruption_test
   WHERE (ctid::text::point)[0] != 0
   ORDER BY random() LIMIT 5)
  ORDER BY ctid ASC;"
);

# Do the changes below to use tuples in the query.
# "\n" -> ","
# "(" -> "'("
# ")" -> ")'"
(my $tuples_query = $tuples) =~ s/\n/,/g;
$tuples_query =~ s/\(/\'\(/g;
$tuples_query =~ s/\)/\)\'/g;

$node->safe_psql(
 "postgres",
 "DELETE FROM corruption_test
  WHERE ctid in ($tuples_query);"
);

# Overwrite visibility map with the old one.
$node->stop;
move("${vm_file}_temp", "$vm_file");
$node->start;

my $result = $node->safe_psql(
 "postgres",
 "SELECT DISTINCT t_ctid
  FROM pg_check_visible('corruption_test')
  ORDER BY t_ctid ASC;"
);
is($result, $tuples, 'pg_check_visible must report tuples as corrupted');

$result = $node->safe_psql(
 "postgres",
 "SELECT DISTINCT t_ctid
  FROM pg_check_frozen('corruption_test')
  ORDER BY t_ctid ASC;"
);
is($result, $tuples, 'pg_check_frozen must report tuples as corrupted');

$node->stop;
done_testing();