archive.c: avoid access to the_index
[git.git] / resolve-undo.c
blobd2e2d22b7fb6a087eced2320fbd766bfc8ea1efc
1 #include "cache.h"
2 #include "dir.h"
3 #include "resolve-undo.h"
4 #include "string-list.h"
6 /* The only error case is to run out of memory in string-list */
7 void record_resolve_undo(struct index_state *istate, struct cache_entry *ce)
9 struct string_list_item *lost;
10 struct resolve_undo_info *ui;
11 struct string_list *resolve_undo;
12 int stage = ce_stage(ce);
14 if (!stage)
15 return;
17 if (!istate->resolve_undo) {
18 resolve_undo = xcalloc(1, sizeof(*resolve_undo));
19 resolve_undo->strdup_strings = 1;
20 istate->resolve_undo = resolve_undo;
22 resolve_undo = istate->resolve_undo;
23 lost = string_list_insert(resolve_undo, ce->name);
24 if (!lost->util)
25 lost->util = xcalloc(1, sizeof(*ui));
26 ui = lost->util;
27 oidcpy(&ui->oid[stage - 1], &ce->oid);
28 ui->mode[stage - 1] = ce->ce_mode;
31 void resolve_undo_write(struct strbuf *sb, struct string_list *resolve_undo)
33 struct string_list_item *item;
34 for_each_string_list_item(item, resolve_undo) {
35 struct resolve_undo_info *ui = item->util;
36 int i;
38 if (!ui)
39 continue;
40 strbuf_addstr(sb, item->string);
41 strbuf_addch(sb, 0);
42 for (i = 0; i < 3; i++)
43 strbuf_addf(sb, "%o%c", ui->mode[i], 0);
44 for (i = 0; i < 3; i++) {
45 if (!ui->mode[i])
46 continue;
47 strbuf_add(sb, ui->oid[i].hash, the_hash_algo->rawsz);
52 struct string_list *resolve_undo_read(const char *data, unsigned long size)
54 struct string_list *resolve_undo;
55 size_t len;
56 char *endptr;
57 int i;
58 const unsigned rawsz = the_hash_algo->rawsz;
60 resolve_undo = xcalloc(1, sizeof(*resolve_undo));
61 resolve_undo->strdup_strings = 1;
63 while (size) {
64 struct string_list_item *lost;
65 struct resolve_undo_info *ui;
67 len = strlen(data) + 1;
68 if (size <= len)
69 goto error;
70 lost = string_list_insert(resolve_undo, data);
71 if (!lost->util)
72 lost->util = xcalloc(1, sizeof(*ui));
73 ui = lost->util;
74 size -= len;
75 data += len;
77 for (i = 0; i < 3; i++) {
78 ui->mode[i] = strtoul(data, &endptr, 8);
79 if (!endptr || endptr == data || *endptr)
80 goto error;
81 len = (endptr + 1) - (char*)data;
82 if (size <= len)
83 goto error;
84 size -= len;
85 data += len;
88 for (i = 0; i < 3; i++) {
89 if (!ui->mode[i])
90 continue;
91 if (size < rawsz)
92 goto error;
93 oidread(&ui->oid[i], (const unsigned char *)data);
94 size -= rawsz;
95 data += rawsz;
98 return resolve_undo;
100 error:
101 string_list_clear(resolve_undo, 1);
102 error("Index records invalid resolve-undo information");
103 return NULL;
106 void resolve_undo_clear_index(struct index_state *istate)
108 struct string_list *resolve_undo = istate->resolve_undo;
109 if (!resolve_undo)
110 return;
111 string_list_clear(resolve_undo, 1);
112 free(resolve_undo);
113 istate->resolve_undo = NULL;
114 istate->cache_changed |= RESOLVE_UNDO_CHANGED;
117 int unmerge_index_entry_at(struct index_state *istate, int pos)
119 const struct cache_entry *ce;
120 struct string_list_item *item;
121 struct resolve_undo_info *ru;
122 int i, err = 0, matched;
123 char *name;
125 if (!istate->resolve_undo)
126 return pos;
128 ce = istate->cache[pos];
129 if (ce_stage(ce)) {
130 /* already unmerged */
131 while ((pos < istate->cache_nr) &&
132 ! strcmp(istate->cache[pos]->name, ce->name))
133 pos++;
134 return pos - 1; /* return the last entry processed */
136 item = string_list_lookup(istate->resolve_undo, ce->name);
137 if (!item)
138 return pos;
139 ru = item->util;
140 if (!ru)
141 return pos;
142 matched = ce->ce_flags & CE_MATCHED;
143 name = xstrdup(ce->name);
144 remove_index_entry_at(istate, pos);
145 for (i = 0; i < 3; i++) {
146 struct cache_entry *nce;
147 if (!ru->mode[i])
148 continue;
149 nce = make_cache_entry(istate,
150 ru->mode[i],
151 &ru->oid[i],
152 name, i + 1, 0);
153 if (matched)
154 nce->ce_flags |= CE_MATCHED;
155 if (add_index_entry(istate, nce, ADD_CACHE_OK_TO_ADD)) {
156 err = 1;
157 error("cannot unmerge '%s'", name);
160 free(name);
161 if (err)
162 return pos;
163 free(ru);
164 item->util = NULL;
165 return unmerge_index_entry_at(istate, pos);
168 void unmerge_marked_index(struct index_state *istate)
170 int i;
172 if (!istate->resolve_undo)
173 return;
175 for (i = 0; i < istate->cache_nr; i++) {
176 const struct cache_entry *ce = istate->cache[i];
177 if (ce->ce_flags & CE_MATCHED)
178 i = unmerge_index_entry_at(istate, i);
182 void unmerge_index(struct index_state *istate, const struct pathspec *pathspec)
184 int i;
186 if (!istate->resolve_undo)
187 return;
189 for (i = 0; i < istate->cache_nr; i++) {
190 const struct cache_entry *ce = istate->cache[i];
191 if (!ce_path_match(&the_index, ce, pathspec, NULL))
192 continue;
193 i = unmerge_index_entry_at(istate, i);