// sparksandflames.com


#define PI              3.14159265358979323846
#define MAXCMD       1024
#define MAXVARNAME     32

#ifdef __GNUC__
    #define  FD_SETSIZE (1024)
    #include <unistd.h>
    #include <dirent.h>
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <netdb.h>
    #include <errno.h>
    #include <sys/time.h>
    #include <pwd.h>
    #include <err.h>
    #include <sysexits.h>
    #include <sys/resource.h>    
    #include <arpa/nameser.h>
    #include <resolv.h>
    #define  DELIMITER "/"
    #define  min(x,y)   (x<y?x:y)
    #define  max(x,y)   (x>y?x:y)
    #define  O_BINARY   0
    #include <sys/wait.h>
    #include <sys/syslimits.h>
    #include <utime.h>    
    #include <sys/ipc.h>
    #include <sys/sem.h>
    #include <readline/readline.h>
    #include <readline/history.h>
    #define  closesocket close
    #define  winsockerr strerror
    #define  WSAGetLastError() errno
    #define  stricmp strcasecmp
#else   
    #define  FD_SETSIZE 1024
    #include <io.h>
    #include <direct.h>
    #include <winsock2.h>
    #include <stddef.h>
    #include <malloc.h>
    #include <process.h>
    #include <dos.h>
    #define  DELIMITER "\\"
    #include <sys\utime.h>
#endif

   
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <string.h>
#include <time.h>
#include <ctype.h>
#include <signal.h>
#include <stdarg.h>
#include <math.h>
#include <float.h>


#define MEGABYTE (1024*1024)



int main(int argc, char *argv[]);


char buf[MEGABYTE];


// WRITE A BIG FILE AND TIME IT

// FILE MUST BE SIGNIFICANTLY LARGER THAN DISKCONTROLLER CACHE SIZE
//   AND MOTHERBOARD RAM SIZE TO BE A MEANINGFUL TEST

int main(int argc, char *argv[])
{
    int i,handle;

    time_t wallstart,wallend;

    double sizework,timework;
        
    
    char *filename = "diskwrite.tmp";


    // DEFAULT WRITE SIZE = 4 GIGABYTES

    int megabytes = 1024 * 4;


    // PARAMETER 1 OVER RIDES DEFAULT

    if (argc == 2)
    {
        megabytes = atoi(argv[1]);
        if (megabytes < 1)
        {
           printf("diskwrite('%d'): Megabytes to write must be greater than 1.\n",megabytes);
           return 0;
        }
    }

    printf("Writing %d megabytes\n",megabytes);


    // START TIMER
    
    wallstart = time(NULL);
    


    // OPEN FILE
    
    handle = open(filename,O_BINARY|O_CREAT|O_WRONLY|O_TRUNC,S_IRUSR|S_IWUSR|S_IROTH|S_IWOTH|S_IRGRP);
    if (handle == -1)
    {
       printf("open('%s'): #%d. %s\n",filename,errno,strerror(errno));
       return 0;
    }    


    // WRITE IT

    memset(buf,0,MEGABYTE);
    
    for (i=0;i<megabytes;i++)
    {
       if (write(handle,buf,MEGABYTE) != MEGABYTE)
       {
           printf("write(): #%d. %s\n",errno,strerror(errno));
           break;
       }
    }

    // CLOSE

    close(handle);


    // STOP TIMER. DISPLAY STATS.
    
    wallend   = time(NULL);

    timework = wallend - wallstart;
    sizework = megabytes;
   
    printf("Elapsed time: %.0f seconds\n", timework);

   
    if (timework > 0.00)
       printf("Write speed.: %.2f Megabytes / Second\n",sizework/timework);

    
    // REMOVE WORK FILE

    if (remove(filename) != 0)
    {
       printf("remove('%s'): #%d. %s\n",filename,errno,strerror(errno));
       return 0;
    }
    
    return 0;
}