diff options
Diffstat (limited to 'lib9p/tests/test_server/main.c')
-rw-r--r-- | lib9p/tests/test_server/main.c | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib9p/tests/test_server/main.c b/lib9p/tests/test_server/main.c index d7819eb..8490d62 100644 --- a/lib9p/tests/test_server/main.c +++ b/lib9p/tests/test_server/main.c @@ -151,7 +151,11 @@ int main(int argc, char *argv[]) { if (argc != 3) error(2, 0, "usage: %s PORT_NUMBER LOGFILE", argv[0]); - globals.port = atoi(argv[1]); + int _port = atoi(argv[1]); + if (_port == 0 || _port > UINT16_MAX) + error(2, 0, "usage: %s PORT_NUMBER LOGFILE", argv[0]); + + globals.port = (uint16_t)_port; globals.logstream = fopen(argv[2], "w"); if (!globals.logstream) error(2, errno, "fopen"); |