summaryrefslogtreecommitdiff
path: root/code/util
diff options
context:
space:
mode:
authorUros Majstorovic <majstor@majstor.org>2017-05-23 17:28:12 +0200
committerUros Majstorovic <majstor@majstor.org>2017-05-23 17:28:12 +0200
commitab0325ae7906230f1ea82f08b27c72b075e9a13d (patch)
tree5a292826fdea4db2c86b2d82b80fa489c6c131c7 /code/util
parent3ef6719f47b734b12c0b11c725b7f12e3fb3c08a (diff)
build fixed; added lib util
Diffstat (limited to 'code/util')
-rw-r--r--code/util/Makefile21
-rw-r--r--code/util/mknode.c63
-rw-r--r--code/util/util.c91
-rw-r--r--code/util/util.h5
4 files changed, 180 insertions, 0 deletions
diff --git a/code/util/Makefile b/code/util/Makefile
new file mode 100644
index 0000000..c87a1dd
--- /dev/null
+++ b/code/util/Makefile
@@ -0,0 +1,21 @@
+CFLAGS = -I../core -pthread -O3 $(PIC)
+LDFLAGS=-lm -pthread
+
+obj=util.o
+dep=../core/libecpcore.a ../core/crypto/libecpcr.a ../core/htable/libecpht.a ../core/posix/libecptr.a ../core/posix/libecptm.a ./libecputil.a
+
+
+%.o: %.c %.h
+ $(CC) $(CFLAGS) -c $<
+
+all: libecputil.a mknode
+
+libecputil.a: $(obj)
+ $(AR) rcs libecputil.a $(obj)
+
+mknode: mknode.o libecputil.a
+ $(CC) -o $@ $(LDFLAGS) $< $(dep)
+
+clean:
+ rm -f *.o *.a
+ rm -f mknode
diff --git a/code/util/mknode.c b/code/util/mknode.c
new file mode 100644
index 0000000..0c389dc
--- /dev/null
+++ b/code/util/mknode.c
@@ -0,0 +1,63 @@
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <stdio.h>
+
+#include <core.h>
+#include "util.h"
+
+#define FN_LEN 256
+
+static char fn_key[FN_LEN];
+static char fn_node[FN_LEN];
+
+static int v_rng(void *buf, size_t bufsize) {
+ int fd;
+
+ if((fd = open("/dev/urandom", O_RDONLY)) < 0) return -1;
+ size_t nb = read(fd, buf, bufsize);
+ close(fd);
+ if (nb != bufsize) return -1;
+ return 0;
+}
+
+static void usage(char *arg) {
+ fprintf(stderr, "Usage: %s <name> [address]\n", arg);
+ exit(1);
+}
+
+int main(int argc, char *argv[]) {
+ int rv;
+ ECPContext ctx;
+ ECPDHKey key;
+ ECPNode node;
+
+ if ((argc < 2) || (argc > 3)) usage(argv[0]);
+
+ if (strlen(argv[1]) > FN_LEN - 6) usage(argv[0]);
+ strcpy(fn_node, argv[1]);
+ strcpy(fn_key, argv[1]);
+ strcat(fn_key, ".priv");
+ strcat(fn_node, ".pub");
+
+ rv = ecp_ctx_create(&ctx);
+ if (rv) goto err;
+ ctx.rng = v_rng;
+
+ rv = ecp_dhkey_generate(&ctx, &key);
+ if (rv) goto err;
+
+ rv = ecp_node_init(&ctx, &node, &key.public, (argc == 3) ? argv[2] : NULL);
+ if (rv) goto err;
+
+ rv = ecp_util_key_save(&ctx, &key, fn_key);
+ if (rv) goto err;
+
+ rv = ecp_util_node_save(&ctx, &node, fn_node);
+ if (rv) goto err;
+
+ return 0;
+ err:
+ printf("ERR:%d\n", rv);
+ return 1;
+} \ No newline at end of file
diff --git a/code/util/util.c b/code/util/util.c
new file mode 100644
index 0000000..e225ea1
--- /dev/null
+++ b/code/util/util.c
@@ -0,0 +1,91 @@
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+
+#include <core.h>
+#include "util.h"
+
+int ecp_util_key_save(ECPContext *ctx, ECPDHKey *key, char *filename) {
+ int fd;
+ ssize_t rv;
+
+ if((fd = open(filename, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR)) < 0) return ECP_ERR;
+ rv = write(fd, ctx->cr.dh_pub_get_buf(&key->public), ECP_ECDH_SIZE_KEY);
+ if (rv != ECP_ECDH_SIZE_KEY) {
+ close(fd);
+ return ECP_ERR;
+ }
+ rv = write(fd, &key->private, sizeof(key->private));
+ if (rv != sizeof(key->private)) {
+ close(fd);
+ return ECP_ERR;
+ }
+ close(fd);
+ return ECP_OK;
+}
+
+int ecp_util_key_load(ECPContext *ctx, ECPDHKey *key, char *filename) {
+ int fd;
+ ssize_t rv;
+ unsigned char buf[ECP_ECDH_SIZE_KEY];
+
+ if((fd = open(filename, O_RDONLY)) < 0) return ECP_ERR;
+ rv = read(fd, buf, ECP_ECDH_SIZE_KEY);
+ if (rv != ECP_ECDH_SIZE_KEY) {
+ close(fd);
+ return ECP_ERR;
+ }
+ rv = read(fd, &key->private, sizeof(key->private));
+ if (rv != sizeof(key->private)) {
+ close(fd);
+ return ECP_ERR;
+ }
+ close(fd);
+
+ ctx->cr.dh_pub_from_buf(&key->public, buf);
+
+ key->valid = 1;
+ return ECP_OK;
+}
+
+int ecp_util_node_save(ECPContext *ctx, ECPNode *node, char *filename) {
+ int fd;
+ ssize_t rv;
+
+ if((fd = open(filename, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR)) < 0) return ECP_ERR;
+ rv = write(fd, ctx->cr.dh_pub_get_buf(&node->public), ECP_ECDH_SIZE_KEY);
+ if (rv != ECP_ECDH_SIZE_KEY) {
+ close(fd);
+ return ECP_ERR;
+ }
+ rv = write(fd, &node->addr, sizeof(node->addr));
+ if (rv != sizeof(node->addr)) {
+ close(fd);
+ return ECP_ERR;
+ }
+ close(fd);
+ return ECP_OK;
+}
+
+int ecp_util_node_load(ECPContext *ctx, ECPNode *node, char *filename) {
+ int fd;
+ ssize_t rv;
+ unsigned char buf[ECP_ECDH_SIZE_KEY];
+
+ if((fd = open(filename, O_RDONLY)) < 0) return ECP_ERR;
+ rv = read(fd, buf, ECP_ECDH_SIZE_KEY);
+ if (rv != ECP_ECDH_SIZE_KEY) {
+ close(fd);
+ return ECP_ERR;
+ }
+ rv = read(fd, &node->addr, sizeof(node->addr));
+ if (rv != sizeof(node->addr)) {
+ close(fd);
+ return ECP_ERR;
+ }
+ close(fd);
+
+ ctx->cr.dh_pub_from_buf(&node->public, buf);
+
+ return ECP_OK;
+} \ No newline at end of file
diff --git a/code/util/util.h b/code/util/util.h
new file mode 100644
index 0000000..3d07588
--- /dev/null
+++ b/code/util/util.h
@@ -0,0 +1,5 @@
+int ecp_util_key_save(ECPContext *ctx, ECPDHKey *key, char *filename);
+int ecp_util_key_load(ECPContext *ctx, ECPDHKey *key, char *filename);
+
+int ecp_util_node_save(ECPContext *ctx, ECPNode *node, char *filename);
+int ecp_util_node_load(ECPContext *ctx, ECPNode *node, char *filename); \ No newline at end of file