1 files changed, 91 insertions, 0 deletions
diff --git a/clib/cunit/ecc.c b/clib/cunit/ecc.c
new file mode 100644
index 0000000..3ae6303
--- /dev/null
+++ b/clib/cunit/ecc.c
@@ -0,0 +1,91 @@
+/*
+ * Copyright (c) International Business Machines Corp., 2014
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <clib/exception.h>
+#include <clib/assert.h>
+#include <clib/type.h>
+
+#include <clib/ecc.h>
+#include <clib/misc.h>
+
+#include <CUnit/Basic.h>
+
+#define COUNT	10000
+#define SEED    41
+
+static int init_ecc(void) {
+    return 0;
+}
+
+static int clean_ecc(void) {
+    return 0;
+}
+
+static void ecc_1(void) {
+    int size[] = {7, 8, 16, 1024+16, 4096+32, 8*1024*4+64, 0};
+
+    for (int * s = size; *s != 0; s++) {
+        unsigned char in[*s];
+        unsigned char out[*s + (*s / 8)];
+
+        ssize_t in_sz = sizeof in;
+        ssize_t out_sz = sizeof out;
+
+        memset(in, 0, in_sz);
+        memset(out, 0, out_sz);
+
+        FILE * f = fopen("/dev/urandom", "r");
+        CU_ASSERT_FATAL(f != NULL);
+        CU_ASSERT_FATAL(fread(in, in_sz, 1, f) == 1);
+        fclose(f);
+
+        ssize_t rc = sfc_ecc_inject(out, out_sz, in, in_sz);
+        if ((*s % 8) != 0) {
+            CU_ASSERT(rc == -1);
+        } else {
+            CU_ASSERT(rc == in_sz + (in_sz / 8));
+        }
+
+        unsigned char cmp[*s];
+        ssize_t cmp_sz = sizeof cmp;
+        memset(cmp, 0, cmp_sz);
+
+        rc = sfc_ecc_remove(cmp, cmp_sz, out, out_sz);
+        if ((out_sz % 9) != 0) {
+            CU_ASSERT(rc == -1);
+        } else {
+            CU_ASSERT(rc == in_sz)
+            CU_ASSERT_FATAL(memcmp(in, cmp, in_sz) == 0);
+#ifdef DEBUG
+            dump_memory(stdout, 0, in, in_sz);
+            dump_memory(stdout, 0, cmp, cmp_sz);
+#endif
+        }
+    }
+}
+
+void ecc_test(void) {
+    CU_pSuite suite = CU_add_suite("ecc", init_ecc, clean_ecc);
+    if (NULL == suite)
+	return;
+
+    if (CU_add_test(suite, "test of --> ecc_1", ecc_1) == NULL) return;
+}