macstore.c

putty

    write_setting_i(handle, settingname, font.size);
    sfree(settingname);
}

int read_setting_filename(void *handle, const char *key, Filename *result)
{
    int fd;
    AliasHandle h;
    Boolean changed;
    OSErr err;
    Str255 pkey;

    if (handle == NULL) goto out;
    fd = *(int *)handle;
    UseResFile(fd);
    if (ResError() != noErr) goto out;
    c2pstrcpy(pkey, key);
    h = (AliasHandle)Get1NamedResource(rAliasType, pkey);
    if (h == NULL) goto out;
    if ((*h)->userType == 'pTTY' && (*h)->aliasSize == sizeof(**h))
	memset(result, 0, sizeof(*result));
    else {
	err = ResolveAlias(NULL, h, &result->fss, &changed);
	if (err != noErr && err != fnfErr) goto out;
	if ((*h)->userType == 'pTTY') {
	    long dirid;
	    StrFileName fname;

	    /* Tail of record is pascal string contaning leafname */
	    if (FSpGetDirID(&result->fss, &dirid, FALSE) != noErr) goto out;
	    memcpy(fname, (char *)*h + (*h)->aliasSize,
		   GetHandleSize((Handle)h) - (*h)->aliasSize);
	    err = FSMakeFSSpec(result->fss.vRefNum, dirid, fname,
			       &result->fss);
	    if (err != noErr && err != fnfErr) goto out;
	}
    }
    ReleaseResource((Handle)h);
    if (ResError() != noErr) goto out;
    return 1;

  out:
    return 0;
}

void write_setting_filename(void *handle, const char *key, Filename fn)
{
    int fd = *(int *)handle;
    AliasHandle h;
    int id;
    OSErr error;
    Str255 pkey;

    UseResFile(fd);
    if (ResError() != noErr)
        fatalbox("Failed to open saved session (%d)", ResError());

    if (filename_is_null(fn)) {
	/* Generate a special "null" alias */
	h = (AliasHandle)NewHandle(sizeof(**h));
	if (h == NULL)
	    fatalbox("Failed to create fake alias");
	(*h)->userType = 'pTTY';
	(*h)->aliasSize = sizeof(**h);
    } else {
	error = NewAlias(NULL, &fn.fss, &h);
	if (error == fnfErr) {
	    /*
	     * NewAlias can't create an alias for a nonexistent file.
	     * Create an alias for the directory, and record the
	     * filename as well.
	     */
	    FSSpec tmpfss;

	    FSMakeFSSpec(fn.fss.vRefNum, fn.fss.parID, NULL, &tmpfss);
	    error = NewAlias(NULL, &tmpfss, &h);
	    if (error != noErr)
		fatalbox("Failed to create alias");
	    (*h)->userType = 'pTTY';
	    SetHandleSize((Handle)h, (*h)->aliasSize + fn.fss.name[0] + 1);
	    if (MemError() != noErr)
		fatalbox("Failed to create alias");
	    memcpy((char *)*h + (*h)->aliasSize, fn.fss.name,
		   fn.fss.name[0] + 1);
	}
	if (error != noErr)
	    fatalbox("Failed to create alias");
    }
    /* Put the data in a resource. */
    id = Unique1ID(rAliasType);
    if (ResError() != noErr)
	fatalbox("Failed to get ID for resource %s (%d)", key, ResError());
    c2pstrcpy(pkey, key);
    AddResource((Handle)h, rAliasType, id, pkey);
    if (ResError() != noErr)
	fatalbox("Failed to add resource %s (%d)", key, ResError());
}

void close_settings_r(void *handle) {
    int fd;

    if (handle == NULL) return;
    fd = *(int *)handle;
    CloseResFile(fd);
    if (ResError() != noErr)
	fatalbox("Close of saved session failed (%d)", ResError());
    sfree(handle);
}

void del_settings(char const *sessionname) {
    OSErr error;
    FSSpec sessfile;
    short sessVRefNum;
    long sessDirID;
    Str255 psessionname;

    error = get_session_dir(kDontCreateFolder, &sessVRefNum, &sessDirID);

    c2pstrcpy(psessionname, sessionname);
    error = FSMakeFSSpec(sessVRefNum, sessDirID, psessionname, &sessfile);
    if (error != noErr) goto out;

    error = FSpDelete(&sessfile);
    return;
  out:
    fatalbox("Delete session failed (%d)", error);
}

struct enum_settings_state {
    short vRefNum;
    long dirID;
    int index;
};

void *enum_settings_start(void) {
    OSErr error;
    struct enum_settings_state *state;

    state = snew(struct enum_settings_state);
    error = get_session_dir(kDontCreateFolder, &state->vRefNum, &state->dirID);
    if (error != noErr) {
	sfree(state);
	return NULL;
    }
    state->index = 1;
    return state;
}

char *enum_settings_next(void *handle, char *buffer, int buflen) {
    struct enum_settings_state *e = handle;
    CInfoPBRec pb;
    OSErr error = noErr;
    Str255 name;

    if (e == NULL) return NULL;
    do {
	pb.hFileInfo.ioNamePtr = name;
	pb.hFileInfo.ioVRefNum = e->vRefNum;
	pb.hFileInfo.ioDirID = e->dirID;
	pb.hFileInfo.ioFDirIndex = e->index++;
	error = PBGetCatInfoSync(&pb);
	if (error != noErr) return NULL;
    } while (!((pb.hFileInfo.ioFlAttrib & ioDirMask) == 0 &&
	       pb.hFileInfo.ioFlFndrInfo.fdCreator == PUTTY_CREATOR &&
	       pb.hFileInfo.ioFlFndrInfo.fdType == SESS_TYPE &&
	       name[0] < buflen));

    p2cstrcpy(buffer, name);
    return buffer;
}

void enum_settings_finish(void *handle) {

    safefree(handle);
}

#define SEED_SIZE 512

void read_random_seed(noise_consumer_t consumer)
{
    short puttyVRefNum;
    long puttyDirID;
    OSErr error;
    char buf[SEED_SIZE];
    short refnum;
    long count = SEED_SIZE;

    if (get_putty_dir(kDontCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
	return;
    if (HOpenDF(puttyVRefNum, puttyDirID, "\pPuTTY Random Seed", fsRdPerm,
		&refnum) != noErr)
	return;
    error = FSRead(refnum, &count, buf);
    if (error != noErr && error != eofErr)
	return;
    (*consumer)(buf, count);
    FSClose(refnum);
}

/*
 * We don't bother with the usual FSpExchangeFiles dance here because
 * it doesn't really matter if the old random seed gets lost.
 */
void write_random_seed(void *data, int len)
{
    short puttyVRefNum;
    long puttyDirID;
    OSErr error;
    FSSpec dstfile;
    short refnum;
    long count = len;

    if (get_putty_dir(kCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
	return;

    error = FSMakeFSSpec(puttyVRefNum, puttyDirID, "\pPuTTY Random Seed",
			 &dstfile);
    if (error == fnfErr) {
	/* Set up standard resources */
	FSpCreateResFile(&dstfile, INTERNAL_CREATOR, SEED_TYPE, smRoman);
	refnum = FSpOpenResFile(&dstfile, fsWrPerm);
	if (ResError() == noErr) {
	    copy_resource('STR ', -16397);
	    CloseResFile(refnum);
	}
    } else if (error != noErr) return;

    if (FSpOpenDF(&dstfile, fsWrPerm, &refnum) != noErr) return;
    FSWrite(refnum, &count, data);
    FSClose(refnum);

    return;
}

/*
 * This host key cache uses a file in the PuTTY Preferences folder and
 * stores keys as individual TEXT resources in the resource fork of
 * that file.  This has two problems.  Firstly, a resource fork can
 * contain no more than 2727 resources.  Secondly, the Resource
 * Manager uses a linear search to find a particular resource, which
 * could make having lots of host keys quite slow.
 */

int verify_host_key(const char *hostname, int port,
		    const char *keytype, const char *key)
{
    short puttyVRefNum;
    long puttyDirID;
    OSErr error;
    FSSpec keyfile;
    short refnum;
    char *resname;
    Str255 presname;
    char *resvalue;
    Handle reshandle;
    int len, compare;

    if (get_putty_dir(kCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
	return 1;

    error = FSMakeFSSpec(puttyVRefNum, puttyDirID, "\pSSH Host Keys",
			 &keyfile);
    if (error == fnfErr) {
	/* Keys file doesn't exist yet, so we can't match the key */
	return 1;
    }

    refnum = FSpOpenResFile(&keyfile, fsRdPerm);

    if (refnum == -1) {
	/* We couldn't open the resource fork, so we can't match the key */
	return 1;
    }

    UseResFile(refnum);

    resname = dupprintf("%s@%d:%s", keytype, port, hostname);
    c2pstrcpy(presname, resname);
    reshandle = Get1NamedResource(FOUR_CHAR_CODE('TEXT'), presname);
    if (ResError() != noErr) {
	/* Couldn't open the specific resource */
	return 1;
    }

    len = GetHandleSize(reshandle);
    resvalue = snewn(len+1, char);
    memcpy(resvalue, *reshandle, len);
    resvalue[len]='\0';
    ReleaseResource(reshandle);
    CloseResFile(refnum);

    compare = strncmp(resvalue, key, strlen(resvalue));
    sfree(resname);
    sfree(resvalue);

    if (compare) {
	/* Key different */
	return 2;
    } else {
	/* Key matched */
	return 0;
    }
}

void store_host_key(const char *hostname, int port,
		    const char *keytype, const char *key)
{
    short puttyVRefNum;
    long puttyDirID;
    OSErr error;
    FSSpec keyfile;
    short keyrefnum;
    char *resname;
    Str255 presname;
    Handle resvalue;
    Handle reshandle;
    int id;

    /* Open the host key file */

    if (get_putty_dir(~kCreateFolder, &puttyVRefNum, &puttyDirID) != noErr)
	goto out;

    error = FSMakeFSSpec(puttyVRefNum, puttyDirID, "\pSSH Host Keys",
			 &keyfile);
    if (error == fnfErr) {
	/* It doesn't exist, so create it */
	FSpCreateResFile(&keyfile, INTERNAL_CREATOR, HKYS_TYPE, smRoman);
	keyrefnum = FSpOpenResFile(&keyfile, fsWrPerm);
	if (ResError() == noErr) {
	    copy_resource('STR', -16397); /* XXX: wtf is this? */
	    CloseResFile(keyrefnum);
	}
    } else if (error != noErr) goto out;

    keyrefnum = FSpOpenResFile(&keyfile, fsWrPerm);
    if (keyrefnum == -1) goto out;

    UseResFile(keyrefnum);
    resname = dupprintf("%s@%d:%s", keytype, port, hostname);
    c2pstrcpy(presname, resname);

    reshandle = Get1NamedResource(FOUR_CHAR_CODE('TEXT'), presname);
    if (reshandle != NULL) {
	/* The resource exists, we're replacing a host key */
	RemoveResource(reshandle);
    }
    error = PtrToHand(key, &resvalue, strlen(key));
    if (error != noErr) goto out;

    id = Unique1ID(FOUR_CHAR_CODE('TEXT'));
    if (ResError() != noErr) goto out;
    AddResource(resvalue, FOUR_CHAR_CODE('TEXT'), id, presname);
    if (ResError() != noErr) goto out;

    CloseResFile(keyrefnum);
    return;

  out:
    fatalbox("Writing host key failed (%d)", error);
    sfree(resname);
}
  
/*
 * Emacs magic:
 * Local Variables:
 * c-file-style: "simon"
 * End:
 */