X-Git-Url: https://git.llucax.com/z.facultad/75.06/jacu.git/blobdiff_plain/a63c710f7fbc74f701af980d7d6ac9444298f5c7..66f873a3de4adb65bbad6fb09cd57272216d332f:/src/vfile/vfile.c

diff --git a/src/vfile/vfile.c b/src/vfile/vfile.c
index bba064f..7136eed 100644
--- a/src/vfile/vfile.c
+++ b/src/vfile/vfile.c
@@ -132,43 +132,58 @@ int vfputc(int c, VFILE* vfp)
 
 size_t vfread(void* ptr, size_t size, size_t nmemb, VFILE* vfp)
 {
-	int i = 0;
+	int c;
+	size_t i = 0;
 	size_t total = size * nmemb;
 	/* leo uno a uno y si hay error salgo. */
-	while (i < total && (((char*)ptr)[i++] = vfgetc(vfp)) != EOF);
-	return i % size;
+	while (i < total)
+	{
+		if ((c = vfgetc(vfp)) == EOF)
+		{
+			PERR("vfread: EOF");
+			break;
+		}
+		else ((char*)ptr)[i++] = c;
+	}
+	return i / size;
 }
 
 size_t vfwrite(const void *ptr, size_t size, size_t nmemb, VFILE* vfp)
 {
-	int i = 0;
+	size_t i = 0;
 	size_t total = size * nmemb;
 	/* escribo uno a uno y si hay error salgo. */
 	while (i < total && (vfputc(((char*)ptr)[i++], vfp)) != EOF);
-	return i % size;
+	return i / size;
 }
 
 int vfvol_close(VFILE* vfp)
 {
-	/* Si es de escritura y el último guardo la cabecera. */
-	if (vfp->mode == VFWRITE && vfp->lastvol)
+	int ret = 0;
+	if (vfp->fp)
 	{
-		int ret;
-		PERR("vfvol_close: modo == VFWRITE");
-		/* Me posiciono al principio del archivo. */
-		if ((ret = fseek(vfp->fp, 0l, SEEK_SET)))
-		{
-			PERR("vfvol_close: fseek error");
-			return ret; /* fseek error. */
-		}
-		/* Guardo cabecera para indicar si es el último volumen o no. */
-		if ((ret = fputc(vfp->lastvol, vfp->fp)) == EOF)
+		/* Si es de escritura y el último guardo la cabecera. */
+		if (vfp->mode == VFWRITE && vfp->lastvol)
 		{
-			PERR("vfvol_close: fputc error");
-			return ret; /* fputc error. */
+			int ret;
+			PERR("vfvol_close: modo == VFWRITE");
+			/* Me posiciono al principio del archivo. */
+			if ((ret = fseek(vfp->fp, 0l, SEEK_SET)))
+			{
+				PERR("vfvol_close: fseek error");
+				return ret; /* fseek error. */
+			}
+			/* Guardo cabecera para indicar si es el último volumen o no. */
+			if ((ret = fputc(vfp->lastvol, vfp->fp)) == EOF)
+			{
+				PERR("vfvol_close: fputc error");
+				return ret; /* fputc error. */
+			}
 		}
+		ret = fclose(vfp->fp);
+		vfp->fp = 0;
 	}
-	return fclose(vfp->fp);
+	return ret;
 }
 
 int vfvol_open_next(VFILE* vfp)
@@ -206,3 +221,54 @@ int vfvol_open_next(VFILE* vfp)
 	return 0;
 }
 
+long vfsize(const char* path)
+{
+	VFILE* vfp = vfopen(path, "r", 0);
+	long size;
+	if (!vfp) return -1; /* error */
+	if (fseek(vfp->fp, 0l, SEEK_END) == -1)
+	{
+		vfclose(vfp);
+		return -1; /* error */
+	}
+	else
+	{
+		size = ftell(vfp->fp);
+		if (size == -1)
+		{
+			vfclose(vfp);
+			return -1; /* error */
+		}
+	}
+	while (!vfp->lastvol) /* mientras no sea el último volumen */
+	{
+		if (vfvol_open_next(vfp))
+		{
+			vfclose(vfp);
+			return -1; /* error */
+		}
+		if (fseek(vfp->fp, 0l, SEEK_END) == -1)
+		{
+			vfclose(vfp);
+			return -1; /* error */
+		}
+		else
+		{
+			long curr_size = ftell(vfp->fp);
+			if (curr_size == -1)
+			{
+				vfclose(vfp);
+				return -1; /* error */
+			}
+			size += curr_size;
+		}
+		if (vfvol_close(vfp))
+		{
+			vfclose(vfp);
+			return -1; /* error */
+		}
+	}
+	vfclose(vfp);
+	return size;
+}
+