Sync with 'maint'
[git/gitster.git] / builtin / credential-cache--daemon.c
blob4952b225477227862945e5f987582a76abdaf07e
1 #include "builtin.h"
2 #include "abspath.h"
3 #include "gettext.h"
4 #include "object-file.h"
5 #include "parse-options.h"
7 #ifndef NO_UNIX_SOCKETS
9 #include "config.h"
10 #include "tempfile.h"
11 #include "credential.h"
12 #include "unix-socket.h"
14 struct credential_cache_entry {
15 struct credential item;
16 timestamp_t expiration;
18 static struct credential_cache_entry *entries;
19 static int entries_nr;
20 static int entries_alloc;
22 static void cache_credential(struct credential *c, int timeout)
24 struct credential_cache_entry *e;
26 ALLOC_GROW(entries, entries_nr + 1, entries_alloc);
27 e = &entries[entries_nr++];
29 /* take ownership of pointers */
30 memcpy(&e->item, c, sizeof(*c));
31 memset(c, 0, sizeof(*c));
32 e->expiration = time(NULL) + timeout;
35 static struct credential_cache_entry *lookup_credential(const struct credential *c)
37 int i;
38 for (i = 0; i < entries_nr; i++) {
39 struct credential *e = &entries[i].item;
40 if (credential_match(c, e, 0))
41 return &entries[i];
43 return NULL;
46 static void remove_credential(const struct credential *c, int match_password)
48 struct credential_cache_entry *e;
50 int i;
51 for (i = 0; i < entries_nr; i++) {
52 e = &entries[i];
53 if (credential_match(c, &e->item, match_password))
54 e->expiration = 0;
58 static timestamp_t check_expirations(void)
60 static timestamp_t wait_for_entry_until;
61 int i = 0;
62 timestamp_t now = time(NULL);
63 timestamp_t next = TIME_MAX;
66 * Initially give the client 30 seconds to actually contact us
67 * and store a credential before we decide there's no point in
68 * keeping the daemon around.
70 if (!wait_for_entry_until)
71 wait_for_entry_until = now + 30;
73 while (i < entries_nr) {
74 if (entries[i].expiration <= now) {
75 entries_nr--;
76 credential_clear(&entries[i].item);
77 if (i != entries_nr)
78 memcpy(&entries[i], &entries[entries_nr], sizeof(*entries));
80 * Stick around 30 seconds in case a new credential
81 * shows up (e.g., because we just removed a failed
82 * one, and we will soon get the correct one).
84 wait_for_entry_until = now + 30;
86 else {
87 if (entries[i].expiration < next)
88 next = entries[i].expiration;
89 i++;
93 if (!entries_nr) {
94 if (wait_for_entry_until <= now)
95 return 0;
96 next = wait_for_entry_until;
99 return next - now;
102 static int read_request(FILE *fh, struct credential *c,
103 struct strbuf *action, int *timeout)
105 static struct strbuf item = STRBUF_INIT;
106 const char *p;
108 strbuf_getline_lf(&item, fh);
109 if (!skip_prefix(item.buf, "action=", &p))
110 return error("client sent bogus action line: %s", item.buf);
111 strbuf_addstr(action, p);
113 strbuf_getline_lf(&item, fh);
114 if (!skip_prefix(item.buf, "timeout=", &p))
115 return error("client sent bogus timeout line: %s", item.buf);
116 *timeout = atoi(p);
118 credential_set_all_capabilities(c, CREDENTIAL_OP_INITIAL);
120 if (credential_read(c, fh, CREDENTIAL_OP_HELPER) < 0)
121 return -1;
122 return 0;
125 static void serve_one_client(FILE *in, FILE *out)
127 struct credential c = CREDENTIAL_INIT;
128 struct strbuf action = STRBUF_INIT;
129 int timeout = -1;
131 if (read_request(in, &c, &action, &timeout) < 0)
132 /* ignore error */ ;
133 else if (!strcmp(action.buf, "get")) {
134 struct credential_cache_entry *e = lookup_credential(&c);
135 if (e) {
136 e->item.capa_authtype.request_initial = 1;
137 e->item.capa_authtype.request_helper = 1;
139 fprintf(out, "capability[]=authtype\n");
140 if (e->item.username)
141 fprintf(out, "username=%s\n", e->item.username);
142 if (e->item.password)
143 fprintf(out, "password=%s\n", e->item.password);
144 if (credential_has_capability(&c.capa_authtype, CREDENTIAL_OP_HELPER) && e->item.authtype)
145 fprintf(out, "authtype=%s\n", e->item.authtype);
146 if (credential_has_capability(&c.capa_authtype, CREDENTIAL_OP_HELPER) && e->item.credential)
147 fprintf(out, "credential=%s\n", e->item.credential);
148 if (e->item.password_expiry_utc != TIME_MAX)
149 fprintf(out, "password_expiry_utc=%"PRItime"\n",
150 e->item.password_expiry_utc);
151 if (e->item.oauth_refresh_token)
152 fprintf(out, "oauth_refresh_token=%s\n",
153 e->item.oauth_refresh_token);
156 else if (!strcmp(action.buf, "exit")) {
158 * It's important that we clean up our socket first, and then
159 * signal the client only once we have finished the cleanup.
160 * Calling exit() directly does this, because we clean up in
161 * our atexit() handler, and then signal the client when our
162 * process actually ends, which closes the socket and gives
163 * them EOF.
165 exit(0);
167 else if (!strcmp(action.buf, "erase"))
168 remove_credential(&c, 1);
169 else if (!strcmp(action.buf, "store")) {
170 if (timeout < 0)
171 warning("cache client didn't specify a timeout");
172 else if ((!c.username || !c.password) && (!c.authtype && !c.credential))
173 warning("cache client gave us a partial credential");
174 else if (c.ephemeral)
175 warning("not storing ephemeral credential");
176 else {
177 remove_credential(&c, 0);
178 cache_credential(&c, timeout);
181 else
182 warning("cache client sent unknown action: %s", action.buf);
184 credential_clear(&c);
185 strbuf_release(&action);
188 static int serve_cache_loop(int fd)
190 struct pollfd pfd;
191 timestamp_t wakeup;
193 wakeup = check_expirations();
194 if (!wakeup)
195 return 0;
197 pfd.fd = fd;
198 pfd.events = POLLIN;
199 if (poll(&pfd, 1, 1000 * wakeup) < 0) {
200 if (errno != EINTR)
201 die_errno("poll failed");
202 return 1;
205 if (pfd.revents & POLLIN) {
206 int client, client2;
207 FILE *in, *out;
209 client = accept(fd, NULL, NULL);
210 if (client < 0) {
211 warning_errno("accept failed");
212 return 1;
214 client2 = dup(client);
215 if (client2 < 0) {
216 warning_errno("dup failed");
217 close(client);
218 return 1;
221 in = xfdopen(client, "r");
222 out = xfdopen(client2, "w");
223 serve_one_client(in, out);
224 fclose(in);
225 fclose(out);
227 return 1;
230 static void serve_cache(const char *socket_path, int debug)
232 struct unix_stream_listen_opts opts = UNIX_STREAM_LISTEN_OPTS_INIT;
233 int fd;
235 fd = unix_stream_listen(socket_path, &opts);
236 if (fd < 0)
237 die_errno("unable to bind to '%s'", socket_path);
239 printf("ok\n");
240 fclose(stdout);
241 if (!debug) {
242 if (!freopen("/dev/null", "w", stderr))
243 die_errno("unable to point stderr to /dev/null");
246 while (serve_cache_loop(fd))
247 ; /* nothing */
249 close(fd);
252 static const char permissions_advice[] = N_(
253 "The permissions on your socket directory are too loose; other\n"
254 "users may be able to read your cached credentials. Consider running:\n"
255 "\n"
256 " chmod 0700 %s");
257 static void init_socket_directory(const char *path)
259 struct stat st;
260 char *path_copy = xstrdup(path);
261 char *dir = dirname(path_copy);
263 if (!stat(dir, &st)) {
264 if (st.st_mode & 077)
265 die(_(permissions_advice), dir);
266 } else {
268 * We must be sure to create the directory with the correct mode,
269 * not just chmod it after the fact; otherwise, there is a race
270 * condition in which somebody can chdir to it, sleep, then try to open
271 * our protected socket.
273 if (safe_create_leading_directories_const(dir) < 0)
274 die_errno("unable to create directories for '%s'", dir);
275 if (mkdir(dir, 0700) < 0)
276 die_errno("unable to mkdir '%s'", dir);
279 if (chdir(dir))
281 * We don't actually care what our cwd is; we chdir here just to
282 * be a friendly daemon and avoid tying up our original cwd.
283 * If this fails, it's OK to just continue without that benefit.
287 free(path_copy);
290 int cmd_credential_cache_daemon(int argc, const char **argv, const char *prefix)
292 struct tempfile *socket_file;
293 const char *socket_path;
294 int ignore_sighup = 0;
295 static const char *usage[] = {
296 "git credential-cache--daemon [--debug] <socket-path>",
297 NULL
299 int debug = 0;
300 const struct option options[] = {
301 OPT_BOOL(0, "debug", &debug,
302 N_("print debugging messages to stderr")),
303 OPT_END()
306 git_config_get_bool("credentialcache.ignoresighup", &ignore_sighup);
308 argc = parse_options(argc, argv, prefix, options, usage, 0);
309 socket_path = argv[0];
311 if (!have_unix_sockets())
312 die(_("credential-cache--daemon unavailable; no unix socket support"));
313 if (!socket_path)
314 usage_with_options(usage, options);
316 if (!is_absolute_path(socket_path))
317 die("socket directory must be an absolute path");
319 init_socket_directory(socket_path);
320 socket_file = register_tempfile(socket_path);
322 if (ignore_sighup)
323 signal(SIGHUP, SIG_IGN);
325 serve_cache(socket_path, debug);
326 delete_tempfile(&socket_file);
328 return 0;
331 #else
333 int cmd_credential_cache_daemon(int argc, const char **argv, const char *prefix)
335 const char * const usage[] = {
336 "git credential-cache--daemon [--debug] <socket-path>",
338 "credential-cache--daemon is disabled in this build of Git",
339 NULL
341 struct option options[] = { OPT_END() };
343 argc = parse_options(argc, argv, prefix, options, usage, 0);
344 die(_("credential-cache--daemon unavailable; no unix socket support"));
347 #endif /* NO_UNIX_SOCKET */