Added chdir and parameter

src/wstationd.c

@@ -19,7 +19,7 @@ static char *exec_name;
 
 static void print_help()
 {
-	fprintf(stdout, "usage: %s\n", exec_name);
+	fprintf(stdout, "usage: %s [-p port] [-b addr] directory\n", exec_name);
 	fprintf(stdout, "  -p, --port <port>     Port to listen on (default: 10800)\n");
 	fprintf(stdout, "  -b, --bind <address>  Address to bind to (default: 0.0.0.0)\n");
 	fprintf(stdout, "  -h, --help            Print this message and exit\n");
@@ -52,7 +52,7 @@ int main(int argc, char *argv[])
 
 	exec_name = basename(argv[0]);
 
-	for(int ch; (ch = getopt_long(argc, argv, "b:p:vh", longopts, NULL)) != -1;){
+	for(int ch; (ch = getopt_long(argc, argv, "b:p:vh0", longopts, NULL)) != -1;){
 		switch(ch){
 			case 'p':
 				port = (int)strtol(optarg, NULL, 10);
@@ -82,6 +82,20 @@ int main(int argc, char *argv[])
 		}
 	}
 
+	// Accept final argument, the directory
+	if(argc == (optind+1)){
+		// Fail if we can't chdir to that directory
+		if(chdir(argv[argc-1]) == -1){
+			fprintf(stderr, "%s: cannot change directory - %s\n",
+				exec_name, strerror(errno)
+			);
+			goto shutdown_error;
+		}
+	} else {
+		print_help();
+		goto shutdown_error;
+	}
+	
 	// Establish the socket
 	sockfd = socket(AF_INET, SOCK_STREAM, 0);
 	if(sockfd == -1){