/*--------------------------------------------------------------------------
 *
 * test_lfind.c
 *  Test correctness of optimized linear search functions.
 *
 * Copyright (c) 2022-2025, PostgreSQL Global Development Group
 *
 * IDENTIFICATION
 *  src/test/modules/test_lfind/test_lfind.c
 *
 * -------------------------------------------------------------------------
 */

#include "postgres.h"

#include "fmgr.h"
#include "port/pg_lfind.h"

/*
 * Convenience macros for testing both vector and scalar operations. The 2x
 * factor is to make sure iteration works
 */
#define LEN_NO_TAIL(vectortype) (2 * sizeof(vectortype))
#define LEN_WITH_TAIL(vectortype) (LEN_NO_TAIL(vectortype) + 3)

PG_MODULE_MAGIC;

/* workhorse for test_lfind8 */
static void
test_lfind8_internal(uint8 key)
{
 uint8  charbuf[LEN_WITH_TAIL(Vector8)];
 const int len_no_tail = LEN_NO_TAIL(Vector8);
 const int len_with_tail = LEN_WITH_TAIL(Vector8);

 memset(charbuf, 0xFF, len_with_tail);
 /* search tail to test one-byte-at-a-time path */
 charbuf[len_with_tail - 1] = key;
 if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
 if (key < 0xFF && !pg_lfind8(key, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
 if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);

 memset(charbuf, 0xFF, len_with_tail);
 /* search with vector operations */
 charbuf[len_no_tail - 1] = key;
 if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
 if (key < 0xFF && !pg_lfind8(key, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
 if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);
}

PG_FUNCTION_INFO_V1(test_lfind8);
Datum
test_lfind8(PG_FUNCTION_ARGS)
{
 test_lfind8_internal(0);
 test_lfind8_internal(1);
 test_lfind8_internal(0x7F);
 test_lfind8_internal(0x80);
 test_lfind8_internal(0x81);
 test_lfind8_internal(0xFD);
 test_lfind8_internal(0xFE);
 test_lfind8_internal(0xFF);

 PG_RETURN_VOID();
}

/* workhorse for test_lfind8_le */
static void
test_lfind8_le_internal(uint8 key)
{
 uint8  charbuf[LEN_WITH_TAIL(Vector8)];
 const int len_no_tail = LEN_NO_TAIL(Vector8);
 const int len_with_tail = LEN_WITH_TAIL(Vector8);

 memset(charbuf, 0xFF, len_with_tail);
 /* search tail to test one-byte-at-a-time path */
 charbuf[len_with_tail - 1] = key;
 if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
 if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
 if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_with_tail))
  elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);

 memset(charbuf, 0xFF, len_with_tail);
 /* search with vector operations */
 charbuf[len_no_tail - 1] = key;
 if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
 if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
 if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_no_tail))
  elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);
}

PG_FUNCTION_INFO_V1(test_lfind8_le);
Datum
test_lfind8_le(PG_FUNCTION_ARGS)
{
 test_lfind8_le_internal(0);
 test_lfind8_le_internal(1);
 test_lfind8_le_internal(0x7F);
 test_lfind8_le_internal(0x80);
 test_lfind8_le_internal(0x81);
 test_lfind8_le_internal(0xFD);
 test_lfind8_le_internal(0xFE);
 test_lfind8_le_internal(0xFF);

 PG_RETURN_VOID();
}

PG_FUNCTION_INFO_V1(test_lfind32);
Datum
test_lfind32(PG_FUNCTION_ARGS)
{
#define TEST_ARRAY_SIZE 135
 uint32  test_array[TEST_ARRAY_SIZE] = {0};

 test_array[8] = 1;
 test_array[64] = 2;
 test_array[TEST_ARRAY_SIZE - 1] = 3;

 if (pg_lfind32(1, test_array, 4))
  elog(ERROR, "pg_lfind32() found nonexistent element");
 if (!pg_lfind32(1, test_array, TEST_ARRAY_SIZE))
  elog(ERROR, "pg_lfind32() did not find existing element");

 if (pg_lfind32(2, test_array, 32))
  elog(ERROR, "pg_lfind32() found nonexistent element");
 if (!pg_lfind32(2, test_array, TEST_ARRAY_SIZE))
  elog(ERROR, "pg_lfind32() did not find existing element");

 if (pg_lfind32(3, test_array, 96))
  elog(ERROR, "pg_lfind32() found nonexistent element");
 if (!pg_lfind32(3, test_array, TEST_ARRAY_SIZE))
  elog(ERROR, "pg_lfind32() did not find existing element");

 if (pg_lfind32(4, test_array, TEST_ARRAY_SIZE))
  elog(ERROR, "pg_lfind32() found nonexistent element");

 PG_RETURN_VOID();
}