X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=crypt.c;h=085c0563d0afb2eb23669d4084e4f83c92ff5540;hp=f227eb39907691d24f0eae360c04027c1653be75;hb=b4c9282be012ea84f7154684e9b2a469171e5e2b;hpb=35fcd604e2ec04f8667c85fa1552932b1a0e6efb

diff --git a/crypt.c b/crypt.c
index f227eb39..085c0563 100644
--- a/crypt.c
+++ b/crypt.c
@@ -65,12 +65,7 @@ static EVP_PKEY *load_key(const char *file, int private)
 {
 	BIO *key;
 	EVP_PKEY *pkey = NULL;
-	int ret = check_key_file(file, private);
 
-	if (ret < 0) {
-		PARA_ERROR_LOG("%s\n", para_strerror(-ret));
-		return NULL;
-	}
 	key = BIO_new(BIO_s_file());
 	if (!key)
 		return NULL;
@@ -134,18 +129,25 @@ static int read_rsa_bignums(const unsigned char *blob, int blen, RSA **result)
 {
 	int ret;
 	RSA *rsa;
+	BIGNUM *n, *e;
 	const unsigned char *p = blob, *end = blob + blen;
 
 	rsa = RSA_new();
 	if (!rsa)
 		return -E_BIGNUM;
-	ret = read_bignum(p, end - p, &rsa->e);
+	ret = read_bignum(p, end - p, &e);
 	if (ret < 0)
 		goto fail;
 	p += ret;
-	ret = read_bignum(p, end - p, &rsa->n);
+	ret = read_bignum(p, end - p, &n);
 	if (ret < 0)
 		goto fail;
+#ifdef HAVE_RSA_SET0_KEY
+	RSA_set0_key(rsa, n, e, NULL);
+#else
+	rsa->n = n;
+	rsa->e = e;
+#endif
 	*result = rsa;
 	return 1;
 fail:
@@ -222,6 +224,9 @@ int priv_decrypt(const char *key_file, unsigned char *outbuf,
 	struct asymmetric_key *priv;
 	int ret;
 
+	ret = check_private_key_file(key_file);
+	if (ret < 0)
+		return ret;
 	if (inlen < 0)
 		return -E_RSA;
 	ret = get_asymmetric_key(key_file, LOAD_PRIVATE_KEY, &priv);
@@ -256,27 +261,18 @@ int pub_encrypt(struct asymmetric_key *pub, unsigned char *inbuf,
 	return ret < 0? -E_ENCRYPT : ret;
 }
 
-struct aes_ctr_128_context {
-	AES_KEY key;
-	unsigned char ivec[AES_CRT128_BLOCK_SIZE];
-	unsigned char ecount[AES_CRT128_BLOCK_SIZE];
-	unsigned int num;
-};
-
 struct stream_cipher {
 	bool use_aes;
 	union {
 		RC4_KEY rc4_key;
-		struct aes_ctr_128_context aes;
+		EVP_CIPHER_CTX *aes;
 	} context;
 };
 
 struct stream_cipher *sc_new(const unsigned char *data, int len,
 		bool use_aes)
 {
-	int ret;
 	struct stream_cipher *sc = para_malloc(sizeof(*sc));
-	struct aes_ctr_128_context *aes;
 
 	sc->use_aes = use_aes;
 	if (!use_aes) {
@@ -284,17 +280,17 @@ struct stream_cipher *sc_new(const unsigned char *data, int len,
 		return sc;
 	}
 	assert(len >= 2 * AES_CRT128_BLOCK_SIZE);
-	aes = &sc->context.aes;
-	ret = AES_set_encrypt_key(data, AES_CRT128_BLOCK_SIZE * 8 /* bits */,
-		&aes->key);
-	assert(ret == 0);
-	memcpy(aes->ivec, data + AES_CRT128_BLOCK_SIZE, AES_CRT128_BLOCK_SIZE);
-	aes->num = 0;
+	sc->context.aes = EVP_CIPHER_CTX_new();
+	EVP_EncryptInit_ex(sc->context.aes, EVP_aes_128_ctr(), NULL, data,
+		data + AES_CRT128_BLOCK_SIZE);
 	return sc;
 }
 
 void sc_free(struct stream_cipher *sc)
 {
+	if (!sc)
+		return;
+	EVP_CIPHER_CTX_free(sc->context.aes);
 	free(sc);
 }
 
@@ -328,25 +324,29 @@ static void rc4_crypt(RC4_KEY *key, struct iovec *src, struct iovec *dst)
 	((char *)dst->iov_base)[len] = '\0';
 }
 
-static void aes_ctr128_crypt(struct aes_ctr_128_context *aes, struct iovec *src,
+static void aes_ctr128_crypt(EVP_CIPHER_CTX *ctx, struct iovec *src,
 		struct iovec *dst)
 {
-	size_t len = src->iov_len;
+	int ret, inlen = src->iov_len, outlen, tmplen;
 
 	*dst = (typeof(*dst)) {
 		/* Add one for the terminating zero byte. */
-		.iov_base = para_malloc(len + 1),
-		.iov_len = len
+		.iov_base = para_malloc(inlen + 1),
+		.iov_len = inlen
 	};
-	AES_ctr128_encrypt(src->iov_base, dst->iov_base, len,
-		&aes->key, aes->ivec, aes->ecount, &aes->num);
-	((char *)dst->iov_base)[len] = '\0';
+	ret = EVP_EncryptUpdate(ctx, dst->iov_base, &outlen, src->iov_base, inlen);
+	assert(ret != 0);
+	ret = EVP_EncryptFinal_ex(ctx, dst->iov_base + outlen, &tmplen);
+	assert(ret != 0);
+	outlen += tmplen;
+	((char *)dst->iov_base)[outlen] = '\0';
+	dst->iov_len = outlen;
 }
 
 void sc_crypt(struct stream_cipher *sc, struct iovec *src, struct iovec *dst)
 {
 	if (sc->use_aes)
-		return aes_ctr128_crypt(&sc->context.aes, src, dst);
+		return aes_ctr128_crypt(sc->context.aes, src, dst);
 	return rc4_crypt(&sc->context.rc4_key, src, dst);
 }