From 575ff1f2b95ca64636cec7db7d1e54342eed4177 Mon Sep 17 00:00:00 2001 From: Andrew Pamment Date: Mon, 2 Dec 2019 10:20:16 +1000 Subject: [PATCH] add a few checks --- MD_Init.c | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/MD_Init.c b/MD_Init.c index d0dada4..a1fa100 100644 --- a/MD_Init.c +++ b/MD_Init.c @@ -183,12 +183,12 @@ void md_exit(int exitcode) { exit(exitcode); } -int md_init(const char *dropfile, int socket) { +void md_init(const char *dropfile, int socket) { char *filename = strrchr(dropfile, PATH_SEP); #if !defined(WIN32) && !defined(_MSC_VER) struct termios ttystate; #endif - int ret; + int ret = -1; mdcontrol.socket = socket; @@ -202,6 +202,14 @@ int md_init(const char *dropfile, int socket) { ret = read_doorsys(dropfile); } else if (strcasecmp(filename, "door32.sys") == 0) { ret = read_door32(dropfile); + } else { + fprintf(stderr, "Unsupported dropfile: %s\n", filename); + exit(-1); + } + + if (ret == -1) { + fprintf(stderr, "Unable to open dropfile: %s\n", dropfile); + exit(-1); } if (mdcontrol.socket == -1) {