1 files changed, 25 insertions, 16 deletions

diff --git a/src/encryption.c b/src/encryption.c
index a01d4ff..606be03 100644
--- a/src/encryption.c
+++ b/src/encryption.c
@@ -16,6 +16,8 @@
 #include <stdio.h>
 
 #if ___VXGG___ALWAYS_CHECK_LIBSODIUM___ > 0
+#if ___VXGG___USE_CLS_CALLBACK___ > 0
+
 void naclfaildefault(void *none) {
     none = none; // Makes gcc happy    
     if(___VXGG___VERBOSE_ERRORS___)
@@ -23,31 +25,36 @@ void naclfaildefault(void *none) {
     exit(EXIT_FAILURE);
 }
 
-int checksodiumcb(const vxgg_naclfailcb callback, void *data) {
+int checksodiumcb(const vxgg_naclfailcb callback, void *data, unsigned char set) {
     static vxgg_naclfailcb cb = naclfaildefault;
     static void *usr = NULL;
-
-    if(callback != NULL) {
+    int ret;
+    
+    if(set) {
         cb = callback;
         usr = data;
         return 2;   // libsodium normally returns 1 if the library is already initialized, so this is to signal that the callback has been updated
     }
 
-    int ret = sodium_init();
-    if(ret < 0)
-        cb(usr);
+    if((ret = sodium_init()) < 0) {
+        if(cb == NULL)
+            error(0, EINVAL, "<checksodiumcb> refusing to run a null callback");
+        else
+            cb(usr);
+    }
 
     return ret;
 }
 
 void vxgg_setsodiumfailcb(vxgg_naclfailcb cb, void *data) {
-    checksodiumcb(cb, data);
+    checksodiumcb(cb, data, 1);
 }
+
 #endif
 
 void checksodium(void) {
-    #if ___VXGG___ALWAYS_CHECK_LIBSODIUM___ > 0
-    checksodiumcb(NULL, NULL);
+    #if ___VXGG___USE_CLS_CALLBACK___ > 0
+    checksodiumcb(NULL, NULL, 0);
     #else
 
     if(sodium_init() < 0) {
@@ -60,13 +67,15 @@ void checksodium(void) {
     return;
 }
 
+#endif
+
 int maketmp(const char * const dest) {
     return open(dest, (O_TMPFILE | O_WRONLY | O_CLOEXEC | O_SYNC), (S_IRUSR | S_IWUSR));
 }
 
 int encrypttotmp(const char * const target, const char * const output, const unsigned char key[crypto_secretstream_xchacha20poly1305_KEYBYTES]) {
     #if ___VXGG___ALWAYS_CHECK_LIBSODIUM___ > 0
-        checksodium();
+    checksodium();
     #endif
 
     int fd = -1, tfd = -1;
@@ -163,6 +172,9 @@ int decrypttofile(FILE *dst, FILE *src, unsigned char key[crypto_secretstream_xc
     FILE *dst, *src;
     int eof;
     
+    #if ___VXGG___ALWAYS_CHECK_LIBSODIUM___ > 0
+        checksodium();
+    #endif
 
     if(!src)
         RETURNWERR(EINVAL, -1);
@@ -244,9 +256,9 @@ void* xsodium_malloc(size_t size) {
 }
 
 
-#ifdef TESTING
+/*
 int main(void) {
-    /*// Example code for creating a temp file, writing to it, then linking it back into the fs
+    // Example code for creating a temp file, writing to it, then linking it back into the fs
     const char *dir = ".", *testmsg = "we do a little testing\n";
     char *path = NULL;
 
@@ -313,9 +325,6 @@ int main(void) {
     sodium_munlock(uin, strlen(uin) + 1);
     free(uin);
 
-    //*///
-
     return 0;
 }
-
-#endif
\ No newline at end of file
+*/
\ No newline at end of file