object-store.h

   1 #ifndef OBJECT_STORE_H
   2 #define OBJECT_STORE_H
   3
   4 #include "cache.h"
   5 #include "oidmap.h"
   6 #include "list.h"
   7 #include "sha1-array.h"
   8 #include "strbuf.h"
   9
  10 struct object_directory {
  11         struct object_directory *next;
  12
  13         /*
  14          * Used to store the results of readdir(3) calls when we are OK
  15          * sacrificing accuracy due to races for speed. That includes
  16          * object existence with OBJECT_INFO_QUICK, as well as
  17          * our search for unique abbreviated hashes. Don't use it for tasks
  18          * requiring greater accuracy!
  19          *
  20          * Be sure to call odb_load_loose_cache() before using.
  21          */
  22         char loose_objects_subdir_seen[256];
  23         struct oid_array loose_objects_cache[256];
  24
  25         /*
  26          * Path to the alternative object store. If this is a relative path,
  27          * it is relative to the current working directory.
  28          */
  29         char *path;
  30 };
  31
  32 void prepare_alt_odb(struct repository *r);
  33 char *compute_alternate_path(const char *path, struct strbuf *err);
  34 typedef int alt_odb_fn(struct object_directory *, void *);
  35 int foreach_alt_odb(alt_odb_fn, void*);
  36 typedef void alternate_ref_fn(const struct object_id *oid, void *);
  37 void for_each_alternate_ref(alternate_ref_fn, void *);
  38
  39 /*
  40  * Add the directory to the on-disk alternates file; the new entry will also
  41  * take effect in the current process.
  42  */
  43 void add_to_alternates_file(const char *dir);
  44
  45 /*
  46  * Add the directory to the in-memory list of alternates (along with any
  47  * recursive alternates it points to), but do not modify the on-disk alternates
  48  * file.
  49  */
  50 void add_to_alternates_memory(const char *dir);
  51
  52 /*
  53  * Populate and return the loose object cache array corresponding to the
  54  * given object ID.
  55  */
  56 struct oid_array *odb_loose_cache(struct object_directory *odb,
  57                                   const struct object_id *oid);
  58
  59 /* Empty the loose object cache for the specified object directory. */
  60 void odb_clear_loose_cache(struct object_directory *odb);
  61
  62 struct packed_git {
  63         struct hashmap_entry packmap_ent;
  64         struct packed_git *next;
  65         struct list_head mru;
  66         struct pack_window *windows;
  67         off_t pack_size;
  68         const void *index_data;
  69         size_t index_size;
  70         uint32_t num_objects;
  71         uint32_t num_bad_objects;
  72         unsigned char *bad_object_sha1;
  73         int index_version;
  74         time_t mtime;
  75         int pack_fd;
  76         int index;              /* for builtin/pack-objects.c */
  77         unsigned pack_local:1,
  78                  pack_keep:1,
  79                  pack_keep_in_core:1,
  80                  freshened:1,
  81                  do_not_close:1,
  82                  pack_promisor:1,
  83                  multi_pack_index:1;
  84         unsigned char hash[GIT_MAX_RAWSZ];
  85         struct revindex_entry *revindex;
  86         /* something like ".git/objects/pack/xxxxx.pack" */
  87         char pack_name[FLEX_ARRAY]; /* more */
  88 };
  89
  90 struct multi_pack_index;
  91
  92 static inline int pack_map_entry_cmp(const void *unused_cmp_data,
  93                                      const struct hashmap_entry *entry,
  94                                      const struct hashmap_entry *entry2,
  95                                      const void *keydata)
  96 {
  97         const char *key = keydata;
  98         const struct packed_git *pg1, *pg2;
  99
 100         pg1 = container_of(entry, const struct packed_git, packmap_ent);
 101         pg2 = container_of(entry2, const struct packed_git, packmap_ent);
 102
 103         return strcmp(pg1->pack_name, key ? key : pg2->pack_name);
 104 }
 105
 106 struct raw_object_store {
 107         /*
 108          * Set of all object directories; the main directory is first (and
 109          * cannot be NULL after initialization). Subsequent directories are
 110          * alternates.
 111          */
 112         struct object_directory *odb;
 113         struct object_directory **odb_tail;
 114         int loaded_alternates;
 115
 116         /*
 117          * A list of alternate object directories loaded from the environment;
 118          * this should not generally need to be accessed directly, but will
 119          * populate the "odb" list when prepare_alt_odb() is run.
 120          */
 121         char *alternate_db;
 122
 123         /*
 124          * Objects that should be substituted by other objects
 125          * (see git-replace(1)).
 126          */
 127         struct oidmap *replace_map;
 128
 129         struct commit_graph *commit_graph;
 130         unsigned commit_graph_attempted : 1; /* if loading has been attempted */
 131
 132         /*
 133          * private data
 134          *
 135          * should only be accessed directly by packfile.c and midx.c
 136          */
 137         struct multi_pack_index *multi_pack_index;
 138
 139         /*
 140          * private data
 141          *
 142          * should only be accessed directly by packfile.c
 143          */
 144
 145         struct packed_git *packed_git;
 146         /* A most-recently-used ordered version of the packed_git list. */
 147         struct list_head packed_git_mru;
 148
 149         /*
 150          * A map of packfiles to packed_git structs for tracking which
 151          * packs have been loaded already.
 152          */
 153         struct hashmap pack_map;
 154
 155         /*
 156          * A fast, rough count of the number of objects in the repository.
 157          * These two fields are not meant for direct access. Use
 158          * approximate_object_count() instead.
 159          */
 160         unsigned long approximate_object_count;
 161         unsigned approximate_object_count_valid : 1;
 162
 163         /*
 164          * Whether packed_git has already been populated with this repository's
 165          * packs.
 166          */
 167         unsigned packed_git_initialized : 1;
 168 };
 169
 170 struct raw_object_store *raw_object_store_new(void);
 171 void raw_object_store_clear(struct raw_object_store *o);
 172
 173 /*
 174  * Put in `buf` the name of the file in the local object database that
 175  * would be used to store a loose object with the specified oid.
 176  */
 177 const char *loose_object_path(struct repository *r, struct strbuf *buf,
 178                               const struct object_id *oid);
 179
 180 void *map_loose_object(struct repository *r, const struct object_id *oid,
 181                        unsigned long *size);
 182
 183 void *read_object_file_extended(struct repository *r,
 184                                 const struct object_id *oid,
 185                                 enum object_type *type,
 186                                 unsigned long *size, int lookup_replace);
 187 static inline void *repo_read_object_file(struct repository *r,
 188                                           const struct object_id *oid,
 189                                           enum object_type *type,
 190                                           unsigned long *size)
 191 {
 192         return read_object_file_extended(r, oid, type, size, 1);
 193 }
 194 #ifndef NO_THE_REPOSITORY_COMPATIBILITY_MACROS
 195 #define read_object_file(oid, type, size) repo_read_object_file(the_repository, oid, type, size)
 196 #endif
 197
 198 /* Read and unpack an object file into memory, write memory to an object file */
 199 int oid_object_info(struct repository *r, const struct object_id *, unsigned long *);
 200
 201 int hash_object_file(const void *buf, unsigned long len,
 202                      const char *type, struct object_id *oid);
 203
 204 int write_object_file(const void *buf, unsigned long len,
 205                       const char *type, struct object_id *oid);
 206
 207 int hash_object_file_literally(const void *buf, unsigned long len,
 208                                const char *type, struct object_id *oid,
 209                                unsigned flags);
 210
 211 /*
 212  * Add an object file to the in-memory object store, without writing it
 213  * to disk.
 214  *
 215  * Callers are responsible for calling write_object_file to record the
 216  * object in persistent storage before writing any other new objects
 217  * that reference it.
 218  */
 219 int pretend_object_file(void *, unsigned long, enum object_type,
 220                         struct object_id *oid);
 221
 222 int force_object_loose(const struct object_id *oid, time_t mtime);
 223
 224 /*
 225  * Open the loose object at path, check its hash, and return the contents,
 226  * type, and size. If the object is a blob, then "contents" may return NULL,
 227  * to allow streaming of large blobs.
 228  *
 229  * Returns 0 on success, negative on error (details may be written to stderr).
 230  */
 231 int read_loose_object(const char *path,
 232                       const struct object_id *expected_oid,
 233                       enum object_type *type,
 234                       unsigned long *size,
 235                       void **contents);
 236
 237 #ifndef NO_THE_REPOSITORY_COMPATIBILITY_MACROS
 238 #define has_sha1_file_with_flags(sha1, flags) repo_has_sha1_file_with_flags(the_repository, sha1, flags)
 239 #define has_sha1_file(sha1) repo_has_sha1_file(the_repository, sha1)
 240 #endif
 241
 242 /* Same as the above, except for struct object_id. */
 243 int repo_has_object_file(struct repository *r, const struct object_id *oid);
 244 int repo_has_object_file_with_flags(struct repository *r,
 245                                     const struct object_id *oid, int flags);
 246 #ifndef NO_THE_REPOSITORY_COMPATIBILITY_MACROS
 247 #define has_object_file(oid) repo_has_object_file(the_repository, oid)
 248 #define has_object_file_with_flags(oid, flags) repo_has_object_file_with_flags(the_repository, oid, flags)
 249 #endif
 250
 251 /*
 252  * Return true iff an alternate object database has a loose object
 253  * with the specified name.  This function does not respect replace
 254  * references.
 255  */
 256 int has_loose_object_nonlocal(const struct object_id *);
 257
 258 void assert_oid_type(const struct object_id *oid, enum object_type expect);
 259
 260 struct object_info {
 261         /* Request */
 262         enum object_type *typep;
 263         unsigned long *sizep;
 264         off_t *disk_sizep;
 265         unsigned char *delta_base_sha1;
 266         struct strbuf *type_name;
 267         void **contentp;
 268
 269         /* Response */
 270         enum {
 271                 OI_CACHED,
 272                 OI_LOOSE,
 273                 OI_PACKED,
 274                 OI_DBCACHED
 275         } whence;
 276         union {
 277                 /*
 278                  * struct {
 279                  *      ... Nothing to expose in this case
 280                  * } cached;
 281                  * struct {
 282                  *      ... Nothing to expose in this case
 283                  * } loose;
 284                  */
 285                 struct {
 286                         struct packed_git *pack;
 287                         off_t offset;
 288                         unsigned int is_delta;
 289                 } packed;
 290         } u;
 291 };
 292
 293 /*
 294  * Initializer for a "struct object_info" that wants no items. You may
 295  * also memset() the memory to all-zeroes.
 296  */
 297 #define OBJECT_INFO_INIT {NULL}
 298
 299 /* Invoke lookup_replace_object() on the given hash */
 300 #define OBJECT_INFO_LOOKUP_REPLACE 1
 301 /* Allow reading from a loose object file of unknown/bogus type */
 302 #define OBJECT_INFO_ALLOW_UNKNOWN_TYPE 2
 303 /* Do not retry packed storage after checking packed and loose storage */
 304 #define OBJECT_INFO_QUICK 8
 305 /* Do not check loose object */
 306 #define OBJECT_INFO_IGNORE_LOOSE 16
 307 /*
 308  * Do not attempt to fetch the object if missing (even if fetch_is_missing is
 309  * nonzero).
 310  */
 311 #define OBJECT_INFO_SKIP_FETCH_OBJECT 32
 312 /*
 313  * This is meant for bulk prefetching of missing blobs in a partial
 314  * clone. Implies OBJECT_INFO_SKIP_FETCH_OBJECT and OBJECT_INFO_QUICK
 315  */
 316 #define OBJECT_INFO_FOR_PREFETCH (OBJECT_INFO_SKIP_FETCH_OBJECT | OBJECT_INFO_QUICK)
 317
 318 int oid_object_info_extended(struct repository *r,
 319                              const struct object_id *,
 320                              struct object_info *, unsigned flags);
 321
 322 /*
 323  * Iterate over the files in the loose-object parts of the object
 324  * directory "path", triggering the following callbacks:
 325  *
 326  *  - loose_object is called for each loose object we find.
 327  *
 328  *  - loose_cruft is called for any files that do not appear to be
 329  *    loose objects. Note that we only look in the loose object
 330  *    directories "objects/[0-9a-f]{2}/", so we will not report
 331  *    "objects/foobar" as cruft.
 332  *
 333  *  - loose_subdir is called for each top-level hashed subdirectory
 334  *    of the object directory (e.g., "$OBJDIR/f0"). It is called
 335  *    after the objects in the directory are processed.
 336  *
 337  * Any callback that is NULL will be ignored. Callbacks returning non-zero
 338  * will end the iteration.
 339  *
 340  * In the "buf" variant, "path" is a strbuf which will also be used as a
 341  * scratch buffer, but restored to its original contents before
 342  * the function returns.
 343  */
 344 typedef int each_loose_object_fn(const struct object_id *oid,
 345                                  const char *path,
 346                                  void *data);
 347 typedef int each_loose_cruft_fn(const char *basename,
 348                                 const char *path,
 349                                 void *data);
 350 typedef int each_loose_subdir_fn(unsigned int nr,
 351                                  const char *path,
 352                                  void *data);
 353 int for_each_file_in_obj_subdir(unsigned int subdir_nr,
 354                                 struct strbuf *path,
 355                                 each_loose_object_fn obj_cb,
 356                                 each_loose_cruft_fn cruft_cb,
 357                                 each_loose_subdir_fn subdir_cb,
 358                                 void *data);
 359 int for_each_loose_file_in_objdir(const char *path,
 360                                   each_loose_object_fn obj_cb,
 361                                   each_loose_cruft_fn cruft_cb,
 362                                   each_loose_subdir_fn subdir_cb,
 363                                   void *data);
 364 int for_each_loose_file_in_objdir_buf(struct strbuf *path,
 365                                       each_loose_object_fn obj_cb,
 366                                       each_loose_cruft_fn cruft_cb,
 367                                       each_loose_subdir_fn subdir_cb,
 368                                       void *data);
 369
 370 /* Flags for for_each_*_object() below. */
 371 enum for_each_object_flags {
 372         /* Iterate only over local objects, not alternates. */
 373         FOR_EACH_OBJECT_LOCAL_ONLY = (1<<0),
 374
 375         /* Only iterate over packs obtained from the promisor remote. */
 376         FOR_EACH_OBJECT_PROMISOR_ONLY = (1<<1),
 377
 378         /*
 379          * Visit objects within a pack in packfile order rather than .idx order
 380          */
 381         FOR_EACH_OBJECT_PACK_ORDER = (1<<2),
 382 };
 383
 384 /*
 385  * Iterate over all accessible loose objects without respect to
 386  * reachability. By default, this includes both local and alternate objects.
 387  * The order in which objects are visited is unspecified.
 388  *
 389  * Any flags specific to packs are ignored.
 390  */
 391 int for_each_loose_object(each_loose_object_fn, void *,
 392                           enum for_each_object_flags flags);
 393
 394 /*
 395  * Iterate over all accessible packed objects without respect to reachability.
 396  * By default, this includes both local and alternate packs.
 397  *
 398  * Note that some objects may appear twice if they are found in multiple packs.
 399  * Each pack is visited in an unspecified order. By default, objects within a
 400  * pack are visited in pack-idx order (i.e., sorted by oid).
 401  */
 402 typedef int each_packed_object_fn(const struct object_id *oid,
 403                                   struct packed_git *pack,
 404                                   uint32_t pos,
 405                                   void *data);
 406 int for_each_object_in_pack(struct packed_git *p,
 407                             each_packed_object_fn, void *data,
 408                             enum for_each_object_flags flags);
 409 int for_each_packed_object(each_packed_object_fn, void *,
 410                            enum for_each_object_flags flags);
 411
 412 #endif /* OBJECT_STORE_H */