Write some sample code to encrypt a file via password

1 files changed, 21 insertions, 1 deletions

diff --git a/src/shared.c b/src/shared.c
index f3e98ee..40240d5 100644
--- a/src/shared.c
+++ b/src/shared.c
@@ -81,10 +81,30 @@ int readwholebuffer(char **str, unsigned long int initsize, int fd) {
                 lstr = tmp;
         }
 
-    if(lstr) {
+    if(lstr && fd == STDIN_FILENO) {
         lstr[csize - ccap - 1] = '\0'; // Don't include the newline
     }
 
         *str = lstr;
         return ((bytesread == 0) ? (csize - ccap) : -1);
 }
+
+
+int writewholebuffer(int fd, const unsigned char *buf, int len) {
+    int total = 0;
+    int left = len;
+    int n;
+
+    while(total < len) {
+        if((n = write(fd, buf + total, left)) < 0) 
+            break;
+
+        total += n;
+        left -= n;
+    }
+
+    return (n < 0) ? -1 : 0;
+}
+// Adapted from Beej's `sendall()` function
+// https://beej.us/guide/bgnet/html/split/slightly-advanced-techniques.html#sendall
+    // Thanks Beej!
\ No newline at end of file