Merge branch 'jc/maint-fix-test-perm' into maint
[tgit.git] / resolve-undo.c
blob0f50ee0484776c545e632cde69b0646629a3b190
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(ce->name, resolve_undo);
24 if (!lost->util)
25 lost->util = xcalloc(1, sizeof(*ui));
26 ui = lost->util;
27 hashcpy(ui->sha1[stage - 1], ce->sha1);
28 ui->mode[stage - 1] = ce->ce_mode;
31 static int write_one(struct string_list_item *item, void *cbdata)
33 struct strbuf *sb = cbdata;
34 struct resolve_undo_info *ui = item->util;
35 int i;
37 if (!ui)
38 return 0;
39 strbuf_addstr(sb, item->string);
40 strbuf_addch(sb, 0);
41 for (i = 0; i < 3; i++)
42 strbuf_addf(sb, "%o%c", ui->mode[i], 0);
43 for (i = 0; i < 3; i++) {
44 if (!ui->mode[i])
45 continue;
46 strbuf_add(sb, ui->sha1[i], 20);
48 return 0;
51 void resolve_undo_write(struct strbuf *sb, struct string_list *resolve_undo)
53 for_each_string_list(write_one, resolve_undo, sb);
56 struct string_list *resolve_undo_read(const char *data, unsigned long size)
58 struct string_list *resolve_undo;
59 size_t len;
60 char *endptr;
61 int i;
63 resolve_undo = xcalloc(1, sizeof(*resolve_undo));
64 resolve_undo->strdup_strings = 1;
66 while (size) {
67 struct string_list_item *lost;
68 struct resolve_undo_info *ui;
70 len = strlen(data) + 1;
71 if (size <= len)
72 goto error;
73 lost = string_list_insert(data, resolve_undo);
74 if (!lost->util)
75 lost->util = xcalloc(1, sizeof(*ui));
76 ui = lost->util;
77 size -= len;
78 data += len;
80 for (i = 0; i < 3; i++) {
81 ui->mode[i] = strtoul(data, &endptr, 8);
82 if (!endptr || endptr == data || *endptr)
83 goto error;
84 len = (endptr + 1) - (char*)data;
85 if (size <= len)
86 goto error;
87 size -= len;
88 data += len;
91 for (i = 0; i < 3; i++) {
92 if (!ui->mode[i])
93 continue;
94 if (size < 20)
95 goto error;
96 hashcpy(ui->sha1[i], (const unsigned char *)data);
97 size -= 20;
98 data += 20;
101 return resolve_undo;
103 error:
104 string_list_clear(resolve_undo, 1);
105 error("Index records invalid resolve-undo information");
106 return NULL;
109 void resolve_undo_clear_index(struct index_state *istate)
111 struct string_list *resolve_undo = istate->resolve_undo;
112 if (!resolve_undo)
113 return;
114 string_list_clear(resolve_undo, 1);
115 free(resolve_undo);
116 istate->resolve_undo = NULL;
117 istate->cache_changed = 1;
120 int unmerge_index_entry_at(struct index_state *istate, int pos)
122 struct cache_entry *ce;
123 struct string_list_item *item;
124 struct resolve_undo_info *ru;
125 int i, err = 0;
127 if (!istate->resolve_undo)
128 return pos;
130 ce = istate->cache[pos];
131 if (ce_stage(ce)) {
132 /* already unmerged */
133 while ((pos < istate->cache_nr) &&
134 ! strcmp(istate->cache[pos]->name, ce->name))
135 pos++;
136 return pos - 1; /* return the last entry processed */
138 item = string_list_lookup(ce->name, istate->resolve_undo);
139 if (!item)
140 return pos;
141 ru = item->util;
142 if (!ru)
143 return pos;
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(ru->mode[i], ru->sha1[i],
150 ce->name, i + 1, 0);
151 if (add_index_entry(istate, nce, ADD_CACHE_OK_TO_ADD)) {
152 err = 1;
153 error("cannot unmerge '%s'", ce->name);
156 if (err)
157 return pos;
158 free(ru);
159 item->util = NULL;
160 return unmerge_index_entry_at(istate, pos);
163 void unmerge_index(struct index_state *istate, const char **pathspec)
165 int i;
167 if (!istate->resolve_undo)
168 return;
170 for (i = 0; i < istate->cache_nr; i++) {
171 struct cache_entry *ce = istate->cache[i];
172 if (!match_pathspec(pathspec, ce->name, ce_namelen(ce), 0, NULL))
173 continue;
174 i = unmerge_index_entry_at(istate, i);