Line data Source code
1 : /*--------------------------------------------------------------------------
2 : *
3 : * test_lfind.c
4 : * Test correctness of optimized linear search functions.
5 : *
6 : * Copyright (c) 2022-2026, PostgreSQL Global Development Group
7 : *
8 : * IDENTIFICATION
9 : * src/test/modules/test_lfind/test_lfind.c
10 : *
11 : * -------------------------------------------------------------------------
12 : */
13 :
14 : #include "postgres.h"
15 :
16 : #include "fmgr.h"
17 : #include "port/pg_lfind.h"
18 :
19 : /*
20 : * Convenience macros for testing both vector and scalar operations. The 2x
21 : * factor is to make sure iteration works
22 : */
23 : #define LEN_NO_TAIL(vectortype) (2 * sizeof(vectortype))
24 : #define LEN_WITH_TAIL(vectortype) (LEN_NO_TAIL(vectortype) + 3)
25 :
26 1 : PG_MODULE_MAGIC;
27 :
28 : /* workhorse for test_lfind8 */
29 : static void
30 8 : test_lfind8_internal(uint8 key)
31 : {
32 : uint8 charbuf[LEN_WITH_TAIL(Vector8)];
33 8 : const int len_no_tail = LEN_NO_TAIL(Vector8);
34 8 : const int len_with_tail = LEN_WITH_TAIL(Vector8);
35 :
36 8 : memset(charbuf, 0xFF, len_with_tail);
37 : /* search tail to test one-byte-at-a-time path */
38 8 : charbuf[len_with_tail - 1] = key;
39 8 : if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_with_tail))
40 0 : elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
41 8 : if (key < 0xFF && !pg_lfind8(key, charbuf, len_with_tail))
42 0 : elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
43 8 : if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_with_tail))
44 0 : elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);
45 :
46 8 : memset(charbuf, 0xFF, len_with_tail);
47 : /* search with vector operations */
48 8 : charbuf[len_no_tail - 1] = key;
49 8 : if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_no_tail))
50 0 : elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
51 8 : if (key < 0xFF && !pg_lfind8(key, charbuf, len_no_tail))
52 0 : elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
53 8 : if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_no_tail))
54 0 : elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);
55 8 : }
56 :
57 2 : PG_FUNCTION_INFO_V1(test_lfind8);
58 : Datum
59 1 : test_lfind8(PG_FUNCTION_ARGS)
60 : {
61 1 : test_lfind8_internal(0);
62 1 : test_lfind8_internal(1);
63 1 : test_lfind8_internal(0x7F);
64 1 : test_lfind8_internal(0x80);
65 1 : test_lfind8_internal(0x81);
66 1 : test_lfind8_internal(0xFD);
67 1 : test_lfind8_internal(0xFE);
68 1 : test_lfind8_internal(0xFF);
69 :
70 1 : PG_RETURN_VOID();
71 : }
72 :
73 : /* workhorse for test_lfind8_le */
74 : static void
75 8 : test_lfind8_le_internal(uint8 key)
76 : {
77 : uint8 charbuf[LEN_WITH_TAIL(Vector8)];
78 8 : const int len_no_tail = LEN_NO_TAIL(Vector8);
79 8 : const int len_with_tail = LEN_WITH_TAIL(Vector8);
80 :
81 8 : memset(charbuf, 0xFF, len_with_tail);
82 : /* search tail to test one-byte-at-a-time path */
83 8 : charbuf[len_with_tail - 1] = key;
84 8 : if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_with_tail))
85 0 : elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
86 8 : if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_with_tail))
87 0 : elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
88 8 : if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_with_tail))
89 0 : elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);
90 :
91 8 : memset(charbuf, 0xFF, len_with_tail);
92 : /* search with vector operations */
93 8 : charbuf[len_no_tail - 1] = key;
94 8 : if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_no_tail))
95 0 : elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
96 8 : if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_no_tail))
97 0 : elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
98 8 : if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_no_tail))
99 0 : elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);
100 8 : }
101 :
102 2 : PG_FUNCTION_INFO_V1(test_lfind8_le);
103 : Datum
104 1 : test_lfind8_le(PG_FUNCTION_ARGS)
105 : {
106 1 : test_lfind8_le_internal(0);
107 1 : test_lfind8_le_internal(1);
108 1 : test_lfind8_le_internal(0x7F);
109 1 : test_lfind8_le_internal(0x80);
110 1 : test_lfind8_le_internal(0x81);
111 1 : test_lfind8_le_internal(0xFD);
112 1 : test_lfind8_le_internal(0xFE);
113 1 : test_lfind8_le_internal(0xFF);
114 :
115 1 : PG_RETURN_VOID();
116 : }
117 :
118 2 : PG_FUNCTION_INFO_V1(test_lfind32);
119 : Datum
120 1 : test_lfind32(PG_FUNCTION_ARGS)
121 : {
122 : #define TEST_ARRAY_SIZE 135
123 1 : uint32 test_array[TEST_ARRAY_SIZE] = {0};
124 :
125 1 : test_array[8] = 1;
126 1 : test_array[64] = 2;
127 1 : test_array[TEST_ARRAY_SIZE - 1] = 3;
128 :
129 1 : if (pg_lfind32(1, test_array, 4))
130 0 : elog(ERROR, "pg_lfind32() found nonexistent element");
131 1 : if (!pg_lfind32(1, test_array, TEST_ARRAY_SIZE))
132 0 : elog(ERROR, "pg_lfind32() did not find existing element");
133 :
134 1 : if (pg_lfind32(2, test_array, 32))
135 0 : elog(ERROR, "pg_lfind32() found nonexistent element");
136 1 : if (!pg_lfind32(2, test_array, TEST_ARRAY_SIZE))
137 0 : elog(ERROR, "pg_lfind32() did not find existing element");
138 :
139 1 : if (pg_lfind32(3, test_array, 96))
140 0 : elog(ERROR, "pg_lfind32() found nonexistent element");
141 1 : if (!pg_lfind32(3, test_array, TEST_ARRAY_SIZE))
142 0 : elog(ERROR, "pg_lfind32() did not find existing element");
143 :
144 1 : if (pg_lfind32(4, test_array, TEST_ARRAY_SIZE))
145 0 : elog(ERROR, "pg_lfind32() found nonexistent element");
146 :
147 1 : PG_RETURN_VOID();
148 : }
|
