From 846c9f6bcc9aa86067850088808ff8d724a0d18f Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Sat, 17 Oct 2009 03:14:47 +0000
Subject: checkpoint -- started doing some R4 encryption support

git-svn-id: svn+q:///qpdf/trunk@807 71b93d88-0707-0410-a8cf-f5a4172ac649
---
 libtests/aes.cc | 97 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 libtests/aes.cc

(limited to 'libtests/aes.cc')

diff --git a/libtests/aes.cc b/libtests/aes.cc
new file mode 100644
index 00000000..0c4b8e19
--- /dev/null
+++ b/libtests/aes.cc
@@ -0,0 +1,97 @@
+#include <qpdf/Pl_AES_PDF.hh>
+#include <qpdf/Pl_StdioFile.hh>
+
+#include <stdio.h>
+#include <string.h>
+#include <iostream>
+#include <stdlib.h>
+
+static void usage()
+{
+    std::cerr << "Usage: aes { -encrypt | -decrypt }"
+	      << " hex-key infile outfile" << std::endl;
+    exit(2);
+}
+
+int main(int argc, char* argv[])
+{
+    if (argc != 5)
+    {
+	usage();
+   }
+
+    char* action = argv[1];
+    char* hexkey = argv[2];
+    char* infilename = argv[3];
+    char* outfilename = argv[4];
+
+    bool encrypt = true;
+    if (strcmp(action, "-decrypt") == 0)
+    {
+	encrypt = false;
+    }
+    else if (strcmp(action, "-encrypt") != 0)
+    {
+	usage();
+    }
+
+    unsigned int hexkeylen = strlen(hexkey);
+    unsigned int keylen = hexkeylen / 2;
+    if (keylen != Pl_AES_PDF::key_size)
+    {
+	std::cerr << "key length must be " << Pl_AES_PDF::key_size
+		  << " bytes" << std::endl;
+	exit(2);
+    }
+
+    FILE* infile = fopen(infilename, "rb");
+    if (infile == 0)
+    {
+	std::cerr << "can't open " << infilename << std::endl;
+	exit(2);
+    }
+
+    FILE* outfile = fopen(outfilename, "wb");
+    if (outfile == 0)
+    {
+	std::cerr << "can't open " << outfilename << std::endl;
+	exit(2);
+    }
+
+    unsigned char key[Pl_AES_PDF::key_size];
+    for (unsigned int i = 0; i < strlen(hexkey); i += 2)
+    {
+	char t[3];
+	t[0] = hexkey[i];
+	t[1] = hexkey[i + 1];
+	t[2] = '\0';
+
+	long val = strtol(t, 0, 16);
+	key[i/2] = (unsigned char) val;
+    }
+
+    Pl_StdioFile* out = new Pl_StdioFile("stdout", outfile);
+    Pl_AES_PDF* aes = new Pl_AES_PDF("aes_128_cbc", out, encrypt, key);
+
+    // 16 < buffer size, buffer_size is not a multiple of 8 for testing
+    unsigned char buf[83];
+    bool done = false;
+    while (! done)
+    {
+	int len = fread(buf, 1, sizeof(buf), infile);
+	if (len <= 0)
+	{
+	    done = true;
+	}
+	else
+	{
+	    aes->write(buf, len);
+	}
+    }
+    aes->finish();
+    delete aes;
+    delete out;
+    fclose(infile);
+    fclose(outfile);
+    return 0;
+}
-- 
cgit v1.2.3-54-g00ecf