Support for saving sessions on the Mac. This is slightly useful even in the
[u/mdw/putty] / mac / macstore.c
CommitLineData
b537dd42 1/* $Id: macstore.c,v 1.11 2003/01/18 20:09:21 ben Exp $ */
d082ac49 2
3/*
4 * macstore.c: Macintosh-specific impementation of the interface
5 * defined in storage.h
6 */
7
8#include <MacTypes.h>
9#include <Folders.h>
10#include <Memory.h>
11#include <Resources.h>
12#include <TextUtils.h>
13
b537dd42 14#include <stdio.h>
d082ac49 15#include <string.h>
16
17#include "putty.h"
18#include "storage.h"
ce283213 19#include "mac.h"
1d4cb74b 20#include "macresid.h"
d082ac49 21
d082ac49 22
d082ac49 23OSErr FSpGetDirID(FSSpec *f, long *idp, Boolean makeit);
24
25/*
26 * We store each session as a file in the "PuTTY" sub-directory of the
27 * preferences folder. Each (key,value) pair is stored as a resource.
28 */
29
0c4b7799 30OSErr get_putty_dir(Boolean makeit, short *pVRefNum, long *pDirID)
31{
d082ac49 32 OSErr error = noErr;
33 short prefVRefNum;
0c4b7799 34 FSSpec puttydir;
35 long prefDirID, puttyDirID;
d082ac49 36
37 error = FindFolder(kOnSystemDisk, kPreferencesFolderType, makeit,
38 &prefVRefNum, &prefDirID);
39 if (error != noErr) goto out;
40
41 error = FSMakeFSSpec(prefVRefNum, prefDirID, "\pPuTTY", &puttydir);
42 if (error != noErr && error != fnfErr) goto out;
43 error = FSpGetDirID(&puttydir, &puttyDirID, makeit);
44 if (error != noErr) goto out;
45
0c4b7799 46 *pVRefNum = prefVRefNum;
47 *pDirID = puttyDirID;
48
49 out:
50 return error;
51}
52
53OSErr get_session_dir(Boolean makeit, short *pVRefNum, long *pDirID) {
54 OSErr error = noErr;
55 short puttyVRefNum;
56 FSSpec sessdir;
57 long puttyDirID, sessDirID;
58
59 error = get_putty_dir(makeit, &puttyVRefNum, &puttyDirID);
60 if (error != noErr) goto out;
61 error = FSMakeFSSpec(puttyVRefNum, puttyDirID, "\pSaved Sessions",
d082ac49 62 &sessdir);
63 if (error != noErr && error != fnfErr) goto out;
64 error = FSpGetDirID(&sessdir, &sessDirID, makeit);
65 if (error != noErr) goto out;
66
0c4b7799 67 *pVRefNum = puttyVRefNum;
d082ac49 68 *pDirID = sessDirID;
69
70 out:
71 return error;
72}
73
74OSErr FSpGetDirID(FSSpec *f, long *idp, Boolean makeit) {
75 CInfoPBRec pb;
76 OSErr error = noErr;
77
78 pb.dirInfo.ioNamePtr = f->name;
79 pb.dirInfo.ioVRefNum = f->vRefNum;
80 pb.dirInfo.ioDrDirID = f->parID;
81 pb.dirInfo.ioFDirIndex = 0;
82 error = PBGetCatInfoSync(&pb);
83 if (error == fnfErr && makeit)
84 return FSpDirCreate(f, smSystemScript, idp);
85 if (error != noErr) goto out;
86 if ((pb.dirInfo.ioFlAttrib & ioDirMask) == 0) {
87 error = dirNFErr;
88 goto out;
89 }
90 *idp = pb.dirInfo.ioDrDirID;
91
92 out:
93 return error;
94}
95
1d4cb74b 96/* Copy a resource into the current resource file */
97static OSErr copy_resource(ResType restype, short resid)
98{
99 Handle h;
100 Str255 resname;
101
102 fprintf(stderr, "getting resource %x, id %d\n", restype, resid);
103 h = GetResource(restype, resid);
104 if (h != NULL) {
105 GetResInfo(h, &resid, &restype, resname);
106 DetachResource(h);
107 AddResource(h, restype, resid, resname);
108 if (ResError() == noErr)
109 WriteResource(h);
110 }
111 fprintf(stderr, "ResError() == %d\n", ResError());
112 return ResError();
113}
114
d082ac49 115struct write_settings {
116 int fd;
117 FSSpec tmpfile;
118 FSSpec dstfile;
119};
120
f5d2d791 121void *open_settings_w(char const *sessionname) {
b537dd42 122 short sessVRefNum;
123 long sessDirID;
d082ac49 124 OSErr error;
125 Str255 psessionname;
b537dd42 126 FSSpec dstfile;
d082ac49 127
d082ac49 128 error = get_session_dir(kCreateFolder, &sessVRefNum, &sessDirID);
b537dd42 129 if (error != noErr) return NULL;
d082ac49 130
131 c2pstrcpy(psessionname, sessionname);
b537dd42 132 error = FSMakeFSSpec(sessVRefNum, sessDirID, psessionname, &dstfile);
d082ac49 133 if (error == fnfErr) {
b537dd42 134 FSpCreateResFile(&dstfile, PUTTY_CREATOR, SESS_TYPE, smSystemScript);
135 if ((error = ResError()) != noErr) return NULL;
136 } else if (error != noErr) return NULL;
137
138 return open_settings_w_fsp(&dstfile);
139}
140
141/*
142 * NB: Destination file must exist.
143 */
144void *open_settings_w_fsp(FSSpec *dstfile)
145{
146 short tmpVRefNum;
147 long tmpDirID;
148 struct write_settings *ws;
149 OSErr error;
150 Str255 tmpname;
151
152 ws = smalloc(sizeof *ws);
153 ws->dstfile = *dstfile;
d082ac49 154
155 /* Create a temporary file to save to first. */
b537dd42 156 error = FindFolder(ws->dstfile.vRefNum, kTemporaryFolderType,
157 kCreateFolder, &tmpVRefNum, &tmpDirID);
d082ac49 158 if (error != noErr) goto out;
b537dd42 159 c2pstrcpy(tmpname, tmpnam(NULL));
160 error = FSMakeFSSpec(tmpVRefNum, tmpDirID, tmpname, &ws->tmpfile);
d082ac49 161 if (error != noErr && error != fnfErr) goto out;
162 if (error == noErr) {
163 error = FSpDelete(&ws->tmpfile);
164 if (error != noErr) goto out;
165 }
166 FSpCreateResFile(&ws->tmpfile, PUTTY_CREATOR, SESS_TYPE, smSystemScript);
167 if ((error = ResError()) != noErr) goto out;
168
169 ws->fd = FSpOpenResFile(&ws->tmpfile, fsWrPerm);
170 if (ws->fd == -1) {error = ResError(); goto out;}
171
1d4cb74b 172 /* Set up standard resources. Doesn't matter if these fail. */
173 copy_resource('STR ', -16396);
174 copy_resource('TMPL', TMPL_Int);
175
d082ac49 176 return ws;
177
178 out:
179 safefree(ws);
180 fatalbox("Failed to open session for write (%d)", error);
181}
182
f5d2d791 183void write_setting_s(void *handle, char const *key, char const *value) {
d082ac49 184 int fd = *(int *)handle;
185 Handle h;
186 int id;
803efcdf 187 OSErr error;
d082ac49 188
189 UseResFile(fd);
190 if (ResError() != noErr)
191 fatalbox("Failed to open saved session (%d)", ResError());
192
193 error = PtrToHand(value, &h, strlen(value));
194 if (error != noErr)
195 fatalbox("Failed to allocate memory");
196 /* Put the data in a resource. */
197 id = Unique1ID(FOUR_CHAR_CODE('TEXT'));
198 if (ResError() != noErr)
199 fatalbox("Failed to get ID for resource %s (%d)", key, ResError());
200 addresource(h, FOUR_CHAR_CODE('TEXT'), id, key);
201 if (ResError() != noErr)
202 fatalbox("Failed to add resource %s (%d)", key, ResError());
203}
204
f5d2d791 205void write_setting_i(void *handle, char const *key, int value) {
d082ac49 206 int fd = *(int *)handle;
207 Handle h;
208 int id;
209 OSErr error;
210
211 UseResFile(fd);
212 if (ResError() != noErr)
213 fatalbox("Failed to open saved session (%d)", ResError());
214
215 /* XXX assume all systems have the same "int" format */
216 error = PtrToHand(&value, &h, sizeof(int));
217 if (error != noErr)
218 fatalbox("Failed to allocate memory (%d)", error);
219
220 /* Put the data in a resource. */
221 id = Unique1ID(FOUR_CHAR_CODE('Int '));
222 if (ResError() != noErr)
223 fatalbox("Failed to get ID for resource %s (%d)", key, ResError());
224 addresource(h, FOUR_CHAR_CODE('Int '), id, key);
225 if (ResError() != noErr)
226 fatalbox("Failed to add resource %s (%d)", key, ResError());
227}
228
229void close_settings_w(void *handle) {
230 struct write_settings *ws = handle;
231 OSErr error;
232
233 CloseResFile(ws->fd);
234 if ((error = ResError()) != noErr)
235 goto out;
236 error = FSpExchangeFiles(&ws->tmpfile, &ws->dstfile);
237 if (error != noErr) goto out;
238 error = FSpDelete(&ws->tmpfile);
239 if (error != noErr) goto out;
240 return;
241
242 out:
243 fatalbox("Close of saved session failed (%d)", error);
244 safefree(handle);
245}
246
f5d2d791 247void *open_settings_r(char const *sessionname)
ce283213 248{
d082ac49 249 short sessVRefNum;
250 long sessDirID;
251 FSSpec sessfile;
252 OSErr error;
253 Str255 psessionname;
d082ac49 254
255 error = get_session_dir(kDontCreateFolder, &sessVRefNum, &sessDirID);
256
257 c2pstrcpy(psessionname, sessionname);
258 error = FSMakeFSSpec(sessVRefNum, sessDirID, psessionname, &sessfile);
259 if (error != noErr) goto out;
ce283213 260 return open_settings_r_fsp(&sessfile);
261
262 out:
263 return NULL;
264}
265
266void *open_settings_r_fsp(FSSpec *sessfile)
267{
268 OSErr error;
269 int fd;
270 int *handle;
271
272 fd = FSpOpenResFile(sessfile, fsRdPerm);
d082ac49 273 if (fd == 0) {error = ResError(); goto out;}
274
275 handle = safemalloc(sizeof *handle);
276 *handle = fd;
277 return handle;
278
279 out:
280 return NULL;
281}
282
f5d2d791 283char *read_setting_s(void *handle, char const *key, char *buffer, int buflen) {
d082ac49 284 int fd;
285 Handle h;
086efcde 286 size_t len;
d082ac49 287
288 if (handle == NULL) goto out;
289 fd = *(int *)handle;
290 UseResFile(fd);
291 if (ResError() != noErr) goto out;
292 h = get1namedresource(FOUR_CHAR_CODE('TEXT'), key);
293 if (h == NULL) goto out;
294
086efcde 295 len = GetHandleSize(h);
296 if (len + 1 > buflen) goto out;
297 memcpy(buffer, *h, len);
298 buffer[len] = '\0';
299
d082ac49 300 ReleaseResource(h);
301 if (ResError() != noErr) goto out;
302 return buffer;
303
304 out:
305 return NULL;
306}
307
f5d2d791 308int read_setting_i(void *handle, char const *key, int defvalue) {
d082ac49 309 int fd;
310 Handle h;
d082ac49 311 int value;
312
313 if (handle == NULL) goto out;
314 fd = *(int *)handle;
315 UseResFile(fd);
316 if (ResError() != noErr) goto out;
317 h = get1namedresource(FOUR_CHAR_CODE('Int '), key);
318 if (h == NULL) goto out;
319 value = *(int *)*h;
320 ReleaseResource(h);
321 if (ResError() != noErr) goto out;
322 return value;
323
324 out:
325 return defvalue;
326}
327
328void close_settings_r(void *handle) {
329 int fd;
330
331 if (handle == NULL) return;
332 fd = *(int *)handle;
333 CloseResFile(fd);
334 if (ResError() != noErr)
335 fatalbox("Close of saved session failed (%d)", ResError());
336 safefree(handle);
337}
338
f5d2d791 339void del_settings(char const *sessionname) {
d082ac49 340 OSErr error;
341 FSSpec sessfile;
342 short sessVRefNum;
343 long sessDirID;
344 Str255 psessionname;
345
346 error = get_session_dir(kDontCreateFolder, &sessVRefNum, &sessDirID);
347
348 c2pstrcpy(psessionname, sessionname);
349 error = FSMakeFSSpec(sessVRefNum, sessDirID, psessionname, &sessfile);
350 if (error != noErr) goto out;
351
352 error = FSpDelete(&sessfile);
353 return;
354 out:
355 fatalbox("Delete session failed (%d)", error);
356}
357
358struct enum_settings_state {
359 short vRefNum;
360 long dirID;
361 int index;
362};
363
364void *enum_settings_start(void) {
365 OSErr error;
366 struct enum_settings_state *state;
367
368 state = safemalloc(sizeof(*state));
369 error = get_session_dir(kDontCreateFolder, &state->vRefNum, &state->dirID);
370 if (error != noErr) {
371 safefree(state);
372 return NULL;
373 }
374 state->index = 1;
375 return state;
376}
377
378char *enum_settings_next(void *handle, char *buffer, int buflen) {
379 struct enum_settings_state *e = handle;
380 CInfoPBRec pb;
381 OSErr error = noErr;
382 Str255 name;
383
384 if (e == NULL) return NULL;
385 do {
386 pb.hFileInfo.ioNamePtr = name;
387 pb.hFileInfo.ioVRefNum = e->vRefNum;
388 pb.hFileInfo.ioDirID = e->dirID;
389 pb.hFileInfo.ioFDirIndex = e->index++;
390 error = PBGetCatInfoSync(&pb);
391 if (error != noErr) return NULL;
392 } while (!((pb.hFileInfo.ioFlAttrib & ioDirMask) == 0 &&
393 pb.hFileInfo.ioFlFndrInfo.fdCreator == PUTTY_CREATOR &&
394 pb.hFileInfo.ioFlFndrInfo.fdType == SESS_TYPE &&
395 name[0] < buflen));
396
397 p2cstrcpy(buffer, name);
398 return buffer;
399}
400
401void enum_settings_finish(void *handle) {
402
403 safefree(handle);
404}
405
0c4b7799 406#define SEED_SIZE 512
407
408void read_random_seed(noise_consumer_t consumer)
409{
410 short puttyVRefNum;
411 long puttyDirID;
412 OSErr error;
413 char buf[SEED_SIZE];
414 short refnum;
415 long count = SEED_SIZE;
416
417 if (get_putty_dir(kDontCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
418 return;
419 if (HOpenDF(puttyVRefNum, puttyDirID, "\pPuTTY Random Seed", fsRdPerm,
420 &refnum) != noErr)
421 return;
422 error = FSRead(refnum, &count, buf);
423 if (error != noErr && error != eofErr)
424 return;
425 (*consumer)(buf, count);
426 FSClose(refnum);
427}
428
604379f1 429/*
430 * We don't bother with the usual FSpExchangeFiles dance here because
431 * it doesn't really matter if the old random seed gets lost.
432 */
0c4b7799 433void write_random_seed(void *data, int len)
434{
604379f1 435 short puttyVRefNum;
436 long puttyDirID;
0c4b7799 437 OSErr error;
604379f1 438 FSSpec dstfile;
0c4b7799 439 short refnum;
440 long count = len;
441
442 if (get_putty_dir(kCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
443 return;
444
445 error = FSMakeFSSpec(puttyVRefNum, puttyDirID, "\pPuTTY Random Seed",
446 &dstfile);
1d4cb74b 447 if (error == fnfErr) {
448 /* Set up standard resources */
449 FSpCreateResFile(&dstfile, INTERNAL_CREATOR, SEED_TYPE, smRoman);
450 refnum = FSpOpenResFile(&dstfile, fsWrPerm);
451 if (ResError() == noErr) {
452 copy_resource('STR ', -16397);
453 CloseResFile(refnum);
454 }
455 } else if (error != noErr) return;
0c4b7799 456
604379f1 457 if (FSpOpenDF(&dstfile, fsWrPerm, &refnum) != noErr) return;
604379f1 458 FSWrite(refnum, &count, data);
0c4b7799 459 FSClose(refnum);
1d4cb74b 460
604379f1 461 return;
0c4b7799 462}
d082ac49 463
464/*
465 * Emacs magic:
466 * Local Variables:
467 * c-file-style: "simon"
468 * End:
469 */