mirror of
https://github.com/git/git
synced 2024-08-28 12:09:19 +00:00
![Han-Wen Nienhuys](/assets/img/avatar_default.png)
This function is part of the reftable API, so it should use the reftable_ prefix Signed-off-by: Han-Wen Nienhuys <hanwen@google.com> Signed-off-by: Junio C Hamano <gitster@pobox.com>
852 lines
20 KiB
C
852 lines
20 KiB
C
/*
|
|
Copyright 2020 Google LLC
|
|
|
|
Use of this source code is governed by a BSD-style
|
|
license that can be found in the LICENSE file or at
|
|
https://developers.google.com/open-source/licenses/bsd
|
|
*/
|
|
|
|
#include "system.h"
|
|
|
|
#include "basics.h"
|
|
#include "block.h"
|
|
#include "blocksource.h"
|
|
#include "constants.h"
|
|
#include "reader.h"
|
|
#include "record.h"
|
|
#include "test_framework.h"
|
|
#include "reftable-tests.h"
|
|
#include "reftable-writer.h"
|
|
|
|
static const int update_index = 5;
|
|
|
|
static void test_buffer(void)
|
|
{
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_block_source source = { NULL };
|
|
struct reftable_block out = { NULL };
|
|
int n;
|
|
uint8_t in[] = "hello";
|
|
strbuf_add(&buf, in, sizeof(in));
|
|
block_source_from_strbuf(&source, &buf);
|
|
EXPECT(block_source_size(&source) == 6);
|
|
n = block_source_read_block(&source, &out, 0, sizeof(in));
|
|
EXPECT(n == sizeof(in));
|
|
EXPECT(!memcmp(in, out.data, n));
|
|
reftable_block_done(&out);
|
|
|
|
n = block_source_read_block(&source, &out, 1, 2);
|
|
EXPECT(n == 2);
|
|
EXPECT(!memcmp(out.data, "el", 2));
|
|
|
|
reftable_block_done(&out);
|
|
block_source_close(&source);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void write_table(char ***names, struct strbuf *buf, int N,
|
|
int block_size, uint32_t hash_id)
|
|
{
|
|
struct reftable_write_options opts = {
|
|
.block_size = block_size,
|
|
.hash_id = hash_id,
|
|
};
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, buf, &opts);
|
|
struct reftable_ref_record ref = { NULL };
|
|
int i = 0, n;
|
|
struct reftable_log_record log = { NULL };
|
|
const struct reftable_stats *stats = NULL;
|
|
*names = reftable_calloc(sizeof(char *) * (N + 1));
|
|
reftable_writer_set_limits(w, update_index, update_index);
|
|
for (i = 0; i < N; i++) {
|
|
uint8_t hash[GIT_SHA256_RAWSZ] = { 0 };
|
|
char name[100];
|
|
int n;
|
|
|
|
set_test_hash(hash, i);
|
|
|
|
snprintf(name, sizeof(name), "refs/heads/branch%02d", i);
|
|
|
|
ref.refname = name;
|
|
ref.update_index = update_index;
|
|
ref.value_type = REFTABLE_REF_VAL1;
|
|
ref.value.val1 = hash;
|
|
(*names)[i] = xstrdup(name);
|
|
|
|
n = reftable_writer_add_ref(w, &ref);
|
|
EXPECT(n == 0);
|
|
}
|
|
|
|
for (i = 0; i < N; i++) {
|
|
uint8_t hash[GIT_SHA256_RAWSZ] = { 0 };
|
|
char name[100];
|
|
int n;
|
|
|
|
set_test_hash(hash, i);
|
|
|
|
snprintf(name, sizeof(name), "refs/heads/branch%02d", i);
|
|
|
|
log.refname = name;
|
|
log.update_index = update_index;
|
|
log.value_type = REFTABLE_LOG_UPDATE;
|
|
log.value.update.new_hash = hash;
|
|
log.value.update.message = "message";
|
|
|
|
n = reftable_writer_add_log(w, &log);
|
|
EXPECT(n == 0);
|
|
}
|
|
|
|
n = reftable_writer_close(w);
|
|
EXPECT(n == 0);
|
|
|
|
stats = reftable_writer_stats(w);
|
|
for (i = 0; i < stats->ref_stats.blocks; i++) {
|
|
int off = i * opts.block_size;
|
|
if (off == 0) {
|
|
off = header_size(
|
|
(hash_id == GIT_SHA256_FORMAT_ID) ? 2 : 1);
|
|
}
|
|
EXPECT(buf->buf[off] == 'r');
|
|
}
|
|
|
|
EXPECT(stats->log_stats.blocks > 0);
|
|
reftable_writer_free(w);
|
|
}
|
|
|
|
static void test_log_buffer_size(void)
|
|
{
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_write_options opts = {
|
|
.block_size = 4096,
|
|
};
|
|
int err;
|
|
int i;
|
|
struct reftable_log_record
|
|
log = { .refname = "refs/heads/master",
|
|
.update_index = 0xa,
|
|
.value_type = REFTABLE_LOG_UPDATE,
|
|
.value = { .update = {
|
|
.name = "Han-Wen Nienhuys",
|
|
.email = "hanwen@google.com",
|
|
.tz_offset = 100,
|
|
.time = 0x5e430672,
|
|
.message = "commit: 9\n",
|
|
} } };
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
|
|
/* This tests buffer extension for log compression. Must use a random
|
|
hash, to ensure that the compressed part is larger than the original.
|
|
*/
|
|
uint8_t hash1[GIT_SHA1_RAWSZ], hash2[GIT_SHA1_RAWSZ];
|
|
for (i = 0; i < GIT_SHA1_RAWSZ; i++) {
|
|
hash1[i] = (uint8_t)(rand() % 256);
|
|
hash2[i] = (uint8_t)(rand() % 256);
|
|
}
|
|
log.value.update.old_hash = hash1;
|
|
log.value.update.new_hash = hash2;
|
|
reftable_writer_set_limits(w, update_index, update_index);
|
|
err = reftable_writer_add_log(w, &log);
|
|
EXPECT_ERR(err);
|
|
err = reftable_writer_close(w);
|
|
EXPECT_ERR(err);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_log_overflow(void)
|
|
{
|
|
struct strbuf buf = STRBUF_INIT;
|
|
char msg[256] = { 0 };
|
|
struct reftable_write_options opts = {
|
|
.block_size = ARRAY_SIZE(msg),
|
|
};
|
|
int err;
|
|
struct reftable_log_record
|
|
log = { .refname = "refs/heads/master",
|
|
.update_index = 0xa,
|
|
.value_type = REFTABLE_LOG_UPDATE,
|
|
.value = { .update = {
|
|
.name = "Han-Wen Nienhuys",
|
|
.email = "hanwen@google.com",
|
|
.tz_offset = 100,
|
|
.time = 0x5e430672,
|
|
.message = msg,
|
|
} } };
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
|
|
uint8_t hash1[GIT_SHA1_RAWSZ] = {1}, hash2[GIT_SHA1_RAWSZ] = { 2 };
|
|
|
|
memset(msg, 'x', sizeof(msg) - 1);
|
|
log.value.update.old_hash = hash1;
|
|
log.value.update.new_hash = hash2;
|
|
reftable_writer_set_limits(w, update_index, update_index);
|
|
err = reftable_writer_add_log(w, &log);
|
|
EXPECT(err == REFTABLE_ENTRY_TOO_BIG_ERROR);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_log_write_read(void)
|
|
{
|
|
int N = 2;
|
|
char **names = reftable_calloc(sizeof(char *) * (N + 1));
|
|
int err;
|
|
struct reftable_write_options opts = {
|
|
.block_size = 256,
|
|
};
|
|
struct reftable_ref_record ref = { NULL };
|
|
int i = 0;
|
|
struct reftable_log_record log = { NULL };
|
|
int n;
|
|
struct reftable_iterator it = { NULL };
|
|
struct reftable_reader rd = { NULL };
|
|
struct reftable_block_source source = { NULL };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
const struct reftable_stats *stats = NULL;
|
|
reftable_writer_set_limits(w, 0, N);
|
|
for (i = 0; i < N; i++) {
|
|
char name[256];
|
|
struct reftable_ref_record ref = { NULL };
|
|
snprintf(name, sizeof(name), "b%02d%0*d", i, 130, 7);
|
|
names[i] = xstrdup(name);
|
|
ref.refname = name;
|
|
ref.update_index = i;
|
|
|
|
err = reftable_writer_add_ref(w, &ref);
|
|
EXPECT_ERR(err);
|
|
}
|
|
for (i = 0; i < N; i++) {
|
|
uint8_t hash1[GIT_SHA1_RAWSZ], hash2[GIT_SHA1_RAWSZ];
|
|
struct reftable_log_record log = { NULL };
|
|
set_test_hash(hash1, i);
|
|
set_test_hash(hash2, i + 1);
|
|
|
|
log.refname = names[i];
|
|
log.update_index = i;
|
|
log.value_type = REFTABLE_LOG_UPDATE;
|
|
log.value.update.old_hash = hash1;
|
|
log.value.update.new_hash = hash2;
|
|
|
|
err = reftable_writer_add_log(w, &log);
|
|
EXPECT_ERR(err);
|
|
}
|
|
|
|
n = reftable_writer_close(w);
|
|
EXPECT(n == 0);
|
|
|
|
stats = reftable_writer_stats(w);
|
|
EXPECT(stats->log_stats.blocks > 0);
|
|
reftable_writer_free(w);
|
|
w = NULL;
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.log");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_reader_seek_ref(&rd, &it, names[N - 1]);
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT_ERR(err);
|
|
|
|
/* end of iteration. */
|
|
err = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT(0 < err);
|
|
|
|
reftable_iterator_destroy(&it);
|
|
reftable_ref_record_release(&ref);
|
|
|
|
err = reftable_reader_seek_log(&rd, &it, "");
|
|
EXPECT_ERR(err);
|
|
|
|
i = 0;
|
|
while (1) {
|
|
int err = reftable_iterator_next_log(&it, &log);
|
|
if (err > 0) {
|
|
break;
|
|
}
|
|
|
|
EXPECT_ERR(err);
|
|
EXPECT_STREQ(names[i], log.refname);
|
|
EXPECT(i == log.update_index);
|
|
i++;
|
|
reftable_log_record_release(&log);
|
|
}
|
|
|
|
EXPECT(i == N);
|
|
reftable_iterator_destroy(&it);
|
|
|
|
/* cleanup. */
|
|
strbuf_release(&buf);
|
|
free_names(names);
|
|
reader_close(&rd);
|
|
}
|
|
|
|
static void test_log_zlib_corruption(void)
|
|
{
|
|
struct reftable_write_options opts = {
|
|
.block_size = 256,
|
|
};
|
|
struct reftable_iterator it = { 0 };
|
|
struct reftable_reader rd = { 0 };
|
|
struct reftable_block_source source = { 0 };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
const struct reftable_stats *stats = NULL;
|
|
uint8_t hash1[GIT_SHA1_RAWSZ] = { 1 };
|
|
uint8_t hash2[GIT_SHA1_RAWSZ] = { 2 };
|
|
char message[100] = { 0 };
|
|
int err, i, n;
|
|
|
|
struct reftable_log_record log = {
|
|
.refname = "refname",
|
|
.value_type = REFTABLE_LOG_UPDATE,
|
|
.value = {
|
|
.update = {
|
|
.new_hash = hash1,
|
|
.old_hash = hash2,
|
|
.name = "My Name",
|
|
.email = "myname@invalid",
|
|
.message = message,
|
|
},
|
|
},
|
|
};
|
|
|
|
for (i = 0; i < sizeof(message) - 1; i++)
|
|
message[i] = (uint8_t)(rand() % 64 + ' ');
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
|
|
err = reftable_writer_add_log(w, &log);
|
|
EXPECT_ERR(err);
|
|
|
|
n = reftable_writer_close(w);
|
|
EXPECT(n == 0);
|
|
|
|
stats = reftable_writer_stats(w);
|
|
EXPECT(stats->log_stats.blocks > 0);
|
|
reftable_writer_free(w);
|
|
w = NULL;
|
|
|
|
/* corrupt the data. */
|
|
buf.buf[50] ^= 0x99;
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.log");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_reader_seek_log(&rd, &it, "refname");
|
|
EXPECT(err == REFTABLE_ZLIB_ERROR);
|
|
|
|
reftable_iterator_destroy(&it);
|
|
|
|
/* cleanup. */
|
|
strbuf_release(&buf);
|
|
reader_close(&rd);
|
|
}
|
|
|
|
static void test_table_read_write_sequential(void)
|
|
{
|
|
char **names;
|
|
struct strbuf buf = STRBUF_INIT;
|
|
int N = 50;
|
|
struct reftable_iterator it = { NULL };
|
|
struct reftable_block_source source = { NULL };
|
|
struct reftable_reader rd = { NULL };
|
|
int err = 0;
|
|
int j = 0;
|
|
|
|
write_table(&names, &buf, N, 256, GIT_SHA1_FORMAT_ID);
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.ref");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_reader_seek_ref(&rd, &it, "");
|
|
EXPECT_ERR(err);
|
|
|
|
while (1) {
|
|
struct reftable_ref_record ref = { NULL };
|
|
int r = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT(r >= 0);
|
|
if (r > 0) {
|
|
break;
|
|
}
|
|
EXPECT(0 == strcmp(names[j], ref.refname));
|
|
EXPECT(update_index == ref.update_index);
|
|
|
|
j++;
|
|
reftable_ref_record_release(&ref);
|
|
}
|
|
EXPECT(j == N);
|
|
reftable_iterator_destroy(&it);
|
|
strbuf_release(&buf);
|
|
free_names(names);
|
|
|
|
reader_close(&rd);
|
|
}
|
|
|
|
static void test_table_write_small_table(void)
|
|
{
|
|
char **names;
|
|
struct strbuf buf = STRBUF_INIT;
|
|
int N = 1;
|
|
write_table(&names, &buf, N, 4096, GIT_SHA1_FORMAT_ID);
|
|
EXPECT(buf.len < 200);
|
|
strbuf_release(&buf);
|
|
free_names(names);
|
|
}
|
|
|
|
static void test_table_read_api(void)
|
|
{
|
|
char **names;
|
|
struct strbuf buf = STRBUF_INIT;
|
|
int N = 50;
|
|
struct reftable_reader rd = { NULL };
|
|
struct reftable_block_source source = { NULL };
|
|
int err;
|
|
int i;
|
|
struct reftable_log_record log = { NULL };
|
|
struct reftable_iterator it = { NULL };
|
|
|
|
write_table(&names, &buf, N, 256, GIT_SHA1_FORMAT_ID);
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.ref");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_reader_seek_ref(&rd, &it, names[0]);
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_iterator_next_log(&it, &log);
|
|
EXPECT(err == REFTABLE_API_ERROR);
|
|
|
|
strbuf_release(&buf);
|
|
for (i = 0; i < N; i++) {
|
|
reftable_free(names[i]);
|
|
}
|
|
reftable_iterator_destroy(&it);
|
|
reftable_free(names);
|
|
reader_close(&rd);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_table_read_write_seek(int index, int hash_id)
|
|
{
|
|
char **names;
|
|
struct strbuf buf = STRBUF_INIT;
|
|
int N = 50;
|
|
struct reftable_reader rd = { NULL };
|
|
struct reftable_block_source source = { NULL };
|
|
int err;
|
|
int i = 0;
|
|
|
|
struct reftable_iterator it = { NULL };
|
|
struct strbuf pastLast = STRBUF_INIT;
|
|
struct reftable_ref_record ref = { NULL };
|
|
|
|
write_table(&names, &buf, N, 256, hash_id);
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.ref");
|
|
EXPECT_ERR(err);
|
|
EXPECT(hash_id == reftable_reader_hash_id(&rd));
|
|
|
|
if (!index) {
|
|
rd.ref_offsets.index_offset = 0;
|
|
} else {
|
|
EXPECT(rd.ref_offsets.index_offset > 0);
|
|
}
|
|
|
|
for (i = 1; i < N; i++) {
|
|
int err = reftable_reader_seek_ref(&rd, &it, names[i]);
|
|
EXPECT_ERR(err);
|
|
err = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT_ERR(err);
|
|
EXPECT(0 == strcmp(names[i], ref.refname));
|
|
EXPECT(REFTABLE_REF_VAL1 == ref.value_type);
|
|
EXPECT(i == ref.value.val1[0]);
|
|
|
|
reftable_ref_record_release(&ref);
|
|
reftable_iterator_destroy(&it);
|
|
}
|
|
|
|
strbuf_addstr(&pastLast, names[N - 1]);
|
|
strbuf_addstr(&pastLast, "/");
|
|
|
|
err = reftable_reader_seek_ref(&rd, &it, pastLast.buf);
|
|
if (err == 0) {
|
|
struct reftable_ref_record ref = { NULL };
|
|
int err = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT(err > 0);
|
|
} else {
|
|
EXPECT(err > 0);
|
|
}
|
|
|
|
strbuf_release(&pastLast);
|
|
reftable_iterator_destroy(&it);
|
|
|
|
strbuf_release(&buf);
|
|
for (i = 0; i < N; i++) {
|
|
reftable_free(names[i]);
|
|
}
|
|
reftable_free(names);
|
|
reader_close(&rd);
|
|
}
|
|
|
|
static void test_table_read_write_seek_linear(void)
|
|
{
|
|
test_table_read_write_seek(0, GIT_SHA1_FORMAT_ID);
|
|
}
|
|
|
|
static void test_table_read_write_seek_linear_sha256(void)
|
|
{
|
|
test_table_read_write_seek(0, GIT_SHA256_FORMAT_ID);
|
|
}
|
|
|
|
static void test_table_read_write_seek_index(void)
|
|
{
|
|
test_table_read_write_seek(1, GIT_SHA1_FORMAT_ID);
|
|
}
|
|
|
|
static void test_table_refs_for(int indexed)
|
|
{
|
|
int N = 50;
|
|
char **want_names = reftable_calloc(sizeof(char *) * (N + 1));
|
|
int want_names_len = 0;
|
|
uint8_t want_hash[GIT_SHA1_RAWSZ];
|
|
|
|
struct reftable_write_options opts = {
|
|
.block_size = 256,
|
|
};
|
|
struct reftable_ref_record ref = { NULL };
|
|
int i = 0;
|
|
int n;
|
|
int err;
|
|
struct reftable_reader rd;
|
|
struct reftable_block_source source = { NULL };
|
|
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
|
|
struct reftable_iterator it = { NULL };
|
|
int j;
|
|
|
|
set_test_hash(want_hash, 4);
|
|
|
|
for (i = 0; i < N; i++) {
|
|
uint8_t hash[GIT_SHA1_RAWSZ];
|
|
char fill[51] = { 0 };
|
|
char name[100];
|
|
uint8_t hash1[GIT_SHA1_RAWSZ];
|
|
uint8_t hash2[GIT_SHA1_RAWSZ];
|
|
struct reftable_ref_record ref = { NULL };
|
|
|
|
memset(hash, i, sizeof(hash));
|
|
memset(fill, 'x', 50);
|
|
/* Put the variable part in the start */
|
|
snprintf(name, sizeof(name), "br%02d%s", i, fill);
|
|
name[40] = 0;
|
|
ref.refname = name;
|
|
|
|
set_test_hash(hash1, i / 4);
|
|
set_test_hash(hash2, 3 + i / 4);
|
|
ref.value_type = REFTABLE_REF_VAL2;
|
|
ref.value.val2.value = hash1;
|
|
ref.value.val2.target_value = hash2;
|
|
|
|
/* 80 bytes / entry, so 3 entries per block. Yields 17
|
|
*/
|
|
/* blocks. */
|
|
n = reftable_writer_add_ref(w, &ref);
|
|
EXPECT(n == 0);
|
|
|
|
if (!memcmp(hash1, want_hash, GIT_SHA1_RAWSZ) ||
|
|
!memcmp(hash2, want_hash, GIT_SHA1_RAWSZ)) {
|
|
want_names[want_names_len++] = xstrdup(name);
|
|
}
|
|
}
|
|
|
|
n = reftable_writer_close(w);
|
|
EXPECT(n == 0);
|
|
|
|
reftable_writer_free(w);
|
|
w = NULL;
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = init_reader(&rd, &source, "file.ref");
|
|
EXPECT_ERR(err);
|
|
if (!indexed) {
|
|
rd.obj_offsets.is_present = 0;
|
|
}
|
|
|
|
err = reftable_reader_seek_ref(&rd, &it, "");
|
|
EXPECT_ERR(err);
|
|
reftable_iterator_destroy(&it);
|
|
|
|
err = reftable_reader_refs_for(&rd, &it, want_hash);
|
|
EXPECT_ERR(err);
|
|
|
|
j = 0;
|
|
while (1) {
|
|
int err = reftable_iterator_next_ref(&it, &ref);
|
|
EXPECT(err >= 0);
|
|
if (err > 0) {
|
|
break;
|
|
}
|
|
|
|
EXPECT(j < want_names_len);
|
|
EXPECT(0 == strcmp(ref.refname, want_names[j]));
|
|
j++;
|
|
reftable_ref_record_release(&ref);
|
|
}
|
|
EXPECT(j == want_names_len);
|
|
|
|
strbuf_release(&buf);
|
|
free_names(want_names);
|
|
reftable_iterator_destroy(&it);
|
|
reader_close(&rd);
|
|
}
|
|
|
|
static void test_table_refs_for_no_index(void)
|
|
{
|
|
test_table_refs_for(0);
|
|
}
|
|
|
|
static void test_table_refs_for_obj_index(void)
|
|
{
|
|
test_table_refs_for(1);
|
|
}
|
|
|
|
static void test_write_empty_table(void)
|
|
{
|
|
struct reftable_write_options opts = { 0 };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
struct reftable_block_source source = { NULL };
|
|
struct reftable_reader *rd = NULL;
|
|
struct reftable_ref_record rec = { NULL };
|
|
struct reftable_iterator it = { NULL };
|
|
int err;
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
|
|
err = reftable_writer_close(w);
|
|
EXPECT(err == REFTABLE_EMPTY_TABLE_ERROR);
|
|
reftable_writer_free(w);
|
|
|
|
EXPECT(buf.len == header_size(1) + footer_size(1));
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
|
|
err = reftable_new_reader(&rd, &source, "filename");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_reader_seek_ref(rd, &it, "");
|
|
EXPECT_ERR(err);
|
|
|
|
err = reftable_iterator_next_ref(&it, &rec);
|
|
EXPECT(err > 0);
|
|
|
|
reftable_iterator_destroy(&it);
|
|
reftable_reader_free(rd);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_write_object_id_min_length(void)
|
|
{
|
|
struct reftable_write_options opts = {
|
|
.block_size = 75,
|
|
};
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
uint8_t hash[GIT_SHA1_RAWSZ] = {42};
|
|
struct reftable_ref_record ref = {
|
|
.update_index = 1,
|
|
.value_type = REFTABLE_REF_VAL1,
|
|
.value.val1 = hash,
|
|
};
|
|
int err;
|
|
int i;
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
|
|
/* Write the same hash in many refs. If there is only 1 hash, the
|
|
* disambiguating prefix is length 0 */
|
|
for (i = 0; i < 256; i++) {
|
|
char name[256];
|
|
snprintf(name, sizeof(name), "ref%05d", i);
|
|
ref.refname = name;
|
|
err = reftable_writer_add_ref(w, &ref);
|
|
EXPECT_ERR(err);
|
|
}
|
|
|
|
err = reftable_writer_close(w);
|
|
EXPECT_ERR(err);
|
|
EXPECT(reftable_writer_stats(w)->object_id_len == 2);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_write_object_id_length(void)
|
|
{
|
|
struct reftable_write_options opts = {
|
|
.block_size = 75,
|
|
};
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
uint8_t hash[GIT_SHA1_RAWSZ] = {42};
|
|
struct reftable_ref_record ref = {
|
|
.update_index = 1,
|
|
.value_type = REFTABLE_REF_VAL1,
|
|
.value.val1 = hash,
|
|
};
|
|
int err;
|
|
int i;
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
|
|
/* Write the same hash in many refs. If there is only 1 hash, the
|
|
* disambiguating prefix is length 0 */
|
|
for (i = 0; i < 256; i++) {
|
|
char name[256];
|
|
snprintf(name, sizeof(name), "ref%05d", i);
|
|
ref.refname = name;
|
|
ref.value.val1[15] = i;
|
|
err = reftable_writer_add_ref(w, &ref);
|
|
EXPECT_ERR(err);
|
|
}
|
|
|
|
err = reftable_writer_close(w);
|
|
EXPECT_ERR(err);
|
|
EXPECT(reftable_writer_stats(w)->object_id_len == 16);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_write_empty_key(void)
|
|
{
|
|
struct reftable_write_options opts = { 0 };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
struct reftable_ref_record ref = {
|
|
.refname = "",
|
|
.update_index = 1,
|
|
.value_type = REFTABLE_REF_DELETION,
|
|
};
|
|
int err;
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
err = reftable_writer_add_ref(w, &ref);
|
|
EXPECT(err == REFTABLE_API_ERROR);
|
|
|
|
err = reftable_writer_close(w);
|
|
EXPECT(err == REFTABLE_EMPTY_TABLE_ERROR);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_write_key_order(void)
|
|
{
|
|
struct reftable_write_options opts = { 0 };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_writer *w =
|
|
reftable_new_writer(&strbuf_add_void, &buf, &opts);
|
|
struct reftable_ref_record refs[2] = {
|
|
{
|
|
.refname = "b",
|
|
.update_index = 1,
|
|
.value_type = REFTABLE_REF_SYMREF,
|
|
.value = {
|
|
.symref = "target",
|
|
},
|
|
}, {
|
|
.refname = "a",
|
|
.update_index = 1,
|
|
.value_type = REFTABLE_REF_SYMREF,
|
|
.value = {
|
|
.symref = "target",
|
|
},
|
|
}
|
|
};
|
|
int err;
|
|
|
|
reftable_writer_set_limits(w, 1, 1);
|
|
err = reftable_writer_add_ref(w, &refs[0]);
|
|
EXPECT_ERR(err);
|
|
err = reftable_writer_add_ref(w, &refs[1]);
|
|
EXPECT(err == REFTABLE_API_ERROR);
|
|
reftable_writer_close(w);
|
|
reftable_writer_free(w);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
static void test_corrupt_table_empty(void)
|
|
{
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_block_source source = { NULL };
|
|
struct reftable_reader rd = { NULL };
|
|
int err;
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
err = init_reader(&rd, &source, "file.log");
|
|
EXPECT(err == REFTABLE_FORMAT_ERROR);
|
|
}
|
|
|
|
static void test_corrupt_table(void)
|
|
{
|
|
uint8_t zeros[1024] = { 0 };
|
|
struct strbuf buf = STRBUF_INIT;
|
|
struct reftable_block_source source = { NULL };
|
|
struct reftable_reader rd = { NULL };
|
|
int err;
|
|
strbuf_add(&buf, zeros, sizeof(zeros));
|
|
|
|
block_source_from_strbuf(&source, &buf);
|
|
err = init_reader(&rd, &source, "file.log");
|
|
EXPECT(err == REFTABLE_FORMAT_ERROR);
|
|
strbuf_release(&buf);
|
|
}
|
|
|
|
int readwrite_test_main(int argc, const char *argv[])
|
|
{
|
|
RUN_TEST(test_log_zlib_corruption);
|
|
RUN_TEST(test_corrupt_table);
|
|
RUN_TEST(test_corrupt_table_empty);
|
|
RUN_TEST(test_log_write_read);
|
|
RUN_TEST(test_write_key_order);
|
|
RUN_TEST(test_table_read_write_seek_linear_sha256);
|
|
RUN_TEST(test_log_buffer_size);
|
|
RUN_TEST(test_table_write_small_table);
|
|
RUN_TEST(test_buffer);
|
|
RUN_TEST(test_table_read_api);
|
|
RUN_TEST(test_table_read_write_sequential);
|
|
RUN_TEST(test_table_read_write_seek_linear);
|
|
RUN_TEST(test_table_read_write_seek_index);
|
|
RUN_TEST(test_table_refs_for_no_index);
|
|
RUN_TEST(test_table_refs_for_obj_index);
|
|
RUN_TEST(test_write_empty_key);
|
|
RUN_TEST(test_write_empty_table);
|
|
RUN_TEST(test_log_overflow);
|
|
RUN_TEST(test_write_object_id_length);
|
|
RUN_TEST(test_write_object_id_min_length);
|
|
return 0;
|
|
}
|