GUAC-847: Read from file (stub). Override with args.

This commit is contained in:
Michael Jumper 2014-09-08 11:09:29 -07:00
parent 8c78f105cc
commit 9e57d01c99
6 changed files with 219 additions and 73 deletions

View File

@ -30,6 +30,7 @@ man_MANS = man/guacd.8
noinst_HEADERS = \ noinst_HEADERS = \
client.h \ client.h \
client-map.h \ client-map.h \
conf-args.h \
conf-file.h \ conf-file.h \
conf-parse.h \ conf-parse.h \
log.h log.h
@ -38,6 +39,7 @@ guacd_SOURCES = \
daemon.c \ daemon.c \
client.c \ client.c \
client-map.c \ client-map.c \
conf-args.c \
conf-file.c \ conf-file.c \
conf-parse.c \ conf-parse.c \
log.c log.c

105
src/guacd/conf-args.c Normal file
View File

@ -0,0 +1,105 @@
/*
* Copyright (C) 2014 Glyptodon LLC
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "config.h"
#include "conf-args.h"
#include "conf-file.h"
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int guacd_conf_parse_args(guacd_config* config, int argc, char** argv) {
/* Parse arguments */
int opt;
while ((opt = getopt(argc, argv, "l:b:p:C:K:f")) != -1) {
/* -l: Bind port */
if (opt == 'l') {
free(config->bind_port);
config->bind_port = strdup(optarg);
}
/* -b: Bind host */
else if (opt == 'b') {
free(config->bind_host);
config->bind_host = strdup(optarg);
}
/* -f: Run in foreground */
else if (opt == 'f') {
config->foreground = 1;
}
/* -p: PID file */
else if (opt == 'p') {
free(config->pidfile);
config->pidfile = strdup(optarg);
}
#ifdef ENABLE_SSL
/* -C SSL certificate */
else if (opt == 'C') {
free(config->cert_file);
config->cert_file = strdup(optarg);
}
/* -K SSL key */
else if (opt == 'K') {
free(config->key_file);
config->key_file = strdup(optarg);
}
#else
else if (opt == 'C' || opt == 'K') {
fprintf(stderr,
"This guacd does not have SSL/TLS support compiled in.\n\n"
"If you wish to enable support for the -%c option, please install libssl and\n"
"recompile guacd.\n",
opt);
return 1;
}
#endif
else {
fprintf(stderr, "USAGE: %s"
" [-l LISTENPORT]"
" [-b LISTENADDRESS]"
" [-p PIDFILE]"
#ifdef ENABLE_SSL
" [-C CERTIFICATE_FILE]"
" [-K PEM_FILE]"
#endif
" [-f]\n", argv[0]);
return 1;
}
}
/* Success */
return 0;
}

37
src/guacd/conf-args.h Normal file
View File

@ -0,0 +1,37 @@
/*
* Copyright (C) 2014 Glyptodon LLC
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#ifndef _GUACD_CONF_ARGS_H
#define _GUACD_CONF_ARGS_H
#include "config.h"
#include "conf-file.h"
/**
* Parses the given arguments into the given configuration. Zero is returned on
* success, and non-zero is returned if arguments cannot be parsed.
*/
int guacd_conf_parse_args(guacd_config* config, int argc, char** argv);
#endif

View File

@ -25,12 +25,33 @@
#include "conf-file.h" #include "conf-file.h"
#include "conf-parse.h" #include "conf-parse.h"
int guacd_conf_parse_file(guacd_config* conf, int fd) { #include <stdlib.h>
#include <string.h>
/* Load defaults */ int guacd_conf_parse_file(guacd_config* conf, int fd) {
conf->bind_port = 4822;
/* STUB */ /* STUB */
return 0; return 0;
} }
guacd_config* guacd_conf_load() {
guacd_config* conf = malloc(sizeof(guacd_config));
if (conf == NULL)
return NULL;
/* Load defaults */
conf->bind_host = NULL;
conf->bind_port = strdup("4822");
conf->pidfile = NULL;
conf->foreground = 0;
#ifdef ENABLE_SSL
conf->cert_file = NULL;
conf->key_file = NULL;
#endif
return conf;
}

View File

@ -30,10 +30,37 @@
*/ */
typedef struct guacd_config { typedef struct guacd_config {
/**
* The host to bind on.
*/
char* bind_host;
/** /**
* The port to bind on. * The port to bind on.
*/ */
int bind_port; char* bind_port;
/**
* The file to write the PID in, if any.
*/
char* pidfile;
/**
* Whether guacd should run in the foreground.
*/
int foreground;
#ifdef ENABLE_SSL
/**
* SSL certificate file.
*/
char* cert_file;
/**
* SSL private key file.
*/
char* key_file;
#endif
} guacd_config; } guacd_config;
@ -44,5 +71,11 @@ typedef struct guacd_config {
*/ */
int guacd_conf_parse_file(guacd_config* conf, int fd); int guacd_conf_parse_file(guacd_config* conf, int fd);
/**
* Loads the configuration from any of several default locations, if found. If
* parsing fails, NULL is returned, and an error message is printed to stderr.
*/
guacd_config* guacd_conf_load();
#endif #endif

View File

@ -24,6 +24,8 @@
#include "client.h" #include "client.h"
#include "client-map.h" #include "client-map.h"
#include "conf-args.h"
#include "conf-file.h"
#include "log.h" #include "log.h"
#include <guacamole/client.h> #include <guacamole/client.h>
@ -40,7 +42,6 @@
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <getopt.h>
#include <libgen.h> #include <libgen.h>
#include <netdb.h> #include <netdb.h>
#include <netinet/in.h> #include <netinet/in.h>
@ -375,17 +376,7 @@ int main(int argc, char* argv[]) {
socklen_t client_addr_len; socklen_t client_addr_len;
int connected_socket_fd; int connected_socket_fd;
/* Arguments */
char* listen_address = NULL; /* Default address of INADDR_ANY */
char* listen_port = "4822"; /* Default port */
char* pidfile = NULL;
int opt;
int foreground = 0;
#ifdef ENABLE_SSL #ifdef ENABLE_SSL
/* SSL */
char* cert_file = NULL;
char* key_file = NULL;
SSL_CTX* ssl_context = NULL; SSL_CTX* ssl_context = NULL;
#endif #endif
@ -394,53 +385,10 @@ int main(int argc, char* argv[]) {
/* General */ /* General */
int retval; int retval;
/* Parse arguments */ /* Load configuration */
while ((opt = getopt(argc, argv, "l:b:p:C:K:f")) != -1) { guacd_config* config = guacd_conf_load();
if (opt == 'l') { if (config == NULL || guacd_conf_parse_args(config, argc, argv))
listen_port = strdup(optarg);
}
else if (opt == 'b') {
listen_address = strdup(optarg);
}
else if (opt == 'f') {
foreground = 1;
}
else if (opt == 'p') {
pidfile = strdup(optarg);
}
#ifdef ENABLE_SSL
else if (opt == 'C') {
cert_file = strdup(optarg);
}
else if (opt == 'K') {
key_file = strdup(optarg);
}
#else
else if (opt == 'C' || opt == 'K') {
fprintf(stderr,
"This guacd does not have SSL/TLS support compiled in.\n\n"
"If you wish to enable support for the -%c option, please install libssl and\n"
"recompile guacd.\n",
opt);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
}
#endif
else {
fprintf(stderr, "USAGE: %s"
" [-l LISTENPORT]"
" [-b LISTENADDRESS]"
" [-p PIDFILE]"
#ifdef ENABLE_SSL
" [-C CERTIFICATE_FILE]"
" [-K PEM_FILE]"
#endif
" [-f]\n", argv[0]);
exit(EXIT_FAILURE);
}
}
/* Set up logging prefix */ /* Set up logging prefix */
strncpy(log_prefix, basename(argv[0]), sizeof(log_prefix)); strncpy(log_prefix, basename(argv[0]), sizeof(log_prefix));
@ -452,7 +400,7 @@ int main(int argc, char* argv[]) {
guacd_log_info("Guacamole proxy daemon (guacd) version " VERSION); guacd_log_info("Guacamole proxy daemon (guacd) version " VERSION);
/* Get addresses for binding */ /* Get addresses for binding */
if ((retval = getaddrinfo(listen_address, listen_port, if ((retval = getaddrinfo(config->bind_host, config->bind_port,
&hints, &addresses))) { &hints, &addresses))) {
guacd_log_error("Error parsing given address or port: %s", guacd_log_error("Error parsing given address or port: %s",
@ -521,7 +469,7 @@ int main(int argc, char* argv[]) {
#ifdef ENABLE_SSL #ifdef ENABLE_SSL
/* Init SSL if enabled */ /* Init SSL if enabled */
if (key_file != NULL || cert_file != NULL) { if (config->key_file != NULL || config->cert_file != NULL) {
/* Init SSL */ /* Init SSL */
guacd_log_info("Communication will require SSL/TLS."); guacd_log_info("Communication will require SSL/TLS.");
@ -530,9 +478,9 @@ int main(int argc, char* argv[]) {
ssl_context = SSL_CTX_new(SSLv23_server_method()); ssl_context = SSL_CTX_new(SSLv23_server_method());
/* Load key */ /* Load key */
if (key_file != NULL) { if (config->key_file != NULL) {
guacd_log_info("Using PEM keyfile %s", key_file); guacd_log_info("Using PEM keyfile %s", config->key_file);
if (!SSL_CTX_use_PrivateKey_file(ssl_context, key_file, SSL_FILETYPE_PEM)) { if (!SSL_CTX_use_PrivateKey_file(ssl_context, config->key_file, SSL_FILETYPE_PEM)) {
guacd_log_error("Unable to load keyfile."); guacd_log_error("Unable to load keyfile.");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
@ -541,9 +489,9 @@ int main(int argc, char* argv[]) {
guacd_log_info("No PEM keyfile given - SSL/TLS may not work."); guacd_log_info("No PEM keyfile given - SSL/TLS may not work.");
/* Load cert file if specified */ /* Load cert file if specified */
if (cert_file != NULL) { if (config->cert_file != NULL) {
guacd_log_info("Using certificate file %s", cert_file); guacd_log_info("Using certificate file %s", config->cert_file);
if (!SSL_CTX_use_certificate_chain_file(ssl_context, cert_file)) { if (!SSL_CTX_use_certificate_chain_file(ssl_context, config->cert_file)) {
guacd_log_error("Unable to load certificate."); guacd_log_error("Unable to load certificate.");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
@ -555,7 +503,7 @@ int main(int argc, char* argv[]) {
#endif #endif
/* Daemonize if requested */ /* Daemonize if requested */
if (!foreground) { if (!config->foreground) {
/* Attempt to daemonize process */ /* Attempt to daemonize process */
if (daemonize()) { if (daemonize()) {
@ -566,10 +514,10 @@ int main(int argc, char* argv[]) {
} }
/* Write PID file if requested */ /* Write PID file if requested */
if (pidfile != NULL) { if (config->pidfile != NULL) {
/* Attempt to open pidfile and write PID */ /* Attempt to open pidfile and write PID */
FILE* pidf = fopen(pidfile, "w"); FILE* pidf = fopen(config->pidfile, "w");
if (pidf) { if (pidf) {
fprintf(pidf, "%d\n", getpid()); fprintf(pidf, "%d\n", getpid());
fclose(pidf); fclose(pidf);