From b2ca4dfa7dc64b9c03f4c12968bcf9179a04d5a4 Mon Sep 17 00:00:00 2001 From: David Phillips Date: Sat, 6 Jun 2015 23:16:18 +1200 Subject: Switched to using basename --- fractal-gen.c | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'fractal-gen.c') diff --git a/fractal-gen.c b/fractal-gen.c index 6f3cb5f..13bd6ba 100644 --- a/fractal-gen.c +++ b/fractal-gen.c @@ -4,16 +4,20 @@ int main(int argc, char **argv) { unsigned int size, iterat, cores, i, x, y, s; double power; + char* bname; void *(*generator)(void *); // Select correct generator for the fractal type - // TO DO: use basename - if (strcmp(argv[0], "./mbrot-gen") == 0) + bname = basename(argv[0]); + if (strcmp(bname, "mbrot-gen") == 0) + { generator = &generate_mbrot_section; - else if (strcmp(argv[0], "./bship-gen") == 0) + } else if (strcmp(bname, "bship-gen") == 0) { generator = &generate_bship_section; - else + } else { fprintf(stderr, "Don't call this directly, call a symlink to me\n"); + return EXIT_FAILURE; + } if (argc != 4) { -- cgit v1.1