From 588976a9a16eaa0f1b694587b459b2b264e5fe49 Mon Sep 17 00:00:00 2001 From: justanothercatgirl Date: Tue, 1 Apr 2025 08:27:33 +0300 Subject: added -o flag to STE --- ste/ste.c | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) (limited to 'ste/ste.c') 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] \n", *argv); + fprintf(stderr, "Usage: %s [-m name] [-o out] \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; -- cgit v1.2.3-70-g09d2