1 files changed, 16 insertions, 3 deletions

diff --git a/src/tests/md5test.cpp b/src/tests/md5test.cpp
index 6f832df..2cc8868 100644
--- a/src/tests/md5test.cpp
+++ b/src/tests/md5test.cpp
@@ -2,18 +2,31 @@
 #include <string.h>
 #include "md5.h"
 
-int main()
+       #include <sys/types.h>
+       #include <sys/stat.h>
+       #include <unistd.h>
+
+
+int main( int argc, char *argv[] )
 {
         md5 mproc;
         md5sum sum;
         char hexstr[33];
 
+        struct stat s;
+        stat( argv[1], &s );
+        char *buf = new char[s.st_size];
+        FILE *fh = fopen( argv[1], "rb" );
+        fread( buf, 1, s.st_size, fh );
+        fclose( fh );
+
         memset( hexstr, 0, 33 );
 
-        mproc.sumString( &sum, "qwertyuiopasdfgh" );
+        mproc.sumData( &sum, buf, s.st_size );
         mproc.sumToHex( &sum, hexstr );
         printf("sum: %s\n", hexstr );
-        printf("chk: 1ebfc043d8880b758b13ddc8aa1638ef\n");
+
+        delete[] buf;
 
         return 0;
 }