ss_testseek.c

Go to the documentation of this file.

/* Test for securestorage_seek 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

/* MAIN */
int main (int argc, char *argv[]) {
        
        int fd;
        unsigned char *buffer = (unsigned char *) calloc(1, BUFLEN+1);
        int read_byte=0;
        off_t remote_file_position;
        off_t offset;
        int whence;
        
        if (argc != 5) {
                fprintf(stderr,"Usage: %s <lfn> <destfile> <offset> <whence>\n",argv[0]);
                free(buffer);
                exit(1);
        }
        
        offset = atoi(argv[3]);
        if (!strncmp(argv[4], "SEEK_SET", 8)) {
                        whence = SEEK_SET;
        } else if (!strncmp(argv[4], "SEEK_CUR", 8)) {
                        whence = SEEK_CUR;
        } else if (!strncmp(argv[4], "SEEK_END", 8)) {
                        whence = SEEK_END;
        } else {
                fprintf(stderr,"Invalid whence specified: only SEEK_SET, SEEK_CUR and SEEK_END are allowed.\n");
                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);
        }
        
        if ((remote_file_position = securestorage_lseek (fd, offset, whence)) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_lseek: %s\n", securestorage_error(errno_save));
                fclose(fdout);
                // Delete dest file
                if (remove(argv[2]) < 0 ) {
                        perror ("remove");
        }
        free(buffer);
        securestorage_close (fd);
                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);
                securestorage_close (fd);
                        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);
                    securestorage_close (fd);
                                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);
}

---