blob: a5058469e9b3a83fdd58a04f2af72690742d08bd [file] [log] [blame]
Junio C Hamano1b0c7172006-03-29 22:55:43 -08001#ifndef TREE_WALK_H
2#define TREE_WALK_H
3
brian m. carlsonea82b2a2019-01-15 00:39:44 +00004#include "cache.h"
Elijah Newrenef3ca952018-08-15 10:54:05 -07005
Jeff King5290d452020-02-01 06:39:22 -05006#define MAX_TRAVERSE_TREES 8
7
Heba Walybbcfa302019-11-17 21:04:57 +00008/**
9 * The tree walking API is used to traverse and inspect trees.
10 */
11
12/**
13 * An entry in a tree. Each entry has a sha1 identifier, pathname, and mode.
14 */
Junio C Hamano1b0c7172006-03-29 22:55:43 -080015struct name_entry {
brian m. carlsonea82b2a2019-01-15 00:39:44 +000016 struct object_id oid;
Junio C Hamano1b0c7172006-03-29 22:55:43 -080017 const char *path;
brian m. carlsonea82b2a2019-01-15 00:39:44 +000018 int pathlen;
Junio C Hamano1b0c7172006-03-29 22:55:43 -080019 unsigned int mode;
Junio C Hamano1b0c7172006-03-29 22:55:43 -080020};
21
Heba Walybbcfa302019-11-17 21:04:57 +000022/**
23 * A semi-opaque data structure used to maintain the current state of the walk.
24 */
Linus Torvalds4651ece2007-03-21 10:09:56 -070025struct tree_desc {
Heba Walybbcfa302019-11-17 21:04:57 +000026 /*
27 * pointer into the memory representation of the tree. It always
28 * points at the current entry being visited.
29 */
Linus Torvalds4651ece2007-03-21 10:09:56 -070030 const void *buffer;
Heba Walybbcfa302019-11-17 21:04:57 +000031
32 /* points to the current entry being visited. */
Linus Torvalds4651ece2007-03-21 10:09:56 -070033 struct name_entry entry;
Heba Walybbcfa302019-11-17 21:04:57 +000034
35 /* counts the number of bytes left in the `buffer`. */
Linus Torvalds4651ece2007-03-21 10:09:56 -070036 unsigned int size;
37};
38
Heba Walybbcfa302019-11-17 21:04:57 +000039/**
40 * Decode the entry currently being visited (the one pointed to by
41 * `tree_desc's` `entry` member) and return the sha1 of the entry. The
42 * `pathp` and `modep` arguments are set to the entry's pathname and mode
43 * respectively.
44 */
Elijah Newren5ec1e722019-04-05 08:00:12 -070045static inline const struct object_id *tree_entry_extract(struct tree_desc *desc, const char **pathp, unsigned short *modep)
Linus Torvalds4651ece2007-03-21 10:09:56 -070046{
47 *pathp = desc->entry.path;
Kirill Smelkov7146e662014-02-06 15:36:31 +040048 *modep = desc->entry.mode;
brian m. carlsonea82b2a2019-01-15 00:39:44 +000049 return &desc->entry.oid;
Linus Torvalds4651ece2007-03-21 10:09:56 -070050}
51
Heba Walybbcfa302019-11-17 21:04:57 +000052/**
53 * Calculate the length of a tree entry's pathname. This utilizes the
54 * memory structure of a tree entry to avoid the overhead of using a
55 * generic strlen().
56 */
Nguyễn Thái Ngọc Duy0de16332011-10-24 17:36:09 +110057static inline int tree_entry_len(const struct name_entry *ne)
Linus Torvalds304de2d2007-03-17 20:06:24 -070058{
brian m. carlsonea82b2a2019-01-15 00:39:44 +000059 return ne->pathlen;
Linus Torvalds304de2d2007-03-17 20:06:24 -070060}
61
David Turner8354fa32016-09-27 16:59:51 -040062/*
63 * The _gently versions of these functions warn and return false on a
64 * corrupt tree entry rather than dying,
65 */
66
Heba Walybbcfa302019-11-17 21:04:57 +000067/**
68 * Walk to the next entry in a tree. This is commonly used in conjunction
69 * with `tree_entry_extract` to inspect the current entry.
70 */
Junio C Hamano1b0c7172006-03-29 22:55:43 -080071void update_tree_entry(struct tree_desc *);
Heba Walybbcfa302019-11-17 21:04:57 +000072
David Turner8354fa32016-09-27 16:59:51 -040073int update_tree_entry_gently(struct tree_desc *);
Heba Walybbcfa302019-11-17 21:04:57 +000074
75/**
76 * Initialize a `tree_desc` and decode its first entry. The buffer and
77 * size parameters are assumed to be the same as the buffer and size
78 * members of `struct tree`.
79 */
Linus Torvalds6fda5e52007-03-21 10:08:25 -070080void init_tree_desc(struct tree_desc *desc, const void *buf, unsigned long size);
Heba Walybbcfa302019-11-17 21:04:57 +000081
David Turner8354fa32016-09-27 16:59:51 -040082int init_tree_desc_gently(struct tree_desc *desc, const void *buf, unsigned long size);
Junio C Hamano1b0c7172006-03-29 22:55:43 -080083
Elijah Newren2244eab2010-08-24 20:53:11 -060084/*
Heba Walybbcfa302019-11-17 21:04:57 +000085 * Visit the next entry in a tree. Returns 1 when there are more entries
86 * left to visit and 0 when all entries have been visited. This is
87 * commonly used in the test of a while loop.
Elijah Newren2244eab2010-08-24 20:53:11 -060088 */
Linus Torvalds4c068a92006-05-30 09:45:45 -070089int tree_entry(struct tree_desc *, struct name_entry *);
Heba Walybbcfa302019-11-17 21:04:57 +000090
David Turner8354fa32016-09-27 16:59:51 -040091int tree_entry_gently(struct tree_desc *, struct name_entry *);
Linus Torvalds4c068a92006-05-30 09:45:45 -070092
Heba Walybbcfa302019-11-17 21:04:57 +000093/**
94 * Initialize a `tree_desc` and decode its first entry given the
95 * object ID of a tree. Returns the `buffer` member if the latter
96 * is a valid tree identifier and NULL otherwise.
97 */
Nguyễn Thái Ngọc Duy5e575802019-06-27 16:28:48 +070098void *fill_tree_descriptor(struct repository *r,
99 struct tree_desc *desc,
100 const struct object_id *oid);
Junio C Hamano1b0c7172006-03-29 22:55:43 -0800101
Linus Torvalds40d934d2008-03-05 18:59:29 -0800102struct traverse_info;
Linus Torvalds91e4f032008-03-05 20:06:18 -0800103typedef int (*traverse_callback_t)(int n, unsigned long mask, unsigned long dirmask, struct name_entry *entry, struct traverse_info *);
Heba Walybbcfa302019-11-17 21:04:57 +0000104
105/**
106 * Traverse `n` number of trees in parallel. The `fn` callback member of
107 * `traverse_info` is called once for each tree entry.
108 */
Nguyễn Thái Ngọc Duy67022e02018-11-18 17:47:57 +0100109int traverse_trees(struct index_state *istate, int n, struct tree_desc *t, struct traverse_info *info);
Junio C Hamano1b0c7172006-03-29 22:55:43 -0800110
Nguyễn Thái Ngọc Duy0dd1f0c2019-06-27 16:28:50 +0700111enum get_oid_result get_tree_entry_follow_symlinks(struct repository *r, struct object_id *tree_oid, const char *name, struct object_id *result, struct strbuf *result_path, unsigned short *mode);
David Turner275721c2015-05-20 13:03:38 -0400112
Heba Walybbcfa302019-11-17 21:04:57 +0000113/**
114 * A structure used to maintain the state of a traversal.
115 */
Linus Torvalds40d934d2008-03-05 18:59:29 -0800116struct traverse_info {
David Turnerd9c2bd52015-12-21 17:34:20 -0500117 const char *traverse_path;
Heba Walybbcfa302019-11-17 21:04:57 +0000118
119 /*
120 * points to the traverse_info which was used to descend into the
121 * current tree. If this is the top-level tree `prev` will point to
122 * a dummy traverse_info.
123 */
Linus Torvalds40d934d2008-03-05 18:59:29 -0800124 struct traverse_info *prev;
Heba Walybbcfa302019-11-17 21:04:57 +0000125
126 /* is the entry for the current tree (if the tree is a subtree). */
Jeff King90553842019-07-31 00:38:15 -0400127 const char *name;
Heba Walybbcfa302019-11-17 21:04:57 +0000128
Jeff King90553842019-07-31 00:38:15 -0400129 size_t namelen;
130 unsigned mode;
131
Heba Walybbcfa302019-11-17 21:04:57 +0000132 /* is the length of the full path for the current tree. */
Jeff King37806082019-07-31 00:38:18 -0400133 size_t pathlen;
Heba Walybbcfa302019-11-17 21:04:57 +0000134
Junio C Hamano2842c0f2011-08-29 12:26:05 -0700135 struct pathspec *pathspec;
Linus Torvalds40d934d2008-03-05 18:59:29 -0800136
Heba Walybbcfa302019-11-17 21:04:57 +0000137 /* can be used by callbacks to maintain directory-file conflicts. */
René Scharfe603d2492013-06-16 01:44:43 +0200138 unsigned long df_conflicts;
Heba Walybbcfa302019-11-17 21:04:57 +0000139
140 /* a callback called for each entry in the tree.
141 *
142 * The arguments passed to the traverse callback are as follows:
143 *
144 * - `n` counts the number of trees being traversed.
145 *
146 * - `mask` has its nth bit set if something exists in the nth entry.
147 *
148 * - `dirmask` has its nth bit set if the nth tree's entry is a directory.
149 *
150 * - `entry` is an array of size `n` where the nth entry is from the nth tree.
151 *
152 * - `info` maintains the state of the traversal.
153 *
154 * Returning a negative value will terminate the traversal. Otherwise the
155 * return value is treated as an update mask. If the nth bit is set the nth tree
156 * will be updated and if the bit is not set the nth tree entry will be the
157 * same in the next callback invocation.
158 */
Linus Torvalds40d934d2008-03-05 18:59:29 -0800159 traverse_callback_t fn;
Heba Walybbcfa302019-11-17 21:04:57 +0000160
161 /* can be anything the `fn` callback would want to use. */
Linus Torvalds40d934d2008-03-05 18:59:29 -0800162 void *data;
Heba Walybbcfa302019-11-17 21:04:57 +0000163
164 /* tells whether to stop at the first error or not. */
Matthieu Moye6c111b2010-08-11 10:38:07 +0200165 int show_all_errors;
Linus Torvalds40d934d2008-03-05 18:59:29 -0800166};
Junio C Hamano1b0c7172006-03-29 22:55:43 -0800167
Heba Walybbcfa302019-11-17 21:04:57 +0000168/**
169 * Find an entry in a tree given a pathname and the sha1 of a tree to
170 * search. Returns 0 if the entry is found and -1 otherwise. The third
171 * and fourth parameters are set to the entry's sha1 and mode respectively.
172 */
Nguyễn Thái Ngọc Duy50ddb082019-06-27 16:28:49 +0700173int get_tree_entry(struct repository *, const struct object_id *, const char *, struct object_id *, unsigned short *);
Heba Walybbcfa302019-11-17 21:04:57 +0000174
175/**
176 * Generate the full pathname of a tree entry based from the root of the
177 * traversal. For example, if the traversal has recursed into another
178 * tree named "bar" the pathname of an entry "baz" in the "bar"
179 * tree would be "bar/baz".
180 */
Jeff King5aa02f92019-07-31 00:38:25 -0400181char *make_traverse_path(char *path, size_t pathlen, const struct traverse_info *info,
Jeff King90553842019-07-31 00:38:15 -0400182 const char *name, size_t namelen);
Heba Walybbcfa302019-11-17 21:04:57 +0000183
184/**
185 * Convenience wrapper to `make_traverse_path` into a strbuf.
186 */
Jeff Kingc43ab062019-07-31 00:38:23 -0400187void strbuf_make_traverse_path(struct strbuf *out,
188 const struct traverse_info *info,
189 const char *name, size_t namelen);
Heba Walybbcfa302019-11-17 21:04:57 +0000190
191/**
192 * Initialize a `traverse_info` given the pathname of the tree to start
193 * traversing from.
194 */
Denton Liu55454422019-04-29 04:28:14 -0400195void setup_traverse_info(struct traverse_info *info, const char *base);
Linus Torvalds40d934d2008-03-05 18:59:29 -0800196
Heba Walybbcfa302019-11-17 21:04:57 +0000197/**
198 * Calculate the length of a pathname returned by `make_traverse_path`.
199 * This utilizes the memory structure of a tree entry to avoid the
200 * overhead of using a generic strlen().
201 */
Jeff Kingb3b3cbc2019-07-31 00:38:20 -0400202static inline size_t traverse_path_len(const struct traverse_info *info,
203 size_t namelen)
Linus Torvalds40d934d2008-03-05 18:59:29 -0800204{
Jeff Kingb3b3cbc2019-07-31 00:38:20 -0400205 return st_add(info->pathlen, namelen);
Linus Torvalds40d934d2008-03-05 18:59:29 -0800206}
Junio C Hamano4dcff632006-04-19 14:05:47 -0700207
Nguyễn Thái Ngọc Duyd688cf02011-10-24 17:36:10 +1100208/* in general, positive means "kind of interesting" */
209enum interesting {
210 all_entries_not_interesting = -1, /* no, and no subsequent entries will be either */
211 entry_not_interesting = 0,
212 entry_interesting = 1,
213 all_entries_interesting = 2 /* yes, and all subsequent entries will be */
214};
215
Nguyễn Thái Ngọc Duy67022e02018-11-18 17:47:57 +0100216enum interesting tree_entry_interesting(struct index_state *istate,
217 const struct name_entry *,
218 struct strbuf *, int,
219 const struct pathspec *ps);
Nguyễn Thái Ngọc Duy2c389fc2010-12-15 22:02:40 +0700220
Junio C Hamano1b0c7172006-03-29 22:55:43 -0800221#endif