ml_set_crypt_key(buf, old_key, old_cm)
buf_T *buf;
char_u *old_key;
- int old_cm;
+ char_u *old_cm;
{
memfile_T *mfp = buf->b_ml.ml_mfp;
bhdr_T *hp;
DATA_BL *dp;
blocknr_T bnum;
int top;
+ int old_method;
if (mfp == NULL)
return; /* no memfile yet, nothing to do */
+ old_method = crypt_method_nr_from_name(old_cm);
+
+ /* First make sure the swapfile is in a consistent state, using the old
+ * key and method. */
+ {
+ char_u *new_key = buf->b_p_key;
+ char_u *new_buf_cm = buf->b_p_cm;
+
+ buf->b_p_key = old_key;
+ buf->b_p_cm = old_cm;
+ ml_preserve(buf, FALSE);
+ buf->b_p_key = new_key;
+ buf->b_p_cm = new_buf_cm;
+ }
/* Set the key, method and seed to be used for reading, these must be the
* old values. */
mfp->mf_old_key = old_key;
- mfp->mf_old_cm = old_cm;
- if (old_cm > 0)
+ mfp->mf_old_cm = old_method;
+ if (old_method > 0 && *old_key != NUL)
mch_memmove(mfp->mf_old_seed, mfp->mf_seed, MF_SEED_LEN);
/* Update block 0 with the crypt flag and may set a new seed. */
{
if (pp->pb_pointer[idx].pe_bnum < 0)
{
- /* Skip data block with negative block number. */
- ++idx; /* get same block again for next index */
+ /* Skip data block with negative block number.
+ * Should not happen, because of the ml_preserve()
+ * above. Get same block again for next index. */
+ ++idx;
continue;
}
bnum = pp->pb_pointer[idx].pe_bnum;
page_count = pp->pb_pointer[idx].pe_page_count;
+ idx = 0;
continue;
}
}
idx = ip->ip_index + 1; /* go to next index */
page_count = 1;
}
+ if (hp != NULL)
+ mf_put(mfp, hp, FALSE, FALSE); /* release previous block */
if (error > 0)
EMSG(_("E843: Error while updating swap file crypt"));
if (dp->db_id != DATA_ID)
return data;
+ state = ml_crypt_prepare(mfp, offset, FALSE);
+ if (state == NULL)
+ return data;
+
new_data = (char_u *)alloc(size);
if (new_data == NULL)
return NULL;
mch_memmove(new_data, dp, head_end - (char_u *)dp);
/* Encrypt the text. */
- state = ml_crypt_prepare(mfp, offset, FALSE);
crypt_encode(state, text_start, text_len, new_data + dp->db_txt_start);
crypt_free_state(state);
}
/*
- * Decrypt the text in "data" if it points to a data block.
+ * Decrypt the text in "data" if it points to an encrypted data block.
*/
void
ml_decrypt_data(mfp, data, offset, size)
|| dp->db_txt_end > size)
return; /* data was messed up */
- /* Decrypt the text in place. */
state = ml_crypt_prepare(mfp, offset, TRUE);
- crypt_decode_inplace(state, text_start, text_len);
- crypt_free_state(state);
+ if (state != NULL)
+ {
+ /* Decrypt the text in place. */
+ crypt_decode_inplace(state, text_start, text_len);
+ crypt_free_state(state);
+ }
}
}
key = buf->b_p_key;
seed = mfp->mf_seed;
}
+ if (*key == NUL)
+ return NULL;
if (method_nr == CRYPT_M_ZIP)
{
# endif
if (STRCMP(curbuf->b_p_key, oldval) != 0)
/* Need to update the swapfile. */
- ml_set_crypt_key(curbuf, oldval, crypt_get_method_nr(curbuf));
+ ml_set_crypt_key(curbuf, oldval,
+ *curbuf->b_p_cm == NUL ? p_cm : curbuf->b_p_cm);
}
else if (gvarp == &p_cm)
else
p = curbuf->b_p_cm;
if (STRCMP(s, p) != 0)
- ml_set_crypt_key(curbuf, curbuf->b_p_key,
- crypt_method_nr_from_name(s));
+ ml_set_crypt_key(curbuf, curbuf->b_p_key, s);
/* If the global value changes need to update the swapfile for all
* buffers using that value. */
for (buf = firstbuf; buf != NULL; buf = buf->b_next)
if (buf != curbuf && *buf->b_p_cm == NUL)
- ml_set_crypt_key(buf, buf->b_p_key,
- crypt_method_nr_from_name(oldval));
+ ml_set_crypt_key(buf, buf->b_p_key, oldval);
}
}
}