Andrew's git / gitweb.git / credential.c
?
credential.con commit git-p4: correct --prepare-p4-only instructions (10de86d)
   1#include "cache.h"
   2#include "credential.h"
   3#include "string-list.h"
   4#include "run-command.h"
   5#include "url.h"
   6#include "prompt.h"
   7
   8void credential_init(struct credential *c)
   9{
  10        memset(c, 0, sizeof(*c));
  11        c->helpers.strdup_strings = 1;
  12}
  13
  14void credential_clear(struct credential *c)
  15{
  16        free(c->protocol);
  17        free(c->host);
  18        free(c->path);
  19        free(c->username);
  20        free(c->password);
  21        string_list_clear(&c->helpers, 0);
  22
  23        credential_init(c);
  24}
  25
  26int credential_match(const struct credential *want,
  27                     const struct credential *have)
  28{
  29#define CHECK(x) (!want->x || (have->x && !strcmp(want->x, have->x)))
  30        return CHECK(protocol) &&
  31               CHECK(host) &&
  32               CHECK(path) &&
  33               CHECK(username);
  34#undef CHECK
  35}
  36
  37static int credential_config_callback(const char *var, const char *value,
  38                                      void *data)
  39{
  40        struct credential *c = data;
  41        const char *key, *dot;
  42
  43        if (!skip_prefix(var, "credential.", &key))
  44                return 0;
  45
  46        if (!value)
  47                return config_error_nonbool(var);
  48
  49        dot = strrchr(key, '.');
  50        if (dot) {
  51                struct credential want = CREDENTIAL_INIT;
  52                char *url = xmemdupz(key, dot - key);
  53                int matched;
  54
  55                credential_from_url(&want, url);
  56                matched = credential_match(&want, c);
  57
  58                credential_clear(&want);
  59                free(url);
  60
  61                if (!matched)
  62                        return 0;
  63                key = dot + 1;
  64        }
  65
  66        if (!strcmp(key, "helper"))
  67                string_list_append(&c->helpers, value);
  68        else if (!strcmp(key, "username")) {
  69                if (!c->username)
  70                        c->username = xstrdup(value);
  71        }
  72        else if (!strcmp(key, "usehttppath"))
  73                c->use_http_path = git_config_bool(var, value);
  74
  75        return 0;
  76}
  77
  78static int proto_is_http(const char *s)
  79{
  80        if (!s)
  81                return 0;
  82        return !strcmp(s, "https") || !strcmp(s, "http");
  83}
  84
  85static void credential_apply_config(struct credential *c)
  86{
  87        if (c->configured)
  88                return;
  89        git_config(credential_config_callback, c);
  90        c->configured = 1;
  91
  92        if (!c->use_http_path && proto_is_http(c->protocol)) {
  93                free(c->path);
  94                c->path = NULL;
  95        }
  96}
  97
  98static void credential_describe(struct credential *c, struct strbuf *out)
  99{
 100        if (!c->protocol)
 101                return;
 102        strbuf_addf(out, "%s://", c->protocol);
 103        if (c->username && *c->username)
 104                strbuf_addf(out, "%s@", c->username);
 105        if (c->host)
 106                strbuf_addstr(out, c->host);
 107        if (c->path)
 108                strbuf_addf(out, "/%s", c->path);
 109}
 110
 111static char *credential_ask_one(const char *what, struct credential *c,
 112                                int flags)
 113{
 114        struct strbuf desc = STRBUF_INIT;
 115        struct strbuf prompt = STRBUF_INIT;
 116        char *r;
 117
 118        credential_describe(c, &desc);
 119        if (desc.len)
 120                strbuf_addf(&prompt, "%s for '%s': ", what, desc.buf);
 121        else
 122                strbuf_addf(&prompt, "%s: ", what);
 123
 124        r = git_prompt(prompt.buf, flags);
 125
 126        strbuf_release(&desc);
 127        strbuf_release(&prompt);
 128        return xstrdup(r);
 129}
 130
 131static void credential_getpass(struct credential *c)
 132{
 133        if (!c->username)
 134                c->username = credential_ask_one("Username", c,
 135                                                 PROMPT_ASKPASS|PROMPT_ECHO);
 136        if (!c->password)
 137                c->password = credential_ask_one("Password", c,
 138                                                 PROMPT_ASKPASS);
 139}
 140
 141int credential_read(struct credential *c, FILE *fp)
 142{
 143        struct strbuf line = STRBUF_INIT;
 144
 145        while (strbuf_getline(&line, fp, '\n') != EOF) {
 146                char *key = line.buf;
 147                char *value = strchr(key, '=');
 148
 149                if (!line.len)
 150                        break;
 151
 152                if (!value) {
 153                        warning("invalid credential line: %s", key);
 154                        strbuf_release(&line);
 155                        return -1;
 156                }
 157                *value++ = '\0';
 158
 159                if (!strcmp(key, "username")) {
 160                        free(c->username);
 161                        c->username = xstrdup(value);
 162                } else if (!strcmp(key, "password")) {
 163                        free(c->password);
 164                        c->password = xstrdup(value);
 165                } else if (!strcmp(key, "protocol")) {
 166                        free(c->protocol);
 167                        c->protocol = xstrdup(value);
 168                } else if (!strcmp(key, "host")) {
 169                        free(c->host);
 170                        c->host = xstrdup(value);
 171                } else if (!strcmp(key, "path")) {
 172                        free(c->path);
 173                        c->path = xstrdup(value);
 174                } else if (!strcmp(key, "url")) {
 175                        credential_from_url(c, value);
 176                }
 177                /*
 178                 * Ignore other lines; we don't know what they mean, but
 179                 * this future-proofs us when later versions of git do
 180                 * learn new lines, and the helpers are updated to match.
 181                 */
 182        }
 183
 184        strbuf_release(&line);
 185        return 0;
 186}
 187
 188static void credential_write_item(FILE *fp, const char *key, const char *value)
 189{
 190        if (!value)
 191                return;
 192        fprintf(fp, "%s=%s\n", key, value);
 193}
 194
 195void credential_write(const struct credential *c, FILE *fp)
 196{
 197        credential_write_item(fp, "protocol", c->protocol);
 198        credential_write_item(fp, "host", c->host);
 199        credential_write_item(fp, "path", c->path);
 200        credential_write_item(fp, "username", c->username);
 201        credential_write_item(fp, "password", c->password);
 202}
 203
 204static int run_credential_helper(struct credential *c,
 205                                 const char *cmd,
 206                                 int want_output)
 207{
 208        struct child_process helper = CHILD_PROCESS_INIT;
 209        const char *argv[] = { NULL, NULL };
 210        FILE *fp;
 211
 212        argv[0] = cmd;
 213        helper.argv = argv;
 214        helper.use_shell = 1;
 215        helper.in = -1;
 216        if (want_output)
 217                helper.out = -1;
 218        else
 219                helper.no_stdout = 1;
 220
 221        if (start_command(&helper) < 0)
 222                return -1;
 223
 224        fp = xfdopen(helper.in, "w");
 225        credential_write(c, fp);
 226        fclose(fp);
 227
 228        if (want_output) {
 229                int r;
 230                fp = xfdopen(helper.out, "r");
 231                r = credential_read(c, fp);
 232                fclose(fp);
 233                if (r < 0) {
 234                        finish_command(&helper);
 235                        return -1;
 236                }
 237        }
 238
 239        if (finish_command(&helper))
 240                return -1;
 241        return 0;
 242}
 243
 244static int credential_do(struct credential *c, const char *helper,
 245                         const char *operation)
 246{
 247        struct strbuf cmd = STRBUF_INIT;
 248        int r;
 249
 250        if (helper[0] == '!')
 251                strbuf_addstr(&cmd, helper + 1);
 252        else if (is_absolute_path(helper))
 253                strbuf_addstr(&cmd, helper);
 254        else
 255                strbuf_addf(&cmd, "git credential-%s", helper);
 256
 257        strbuf_addf(&cmd, " %s", operation);
 258        r = run_credential_helper(c, cmd.buf, !strcmp(operation, "get"));
 259
 260        strbuf_release(&cmd);
 261        return r;
 262}
 263
 264void credential_fill(struct credential *c)
 265{
 266        int i;
 267
 268        if (c->username && c->password)
 269                return;
 270
 271        credential_apply_config(c);
 272
 273        for (i = 0; i < c->helpers.nr; i++) {
 274                credential_do(c, c->helpers.items[i].string, "get");
 275                if (c->username && c->password)
 276                        return;
 277        }
 278
 279        credential_getpass(c);
 280        if (!c->username && !c->password)
 281                die("unable to get password from user");
 282}
 283
 284void credential_approve(struct credential *c)
 285{
 286        int i;
 287
 288        if (c->approved)
 289                return;
 290        if (!c->username || !c->password)
 291                return;
 292
 293        credential_apply_config(c);
 294
 295        for (i = 0; i < c->helpers.nr; i++)
 296                credential_do(c, c->helpers.items[i].string, "store");
 297        c->approved = 1;
 298}
 299
 300void credential_reject(struct credential *c)
 301{
 302        int i;
 303
 304        credential_apply_config(c);
 305
 306        for (i = 0; i < c->helpers.nr; i++)
 307                credential_do(c, c->helpers.items[i].string, "erase");
 308
 309        free(c->username);
 310        c->username = NULL;
 311        free(c->password);
 312        c->password = NULL;
 313        c->approved = 0;
 314}
 315
 316void credential_from_url(struct credential *c, const char *url)
 317{
 318        const char *at, *colon, *cp, *slash, *host, *proto_end;
 319
 320        credential_clear(c);
 321
 322        /*
 323         * Match one of:
 324         *   (1) proto://<host>/...
 325         *   (2) proto://<user>@<host>/...
 326         *   (3) proto://<user>:<pass>@<host>/...
 327         */
 328        proto_end = strstr(url, "://");
 329        if (!proto_end)
 330                return;
 331        cp = proto_end + 3;
 332        at = strchr(cp, '@');
 333        colon = strchr(cp, ':');
 334        slash = strchrnul(cp, '/');
 335
 336        if (!at || slash <= at) {
 337                /* Case (1) */
 338                host = cp;
 339        }
 340        else if (!colon || at <= colon) {
 341                /* Case (2) */
 342                c->username = url_decode_mem(cp, at - cp);
 343                host = at + 1;
 344        } else {
 345                /* Case (3) */
 346                c->username = url_decode_mem(cp, colon - cp);
 347                c->password = url_decode_mem(colon + 1, at - (colon + 1));
 348                host = at + 1;
 349        }
 350
 351        if (proto_end - url > 0)
 352                c->protocol = xmemdupz(url, proto_end - url);
 353        if (slash - host > 0)
 354                c->host = url_decode_mem(host, slash - host);
 355        /* Trim leading and trailing slashes from path */
 356        while (*slash == '/')
 357                slash++;
 358        if (*slash) {
 359                char *p;
 360                c->path = url_decode(slash);
 361                p = c->path + strlen(c->path) - 1;
 362                while (p > c->path && *p == '/')
 363                        *p-- = '\0';
 364        }
 365}