10 #include "list-objects.h"
11 #include "list-objects-filter.h"
12 #include "list-objects-filter-options.h"
15 #include "object-store.h"
17 /* Remember to update object flag allocation in object.h */
19 * FILTER_SHOWN_BUT_REVISIT -- we set this bit on tree objects
20 * that have been shown, but should be revisited if they appear
21 * in the traversal (until we mark it SEEN). This is a way to
22 * let us silently de-dup calls to show() in the caller. This
23 * is subtly different from the "revision.h:SHOWN" and the
24 * "object-name.c:ONELINE_SEEN" bits. And also different from
25 * the non-de-dup usage in pack-bitmap.c
27 #define FILTER_SHOWN_BUT_REVISIT (1<<21)
30 struct filter
*filter
;
33 struct object_id skip_tree
;
34 unsigned is_skipping_tree
: 1;
38 enum list_objects_filter_result (*filter_object_fn
)(
40 enum list_objects_filter_situation filter_situation
,
48 * Optional. If this function is supplied and the filter needs
49 * to collect omits, then this function is called once before
52 * This is required because the following two conditions hold:
54 * a. A tree filter can add and remove objects as an object
56 * b. A combine filter's omit set is the union of all its
57 * subfilters, which may include tree: filters.
59 * As such, the omits sets must be separate sets, and can only
60 * be unioned after the traversal is completed.
62 void (*finalize_omits_fn
)(struct oidset
*omits
, void *filter_data
);
64 void (*free_fn
)(void *filter_data
);
68 /* If non-NULL, the filter collects a list of the omitted OIDs here. */
72 static enum list_objects_filter_result
filter_blobs_none(
74 enum list_objects_filter_situation filter_situation
,
81 switch (filter_situation
) {
83 BUG("unknown filter_situation: %d", filter_situation
);
86 assert(obj
->type
== OBJ_TAG
);
87 /* always include all tag objects */
88 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
91 assert(obj
->type
== OBJ_COMMIT
);
92 /* always include all commit objects */
93 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
96 assert(obj
->type
== OBJ_TREE
);
97 /* always include all tree objects */
98 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
101 assert(obj
->type
== OBJ_TREE
);
105 assert(obj
->type
== OBJ_BLOB
);
106 assert((obj
->flags
& SEEN
) == 0);
109 oidset_insert(omits
, &obj
->oid
);
110 return LOFR_MARK_SEEN
; /* but not LOFR_DO_SHOW (hard omit) */
114 static void filter_blobs_none__init(
115 struct list_objects_filter_options
*filter_options
,
116 struct filter
*filter
)
118 filter
->filter_object_fn
= filter_blobs_none
;
119 filter
->free_fn
= free
;
123 * A filter for list-objects to omit ALL trees and blobs from the traversal.
124 * Can OPTIONALLY collect a list of the omitted OIDs.
126 struct filter_trees_depth_data
{
128 * Maps trees to the minimum depth at which they were seen. It is not
129 * necessary to re-traverse a tree at deeper or equal depths than it has
130 * already been traversed.
132 * We can't use LOFR_MARK_SEEN for tree objects since this will prevent
133 * it from being traversed at shallower depths.
135 struct oidmap seen_at_depth
;
137 unsigned long exclude_depth
;
138 unsigned long current_depth
;
141 struct seen_map_entry
{
142 struct oidmap_entry base
;
146 /* Returns 1 if the oid was in the omits set before it was invoked. */
147 static int filter_trees_update_omits(
149 struct oidset
*omits
,
156 return oidset_remove(omits
, &obj
->oid
);
158 return oidset_insert(omits
, &obj
->oid
);
161 static enum list_objects_filter_result
filter_trees_depth(
162 struct repository
*r
,
163 enum list_objects_filter_situation filter_situation
,
165 const char *pathname
,
166 const char *filename
,
167 struct oidset
*omits
,
170 struct filter_trees_depth_data
*filter_data
= filter_data_
;
171 struct seen_map_entry
*seen_info
;
172 int include_it
= filter_data
->current_depth
<
173 filter_data
->exclude_depth
;
178 * Note that we do not use _MARK_SEEN in order to allow re-traversal in
179 * case we encounter a tree or blob again at a shallower depth.
182 switch (filter_situation
) {
184 BUG("unknown filter_situation: %d", filter_situation
);
187 assert(obj
->type
== OBJ_TAG
);
188 /* always include all tag objects */
189 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
192 assert(obj
->type
== OBJ_COMMIT
);
193 /* always include all commit objects */
194 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
197 assert(obj
->type
== OBJ_TREE
);
198 filter_data
->current_depth
--;
202 filter_trees_update_omits(obj
, omits
, include_it
);
203 return include_it
? LOFR_MARK_SEEN
| LOFR_DO_SHOW
: LOFR_ZERO
;
205 case LOFS_BEGIN_TREE
:
206 seen_info
= oidmap_get(
207 &filter_data
->seen_at_depth
, &obj
->oid
);
209 CALLOC_ARRAY(seen_info
, 1);
210 oidcpy(&seen_info
->base
.oid
, &obj
->oid
);
211 seen_info
->depth
= filter_data
->current_depth
;
212 oidmap_put(&filter_data
->seen_at_depth
, seen_info
);
216 filter_data
->current_depth
>= seen_info
->depth
;
220 filter_res
= LOFR_SKIP_TREE
;
222 int been_omitted
= filter_trees_update_omits(
223 obj
, omits
, include_it
);
224 seen_info
->depth
= filter_data
->current_depth
;
227 filter_res
= LOFR_DO_SHOW
;
228 else if (omits
&& !been_omitted
)
230 * Must update omit information of children
231 * recursively; they have not been omitted yet.
233 filter_res
= LOFR_ZERO
;
235 filter_res
= LOFR_SKIP_TREE
;
238 filter_data
->current_depth
++;
243 static void filter_trees_free(void *filter_data
) {
244 struct filter_trees_depth_data
*d
= filter_data
;
247 oidmap_free(&d
->seen_at_depth
, 1);
251 static void filter_trees_depth__init(
252 struct list_objects_filter_options
*filter_options
,
253 struct filter
*filter
)
255 struct filter_trees_depth_data
*d
= xcalloc(1, sizeof(*d
));
256 oidmap_init(&d
->seen_at_depth
, 0);
257 d
->exclude_depth
= filter_options
->tree_exclude_depth
;
258 d
->current_depth
= 0;
260 filter
->filter_data
= d
;
261 filter
->filter_object_fn
= filter_trees_depth
;
262 filter
->free_fn
= filter_trees_free
;
266 * A filter for list-objects to omit large blobs.
267 * And to OPTIONALLY collect a list of the omitted OIDs.
269 struct filter_blobs_limit_data
{
270 unsigned long max_bytes
;
273 static enum list_objects_filter_result
filter_blobs_limit(
274 struct repository
*r
,
275 enum list_objects_filter_situation filter_situation
,
277 const char *pathname
,
278 const char *filename
,
279 struct oidset
*omits
,
282 struct filter_blobs_limit_data
*filter_data
= filter_data_
;
283 unsigned long object_length
;
286 switch (filter_situation
) {
288 BUG("unknown filter_situation: %d", filter_situation
);
291 assert(obj
->type
== OBJ_TAG
);
292 /* always include all tag objects */
293 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
296 assert(obj
->type
== OBJ_COMMIT
);
297 /* always include all commit objects */
298 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
300 case LOFS_BEGIN_TREE
:
301 assert(obj
->type
== OBJ_TREE
);
302 /* always include all tree objects */
303 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
306 assert(obj
->type
== OBJ_TREE
);
310 assert(obj
->type
== OBJ_BLOB
);
311 assert((obj
->flags
& SEEN
) == 0);
313 t
= oid_object_info(r
, &obj
->oid
, &object_length
);
314 if (t
!= OBJ_BLOB
) { /* probably OBJ_NONE */
316 * We DO NOT have the blob locally, so we cannot
317 * apply the size filter criteria. Be conservative
318 * and force show it (and let the caller deal with
324 if (object_length
< filter_data
->max_bytes
)
328 oidset_insert(omits
, &obj
->oid
);
329 return LOFR_MARK_SEEN
; /* but not LOFR_DO_SHOW (hard omit) */
334 oidset_remove(omits
, &obj
->oid
);
335 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
338 static void filter_blobs_limit__init(
339 struct list_objects_filter_options
*filter_options
,
340 struct filter
*filter
)
342 struct filter_blobs_limit_data
*d
= xcalloc(1, sizeof(*d
));
343 d
->max_bytes
= filter_options
->blob_limit_value
;
345 filter
->filter_data
= d
;
346 filter
->filter_object_fn
= filter_blobs_limit
;
347 filter
->free_fn
= free
;
351 * A filter driven by a sparse-checkout specification to only
352 * include blobs that a sparse checkout would populate.
354 * The sparse-checkout spec can be loaded from a blob with the
355 * given OID or from a local pathname. We allow an OID because
356 * the repo may be bare or we may be doing the filtering on the
361 * default_match is the usual default include/exclude value that
362 * should be inherited as we recurse into directories based
363 * upon pattern matching of the directory itself or of a
364 * containing directory.
366 enum pattern_match_result default_match
;
369 * 1 if the directory (recursively) contains any provisionally
372 * 0 if everything (recursively) contained in this directory
373 * has been explicitly included (SHOWN) in the result and
374 * the directory may be short-cut later in the traversal.
376 unsigned child_prov_omit
: 1;
379 struct filter_sparse_data
{
380 struct pattern_list pl
;
383 struct frame
*array_frame
;
386 static enum list_objects_filter_result
filter_sparse(
387 struct repository
*r
,
388 enum list_objects_filter_situation filter_situation
,
390 const char *pathname
,
391 const char *filename
,
392 struct oidset
*omits
,
395 struct filter_sparse_data
*filter_data
= filter_data_
;
398 enum pattern_match_result match
;
400 switch (filter_situation
) {
402 BUG("unknown filter_situation: %d", filter_situation
);
405 assert(obj
->type
== OBJ_TAG
);
406 /* always include all tag objects */
407 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
410 assert(obj
->type
== OBJ_COMMIT
);
411 /* always include all commit objects */
412 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
414 case LOFS_BEGIN_TREE
:
415 assert(obj
->type
== OBJ_TREE
);
417 match
= path_matches_pattern_list(pathname
, strlen(pathname
),
418 filename
, &dtype
, &filter_data
->pl
,
420 if (match
== UNDECIDED
)
421 match
= filter_data
->array_frame
[filter_data
->nr
- 1].default_match
;
423 ALLOC_GROW(filter_data
->array_frame
, filter_data
->nr
+ 1,
425 filter_data
->array_frame
[filter_data
->nr
].default_match
= match
;
426 filter_data
->array_frame
[filter_data
->nr
].child_prov_omit
= 0;
430 * A directory with this tree OID may appear in multiple
431 * places in the tree. (Think of a directory move or copy,
432 * with no other changes, so the OID is the same, but the
433 * full pathnames of objects within this directory are new
434 * and may match is_excluded() patterns differently.)
435 * So we cannot mark this directory as SEEN (yet), since
436 * that will prevent process_tree() from revisiting this
437 * tree object with other pathname prefixes.
439 * Only _DO_SHOW the tree object the first time we visit
442 * We always show all tree objects. A future optimization
443 * may want to attempt to narrow this.
445 if (obj
->flags
& FILTER_SHOWN_BUT_REVISIT
)
447 obj
->flags
|= FILTER_SHOWN_BUT_REVISIT
;
451 assert(obj
->type
== OBJ_TREE
);
452 assert(filter_data
->nr
> 1);
454 frame
= &filter_data
->array_frame
[--filter_data
->nr
];
457 * Tell our parent directory if any of our children were
458 * provisionally omitted.
460 filter_data
->array_frame
[filter_data
->nr
- 1].child_prov_omit
|=
461 frame
->child_prov_omit
;
464 * If there are NO provisionally omitted child objects (ALL child
465 * objects in this folder were INCLUDED), then we can mark the
466 * folder as SEEN (so we will not have to revisit it again).
468 if (!frame
->child_prov_omit
)
469 return LOFR_MARK_SEEN
;
473 assert(obj
->type
== OBJ_BLOB
);
474 assert((obj
->flags
& SEEN
) == 0);
476 frame
= &filter_data
->array_frame
[filter_data
->nr
- 1];
479 match
= path_matches_pattern_list(pathname
, strlen(pathname
),
480 filename
, &dtype
, &filter_data
->pl
,
482 if (match
== UNDECIDED
)
483 match
= frame
->default_match
;
484 if (match
== MATCHED
) {
486 oidset_remove(omits
, &obj
->oid
);
487 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
491 * Provisionally omit it. We've already established that
492 * this pathname is not in the sparse-checkout specification
493 * with the CURRENT pathname, so we *WANT* to omit this blob.
495 * However, a pathname elsewhere in the tree may also
496 * reference this same blob, so we cannot reject it yet.
497 * Leave the LOFR_ bits unset so that if the blob appears
498 * again in the traversal, we will be asked again.
501 oidset_insert(omits
, &obj
->oid
);
504 * Remember that at least 1 blob in this tree was
505 * provisionally omitted. This prevents us from short
506 * cutting the tree in future iterations.
508 frame
->child_prov_omit
= 1;
514 static void filter_sparse_free(void *filter_data
)
516 struct filter_sparse_data
*d
= filter_data
;
517 free(d
->array_frame
);
521 static void filter_sparse_oid__init(
522 struct list_objects_filter_options
*filter_options
,
523 struct filter
*filter
)
525 struct filter_sparse_data
*d
= xcalloc(1, sizeof(*d
));
526 struct object_context oc
;
527 struct object_id sparse_oid
;
529 if (get_oid_with_context(the_repository
,
530 filter_options
->sparse_oid_name
,
531 GET_OID_BLOB
, &sparse_oid
, &oc
))
532 die(_("unable to access sparse blob in '%s'"),
533 filter_options
->sparse_oid_name
);
534 if (add_patterns_from_blob_to_list(&sparse_oid
, "", 0, &d
->pl
) < 0)
535 die(_("unable to parse sparse filter data in %s"),
536 oid_to_hex(&sparse_oid
));
538 ALLOC_GROW(d
->array_frame
, d
->nr
+ 1, d
->alloc
);
539 d
->array_frame
[d
->nr
].default_match
= 0; /* default to include */
540 d
->array_frame
[d
->nr
].child_prov_omit
= 0;
543 filter
->filter_data
= d
;
544 filter
->filter_object_fn
= filter_sparse
;
545 filter
->free_fn
= filter_sparse_free
;
549 * A filter for list-objects to omit large blobs.
550 * And to OPTIONALLY collect a list of the omitted OIDs.
552 struct filter_object_type_data
{
553 enum object_type object_type
;
556 static enum list_objects_filter_result
filter_object_type(
557 struct repository
*r
,
558 enum list_objects_filter_situation filter_situation
,
560 const char *pathname
,
561 const char *filename
,
562 struct oidset
*omits
,
565 struct filter_object_type_data
*filter_data
= filter_data_
;
567 switch (filter_situation
) {
569 BUG("unknown filter_situation: %d", filter_situation
);
572 assert(obj
->type
== OBJ_TAG
);
573 if (filter_data
->object_type
== OBJ_TAG
)
574 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
575 return LOFR_MARK_SEEN
;
578 assert(obj
->type
== OBJ_COMMIT
);
579 if (filter_data
->object_type
== OBJ_COMMIT
)
580 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
581 return LOFR_MARK_SEEN
;
583 case LOFS_BEGIN_TREE
:
584 assert(obj
->type
== OBJ_TREE
);
587 * If we only want to show commits or tags, then there is no
588 * need to walk down trees.
590 if (filter_data
->object_type
== OBJ_COMMIT
||
591 filter_data
->object_type
== OBJ_TAG
)
592 return LOFR_SKIP_TREE
;
594 if (filter_data
->object_type
== OBJ_TREE
)
595 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
597 return LOFR_MARK_SEEN
;
600 assert(obj
->type
== OBJ_BLOB
);
602 if (filter_data
->object_type
== OBJ_BLOB
)
603 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
604 return LOFR_MARK_SEEN
;
611 static void filter_object_type__init(
612 struct list_objects_filter_options
*filter_options
,
613 struct filter
*filter
)
615 struct filter_object_type_data
*d
= xcalloc(1, sizeof(*d
));
616 d
->object_type
= filter_options
->object_type
;
618 filter
->filter_data
= d
;
619 filter
->filter_object_fn
= filter_object_type
;
620 filter
->free_fn
= free
;
623 /* A filter which only shows objects shown by all sub-filters. */
624 struct combine_filter_data
{
625 struct subfilter
*sub
;
629 static enum list_objects_filter_result
process_subfilter(
630 struct repository
*r
,
631 enum list_objects_filter_situation filter_situation
,
633 const char *pathname
,
634 const char *filename
,
635 struct subfilter
*sub
)
637 enum list_objects_filter_result result
;
640 * Check and update is_skipping_tree before oidset_contains so
641 * that is_skipping_tree gets unset even when the object is
642 * marked as seen. As of this writing, no filter uses
643 * LOFR_MARK_SEEN on trees that also uses LOFR_SKIP_TREE, so the
644 * ordering is only theoretically important. Be cautious if you
645 * change the order of the below checks and more filters have
648 if (sub
->is_skipping_tree
) {
649 if (filter_situation
== LOFS_END_TREE
&&
650 oideq(&obj
->oid
, &sub
->skip_tree
))
651 sub
->is_skipping_tree
= 0;
655 if (oidset_contains(&sub
->seen
, &obj
->oid
))
658 result
= list_objects_filter__filter_object(
659 r
, filter_situation
, obj
, pathname
, filename
, sub
->filter
);
661 if (result
& LOFR_MARK_SEEN
)
662 oidset_insert(&sub
->seen
, &obj
->oid
);
664 if (result
& LOFR_SKIP_TREE
) {
665 sub
->is_skipping_tree
= 1;
666 sub
->skip_tree
= obj
->oid
;
672 static enum list_objects_filter_result
filter_combine(
673 struct repository
*r
,
674 enum list_objects_filter_situation filter_situation
,
676 const char *pathname
,
677 const char *filename
,
678 struct oidset
*omits
,
681 struct combine_filter_data
*d
= filter_data
;
682 enum list_objects_filter_result combined_result
=
683 LOFR_DO_SHOW
| LOFR_MARK_SEEN
| LOFR_SKIP_TREE
;
686 for (sub
= 0; sub
< d
->nr
; sub
++) {
687 enum list_objects_filter_result sub_result
= process_subfilter(
688 r
, filter_situation
, obj
, pathname
, filename
,
690 if (!(sub_result
& LOFR_DO_SHOW
))
691 combined_result
&= ~LOFR_DO_SHOW
;
692 if (!(sub_result
& LOFR_MARK_SEEN
))
693 combined_result
&= ~LOFR_MARK_SEEN
;
694 if (!d
->sub
[sub
].is_skipping_tree
)
695 combined_result
&= ~LOFR_SKIP_TREE
;
698 return combined_result
;
701 static void filter_combine__free(void *filter_data
)
703 struct combine_filter_data
*d
= filter_data
;
705 for (sub
= 0; sub
< d
->nr
; sub
++) {
706 list_objects_filter__free(d
->sub
[sub
].filter
);
707 oidset_clear(&d
->sub
[sub
].seen
);
708 if (d
->sub
[sub
].omits
.set
.size
)
709 BUG("expected oidset to be cleared already");
715 static void add_all(struct oidset
*dest
, struct oidset
*src
) {
716 struct oidset_iter iter
;
717 struct object_id
*src_oid
;
719 oidset_iter_init(src
, &iter
);
720 while ((src_oid
= oidset_iter_next(&iter
)) != NULL
)
721 oidset_insert(dest
, src_oid
);
724 static void filter_combine__finalize_omits(
725 struct oidset
*omits
,
728 struct combine_filter_data
*d
= filter_data
;
731 for (sub
= 0; sub
< d
->nr
; sub
++) {
732 add_all(omits
, &d
->sub
[sub
].omits
);
733 oidset_clear(&d
->sub
[sub
].omits
);
737 static void filter_combine__init(
738 struct list_objects_filter_options
*filter_options
,
739 struct filter
* filter
)
741 struct combine_filter_data
*d
= xcalloc(1, sizeof(*d
));
744 d
->nr
= filter_options
->sub_nr
;
745 CALLOC_ARRAY(d
->sub
, d
->nr
);
746 for (sub
= 0; sub
< d
->nr
; sub
++)
747 d
->sub
[sub
].filter
= list_objects_filter__init(
748 filter
->omits
? &d
->sub
[sub
].omits
: NULL
,
749 &filter_options
->sub
[sub
]);
751 filter
->filter_data
= d
;
752 filter
->filter_object_fn
= filter_combine
;
753 filter
->free_fn
= filter_combine__free
;
754 filter
->finalize_omits_fn
= filter_combine__finalize_omits
;
757 typedef void (*filter_init_fn
)(
758 struct list_objects_filter_options
*filter_options
,
759 struct filter
*filter
);
762 * Must match "enum list_objects_filter_choice".
764 static filter_init_fn s_filters
[] = {
766 filter_blobs_none__init
,
767 filter_blobs_limit__init
,
768 filter_trees_depth__init
,
769 filter_sparse_oid__init
,
770 filter_object_type__init
,
771 filter_combine__init
,
774 struct filter
*list_objects_filter__init(
775 struct oidset
*omitted
,
776 struct list_objects_filter_options
*filter_options
)
778 struct filter
*filter
;
779 filter_init_fn init_fn
;
781 assert((sizeof(s_filters
) / sizeof(s_filters
[0])) == LOFC__COUNT
);
786 if (filter_options
->choice
>= LOFC__COUNT
)
787 BUG("invalid list-objects filter choice: %d",
788 filter_options
->choice
);
790 init_fn
= s_filters
[filter_options
->choice
];
794 CALLOC_ARRAY(filter
, 1);
795 filter
->omits
= omitted
;
796 init_fn(filter_options
, filter
);
800 enum list_objects_filter_result
list_objects_filter__filter_object(
801 struct repository
*r
,
802 enum list_objects_filter_situation filter_situation
,
804 const char *pathname
,
805 const char *filename
,
806 struct filter
*filter
)
808 if (filter
&& (obj
->flags
& NOT_USER_GIVEN
))
809 return filter
->filter_object_fn(r
, filter_situation
, obj
,
812 filter
->filter_data
);
814 * No filter is active or user gave object explicitly. In this case,
815 * always show the object (except when LOFS_END_TREE, since this tree
816 * had already been shown when LOFS_BEGIN_TREE).
818 if (filter_situation
== LOFS_END_TREE
)
820 return LOFR_MARK_SEEN
| LOFR_DO_SHOW
;
823 void list_objects_filter__free(struct filter
*filter
)
827 if (filter
->finalize_omits_fn
&& filter
->omits
)
828 filter
->finalize_omits_fn(filter
->omits
, filter
->filter_data
);
829 filter
->free_fn(filter
->filter_data
);