Merge branch 'jh/memihash-opt' into next
[git] / builtin / hash-object.c
1 /*
2  * GIT - The information manager from hell
3  *
4  * Copyright (C) Linus Torvalds, 2005
5  * Copyright (C) Junio C Hamano, 2005
6  */
7 #include "builtin.h"
8 #include "blob.h"
9 #include "quote.h"
10 #include "parse-options.h"
11 #include "exec_cmd.h"
12
13 /*
14  * This is to create corrupt objects for debugging and as such it
15  * needs to bypass the data conversion performed by, and the type
16  * limitation imposed by, index_fd() and its callees.
17  */
18 static int hash_literally(unsigned char *sha1, int fd, const char *type, unsigned flags)
19 {
20         struct strbuf buf = STRBUF_INIT;
21         int ret;
22
23         if (strbuf_read(&buf, fd, 4096) < 0)
24                 ret = -1;
25         else
26                 ret = hash_sha1_file_literally(buf.buf, buf.len, type, sha1, flags);
27         strbuf_release(&buf);
28         return ret;
29 }
30
31 static void hash_fd(int fd, const char *type, const char *path, unsigned flags,
32                     int literally)
33 {
34         struct stat st;
35         unsigned char sha1[20];
36
37         if (fstat(fd, &st) < 0 ||
38             (literally
39              ? hash_literally(sha1, fd, type, flags)
40              : index_fd(sha1, fd, &st, type_from_string(type), path, flags)))
41                 die((flags & HASH_WRITE_OBJECT)
42                     ? "Unable to add %s to database"
43                     : "Unable to hash %s", path);
44         printf("%s\n", sha1_to_hex(sha1));
45         maybe_flush_or_die(stdout, "hash to stdout");
46 }
47
48 static void hash_object(const char *path, const char *type, const char *vpath,
49                         unsigned flags, int literally)
50 {
51         int fd;
52         fd = open(path, O_RDONLY);
53         if (fd < 0)
54                 die_errno("Cannot open '%s'", path);
55         hash_fd(fd, type, vpath, flags, literally);
56 }
57
58 static void hash_stdin_paths(const char *type, int no_filters, unsigned flags,
59                              int literally)
60 {
61         struct strbuf buf = STRBUF_INIT;
62         struct strbuf unquoted = STRBUF_INIT;
63
64         while (strbuf_getline(&buf, stdin) != EOF) {
65                 if (buf.buf[0] == '"') {
66                         strbuf_reset(&unquoted);
67                         if (unquote_c_style(&unquoted, buf.buf, NULL))
68                                 die("line is badly quoted");
69                         strbuf_swap(&buf, &unquoted);
70                 }
71                 hash_object(buf.buf, type, no_filters ? NULL : buf.buf, flags,
72                             literally);
73         }
74         strbuf_release(&buf);
75         strbuf_release(&unquoted);
76 }
77
78 int cmd_hash_object(int argc, const char **argv, const char *prefix)
79 {
80         static const char * const hash_object_usage[] = {
81                 N_("git hash-object [-t <type>] [-w] [--path=<file> | --no-filters] [--stdin] [--] <file>..."),
82                 N_("git hash-object  --stdin-paths"),
83                 NULL
84         };
85         const char *type = blob_type;
86         int hashstdin = 0;
87         int stdin_paths = 0;
88         int no_filters = 0;
89         int literally = 0;
90         int nongit = 0;
91         unsigned flags = HASH_FORMAT_CHECK;
92         const char *vpath = NULL;
93         const struct option hash_object_options[] = {
94                 OPT_STRING('t', NULL, &type, N_("type"), N_("object type")),
95                 OPT_BIT('w', NULL, &flags, N_("write the object into the object database"),
96                         HASH_WRITE_OBJECT),
97                 OPT_COUNTUP( 0 , "stdin", &hashstdin, N_("read the object from stdin")),
98                 OPT_BOOL( 0 , "stdin-paths", &stdin_paths, N_("read file names from stdin")),
99                 OPT_BOOL( 0 , "no-filters", &no_filters, N_("store file as is without filters")),
100                 OPT_BOOL( 0, "literally", &literally, N_("just hash any random garbage to create corrupt objects for debugging Git")),
101                 OPT_STRING( 0 , "path", &vpath, N_("file"), N_("process file as it were from this path")),
102                 OPT_END()
103         };
104         int i;
105         const char *errstr = NULL;
106
107         argc = parse_options(argc, argv, NULL, hash_object_options,
108                              hash_object_usage, 0);
109
110         if (flags & HASH_WRITE_OBJECT)
111                 prefix = setup_git_directory();
112         else
113                 prefix = setup_git_directory_gently(&nongit);
114
115         if (vpath && prefix)
116                 vpath = xstrdup(prefix_filename(prefix, vpath));
117
118         git_config(git_default_config, NULL);
119
120         if (stdin_paths) {
121                 if (hashstdin)
122                         errstr = "Can't use --stdin-paths with --stdin";
123                 else if (argc)
124                         errstr = "Can't specify files with --stdin-paths";
125                 else if (vpath)
126                         errstr = "Can't use --stdin-paths with --path";
127         }
128         else {
129                 if (hashstdin > 1)
130                         errstr = "Multiple --stdin arguments are not supported";
131                 if (vpath && no_filters)
132                         errstr = "Can't use --path with --no-filters";
133         }
134
135         if (errstr) {
136                 error("%s", errstr);
137                 usage_with_options(hash_object_usage, hash_object_options);
138         }
139
140         if (hashstdin)
141                 hash_fd(0, type, vpath, flags, literally);
142
143         for (i = 0 ; i < argc; i++) {
144                 const char *arg = argv[i];
145                 char *to_free = NULL;
146
147                 if (prefix)
148                         arg = to_free = prefix_filename(prefix, arg);
149                 hash_object(arg, type, no_filters ? NULL : vpath ? vpath : arg,
150                             flags, literally);
151                 free(to_free);
152         }
153
154         if (stdin_paths)
155                 hash_stdin_paths(type, no_filters, flags, literally);
156
157         return 0;
158 }