X-Git-Url: https://git.llucax.com/z.facultad/75.06/jacu.git/blobdiff_plain/20b0b8b6db70b1a0c9b4c211f43e91562af6c54c..581c6c4f1d70b1928ae2946aa0482a4d36ea56d9:/src/jacu.c

diff --git a/src/jacu.c b/src/jacu.c
index 75ca87c..042a332 100644
--- a/src/jacu.c
+++ b/src/jacu.c
@@ -12,6 +12,16 @@
 
 long fsize(const char* filename);
 
+/* Flags del archivo comprimido */
+#define FLAGS_ZG 0x1
+#define FLAGS_WS 0x2
+#define FLAGS_RESERVED_1 0x4
+#define FLAGS_RESERVED_2 0x8
+#define FLAGS_RESERVED_3 0x16
+#define FLAGS_RESERVED_4 0x64
+#define FLAGS_RESERVED_5 0x128
+#define FLAGS_RESERVED_6 0x255
+
 typedef struct _flags_ {
 	int cflag;
 	int dflag;
@@ -20,11 +30,27 @@ typedef struct _flags_ {
 	int qflag;
 	int sflag;
 	int mflag;
+	int rflag; /* Richard Dictionary :-) */
 } t_Flags;
 
 int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags *flags, char *staticmodel);
 int descomprimir(char *src, char *dst);
 
+char is_flags_on(unsigned char flags, unsigned char flag)
+{
+	return (flags & flag);
+}
+
+char flag_on(unsigned char flags, unsigned char flag)
+{
+	return (flags | flag);
+}
+
+char flag_off(unsigned char flags, unsigned char flag)
+{
+	return (flags & (~flag));
+}
+
 int main(int argc, char* argv[])
 {	
 	long int volumesize = 0;
@@ -35,7 +61,7 @@ int main(int argc, char* argv[])
 			
 	memset(&flags, 0, sizeof(t_Flags));
 
-	while ((ch = getopt(argc, argv, "scdzm:t:q:")) != -1) { 
+	while ((ch = getopt(argc, argv, "rscdzm:t:q:")) != -1) { 
 		 
 		switch (ch) { 
 			case 'c': flags.cflag = 1; 
@@ -58,6 +84,8 @@ int main(int argc, char* argv[])
 				volumesize = atol(optarg);
 				break; 
 
+			case 'r': flags.rflag = 1;
+				break;
 			case 'q': flags.qflag = 1; 
 				switch (atoi(optarg))
 				{
@@ -85,13 +113,13 @@ int main(int argc, char* argv[])
 				}
 				break; 
 
-			default: fprintf(stderr, "Usage: %s [-cdzs][-q blksize][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
+			default: fprintf(stderr, "Usage: %s [-cdzsr][-q blksize][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
 					 return(2);
 		}
 	}
 		
 	if ( (argc == 1) || (flags.cflag & flags.dflag) || !(flags.cflag | flags.dflag) || ((argc - optind) < 2) || (flags.mflag & flags.sflag)) {
-		fprintf(stderr, "Usage: %s [-cdzs][-q compressionquality][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
+		fprintf(stderr, "Usage: %s [-cdzsr][-q compressionquality][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
 		return (3);
 	}
 	if ((flags.tflag) && (volumesize <= 0l)) {
@@ -135,6 +163,7 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
 	unsigned char *mtf;
 	unsigned char *salida, *data;
 	unsigned char *z;
+	unsigned char file_flags = 0;
 	int z_len;
 	
 	/* Abrimos el archivo a comprimir y encodeamos bloques */
@@ -154,14 +183,16 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
 
 	/* Guardamos cabecera para indicar si usamos ZG (huffencoded) */
 	if (flags->zflag)
-		shuff_scanfreq_chunk(shuff, "\001", 1);
-	else
-		shuff_scanfreq_chunk(shuff, "\000", 1);
+		file_flags = flag_on(file_flags, FLAGS_ZG);
+	if (flags->rflag)
+		file_flags = flag_on(file_flags, FLAGS_WS);
+
+	shuff_scanfreq_chunk(shuff, &file_flags, 1);
 
 	total = 0;
 	while (!feof(fp)) {
 		i = 0;
-		i = bs_readblock(fp, data, pagesize);
+		i = bs_readblock(fp, data, pagesize, flags->rflag);
 		total += i;
 
 
@@ -223,8 +254,8 @@ int descomprimir(char *src, char *dst)
 	Uint32 block_size = 0, k;
 	unsigned char *block, *mtf, *orig;
 	unsigned char *z;
-	int z_len=0,moredata = 0,decoded = 0;
-	unsigned char use_zg = 0,retbytes = 0;
+	Uint32 z_len=0,moredata = 0,decoded = 0;
+	unsigned char file_flags = 0,retbytes = 0;
 	HUFF_STATE *shuff;
 
 	/* Inicializo el descompresor */
@@ -237,7 +268,7 @@ int descomprimir(char *src, char *dst)
 	if (!(moredata = shuff_decode_chunk(shuff,(char*)&block_size,sizeof(Uint32),&decoded))) return 1;
 
 	/* Descomprimo byte que indica si se usa ZG */
-	if (!(moredata = shuff_decode_chunk(shuff, &use_zg, 1, &decoded))) return 1;
+	if (!(moredata = shuff_decode_chunk(shuff, &file_flags, 1, &decoded))) return 1;
 
 	/* Creo buffers */
 	block = malloc(block_size*sizeof(unsigned char)+sizeof(Uint32));
@@ -252,7 +283,7 @@ int descomprimir(char *src, char *dst)
 			moredata = shuff_decode_chunk(shuff,z,z_len,&decoded);				
 			
 			/* Veo si se uso Zero Grouping para comprimir */
-			if (use_zg) {
+			if (is_flags_on(file_flags, FLAGS_ZG)) {
 				ZG zg;
 				unsigned char zgbuffer[255];
 				unsigned char zgbyte = 0;
@@ -295,7 +326,12 @@ int descomprimir(char *src, char *dst)
 			/* Obtengo el chunk original aplicando BS Inverso */
 			bs_restore(orig, mtf+sizeof(Uint32), k, decoded - sizeof(Uint32));
 
-			fwrite(orig, decoded - sizeof(Uint32), sizeof(unsigned char), fp_out);
+			decoded -= sizeof(Uint32);
+			if (is_flags_on(file_flags, FLAGS_WS)) {
+				orig = bs_finalblock(orig, decoded, &decoded);
+			}
+
+			fwrite(orig, decoded, sizeof(unsigned char), fp_out);
 			free(mtf);
 			free(z);
 		}