From b98b5b7d61a5285363c8cec4817f680ad96a12df Mon Sep 17 00:00:00 2001 From: Joseph Hunkeler Date: Fri, 8 Nov 2019 23:40:01 -0500 Subject: Fix segfault while calculating base name --- main.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'main.c') diff --git a/main.c b/main.c index f590126..dadc455 100644 --- a/main.c +++ b/main.c @@ -116,7 +116,12 @@ void reloc_replace(RelocMatch *match, const char *rstr) { int main(int argc, char *argv[]) { - const char *program = strrchr(argv[0], DIRSEP) + 1; + char *program = argv[0]; + char *program_relative = strrchr(program, DIRSEP); + if (program_relative) { + program = program_relative + 1; + } + if (argc < 5) { printf("%s \n" "\n" -- cgit