git-remote-mediawiki: change return type of get_mw_pages
[git] / credential-store.c
1 #include "cache.h"
2 #include "credential.h"
3 #include "string-list.h"
4 #include "parse-options.h"
5
6 static struct lock_file credential_lock;
7
8 static void parse_credential_file(const char *fn,
9                                   struct credential *c,
10                                   void (*match_cb)(struct credential *),
11                                   void (*other_cb)(struct strbuf *))
12 {
13         FILE *fh;
14         struct strbuf line = STRBUF_INIT;
15         struct credential entry = CREDENTIAL_INIT;
16
17         fh = fopen(fn, "r");
18         if (!fh) {
19                 if (errno != ENOENT)
20                         die_errno("unable to open %s", fn);
21                 return;
22         }
23
24         while (strbuf_getline(&line, fh, '\n') != EOF) {
25                 credential_from_url(&entry, line.buf);
26                 if (entry.username && entry.password &&
27                     credential_match(c, &entry)) {
28                         if (match_cb) {
29                                 match_cb(&entry);
30                                 break;
31                         }
32                 }
33                 else if (other_cb)
34                         other_cb(&line);
35         }
36
37         credential_clear(&entry);
38         strbuf_release(&line);
39         fclose(fh);
40 }
41
42 static void print_entry(struct credential *c)
43 {
44         printf("username=%s\n", c->username);
45         printf("password=%s\n", c->password);
46 }
47
48 static void print_line(struct strbuf *buf)
49 {
50         strbuf_addch(buf, '\n');
51         write_or_die(credential_lock.fd, buf->buf, buf->len);
52 }
53
54 static void rewrite_credential_file(const char *fn, struct credential *c,
55                                     struct strbuf *extra)
56 {
57         if (hold_lock_file_for_update(&credential_lock, fn, 0) < 0)
58                 die_errno("unable to get credential storage lock");
59         if (extra)
60                 print_line(extra);
61         parse_credential_file(fn, c, NULL, print_line);
62         if (commit_lock_file(&credential_lock) < 0)
63                 die_errno("unable to commit credential store");
64 }
65
66 static void store_credential(const char *fn, struct credential *c)
67 {
68         struct strbuf buf = STRBUF_INIT;
69
70         /*
71          * Sanity check that what we are storing is actually sensible.
72          * In particular, we can't make a URL without a protocol field.
73          * Without either a host or pathname (depending on the scheme),
74          * we have no primary key. And without a username and password,
75          * we are not actually storing a credential.
76          */
77         if (!c->protocol || !(c->host || c->path) ||
78             !c->username || !c->password)
79                 return;
80
81         strbuf_addf(&buf, "%s://", c->protocol);
82         strbuf_addstr_urlencode(&buf, c->username, 1);
83         strbuf_addch(&buf, ':');
84         strbuf_addstr_urlencode(&buf, c->password, 1);
85         strbuf_addch(&buf, '@');
86         if (c->host)
87                 strbuf_addstr_urlencode(&buf, c->host, 1);
88         if (c->path) {
89                 strbuf_addch(&buf, '/');
90                 strbuf_addstr_urlencode(&buf, c->path, 0);
91         }
92
93         rewrite_credential_file(fn, c, &buf);
94         strbuf_release(&buf);
95 }
96
97 static void remove_credential(const char *fn, struct credential *c)
98 {
99         /*
100          * Sanity check that we actually have something to match
101          * against. The input we get is a restrictive pattern,
102          * so technically a blank credential means "erase everything".
103          * But it is too easy to accidentally send this, since it is equivalent
104          * to empty input. So explicitly disallow it, and require that the
105          * pattern have some actual content to match.
106          */
107         if (c->protocol || c->host || c->path || c->username)
108                 rewrite_credential_file(fn, c, NULL);
109 }
110
111 static int lookup_credential(const char *fn, struct credential *c)
112 {
113         parse_credential_file(fn, c, print_entry, NULL);
114         return c->username && c->password;
115 }
116
117 int main(int argc, const char **argv)
118 {
119         const char * const usage[] = {
120                 "git credential-store [options] <action>",
121                 NULL
122         };
123         const char *op;
124         struct credential c = CREDENTIAL_INIT;
125         char *file = NULL;
126         struct option options[] = {
127                 OPT_STRING(0, "file", &file, "path",
128                            "fetch and store credentials in <path>"),
129                 OPT_END()
130         };
131
132         umask(077);
133
134         argc = parse_options(argc, argv, NULL, options, usage, 0);
135         if (argc != 1)
136                 usage_with_options(usage, options);
137         op = argv[0];
138
139         if (!file)
140                 file = expand_user_path("~/.git-credentials");
141         if (!file)
142                 die("unable to set up default path; use --file");
143
144         if (credential_read(&c, stdin) < 0)
145                 die("unable to read credential");
146
147         if (!strcmp(op, "get"))
148                 lookup_credential(file, &c);
149         else if (!strcmp(op, "erase"))
150                 remove_credential(file, &c);
151         else if (!strcmp(op, "store"))
152                 store_credential(file, &c);
153         else
154                 ; /* Ignore unknown operation. */
155
156         return 0;
157 }