1 files changed, 12 insertions, 7 deletions

diff --git a/ste/ste.c b/ste/ste.c
index 43252b9..7ec1fe4 100644
--- a/ste/ste.c
+++ b/ste/ste.c
@@ -21,29 +21,34 @@ void process_files(FILE* in, FILE* out, const char *macro) {
 
 int main(int argc, char **argv) {
 	const char *macro = "OUT";
-	char *filename = NULL;
+	char *filename = NULL, *outname = NULL;
 	FILE *in, *out;
 
 	if (argc < 2) {
 		usage:
-		fprintf(stderr, "Usage: %s [-m name] <file.ste>\n", *argv);
+		fprintf(stderr, "Usage: %s [-m name] [-o out] <file.ste>\n", *argv);
 		return 1;
 	}
 
 	for (int i = 0; i < argc; ++i) {
 		if (!strcmp("-m", argv[i])) macro = argv[++i];
+		else if (!strcmp("-o", argv[i])) outname = argv[++i];
 		else filename = argv[i];
 	}
 	if (!filename || strlen(filename) <= 4) goto usage;
 
 	in = fopen(filename, "r");
-	for (char *fn = filename; fn[3]; ++fn) {
-		if (!strncmp(".ste", fn, 4)) {
-			*fn = '\0';
-			break;
+	if (!outname) {
+		for (char *fn = filename; fn[3]; ++fn) {
+			if (!strncmp(".ste", fn, 4)) {
+				*fn = '\0';
+				break;
+			}
 		}
+		out = fopen(filename, "w");
+	} else {
+		out = fopen(outname, "w");
 	}
-	out = fopen(filename, "w");
 	if (!in || !out) {
 		perror("fopen");
 		goto usage;