X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=crypt.c;h=b754c0919e6fbccca77c0d662bddf59c6351fca6;hp=207ad5db4e9f264091076d404adb06e12b95e8e4;hb=4744d937c4160898d1fe151257606430750e580c;hpb=6f24630aa6d5ea343f0f0626b356d03fa36f890e

diff --git a/crypt.c b/crypt.c
index 207ad5db..b754c091 100644
--- a/crypt.c
+++ b/crypt.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2005-2011 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2005-2012 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
@@ -7,7 +7,6 @@
 /** \file crypt.c Openssl-based encryption/decryption routines. */
 
 #include <regex.h>
-#include <stdbool.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <openssl/rand.h>
@@ -299,7 +298,7 @@ int sc_send_bin_buffer(struct stream_cipher_context *scc, char *buf,
 		memcpy(remainder, buf + l1, len - l1);
 		RC4(&scc->send->key, len - l1, remainder, tmp + l1);
 	}
-	ret = write_all(scc->fd, (char *)tmp, &len);
+	ret = xwrite(scc->fd, (char *)tmp, len);
 	free(tmp);
 	return ret;
 }
@@ -307,7 +306,7 @@ int sc_send_bin_buffer(struct stream_cipher_context *scc, char *buf,
 int sc_recv_bin_buffer(struct stream_cipher_context *scc, char *buf,
 		size_t size)
 {
-	unsigned char *tmp = para_malloc(size);
+	unsigned char *tmp = para_malloc(ROUND_UP(size, RC4_ALIGN));
 	ssize_t ret = recv(scc->fd, tmp, size, 0);
 
 	if (ret > 0)
@@ -318,6 +317,23 @@ int sc_recv_bin_buffer(struct stream_cipher_context *scc, char *buf,
 	return ret;
 }
 
+void sc_crypt(struct stream_cipher *sc, struct iovec *src, struct iovec *dst)
+{
+	RC4_KEY *key = &sc->key;
+
+	*dst = (typeof(*dst)) {
+		/*
+		 * Add one for the terminating zero byte. Integer overflow is
+		 * no problem here as para_malloc() aborts when given a zero
+		 * size argument.
+		 */
+		.iov_base = para_malloc(src->iov_len + 1),
+		.iov_len = src->iov_len
+	};
+	RC4(key, src->iov_len, src->iov_base, dst->iov_base);
+	((char *)dst->iov_base)[dst->iov_len] = '\0';
+}
+
 void hash_function(const char *data, unsigned long len, unsigned char *hash)
 {
 	SHA_CTX c;