/* Test 14 -> Checks whether the read call in library
/* detects multiple block corruption and returns an error
**/

#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <string.h>
#include <dlfcn.h>

int fill_random(char *buf)
{
    int i;
    for(i=0; i<4096; i++) {
         buf[i] = 'a' + random() % 26;
    }
}

int main()
{
   int fd1 = -1;
   int err;
   void * handle = NULL;
   char buf[4096];
   char new_buf[4096];
   off_t cur=0;
   int i=0;
   int data_read = 0;

    handle = dlopen("/lib/libc.so.6",  RTLD_LAZY);
    if (handle == NULL) {
	fprintf(stderr, "%s", dlerror());
	return -1;
    }
    int (*o_open)(const char *, int) = (int (*)(const char *, int)) dlsym(handle, "open");
    ssize_t (*o_read)(int, void *, size_t) = (ssize_t (*)(int, void *, size_t)) dlsym(handle, "read");
    ssize_t (*o_write)(int, void *, size_t) = (ssize_t (*)(int, void *, size_t)) dlsym(handle, "write");
    off_t (*o_lseek)(int, off_t, int) = (off_t (*)(int, off_t, int)) dlsym(handle, "lseek");
    int (*o_close)(int) = (int (*)(int)) dlsym(handle, "close");


/* Bootstrap the file with libfsprotect.c */
   fd1 = open("file5", O_RDONLY);
   if(fd1 == -1) {
         return -1;
   }
   err = close(fd1);
   if(err == -1)
        return -1;

	printf("STarting ------------------------------------------------------------------------\n");


/* Corrupt 1st and 3rd blocks outside the library */
   fd1 = (*o_open)("file5", O_RDWR);  
   if(fd1 == -1) {
         return -1;
   }
   err = (*o_read)(fd1, buf, 4096);
   if(err == -1)
        return -1;
   memcpy(new_buf, buf, 4096);
   for(i=130; i<160; i++) {
        new_buf[i]++;
   }
   (*o_lseek)(fd1, 0, SEEK_SET);
   err = (*o_write)(fd1, new_buf, 4096);
   if(err == -1)
        return -1;

   err = (*o_read)(fd1, buf, 4096);
   if(err == -1)
        return -1;
   err = (*o_read)(fd1, buf, 4096);
   if(err == -1)
        return -1;
   memcpy(new_buf, buf, 4096);
   for(i=130; i<160; i++) {
        new_buf[i]++;
   }
   (*o_lseek)(fd1, 8192, SEEK_SET);
   err = (*o_write)(fd1, new_buf, 4096);
   if(err == -1)
        return -1;
   (*o_close)(fd1);

	printf("DONE CORRUPTING----------------------------------------------------------------------\n");

/* Re-open the file with the library. Do a write to indicate the error */
   fd1 = open("file5",O_RDWR);
   if(fd1 == -1) {
         return -1;
   }

   //printf("DOne Opening\n") 	 
   fill_random(new_buf);
   //printf("about to read\n");
   data_read = write(fd1, new_buf, 4096);
   if(data_read == -1)
        goto pass;  
   else 
        goto fail;

pass:
   err = close(fd1);
   if(err == -1)
        return -1;

   return 0;

fail:
   close(fd1);
   return -1;

}
