Merge branch 'jt/fetch-pack-request-fix'
[git] / list-objects-filter.c
1 #include "cache.h"
2 #include "dir.h"
3 #include "tag.h"
4 #include "commit.h"
5 #include "tree.h"
6 #include "blob.h"
7 #include "diff.h"
8 #include "tree-walk.h"
9 #include "revision.h"
10 #include "list-objects.h"
11 #include "list-objects-filter.h"
12 #include "list-objects-filter-options.h"
13 #include "oidmap.h"
14 #include "oidset.h"
15 #include "object-store.h"
16
17 /* Remember to update object flag allocation in object.h */
18 /*
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
26  */
27 #define FILTER_SHOWN_BUT_REVISIT (1<<21)
28
29 struct subfilter {
30         struct filter *filter;
31         struct oidset seen;
32         struct oidset omits;
33         struct object_id skip_tree;
34         unsigned is_skipping_tree : 1;
35 };
36
37 struct filter {
38         enum list_objects_filter_result (*filter_object_fn)(
39                 struct repository *r,
40                 enum list_objects_filter_situation filter_situation,
41                 struct object *obj,
42                 const char *pathname,
43                 const char *filename,
44                 struct oidset *omits,
45                 void *filter_data);
46
47         /*
48          * Optional. If this function is supplied and the filter needs
49          * to collect omits, then this function is called once before
50          * free_fn is called.
51          *
52          * This is required because the following two conditions hold:
53          *
54          *   a. A tree filter can add and remove objects as an object
55          *      graph is traversed.
56          *   b. A combine filter's omit set is the union of all its
57          *      subfilters, which may include tree: filters.
58          *
59          * As such, the omits sets must be separate sets, and can only
60          * be unioned after the traversal is completed.
61          */
62         void (*finalize_omits_fn)(struct oidset *omits, void *filter_data);
63
64         void (*free_fn)(void *filter_data);
65
66         void *filter_data;
67
68         /* If non-NULL, the filter collects a list of the omitted OIDs here. */
69         struct oidset *omits;
70 };
71
72 static enum list_objects_filter_result filter_blobs_none(
73         struct repository *r,
74         enum list_objects_filter_situation filter_situation,
75         struct object *obj,
76         const char *pathname,
77         const char *filename,
78         struct oidset *omits,
79         void *filter_data_)
80 {
81         switch (filter_situation) {
82         default:
83                 BUG("unknown filter_situation: %d", filter_situation);
84
85         case LOFS_BEGIN_TREE:
86                 assert(obj->type == OBJ_TREE);
87                 /* always include all tree objects */
88                 return LOFR_MARK_SEEN | LOFR_DO_SHOW;
89
90         case LOFS_END_TREE:
91                 assert(obj->type == OBJ_TREE);
92                 return LOFR_ZERO;
93
94         case LOFS_BLOB:
95                 assert(obj->type == OBJ_BLOB);
96                 assert((obj->flags & SEEN) == 0);
97
98                 if (omits)
99                         oidset_insert(omits, &obj->oid);
100                 return LOFR_MARK_SEEN; /* but not LOFR_DO_SHOW (hard omit) */
101         }
102 }
103
104 static void filter_blobs_none__init(
105         struct list_objects_filter_options *filter_options,
106         struct filter *filter)
107 {
108         filter->filter_object_fn = filter_blobs_none;
109         filter->free_fn = free;
110 }
111
112 /*
113  * A filter for list-objects to omit ALL trees and blobs from the traversal.
114  * Can OPTIONALLY collect a list of the omitted OIDs.
115  */
116 struct filter_trees_depth_data {
117         /*
118          * Maps trees to the minimum depth at which they were seen. It is not
119          * necessary to re-traverse a tree at deeper or equal depths than it has
120          * already been traversed.
121          *
122          * We can't use LOFR_MARK_SEEN for tree objects since this will prevent
123          * it from being traversed at shallower depths.
124          */
125         struct oidmap seen_at_depth;
126
127         unsigned long exclude_depth;
128         unsigned long current_depth;
129 };
130
131 struct seen_map_entry {
132         struct oidmap_entry base;
133         size_t depth;
134 };
135
136 /* Returns 1 if the oid was in the omits set before it was invoked. */
137 static int filter_trees_update_omits(
138         struct object *obj,
139         struct oidset *omits,
140         int include_it)
141 {
142         if (!omits)
143                 return 0;
144
145         if (include_it)
146                 return oidset_remove(omits, &obj->oid);
147         else
148                 return oidset_insert(omits, &obj->oid);
149 }
150
151 static enum list_objects_filter_result filter_trees_depth(
152         struct repository *r,
153         enum list_objects_filter_situation filter_situation,
154         struct object *obj,
155         const char *pathname,
156         const char *filename,
157         struct oidset *omits,
158         void *filter_data_)
159 {
160         struct filter_trees_depth_data *filter_data = filter_data_;
161         struct seen_map_entry *seen_info;
162         int include_it = filter_data->current_depth <
163                 filter_data->exclude_depth;
164         int filter_res;
165         int already_seen;
166
167         /*
168          * Note that we do not use _MARK_SEEN in order to allow re-traversal in
169          * case we encounter a tree or blob again at a shallower depth.
170          */
171
172         switch (filter_situation) {
173         default:
174                 BUG("unknown filter_situation: %d", filter_situation);
175
176         case LOFS_END_TREE:
177                 assert(obj->type == OBJ_TREE);
178                 filter_data->current_depth--;
179                 return LOFR_ZERO;
180
181         case LOFS_BLOB:
182                 filter_trees_update_omits(obj, omits, include_it);
183                 return include_it ? LOFR_MARK_SEEN | LOFR_DO_SHOW : LOFR_ZERO;
184
185         case LOFS_BEGIN_TREE:
186                 seen_info = oidmap_get(
187                         &filter_data->seen_at_depth, &obj->oid);
188                 if (!seen_info) {
189                         CALLOC_ARRAY(seen_info, 1);
190                         oidcpy(&seen_info->base.oid, &obj->oid);
191                         seen_info->depth = filter_data->current_depth;
192                         oidmap_put(&filter_data->seen_at_depth, seen_info);
193                         already_seen = 0;
194                 } else {
195                         already_seen =
196                                 filter_data->current_depth >= seen_info->depth;
197                 }
198
199                 if (already_seen) {
200                         filter_res = LOFR_SKIP_TREE;
201                 } else {
202                         int been_omitted = filter_trees_update_omits(
203                                 obj, omits, include_it);
204                         seen_info->depth = filter_data->current_depth;
205
206                         if (include_it)
207                                 filter_res = LOFR_DO_SHOW;
208                         else if (omits && !been_omitted)
209                                 /*
210                                  * Must update omit information of children
211                                  * recursively; they have not been omitted yet.
212                                  */
213                                 filter_res = LOFR_ZERO;
214                         else
215                                 filter_res = LOFR_SKIP_TREE;
216                 }
217
218                 filter_data->current_depth++;
219                 return filter_res;
220         }
221 }
222
223 static void filter_trees_free(void *filter_data) {
224         struct filter_trees_depth_data *d = filter_data;
225         if (!d)
226                 return;
227         oidmap_free(&d->seen_at_depth, 1);
228         free(d);
229 }
230
231 static void filter_trees_depth__init(
232         struct list_objects_filter_options *filter_options,
233         struct filter *filter)
234 {
235         struct filter_trees_depth_data *d = xcalloc(1, sizeof(*d));
236         oidmap_init(&d->seen_at_depth, 0);
237         d->exclude_depth = filter_options->tree_exclude_depth;
238         d->current_depth = 0;
239
240         filter->filter_data = d;
241         filter->filter_object_fn = filter_trees_depth;
242         filter->free_fn = filter_trees_free;
243 }
244
245 /*
246  * A filter for list-objects to omit large blobs.
247  * And to OPTIONALLY collect a list of the omitted OIDs.
248  */
249 struct filter_blobs_limit_data {
250         unsigned long max_bytes;
251 };
252
253 static enum list_objects_filter_result filter_blobs_limit(
254         struct repository *r,
255         enum list_objects_filter_situation filter_situation,
256         struct object *obj,
257         const char *pathname,
258         const char *filename,
259         struct oidset *omits,
260         void *filter_data_)
261 {
262         struct filter_blobs_limit_data *filter_data = filter_data_;
263         unsigned long object_length;
264         enum object_type t;
265
266         switch (filter_situation) {
267         default:
268                 BUG("unknown filter_situation: %d", filter_situation);
269
270         case LOFS_BEGIN_TREE:
271                 assert(obj->type == OBJ_TREE);
272                 /* always include all tree objects */
273                 return LOFR_MARK_SEEN | LOFR_DO_SHOW;
274
275         case LOFS_END_TREE:
276                 assert(obj->type == OBJ_TREE);
277                 return LOFR_ZERO;
278
279         case LOFS_BLOB:
280                 assert(obj->type == OBJ_BLOB);
281                 assert((obj->flags & SEEN) == 0);
282
283                 t = oid_object_info(r, &obj->oid, &object_length);
284                 if (t != OBJ_BLOB) { /* probably OBJ_NONE */
285                         /*
286                          * We DO NOT have the blob locally, so we cannot
287                          * apply the size filter criteria.  Be conservative
288                          * and force show it (and let the caller deal with
289                          * the ambiguity).
290                          */
291                         goto include_it;
292                 }
293
294                 if (object_length < filter_data->max_bytes)
295                         goto include_it;
296
297                 if (omits)
298                         oidset_insert(omits, &obj->oid);
299                 return LOFR_MARK_SEEN; /* but not LOFR_DO_SHOW (hard omit) */
300         }
301
302 include_it:
303         if (omits)
304                 oidset_remove(omits, &obj->oid);
305         return LOFR_MARK_SEEN | LOFR_DO_SHOW;
306 }
307
308 static void filter_blobs_limit__init(
309         struct list_objects_filter_options *filter_options,
310         struct filter *filter)
311 {
312         struct filter_blobs_limit_data *d = xcalloc(1, sizeof(*d));
313         d->max_bytes = filter_options->blob_limit_value;
314
315         filter->filter_data = d;
316         filter->filter_object_fn = filter_blobs_limit;
317         filter->free_fn = free;
318 }
319
320 /*
321  * A filter driven by a sparse-checkout specification to only
322  * include blobs that a sparse checkout would populate.
323  *
324  * The sparse-checkout spec can be loaded from a blob with the
325  * given OID or from a local pathname.  We allow an OID because
326  * the repo may be bare or we may be doing the filtering on the
327  * server.
328  */
329 struct frame {
330         /*
331          * default_match is the usual default include/exclude value that
332          * should be inherited as we recurse into directories based
333          * upon pattern matching of the directory itself or of a
334          * containing directory.
335          */
336         enum pattern_match_result default_match;
337
338         /*
339          * 1 if the directory (recursively) contains any provisionally
340          * omitted objects.
341          *
342          * 0 if everything (recursively) contained in this directory
343          * has been explicitly included (SHOWN) in the result and
344          * the directory may be short-cut later in the traversal.
345          */
346         unsigned child_prov_omit : 1;
347 };
348
349 struct filter_sparse_data {
350         struct pattern_list pl;
351
352         size_t nr, alloc;
353         struct frame *array_frame;
354 };
355
356 static enum list_objects_filter_result filter_sparse(
357         struct repository *r,
358         enum list_objects_filter_situation filter_situation,
359         struct object *obj,
360         const char *pathname,
361         const char *filename,
362         struct oidset *omits,
363         void *filter_data_)
364 {
365         struct filter_sparse_data *filter_data = filter_data_;
366         int dtype;
367         struct frame *frame;
368         enum pattern_match_result match;
369
370         switch (filter_situation) {
371         default:
372                 BUG("unknown filter_situation: %d", filter_situation);
373
374         case LOFS_BEGIN_TREE:
375                 assert(obj->type == OBJ_TREE);
376                 dtype = DT_DIR;
377                 match = path_matches_pattern_list(pathname, strlen(pathname),
378                                                   filename, &dtype, &filter_data->pl,
379                                                   r->index);
380                 if (match == UNDECIDED)
381                         match = filter_data->array_frame[filter_data->nr - 1].default_match;
382
383                 ALLOC_GROW(filter_data->array_frame, filter_data->nr + 1,
384                            filter_data->alloc);
385                 filter_data->array_frame[filter_data->nr].default_match = match;
386                 filter_data->array_frame[filter_data->nr].child_prov_omit = 0;
387                 filter_data->nr++;
388
389                 /*
390                  * A directory with this tree OID may appear in multiple
391                  * places in the tree. (Think of a directory move or copy,
392                  * with no other changes, so the OID is the same, but the
393                  * full pathnames of objects within this directory are new
394                  * and may match is_excluded() patterns differently.)
395                  * So we cannot mark this directory as SEEN (yet), since
396                  * that will prevent process_tree() from revisiting this
397                  * tree object with other pathname prefixes.
398                  *
399                  * Only _DO_SHOW the tree object the first time we visit
400                  * this tree object.
401                  *
402                  * We always show all tree objects.  A future optimization
403                  * may want to attempt to narrow this.
404                  */
405                 if (obj->flags & FILTER_SHOWN_BUT_REVISIT)
406                         return LOFR_ZERO;
407                 obj->flags |= FILTER_SHOWN_BUT_REVISIT;
408                 return LOFR_DO_SHOW;
409
410         case LOFS_END_TREE:
411                 assert(obj->type == OBJ_TREE);
412                 assert(filter_data->nr > 1);
413
414                 frame = &filter_data->array_frame[--filter_data->nr];
415
416                 /*
417                  * Tell our parent directory if any of our children were
418                  * provisionally omitted.
419                  */
420                 filter_data->array_frame[filter_data->nr - 1].child_prov_omit |=
421                         frame->child_prov_omit;
422
423                 /*
424                  * If there are NO provisionally omitted child objects (ALL child
425                  * objects in this folder were INCLUDED), then we can mark the
426                  * folder as SEEN (so we will not have to revisit it again).
427                  */
428                 if (!frame->child_prov_omit)
429                         return LOFR_MARK_SEEN;
430                 return LOFR_ZERO;
431
432         case LOFS_BLOB:
433                 assert(obj->type == OBJ_BLOB);
434                 assert((obj->flags & SEEN) == 0);
435
436                 frame = &filter_data->array_frame[filter_data->nr - 1];
437
438                 dtype = DT_REG;
439                 match = path_matches_pattern_list(pathname, strlen(pathname),
440                                             filename, &dtype, &filter_data->pl,
441                                             r->index);
442                 if (match == UNDECIDED)
443                         match = frame->default_match;
444                 if (match == MATCHED) {
445                         if (omits)
446                                 oidset_remove(omits, &obj->oid);
447                         return LOFR_MARK_SEEN | LOFR_DO_SHOW;
448                 }
449
450                 /*
451                  * Provisionally omit it.  We've already established that
452                  * this pathname is not in the sparse-checkout specification
453                  * with the CURRENT pathname, so we *WANT* to omit this blob.
454                  *
455                  * However, a pathname elsewhere in the tree may also
456                  * reference this same blob, so we cannot reject it yet.
457                  * Leave the LOFR_ bits unset so that if the blob appears
458                  * again in the traversal, we will be asked again.
459                  */
460                 if (omits)
461                         oidset_insert(omits, &obj->oid);
462
463                 /*
464                  * Remember that at least 1 blob in this tree was
465                  * provisionally omitted.  This prevents us from short
466                  * cutting the tree in future iterations.
467                  */
468                 frame->child_prov_omit = 1;
469                 return LOFR_ZERO;
470         }
471 }
472
473
474 static void filter_sparse_free(void *filter_data)
475 {
476         struct filter_sparse_data *d = filter_data;
477         free(d->array_frame);
478         free(d);
479 }
480
481 static void filter_sparse_oid__init(
482         struct list_objects_filter_options *filter_options,
483         struct filter *filter)
484 {
485         struct filter_sparse_data *d = xcalloc(1, sizeof(*d));
486         struct object_context oc;
487         struct object_id sparse_oid;
488
489         if (get_oid_with_context(the_repository,
490                                  filter_options->sparse_oid_name,
491                                  GET_OID_BLOB, &sparse_oid, &oc))
492                 die(_("unable to access sparse blob in '%s'"),
493                     filter_options->sparse_oid_name);
494         if (add_patterns_from_blob_to_list(&sparse_oid, "", 0, &d->pl) < 0)
495                 die(_("unable to parse sparse filter data in %s"),
496                     oid_to_hex(&sparse_oid));
497
498         ALLOC_GROW(d->array_frame, d->nr + 1, d->alloc);
499         d->array_frame[d->nr].default_match = 0; /* default to include */
500         d->array_frame[d->nr].child_prov_omit = 0;
501         d->nr++;
502
503         filter->filter_data = d;
504         filter->filter_object_fn = filter_sparse;
505         filter->free_fn = filter_sparse_free;
506 }
507
508 /* A filter which only shows objects shown by all sub-filters. */
509 struct combine_filter_data {
510         struct subfilter *sub;
511         size_t nr;
512 };
513
514 static enum list_objects_filter_result process_subfilter(
515         struct repository *r,
516         enum list_objects_filter_situation filter_situation,
517         struct object *obj,
518         const char *pathname,
519         const char *filename,
520         struct subfilter *sub)
521 {
522         enum list_objects_filter_result result;
523
524         /*
525          * Check and update is_skipping_tree before oidset_contains so
526          * that is_skipping_tree gets unset even when the object is
527          * marked as seen.  As of this writing, no filter uses
528          * LOFR_MARK_SEEN on trees that also uses LOFR_SKIP_TREE, so the
529          * ordering is only theoretically important. Be cautious if you
530          * change the order of the below checks and more filters have
531          * been added!
532          */
533         if (sub->is_skipping_tree) {
534                 if (filter_situation == LOFS_END_TREE &&
535                     oideq(&obj->oid, &sub->skip_tree))
536                         sub->is_skipping_tree = 0;
537                 else
538                         return LOFR_ZERO;
539         }
540         if (oidset_contains(&sub->seen, &obj->oid))
541                 return LOFR_ZERO;
542
543         result = list_objects_filter__filter_object(
544                 r, filter_situation, obj, pathname, filename, sub->filter);
545
546         if (result & LOFR_MARK_SEEN)
547                 oidset_insert(&sub->seen, &obj->oid);
548
549         if (result & LOFR_SKIP_TREE) {
550                 sub->is_skipping_tree = 1;
551                 sub->skip_tree = obj->oid;
552         }
553
554         return result;
555 }
556
557 static enum list_objects_filter_result filter_combine(
558         struct repository *r,
559         enum list_objects_filter_situation filter_situation,
560         struct object *obj,
561         const char *pathname,
562         const char *filename,
563         struct oidset *omits,
564         void *filter_data)
565 {
566         struct combine_filter_data *d = filter_data;
567         enum list_objects_filter_result combined_result =
568                 LOFR_DO_SHOW | LOFR_MARK_SEEN | LOFR_SKIP_TREE;
569         size_t sub;
570
571         for (sub = 0; sub < d->nr; sub++) {
572                 enum list_objects_filter_result sub_result = process_subfilter(
573                         r, filter_situation, obj, pathname, filename,
574                         &d->sub[sub]);
575                 if (!(sub_result & LOFR_DO_SHOW))
576                         combined_result &= ~LOFR_DO_SHOW;
577                 if (!(sub_result & LOFR_MARK_SEEN))
578                         combined_result &= ~LOFR_MARK_SEEN;
579                 if (!d->sub[sub].is_skipping_tree)
580                         combined_result &= ~LOFR_SKIP_TREE;
581         }
582
583         return combined_result;
584 }
585
586 static void filter_combine__free(void *filter_data)
587 {
588         struct combine_filter_data *d = filter_data;
589         size_t sub;
590         for (sub = 0; sub < d->nr; sub++) {
591                 list_objects_filter__free(d->sub[sub].filter);
592                 oidset_clear(&d->sub[sub].seen);
593                 if (d->sub[sub].omits.set.size)
594                         BUG("expected oidset to be cleared already");
595         }
596         free(d->sub);
597 }
598
599 static void add_all(struct oidset *dest, struct oidset *src) {
600         struct oidset_iter iter;
601         struct object_id *src_oid;
602
603         oidset_iter_init(src, &iter);
604         while ((src_oid = oidset_iter_next(&iter)) != NULL)
605                 oidset_insert(dest, src_oid);
606 }
607
608 static void filter_combine__finalize_omits(
609         struct oidset *omits,
610         void *filter_data)
611 {
612         struct combine_filter_data *d = filter_data;
613         size_t sub;
614
615         for (sub = 0; sub < d->nr; sub++) {
616                 add_all(omits, &d->sub[sub].omits);
617                 oidset_clear(&d->sub[sub].omits);
618         }
619 }
620
621 static void filter_combine__init(
622         struct list_objects_filter_options *filter_options,
623         struct filter* filter)
624 {
625         struct combine_filter_data *d = xcalloc(1, sizeof(*d));
626         size_t sub;
627
628         d->nr = filter_options->sub_nr;
629         CALLOC_ARRAY(d->sub, d->nr);
630         for (sub = 0; sub < d->nr; sub++)
631                 d->sub[sub].filter = list_objects_filter__init(
632                         filter->omits ? &d->sub[sub].omits : NULL,
633                         &filter_options->sub[sub]);
634
635         filter->filter_data = d;
636         filter->filter_object_fn = filter_combine;
637         filter->free_fn = filter_combine__free;
638         filter->finalize_omits_fn = filter_combine__finalize_omits;
639 }
640
641 typedef void (*filter_init_fn)(
642         struct list_objects_filter_options *filter_options,
643         struct filter *filter);
644
645 /*
646  * Must match "enum list_objects_filter_choice".
647  */
648 static filter_init_fn s_filters[] = {
649         NULL,
650         filter_blobs_none__init,
651         filter_blobs_limit__init,
652         filter_trees_depth__init,
653         filter_sparse_oid__init,
654         filter_combine__init,
655 };
656
657 struct filter *list_objects_filter__init(
658         struct oidset *omitted,
659         struct list_objects_filter_options *filter_options)
660 {
661         struct filter *filter;
662         filter_init_fn init_fn;
663
664         assert((sizeof(s_filters) / sizeof(s_filters[0])) == LOFC__COUNT);
665
666         if (!filter_options)
667                 return NULL;
668
669         if (filter_options->choice >= LOFC__COUNT)
670                 BUG("invalid list-objects filter choice: %d",
671                     filter_options->choice);
672
673         init_fn = s_filters[filter_options->choice];
674         if (!init_fn)
675                 return NULL;
676
677         CALLOC_ARRAY(filter, 1);
678         filter->omits = omitted;
679         init_fn(filter_options, filter);
680         return filter;
681 }
682
683 enum list_objects_filter_result list_objects_filter__filter_object(
684         struct repository *r,
685         enum list_objects_filter_situation filter_situation,
686         struct object *obj,
687         const char *pathname,
688         const char *filename,
689         struct filter *filter)
690 {
691         if (filter && (obj->flags & NOT_USER_GIVEN))
692                 return filter->filter_object_fn(r, filter_situation, obj,
693                                                 pathname, filename,
694                                                 filter->omits,
695                                                 filter->filter_data);
696         /*
697          * No filter is active or user gave object explicitly. In this case,
698          * always show the object (except when LOFS_END_TREE, since this tree
699          * had already been shown when LOFS_BEGIN_TREE).
700          */
701         if (filter_situation == LOFS_END_TREE)
702                 return 0;
703         return LOFR_MARK_SEEN | LOFR_DO_SHOW;
704 }
705
706 void list_objects_filter__free(struct filter *filter)
707 {
708         if (!filter)
709                 return;
710         if (filter->finalize_omits_fn && filter->omits)
711                 filter->finalize_omits_fn(filter->omits, filter->filter_data);
712         filter->free_fn(filter->filter_data);
713         free(filter);
714 }