builtin/credential-cache.c

   1 #include "builtin.h"
   2 #include "parse-options.h"
   3
   4 #ifndef NO_UNIX_SOCKETS
   5
   6 #include "credential.h"
   7 #include "string-list.h"
   8 #include "unix-socket.h"
   9 #include "run-command.h"
  10
  11 #define FLAG_SPAWN 0x1
  12 #define FLAG_RELAY 0x2
  13
  14 #ifdef GIT_WINDOWS_NATIVE
  15
  16 static int connection_closed(int error)
  17 {
  18         return (error == EINVAL);
  19 }
  20
  21 static int connection_fatally_broken(int error)
  22 {
  23         return (error != ENOENT) && (error != ENETDOWN);
  24 }
  25
  26 #else
  27
  28 static int connection_closed(int error)
  29 {
  30         return (error == ECONNRESET);
  31 }
  32
  33 static int connection_fatally_broken(int error)
  34 {
  35         return (error != ENOENT) && (error != ECONNREFUSED);
  36 }
  37
  38 #endif
  39
  40 static int send_request(const char *socket, const struct strbuf *out)
  41 {
  42         int got_data = 0;
  43         int fd = unix_stream_connect(socket, 0);
  44
  45         if (fd < 0)
  46                 return -1;
  47
  48         if (write_in_full(fd, out->buf, out->len) < 0)
  49                 die_errno("unable to write to cache daemon");
  50         shutdown(fd, SHUT_WR);
  51
  52         while (1) {
  53                 char in[1024];
  54                 int r;
  55
  56                 r = read_in_full(fd, in, sizeof(in));
  57                 if (r == 0 || (r < 0 && connection_closed(errno)))
  58                         break;
  59                 if (r < 0)
  60                         die_errno("read error from cache daemon");
  61                 write_or_die(1, in, r);
  62                 got_data = 1;
  63         }
  64         close(fd);
  65         return got_data;
  66 }
  67
  68 static void spawn_daemon(const char *socket)
  69 {
  70         struct child_process daemon = CHILD_PROCESS_INIT;
  71         char buf[128];
  72         int r;
  73
  74         strvec_pushl(&daemon.args,
  75                      "credential-cache--daemon", socket,
  76                      NULL);
  77         daemon.git_cmd = 1;
  78         daemon.no_stdin = 1;
  79         daemon.out = -1;
  80
  81         if (start_command(&daemon))
  82                 die_errno("unable to start cache daemon");
  83         r = read_in_full(daemon.out, buf, sizeof(buf));
  84         if (r < 0)
  85                 die_errno("unable to read result code from cache daemon");
  86         if (r != 3 || memcmp(buf, "ok\n", 3))
  87                 die("cache daemon did not start: %.*s", r, buf);
  88         close(daemon.out);
  89 }
  90
  91 static void do_cache(const char *socket, const char *action, int timeout,
  92                      int flags)
  93 {
  94         struct strbuf buf = STRBUF_INIT;
  95
  96         strbuf_addf(&buf, "action=%s\n", action);
  97         strbuf_addf(&buf, "timeout=%d\n", timeout);
  98         if (flags & FLAG_RELAY) {
  99                 if (strbuf_read(&buf, 0, 0) < 0)
 100                         die_errno("unable to relay credential");
 101         }
 102
 103         if (send_request(socket, &buf) < 0) {
 104                 if (connection_fatally_broken(errno))
 105                         die_errno("unable to connect to cache daemon");
 106                 if (flags & FLAG_SPAWN) {
 107                         spawn_daemon(socket);
 108                         if (send_request(socket, &buf) < 0)
 109                                 die_errno("unable to connect to cache daemon");
 110                 }
 111         }
 112         strbuf_release(&buf);
 113 }
 114
 115 static char *get_socket_path(void)
 116 {
 117         struct stat sb;
 118         char *old_dir, *socket;
 119         old_dir = interpolate_path("~/.git-credential-cache", 0);
 120         if (old_dir && !stat(old_dir, &sb) && S_ISDIR(sb.st_mode))
 121                 socket = xstrfmt("%s/socket", old_dir);
 122         else
 123                 socket = xdg_cache_home("credential/socket");
 124         free(old_dir);
 125         return socket;
 126 }
 127
 128 int cmd_credential_cache(int argc, const char **argv, const char *prefix)
 129 {
 130         char *socket_path = NULL;
 131         int timeout = 900;
 132         const char *op;
 133         const char * const usage[] = {
 134                 "git credential-cache [<options>] <action>",
 135                 NULL
 136         };
 137         struct option options[] = {
 138                 OPT_INTEGER(0, "timeout", &timeout,
 139                             "number of seconds to cache credentials"),
 140                 OPT_STRING(0, "socket", &socket_path, "path",
 141                            "path of cache-daemon socket"),
 142                 OPT_END()
 143         };
 144
 145         argc = parse_options(argc, argv, prefix, options, usage, 0);
 146         if (!argc)
 147                 usage_with_options(usage, options);
 148         op = argv[0];
 149
 150         if (!socket_path)
 151                 socket_path = get_socket_path();
 152         if (!socket_path)
 153                 die("unable to find a suitable socket path; use --socket");
 154
 155         if (!strcmp(op, "exit"))
 156                 do_cache(socket_path, op, timeout, 0);
 157         else if (!strcmp(op, "get") || !strcmp(op, "erase"))
 158                 do_cache(socket_path, op, timeout, FLAG_RELAY);
 159         else if (!strcmp(op, "store"))
 160                 do_cache(socket_path, op, timeout, FLAG_RELAY|FLAG_SPAWN);
 161         else
 162                 ; /* ignore unknown operation */
 163
 164         return 0;
 165 }
 166
 167 #else
 168
 169 int cmd_credential_cache(int argc, const char **argv, const char *prefix)
 170 {
 171         const char * const usage[] = {
 172                 "git credential-cache [options] <action>",
 173                 "",
 174                 "credential-cache is disabled in this build of Git",
 175                 NULL
 176         };
 177         struct option options[] = { OPT_END() };
 178
 179         argc = parse_options(argc, argv, prefix, options, usage, 0);
 180         die(_("credential-cache unavailable; no unix socket support"));
 181 }
 182
 183 #endif /* NO_UNIX_SOCKETS */