]>
git.proxmox.com Git - libgit2.git/blob - src/odb.c
2 * Copyright (C) 2009-2012 the libgit2 contributors
4 * This file is part of libgit2, distributed under the GNU GPL v2 with
5 * a Linking Exception. For full terms see the included COPYING file.
10 #include "git2/object.h"
14 #include "delta-apply.h"
17 #include "git2/odb_backend.h"
20 #define GIT_ALTERNATES_FILE "info/alternates"
22 /* TODO: is this correct? */
23 #define GIT_LOOSE_PRIORITY 2
24 #define GIT_PACKED_PRIORITY 1
28 git_odb_backend
*backend
;
33 static int format_object_header(char *hdr
, size_t n
, size_t obj_len
, git_otype obj_type
)
35 const char *type_str
= git_object_type2string(obj_type
);
36 int len
= p_snprintf(hdr
, n
, "%s %"PRIuZ
, type_str
, obj_len
);
37 assert(len
> 0 && len
<= (int)n
);
41 int git_odb__hashobj(git_oid
*id
, git_rawobj
*obj
)
49 if (!git_object_typeisloose(obj
->type
))
51 if (!obj
->data
&& obj
->len
!= 0)
54 hdrlen
= format_object_header(header
, sizeof(header
), obj
->len
, obj
->type
);
58 vec
[1].data
= obj
->data
;
59 vec
[1].len
= obj
->len
;
61 git_hash_vec(id
, vec
, 2);
67 static git_odb_object
*new_odb_object(const git_oid
*oid
, git_rawobj
*source
)
69 git_odb_object
*object
= git__malloc(sizeof(git_odb_object
));
70 memset(object
, 0x0, sizeof(git_odb_object
));
72 git_oid_cpy(&object
->cached
.oid
, oid
);
73 memcpy(&object
->raw
, source
, sizeof(git_rawobj
));
78 static void free_odb_object(void *o
)
80 git_odb_object
*object
= (git_odb_object
*)o
;
83 git__free(object
->raw
.data
);
88 const git_oid
*git_odb_object_id(git_odb_object
*object
)
90 return &object
->cached
.oid
;
93 const void *git_odb_object_data(git_odb_object
*object
)
95 return object
->raw
.data
;
98 size_t git_odb_object_size(git_odb_object
*object
)
100 return object
->raw
.len
;
103 git_otype
git_odb_object_type(git_odb_object
*object
)
105 return object
->raw
.type
;
108 void git_odb_object_free(git_odb_object
*object
)
110 git_cached_obj_decref((git_cached_obj
*)object
, &free_odb_object
);
113 int git_odb__hashfd(git_oid
*out
, git_file fd
, size_t size
, git_otype type
)
116 char hdr
[64], buffer
[2048];
119 hdr_len
= format_object_header(hdr
, sizeof(hdr
), size
, type
);
121 ctx
= git_hash_new_ctx();
122 GITERR_CHECK_ALLOC(ctx
);
124 git_hash_update(ctx
, hdr
, hdr_len
);
127 ssize_t read_len
= p_read(fd
, buffer
, sizeof(buffer
));
130 git_hash_free_ctx(ctx
);
131 giterr_set(GITERR_OS
, "Error reading file");
135 git_hash_update(ctx
, buffer
, read_len
);
139 git_hash_final(out
, ctx
);
140 git_hash_free_ctx(ctx
);
145 int git_odb__hashfd_filtered(
146 git_oid
*out
, git_file fd
, size_t size
, git_otype type
, git_vector
*filters
)
149 git_buf raw
= GIT_BUF_INIT
;
150 git_buf filtered
= GIT_BUF_INIT
;
152 if (!filters
|| !filters
->length
)
153 return git_odb__hashfd(out
, fd
, size
, type
);
155 /* size of data is used in header, so we have to read the whole file
156 * into memory to apply filters before beginning to calculate the hash
159 if (!(error
= git_futils_readbuffer_fd(&raw
, fd
, size
)))
160 error
= git_filters_apply(&filtered
, &raw
, filters
);
165 error
= git_odb_hash(out
, filtered
.ptr
, filtered
.size
, type
);
167 git_buf_free(&filtered
);
172 int git_odb__hashlink(git_oid
*out
, const char *path
)
178 if (git_path_lstat(path
, &st
) < 0)
183 if (!git__is_sizet(size
)) {
184 giterr_set(GITERR_OS
, "File size overflow for 32-bit systems");
188 if (S_ISLNK(st
.st_mode
)) {
192 link_data
= git__malloc((size_t)size
);
193 GITERR_CHECK_ALLOC(link_data
);
195 read_len
= p_readlink(path
, link_data
, (size_t)(size
+ 1));
196 if (read_len
!= (ssize_t
)size
) {
197 giterr_set(GITERR_OS
, "Failed to read symlink data for '%s'", path
);
201 result
= git_odb_hash(out
, link_data
, (size_t)size
, GIT_OBJ_BLOB
);
202 git__free(link_data
);
204 int fd
= git_futils_open_ro(path
);
207 result
= git_odb__hashfd(out
, fd
, (size_t)size
, GIT_OBJ_BLOB
);
214 int git_odb_hashfile(git_oid
*out
, const char *path
, git_otype type
)
217 int result
, fd
= git_futils_open_ro(path
);
221 if ((size
= git_futils_filesize(fd
)) < 0 || !git__is_sizet(size
)) {
222 giterr_set(GITERR_OS
, "File size overflow for 32-bit systems");
227 result
= git_odb__hashfd(out
, fd
, (size_t)size
, type
);
232 int git_odb_hash(git_oid
*id
, const void *data
, size_t len
, git_otype type
)
238 raw
.data
= (void *)data
;
242 return git_odb__hashobj(id
, &raw
);
250 git_odb_stream stream
;
252 size_t size
, written
;
256 static int fake_wstream__fwrite(git_oid
*oid
, git_odb_stream
*_stream
)
258 fake_wstream
*stream
= (fake_wstream
*)_stream
;
259 return _stream
->backend
->write(oid
, _stream
->backend
, stream
->buffer
, stream
->size
, stream
->type
);
262 static int fake_wstream__write(git_odb_stream
*_stream
, const char *data
, size_t len
)
264 fake_wstream
*stream
= (fake_wstream
*)_stream
;
266 if (stream
->written
+ len
> stream
->size
)
269 memcpy(stream
->buffer
+ stream
->written
, data
, len
);
270 stream
->written
+= len
;
274 static void fake_wstream__free(git_odb_stream
*_stream
)
276 fake_wstream
*stream
= (fake_wstream
*)_stream
;
278 git__free(stream
->buffer
);
282 static int init_fake_wstream(git_odb_stream
**stream_p
, git_odb_backend
*backend
, size_t size
, git_otype type
)
284 fake_wstream
*stream
;
286 stream
= git__calloc(1, sizeof(fake_wstream
));
287 GITERR_CHECK_ALLOC(stream
);
291 stream
->buffer
= git__malloc(size
);
292 if (stream
->buffer
== NULL
) {
297 stream
->stream
.backend
= backend
;
298 stream
->stream
.read
= NULL
; /* read only */
299 stream
->stream
.write
= &fake_wstream__write
;
300 stream
->stream
.finalize_write
= &fake_wstream__fwrite
;
301 stream
->stream
.free
= &fake_wstream__free
;
302 stream
->stream
.mode
= GIT_STREAM_WRONLY
;
304 *stream_p
= (git_odb_stream
*)stream
;
308 /***********************************************************
310 * OBJECT DATABASE PUBLIC API
312 * Public calls for the ODB functionality
314 ***********************************************************/
316 static int backend_sort_cmp(const void *a
, const void *b
)
318 const backend_internal
*backend_a
= (const backend_internal
*)(a
);
319 const backend_internal
*backend_b
= (const backend_internal
*)(b
);
321 if (backend_a
->is_alternate
== backend_b
->is_alternate
)
322 return (backend_b
->priority
- backend_a
->priority
);
324 return backend_a
->is_alternate
? 1 : -1;
327 int git_odb_new(git_odb
**out
)
329 git_odb
*db
= git__calloc(1, sizeof(*db
));
330 GITERR_CHECK_ALLOC(db
);
332 if (git_cache_init(&db
->cache
, GIT_DEFAULT_CACHE_SIZE
, &free_odb_object
) < 0 ||
333 git_vector_init(&db
->backends
, 4, backend_sort_cmp
) < 0)
340 GIT_REFCOUNT_INC(db
);
344 static int add_backend_internal(git_odb
*odb
, git_odb_backend
*backend
, int priority
, int is_alternate
)
346 backend_internal
*internal
;
348 assert(odb
&& backend
);
350 /* Check if the backend is already owned by another ODB */
351 assert(!backend
->odb
|| backend
->odb
== odb
);
353 internal
= git__malloc(sizeof(backend_internal
));
354 GITERR_CHECK_ALLOC(internal
);
356 internal
->backend
= backend
;
357 internal
->priority
= priority
;
358 internal
->is_alternate
= is_alternate
;
360 if (git_vector_insert(&odb
->backends
, internal
) < 0) {
365 git_vector_sort(&odb
->backends
);
366 internal
->backend
->odb
= odb
;
370 int git_odb_add_backend(git_odb
*odb
, git_odb_backend
*backend
, int priority
)
372 return add_backend_internal(odb
, backend
, priority
, 0);
375 int git_odb_add_alternate(git_odb
*odb
, git_odb_backend
*backend
, int priority
)
377 return add_backend_internal(odb
, backend
, priority
, 1);
380 static int add_default_backends(git_odb
*db
, const char *objects_dir
, int as_alternates
)
382 git_odb_backend
*loose
, *packed
;
384 /* add the loose object backend */
385 if (git_odb_backend_loose(&loose
, objects_dir
, -1, 0) < 0 ||
386 add_backend_internal(db
, loose
, GIT_LOOSE_PRIORITY
, as_alternates
) < 0)
389 /* add the packed file backend */
390 if (git_odb_backend_pack(&packed
, objects_dir
) < 0 ||
391 add_backend_internal(db
, packed
, GIT_PACKED_PRIORITY
, as_alternates
) < 0)
397 static int load_alternates(git_odb
*odb
, const char *objects_dir
)
399 git_buf alternates_path
= GIT_BUF_INIT
;
400 git_buf alternates_buf
= GIT_BUF_INIT
;
402 const char *alternate
;
405 if (git_buf_joinpath(&alternates_path
, objects_dir
, GIT_ALTERNATES_FILE
) < 0)
408 if (git_path_exists(alternates_path
.ptr
) == false) {
409 git_buf_free(&alternates_path
);
413 if (git_futils_readbuffer(&alternates_buf
, alternates_path
.ptr
) < 0) {
414 git_buf_free(&alternates_path
);
418 buffer
= (char *)alternates_buf
.ptr
;
420 /* add each alternate as a new backend; one alternate per line */
421 while ((alternate
= git__strtok(&buffer
, "\r\n")) != NULL
) {
422 if (*alternate
== '\0' || *alternate
== '#')
425 /* relative path: build based on the current `objects` folder */
426 if (*alternate
== '.') {
427 if ((result
= git_buf_joinpath(&alternates_path
, objects_dir
, alternate
)) < 0)
429 alternate
= git_buf_cstr(&alternates_path
);
432 if ((result
= add_default_backends(odb
, alternate
, 1)) < 0)
436 git_buf_free(&alternates_path
);
437 git_buf_free(&alternates_buf
);
442 int git_odb_open(git_odb
**out
, const char *objects_dir
)
446 assert(out
&& objects_dir
);
450 if (git_odb_new(&db
) < 0)
453 if (add_default_backends(db
, objects_dir
, 0) < 0 ||
454 load_alternates(db
, objects_dir
) < 0)
464 static void odb_free(git_odb
*db
)
468 for (i
= 0; i
< db
->backends
.length
; ++i
) {
469 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
470 git_odb_backend
*backend
= internal
->backend
;
472 if (backend
->free
) backend
->free(backend
);
473 else git__free(backend
);
478 git_vector_free(&db
->backends
);
479 git_cache_free(&db
->cache
);
483 void git_odb_free(git_odb
*db
)
488 GIT_REFCOUNT_DEC(db
, odb_free
);
491 int git_odb_exists(git_odb
*db
, const git_oid
*id
)
493 git_odb_object
*object
;
499 if ((object
= git_cache_get(&db
->cache
, id
)) != NULL
) {
500 git_odb_object_free(object
);
504 for (i
= 0; i
< db
->backends
.length
&& !found
; ++i
) {
505 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
506 git_odb_backend
*b
= internal
->backend
;
508 if (b
->exists
!= NULL
)
509 found
= b
->exists(b
, id
);
515 int git_odb_read_header(size_t *len_p
, git_otype
*type_p
, git_odb
*db
, const git_oid
*id
)
518 int error
= GIT_ENOTFOUND
;
519 git_odb_object
*object
;
523 if ((object
= git_cache_get(&db
->cache
, id
)) != NULL
) {
524 *len_p
= object
->raw
.len
;
525 *type_p
= object
->raw
.type
;
526 git_odb_object_free(object
);
530 for (i
= 0; i
< db
->backends
.length
&& error
< 0; ++i
) {
531 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
532 git_odb_backend
*b
= internal
->backend
;
534 if (b
->read_header
!= NULL
)
535 error
= b
->read_header(len_p
, type_p
, b
, id
);
538 if (!error
|| error
== GIT_PASSTHROUGH
)
542 * no backend could read only the header.
543 * try reading the whole object and freeing the contents
545 if ((error
= git_odb_read(&object
, db
, id
)) < 0)
546 return error
; /* error already set - pass along */
548 *len_p
= object
->raw
.len
;
549 *type_p
= object
->raw
.type
;
550 git_odb_object_free(object
);
554 int git_odb_read(git_odb_object
**out
, git_odb
*db
, const git_oid
*id
)
557 int error
= GIT_ENOTFOUND
;
560 assert(out
&& db
&& id
);
562 *out
= git_cache_get(&db
->cache
, id
);
566 for (i
= 0; i
< db
->backends
.length
&& error
< 0; ++i
) {
567 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
568 git_odb_backend
*b
= internal
->backend
;
571 error
= b
->read(&raw
.data
, &raw
.len
, &raw
.type
, b
, id
);
574 /* TODO: If no backends are configured, this returns GIT_ENOTFOUND but
575 * will never have called giterr_set().
578 if (error
&& error
!= GIT_PASSTHROUGH
)
581 *out
= git_cache_try_store(&db
->cache
, new_odb_object(id
, &raw
));
585 int git_odb_read_prefix(
586 git_odb_object
**out
, git_odb
*db
, const git_oid
*short_id
, size_t len
)
589 int error
= GIT_ENOTFOUND
;
590 git_oid found_full_oid
= {{0}};
597 if (len
< GIT_OID_MINPREFIXLEN
)
598 return git_odb__error_ambiguous("prefix length too short");
600 if (len
> GIT_OID_HEXSZ
)
603 if (len
== GIT_OID_HEXSZ
) {
604 *out
= git_cache_get(&db
->cache
, short_id
);
609 for (i
= 0; i
< db
->backends
.length
; ++i
) {
610 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
611 git_odb_backend
*b
= internal
->backend
;
613 if (b
->read
!= NULL
) {
615 error
= b
->read_prefix(&full_oid
, &raw
.data
, &raw
.len
, &raw
.type
, b
, short_id
, len
);
616 if (error
== GIT_ENOTFOUND
|| error
== GIT_PASSTHROUGH
)
624 if (found
&& git_oid_cmp(&full_oid
, &found_full_oid
))
625 return git_odb__error_ambiguous("multiple matches for prefix");
626 found_full_oid
= full_oid
;
632 return git_odb__error_notfound("no match for prefix", short_id
);
634 *out
= git_cache_try_store(&db
->cache
, new_odb_object(&found_full_oid
, &raw
));
638 int git_odb_foreach(git_odb
*db
, int (*cb
)(git_oid
*oid
, void *data
), void *data
)
641 backend_internal
*internal
;
643 git_vector_foreach(&db
->backends
, i
, internal
) {
644 git_odb_backend
*b
= internal
->backend
;
645 int error
= b
->foreach(b
, cb
, data
);
654 git_oid
*oid
, git_odb
*db
, const void *data
, size_t len
, git_otype type
)
657 int error
= GIT_ERROR
;
658 git_odb_stream
*stream
;
662 for (i
= 0; i
< db
->backends
.length
&& error
< 0; ++i
) {
663 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
664 git_odb_backend
*b
= internal
->backend
;
666 /* we don't write in alternates! */
667 if (internal
->is_alternate
)
670 if (b
->write
!= NULL
)
671 error
= b
->write(oid
, b
, data
, len
, type
);
674 if (!error
|| error
== GIT_PASSTHROUGH
)
677 /* if no backends were able to write the object directly, we try a streaming
678 * write to the backends; just write the whole object into the stream in one
681 if ((error
= git_odb_open_wstream(&stream
, db
, len
, type
)) != 0)
684 stream
->write(stream
, data
, len
);
685 error
= stream
->finalize_write(oid
, stream
);
686 stream
->free(stream
);
691 int git_odb_open_wstream(
692 git_odb_stream
**stream
, git_odb
*db
, size_t size
, git_otype type
)
695 int error
= GIT_ERROR
;
697 assert(stream
&& db
);
699 for (i
= 0; i
< db
->backends
.length
&& error
< 0; ++i
) {
700 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
701 git_odb_backend
*b
= internal
->backend
;
703 /* we don't write in alternates! */
704 if (internal
->is_alternate
)
707 if (b
->writestream
!= NULL
)
708 error
= b
->writestream(stream
, b
, size
, type
);
709 else if (b
->write
!= NULL
)
710 error
= init_fake_wstream(stream
, b
, size
, type
);
713 if (error
== GIT_PASSTHROUGH
)
719 int git_odb_open_rstream(git_odb_stream
**stream
, git_odb
*db
, const git_oid
*oid
)
722 int error
= GIT_ERROR
;
724 assert(stream
&& db
);
726 for (i
= 0; i
< db
->backends
.length
&& error
< 0; ++i
) {
727 backend_internal
*internal
= git_vector_get(&db
->backends
, i
);
728 git_odb_backend
*b
= internal
->backend
;
730 if (b
->readstream
!= NULL
)
731 error
= b
->readstream(stream
, b
, oid
);
734 if (error
== GIT_PASSTHROUGH
)
740 void * git_odb_backend_malloc(git_odb_backend
*backend
, size_t len
)
743 return git__malloc(len
);
746 int git_odb__error_notfound(const char *message
, const git_oid
*oid
)
749 char oid_str
[GIT_OID_HEXSZ
+ 1];
750 git_oid_tostr(oid_str
, sizeof(oid_str
), oid
);
751 giterr_set(GITERR_ODB
, "Object not found - %s (%s)", message
, oid_str
);
753 giterr_set(GITERR_ODB
, "Object not found - %s", message
);
755 return GIT_ENOTFOUND
;
758 int git_odb__error_ambiguous(const char *message
)
760 giterr_set(GITERR_ODB
, "Ambiguous SHA1 prefix - %s", message
);
761 return GIT_EAMBIGUOUS
;