--- libaitio/src/crypt.c	2011/04/19 20:00:31	1.2
+++ libaitio/src/crypt.c	2011/10/31 13:53:51	1.4
@@ -1,11 +1,48 @@
 /*************************************************************************
-* (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
-*  by Michael Pounov <misho@openbsd-bg.org>
+* (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
+*  by Michael Pounov <misho@elwix.org>
 *
 * $Author: misho $
-* $Id: crypt.c,v 1.2 2011/04/19 20:00:31 misho Exp $
+* $Id: crypt.c,v 1.4 2011/10/31 13:53:51 misho Exp $
 *
-*************************************************************************/
+**************************************************************************
+The ELWIX and AITNET software is distributed under the following
+terms:
+
+All of the documentation and software included in the ELWIX and AITNET
+Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
+
+Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011
+	by Michael Pounov <misho@elwix.org>.  All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions
+are met:
+1. Redistributions of source code must retain the above copyright
+   notice, this list of conditions and the following disclaimer.
+2. Redistributions in binary form must reproduce the above copyright
+   notice, this list of conditions and the following disclaimer in the
+   documentation and/or other materials provided with the distribution.
+3. All advertising materials mentioning features or use of this software
+   must display the following acknowledgement:
+This product includes software developed by Michael Pounov <misho@elwix.org>
+ELWIX - Embedded LightWeight unIX and its contributors.
+4. Neither the name of AITNET nor the names of its contributors
+   may be used to endorse or promote products derived from this software
+   without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+SUCH DAMAGE.
+*/
 #include "global.h"
 
 
@@ -131,4 +168,53 @@ io_Blowfish(u_char *pInput, int inLen, u_char **ppOutp
 			*ppOutput = (u_char*) strdup((char*) buf);
 
 	return ret;
+}
+
+/*
+ * io_ctr_AES() Encrypt/Decrypt stream cipher CTR_AES
+ * @pInput = Input buffer with ASCII
+ * @inLen = Input buffer data length
+ * @ppOutput = Output buffer with cipher data, must be free after use
+ * @pKey = Key
+ * @IV = IVector/Nonce/Counter, Warning: IV must be variable, because we write there!!!
+ * return: -1 error or >-1 how many cipher blocks proceeded
+ */
+int
+io_ctr_AES(u_char *pInput, int inLen, u_char **ppOutput, u_char *pKey, u_char IV[AES_BLOCK_SIZE])
+{
+	u_int num;
+	AES_KEY key;
+	u_char ecount_buf[AES_BLOCK_SIZE] = { 0 };
+	int total = 0;
+
+	if (!pInput || !inLen || !ppOutput)
+		return -1;
+
+	*ppOutput = malloc(inLen);
+	if (!*ppOutput) {
+		LOGERR;
+		return -1;
+	} else
+		memset(*ppOutput, 0, inLen);
+
+	AES_set_encrypt_key(pKey, 128, &key);
+
+	while (inLen) {
+		num = 0;
+		memset(ecount_buf, 0, sizeof ecount_buf);
+		AES_ctr128_encrypt(pInput + total, (*ppOutput) + total, 
+				(inLen / (AES_BLOCK_SIZE - 1)) ? (AES_BLOCK_SIZE - 1) : inLen, 
+				&key, IV, ecount_buf, &num);
+		if (num < 1) {
+			free(*ppOutput);
+			*ppOutput = NULL;
+			total = -1;
+			break;
+		} else {
+			total += num;
+			inLen -= num;
+		}
+	}
+
+	return total;
 }