X-Git-Url: https://git.draconx.ca/gitweb/liblbx.git/blobdiff_plain/b0910e37f2dbf5b4e3dcd8bdfcef16d5d6b597cb..4a5ac8356dbc71d84ad5e6b04499597eedf0d89d:/src/lbximg.c

diff --git a/src/lbximg.c b/src/lbximg.c
index d56824c..4b664b7 100644
--- a/src/lbximg.c
+++ b/src/lbximg.c
@@ -1,22 +1,22 @@
-/* 2ooM: The Master of Orion II Reverse Engineering Project
- * Simple command-line tool to convert an LBX image to a set of PNGs.
- * Copyright (C) 2006-2008 Nick Bowler
+/*
+ *  2ooM: The Master of Orion II Reverse Engineering Project
+ *  Simple command-line tool to convert an LBX image to a set of PNGs.
+ *  Copyright (C) 2006-2008 Nick Bowler
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
  *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
  *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
-#define _GNU_SOURCE
+#include <config.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -364,11 +364,11 @@ err:
 
 int main(int argc, char **argv)
 {
-	int mode = MODE_NONE;
-	FILE *inf = stdin, *palf = NULL, *overf = NULL;
+	int mode = MODE_NONE, opt, rc = EXIT_FAILURE;
+	struct lbx_pipe_state state = { .f = stdin };
+	FILE *palf = NULL, *overf = NULL;
 	const char *name = "stdin";
 	LBX_IMG *img;
-	int opt;
 
 	static const char *sopts = "idvf:p:O:V";
 	static const struct option lopts[] = {
@@ -401,14 +401,10 @@ int main(int argc, char **argv)
 			verbose = 1;
 			break;
 		case 'f':
-			if (strcmp(optarg, "-") == 0)
-				break;
-
 			name = strrchr(optarg, '/');
 			name = name ? name+1 : optarg;
 
-			inf = fopen(optarg, "rb");
-			if (!inf) {
+			if (!freopen(optarg, "rb", state.f)) {
 				errmsg("failed to open %s: %m\n", optarg);
 				return EXIT_FAILURE;
 			}
@@ -448,7 +444,11 @@ int main(int argc, char **argv)
 		return EXIT_FAILURE;
 	}
 
-	img = lbximg_fopen(inf);
+	if (fseek(state.f, 0, SEEK_CUR) == 0)
+		img = lbximg_open(state.f, &lbx_default_fops, NULL);
+	else
+		img = lbximg_open(&state, &lbx_pipe_fops, NULL);
+
 	if (!img) {
 		errmsg("failed to open image: %s.\n", lbx_strerror());
 		return EXIT_FAILURE;
@@ -466,13 +466,10 @@ int main(int argc, char **argv)
 
 	switch (mode) {
 	case MODE_DECODE:
-		if (decode(img, palf, overf, &argv[optind])) {
-			lbximg_close(img);
-			return EXIT_FAILURE;
-		}
+		rc = decode(img, palf, overf, &argv[optind]);
 		break;
 	}
 
 	lbximg_close(img);
-	return EXIT_SUCCESS;
+	return rc;
 }