2 * GIT - The information manager from hell
4 * Copyright (C) Linus Torvalds, 2005
9 * Default to not allowing changes to the list of files. The
10 * tool doesn't actually care, but this makes it harder to add
11 * files to the revision control by mistake by doing something
12 * like "update-cache *" and suddenly having all the object
13 * files be revision controlled.
15 static int allow_add
= 0, allow_remove
= 0;
17 static int index_fd(const char *path
, int namelen
, struct cache_entry
*ce
, int fd
, struct stat
*st
)
20 unsigned long size
= st
->st_size
;
21 int max_out_bytes
= namelen
+ size
+ 200;
22 void *out
= malloc(max_out_bytes
);
23 void *metadata
= malloc(namelen
+ 200);
29 in
= mmap(NULL
, size
, PROT_READ
, MAP_PRIVATE
, fd
, 0);
31 if (!out
|| (int)(long)in
== -1)
34 memset(&stream
, 0, sizeof(stream
));
35 deflateInit(&stream
, Z_BEST_COMPRESSION
);
38 * ASCII size + nul byte
40 stream
.next_in
= metadata
;
41 stream
.avail_in
= 1+sprintf(metadata
, "blob %lu", size
);
42 stream
.next_out
= out
;
43 stream
.avail_out
= max_out_bytes
;
44 while (deflate(&stream
, 0) == Z_OK
)
51 stream
.avail_in
= size
;
52 while (deflate(&stream
, Z_FINISH
) == Z_OK
)
58 SHA1_Update(&c
, out
, stream
.total_out
);
59 SHA1_Final(ce
->sha1
, &c
);
61 return write_sha1_buffer(ce
->sha1
, out
, stream
.total_out
);
65 * This only updates the "non-critical" parts of the directory
66 * cache, ie the parts that aren't tracked by GIT, and only used
67 * to validate the cache.
69 static void fill_stat_cache_info(struct cache_entry
*ce
, struct stat
*st
)
71 ce
->ctime
.sec
= st
->st_ctime
;
72 ce
->ctime
.nsec
= st
->st_ctim
.tv_nsec
;
73 ce
->mtime
.sec
= st
->st_mtime
;
74 ce
->mtime
.nsec
= st
->st_mtim
.tv_nsec
;
75 ce
->st_dev
= st
->st_dev
;
76 ce
->st_ino
= st
->st_ino
;
77 ce
->st_uid
= st
->st_uid
;
78 ce
->st_gid
= st
->st_gid
;
81 static int add_file_to_cache(char *path
)
84 struct cache_entry
*ce
;
88 fd
= open(path
, O_RDONLY
);
90 if (errno
== ENOENT
) {
92 return remove_file_from_cache(path
);
96 if (fstat(fd
, &st
) < 0) {
100 namelen
= strlen(path
);
101 size
= cache_entry_size(namelen
);
104 memcpy(ce
->name
, path
, namelen
);
105 fill_stat_cache_info(ce
, &st
);
106 ce
->st_mode
= st
.st_mode
;
107 ce
->st_size
= st
.st_size
;
108 ce
->namelen
= namelen
;
110 if (index_fd(path
, namelen
, ce
, fd
, &st
) < 0)
113 return add_cache_entry(ce
, allow_add
);
116 static int match_data(int fd
, void *buffer
, unsigned long size
)
120 int ret
= read(fd
, compare
, sizeof(compare
));
122 if (ret
<= 0 || ret
> size
|| memcmp(buffer
, compare
, ret
))
130 static int compare_data(struct cache_entry
*ce
, unsigned long expected_size
)
133 int fd
= open(ce
->name
, O_RDONLY
);
140 buffer
= read_sha1_file(ce
->sha1
, type
, &size
);
142 if (size
== expected_size
&& !strcmp(type
, "blob"))
143 match
= match_data(fd
, buffer
, size
);
152 * "refresh" does not calculate a new sha1 file or bring the
153 * cache up-to-date for mode/content changes. But what it
154 * _does_ do is to "re-match" the stat information of a file
155 * with the cache, so that you can refresh the cache for a
156 * file that hasn't been changed but where the stat entry is
159 * For example, you'd want to do this after doing a "read-tree",
160 * to link up the stat cache details with the proper files.
162 static struct cache_entry
*refresh_entry(struct cache_entry
*ce
)
165 struct cache_entry
*updated
;
168 if (stat(ce
->name
, &st
) < 0)
171 changed
= cache_match_stat(ce
, &st
);
176 * If the mode has changed, there's no point in trying
177 * to refresh the entry - it's not going to match
179 if (changed
& MODE_CHANGED
)
182 if (compare_data(ce
, st
.st_size
))
186 updated
= malloc(size
);
187 memcpy(updated
, ce
, size
);
188 fill_stat_cache_info(updated
, &st
);
189 updated
->st_size
= st
.st_size
;
193 static void refresh_cache(void)
197 for (i
= 0; i
< active_nr
; i
++) {
198 struct cache_entry
*ce
= active_cache
[i
];
199 struct cache_entry
*new = refresh_entry(ce
);
202 printf("%s: needs update\n", ce
->name
);
205 active_cache
[i
] = new;
210 * We fundamentally don't like some paths: we don't want
211 * dot or dot-dot anywhere, and in fact, we don't even want
212 * any other dot-files (.git or anything else). They
213 * are hidden, for chist sake.
215 * Also, we don't want double slashes or slashes at the
216 * end that can make pathnames ambiguous.
218 static int verify_path(char *path
)
229 if (c
!= '/' && c
!= '.' && c
!= '\0')
237 static int remove_lock
= 0;
239 static void remove_lock_file(void)
242 unlink(".git/index.lock");
245 int main(int argc
, char **argv
)
247 int i
, newfd
, entries
;
248 int allow_options
= 1;
250 newfd
= open(".git/index.lock", O_RDWR
| O_CREAT
| O_EXCL
, 0600);
252 die("unable to create new cachefile");
254 atexit(remove_lock_file
);
257 entries
= read_cache();
259 die("cache corrupted");
261 for (i
= 1 ; i
< argc
; i
++) {
262 char *path
= argv
[i
];
264 if (allow_options
&& *path
== '-') {
265 if (!strcmp(path
, "--")) {
269 if (!strcmp(path
, "--add")) {
273 if (!strcmp(path
, "--remove")) {
277 if (!strcmp(path
, "--refresh")) {
281 die("unknown option %s", path
);
283 if (!verify_path(path
)) {
284 fprintf(stderr
, "Ignoring path %s\n", argv
[i
]);
287 if (add_file_to_cache(path
))
288 die("Unable to add %s to database", path
);
290 if (write_cache(newfd
, active_cache
, active_nr
) ||
291 rename(".git/index.lock", ".git/index"))
292 die("Unable to write new cachefile");