First cut adding random data generation function.

michelp · Apr 16, 2017 · 0225515 · 0225515
1 parent 73c16ba
commit 0225515
Show file tree

Hide file tree

Showing 6 changed files with 73 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -0,0 +1,12 @@
+#http://blog.pgxn.org/post/4783001135/extension-makefiles pg makefiles
+EXTENSION = pgsodium
+PG_CONFIG ?= pg_config
+DATA = $(wildcard *--*.sql)
+PGXS := $(shell $(PG_CONFIG) --pgxs)
+MODULE_big = pgsodium
+OBJS = $(patsubst %.c,%.o,$(wildcard src/*.c))
+#TESTS        = $(wildcard test/sql/*.sql)
+#REGRESS      = $(patsubst test/sql/%.sql,%,$(TESTS))
+#REGRESS_OPTS = --inputdir=test --load-language=plpgsql
+include $(PGXS)
+
diff --git a/README.md b/README.md
@@ -1 +1,3 @@
-# pgsodium
+# pgsodium
+
+Postgres extension to expose libsodium functions to SQL.
diff --git a/pgsodium--0.0.1.sql b/pgsodium--0.0.1.sql
@@ -0,0 +1,3 @@
+-- complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION pgsodium" to load this file. \quit
+
diff --git a/pgsodium.control b/pgsodium.control
@@ -0,0 +1,5 @@
+# pgsodium extension
+comment = 'Postgres extension for libsodium functions'
+default_version = '0.0.1'
+relocatable = true
+requires = ''
diff --git a/src/pgsodium.c b/src/pgsodium.c
@@ -0,0 +1,36 @@
+#include "pgsodium.h"
+#include <sodium.h>
+PG_MODULE_MAGIC;
+
+PG_FUNCTION_INFO_V1(pgsodium_randombytes_random);
+Datum
+pgsodium_randombytes_random(PG_FUNCTION_ARGS)
+{
+	PG_RETURN_UINT32(randombytes_random());
+}
+
+PG_FUNCTION_INFO_V1(pgsodium_randombytes_uniform);
+Datum
+pgsodium_randombytes_uniform(PG_FUNCTION_ARGS)
+{
+	uint32_t upper_bound = PG_GETARG_UINT32(0);
+	PG_RETURN_UINT32(randombytes_uniform(upper_bound));
+}
+
+PG_FUNCTION_INFO_V1(pgsodium_randombytes_buf);
+Datum
+pgsodium_randombytes_buf(PG_FUNCTION_ARGS)
+{
+	size_t size = PG_GETARG_UINT32(0);
+	bytea *ret = (bytea *) palloc(VARHDRSZ + size);
+	SET_VARSIZE(ret, VARHDRSZ + size);
+	randombytes_buf(VARDATA(ret), size);
+	PG_RETURN_BYTEA_P(ret);
+}
+
+void _PG_init(void) {
+	if (sodium_init() == -1) {
+		elog(ERROR, "_PG_init: sodium_init() failed cannot initialize pgsodium");
+		return;
+	}
+}
diff --git a/src/pgsodium.h b/src/pgsodium.h
@@ -0,0 +1,14 @@
+#ifndef PGSODIUM_H
+#define PGSODIUM_H
+
+#include "postgres.h"
+#include "utils/builtins.h"
+#include "libpq/pqformat.h"
+
+void _PG_init(void);
+
+Datum pgsodium_randombytes_random(PG_FUNCTION_ARGS);
+Datum pgsodium_randombytes_uniform(PG_FUNCTION_ARGS);
+Datum pgsodium_randombytes_buf(PG_FUNCTION_ARGS);
+
+#endif // PGSODIUM_H
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		-- complain if script is sourced in psql, rather than via CREATE EXTENSION
		\echo Use "CREATE EXTENSION pgsodium" to load this file. \quit