/* apps/enc.c */

/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)

 * All rights reserved.

 *

 * This package is an SSL implementation written

 * by Eric Young (eay@cryptsoft.com).

 * The implementation was written so as to conform with Netscapes SSL.

 * 

 * This library is free for commercial and non-commercial use as long as

 * the following conditions are aheared to.  The following conditions

 * apply to all code found in this distribution, be it the RC4, RSA,

 * lhash, DES, etc., code; not just the SSL code.  The SSL documentation

 * included with this distribution is covered by the same copyright terms

 * except that the holder is Tim Hudson (tjh@cryptsoft.com).

 * 

 * Copyright remains Eric Young's, and as such any Copyright notices in

 * the code are not to be removed.

 * If this package is used in a product, Eric Young should be given attribution

 * as the author of the parts of the library used.

 * This can be in the form of a textual message at program startup or

 * in documentation (online or textual) provided with the package.

 * 

 * 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 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 cryptographic software written by

 *     Eric Young (eay@cryptsoft.com)"

 *    The word 'cryptographic' can be left out if the rouines from the library

 *    being used are not cryptographic related :-).

 * 4. If you include any Windows specific code (or a derivative thereof) from 

 *    the apps directory (application code) you must include an acknowledgement:

 *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"

 * 

 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``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 AUTHOR 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.

 * 

 * The licence and distribution terms for any publically available version or

 * derivative of this code cannot be changed.  i.e. this code cannot simply be

 * copied and put under another distribution licence

 * [including the GNU Public Licence.]

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include "apps.h"

#include <openssl/bio.h>

#include <openssl/err.h>

#include <openssl/evp.h>

#include <openssl/objects.h>

#include <openssl/x509.h>

#include <openssl/rand.h>

#include <openssl/pem.h>

#include <ctype.h>

int set_hex(char *in,unsigned char *out,int size);

#undef SIZE

#undef BSIZE

#undef PROG

#define SIZE	(512)

#define BSIZE	(8*1024)

#define	PROG	enc_main

static void show_ciphers(const OBJ_NAME *name,void *bio_)

	{

	BIO *bio=bio_;

	static int n;

	if(!islower((unsigned char)*name->name))

		return;

	BIO_printf(bio,"-%-25s",name->name);

	if(++n == 3)

		{

		BIO_printf(bio,"\n");

		n=0;

		}

	else

		BIO_printf(bio," ");

	}

int MAIN(int, char **);

int MAIN(int argc, char **argv)

	{

#ifndef OPENSSL_NO_ENGINE

	ENGINE *e = NULL;

#endif

	static const char magic[]="Salted__";

	char mbuf[sizeof magic-1];

	char *strbuf=NULL;

	unsigned char *buff=NULL,*bufsize=NULL;

	int bsize=BSIZE,verbose=0;

	int ret=1,inl;

	int nopad = 0;

	unsigned char key[EVP_MAX_KEY_LENGTH],iv[EVP_MAX_IV_LENGTH];

	unsigned char salt[PKCS5_SALT_LEN];

	char *str=NULL, *passarg = NULL, *pass = NULL;

	char *hkey=NULL,*hiv=NULL,*hsalt = NULL;

	char *md=NULL;

	int enc=1,printkey=0,i,base64=0;

	int debug=0,olb64=0,nosalt=0;

	const EVP_CIPHER *cipher=NULL,*c;

	EVP_CIPHER_CTX *ctx = NULL;

	char *inf=NULL,*outf=NULL;

	BIO *in=NULL,*out=NULL,*b64=NULL,*benc=NULL,*rbio=NULL,*wbio=NULL;

#define PROG_NAME_SIZE  39

	char pname[PROG_NAME_SIZE+1];

#ifndef OPENSSL_NO_ENGINE

	char *engine = NULL;

#endif

	const EVP_MD *dgst=NULL;

	apps_startup();

	if (bio_err == NULL)

		if ((bio_err=BIO_new(BIO_s_file())) != NULL)

			BIO_set_fp(bio_err,stderr,BIO_NOCLOSE|BIO_FP_TEXT);

	if (!load_config(bio_err, NULL))

		goto end;

	/* first check the program name */

	program_name(argv[0],pname,sizeof pname);

	if (strcmp(pname,"base64") == 0)

		base64=1;

	cipher=EVP_get_cipherbyname(pname);

	if (!base64 && (cipher == NULL) && (strcmp(pname,"enc") != 0))

		{

		BIO_printf(bio_err,"%s is an unknown cipher\n",pname);

		goto bad;

		}

	argc--;

	argv++;

	while (argc >= 1)

		{

		if	(strcmp(*argv,"-e") == 0)

			enc=1;

		else if (strcmp(*argv,"-in") == 0)

			{

			if (--argc < 1) goto bad;

			inf= *(++argv);

			}

		else if (strcmp(*argv,"-out") == 0)

			{

			if (--argc < 1) goto bad;

			outf= *(++argv);

			}

		else if (strcmp(*argv,"-pass") == 0)

			{

			if (--argc < 1) goto bad;

			passarg= *(++argv);

			}

#ifndef OPENSSL_NO_ENGINE

		else if (strcmp(*argv,"-engine") == 0)

			{

			if (--argc < 1) goto bad;

			engine= *(++argv);

			}

#endif

		else if	(strcmp(*argv,"-d") == 0)

			enc=0;

		else if	(strcmp(*argv,"-p") == 0)

			printkey=1;

		else if	(strcmp(*argv,"-v") == 0)

			verbose=1;

		else if	(strcmp(*argv,"-nopad") == 0)

			nopad=1;

		else if	(strcmp(*argv,"-salt") == 0)

			nosalt=0;

		else if	(strcmp(*argv,"-nosalt") == 0)

			nosalt=1;

		else if	(strcmp(*argv,"-debug") == 0)

			debug=1;

		else if	(strcmp(*argv,"-P") == 0)

			printkey=2;

		else if	(strcmp(*argv,"-A") == 0)

			olb64=1;

		else if	(strcmp(*argv,"-a") == 0)

			base64=1;

		else if	(strcmp(*argv,"-base64") == 0)

			base64=1;

		else if (strcmp(*argv,"-bufsize") == 0)

			{

			if (--argc < 1) goto bad;

			bufsize=(unsigned char *)*(++argv);

			}

		else if (strcmp(*argv,"-k") == 0)

			{

			if (--argc < 1) goto bad;

			str= *(++argv);

			}

		else if (strcmp(*argv,"-kfile") == 0)

			{

			static char buf[128];

			FILE *infile;

			char *file;

			if (--argc < 1) goto bad;

			file= *(++argv);

			infile=fopen(file,"r");

			if (infile == NULL)

				{

				BIO_printf(bio_err,"unable to read key from '%s'\n",

					file);

				goto bad;

				}

			buf[0]='\0';

			fgets(buf,sizeof buf,infile);

			fclose(infile);

			i=strlen(buf);

			if ((i > 0) &&

				((buf[i-1] == '\n') || (buf[i-1] == '\r')))

				buf[--i]='\0';

			if ((i > 0) &&

				((buf[i-1] == '\n') || (buf[i-1] == '\r')))

				buf[--i]='\0';

			if (i < 1)

				{

				BIO_printf(bio_err,"zero length password\n");

				goto bad;

				}

			str=buf;

			}

		else if (strcmp(*argv,"-K") == 0)

			{

			if (--argc < 1) goto bad;

			hkey= *(++argv);

			}

		else if (strcmp(*argv,"-S") == 0)

			{

			if (--argc < 1) goto bad;

			hsalt= *(++argv);

			}

		else if (strcmp(*argv,"-iv") == 0)

			{

			if (--argc < 1) goto bad;

			hiv= *(++argv);

			}

		else if (strcmp(*argv,"-md") == 0)

			{

			if (--argc < 1) goto bad;

			md= *(++argv);

			}

		else if	((argv[0][0] == '-') &&

			((c=EVP_get_cipherbyname(&(argv[0][1]))) != NULL))

			{

			cipher=c;

			}

		else if (strcmp(*argv,"-none") == 0)

			cipher=NULL;

		else

			{

			BIO_printf(bio_err,"unknown option '%s'\n",*argv);

bad:

			BIO_printf(bio_err,"options are\n");

			BIO_printf(bio_err,"%-14s input file\n","-in <file>");

			BIO_printf(bio_err,"%-14s output file\n","-out <file>");

			BIO_printf(bio_err,"%-14s pass phrase source\n","-pass <arg>");

			BIO_printf(bio_err,"%-14s encrypt\n","-e");

			BIO_printf(bio_err,"%-14s decrypt\n","-d");

			BIO_printf(bio_err,"%-14s base64 encode/decode, depending on encryption flag\n","-a/-base64");

			BIO_printf(bio_err,"%-14s passphrase is the next argument\n","-k");

			BIO_printf(bio_err,"%-14s passphrase is the first line of the file argument\n","-kfile");

			BIO_printf(bio_err,"%-14s the next argument is the md to use to create a key\n","-md");

			BIO_printf(bio_err,"%-14s   from a passphrase.  One of md2, md5, sha or sha1\n","");

			BIO_printf(bio_err,"%-14s key/iv in hex is the next argument\n","-K/-iv");

			BIO_printf(bio_err,"%-14s print the iv/key (then exit if -P)\n","-[pP]");

			BIO_printf(bio_err,"%-14s buffer size\n","-bufsize <n>");

#ifndef OPENSSL_NO_ENGINE

			BIO_printf(bio_err,"%-14s use engine e, possibly a hardware device.\n","-engine e");

#endif

			BIO_printf(bio_err,"Cipher Types\n");

			OBJ_NAME_do_all_sorted(OBJ_NAME_TYPE_CIPHER_METH,

					       show_ciphers,

					       bio_err);

			BIO_printf(bio_err,"\n");

			goto end;

			}

		argc--;

		argv++;

		}

#ifndef OPENSSL_NO_ENGINE

        e = setup_engine(bio_err, engine, 0);

#endif

	if (md && (dgst=EVP_get_digestbyname(md)) == NULL)

		{

		BIO_printf(bio_err,"%s is an unsupported message digest type\n",md);

		goto end;

		}

	if (dgst == NULL)

		{

		dgst = EVP_md5();

		}

	if (bufsize != NULL)

		{

		unsigned long n;

		for (n=0; *bufsize; bufsize++)

			{

			i= *bufsize;

			if ((i <= '9') && (i >= '0'))

				n=n*10+i-'0';

			else if (i == 'k')

				{

				n*=1024;

				bufsize++;

				break;

				}

			}

		if (*bufsize != '\0')

			{

			BIO_printf(bio_err,"invalid 'bufsize' specified.\n");

			goto end;

			}

		/* It must be large enough for a base64 encoded line */

		if (base64 && n < 80) n=80;

		bsize=(int)n;

		if (verbose) BIO_printf(bio_err,"bufsize=%d\n",bsize);

		}

	strbuf=OPENSSL_malloc(SIZE);

	buff=(unsigned char *)OPENSSL_malloc(EVP_ENCODE_LENGTH(bsize));

	if ((buff == NULL) || (strbuf == NULL))

		{

		BIO_printf(bio_err,"OPENSSL_malloc failure %ld\n",(long)EVP_ENCODE_LENGTH(bsize));

		goto end;

		}

	in=BIO_new(BIO_s_file());

	out=BIO_new(BIO_s_file());

	if ((in == NULL) || (out == NULL))

		{

		ERR_print_errors(bio_err);

		goto end;

		}

	if (debug)

		{

		BIO_set_callback(in,BIO_debug_callback);

		BIO_set_callback(out,BIO_debug_callback);

		BIO_set_callback_arg(in,(char *)bio_err);

		BIO_set_callback_arg(out,(char *)bio_err);

		}

	if (inf == NULL)

	        {

		if (bufsize != NULL)

			setvbuf(stdin, (char *)NULL, _IONBF, 0);

		BIO_set_fp(in,stdin,BIO_NOCLOSE);

	        }

	else

		{

		if (BIO_read_filename(in,inf) <= 0)

			{

			perror(inf);

			goto end;

			}

		}

	if(!str && passarg) {

		if(!app_passwd(bio_err, passarg, NULL, &pass, NULL)) {

			BIO_printf(bio_err, "Error getting password\n");

			goto end;

		}

		str = pass;

	}

	if ((str == NULL) && (cipher != NULL) && (hkey == NULL))

		{

		for (;;)

			{

			char buf[200];

			BIO_snprintf(buf,sizeof buf,"enter %s %s password:",

				     OBJ_nid2ln(EVP_CIPHER_nid(cipher)),

				     (enc)?"encryption":"decryption");

			strbuf[0]='\0';

			i=EVP_read_pw_string((char *)strbuf,SIZE,buf,enc);

			if (i == 0)

				{

				if (strbuf[0] == '\0')

					{

					ret=1;

					goto end;

					}

				str=strbuf;

				break;

				}

			if (i < 0)

				{

				BIO_printf(bio_err,"bad password read\n");

				goto end;

				}

			}

		}

	if (outf == NULL)

		{

		BIO_set_fp(out,stdout,BIO_NOCLOSE);

		if (bufsize != NULL)

			setvbuf(stdout, (char *)NULL, _IONBF, 0);

#ifdef OPENSSL_SYS_VMS

		{

		BIO *tmpbio = BIO_new(BIO_f_linebuffer());

		out = BIO_push(tmpbio, out);

		}

#endif

		}

	else

		{

		if (BIO_write_filename(out,outf) <= 0)

			{

			perror(outf);

			goto end;

			}

		}

	rbio=in;

	wbio=out;

	if (base64)

		{

		if ((b64=BIO_new(BIO_f_base64())) == NULL)

			goto end;

		if (debug)

			{

			BIO_set_callback(b64,BIO_debug_callback);

			BIO_set_callback_arg(b64,(char *)bio_err);

			}

		if (olb64)

			BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL);

		if (enc)

			wbio=BIO_push(b64,wbio);

		else

			rbio=BIO_push(b64,rbio);

		}

	if (cipher != NULL)

		{

		/* Note that str is NULL if a key was passed on the command

		 * line, so we get no salt in that case. Is this a bug?

		 */

		if (str != NULL)

			{

			/* Salt handling: if encrypting generate a salt and

			 * write to output BIO. If decrypting read salt from

			 * input BIO.

			 */

			unsigned char *sptr;

			if(nosalt) sptr = NULL;

			else {

				if(enc) {

					if(hsalt) {

						if(!set_hex(hsalt,salt,sizeof salt)) {

							BIO_printf(bio_err,

								"invalid hex salt value\n");

							goto end;

						}

					} else if (RAND_pseudo_bytes(salt, sizeof salt) < 0)

						goto end;

					/* If -P option then don't bother writing */

					if((printkey != 2)

					   && (BIO_write(wbio,magic,

							 sizeof magic-1) != sizeof magic-1

					       || BIO_write(wbio,

							    (char *)salt,

							    sizeof salt) != sizeof salt)) {

						BIO_printf(bio_err,"error writing output file\n");

						goto end;

					}

				} else if(BIO_read(rbio,mbuf,sizeof mbuf) != sizeof mbuf

					  || BIO_read(rbio,

						      (unsigned char *)salt,

				    sizeof salt) != sizeof salt) {

					BIO_printf(bio_err,"error reading input file\n");

					goto end;

				} else if(memcmp(mbuf,magic,sizeof magic-1)) {

				    BIO_printf(bio_err,"bad magic number\n");

				    goto end;

				}

				sptr = salt;

			}

			EVP_BytesToKey(cipher,dgst,sptr,

				(unsigned char *)str,

				strlen(str),1,key,iv);

			/* zero the complete buffer or the string

			 * passed from the command line

			 * bug picked up by

			 * Larry J. Hughes Jr. <hughes@indiana.edu> */

			if (str == strbuf)

				OPENSSL_cleanse(str,SIZE);

			else

				OPENSSL_cleanse(str,strlen(str));

			}

		if ((hiv != NULL) && !set_hex(hiv,iv,sizeof iv))

			{

			BIO_printf(bio_err,"invalid hex iv value\n");

			goto end;

			}

		if ((hiv == NULL) && (str == NULL))

			{

			/* No IV was explicitly set and no IV was generated

			 * during EVP_BytesToKey. Hence the IV is undefined,

			 * making correct decryption impossible. */

			BIO_printf(bio_err, "iv undefined\n");

			goto end;

			}

		if ((hkey != NULL) && !set_hex(hkey,key,sizeof key))

			{

			BIO_printf(bio_err,"invalid hex key value\n");

			goto end;

			}

		if ((benc=BIO_new(BIO_f_cipher())) == NULL)

			goto end;

		/* Since we may be changing parameters work on the encryption

		 * context rather than calling BIO_set_cipher().

		 */

		BIO_get_cipher_ctx(benc, &ctx);

		if (!EVP_CipherInit_ex(ctx, cipher, NULL, NULL, NULL, enc))

			{

			BIO_printf(bio_err, "Error setting cipher %s\n",

				EVP_CIPHER_name(cipher));

			ERR_print_errors(bio_err);

			goto end;

			}

		if (nopad)

			EVP_CIPHER_CTX_set_padding(ctx, 0);

		if (!EVP_CipherInit_ex(ctx, NULL, NULL, key, iv, enc))

			{

			BIO_printf(bio_err, "Error setting cipher %s\n",

				EVP_CIPHER_name(cipher));

			ERR_print_errors(bio_err);

			goto end;

			}

		if (debug)

			{

			BIO_set_callback(benc,BIO_debug_callback);

			BIO_set_callback_arg(benc,(char *)bio_err);

			}

		if (printkey)

			{

			if (!nosalt)

				{

				printf("salt=");

				for (i=0; i<(int)sizeof(salt); i++)

					printf("%02X",salt[i]);

				printf("\n");

				}

			if (cipher->key_len > 0)

				{

				printf("key=");

				for (i=0; i<cipher->key_len; i++)

					printf("%02X",key[i]);

				printf("\n");

				}

			if (cipher->iv_len > 0)

				{

				printf("iv =");

				for (i=0; i<cipher->iv_len; i++)

					printf("%02X",iv[i]);

				printf("\n");

				}

			if (printkey == 2)

				{

				ret=0;

				goto end;

				}

			}

		}

	/* Only encrypt/decrypt as we write the file */

	if (benc != NULL)

		wbio=BIO_push(benc,wbio);

	for (;;)

		{

		inl=BIO_read(rbio,(char *)buff,bsize);

		if (inl <= 0) break;

		if (BIO_write(wbio,(char *)buff,inl) != inl)

			{

			BIO_printf(bio_err,"error writing output file\n");

			goto end;

			}

		}

	if (!BIO_flush(wbio))

		{

		BIO_printf(bio_err,"bad decrypt\n");

		goto end;

		}

	ret=0;

	if (verbose)

		{

		BIO_printf(bio_err,"bytes read   :%8ld\n",BIO_number_read(in));

		BIO_printf(bio_err,"bytes written:%8ld\n",BIO_number_written(out));

		}

end:

	ERR_print_errors(bio_err);

	if (strbuf != NULL) OPENSSL_free(strbuf);

	if (buff != NULL) OPENSSL_free(buff);

	if (in != NULL) BIO_free(in);

	if (out != NULL) BIO_free_all(out);

	if (benc != NULL) BIO_free(benc);

	if (b64 != NULL) BIO_free(b64);

	if(pass) OPENSSL_free(pass);

	apps_shutdown();

	OPENSSL_EXIT(ret);

	}

int set_hex(char *in, unsigned char *out, int size)

	{

	int i,n;

	unsigned char j;

	n=strlen(in);

	if (n > (size*2))

		{

		BIO_printf(bio_err,"hex string is too long\n");

		return(0);

		}

	memset(out,0,size);

	for (i=0; i<n; i++)

		{

		j=(unsigned char)*in;

		*(in++)='\0';

		if (j == 0) break;

		if ((j >= '0') && (j <= '9'))

			j-='0';

		else if ((j >= 'A') && (j <= 'F'))

			j=j-'A'+10;

		else if ((j >= 'a') && (j <= 'f'))

			j=j-'a'+10;

		else

			{

			BIO_printf(bio_err,"non-hex digit\n");

			return(0);

			}

		if (i&1)

			out[i/2]|=j;

		else

			out[i/2]=(j<<4);

		}

	return(1);

	}