undef/git-commit-vandalism
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reftable: make reftable_record a tagged union

Browse Source 
This reduces the amount of glue code, because we don't need a void
pointer or vtable within the structure.

The only snag is that reftable_index_record contain a strbuf, so it
cannot be zero-initialized. To address this, use reftable_new_record()
to return fresh instance, given a record type. Since
reftable_new_record() doesn't cause heap allocation anymore, it should
be balanced with reftable_record_release() rather than
reftable_record_destroy().

Thanks to Peff for the suggestion.

Helped-by: Jeff King <peff@peff.net>
Signed-off-by: Han-Wen Nienhuys <hanwen@google.com>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
This commit is contained in:
Han-Wen Nienhuys 2022-01-20 15:12:13 +00:00 committed by Junio C Hamano
parent 9391b88dab
commit 66c0dabab5
12 changed files with 334 additions and 337 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
reftable/block.c
View File

@ -421,7 +421,7 @@ int block_reader_seek(struct block_reader *br, struct block_iter *it,
done: done:
strbuf_release(&key); strbuf_release(&key);
strbuf_release(&next.last_key); strbuf_release(&next.last_key);
reftable_record_destroy(&rec); reftable_record_release(&rec);
return err; return err;
} }

22
reftable/block_test.c
View File

@ -26,8 +26,9 @@ static void test_block_read_write(void)
struct block_writer bw = { struct block_writer bw = {
.last_key = STRBUF_INIT, .last_key = STRBUF_INIT,
}; };
struct reftable_ref_record ref = { NULL }; struct reftable_record rec = {
struct reftable_record rec = { NULL }; .type = BLOCK_TYPE_REF,
};
int i = 0; int i = 0;
int n; int n;
struct block_reader br = { 0 }; struct block_reader br = { 0 };
@ -40,7 +41,6 @@ static void test_block_read_write(void)
block.source = malloc_block_source(); block.source = malloc_block_source();
block_writer_init(&bw, BLOCK_TYPE_REF, block.data, block_size, block_writer_init(&bw, BLOCK_TYPE_REF, block.data, block_size,
header_off, hash_size(GIT_SHA1_FORMAT_ID)); header_off, hash_size(GIT_SHA1_FORMAT_ID));
reftable_record_from_ref(&rec, &ref);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
char name[100]; char name[100];
@ -48,14 +48,14 @@ static void test_block_read_write(void)
snprintf(name, sizeof(name), "branch%02d", i); snprintf(name, sizeof(name), "branch%02d", i);
memset(hash, i, sizeof(hash)); memset(hash, i, sizeof(hash));
ref.refname = name; rec.u.ref.refname = name;
ref.value_type = REFTABLE_REF_VAL1; rec.u.ref.value_type = REFTABLE_REF_VAL1;
ref.value.val1 = hash; rec.u.ref.value.val1 = hash;
names[i] = xstrdup(name); names[i] = xstrdup(name);
n = block_writer_add(&bw, &rec); n = block_writer_add(&bw, &rec);
ref.refname = NULL; rec.u.ref.refname = NULL;
ref.value_type = REFTABLE_REF_DELETION; rec.u.ref.value_type = REFTABLE_REF_DELETION;
EXPECT(n == 0); EXPECT(n == 0);
} }
@ -74,7 +74,7 @@ static void test_block_read_write(void)
if (r > 0) { if (r > 0) {
break; break;
} }
EXPECT_STREQ(names[j], ref.refname); EXPECT_STREQ(names[j], rec.u.ref.refname);
j++; j++;
} }
@ -92,7 +92,7 @@ static void test_block_read_write(void)
n = block_iter_next(&it, &rec); n = block_iter_next(&it, &rec);
EXPECT(n == 0); EXPECT(n == 0);
EXPECT_STREQ(names[i], ref.refname); EXPECT_STREQ(names[i], rec.u.ref.refname);
want.len--; want.len--;
n = block_reader_seek(&br, &it, &want); n = block_reader_seek(&br, &it, &want);
@ -100,7 +100,7 @@ static void test_block_read_write(void)
n = block_iter_next(&it, &rec); n = block_iter_next(&it, &rec);
EXPECT(n == 0); EXPECT(n == 0);
EXPECT_STREQ(names[10 * (i / 10)], ref.refname); EXPECT_STREQ(names[10 * (i / 10)], rec.u.ref.refname);
block_iter_close(&it); block_iter_close(&it);
} }

41
reftable/generic.c
View File

@ -7,6 +7,7 @@ https://developers.google.com/open-source/licenses/bsd
*/ */
#include "basics.h" #include "basics.h"
#include "constants.h"
#include "record.h" #include "record.h"
#include "generic.h" #include "generic.h"
#include "reftable-iterator.h" #include "reftable-iterator.h"
@ -15,23 +16,21 @@ https://developers.google.com/open-source/licenses/bsd
int reftable_table_seek_ref(struct reftable_table *tab, int reftable_table_seek_ref(struct reftable_table *tab,
struct reftable_iterator *it, const char *name) struct reftable_iterator *it, const char *name)
{ {
struct reftable_ref_record ref = { struct reftable_record rec = { .type = BLOCK_TYPE_REF,
.refname = (char *)name, .u.ref = {
}; .refname = (char *)name,
struct reftable_record rec = { NULL }; } };
reftable_record_from_ref(&rec, &ref);
return tab->ops->seek_record(tab->table_arg, it, &rec); return tab->ops->seek_record(tab->table_arg, it, &rec);
} }
int reftable_table_seek_log(struct reftable_table *tab, int reftable_table_seek_log(struct reftable_table *tab,
struct reftable_iterator *it, const char *name) struct reftable_iterator *it, const char *name)
{ {
struct reftable_log_record log = { struct reftable_record rec = { .type = BLOCK_TYPE_LOG,
.refname = (char *)name, .u.log = {
.update_index = ~((uint64_t)0), .refname = (char *)name,
}; .update_index = ~((uint64_t)0),
struct reftable_record rec = { NULL }; } };
reftable_record_from_log(&rec, &log);
return tab->ops->seek_record(tab->table_arg, it, &rec); return tab->ops->seek_record(tab->table_arg, it, &rec);
} }
@ -129,17 +128,25 @@ void reftable_iterator_destroy(struct reftable_iterator *it)
int reftable_iterator_next_ref(struct reftable_iterator *it, int reftable_iterator_next_ref(struct reftable_iterator *it,
struct reftable_ref_record *ref) struct reftable_ref_record *ref)
{ {
struct reftable_record rec = { NULL }; struct reftable_record rec = {
reftable_record_from_ref(&rec, ref); .type = BLOCK_TYPE_REF,
return iterator_next(it, &rec); .u.ref = *ref,
};
int err = iterator_next(it, &rec);
*ref = rec.u.ref;
return err;
} }
int reftable_iterator_next_log(struct reftable_iterator *it, int reftable_iterator_next_log(struct reftable_iterator *it,
struct reftable_log_record *log) struct reftable_log_record *log)
{ {
struct reftable_record rec = { NULL }; struct reftable_record rec = {
reftable_record_from_log(&rec, log); .type = BLOCK_TYPE_LOG,
return iterator_next(it, &rec); .u.log = *log,
};
int err = iterator_next(it, &rec);
*log = rec.u.log;
return err;
} }
int iterator_next(struct reftable_iterator *it, struct reftable_record *rec) int iterator_next(struct reftable_iterator *it, struct reftable_record *rec)

4
reftable/iter.c
View File

@ -32,7 +32,7 @@ static int filtering_ref_iterator_next(void *iter_arg,
struct reftable_record *rec) struct reftable_record *rec)
{ {
struct filtering_ref_iterator *fri = iter_arg; struct filtering_ref_iterator *fri = iter_arg;
struct reftable_ref_record *ref = rec->data; struct reftable_ref_record *ref = &rec->u.ref;
int err = 0; int err = 0;
while (1) { while (1) {
err = reftable_iterator_next_ref(&fri->it, ref); err = reftable_iterator_next_ref(&fri->it, ref);
@ -127,7 +127,7 @@ static int indexed_table_ref_iter_next_block(struct indexed_table_ref_iter *it)
static int indexed_table_ref_iter_next(void *p, struct reftable_record *rec) static int indexed_table_ref_iter_next(void *p, struct reftable_record *rec)
{ {
struct indexed_table_ref_iter *it = p; struct indexed_table_ref_iter *it = p;
struct reftable_ref_record *ref = rec->data; struct reftable_ref_record *ref = &rec->u.ref;
while (1) { while (1) {
int err = block_iter_next(&it->cur, rec); int err = block_iter_next(&it->cur, rec);

33
reftable/merged.c
View File

@ -30,7 +30,7 @@ static int merged_iter_init(struct merged_iter *mi)
if (err > 0) { if (err > 0) {
reftable_iterator_destroy(&mi->stack[i]); reftable_iterator_destroy(&mi->stack[i]);
reftable_record_destroy(&rec); reftable_record_release(&rec);
} else { } else {
struct pq_entry e = { struct pq_entry e = {
.rec = rec, .rec = rec,
@ -57,18 +57,17 @@ static void merged_iter_close(void *p)
static int merged_iter_advance_nonnull_subiter(struct merged_iter *mi, static int merged_iter_advance_nonnull_subiter(struct merged_iter *mi,
size_t idx) size_t idx)
{ {
struct reftable_record rec = reftable_new_record(mi->typ);
struct pq_entry e = { struct pq_entry e = {
.rec = rec, .rec = reftable_new_record(mi->typ),
.index = idx, .index = idx,
}; };
int err = iterator_next(&mi->stack[idx], &rec); int err = iterator_next(&mi->stack[idx], &e.rec);
if (err < 0) if (err < 0)
return err; return err;
if (err > 0) { if (err > 0) {
reftable_iterator_destroy(&mi->stack[idx]); reftable_iterator_destroy(&mi->stack[idx]);
reftable_record_destroy(&rec); reftable_record_release(&e.rec);
return 0; return 0;
} }
@ -126,11 +125,11 @@ static int merged_iter_next_entry(struct merged_iter *mi,
if (err < 0) { if (err < 0) {
return err; return err;
} }
reftable_record_destroy(&top.rec); reftable_record_release(&top.rec);
} }
reftable_record_copy_from(rec, &entry.rec, hash_size(mi->hash_id)); reftable_record_copy_from(rec, &entry.rec, hash_size(mi->hash_id));
reftable_record_destroy(&entry.rec); reftable_record_release(&entry.rec);
strbuf_release(&entry_key); strbuf_release(&entry_key);
return 0; return 0;
} }
@ -290,11 +289,12 @@ int reftable_merged_table_seek_ref(struct reftable_merged_table *mt,
struct reftable_iterator *it, struct reftable_iterator *it,
const char *name) const char *name)
{ {
struct reftable_ref_record ref = { struct reftable_record rec = {
.refname = (char *)name, .type = BLOCK_TYPE_REF,
.u.ref = {
.refname = (char *)name,
},
}; };
struct reftable_record rec = { NULL };
reftable_record_from_ref(&rec, &ref);
return merged_table_seek_record(mt, it, &rec); return merged_table_seek_record(mt, it, &rec);
} }
@ -302,12 +302,11 @@ int reftable_merged_table_seek_log_at(struct reftable_merged_table *mt,
struct reftable_iterator *it, struct reftable_iterator *it,
const char *name, uint64_t update_index) const char *name, uint64_t update_index)
{ {
struct reftable_log_record log = { struct reftable_record rec = { .type = BLOCK_TYPE_LOG,
.refname = (char *)name, .u.log = {
.update_index = update_index, .refname = (char *)name,
}; .update_index = update_index,
struct reftable_record rec = { NULL }; } };
reftable_record_from_log(&rec, &log);
return merged_table_seek_record(mt, it, &rec); return merged_table_seek_record(mt, it, &rec);
} }

3
reftable/pq.c
View File

@ -74,6 +74,7 @@ struct pq_entry merged_iter_pqueue_remove(struct merged_iter_pqueue *pq)
void merged_iter_pqueue_add(struct merged_iter_pqueue *pq, struct pq_entry e) void merged_iter_pqueue_add(struct merged_iter_pqueue *pq, struct pq_entry e)
{ {
int i = 0; int i = 0;
if (pq->len == pq->cap) { if (pq->len == pq->cap) {
pq->cap = 2 * pq->cap + 1; pq->cap = 2 * pq->cap + 1;
pq->heap = reftable_realloc(pq->heap, pq->heap = reftable_realloc(pq->heap,
@ -98,7 +99,7 @@ void merged_iter_pqueue_release(struct merged_iter_pqueue *pq)
{ {
int i = 0; int i = 0;
for (i = 0; i < pq->len; i++) { for (i = 0; i < pq->len; i++) {
reftable_record_destroy(&pq->heap[i].rec); reftable_record_release(&pq->heap[i].rec);
} }
FREE_AND_NULL(pq->heap); FREE_AND_NULL(pq->heap);
pq->len = pq->cap = 0; pq->len = pq->cap = 0;

27
reftable/pq_test.c
View File

@ -31,7 +31,7 @@ static void test_pq(void)
int N = ARRAY_SIZE(names) - 1; int N = ARRAY_SIZE(names) - 1;
struct merged_iter_pqueue pq = { NULL }; struct merged_iter_pqueue pq = { NULL };
const char *last = NULL; char *last = NULL;
int i = 0; int i = 0;
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
@ -42,12 +42,10 @@ static void test_pq(void)
i = 1; i = 1;
do { do {
struct reftable_record rec = struct pq_entry e = { .rec = { .type = BLOCK_TYPE_REF,
reftable_new_record(BLOCK_TYPE_REF); .u.ref = {
struct pq_entry e = { 0 }; .refname = names[i],
} } };
reftable_record_as_ref(&rec)->refname = names[i];
e.rec = rec;
merged_iter_pqueue_add(&pq, e); merged_iter_pqueue_add(&pq, e);
merged_iter_pqueue_check(pq); merged_iter_pqueue_check(pq);
i = (i * 7) % N; i = (i * 7) % N;
@ -55,19 +53,18 @@ static void test_pq(void)
while (!merged_iter_pqueue_is_empty(pq)) { while (!merged_iter_pqueue_is_empty(pq)) {
struct pq_entry e = merged_iter_pqueue_remove(&pq); struct pq_entry e = merged_iter_pqueue_remove(&pq);
struct reftable_ref_record *ref = struct reftable_record *rec = &e.rec;
reftable_record_as_ref(&e.rec);
merged_iter_pqueue_check(pq); merged_iter_pqueue_check(pq);
EXPECT(reftable_record_type(rec) == BLOCK_TYPE_REF);
if (last) { if (last) {
EXPECT(strcmp(last, ref->refname) < 0); EXPECT(strcmp(last, rec->u.ref.refname) < 0);
} }
last = ref->refname; // this is names[i], so don't dealloc.
ref->refname = NULL; last = rec->u.ref.refname;
reftable_free(ref); rec->u.ref.refname = NULL;
reftable_record_release(rec);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
reftable_free(names[i]); reftable_free(names[i]);
} }

87
reftable/reader.c
View File

@ -239,8 +239,7 @@ static int table_iter_next_in_block(struct table_iter *ti,
{ {
int res = block_iter_next(&ti->bi, rec); int res = block_iter_next(&ti->bi, rec);
if (res == 0 && reftable_record_type(rec) == BLOCK_TYPE_REF) { if (res == 0 && reftable_record_type(rec) == BLOCK_TYPE_REF) {
((struct reftable_ref_record *)rec->data)->update_index += rec->u.ref.update_index += ti->r->min_update_index;
ti->r->min_update_index;
} }
return res; return res;
@ -480,7 +479,7 @@ static int reader_seek_linear(struct reftable_reader *r, struct table_iter *ti,
done: done:
block_iter_close(&next.bi); block_iter_close(&next.bi);
reftable_record_destroy(&rec); reftable_record_release(&rec);
strbuf_release(&want_key); strbuf_release(&want_key);
strbuf_release(&got_key); strbuf_release(&got_key);
return err; return err;
@ -490,34 +489,35 @@ static int reader_seek_indexed(struct reftable_reader *r,
struct reftable_iterator *it, struct reftable_iterator *it,
struct reftable_record *rec) struct reftable_record *rec)
{ {
struct reftable_index_record want_index = { .last_key = STRBUF_INIT }; struct reftable_record want_index = {
struct reftable_record want_index_rec = { NULL }; .type = BLOCK_TYPE_INDEX, .u.idx = { .last_key = STRBUF_INIT }
struct reftable_index_record index_result = { .last_key = STRBUF_INIT }; };
struct reftable_record index_result_rec = { NULL }; struct reftable_record index_result = {
.type = BLOCK_TYPE_INDEX,
.u.idx = { .last_key = STRBUF_INIT },
};
struct table_iter index_iter = TABLE_ITER_INIT; struct table_iter index_iter = TABLE_ITER_INIT;
struct table_iter next = TABLE_ITER_INIT; struct table_iter next = TABLE_ITER_INIT;
int err = 0; int err = 0;
reftable_record_key(rec, &want_index.last_key); reftable_record_key(rec, &want_index.u.idx.last_key);
reftable_record_from_index(&want_index_rec, &want_index);
reftable_record_from_index(&index_result_rec, &index_result);
err = reader_start(r, &index_iter, reftable_record_type(rec), 1); err = reader_start(r, &index_iter, reftable_record_type(rec), 1);
if (err < 0) if (err < 0)
goto done; goto done;
err = reader_seek_linear(r, &index_iter, &want_index_rec); err = reader_seek_linear(r, &index_iter, &want_index);
while (1) { while (1) {
err = table_iter_next(&index_iter, &index_result_rec); err = table_iter_next(&index_iter, &index_result);
table_iter_block_done(&index_iter); table_iter_block_done(&index_iter);
if (err != 0) if (err != 0)
goto done; goto done;
err = reader_table_iter_at(r, &next, index_result.offset, 0); err = reader_table_iter_at(r, &next, index_result.u.idx.offset,
0);
if (err != 0) if (err != 0)
goto done; goto done;
err = block_iter_seek(&next.bi, &want_index.last_key); err = block_iter_seek(&next.bi, &want_index.u.idx.last_key);
if (err < 0) if (err < 0)
goto done; goto done;
@ -545,8 +545,8 @@ static int reader_seek_indexed(struct reftable_reader *r,
done: done:
block_iter_close(&next.bi); block_iter_close(&next.bi);
table_iter_close(&index_iter); table_iter_close(&index_iter);
reftable_record_release(&want_index_rec); reftable_record_release(&want_index);
reftable_record_release(&index_result_rec); reftable_record_release(&index_result);
return err; return err;
} }
@ -595,11 +595,12 @@ static int reader_seek(struct reftable_reader *r, struct reftable_iterator *it,
int reftable_reader_seek_ref(struct reftable_reader *r, int reftable_reader_seek_ref(struct reftable_reader *r,
struct reftable_iterator *it, const char *name) struct reftable_iterator *it, const char *name)
{ {
struct reftable_ref_record ref = { struct reftable_record rec = {
.refname = (char *)name, .type = BLOCK_TYPE_REF,
.u.ref = {
.refname = (char *)name,
},
}; };
struct reftable_record rec = { NULL };
reftable_record_from_ref(&rec, &ref);
return reader_seek(r, it, &rec); return reader_seek(r, it, &rec);
} }
@ -607,12 +608,11 @@ int reftable_reader_seek_log_at(struct reftable_reader *r,
struct reftable_iterator *it, const char *name, struct reftable_iterator *it, const char *name,
uint64_t update_index) uint64_t update_index)
{ {
struct reftable_log_record log = { struct reftable_record rec = { .type = BLOCK_TYPE_LOG,
.refname = (char *)name, .u.log = {
.update_index = update_index, .refname = (char *)name,
}; .update_index = update_index,
struct reftable_record rec = { NULL }; } };
reftable_record_from_log(&rec, &log);
return reader_seek(r, it, &rec); return reader_seek(r, it, &rec);
} }
@ -656,31 +656,33 @@ static int reftable_reader_refs_for_indexed(struct reftable_reader *r,
struct reftable_iterator *it, struct reftable_iterator *it,
uint8_t *oid) uint8_t *oid)
{ {
struct reftable_obj_record want = { struct reftable_record want = {
.hash_prefix = oid, .type = BLOCK_TYPE_OBJ,
.hash_prefix_len = r->object_id_len, .u.obj = {
.hash_prefix = oid,
.hash_prefix_len = r->object_id_len,
},
}; };
struct reftable_record want_rec = { NULL };
struct reftable_iterator oit = { NULL }; struct reftable_iterator oit = { NULL };
struct reftable_obj_record got = { NULL }; struct reftable_record got = {
struct reftable_record got_rec = { NULL }; .type = BLOCK_TYPE_OBJ,
.u.obj = { 0 },
};
int err = 0; int err = 0;
struct indexed_table_ref_iter *itr = NULL; struct indexed_table_ref_iter *itr = NULL;
/* Look through the reverse index. */ /* Look through the reverse index. */
reftable_record_from_obj(&want_rec, &want); err = reader_seek(r, &oit, &want);
err = reader_seek(r, &oit, &want_rec);
if (err != 0) if (err != 0)
goto done; goto done;
/* read out the reftable_obj_record */ /* read out the reftable_obj_record */
reftable_record_from_obj(&got_rec, &got); err = iterator_next(&oit, &got);
err = iterator_next(&oit, &got_rec);
if (err < 0) if (err < 0)
goto done; goto done;
if (err > 0 || if (err > 0 || memcmp(want.u.obj.hash_prefix, got.u.obj.hash_prefix,
memcmp(want.hash_prefix, got.hash_prefix, r->object_id_len)) { r->object_id_len)) {
/* didn't find it; return empty iterator */ /* didn't find it; return empty iterator */
iterator_set_empty(it); iterator_set_empty(it);
err = 0; err = 0;
@ -688,15 +690,16 @@ static int reftable_reader_refs_for_indexed(struct reftable_reader *r,
} }
err = new_indexed_table_ref_iter(&itr, r, oid, hash_size(r->hash_id), err = new_indexed_table_ref_iter(&itr, r, oid, hash_size(r->hash_id),
got.offsets, got.offset_len); got.u.obj.offsets,
got.u.obj.offset_len);
if (err < 0) if (err < 0)
goto done; goto done;
got.offsets = NULL; got.u.obj.offsets = NULL;
iterator_from_indexed_table_ref_iter(it, itr); iterator_from_indexed_table_ref_iter(it, itr);
done: done:
reftable_iterator_destroy(&oit); reftable_iterator_destroy(&oit);
reftable_record_release(&got_rec); reftable_record_release(&got);
return err; return err;
} }

212
reftable/record.c
View File

@ -15,6 +15,10 @@ https://developers.google.com/open-source/licenses/bsd
#include "reftable-error.h" #include "reftable-error.h"
#include "basics.h" #include "basics.h"
static struct reftable_record_vtable *
reftable_record_vtable(struct reftable_record *rec);
static void *reftable_record_data(struct reftable_record *rec);
int get_var_int(uint64_t *dest, struct string_view *in) int get_var_int(uint64_t *dest, struct string_view *in)
{ {
int ptr = 0; int ptr = 0;
@ -475,12 +479,14 @@ static void reftable_obj_record_copy_from(void *rec, const void *src_rec,
(const struct reftable_obj_record *)src_rec; (const struct reftable_obj_record *)src_rec;
reftable_obj_record_release(obj); reftable_obj_record_release(obj);
*obj = *src; obj->hash_prefix = reftable_malloc(src->hash_prefix_len);
obj->hash_prefix = reftable_malloc(obj->hash_prefix_len); obj->hash_prefix_len = src->hash_prefix_len;
memcpy(obj->hash_prefix, src->hash_prefix, obj->hash_prefix_len); if (src->hash_prefix_len)
memcpy(obj->hash_prefix, src->hash_prefix, obj->hash_prefix_len);
obj->offsets = reftable_malloc(obj->offset_len * sizeof(uint64_t)); obj->offsets = reftable_malloc(src->offset_len * sizeof(uint64_t));
COPY_ARRAY(obj->offsets, src->offsets, obj->offset_len); obj->offset_len = src->offset_len;
COPY_ARRAY(obj->offsets, src->offsets, src->offset_len);
} }
static uint8_t reftable_obj_record_val_type(const void *rec) static uint8_t reftable_obj_record_val_type(const void *rec)
@ -965,58 +971,6 @@ static struct reftable_record_vtable reftable_log_record_vtable = {
.equal = &reftable_log_record_equal_void .equal = &reftable_log_record_equal_void
}; };
struct reftable_record reftable_new_record(uint8_t typ)
{
struct reftable_record rec = { NULL };
switch (typ) {
case BLOCK_TYPE_REF: {
struct reftable_ref_record *r =
reftable_calloc(sizeof(struct reftable_ref_record));
reftable_record_from_ref(&rec, r);
return rec;
}
case BLOCK_TYPE_OBJ: {
struct reftable_obj_record *r =
reftable_calloc(sizeof(struct reftable_obj_record));
reftable_record_from_obj(&rec, r);
return rec;
}
case BLOCK_TYPE_LOG: {
struct reftable_log_record *r =
reftable_calloc(sizeof(struct reftable_log_record));
reftable_record_from_log(&rec, r);
return rec;
}
case BLOCK_TYPE_INDEX: {
struct reftable_index_record empty = { .last_key =
STRBUF_INIT };
struct reftable_index_record *r =
reftable_calloc(sizeof(struct reftable_index_record));
*r = empty;
reftable_record_from_index(&rec, r);
return rec;
}
}
abort();
return rec;
}
/* clear out the record, yielding the reftable_record data that was
* encapsulated. */
static void *reftable_record_yield(struct reftable_record *rec)
{
void *p = rec->data;
rec->data = NULL;
return p;
}
void reftable_record_destroy(struct reftable_record *rec)
{
reftable_record_release(rec);
reftable_free(reftable_record_yield(rec));
}
static void reftable_index_record_key(const void *r, struct strbuf *dest) static void reftable_index_record_key(const void *r, struct strbuf *dest)
{ {
const struct reftable_index_record *rec = r; const struct reftable_index_record *rec = r;
@ -1103,98 +1057,60 @@ static struct reftable_record_vtable reftable_index_record_vtable = {
void reftable_record_key(struct reftable_record *rec, struct strbuf *dest) void reftable_record_key(struct reftable_record *rec, struct strbuf *dest)
{ {
rec->ops->key(rec->data, dest); reftable_record_vtable(rec)->key(reftable_record_data(rec), dest);
} }
uint8_t reftable_record_type(struct reftable_record *rec) uint8_t reftable_record_type(struct reftable_record *rec)
{ {
return rec->ops->type; return rec->type;
} }
int reftable_record_encode(struct reftable_record *rec, struct string_view dest, int reftable_record_encode(struct reftable_record *rec, struct string_view dest,
int hash_size) int hash_size)
{ {
return rec->ops->encode(rec->data, dest, hash_size); return reftable_record_vtable(rec)->encode(reftable_record_data(rec),
dest, hash_size);
} }
void reftable_record_copy_from(struct reftable_record *rec, void reftable_record_copy_from(struct reftable_record *rec,
struct reftable_record *src, int hash_size) struct reftable_record *src, int hash_size)
{ {
assert(src->ops->type == rec->ops->type); assert(src->type == rec->type);
rec->ops->copy_from(rec->data, src->data, hash_size); reftable_record_vtable(rec)->copy_from(reftable_record_data(rec),
reftable_record_data(src),
hash_size);
} }
uint8_t reftable_record_val_type(struct reftable_record *rec) uint8_t reftable_record_val_type(struct reftable_record *rec)
{ {
return rec->ops->val_type(rec->data); return reftable_record_vtable(rec)->val_type(reftable_record_data(rec));
} }
int reftable_record_decode(struct reftable_record *rec, struct strbuf key, int reftable_record_decode(struct reftable_record *rec, struct strbuf key,
uint8_t extra, struct string_view src, int hash_size) uint8_t extra, struct string_view src, int hash_size)
{ {
return rec->ops->decode(rec->data, key, extra, src, hash_size); return reftable_record_vtable(rec)->decode(reftable_record_data(rec),
key, extra, src, hash_size);
} }
void reftable_record_release(struct reftable_record *rec) void reftable_record_release(struct reftable_record *rec)
{ {
rec->ops->release(rec->data); reftable_record_vtable(rec)->release(reftable_record_data(rec));
} }
int reftable_record_is_deletion(struct reftable_record *rec) int reftable_record_is_deletion(struct reftable_record *rec)
{ {
return rec->ops->is_deletion(rec->data); return reftable_record_vtable(rec)->is_deletion(
reftable_record_data(rec));
} }
int reftable_record_equal(struct reftable_record *a, struct reftable_record *b, int hash_size) int reftable_record_equal(struct reftable_record *a, struct reftable_record *b, int hash_size)
{ {
if (a->ops != b->ops) if (a->type != b->type)
return 0; return 0;
return a->ops->equal(a->data, b->data, hash_size); return reftable_record_vtable(a)->equal(
} reftable_record_data(a), reftable_record_data(b), hash_size);
void reftable_record_from_ref(struct reftable_record *rec,
struct reftable_ref_record *ref_rec)
{
assert(!rec->ops);
rec->data = ref_rec;
rec->ops = &reftable_ref_record_vtable;
}
void reftable_record_from_obj(struct reftable_record *rec,
struct reftable_obj_record *obj_rec)
{
assert(!rec->ops);
rec->data = obj_rec;
rec->ops = &reftable_obj_record_vtable;
}
void reftable_record_from_index(struct reftable_record *rec,
struct reftable_index_record *index_rec)
{
assert(!rec->ops);
rec->data = index_rec;
rec->ops = &reftable_index_record_vtable;
}
void reftable_record_from_log(struct reftable_record *rec,
struct reftable_log_record *log_rec)
{
assert(!rec->ops);
rec->data = log_rec;
rec->ops = &reftable_log_record_vtable;
}
struct reftable_ref_record *reftable_record_as_ref(struct reftable_record *rec)
{
assert(reftable_record_type(rec) == BLOCK_TYPE_REF);
return rec->data;
}
struct reftable_log_record *reftable_record_as_log(struct reftable_record *rec)
{
assert(reftable_record_type(rec) == BLOCK_TYPE_LOG);
return rec->data;
} }
static int hash_equal(uint8_t *a, uint8_t *b, int hash_size) static int hash_equal(uint8_t *a, uint8_t *b, int hash_size)
@ -1267,3 +1183,75 @@ void string_view_consume(struct string_view *s, int n)
s->buf += n; s->buf += n;
s->len -= n; s->len -= n;
} }
static void *reftable_record_data(struct reftable_record *rec)
{
switch (rec->type) {
case BLOCK_TYPE_REF:
return &rec->u.ref;
case BLOCK_TYPE_LOG:
return &rec->u.log;
case BLOCK_TYPE_INDEX:
return &rec->u.idx;
case BLOCK_TYPE_OBJ:
return &rec->u.obj;
}
abort();
}
static struct reftable_record_vtable *
reftable_record_vtable(struct reftable_record *rec)
{
switch (rec->type) {
case BLOCK_TYPE_REF:
return &reftable_ref_record_vtable;
case BLOCK_TYPE_LOG:
return &reftable_log_record_vtable;
case BLOCK_TYPE_INDEX:
return &reftable_index_record_vtable;
case BLOCK_TYPE_OBJ:
return &reftable_obj_record_vtable;
}
abort();
}
struct reftable_record reftable_new_record(uint8_t typ)
{
struct reftable_record clean = {
.type = typ,
};
/* the following is involved, but the naive solution (just return
* `clean` as is, except for BLOCK_TYPE_INDEX), returns a garbage
* clean.u.obj.offsets pointer on Windows VS CI. Go figure.
*/
switch (typ) {
case BLOCK_TYPE_OBJ:
{
struct reftable_obj_record obj = { 0 };
clean.u.obj = obj;
break;
}
case BLOCK_TYPE_INDEX:
{
struct reftable_index_record idx = {
.last_key = STRBUF_INIT,
};
clean.u.idx = idx;
break;
}
case BLOCK_TYPE_REF:
{
struct reftable_ref_record ref = { 0 };
clean.u.ref = ref;
break;
}
case BLOCK_TYPE_LOG:
{
struct reftable_log_record log = { 0 };
clean.u.log = log;
break;
}
}
return clean;
}

42
reftable/record.h
View File

@ -63,16 +63,10 @@ struct reftable_record_vtable {
int (*equal)(const void *a, const void *b, int hash_size); int (*equal)(const void *a, const void *b, int hash_size);
}; };
/* record is a generic wrapper for different types of records. */
struct reftable_record {
void *data;
struct reftable_record_vtable *ops;
};
/* returns true for recognized block types. Block start with the block type. */ /* returns true for recognized block types. Block start with the block type. */
int reftable_is_block_type(uint8_t typ); int reftable_is_block_type(uint8_t typ);
/* creates a malloced record of the given type. Dispose with record_destroy */ /* return an initialized record for the given type */
struct reftable_record reftable_new_record(uint8_t typ); struct reftable_record reftable_new_record(uint8_t typ);
/* Encode `key` into `dest`. Sets `is_restart` to indicate a restart. Returns /* Encode `key` into `dest`. Sets `is_restart` to indicate a restart. Returns
@ -100,6 +94,22 @@ struct reftable_obj_record {
int offset_len; int offset_len;
}; };
/* record is a generic wrapper for different types of records. It is normally
* created on the stack, or embedded within another struct. If the type is
* known, a fresh instance can be initialized explicitly. Otherwise, use
* reftable_new_record() to initialize generically (as the index_record is not
* valid as 0-initialized structure)
*/
struct reftable_record {
uint8_t type;
union {
struct reftable_ref_record ref;
struct reftable_log_record log;
struct reftable_obj_record obj;
struct reftable_index_record idx;
} u;
};
/* see struct record_vtable */ /* see struct record_vtable */
int reftable_record_equal(struct reftable_record *a, struct reftable_record *b, int hash_size); int reftable_record_equal(struct reftable_record *a, struct reftable_record *b, int hash_size);
void reftable_record_key(struct reftable_record *rec, struct strbuf *dest); void reftable_record_key(struct reftable_record *rec, struct strbuf *dest);
@ -114,25 +124,9 @@ int reftable_record_decode(struct reftable_record *rec, struct strbuf key,
int hash_size); int hash_size);
int reftable_record_is_deletion(struct reftable_record *rec); int reftable_record_is_deletion(struct reftable_record *rec);
/* zeroes out the embedded record */ /* frees and zeroes out the embedded record */
void reftable_record_release(struct reftable_record *rec); void reftable_record_release(struct reftable_record *rec);
/* clear and deallocate embedded record, and zero `rec`. */
void reftable_record_destroy(struct reftable_record *rec);
/* initialize generic records from concrete records. The generic record should
* be zeroed out. */
void reftable_record_from_obj(struct reftable_record *rec,
struct reftable_obj_record *objrec);
void reftable_record_from_index(struct reftable_record *rec,
struct reftable_index_record *idxrec);
void reftable_record_from_ref(struct reftable_record *rec,
struct reftable_ref_record *refrec);
void reftable_record_from_log(struct reftable_record *rec,
struct reftable_log_record *logrec);
struct reftable_ref_record *reftable_record_as_ref(struct reftable_record *ref);
struct reftable_log_record *reftable_record_as_log(struct reftable_record *ref);
/* for qsort. */ /* for qsort. */
int reftable_ref_record_compare_name(const void *a, const void *b); int reftable_ref_record_compare_name(const void *a, const void *b);

157
reftable/record_test.c
View File

@ -16,13 +16,16 @@
static void test_copy(struct reftable_record *rec) static void test_copy(struct reftable_record *rec)
{ {
struct reftable_record copy = struct reftable_record copy = { 0 };
reftable_new_record(reftable_record_type(rec)); uint8_t typ;
typ = reftable_record_type(rec);
copy = reftable_new_record(typ);
reftable_record_copy_from(&copy, rec, GIT_SHA1_RAWSZ); reftable_record_copy_from(&copy, rec, GIT_SHA1_RAWSZ);
/* do it twice to catch memory leaks */ /* do it twice to catch memory leaks */
reftable_record_copy_from(&copy, rec, GIT_SHA1_RAWSZ); reftable_record_copy_from(&copy, rec, GIT_SHA1_RAWSZ);
EXPECT(reftable_record_equal(rec, &copy, GIT_SHA1_RAWSZ)); EXPECT(reftable_record_equal(rec, &copy, GIT_SHA1_RAWSZ));
reftable_record_destroy(&copy); reftable_record_release(&copy);
} }
static void test_varint_roundtrip(void) static void test_varint_roundtrip(void)
@ -95,61 +98,58 @@ static void test_reftable_ref_record_roundtrip(void)
int i = 0; int i = 0;
for (i = REFTABLE_REF_DELETION; i < REFTABLE_NR_REF_VALUETYPES; i++) { for (i = REFTABLE_REF_DELETION; i < REFTABLE_NR_REF_VALUETYPES; i++) {
struct reftable_ref_record in = { NULL }; struct reftable_record in = {
struct reftable_ref_record out = { NULL }; .type = BLOCK_TYPE_REF,
struct reftable_record rec_out = { NULL }; };
struct reftable_record out = { .type = BLOCK_TYPE_REF };
struct strbuf key = STRBUF_INIT; struct strbuf key = STRBUF_INIT;
struct reftable_record rec = { NULL };
uint8_t buffer[1024] = { 0 }; uint8_t buffer[1024] = { 0 };
struct string_view dest = { struct string_view dest = {
.buf = buffer, .buf = buffer,
.len = sizeof(buffer), .len = sizeof(buffer),
}; };
int n, m; int n, m;
in.value_type = i; in.u.ref.value_type = i;
switch (i) { switch (i) {
case REFTABLE_REF_DELETION: case REFTABLE_REF_DELETION:
break; break;
case REFTABLE_REF_VAL1: case REFTABLE_REF_VAL1:
in.value.val1 = reftable_malloc(GIT_SHA1_RAWSZ); in.u.ref.value.val1 = reftable_malloc(GIT_SHA1_RAWSZ);
set_hash(in.value.val1, 1); set_hash(in.u.ref.value.val1, 1);
break; break;
case REFTABLE_REF_VAL2: case REFTABLE_REF_VAL2:
in.value.val2.value = reftable_malloc(GIT_SHA1_RAWSZ); in.u.ref.value.val2.value =
set_hash(in.value.val2.value, 1);
in.value.val2.target_value =
reftable_malloc(GIT_SHA1_RAWSZ); reftable_malloc(GIT_SHA1_RAWSZ);
set_hash(in.value.val2.target_value, 2); set_hash(in.u.ref.value.val2.value, 1);
in.u.ref.value.val2.target_value =
reftable_malloc(GIT_SHA1_RAWSZ);
set_hash(in.u.ref.value.val2.target_value, 2);
break; break;
case REFTABLE_REF_SYMREF: case REFTABLE_REF_SYMREF:
in.value.symref = xstrdup("target"); in.u.ref.value.symref = xstrdup("target");
break; break;
} }
in.refname = xstrdup("refs/heads/master"); in.u.ref.refname = xstrdup("refs/heads/master");
reftable_record_from_ref(&rec, &in); test_copy(&in);
test_copy(&rec);
EXPECT(reftable_record_val_type(&rec) == i); EXPECT(reftable_record_val_type(&in) == i);
reftable_record_key(&rec, &key); reftable_record_key(&in, &key);
n = reftable_record_encode(&rec, dest, GIT_SHA1_RAWSZ); n = reftable_record_encode(&in, dest, GIT_SHA1_RAWSZ);
EXPECT(n > 0); EXPECT(n > 0);
/* decode into a non-zero reftable_record to test for leaks. */ /* decode into a non-zero reftable_record to test for leaks. */
m = reftable_record_decode(&out, key, i, dest, GIT_SHA1_RAWSZ);
reftable_record_from_ref(&rec_out, &out);
m = reftable_record_decode(&rec_out, key, i, dest,
GIT_SHA1_RAWSZ);
EXPECT(n == m); EXPECT(n == m);
EXPECT(reftable_ref_record_equal(&in, &out, GIT_SHA1_RAWSZ)); EXPECT(reftable_ref_record_equal(&in.u.ref, &out.u.ref,
reftable_record_release(&rec_out); GIT_SHA1_RAWSZ));
reftable_record_release(&in);
strbuf_release(&key); strbuf_release(&key);
reftable_ref_record_release(&in); reftable_record_release(&out);
} }
} }
@ -202,7 +202,7 @@ static void test_reftable_log_record_roundtrip(void)
set_test_hash(in[0].value.update.new_hash, 1); set_test_hash(in[0].value.update.new_hash, 1);
set_test_hash(in[0].value.update.old_hash, 2); set_test_hash(in[0].value.update.old_hash, 2);
for (i = 0; i < ARRAY_SIZE(in); i++) { for (i = 0; i < ARRAY_SIZE(in); i++) {
struct reftable_record rec = { NULL }; struct reftable_record rec = { .type = BLOCK_TYPE_LOG };
struct strbuf key = STRBUF_INIT; struct strbuf key = STRBUF_INIT;
uint8_t buffer[1024] = { 0 }; uint8_t buffer[1024] = { 0 };
struct string_view dest = { struct string_view dest = {
@ -210,23 +210,25 @@ static void test_reftable_log_record_roundtrip(void)
.len = sizeof(buffer), .len = sizeof(buffer),
}; };
/* populate out, to check for leaks. */ /* populate out, to check for leaks. */
struct reftable_log_record out = { struct reftable_record out = {
.refname = xstrdup("old name"), .type = BLOCK_TYPE_LOG,
.value_type = REFTABLE_LOG_UPDATE, .u.log = {
.value = { .refname = xstrdup("old name"),
.update = { .value_type = REFTABLE_LOG_UPDATE,
.new_hash = reftable_calloc(GIT_SHA1_RAWSZ), .value = {
.old_hash = reftable_calloc(GIT_SHA1_RAWSZ), .update = {
.name = xstrdup("old name"), .new_hash = reftable_calloc(GIT_SHA1_RAWSZ),
.email = xstrdup("old@email"), .old_hash = reftable_calloc(GIT_SHA1_RAWSZ),
.message = xstrdup("old message"), .name = xstrdup("old name"),
.email = xstrdup("old@email"),
.message = xstrdup("old message"),
},
}, },
}, },
}; };
struct reftable_record rec_out = { NULL };
int n, m, valtype; int n, m, valtype;
reftable_record_from_log(&rec, &in[i]); rec.u.log = in[i];
test_copy(&rec); test_copy(&rec);
@ -234,16 +236,16 @@ static void test_reftable_log_record_roundtrip(void)
n = reftable_record_encode(&rec, dest, GIT_SHA1_RAWSZ); n = reftable_record_encode(&rec, dest, GIT_SHA1_RAWSZ);
EXPECT(n >= 0); EXPECT(n >= 0);
reftable_record_from_log(&rec_out, &out);
valtype = reftable_record_val_type(&rec); valtype = reftable_record_val_type(&rec);
m = reftable_record_decode(&rec_out, key, valtype, dest, m = reftable_record_decode(&out, key, valtype, dest,
GIT_SHA1_RAWSZ); GIT_SHA1_RAWSZ);
EXPECT(n == m); EXPECT(n == m);
EXPECT(reftable_log_record_equal(&in[i], &out, GIT_SHA1_RAWSZ)); EXPECT(reftable_log_record_equal(&in[i], &out.u.log,
GIT_SHA1_RAWSZ));
reftable_log_record_release(&in[i]); reftable_log_record_release(&in[i]);
strbuf_release(&key); strbuf_release(&key);
reftable_record_release(&rec_out); reftable_record_release(&out);
} }
} }
@ -311,41 +313,43 @@ static void test_reftable_obj_record_roundtrip(void)
} }; } };
int i = 0; int i = 0;
for (i = 0; i < ARRAY_SIZE(recs); i++) { for (i = 0; i < ARRAY_SIZE(recs); i++) {
struct reftable_obj_record in = recs[i];
uint8_t buffer[1024] = { 0 }; uint8_t buffer[1024] = { 0 };
struct string_view dest = { struct string_view dest = {
.buf = buffer, .buf = buffer,
.len = sizeof(buffer), .len = sizeof(buffer),
}; };
struct reftable_record rec = { NULL }; struct reftable_record in = {
.type = BLOCK_TYPE_OBJ,
.u.obj = recs[i],
};
struct strbuf key = STRBUF_INIT; struct strbuf key = STRBUF_INIT;
struct reftable_obj_record out = { NULL }; struct reftable_record out = { .type = BLOCK_TYPE_OBJ };
struct reftable_record rec_out = { NULL };
int n, m; int n, m;
uint8_t extra; uint8_t extra;
reftable_record_from_obj(&rec, &in); test_copy(&in);
test_copy(&rec); reftable_record_key(&in, &key);
reftable_record_key(&rec, &key); n = reftable_record_encode(&in, dest, GIT_SHA1_RAWSZ);
n = reftable_record_encode(&rec, dest, GIT_SHA1_RAWSZ);
EXPECT(n > 0); EXPECT(n > 0);
extra = reftable_record_val_type(&rec); extra = reftable_record_val_type(&in);
reftable_record_from_obj(&rec_out, &out); m = reftable_record_decode(&out, key, extra, dest,
m = reftable_record_decode(&rec_out, key, extra, dest,
GIT_SHA1_RAWSZ); GIT_SHA1_RAWSZ);
EXPECT(n == m); EXPECT(n == m);
EXPECT(reftable_record_equal(&rec, &rec_out, GIT_SHA1_RAWSZ)); EXPECT(reftable_record_equal(&in, &out, GIT_SHA1_RAWSZ));
strbuf_release(&key); strbuf_release(&key);
reftable_record_release(&rec_out); reftable_record_release(&out);
} }
} }
static void test_reftable_index_record_roundtrip(void) static void test_reftable_index_record_roundtrip(void)
{ {
struct reftable_index_record in = { struct reftable_record in = {
.offset = 42, .type = BLOCK_TYPE_INDEX,
.last_key = STRBUF_INIT, .u.idx = {
.offset = 42,
.last_key = STRBUF_INIT,
},
}; };
uint8_t buffer[1024] = { 0 }; uint8_t buffer[1024] = { 0 };
struct string_view dest = { struct string_view dest = {
@ -353,31 +357,30 @@ static void test_reftable_index_record_roundtrip(void)
.len = sizeof(buffer), .len = sizeof(buffer),
}; };
struct strbuf key = STRBUF_INIT; struct strbuf key = STRBUF_INIT;
struct reftable_record rec = { NULL }; struct reftable_record out = {
struct reftable_index_record out = { .last_key = STRBUF_INIT }; .type = BLOCK_TYPE_INDEX,
struct reftable_record out_rec = { NULL }; .u.idx = { .last_key = STRBUF_INIT },
};
int n, m; int n, m;
uint8_t extra; uint8_t extra;
strbuf_addstr(&in.last_key, "refs/heads/master"); strbuf_addstr(&in.u.idx.last_key, "refs/heads/master");
reftable_record_from_index(&rec, &in); reftable_record_key(&in, &key);
reftable_record_key(&rec, &key); test_copy(&in);
test_copy(&rec);
EXPECT(0 == strbuf_cmp(&key, &in.last_key)); EXPECT(0 == strbuf_cmp(&key, &in.u.idx.last_key));
n = reftable_record_encode(&rec, dest, GIT_SHA1_RAWSZ); n = reftable_record_encode(&in, dest, GIT_SHA1_RAWSZ);
EXPECT(n > 0); EXPECT(n > 0);
extra = reftable_record_val_type(&rec); extra = reftable_record_val_type(&in);
reftable_record_from_index(&out_rec, &out); m = reftable_record_decode(&out, key, extra, dest, GIT_SHA1_RAWSZ);
m = reftable_record_decode(&out_rec, key, extra, dest, GIT_SHA1_RAWSZ);
EXPECT(m == n); EXPECT(m == n);
EXPECT(reftable_record_equal(&rec, &out_rec, GIT_SHA1_RAWSZ)); EXPECT(reftable_record_equal(&in, &out, GIT_SHA1_RAWSZ));
reftable_record_release(&out_rec); reftable_record_release(&out);
strbuf_release(&key); strbuf_release(&key);
strbuf_release(&in.last_key); strbuf_release(&in.u.idx.last_key);
} }
int record_test_main(int argc, const char *argv[]) int record_test_main(int argc, const char *argv[])

41
reftable/writer.c
View File

@ -253,8 +253,10 @@ done:
int reftable_writer_add_ref(struct reftable_writer *w, int reftable_writer_add_ref(struct reftable_writer *w,
struct reftable_ref_record *ref) struct reftable_ref_record *ref)
{ {
struct reftable_record rec = { NULL }; struct reftable_record rec = {
struct reftable_ref_record copy = *ref; .type = BLOCK_TYPE_REF,
.u.ref = *ref,
};
int err = 0; int err = 0;
if (ref->refname == NULL) if (ref->refname == NULL)
@ -263,8 +265,7 @@ int reftable_writer_add_ref(struct reftable_writer *w,
ref->update_index > w->max_update_index) ref->update_index > w->max_update_index)
return REFTABLE_API_ERROR; return REFTABLE_API_ERROR;
reftable_record_from_ref(&rec, &copy); rec.u.ref.update_index -= w->min_update_index;
copy.update_index -= w->min_update_index;
err = writer_add_record(w, &rec); err = writer_add_record(w, &rec);
if (err < 0) if (err < 0)
@ -303,7 +304,10 @@ int reftable_writer_add_refs(struct reftable_writer *w,
static int reftable_writer_add_log_verbatim(struct reftable_writer *w, static int reftable_writer_add_log_verbatim(struct reftable_writer *w,
struct reftable_log_record *log) struct reftable_log_record *log)
{ {
struct reftable_record rec = { NULL }; struct reftable_record rec = {
.type = BLOCK_TYPE_LOG,
.u.log = *log,
};
if (w->block_writer && if (w->block_writer &&
block_writer_type(w->block_writer) == BLOCK_TYPE_REF) { block_writer_type(w->block_writer) == BLOCK_TYPE_REF) {
int err = writer_finish_public_section(w); int err = writer_finish_public_section(w);
@ -313,8 +317,6 @@ static int reftable_writer_add_log_verbatim(struct reftable_writer *w,
w->next -= w->pending_padding; w->next -= w->pending_padding;
w->pending_padding = 0; w->pending_padding = 0;
reftable_record_from_log(&rec, log);
return writer_add_record(w, &rec); return writer_add_record(w, &rec);
} }
@ -395,8 +397,10 @@ static int writer_finish_section(struct reftable_writer *w)
w->index_len = 0; w->index_len = 0;
w->index_cap = 0; w->index_cap = 0;
for (i = 0; i < idx_len; i++) { for (i = 0; i < idx_len; i++) {
struct reftable_record rec = { NULL }; struct reftable_record rec = {
reftable_record_from_index(&rec, idx + i); .type = BLOCK_TYPE_INDEX,
.u.idx = idx[i],
};
if (block_writer_add(w->block_writer, &rec) == 0) { if (block_writer_add(w->block_writer, &rec) == 0) {
continue; continue;
} }
@ -464,17 +468,17 @@ static void write_object_record(void *void_arg, void *key)
{ {
struct write_record_arg *arg = void_arg; struct write_record_arg *arg = void_arg;
struct obj_index_tree_node *entry = key; struct obj_index_tree_node *entry = key;
struct reftable_obj_record obj_rec = { struct reftable_record
.hash_prefix = (uint8_t *)entry->hash.buf, rec = { .type = BLOCK_TYPE_OBJ,
.hash_prefix_len = arg->w->stats.object_id_len, .u.obj = {
.offsets = entry->offsets, .hash_prefix = (uint8_t *)entry->hash.buf,
.offset_len = entry->offset_len, .hash_prefix_len = arg->w->stats.object_id_len,
}; .offsets = entry->offsets,
struct reftable_record rec = { NULL }; .offset_len = entry->offset_len,
} };
if (arg->err < 0) if (arg->err < 0)
goto done; goto done;
reftable_record_from_obj(&rec, &obj_rec);
arg->err = block_writer_add(arg->w->block_writer, &rec); arg->err = block_writer_add(arg->w->block_writer, &rec);
if (arg->err == 0) if (arg->err == 0)
goto done; goto done;
@ -487,7 +491,8 @@ static void write_object_record(void *void_arg, void *key)
arg->err = block_writer_add(arg->w->block_writer, &rec); arg->err = block_writer_add(arg->w->block_writer, &rec);
if (arg->err == 0) if (arg->err == 0)
goto done; goto done;
obj_rec.offset_len = 0;
rec.u.obj.offset_len = 0;
arg->err = block_writer_add(arg->w->block_writer, &rec); arg->err = block_writer_add(arg->w->block_writer, &rec);
/* Should be able to write into a fresh block. */ /* Should be able to write into a fresh block. */