Test program

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@23631 ec53bebd-3082-4978-b11e-865c3cabbd6b
2008-08-18 20:14:29 +00:00
parent 0077b6dd15
commit a62c81e366
1 changed files with 123 additions and 0 deletions
--- a/lib/hcrypto/example_evp_cipher.c
+++ b/lib/hcrypto/example_evp_cipher.c
@@ -0,0 +1,123 @@
+/*
+ * Copyright (c) 2008 Kungliga Tekniska H<>gskolan
+ * (Royal Institute of Technology, Stockholm, Sweden).
+ * 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. Neither the name of the Institute 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 THE INSTITUTE 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 INSTITUTE 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 <krb5-types.h> /* should really be stdint.h */
+#include <hcrypto/evp.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <err.h>
+
+static char key[16] =
+    "\xaa\xbb\x45\xd4\xaa\xbb\x45\xd4"
+    "\xaa\xbb\x45\xd4\xaa\xbb\x45\xd4";
+static char ivec[16] =
+    "\xaa\xbb\x45\xd4\xaa\xbb\x45\xd4"
+    "\xaa\xbb\x45\xd4\xaa\xbb\x45\xd4";
+
+static void
+usage(int exit_code)
+{
+    printf("usage: %s in out\n", getprogname());
+    exit(exit_code);
+}
+
+
+int
+main(int argc, char **argv)
+{
+    int encryptp = 1;
+    const char *ifn, *ofn;
+    FILE *in, *out;
+    void *ibuf, *obuf;
+    int ilen, olen;
+    size_t block_size; 
+    const EVP_CIPHER *c = EVP_aes_128_cbc();
+    EVP_CIPHER_CTX ctx;
+    int ret;
+
+    setprogname(argv[0]);
+
+    if (argc == 2) {
+	if (strcmp(argv[1], "--version") == 0) {
+	    printf("version");
+	    exit(0);
+	}
+	if (strcmp(argv[1], "--help") == 0)
+	    usage(0);
+    } else if (argc == 3) {
+	ifn = argv[1];
+	ofn = argv[2];
+    } else
+	usage(1);
+
+    /* 
+     * Size is prime to trigger mis-match between io block size and
+     * encryption block size.
+     */
+    block_size = 1021;
+
+    in = fopen("in", "r");
+    if (in == NULL)
+	errx(1, "failed to open input file");
+    out = fopen("out", "w");
+    if (out == NULL)
+	errx(1, "failed to open output file");
+    
+    ibuf = malloc(block_size);
+    obuf = malloc(block_size + EVP_CIPHER_block_size(c));
+
+    EVP_CIPHER_CTX_init(&ctx);
+    EVP_CipherInit_ex(&ctx, c, NULL, key, ivec, encryptp);
+	
+    while ((ilen = fread(ibuf, 1, block_size, in)) > 0) {
+	ret = EVP_CipherUpdate(&ctx, obuf, &olen, ibuf, ilen);
+	if (ret != 1) {
+	    EVP_CIPHER_CTX_cleanup(&ctx);
+	    errx(1, "EVP_CipherUpdate failed");
+	}
+	fwrite(obuf, 1, olen, out);
+    }
+    fclose(in);
+
+    ret = EVP_CipherFinal_ex(&ctx, obuf, &olen);
+    EVP_CIPHER_CTX_cleanup(&ctx);
+    if (ret != 1)
+	errx(1, "EVP_CipherFinal_ex failed");
+
+    fwrite(obuf, 1, olen, out);
+    fclose(out);
+    
+    return 0;
+}