X-Git-Url: http://git.code-monkey.de/?p=ruby-vorbistagger.git;a=blobdiff_plain;f=ext%2Fvcedit.c;h=832f417b2b30519cba6661b3dfb063c52a43f925;hp=073196b3fdb1b1b9ebfcda7166b871f7d1354de3;hb=b1c729d53273c482052dfc99e197463b123e389e;hpb=c4e7d2b3fa117a377a4f33d28be332d904d3d6ea

diff --git a/ext/vcedit.c b/ext/vcedit.c
index 073196b..832f417 100644
--- a/ext/vcedit.c
+++ b/ext/vcedit.c
@@ -31,26 +31,24 @@
 
 #define CHUNKSIZE 4096
 
-static int vcedit_open (vcedit_state *state);
-
 struct vcedit_state_St {
 	int refcount;
 
-	ogg_sync_state *oy;
-	ogg_stream_state *os;
+	ogg_sync_state oy;
+	ogg_stream_state os;
 
-	vorbis_comment *vc;
-	vorbis_info *vi;
+	vorbis_comment vc;
+	vorbis_info vi;
 
 	char filename[PATH_MAX];
 
 	FILE *in;
+	bool opened;
 	long serial;
 	unsigned char *mainbuf;
 	unsigned char *bookbuf;
 	int	mainlen;
 	int	booklen;
-	const char *lasterror;
 	char *vendor;
 	int prevW;
 	int extrapage;
@@ -60,14 +58,15 @@ struct vcedit_state_St {
 static void
 vcedit_state_free (vcedit_state *state)
 {
-	free (state->oy);
-	free (state->os);
-	free (state->vc);
-	free (state->vi);
 	free (state->mainbuf);
 	free (state->bookbuf);
 	free (state->vendor);
 
+	if (state->in) {
+		fclose (state->in);
+		state->in = NULL;
+	}
+
 	free (state);
 }
 
@@ -76,22 +75,6 @@ vcedit_state_init (vcedit_state *state)
 {
 	state->refcount = 1;
 
-	state->oy = malloc (sizeof (ogg_sync_state));
-	if (!state->oy)
-		return false;
-
-	state->os = malloc (sizeof (ogg_stream_state));
-	if (!state->os)
-		return false;
-
-	state->vc = malloc (sizeof (vorbis_comment));
-	if (!state->vc)
-		return false;
-
-	state->vi = malloc (sizeof (vorbis_info));
-	if (!state->vi)
-		return false;
-
 	return true;
 }
 
@@ -114,36 +97,23 @@ vcedit_state_new (const char *filename)
 	snprintf (state->filename, sizeof (state->filename),
 	          "%s", filename);
 
-	state->in = fopen (state->filename, "rb");
-
-	if (vcedit_open (state) < 0) {
-		vcedit_state_free (state);
-		return NULL;
-	}
-
 	return state;
 }
 
-const char *
-vcedit_error (vcedit_state *state)
-{
-	return state->lasterror;
-}
-
 vorbis_comment *
 vcedit_comments (vcedit_state *state)
 {
-	return state->vc;
+	return state->opened ? &state->vc : NULL;
 }
 
 static void
 vcedit_clear_internals (vcedit_state *state)
 {
-	ogg_stream_clear (state->os);
-	ogg_sync_clear (state->oy);
+	ogg_stream_clear (&state->os);
+	ogg_sync_clear (&state->oy);
 
-	vorbis_info_clear (state->vi);
-	vorbis_comment_clear (state->vc);
+	vorbis_info_clear (&state->vi);
+	vorbis_comment_clear (&state->vc);
 
 	free (state->vendor);
 	state->vendor = NULL;
@@ -157,6 +127,7 @@ vcedit_clear_internals (vcedit_state *state)
 	state->booklen = 0;
 
 	state->serial = 0;
+	state->opened = false;
 }
 
 void
@@ -171,7 +142,9 @@ vcedit_state_unref (vcedit_state *state)
 	if (--state->refcount)
 		return;
 
-	vcedit_clear_internals (state);
+	if (state->opened)
+		vcedit_clear_internals (state);
+
 	vcedit_state_free (state);
 }
 
@@ -236,7 +209,7 @@ _blocksize (vcedit_state *s, ogg_packet *p)
 {
 	int this, ret = 0;
 
-	this = vorbis_packet_blocksize (s->vi, p);
+	this = vorbis_packet_blocksize (&s->vi, p);
 
 	if (s->prevW)
 		ret = (this + s->prevW) / 4;
@@ -252,7 +225,7 @@ _fetch_next_packet (vcedit_state *s, ogg_packet *p, ogg_page *page)
 	char *buffer;
 	int result, bytes;
 
-	result = ogg_stream_packetout (s->os, p);
+	result = ogg_stream_packetout (&s->os, p);
 
 	if (result > 0)
 		return 1;
@@ -260,10 +233,10 @@ _fetch_next_packet (vcedit_state *s, ogg_packet *p, ogg_page *page)
 	if (s->eosin)
 		return 0;
 
-	while (ogg_sync_pageout (s->oy, page) <= 0) {
-		buffer = ogg_sync_buffer (s->oy, CHUNKSIZE);
+	while (ogg_sync_pageout (&s->oy, page) <= 0) {
+		buffer = ogg_sync_buffer (&s->oy, CHUNKSIZE);
 		bytes = fread (buffer, 1, CHUNKSIZE, s->in);
-		ogg_sync_wrote (s->oy, bytes);
+		ogg_sync_wrote (&s->oy, bytes);
 
 		if (!bytes)
 			return 0;
@@ -277,14 +250,15 @@ _fetch_next_packet (vcedit_state *s, ogg_packet *p, ogg_page *page)
 		return 0;
 	}
 
-	ogg_stream_pagein (s->os, page);
+	ogg_stream_pagein (&s->os, page);
 
 	return _fetch_next_packet (s, p, page);
 }
 
-static int
+vcedit_error
 vcedit_open (vcedit_state *state)
 {
+	vcedit_error ret;
 	char *buffer;
 	int bytes, i;
 	int chunks = 0;
@@ -292,46 +266,47 @@ vcedit_open (vcedit_state *state)
 	ogg_packet header_main, header_comments, header_codebooks;
 	ogg_page og;
 
-	ogg_sync_init (state->oy);
+	state->in = fopen (state->filename, "rb");
+	if (!state->in)
+		return VCEDIT_ERR_OPEN;
+
+	ogg_sync_init (&state->oy);
 
 	while (1) {
-		buffer = ogg_sync_buffer (state->oy, CHUNKSIZE);
+		buffer = ogg_sync_buffer (&state->oy, CHUNKSIZE);
 		bytes = fread (buffer, 1, CHUNKSIZE, state->in);
 
-		ogg_sync_wrote (state->oy, bytes);
+		ogg_sync_wrote (&state->oy, bytes);
 
-		if (ogg_sync_pageout (state->oy, &og) == 1)
+		if (ogg_sync_pageout (&state->oy, &og) == 1)
 			break;
 
 		/* Bail if we don't find data in the first 40 kB */
 		if (chunks++ >= 10) {
-			if (bytes < CHUNKSIZE)
-				state->lasterror = "Input truncated or empty.";
-			else
-				state->lasterror = "Input is not an Ogg bitstream.";
+			ogg_sync_clear (&state->oy);
 
-			goto err;
+			return VCEDIT_ERR_INVAL;
 		}
 	}
 
 	state->serial = ogg_page_serialno (&og);
 
-	ogg_stream_init (state->os, state->serial);
-	vorbis_info_init (state->vi);
-	vorbis_comment_init (state->vc);
+	ogg_stream_init (&state->os, state->serial);
+	vorbis_info_init (&state->vi);
+	vorbis_comment_init (&state->vc);
 
-	if (ogg_stream_pagein (state->os, &og) < 0) {
-		state->lasterror = "Error reading first page of Ogg bitstream.";
+	if (ogg_stream_pagein (&state->os, &og) < 0) {
+		ret = VCEDIT_ERR_INVAL;
 		goto err;
 	}
 
-	if (ogg_stream_packetout (state->os, &header_main) != 1) {
-		state->lasterror = "Error reading initial header packet.";
+	if (ogg_stream_packetout (&state->os, &header_main) != 1) {
+		ret = VCEDIT_ERR_INVAL;
 		goto err;
 	}
 
-	if (vorbis_synthesis_headerin (state->vi, state->vc, &header_main) < 0) {
-		state->lasterror = "Ogg bitstream does not contain vorbis data.";
+	if (vorbis_synthesis_headerin (&state->vi, &state->vc, &header_main) < 0) {
+		ret = VCEDIT_ERR_INVAL;
 		goto err;
 	}
 
@@ -344,26 +319,26 @@ vcedit_open (vcedit_state *state)
 
 	while (i < 2) {
 		while (i < 2) {
-			int result = ogg_sync_pageout (state->oy, &og);
+			int result = ogg_sync_pageout (&state->oy, &og);
 
 			if (!result)
 				break; /* Too little data so far */
 
 			if (result == 1) {
-				ogg_stream_pagein (state->os, &og);
+				ogg_stream_pagein (&state->os, &og);
 
 				while (i < 2) {
-					result = ogg_stream_packetout (state->os, header);
+					result = ogg_stream_packetout (&state->os, header);
 
 					if (!result)
 						break;
 
 					if (result == -1) {
-						state->lasterror = "Corrupt secondary header.";
+						ret = VCEDIT_ERR_INVAL;
 						goto err;
 					}
 
-					vorbis_synthesis_headerin (state->vi, state->vc, header);
+					vorbis_synthesis_headerin (&state->vi, &state->vc, header);
 
 					if (i == 1) {
 						state->booklen = header->bytes;
@@ -377,30 +352,32 @@ vcedit_open (vcedit_state *state)
 			}
 		}
 
-		buffer = ogg_sync_buffer (state->oy, CHUNKSIZE);
+		buffer = ogg_sync_buffer (&state->oy, CHUNKSIZE);
 		bytes = fread (buffer, 1, CHUNKSIZE, state->in);
 
 		if (bytes == 0 && i < 2) {
-			state->lasterror = "EOF before end of vorbis headers.";
+			ret = VCEDIT_ERR_INVAL;
 			goto err;
 		}
 
-		ogg_sync_wrote (state->oy, bytes);
+		ogg_sync_wrote (&state->oy, bytes);
 	}
 
 	/* Copy the vendor tag */
-	state->vendor = strdup (state->vc->vendor);
+	state->vendor = strdup (state->vc.vendor);
 
 	/* Headers are done! */
-	return 0;
+	state->opened = true;
+
+	return VCEDIT_ERR_SUCCESS;
 
 err:
 	vcedit_clear_internals (state);
 
-	return -1;
+	return ret;
 }
 
-int
+vcedit_error
 vcedit_write (vcedit_state *state)
 {
 	ogg_stream_state streamout;
@@ -412,23 +389,22 @@ vcedit_write (vcedit_state *state)
 	int s, result, bytes, needflush = 0, needout = 0;
 	size_t tmp;
 
+	if (!state->opened)
+		return VCEDIT_ERR_INVAL;
+
 	strcpy (tmpfile, state->filename);
 	strcat (tmpfile, ".XXXXXX");
 
 	s = mkstemp (tmpfile);
-	if (s == -1) {
-		state->lasterror = "Error writing stream to output. "
-		                   "Cannot open temporary file.";
-		return -1;
-	}
+	if (s == -1)
+		return VCEDIT_ERR_TMPFILE;
 
 	out = fdopen (s, "wb");
 	if (!out) {
 		unlink (tmpfile);
 		close (s);
-		state->lasterror = "Error writing stream to output. "
-		                   "Cannot open temporary file.";
-		return -1;
+
+		return VCEDIT_ERR_TMPFILE;
 	}
 
 	state->prevW = state->extrapage = state->eosin = 0;
@@ -447,7 +423,7 @@ vcedit_write (vcedit_state *state)
 
 	ogg_stream_init (&streamout, state->serial);
 
-	_commentheader_out (state->vc, state->vendor, &header_comments);
+	_commentheader_out (&state->vc, state->vendor, &header_comments);
 
 	ogg_stream_packetin (&streamout, &header_main);
 	ogg_stream_packetin (&streamout, &header_comments);
@@ -541,14 +517,12 @@ vcedit_write (vcedit_state *state)
 		 * through, a page at a time.
 		 */
 		while (1) {
-			result = ogg_sync_pageout (state->oy, &ogout);
+			result = ogg_sync_pageout (&state->oy, &ogout);
 
 			if (!result)
                 break;
 
-			if (result < 0)
-				state->lasterror = "Corrupt or missing data, continuing...";
-			else {
+			if (result >= 0) {
 				/* Don't bother going through the rest, we can just
 				 * write the page out now
 				 */
@@ -562,9 +536,9 @@ vcedit_write (vcedit_state *state)
 			}
 		}
 
-		buffer = ogg_sync_buffer (state->oy, CHUNKSIZE);
+		buffer = ogg_sync_buffer (&state->oy, CHUNKSIZE);
 		bytes = fread (buffer, 1, CHUNKSIZE, state->in);
-		ogg_sync_wrote (state->oy, bytes);
+		ogg_sync_wrote (&state->oy, bytes);
 
 		if (!bytes) {
 			state->eosin = 1;
@@ -591,16 +565,11 @@ cleanup:
 
     state->mainbuf = state->bookbuf = NULL;
 
-	if (!state->eosin) {
-		state->lasterror = "Error writing stream to output. "
-		                   "Output stream may be corrupted or truncated.";
-		return -1;
-	}
+	if (!state->eosin)
+		return VCEDIT_ERR_INVAL;
 
 	vcedit_clear_internals (state);
 
-	state->in = fopen (state->filename, "rb");
-	vcedit_open (state);
-
-	return 0;
+	return (vcedit_open (state) == VCEDIT_ERR_SUCCESS) ?
+	       VCEDIT_ERR_SUCCESS : VCEDIT_ERR_REOPEN;
 }