1709795f |
1 | /* |
2 | * uxstore.c: Unix-specific implementation of the interface defined |
3 | * in storage.h. |
4 | */ |
5 | |
6 | #include <stdio.h> |
7 | #include <stdlib.h> |
3f935d5b |
8 | #include <string.h> |
8dacc30e |
9 | #include <assert.h> |
3f935d5b |
10 | #include <errno.h> |
0ac15bdc |
11 | #include <ctype.h> |
4afa6d82 |
12 | #include <limits.h> |
c5e438ec |
13 | #include <unistd.h> |
14 | #include <fcntl.h> |
8dacc30e |
15 | #include <dirent.h> |
c5e438ec |
16 | #include <sys/stat.h> |
17 | #include <sys/types.h> |
c18ed097 |
18 | #include <pwd.h> |
1709795f |
19 | #include "putty.h" |
20 | #include "storage.h" |
0ac15bdc |
21 | #include "tree234.h" |
1709795f |
22 | |
4afa6d82 |
23 | #ifdef PATH_MAX |
24 | #define FNLEN PATH_MAX |
25 | #else |
26 | #define FNLEN 1024 /* XXX */ |
27 | #endif |
28 | |
8dacc30e |
29 | enum { |
be57bcef |
30 | INDEX_DIR, INDEX_HOSTKEYS, INDEX_HOSTKEYS_TMP, INDEX_RANDSEED, |
8dacc30e |
31 | INDEX_SESSIONDIR, INDEX_SESSION, |
32 | }; |
33 | |
34 | static const char hex[16] = "0123456789ABCDEF"; |
35 | |
36 | static char *mungestr(const char *in) |
37 | { |
38 | char *out, *ret; |
39 | |
797f6ff3 |
40 | if (!in || !*in) |
8dacc30e |
41 | in = "Default Settings"; |
42 | |
43 | ret = out = snewn(3*strlen(in)+1, char); |
44 | |
45 | while (*in) { |
46 | /* |
47 | * There are remarkably few punctuation characters that |
48 | * aren't shell-special in some way or likely to be used as |
49 | * separators in some file format or another! Hence we use |
50 | * opt-in for safe characters rather than opt-out for |
51 | * specific unsafe ones... |
52 | */ |
53 | if (*in!='+' && *in!='-' && *in!='.' && *in!='@' && *in!='_' && |
54 | !(*in >= '0' && *in <= '9') && |
55 | !(*in >= 'A' && *in <= 'Z') && |
56 | !(*in >= 'a' && *in <= 'z')) { |
57 | *out++ = '%'; |
58 | *out++ = hex[((unsigned char) *in) >> 4]; |
59 | *out++ = hex[((unsigned char) *in) & 15]; |
60 | } else |
61 | *out++ = *in; |
62 | in++; |
63 | } |
64 | *out = '\0'; |
65 | return ret; |
66 | } |
67 | |
68 | static char *unmungestr(const char *in) |
69 | { |
70 | char *out, *ret; |
71 | out = ret = snewn(strlen(in)+1, char); |
72 | while (*in) { |
73 | if (*in == '%' && in[1] && in[2]) { |
74 | int i, j; |
75 | |
76 | i = in[1] - '0'; |
77 | i -= (i > 9 ? 7 : 0); |
78 | j = in[2] - '0'; |
79 | j -= (j > 9 ? 7 : 0); |
80 | |
81 | *out++ = (i << 4) + j; |
82 | in += 3; |
83 | } else { |
84 | *out++ = *in++; |
85 | } |
86 | } |
87 | *out = '\0'; |
88 | return ret; |
89 | } |
90 | |
c18ed097 |
91 | static char *make_filename(int index, const char *subname) |
8dacc30e |
92 | { |
c18ed097 |
93 | char *env, *tmp, *ret; |
94 | |
95 | /* |
96 | * Allow override of the PuTTY configuration location, and of |
97 | * specific subparts of it, by means of environment variables. |
98 | */ |
99 | if (index == INDEX_DIR) { |
100 | struct passwd *pwd; |
101 | |
102 | env = getenv("PUTTYDIR"); |
103 | if (env) |
104 | return dupstr(env); |
105 | env = getenv("HOME"); |
106 | if (env) |
107 | return dupprintf("%s/.putty", env); |
108 | pwd = getpwuid(getuid()); |
109 | if (pwd && pwd->pw_dir) |
110 | return dupprintf("%s/.putty", pwd->pw_dir); |
111 | return dupstr("/.putty"); |
112 | } |
113 | if (index == INDEX_SESSIONDIR) { |
114 | env = getenv("PUTTYSESSIONS"); |
115 | if (env) |
116 | return dupstr(env); |
117 | tmp = make_filename(INDEX_DIR, NULL); |
118 | ret = dupprintf("%s/sessions", tmp); |
119 | sfree(tmp); |
120 | return ret; |
121 | } |
8dacc30e |
122 | if (index == INDEX_SESSION) { |
123 | char *munged = mungestr(subname); |
c18ed097 |
124 | tmp = make_filename(INDEX_SESSIONDIR, NULL); |
125 | ret = dupprintf("%s/%s", tmp, munged); |
126 | sfree(tmp); |
127 | sfree(munged); |
128 | return ret; |
129 | } |
130 | if (index == INDEX_HOSTKEYS) { |
131 | env = getenv("PUTTYSSHHOSTKEYS"); |
132 | if (env) |
133 | return dupstr(env); |
134 | tmp = make_filename(INDEX_DIR, NULL); |
135 | ret = dupprintf("%s/sshhostkeys", tmp); |
136 | sfree(tmp); |
137 | return ret; |
138 | } |
139 | if (index == INDEX_HOSTKEYS_TMP) { |
140 | tmp = make_filename(INDEX_HOSTKEYS, NULL); |
141 | ret = dupprintf("%s.tmp", tmp); |
142 | sfree(tmp); |
143 | return ret; |
144 | } |
145 | if (index == INDEX_RANDSEED) { |
146 | env = getenv("PUTTYRANDOMSEED"); |
147 | if (env) |
148 | return dupstr(env); |
149 | tmp = make_filename(INDEX_DIR, NULL); |
150 | ret = dupprintf("%s/randomseed", tmp); |
151 | sfree(tmp); |
152 | return ret; |
8dacc30e |
153 | } |
c18ed097 |
154 | tmp = make_filename(INDEX_DIR, NULL); |
155 | ret = dupprintf("%s/ERROR", tmp); |
156 | sfree(tmp); |
157 | return ret; |
8dacc30e |
158 | } |
159 | |
3f935d5b |
160 | void *open_settings_w(const char *sessionname, char **errmsg) |
1709795f |
161 | { |
c18ed097 |
162 | char *filename; |
8dacc30e |
163 | FILE *fp; |
164 | |
3f935d5b |
165 | *errmsg = NULL; |
166 | |
8dacc30e |
167 | /* |
3f935d5b |
168 | * Start by making sure the .putty directory and its sessions |
6702cd9b |
169 | * subdir actually exist. |
8dacc30e |
170 | */ |
6702cd9b |
171 | filename = make_filename(INDEX_DIR, NULL); |
172 | if (mkdir(filename, 0700) < 0 && errno != EEXIST) { |
173 | *errmsg = dupprintf("Unable to save session: mkdir(\"%s\") " |
174 | "returned '%s'", filename, strerror(errno)); |
175 | sfree(filename); |
176 | return NULL; |
177 | } |
178 | sfree(filename); |
179 | |
c18ed097 |
180 | filename = make_filename(INDEX_SESSIONDIR, NULL); |
6702cd9b |
181 | if (mkdir(filename, 0700) < 0 && errno != EEXIST) { |
182 | *errmsg = dupprintf("Unable to save session: mkdir(\"%s\") " |
183 | "returned '%s'", filename, strerror(errno)); |
184 | sfree(filename); |
185 | return NULL; |
c18ed097 |
186 | } |
187 | sfree(filename); |
8dacc30e |
188 | |
c18ed097 |
189 | filename = make_filename(INDEX_SESSION, sessionname); |
8dacc30e |
190 | fp = fopen(filename, "w"); |
3f935d5b |
191 | if (!fp) { |
6702cd9b |
192 | *errmsg = dupprintf("Unable to save session: open(\"%s\") " |
193 | "returned '%s'", filename, strerror(errno)); |
c18ed097 |
194 | sfree(filename); |
3f935d5b |
195 | return NULL; /* can't open */ |
196 | } |
c18ed097 |
197 | sfree(filename); |
8dacc30e |
198 | return fp; |
1709795f |
199 | } |
200 | |
c85623f9 |
201 | void write_setting_s(void *handle, const char *key, const char *value) |
1709795f |
202 | { |
8dacc30e |
203 | FILE *fp = (FILE *)handle; |
204 | fprintf(fp, "%s=%s\n", key, value); |
1709795f |
205 | } |
206 | |
c85623f9 |
207 | void write_setting_i(void *handle, const char *key, int value) |
1709795f |
208 | { |
8dacc30e |
209 | FILE *fp = (FILE *)handle; |
210 | fprintf(fp, "%s=%d\n", key, value); |
1709795f |
211 | } |
212 | |
213 | void close_settings_w(void *handle) |
214 | { |
8dacc30e |
215 | FILE *fp = (FILE *)handle; |
216 | fclose(fp); |
1709795f |
217 | } |
218 | |
faec60ed |
219 | /* |
220 | * Reading settings, for the moment, is done by retrieving X |
221 | * resources from the X display. When we introduce disk files, I |
222 | * think what will happen is that the X resources will override |
223 | * PuTTY's inbuilt defaults, but that the disk files will then |
224 | * override those. This isn't optimal, but it's the best I can |
225 | * immediately work out. |
e50f98bc |
226 | * FIXME: the above comment is a bit out of date. Did it happen? |
faec60ed |
227 | */ |
228 | |
b3d375b2 |
229 | struct skeyval { |
c85623f9 |
230 | const char *key; |
231 | const char *value; |
0ac15bdc |
232 | }; |
233 | |
234 | static tree234 *xrmtree = NULL; |
235 | |
8dacc30e |
236 | int keycmp(void *av, void *bv) |
0ac15bdc |
237 | { |
b3d375b2 |
238 | struct skeyval *a = (struct skeyval *)av; |
239 | struct skeyval *b = (struct skeyval *)bv; |
0ac15bdc |
240 | return strcmp(a->key, b->key); |
241 | } |
242 | |
243 | void provide_xrm_string(char *string) |
244 | { |
c85623f9 |
245 | char *p, *q, *key; |
b3d375b2 |
246 | struct skeyval *xrms, *ret; |
0ac15bdc |
247 | |
248 | p = q = strchr(string, ':'); |
249 | if (!q) { |
250 | fprintf(stderr, "pterm: expected a colon in resource string" |
251 | " \"%s\"\n", string); |
252 | return; |
253 | } |
254 | q++; |
255 | while (p > string && p[-1] != '.' && p[-1] != '*') |
256 | p--; |
b3d375b2 |
257 | xrms = snew(struct skeyval); |
3d88e64d |
258 | key = snewn(q-p, char); |
c85623f9 |
259 | memcpy(key, p, q-p); |
260 | key[q-p-1] = '\0'; |
261 | xrms->key = key; |
e93ed432 |
262 | while (*q && isspace((unsigned char)*q)) |
0ac15bdc |
263 | q++; |
264 | xrms->value = dupstr(q); |
265 | |
266 | if (!xrmtree) |
8dacc30e |
267 | xrmtree = newtree234(keycmp); |
0ac15bdc |
268 | |
269 | ret = add234(xrmtree, xrms); |
270 | if (ret) { |
271 | /* Override an existing string. */ |
272 | del234(xrmtree, ret); |
273 | add234(xrmtree, xrms); |
274 | } |
275 | } |
276 | |
c85623f9 |
277 | const char *get_setting(const char *key) |
0ac15bdc |
278 | { |
b3d375b2 |
279 | struct skeyval tmp, *ret; |
0ac15bdc |
280 | tmp.key = key; |
281 | if (xrmtree) { |
282 | ret = find234(xrmtree, &tmp, NULL); |
283 | if (ret) |
284 | return ret->value; |
285 | } |
c5e438ec |
286 | return x_get_default(key); |
0ac15bdc |
287 | } |
288 | |
c85623f9 |
289 | void *open_settings_r(const char *sessionname) |
1709795f |
290 | { |
c18ed097 |
291 | char *filename; |
8dacc30e |
292 | FILE *fp; |
293 | char *line; |
294 | tree234 *ret; |
295 | |
c18ed097 |
296 | filename = make_filename(INDEX_SESSION, sessionname); |
8dacc30e |
297 | fp = fopen(filename, "r"); |
c18ed097 |
298 | sfree(filename); |
8dacc30e |
299 | if (!fp) |
300 | return NULL; /* can't open */ |
301 | |
302 | ret = newtree234(keycmp); |
303 | |
304 | while ( (line = fgetline(fp)) ) { |
305 | char *value = strchr(line, '='); |
b3d375b2 |
306 | struct skeyval *kv; |
8dacc30e |
307 | |
038ec85e |
308 | if (!value) { |
309 | sfree(line); |
8dacc30e |
310 | continue; |
038ec85e |
311 | } |
8dacc30e |
312 | *value++ = '\0'; |
313 | value[strcspn(value, "\r\n")] = '\0'; /* trim trailing NL */ |
314 | |
b3d375b2 |
315 | kv = snew(struct skeyval); |
8dacc30e |
316 | kv->key = dupstr(line); |
317 | kv->value = dupstr(value); |
318 | add234(ret, kv); |
319 | |
320 | sfree(line); |
321 | } |
322 | |
323 | fclose(fp); |
324 | |
325 | return ret; |
1709795f |
326 | } |
327 | |
4a693cfc |
328 | char *read_setting_s(void *handle, const char *key) |
1709795f |
329 | { |
8dacc30e |
330 | tree234 *tree = (tree234 *)handle; |
331 | const char *val; |
b3d375b2 |
332 | struct skeyval tmp, *kv; |
8dacc30e |
333 | |
334 | tmp.key = key; |
335 | if (tree != NULL && |
336 | (kv = find234(tree, &tmp, NULL)) != NULL) { |
337 | val = kv->value; |
338 | assert(val != NULL); |
339 | } else |
340 | val = get_setting(key); |
341 | |
faec60ed |
342 | if (!val) |
343 | return NULL; |
4a693cfc |
344 | else |
345 | return dupstr(val); |
1709795f |
346 | } |
347 | |
c85623f9 |
348 | int read_setting_i(void *handle, const char *key, int defvalue) |
1709795f |
349 | { |
8dacc30e |
350 | tree234 *tree = (tree234 *)handle; |
351 | const char *val; |
b3d375b2 |
352 | struct skeyval tmp, *kv; |
8dacc30e |
353 | |
354 | tmp.key = key; |
355 | if (tree != NULL && |
356 | (kv = find234(tree, &tmp, NULL)) != NULL) { |
357 | val = kv->value; |
358 | assert(val != NULL); |
359 | } else |
360 | val = get_setting(key); |
361 | |
faec60ed |
362 | if (!val) |
363 | return defvalue; |
364 | else |
365 | return atoi(val); |
1709795f |
366 | } |
367 | |
ae62eaeb |
368 | FontSpec *read_setting_fontspec(void *handle, const char *name) |
9a30e26b |
369 | { |
f160b7b8 |
370 | /* |
371 | * In GTK1-only PuTTY, we used to store font names simply as a |
372 | * valid X font description string (logical or alias), under a |
373 | * bare key such as "Font". |
374 | * |
375 | * In GTK2 PuTTY, we have a prefix system where "client:" |
376 | * indicates a Pango font and "server:" an X one; existing |
377 | * configuration needs to be reinterpreted as having the |
378 | * "server:" prefix, so we change the storage key from the |
379 | * provided name string (e.g. "Font") to a suffixed one |
380 | * ("FontName"). |
381 | */ |
382 | char *suffname = dupcat(name, "Name", NULL); |
4a693cfc |
383 | char *tmp; |
384 | |
385 | if ((tmp = read_setting_s(handle, suffname)) != NULL) { |
ae62eaeb |
386 | FontSpec *fs = fontspec_new(tmp); |
f160b7b8 |
387 | sfree(suffname); |
4a693cfc |
388 | sfree(tmp); |
ae62eaeb |
389 | return fs; /* got new-style name */ |
f160b7b8 |
390 | } |
391 | sfree(suffname); |
392 | |
393 | /* Fall back to old-style name. */ |
4a693cfc |
394 | tmp = read_setting_s(handle, name); |
395 | if (tmp && *tmp) { |
ae62eaeb |
396 | char *tmp2 = dupcat("server:", tmp, NULL); |
397 | FontSpec *fs = fontspec_new(tmp2); |
398 | sfree(tmp2); |
4a693cfc |
399 | sfree(tmp); |
ae62eaeb |
400 | return fs; |
4a693cfc |
401 | } else { |
402 | sfree(tmp); |
ae62eaeb |
403 | return NULL; |
f160b7b8 |
404 | } |
9a30e26b |
405 | } |
962468d4 |
406 | Filename *read_setting_filename(void *handle, const char *name) |
9a30e26b |
407 | { |
4a693cfc |
408 | char *tmp = read_setting_s(handle, name); |
409 | if (tmp) { |
962468d4 |
410 | Filename *ret = filename_from_str(tmp); |
4a693cfc |
411 | sfree(tmp); |
962468d4 |
412 | return ret; |
4a693cfc |
413 | } else |
962468d4 |
414 | return NULL; |
9a30e26b |
415 | } |
416 | |
ae62eaeb |
417 | void write_setting_fontspec(void *handle, const char *name, FontSpec *fs) |
9a30e26b |
418 | { |
f160b7b8 |
419 | /* |
420 | * read_setting_fontspec had to handle two cases, but when |
421 | * writing our settings back out we simply always generate the |
422 | * new-style name. |
423 | */ |
424 | char *suffname = dupcat(name, "Name", NULL); |
ae62eaeb |
425 | write_setting_s(handle, suffname, fs->name); |
f160b7b8 |
426 | sfree(suffname); |
9a30e26b |
427 | } |
962468d4 |
428 | void write_setting_filename(void *handle, const char *name, Filename *result) |
9a30e26b |
429 | { |
962468d4 |
430 | write_setting_s(handle, name, result->path); |
9a30e26b |
431 | } |
432 | |
1709795f |
433 | void close_settings_r(void *handle) |
434 | { |
8dacc30e |
435 | tree234 *tree = (tree234 *)handle; |
b3d375b2 |
436 | struct skeyval *kv; |
8dacc30e |
437 | |
438 | if (!tree) |
439 | return; |
440 | |
441 | while ( (kv = index234(tree, 0)) != NULL) { |
442 | del234(tree, kv); |
443 | sfree((char *)kv->key); |
444 | sfree((char *)kv->value); |
445 | sfree(kv); |
446 | } |
447 | |
448 | freetree234(tree); |
1709795f |
449 | } |
450 | |
c85623f9 |
451 | void del_settings(const char *sessionname) |
1709795f |
452 | { |
c18ed097 |
453 | char *filename; |
454 | filename = make_filename(INDEX_SESSION, sessionname); |
8dacc30e |
455 | unlink(filename); |
c18ed097 |
456 | sfree(filename); |
1709795f |
457 | } |
458 | |
459 | void *enum_settings_start(void) |
460 | { |
8dacc30e |
461 | DIR *dp; |
c18ed097 |
462 | char *filename; |
1709795f |
463 | |
c18ed097 |
464 | filename = make_filename(INDEX_SESSIONDIR, NULL); |
8dacc30e |
465 | dp = opendir(filename); |
c18ed097 |
466 | sfree(filename); |
1709795f |
467 | |
8dacc30e |
468 | return dp; |
1709795f |
469 | } |
470 | |
8dacc30e |
471 | char *enum_settings_next(void *handle, char *buffer, int buflen) |
c5e438ec |
472 | { |
8dacc30e |
473 | DIR *dp = (DIR *)handle; |
474 | struct dirent *de; |
475 | struct stat st; |
c18ed097 |
476 | char *fullpath; |
477 | int maxlen, thislen, len; |
8dacc30e |
478 | char *unmunged; |
479 | |
c18ed097 |
480 | fullpath = make_filename(INDEX_SESSIONDIR, NULL); |
481 | maxlen = len = strlen(fullpath); |
8dacc30e |
482 | |
483 | while ( (de = readdir(dp)) != NULL ) { |
c18ed097 |
484 | thislen = len + 1 + strlen(de->d_name); |
485 | if (maxlen < thislen) { |
486 | maxlen = thislen; |
487 | fullpath = sresize(fullpath, maxlen+1, char); |
488 | } |
489 | fullpath[len] = '/'; |
490 | strncpy(fullpath+len+1, de->d_name, thislen - (len+1)); |
491 | fullpath[thislen] = '\0'; |
8dacc30e |
492 | |
493 | if (stat(fullpath, &st) < 0 || !S_ISREG(st.st_mode)) |
494 | continue; /* try another one */ |
495 | |
496 | unmunged = unmungestr(de->d_name); |
497 | strncpy(buffer, unmunged, buflen); |
498 | buffer[buflen-1] = '\0'; |
499 | sfree(unmunged); |
c18ed097 |
500 | sfree(fullpath); |
8dacc30e |
501 | return buffer; |
502 | } |
503 | |
c18ed097 |
504 | sfree(fullpath); |
8dacc30e |
505 | return NULL; |
c5e438ec |
506 | } |
507 | |
8dacc30e |
508 | void enum_settings_finish(void *handle) |
c5e438ec |
509 | { |
8dacc30e |
510 | DIR *dp = (DIR *)handle; |
511 | closedir(dp); |
c5e438ec |
512 | } |
513 | |
514 | /* |
515 | * Lines in the host keys file are of the form |
516 | * |
517 | * type@port:hostname keydata |
518 | * |
519 | * e.g. |
520 | * |
521 | * rsa@22:foovax.example.org 0x23,0x293487364395345345....2343 |
522 | */ |
c85623f9 |
523 | int verify_host_key(const char *hostname, int port, |
524 | const char *keytype, const char *key) |
1709795f |
525 | { |
c5e438ec |
526 | FILE *fp; |
c18ed097 |
527 | char *filename; |
c5e438ec |
528 | char *line; |
529 | int ret; |
530 | |
c18ed097 |
531 | filename = make_filename(INDEX_HOSTKEYS, NULL); |
c5e438ec |
532 | fp = fopen(filename, "r"); |
c18ed097 |
533 | sfree(filename); |
c5e438ec |
534 | if (!fp) |
535 | return 1; /* key does not exist */ |
536 | |
537 | ret = 1; |
538 | while ( (line = fgetline(fp)) ) { |
539 | int i; |
540 | char *p = line; |
541 | char porttext[20]; |
542 | |
543 | line[strcspn(line, "\n")] = '\0'; /* strip trailing newline */ |
544 | |
545 | i = strlen(keytype); |
546 | if (strncmp(p, keytype, i)) |
547 | goto done; |
548 | p += i; |
549 | |
550 | if (*p != '@') |
551 | goto done; |
552 | p++; |
553 | |
554 | sprintf(porttext, "%d", port); |
555 | i = strlen(porttext); |
556 | if (strncmp(p, porttext, i)) |
557 | goto done; |
558 | p += i; |
559 | |
560 | if (*p != ':') |
561 | goto done; |
562 | p++; |
563 | |
564 | i = strlen(hostname); |
565 | if (strncmp(p, hostname, i)) |
566 | goto done; |
567 | p += i; |
568 | |
569 | if (*p != ' ') |
570 | goto done; |
571 | p++; |
572 | |
573 | /* |
574 | * Found the key. Now just work out whether it's the right |
575 | * one or not. |
576 | */ |
577 | if (!strcmp(p, key)) |
578 | ret = 0; /* key matched OK */ |
579 | else |
580 | ret = 2; /* key mismatch */ |
581 | |
582 | done: |
583 | sfree(line); |
584 | if (ret != 1) |
585 | break; |
586 | } |
587 | |
1957695c |
588 | fclose(fp); |
c5e438ec |
589 | return ret; |
1709795f |
590 | } |
591 | |
c85623f9 |
592 | void store_host_key(const char *hostname, int port, |
593 | const char *keytype, const char *key) |
1709795f |
594 | { |
be57bcef |
595 | FILE *rfp, *wfp; |
596 | char *newtext, *line; |
597 | int headerlen; |
c18ed097 |
598 | char *filename, *tmpfilename; |
c5e438ec |
599 | |
be57bcef |
600 | /* |
601 | * Open both the old file and a new file. |
602 | */ |
c18ed097 |
603 | tmpfilename = make_filename(INDEX_HOSTKEYS_TMP, NULL); |
be57bcef |
604 | wfp = fopen(tmpfilename, "w"); |
c9017afc |
605 | if (!wfp && errno == ENOENT) { |
c18ed097 |
606 | char *dir; |
11cd814b |
607 | |
c18ed097 |
608 | dir = make_filename(INDEX_DIR, NULL); |
c9017afc |
609 | if (mkdir(dir, 0700) < 0) { |
610 | char *msg = dupprintf("Unable to store host key: mkdir(\"%s\") " |
611 | "returned '%s'", dir, strerror(errno)); |
612 | nonfatal(msg); |
613 | sfree(dir); |
614 | sfree(tmpfilename); |
615 | return; |
616 | } |
c18ed097 |
617 | sfree(dir); |
618 | |
11cd814b |
619 | wfp = fopen(tmpfilename, "w"); |
c5e438ec |
620 | } |
c18ed097 |
621 | if (!wfp) { |
c9017afc |
622 | char *msg = dupprintf("Unable to store host key: open(\"%s\") " |
623 | "returned '%s'", tmpfilename, strerror(errno)); |
624 | nonfatal(msg); |
625 | sfree(tmpfilename); |
626 | return; |
c18ed097 |
627 | } |
628 | filename = make_filename(INDEX_HOSTKEYS, NULL); |
11cd814b |
629 | rfp = fopen(filename, "r"); |
be57bcef |
630 | |
038ec85e |
631 | newtext = dupprintf("%s@%d:%s %s\n", keytype, port, hostname, key); |
632 | headerlen = 1 + strcspn(newtext, " "); /* count the space too */ |
633 | |
be57bcef |
634 | /* |
635 | * Copy all lines from the old file to the new one that _don't_ |
636 | * involve the same host key identifier as the one we're adding. |
637 | */ |
11cd814b |
638 | if (rfp) { |
639 | while ( (line = fgetline(rfp)) ) { |
640 | if (strncmp(line, newtext, headerlen)) |
641 | fputs(line, wfp); |
038ec85e |
642 | sfree(line); |
11cd814b |
643 | } |
644 | fclose(rfp); |
c5e438ec |
645 | } |
be57bcef |
646 | |
647 | /* |
648 | * Now add the new line at the end. |
649 | */ |
650 | fputs(newtext, wfp); |
651 | |
be57bcef |
652 | fclose(wfp); |
653 | |
654 | rename(tmpfilename, filename); |
655 | |
c18ed097 |
656 | sfree(tmpfilename); |
657 | sfree(filename); |
be57bcef |
658 | sfree(newtext); |
1709795f |
659 | } |
660 | |
661 | void read_random_seed(noise_consumer_t consumer) |
662 | { |
d9c40fd6 |
663 | int fd; |
c18ed097 |
664 | char *fname; |
d9c40fd6 |
665 | |
c18ed097 |
666 | fname = make_filename(INDEX_RANDSEED, NULL); |
d9c40fd6 |
667 | fd = open(fname, O_RDONLY); |
c18ed097 |
668 | sfree(fname); |
0720879b |
669 | if (fd >= 0) { |
d9c40fd6 |
670 | char buf[512]; |
671 | int ret; |
672 | while ( (ret = read(fd, buf, sizeof(buf))) > 0) |
673 | consumer(buf, ret); |
674 | close(fd); |
675 | } |
1709795f |
676 | } |
677 | |
678 | void write_random_seed(void *data, int len) |
679 | { |
d9c40fd6 |
680 | int fd; |
c18ed097 |
681 | char *fname; |
d9c40fd6 |
682 | |
c18ed097 |
683 | fname = make_filename(INDEX_RANDSEED, NULL); |
e3ac3c05 |
684 | /* |
685 | * Don't truncate the random seed file if it already exists; if |
686 | * something goes wrong half way through writing it, it would |
687 | * be better to leave the old data there than to leave it empty. |
688 | */ |
689 | fd = open(fname, O_CREAT | O_WRONLY, 0600); |
d9c40fd6 |
690 | if (fd < 0) { |
b9fa4c28 |
691 | if (errno != ENOENT) { |
692 | char *msg = dupprintf("Unable to write random seed: open(\"%s\") " |
693 | "returned '%s'", fname, strerror(errno)); |
694 | nonfatal(msg); |
695 | sfree(msg); |
b4bc5384 |
696 | sfree(fname); |
b9fa4c28 |
697 | return; |
698 | } |
c18ed097 |
699 | char *dir; |
d9c40fd6 |
700 | |
c18ed097 |
701 | dir = make_filename(INDEX_DIR, NULL); |
b9fa4c28 |
702 | if (mkdir(dir, 0700) < 0) { |
703 | char *msg = dupprintf("Unable to write random seed: mkdir(\"%s\") " |
704 | "returned '%s'", dir, strerror(errno)); |
705 | nonfatal(msg); |
706 | sfree(msg); |
b4bc5384 |
707 | sfree(fname); |
b9fa4c28 |
708 | sfree(dir); |
709 | return; |
710 | } |
c18ed097 |
711 | sfree(dir); |
712 | |
e3ac3c05 |
713 | fd = open(fname, O_CREAT | O_WRONLY, 0600); |
b9fa4c28 |
714 | if (errno != ENOENT) { |
715 | char *msg = dupprintf("Unable to write random seed: open(\"%s\") " |
716 | "returned '%s'", fname, strerror(errno)); |
717 | nonfatal(msg); |
718 | sfree(msg); |
b4bc5384 |
719 | sfree(fname); |
b9fa4c28 |
720 | return; |
721 | } |
d9c40fd6 |
722 | } |
723 | |
724 | while (len > 0) { |
725 | int ret = write(fd, data, len); |
b9fa4c28 |
726 | if (ret < 0) { |
727 | char *msg = dupprintf("Unable to write random seed: write " |
728 | "returned '%s'", strerror(errno)); |
729 | nonfatal(msg); |
730 | sfree(msg); |
731 | break; |
732 | } |
d9c40fd6 |
733 | len -= ret; |
734 | data = (char *)data + len; |
735 | } |
736 | |
737 | close(fd); |
c18ed097 |
738 | sfree(fname); |
1709795f |
739 | } |
740 | |
741 | void cleanup_all(void) |
742 | { |
743 | } |