ss_testread.c

Go to the documentation of this file.

/* Test for securestorage_read function */
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#if defined(_WIN32)
#include <io.h>
#else
#include <unistd.h>
#endif
#include "../src/securestorage.h"

#define BUFLEN 65536

extern int errno;

/* MAIN */
int main (int argc, char *argv[]) {
        
        int fd;
        unsigned char *buffer = (unsigned char *) calloc(1, BUFLEN+1);
        int read_byte=0;
        
        if (argc != 3) {
                fprintf(stderr,"Usage: %s <lfn> <destfile>\n",argv[0]);
                free(buffer);
                exit(1);
        }
        
        FILE *fdout;
        if ((fdout = fopen(argv[2], "rb"))) {
                fprintf(stderr, "Fatal error: file \"%s\" exist or you don't have write permissions.\n", argv[2]);
                free(buffer);
                exit(1);
        }
        if ( (fdout = fopen(argv[2], "wb")) == NULL ) {
                perror("fopen");
                free(buffer);
                exit(1);
        }
        
        if ( (fd=securestorage_open(argv[1], O_RDONLY, 0644)) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_open: %s\n", securestorage_error(errno_save));
                fclose(fdout);
                // Delete dest file
                if (remove(argv[2]) < 0 ) {
                        perror ("remove");
        }
        free(buffer);
                exit(1);
        }
        
        do {
                if ( (read_byte=securestorage_read ( fd, buffer, BUFLEN )) < 0 ) {
                        int errno_save = errno;
                        fprintf(stderr, "securestorage_read: %s\n", securestorage_error(errno_save));
                        fclose(fdout);
                        // Delete dest file
                        if (remove(argv[2]) < 0 ) {
                                perror ("remove");
                }
                free(buffer);
                        exit(1);
                }
                if (read_byte>0) {
                        if (fwrite(buffer, 1, read_byte, fdout) < 0) {
                                perror ("fwrite");
                    fclose(fdout);
                    // Delete dest file
                    if (remove(argv[2]) < 0 ) {
                        perror ("remove");
                    }
                    free(buffer);
                                exit(1);
                        }
                }
        } while (read_byte>0);
        
        
        if ( securestorage_close (fd) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_close: %s\n", securestorage_error(errno_save));
                fclose(fdout);
                free(buffer);
                exit(1);
        }
        fclose(fdout);
        free(buffer);
        exit(0);
}

---