ss_testseekw.c

Go to the documentation of this file.

/* Test for securestorage_read && write 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 write_byte=0;
        int nread=0;
        FILE *fdin;
        off_t remote_file_position;
        off_t offset;
        int whence;
        
        if (argc != 5) {
                fprintf(stderr,"Usage: %s <sourcefile> <lfn> <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);
        }
        
        // Open source file and verify if it exist
        if ( (fdin = fopen(argv[1], "rb")) == NULL ) {
                fprintf(stderr, "Fatal error: file \"%s\" does not exist or you don't have read permissions.\n",argv[1]);
                free(buffer);
                exit(1);
        }
        
        // Open Destination file on remote SE
        if ( (fd=securestorage_open( argv[2], O_WRONLY|O_CREAT, 0644)) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_open: %s\n", securestorage_error(errno_save));
                fclose(fdin); free(buffer);
                exit(1);
        }

        do {
                // Read source file
                memset(buffer, 0, BUFLEN+1);
                nread =  fread(buffer, 1, BUFLEN, fdin);
                if (nread < 0) {
                        perror ("fread");
                        fclose(fdin); securestorage_close (fd);
                        free(buffer);
                        exit(1);
                }
                else if (nread > 0) {
                        if ( (write_byte=securestorage_write ( fd, buffer, nread )) < 0 ) {
                                int errno_save = errno;
                                fprintf(stderr, "securestorage_write: %s\n", securestorage_error(errno_save));
                                fclose(fdin); securestorage_close (fd);
                                free(buffer);
                                exit(1);
                        }
                }
        } while (nread>0);
        
        if ((remote_file_position = securestorage_lseek (fd, 0, SEEK_SET)) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_lseek: %s\n", securestorage_error(errno_save));
                fclose(fdin); securestorage_close (fd);
                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(fdin); securestorage_close (fd);
                free(buffer);
                exit(1);
        }
        if ( (write_byte=securestorage_write ( fd, "prova", 5 )) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_write: %s\n", securestorage_error(errno_save));
                fclose(fdin); securestorage_close (fd);
                free(buffer);
                exit(1);
        }
        
        if ( securestorage_close (fd) < 0 ) {
                int errno_save = errno;
                fprintf(stderr, "securestorage_close: %s\n", securestorage_error(errno_save));
                fclose(fdin); free(buffer);
                exit(1);
        }
        fclose(fdin);
        free(buffer);
        
        exit (0);
}

---