3 #include "run-command.h"
8 * convert.c - convert a file when checking it out and checking it in.
10 * This should use the pathname to decide on whether it wants to do some
11 * more interesting conversions (automatic gzip/unzip, general format
12 * conversions etc etc), but by default it just does automatic CRLF<->LF
13 * translation when the "text" attribute or "auto_crlf" option is set.
16 /* Stat bits: When BIN is set, the txt bits are unset */
17 #define CONVERT_STAT_BITS_TXT_LF 0x1
18 #define CONVERT_STAT_BITS_TXT_CRLF 0x2
19 #define CONVERT_STAT_BITS_BIN 0x4
33 /* NUL, CR, LF and CRLF counts */
34 unsigned nul
, lonecr
, lonelf
, crlf
;
36 /* These are just approximations! */
37 unsigned printable
, nonprintable
;
40 static void gather_stats(const char *buf
, unsigned long size
, struct text_stat
*stats
)
44 memset(stats
, 0, sizeof(*stats
));
46 for (i
= 0; i
< size
; i
++) {
47 unsigned char c
= buf
[i
];
49 if (i
+1 < size
&& buf
[i
+1] == '\n') {
62 stats
->nonprintable
++;
65 /* BS, HT, ESC and FF */
66 case '\b': case '\t': case '\033': case '\014':
73 stats
->nonprintable
++;
80 /* If file ends with EOF then don't count this EOF as non-printable. */
81 if (size
>= 1 && buf
[size
-1] == '\032')
82 stats
->nonprintable
--;
86 * The same heuristics as diff.c::mmfile_is_binary()
87 * We treat files with bare CR as binary
89 static int convert_is_binary(unsigned long size
, const struct text_stat
*stats
)
95 if ((stats
->printable
>> 7) < stats
->nonprintable
)
100 static unsigned int gather_convert_stats(const char *data
, unsigned long size
)
102 struct text_stat stats
;
106 gather_stats(data
, size
, &stats
);
107 if (convert_is_binary(size
, &stats
))
108 ret
|= CONVERT_STAT_BITS_BIN
;
110 ret
|= CONVERT_STAT_BITS_TXT_CRLF
;
112 ret
|= CONVERT_STAT_BITS_TXT_LF
;
117 static const char *gather_convert_stats_ascii(const char *data
, unsigned long size
)
119 unsigned int convert_stats
= gather_convert_stats(data
, size
);
121 if (convert_stats
& CONVERT_STAT_BITS_BIN
)
123 switch (convert_stats
) {
124 case CONVERT_STAT_BITS_TXT_LF
:
126 case CONVERT_STAT_BITS_TXT_CRLF
:
128 case CONVERT_STAT_BITS_TXT_LF
| CONVERT_STAT_BITS_TXT_CRLF
:
135 const char *get_cached_convert_stats_ascii(const char *path
)
139 void *data
= read_blob_data_from_cache(path
, &sz
);
140 ret
= gather_convert_stats_ascii(data
, sz
);
145 const char *get_wt_convert_stats_ascii(const char *path
)
147 const char *ret
= "";
148 struct strbuf sb
= STRBUF_INIT
;
149 if (strbuf_read_file(&sb
, path
, 0) >= 0)
150 ret
= gather_convert_stats_ascii(sb
.buf
, sb
.len
);
155 static int text_eol_is_crlf(void)
157 if (auto_crlf
== AUTO_CRLF_TRUE
)
159 else if (auto_crlf
== AUTO_CRLF_INPUT
)
161 if (core_eol
== EOL_CRLF
)
163 if (core_eol
== EOL_UNSET
&& EOL_NATIVE
== EOL_CRLF
)
168 static enum eol
output_eol(enum crlf_action crlf_action
)
170 switch (crlf_action
) {
175 case CRLF_TEXT_INPUT
:
180 case CRLF_AUTO_INPUT
:
185 return text_eol_is_crlf() ? EOL_CRLF
: EOL_LF
;
187 warning("Illegal crlf_action %d\n", (int)crlf_action
);
191 static void check_safe_crlf(const char *path
, enum crlf_action crlf_action
,
192 struct text_stat
*old_stats
, struct text_stat
*new_stats
,
193 enum safe_crlf checksafe
)
195 if (old_stats
->crlf
&& !new_stats
->crlf
) {
197 * CRLFs would not be restored by checkout
199 if (checksafe
== SAFE_CRLF_WARN
)
200 warning("CRLF will be replaced by LF in %s.\nThe file will have its original line endings in your working directory.", path
);
201 else /* i.e. SAFE_CRLF_FAIL */
202 die("CRLF would be replaced by LF in %s.", path
);
203 } else if (old_stats
->lonelf
&& !new_stats
->lonelf
) {
205 * CRLFs would be added by checkout
207 if (checksafe
== SAFE_CRLF_WARN
)
208 warning("LF will be replaced by CRLF in %s.\nThe file will have its original line endings in your working directory.", path
);
209 else /* i.e. SAFE_CRLF_FAIL */
210 die("LF would be replaced by CRLF in %s", path
);
214 static int has_cr_in_index(const char *path
)
220 data
= read_blob_data_from_cache(path
, &sz
);
223 has_cr
= memchr(data
, '\r', sz
) != NULL
;
228 static int will_convert_lf_to_crlf(size_t len
, struct text_stat
*stats
,
229 enum crlf_action crlf_action
)
231 if (output_eol(crlf_action
) != EOL_CRLF
)
233 /* No "naked" LF? Nothing to convert, regardless. */
237 if (crlf_action
== CRLF_AUTO
|| crlf_action
== CRLF_AUTO_INPUT
|| crlf_action
== CRLF_AUTO_CRLF
) {
238 /* If we have any CR or CRLF line endings, we do not touch it */
239 /* This is the new safer autocrlf-handling */
240 if (stats
->lonecr
|| stats
->crlf
)
243 if (convert_is_binary(len
, stats
))
250 static int crlf_to_git(const char *path
, const char *src
, size_t len
,
252 enum crlf_action crlf_action
, enum safe_crlf checksafe
)
254 struct text_stat stats
;
256 int convert_crlf_into_lf
;
258 if (crlf_action
== CRLF_BINARY
||
263 * If we are doing a dry-run and have no source buffer, there is
264 * nothing to analyze; we must assume we would convert.
269 gather_stats(src
, len
, &stats
);
270 /* Optimization: No CRLF? Nothing to convert, regardless. */
271 convert_crlf_into_lf
= !!stats
.crlf
;
273 if (crlf_action
== CRLF_AUTO
|| crlf_action
== CRLF_AUTO_INPUT
|| crlf_action
== CRLF_AUTO_CRLF
) {
274 if (convert_is_binary(len
, &stats
))
277 * If the file in the index has any CR in it, do not convert.
278 * This is the new safer autocrlf handling.
280 if (checksafe
== SAFE_CRLF_RENORMALIZE
)
281 checksafe
= SAFE_CRLF_FALSE
;
282 else if (has_cr_in_index(path
))
283 convert_crlf_into_lf
= 0;
285 if (checksafe
&& len
) {
286 struct text_stat new_stats
;
287 memcpy(&new_stats
, &stats
, sizeof(new_stats
));
288 /* simulate "git add" */
289 if (convert_crlf_into_lf
) {
290 new_stats
.lonelf
+= new_stats
.crlf
;
293 /* simulate "git checkout" */
294 if (will_convert_lf_to_crlf(len
, &new_stats
, crlf_action
)) {
295 new_stats
.crlf
+= new_stats
.lonelf
;
296 new_stats
.lonelf
= 0;
298 check_safe_crlf(path
, crlf_action
, &stats
, &new_stats
, checksafe
);
300 if (!convert_crlf_into_lf
)
304 * At this point all of our source analysis is done, and we are sure we
305 * would convert. If we are in dry-run mode, we can give an answer.
310 /* only grow if not in place */
311 if (strbuf_avail(buf
) + buf
->len
< len
)
312 strbuf_grow(buf
, len
- buf
->len
);
314 if (crlf_action
== CRLF_AUTO
|| crlf_action
== CRLF_AUTO_INPUT
|| crlf_action
== CRLF_AUTO_CRLF
) {
316 * If we guessed, we already know we rejected a file with
317 * lone CR, and we can strip a CR without looking at what
321 unsigned char c
= *src
++;
327 unsigned char c
= *src
++;
328 if (! (c
== '\r' && (1 < len
&& *src
== '\n')))
332 strbuf_setlen(buf
, dst
- buf
->buf
);
336 static int crlf_to_worktree(const char *path
, const char *src
, size_t len
,
337 struct strbuf
*buf
, enum crlf_action crlf_action
)
339 char *to_free
= NULL
;
340 struct text_stat stats
;
342 if (!len
|| output_eol(crlf_action
) != EOL_CRLF
)
345 gather_stats(src
, len
, &stats
);
346 if (!will_convert_lf_to_crlf(len
, &stats
, crlf_action
))
349 /* are we "faking" in place editing ? */
351 to_free
= strbuf_detach(buf
, NULL
);
353 strbuf_grow(buf
, len
+ stats
.lonelf
);
355 const char *nl
= memchr(src
, '\n', len
);
358 if (nl
> src
&& nl
[-1] == '\r') {
359 strbuf_add(buf
, src
, nl
+ 1 - src
);
361 strbuf_add(buf
, src
, nl
- src
);
362 strbuf_addstr(buf
, "\r\n");
367 strbuf_add(buf
, src
, len
);
373 struct filter_params
{
381 static int filter_buffer_or_fd(int in
, int out
, void *data
)
384 * Spawn cmd and feed the buffer contents through its stdin.
386 struct child_process child_process
= CHILD_PROCESS_INIT
;
387 struct filter_params
*params
= (struct filter_params
*)data
;
388 int write_err
, status
;
389 const char *argv
[] = { NULL
, NULL
};
391 /* apply % substitution to cmd */
392 struct strbuf cmd
= STRBUF_INIT
;
393 struct strbuf path
= STRBUF_INIT
;
394 struct strbuf_expand_dict_entry dict
[] = {
399 /* quote the path to preserve spaces, etc. */
400 sq_quote_buf(&path
, params
->path
);
401 dict
[0].value
= path
.buf
;
403 /* expand all %f with the quoted path */
404 strbuf_expand(&cmd
, params
->cmd
, strbuf_expand_dict_cb
, &dict
);
405 strbuf_release(&path
);
409 child_process
.argv
= argv
;
410 child_process
.use_shell
= 1;
411 child_process
.in
= -1;
412 child_process
.out
= out
;
414 if (start_command(&child_process
))
415 return error("cannot fork to run external filter %s", params
->cmd
);
417 sigchain_push(SIGPIPE
, SIG_IGN
);
420 write_err
= (write_in_full(child_process
.in
,
421 params
->src
, params
->size
) < 0);
425 write_err
= copy_fd(params
->fd
, child_process
.in
);
426 if (write_err
== COPY_WRITE_ERROR
&& errno
== EPIPE
)
430 if (close(child_process
.in
))
433 error("cannot feed the input to external filter %s", params
->cmd
);
435 sigchain_pop(SIGPIPE
);
437 status
= finish_command(&child_process
);
439 error("external filter %s failed %d", params
->cmd
, status
);
441 strbuf_release(&cmd
);
442 return (write_err
|| status
);
445 static int apply_filter(const char *path
, const char *src
, size_t len
, int fd
,
446 struct strbuf
*dst
, const char *cmd
)
449 * Create a pipeline to have the command filter the buffer's
452 * (child --> cmd) --> us
455 struct strbuf nbuf
= STRBUF_INIT
;
457 struct filter_params params
;
465 memset(&async
, 0, sizeof(async
));
466 async
.proc
= filter_buffer_or_fd
;
467 async
.data
= ¶ms
;
476 if (start_async(&async
))
477 return 0; /* error was already reported */
479 if (strbuf_read(&nbuf
, async
.out
, len
) < 0) {
480 error("read from external filter %s failed", cmd
);
483 if (close(async
.out
)) {
484 error("read from external filter %s failed", cmd
);
487 if (finish_async(&async
)) {
488 error("external filter %s failed", cmd
);
493 strbuf_swap(dst
, &nbuf
);
495 strbuf_release(&nbuf
);
499 static struct convert_driver
{
501 struct convert_driver
*next
;
505 } *user_convert
, **user_convert_tail
;
507 static int read_convert_config(const char *var
, const char *value
, void *cb
)
509 const char *key
, *name
;
511 struct convert_driver
*drv
;
514 * External conversion drivers are configured using
515 * "filter.<name>.variable".
517 if (parse_config_key(var
, "filter", &name
, &namelen
, &key
) < 0 || !name
)
519 for (drv
= user_convert
; drv
; drv
= drv
->next
)
520 if (!strncmp(drv
->name
, name
, namelen
) && !drv
->name
[namelen
])
523 drv
= xcalloc(1, sizeof(struct convert_driver
));
524 drv
->name
= xmemdupz(name
, namelen
);
525 *user_convert_tail
= drv
;
526 user_convert_tail
= &(drv
->next
);
530 * filter.<name>.smudge and filter.<name>.clean specifies
535 * The command-line will not be interpolated in any way.
538 if (!strcmp("smudge", key
))
539 return git_config_string(&drv
->smudge
, var
, value
);
541 if (!strcmp("clean", key
))
542 return git_config_string(&drv
->clean
, var
, value
);
544 if (!strcmp("required", key
)) {
545 drv
->required
= git_config_bool(var
, value
);
552 static int count_ident(const char *cp
, unsigned long size
)
555 * "$Id: 0000000000000000000000000000000000000000 $" <=> "$Id$"
567 if (memcmp("Id", cp
, 2))
578 * "$Id: ... "; scan up to the closing dollar sign and discard.
594 static int ident_to_git(const char *path
, const char *src
, size_t len
,
595 struct strbuf
*buf
, int ident
)
599 if (!ident
|| (src
&& !count_ident(src
, len
)))
605 /* only grow if not in place */
606 if (strbuf_avail(buf
) + buf
->len
< len
)
607 strbuf_grow(buf
, len
- buf
->len
);
610 dollar
= memchr(src
, '$', len
);
613 memmove(dst
, src
, dollar
+ 1 - src
);
614 dst
+= dollar
+ 1 - src
;
615 len
-= dollar
+ 1 - src
;
618 if (len
> 3 && !memcmp(src
, "Id:", 3)) {
619 dollar
= memchr(src
+ 3, '$', len
- 3);
622 if (memchr(src
+ 3, '\n', dollar
- src
- 3)) {
623 /* Line break before the next dollar. */
627 memcpy(dst
, "Id$", 3);
629 len
-= dollar
+ 1 - src
;
633 memmove(dst
, src
, len
);
634 strbuf_setlen(buf
, dst
+ len
- buf
->buf
);
638 static int ident_to_worktree(const char *path
, const char *src
, size_t len
,
639 struct strbuf
*buf
, int ident
)
641 unsigned char sha1
[20];
642 char *to_free
= NULL
, *dollar
, *spc
;
648 cnt
= count_ident(src
, len
);
652 /* are we "faking" in place editing ? */
654 to_free
= strbuf_detach(buf
, NULL
);
655 hash_sha1_file(src
, len
, "blob", sha1
);
657 strbuf_grow(buf
, len
+ cnt
* 43);
659 /* step 1: run to the next '$' */
660 dollar
= memchr(src
, '$', len
);
663 strbuf_add(buf
, src
, dollar
+ 1 - src
);
664 len
-= dollar
+ 1 - src
;
667 /* step 2: does it looks like a bit like Id:xxx$ or Id$ ? */
668 if (len
< 3 || memcmp("Id", src
, 2))
671 /* step 3: skip over Id$ or Id:xxxxx$ */
675 } else if (src
[2] == ':') {
677 * It's possible that an expanded Id has crept its way into the
678 * repository, we cope with that by stripping the expansion out.
679 * This is probably not a good idea, since it will cause changes
680 * on checkout, which won't go away by stash, but let's keep it
683 dollar
= memchr(src
+ 3, '$', len
- 3);
685 /* incomplete keyword, no more '$', so just quit the loop */
689 if (memchr(src
+ 3, '\n', dollar
- src
- 3)) {
690 /* Line break before the next dollar. */
694 spc
= memchr(src
+ 4, ' ', dollar
- src
- 4);
695 if (spc
&& spc
< dollar
-1) {
696 /* There are spaces in unexpected places.
697 * This is probably an id from some other
698 * versioning system. Keep it for now.
703 len
-= dollar
+ 1 - src
;
706 /* it wasn't a "Id$" or "Id:xxxx$" */
710 /* step 4: substitute */
711 strbuf_addstr(buf
, "Id: ");
712 strbuf_add(buf
, sha1_to_hex(sha1
), 40);
713 strbuf_addstr(buf
, " $");
715 strbuf_add(buf
, src
, len
);
721 static enum crlf_action
git_path_check_crlf(struct git_attr_check
*check
)
723 const char *value
= check
->value
;
725 if (ATTR_TRUE(value
))
727 else if (ATTR_FALSE(value
))
729 else if (ATTR_UNSET(value
))
731 else if (!strcmp(value
, "input"))
732 return CRLF_TEXT_INPUT
;
733 else if (!strcmp(value
, "auto"))
735 return CRLF_UNDEFINED
;
738 static enum eol
git_path_check_eol(struct git_attr_check
*check
)
740 const char *value
= check
->value
;
742 if (ATTR_UNSET(value
))
744 else if (!strcmp(value
, "lf"))
746 else if (!strcmp(value
, "crlf"))
751 static struct convert_driver
*git_path_check_convert(struct git_attr_check
*check
)
753 const char *value
= check
->value
;
754 struct convert_driver
*drv
;
756 if (ATTR_TRUE(value
) || ATTR_FALSE(value
) || ATTR_UNSET(value
))
758 for (drv
= user_convert
; drv
; drv
= drv
->next
)
759 if (!strcmp(value
, drv
->name
))
764 static int git_path_check_ident(struct git_attr_check
*check
)
766 const char *value
= check
->value
;
768 return !!ATTR_TRUE(value
);
772 struct convert_driver
*drv
;
773 enum crlf_action attr_action
; /* What attr says */
774 enum crlf_action crlf_action
; /* When no attr is set, use core.autocrlf */
778 static const char *conv_attr_name
[] = {
779 "crlf", "ident", "filter", "eol", "text",
781 #define NUM_CONV_ATTRS ARRAY_SIZE(conv_attr_name)
783 static void convert_attrs(struct conv_attrs
*ca
, const char *path
)
786 static struct git_attr_check ccheck
[NUM_CONV_ATTRS
];
788 if (!ccheck
[0].attr
) {
789 for (i
= 0; i
< NUM_CONV_ATTRS
; i
++)
790 ccheck
[i
].attr
= git_attr(conv_attr_name
[i
]);
791 user_convert_tail
= &user_convert
;
792 git_config(read_convert_config
, NULL
);
795 if (!git_check_attr(path
, NUM_CONV_ATTRS
, ccheck
)) {
796 ca
->crlf_action
= git_path_check_crlf(ccheck
+ 4);
797 if (ca
->crlf_action
== CRLF_UNDEFINED
)
798 ca
->crlf_action
= git_path_check_crlf(ccheck
+ 0);
799 ca
->attr_action
= ca
->crlf_action
;
800 ca
->ident
= git_path_check_ident(ccheck
+ 1);
801 ca
->drv
= git_path_check_convert(ccheck
+ 2);
802 if (ca
->crlf_action
!= CRLF_BINARY
) {
803 enum eol eol_attr
= git_path_check_eol(ccheck
+ 3);
804 if (ca
->crlf_action
== CRLF_AUTO
&& eol_attr
== EOL_LF
)
805 ca
->crlf_action
= CRLF_AUTO_INPUT
;
806 else if (ca
->crlf_action
== CRLF_AUTO
&& eol_attr
== EOL_CRLF
)
807 ca
->crlf_action
= CRLF_AUTO_CRLF
;
808 else if (eol_attr
== EOL_LF
)
809 ca
->crlf_action
= CRLF_TEXT_INPUT
;
810 else if (eol_attr
== EOL_CRLF
)
811 ca
->crlf_action
= CRLF_TEXT_CRLF
;
813 ca
->attr_action
= ca
->crlf_action
;
816 ca
->crlf_action
= CRLF_UNDEFINED
;
819 if (ca
->crlf_action
== CRLF_TEXT
)
820 ca
->crlf_action
= text_eol_is_crlf() ? CRLF_TEXT_CRLF
: CRLF_TEXT_INPUT
;
821 if (ca
->crlf_action
== CRLF_UNDEFINED
&& auto_crlf
== AUTO_CRLF_FALSE
)
822 ca
->crlf_action
= CRLF_BINARY
;
823 if (ca
->crlf_action
== CRLF_UNDEFINED
&& auto_crlf
== AUTO_CRLF_TRUE
)
824 ca
->crlf_action
= CRLF_AUTO_CRLF
;
825 if (ca
->crlf_action
== CRLF_UNDEFINED
&& auto_crlf
== AUTO_CRLF_INPUT
)
826 ca
->crlf_action
= CRLF_AUTO_INPUT
;
829 int would_convert_to_git_filter_fd(const char *path
)
831 struct conv_attrs ca
;
833 convert_attrs(&ca
, path
);
838 * Apply a filter to an fd only if the filter is required to succeed.
839 * We must die if the filter fails, because the original data before
840 * filtering is not available.
842 if (!ca
.drv
->required
)
845 return apply_filter(path
, NULL
, 0, -1, NULL
, ca
.drv
->clean
);
848 const char *get_convert_attr_ascii(const char *path
)
850 struct conv_attrs ca
;
852 convert_attrs(&ca
, path
);
853 switch (ca
.attr_action
) {
860 case CRLF_TEXT_INPUT
:
861 return "text eol=lf";
863 return "text eol=crlf";
867 return "text=auto eol=crlf";
868 case CRLF_AUTO_INPUT
:
869 return "text=auto eol=lf";
874 int convert_to_git(const char *path
, const char *src
, size_t len
,
875 struct strbuf
*dst
, enum safe_crlf checksafe
)
878 const char *filter
= NULL
;
880 struct conv_attrs ca
;
882 convert_attrs(&ca
, path
);
884 filter
= ca
.drv
->clean
;
885 required
= ca
.drv
->required
;
888 ret
|= apply_filter(path
, src
, len
, -1, dst
, filter
);
889 if (!ret
&& required
)
890 die("%s: clean filter '%s' failed", path
, ca
.drv
->name
);
896 ret
|= crlf_to_git(path
, src
, len
, dst
, ca
.crlf_action
, checksafe
);
901 return ret
| ident_to_git(path
, src
, len
, dst
, ca
.ident
);
904 void convert_to_git_filter_fd(const char *path
, int fd
, struct strbuf
*dst
,
905 enum safe_crlf checksafe
)
907 struct conv_attrs ca
;
908 convert_attrs(&ca
, path
);
911 assert(ca
.drv
->clean
);
913 if (!apply_filter(path
, NULL
, 0, fd
, dst
, ca
.drv
->clean
))
914 die("%s: clean filter '%s' failed", path
, ca
.drv
->name
);
916 crlf_to_git(path
, dst
->buf
, dst
->len
, dst
, ca
.crlf_action
, checksafe
);
917 ident_to_git(path
, dst
->buf
, dst
->len
, dst
, ca
.ident
);
920 static int convert_to_working_tree_internal(const char *path
, const char *src
,
921 size_t len
, struct strbuf
*dst
,
924 int ret
= 0, ret_filter
= 0;
925 const char *filter
= NULL
;
927 struct conv_attrs ca
;
929 convert_attrs(&ca
, path
);
931 filter
= ca
.drv
->smudge
;
932 required
= ca
.drv
->required
;
935 ret
|= ident_to_worktree(path
, src
, len
, dst
, ca
.ident
);
941 * CRLF conversion can be skipped if normalizing, unless there
942 * is a smudge filter. The filter might expect CRLFs.
944 if (filter
|| !normalizing
) {
945 ret
|= crlf_to_worktree(path
, src
, len
, dst
, ca
.crlf_action
);
952 ret_filter
= apply_filter(path
, src
, len
, -1, dst
, filter
);
953 if (!ret_filter
&& required
)
954 die("%s: smudge filter %s failed", path
, ca
.drv
->name
);
956 return ret
| ret_filter
;
959 int convert_to_working_tree(const char *path
, const char *src
, size_t len
, struct strbuf
*dst
)
961 return convert_to_working_tree_internal(path
, src
, len
, dst
, 0);
964 int renormalize_buffer(const char *path
, const char *src
, size_t len
, struct strbuf
*dst
)
966 int ret
= convert_to_working_tree_internal(path
, src
, len
, dst
, 1);
971 return ret
| convert_to_git(path
, src
, len
, dst
, SAFE_CRLF_RENORMALIZE
);
974 /*****************************************************************
976 * Streaming conversion support
978 *****************************************************************/
980 typedef int (*filter_fn
)(struct stream_filter
*,
981 const char *input
, size_t *isize_p
,
982 char *output
, size_t *osize_p
);
983 typedef void (*free_fn
)(struct stream_filter
*);
985 struct stream_filter_vtbl
{
990 struct stream_filter
{
991 struct stream_filter_vtbl
*vtbl
;
994 static int null_filter_fn(struct stream_filter
*filter
,
995 const char *input
, size_t *isize_p
,
996 char *output
, size_t *osize_p
)
1001 return 0; /* we do not keep any states */
1003 if (*osize_p
< count
)
1006 memmove(output
, input
, count
);
1013 static void null_free_fn(struct stream_filter
*filter
)
1015 ; /* nothing -- null instances are shared */
1018 static struct stream_filter_vtbl null_vtbl
= {
1023 static struct stream_filter null_filter_singleton
= {
1027 int is_null_stream_filter(struct stream_filter
*filter
)
1029 return filter
== &null_filter_singleton
;
1037 struct lf_to_crlf_filter
{
1038 struct stream_filter filter
;
1039 unsigned has_held
:1;
1043 static int lf_to_crlf_filter_fn(struct stream_filter
*filter
,
1044 const char *input
, size_t *isize_p
,
1045 char *output
, size_t *osize_p
)
1047 size_t count
, o
= 0;
1048 struct lf_to_crlf_filter
*lf_to_crlf
= (struct lf_to_crlf_filter
*)filter
;
1051 * We may be holding onto the CR to see if it is followed by a
1052 * LF, in which case we would need to go to the main loop.
1053 * Otherwise, just emit it to the output stream.
1055 if (lf_to_crlf
->has_held
&& (lf_to_crlf
->held
!= '\r' || !input
)) {
1056 output
[o
++] = lf_to_crlf
->held
;
1057 lf_to_crlf
->has_held
= 0;
1060 /* We are told to drain */
1067 if (count
|| lf_to_crlf
->has_held
) {
1071 if (lf_to_crlf
->has_held
) {
1073 lf_to_crlf
->has_held
= 0;
1076 for (i
= 0; o
< *osize_p
&& i
< count
; i
++) {
1081 } else if (was_cr
) {
1083 * Previous round saw CR and it is not followed
1084 * by a LF; emit the CR before processing the
1085 * current character.
1091 * We may have consumed the last output slot,
1092 * in which case we need to break out of this
1093 * loop; hold the current character before
1096 if (*osize_p
<= o
) {
1097 lf_to_crlf
->has_held
= 1;
1098 lf_to_crlf
->held
= ch
;
1099 continue; /* break but increment i */
1114 if (!lf_to_crlf
->has_held
&& was_cr
) {
1115 lf_to_crlf
->has_held
= 1;
1116 lf_to_crlf
->held
= '\r';
1122 static void lf_to_crlf_free_fn(struct stream_filter
*filter
)
1127 static struct stream_filter_vtbl lf_to_crlf_vtbl
= {
1128 lf_to_crlf_filter_fn
,
1132 static struct stream_filter
*lf_to_crlf_filter(void)
1134 struct lf_to_crlf_filter
*lf_to_crlf
= xcalloc(1, sizeof(*lf_to_crlf
));
1136 lf_to_crlf
->filter
.vtbl
= &lf_to_crlf_vtbl
;
1137 return (struct stream_filter
*)lf_to_crlf
;
1143 #define FILTER_BUFFER 1024
1144 struct cascade_filter
{
1145 struct stream_filter filter
;
1146 struct stream_filter
*one
;
1147 struct stream_filter
*two
;
1148 char buf
[FILTER_BUFFER
];
1152 static int cascade_filter_fn(struct stream_filter
*filter
,
1153 const char *input
, size_t *isize_p
,
1154 char *output
, size_t *osize_p
)
1156 struct cascade_filter
*cas
= (struct cascade_filter
*) filter
;
1158 size_t sz
= *osize_p
;
1159 size_t to_feed
, remaining
;
1162 * input -- (one) --> buf -- (two) --> output
1164 while (filled
< sz
) {
1165 remaining
= sz
- filled
;
1167 /* do we already have something to feed two with? */
1168 if (cas
->ptr
< cas
->end
) {
1169 to_feed
= cas
->end
- cas
->ptr
;
1170 if (stream_filter(cas
->two
,
1171 cas
->buf
+ cas
->ptr
, &to_feed
,
1172 output
+ filled
, &remaining
))
1174 cas
->ptr
+= (cas
->end
- cas
->ptr
) - to_feed
;
1175 filled
= sz
- remaining
;
1179 /* feed one from upstream and have it emit into our buffer */
1180 to_feed
= input
? *isize_p
: 0;
1181 if (input
&& !to_feed
)
1183 remaining
= sizeof(cas
->buf
);
1184 if (stream_filter(cas
->one
,
1186 cas
->buf
, &remaining
))
1188 cas
->end
= sizeof(cas
->buf
) - remaining
;
1191 size_t fed
= *isize_p
- to_feed
;
1196 /* do we know that we drained one completely? */
1197 if (input
|| cas
->end
)
1200 /* tell two to drain; we have nothing more to give it */
1202 remaining
= sz
- filled
;
1203 if (stream_filter(cas
->two
,
1205 output
+ filled
, &remaining
))
1207 if (remaining
== (sz
- filled
))
1208 break; /* completely drained two */
1209 filled
= sz
- remaining
;
1215 static void cascade_free_fn(struct stream_filter
*filter
)
1217 struct cascade_filter
*cas
= (struct cascade_filter
*)filter
;
1218 free_stream_filter(cas
->one
);
1219 free_stream_filter(cas
->two
);
1223 static struct stream_filter_vtbl cascade_vtbl
= {
1228 static struct stream_filter
*cascade_filter(struct stream_filter
*one
,
1229 struct stream_filter
*two
)
1231 struct cascade_filter
*cascade
;
1233 if (!one
|| is_null_stream_filter(one
))
1235 if (!two
|| is_null_stream_filter(two
))
1238 cascade
= xmalloc(sizeof(*cascade
));
1241 cascade
->end
= cascade
->ptr
= 0;
1242 cascade
->filter
.vtbl
= &cascade_vtbl
;
1243 return (struct stream_filter
*)cascade
;
1249 #define IDENT_DRAINING (-1)
1250 #define IDENT_SKIPPING (-2)
1251 struct ident_filter
{
1252 struct stream_filter filter
;
1255 char ident
[45]; /* ": x40 $" */
1258 static int is_foreign_ident(const char *str
)
1262 if (!skip_prefix(str
, "$Id: ", &str
))
1264 for (i
= 0; str
[i
]; i
++) {
1265 if (isspace(str
[i
]) && str
[i
+1] != '$')
1271 static void ident_drain(struct ident_filter
*ident
, char **output_p
, size_t *osize_p
)
1273 size_t to_drain
= ident
->left
.len
;
1275 if (*osize_p
< to_drain
)
1276 to_drain
= *osize_p
;
1278 memcpy(*output_p
, ident
->left
.buf
, to_drain
);
1279 strbuf_remove(&ident
->left
, 0, to_drain
);
1280 *output_p
+= to_drain
;
1281 *osize_p
-= to_drain
;
1283 if (!ident
->left
.len
)
1287 static int ident_filter_fn(struct stream_filter
*filter
,
1288 const char *input
, size_t *isize_p
,
1289 char *output
, size_t *osize_p
)
1291 struct ident_filter
*ident
= (struct ident_filter
*)filter
;
1292 static const char head
[] = "$Id";
1295 /* drain upon eof */
1296 switch (ident
->state
) {
1298 strbuf_add(&ident
->left
, head
, ident
->state
);
1299 case IDENT_SKIPPING
:
1301 case IDENT_DRAINING
:
1302 ident_drain(ident
, &output
, osize_p
);
1307 while (*isize_p
|| (ident
->state
== IDENT_DRAINING
)) {
1310 if (ident
->state
== IDENT_DRAINING
) {
1311 ident_drain(ident
, &output
, osize_p
);
1320 if (ident
->state
== IDENT_SKIPPING
) {
1322 * Skipping until '$' or LF, but keeping them
1323 * in case it is a foreign ident.
1325 strbuf_addch(&ident
->left
, ch
);
1326 if (ch
!= '\n' && ch
!= '$')
1328 if (ch
== '$' && !is_foreign_ident(ident
->left
.buf
)) {
1329 strbuf_setlen(&ident
->left
, sizeof(head
) - 1);
1330 strbuf_addstr(&ident
->left
, ident
->ident
);
1332 ident
->state
= IDENT_DRAINING
;
1336 if (ident
->state
< sizeof(head
) &&
1337 head
[ident
->state
] == ch
) {
1343 strbuf_add(&ident
->left
, head
, ident
->state
);
1344 if (ident
->state
== sizeof(head
) - 1) {
1345 if (ch
!= ':' && ch
!= '$') {
1346 strbuf_addch(&ident
->left
, ch
);
1352 strbuf_addch(&ident
->left
, ch
);
1353 ident
->state
= IDENT_SKIPPING
;
1355 strbuf_addstr(&ident
->left
, ident
->ident
);
1356 ident
->state
= IDENT_DRAINING
;
1361 strbuf_addch(&ident
->left
, ch
);
1362 ident
->state
= IDENT_DRAINING
;
1367 static void ident_free_fn(struct stream_filter
*filter
)
1369 struct ident_filter
*ident
= (struct ident_filter
*)filter
;
1370 strbuf_release(&ident
->left
);
1374 static struct stream_filter_vtbl ident_vtbl
= {
1379 static struct stream_filter
*ident_filter(const unsigned char *sha1
)
1381 struct ident_filter
*ident
= xmalloc(sizeof(*ident
));
1383 xsnprintf(ident
->ident
, sizeof(ident
->ident
),
1384 ": %s $", sha1_to_hex(sha1
));
1385 strbuf_init(&ident
->left
, 0);
1386 ident
->filter
.vtbl
= &ident_vtbl
;
1388 return (struct stream_filter
*)ident
;
1392 * Return an appropriately constructed filter for the path, or NULL if
1393 * the contents cannot be filtered without reading the whole thing
1396 * Note that you would be crazy to set CRLF, smuge/clean or ident to a
1397 * large binary blob you would want us not to slurp into the memory!
1399 struct stream_filter
*get_stream_filter(const char *path
, const unsigned char *sha1
)
1401 struct conv_attrs ca
;
1402 struct stream_filter
*filter
= NULL
;
1404 convert_attrs(&ca
, path
);
1405 if (ca
.drv
&& (ca
.drv
->smudge
|| ca
.drv
->clean
))
1408 if (ca
.crlf_action
== CRLF_AUTO
|| ca
.crlf_action
== CRLF_AUTO_CRLF
)
1412 filter
= ident_filter(sha1
);
1414 if (output_eol(ca
.crlf_action
) == EOL_CRLF
)
1415 filter
= cascade_filter(filter
, lf_to_crlf_filter());
1417 filter
= cascade_filter(filter
, &null_filter_singleton
);
1422 void free_stream_filter(struct stream_filter
*filter
)
1424 filter
->vtbl
->free(filter
);
1427 int stream_filter(struct stream_filter
*filter
,
1428 const char *input
, size_t *isize_p
,
1429 char *output
, size_t *osize_p
)
1431 return filter
->vtbl
->filter(filter
, input
, isize_p
, output
, osize_p
);