mirror of
https://github.com/cesanta/mongoose.git
synced 2024-12-28 07:28:13 +08:00
296 lines
8.9 KiB
C
296 lines
8.9 KiB
C
/*
|
|
* Copyright (c) 2004-2009 Sergey Lyubka
|
|
*
|
|
* 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.
|
|
*
|
|
* $Id: main.c 518 2010-05-03 12:55:35Z valenok $
|
|
*/
|
|
|
|
#if defined(_WIN32)
|
|
#define _CRT_SECURE_NO_WARNINGS /* Disable deprecation warning in VS2005 */
|
|
#endif /* _WIN32 */
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <stddef.h>
|
|
|
|
#include "mongoose.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#include <winsvc.h>
|
|
#define DIRSEP '\\'
|
|
#define snprintf _snprintf
|
|
#if !defined(__LCC__)
|
|
#define strdup(x) _strdup(x)
|
|
#endif /* !MINGW */
|
|
#define sleep(x) Sleep((x) * 1000)
|
|
#else
|
|
#include <sys/wait.h>
|
|
#include <unistd.h> /* For pause() */
|
|
#define DIRSEP '/'
|
|
#endif /* _WIN32 */
|
|
|
|
static int exit_flag; /* Program termination flag */
|
|
|
|
#if !defined(CONFIG_FILE)
|
|
#define CONFIG_FILE "mongoose.conf"
|
|
#endif /* !CONFIG_FILE */
|
|
|
|
static void
|
|
signal_handler(int sig_num)
|
|
{
|
|
#if !defined(_WIN32)
|
|
if (sig_num == SIGCHLD) {
|
|
do {
|
|
} while (waitpid(-1, &sig_num, WNOHANG) > 0);
|
|
} else
|
|
#endif /* !_WIN32 */
|
|
{
|
|
exit_flag = sig_num;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Edit the passwords file.
|
|
*/
|
|
static int
|
|
mg_edit_passwords(const char *fname, const char *domain,
|
|
const char *user, const char *pass)
|
|
{
|
|
struct mg_context *ctx;
|
|
struct mg_config config;
|
|
int retval;
|
|
|
|
memset(&config, 0, sizeof(config));
|
|
config.auth_domain = (char *) domain;
|
|
config.num_threads = "0";
|
|
config.listening_ports = "";
|
|
ctx = mg_start(&config);
|
|
retval = mg_modify_passwords_file(ctx, fname, user, pass);
|
|
mg_stop(ctx);
|
|
|
|
return (retval);
|
|
}
|
|
|
|
#define OFFSET(x) offsetof(struct mg_config, x)
|
|
|
|
static struct option_descriptor {
|
|
const char *name;
|
|
const char *description;
|
|
size_t offset;
|
|
} known_options[] = {
|
|
{"root", "\tWeb root directory", OFFSET(document_root)},
|
|
{"index_files", "Index files", OFFSET(index_files)},
|
|
{"ssl_cert", "SSL certificate file", OFFSET(ssl_certificate)},
|
|
{"ports", "Listening ports", OFFSET(listening_ports)},
|
|
{"dir_list", "Directory listing", OFFSET(enable_directory_listing)},
|
|
{"protect", "URI to htpasswd mapping", OFFSET(protect)},
|
|
{"cgi_ext", "CGI extensions", OFFSET(cgi_extensions)},
|
|
{"cgi_interp", "CGI interpreter to use", OFFSET(cgi_interpreter)},
|
|
{"cgi_env", "Custom CGI enviroment variables", OFFSET(cgi_environment)},
|
|
{"ssi_ext", "SSI extensions", OFFSET(ssi_extensions)},
|
|
{"auth_realm", "Authentication domain name", OFFSET(auth_domain)},
|
|
{"auth_gpass", "Global passwords file", OFFSET(global_passwords_file)},
|
|
{"auth_PUT", "PUT,DELETE auth file", OFFSET(put_delete_passwords_file)},
|
|
{"uid", "\tRun as user", OFFSET(uid)},
|
|
{"access_log", "Access log file", OFFSET(access_log_file)},
|
|
{"error_log", "Error log file", OFFSET(error_log_file)},
|
|
{"acl", "\tAllow/deny IP addresses/subnets", OFFSET(acl)},
|
|
{"num_threads", "Threads to spawn", OFFSET(num_threads)},
|
|
{"mime_types", "Extra mime types to use", OFFSET(mime_types)},
|
|
{NULL, NULL, 0}
|
|
};
|
|
|
|
static void
|
|
show_usage_and_exit(const struct mg_config *config)
|
|
{
|
|
const struct option_descriptor *o;
|
|
const char *value;
|
|
|
|
(void) fprintf(stderr,
|
|
"Mongoose version %s (c) Sergey Lyubka\n"
|
|
"usage: mongoose [options] [config_file]\n", mg_version());
|
|
|
|
fprintf(stderr, " -A <htpasswd_file> <realm> <user> <passwd>\n");
|
|
|
|
for (o = known_options; o->name != NULL; o++) {
|
|
(void) fprintf(stderr, " -%s\t%s", o->name, o->description);
|
|
value = * (char **) ((char *) config + o->offset);
|
|
if (value != NULL)
|
|
fprintf(stderr, " (default: \"%s\")", value);
|
|
fputc('\n', stderr);
|
|
}
|
|
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
static void
|
|
set_option(struct mg_config *config, const char *name, char *value)
|
|
{
|
|
const struct option_descriptor *o;
|
|
|
|
for (o = known_options; o->name != NULL; o++)
|
|
if (strcmp(name, o->name) == 0) {
|
|
* (char **) ((char *) config + o->offset) = value;
|
|
break;
|
|
}
|
|
|
|
if (o->name == NULL)
|
|
show_usage_and_exit(config);
|
|
}
|
|
|
|
static void
|
|
process_command_line_arguments(struct mg_config *config, char *argv[])
|
|
{
|
|
const char *config_file = CONFIG_FILE;
|
|
char line[512], opt[512], *vals[100],
|
|
val[512], path[FILENAME_MAX], *p;
|
|
FILE *fp;
|
|
size_t i, line_no = 0;
|
|
|
|
/* First find out, which config file to open */
|
|
for (i = 1; argv[i] != NULL && argv[i][0] == '-'; i += 2)
|
|
if (argv[i + 1] == NULL)
|
|
show_usage_and_exit(config);
|
|
|
|
if (argv[i] != NULL && argv[i + 1] != NULL) {
|
|
/* More than one non-option arguments are given */
|
|
show_usage_and_exit(config);
|
|
} else if (argv[i] != NULL) {
|
|
/* Just one non-option argument is given, this is config file */
|
|
config_file = argv[i];
|
|
} else {
|
|
/* No config file specified. Look for one where binary lives */
|
|
if ((p = strrchr(argv[0], DIRSEP)) != 0) {
|
|
(void) snprintf(path, sizeof(path), "%.*s%s",
|
|
(int) (p - argv[0]) + 1, argv[0], config_file);
|
|
config_file = path;
|
|
}
|
|
}
|
|
|
|
fp = fopen(config_file, "r");
|
|
|
|
/* If config file was set in command line and open failed, exit */
|
|
if (fp == NULL && argv[i] != NULL) {
|
|
(void) fprintf(stderr, "cannot open config file %s: %s\n",
|
|
config_file, strerror(errno));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
/* Reset temporary value holders */
|
|
(void) memset(vals, 0, sizeof(vals));
|
|
|
|
if (fp != NULL) {
|
|
(void) printf("Loading config file %s, "
|
|
"ignoring command line arguments\n", config_file);
|
|
|
|
/* Loop over the lines in config file */
|
|
while (fgets(line, sizeof(line), fp) != NULL) {
|
|
|
|
line_no++;
|
|
|
|
/* Ignore empty lines and comments */
|
|
if (line[0] == '#' || line[0] == '\n')
|
|
continue;
|
|
|
|
if (sscanf(line, "%s %[^\r\n#]", opt, val) != 2) {
|
|
fprintf(stderr, "%s: line %d is invalid\n",
|
|
config_file, (int) line_no);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
/* TODO(lsm): free this at some point */
|
|
p = malloc(strlen(val) + 1);
|
|
(void) strcpy(p, val);
|
|
set_option(config, opt, p);
|
|
}
|
|
|
|
(void) fclose(fp);
|
|
} else {
|
|
for (i = 1; argv[i] != NULL && argv[i][0] == '-'; i += 2)
|
|
set_option(config, &argv[i][1], argv[i + 1]);
|
|
}
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
struct mg_config config;
|
|
struct mg_context *ctx;
|
|
|
|
/* Initialize configuration with default values */
|
|
(void) memset(&config, 0, sizeof(config));
|
|
config.document_root = ".";
|
|
config.enable_directory_listing = "yes";
|
|
config.auth_domain = "mydomain.com";
|
|
config.num_threads = "20";
|
|
config.index_files = "index.html,index.htm,index.cgi";
|
|
config.cgi_extensions = ".cgi,.pl,.php";
|
|
config.ssi_extensions = ".shtml,.shtm";
|
|
config.listening_ports = "8080";
|
|
|
|
/* Edit passwords file if -A option is specified */
|
|
if (argc > 1 && argv[1][0] == '-' && argv[1][1] == 'A') {
|
|
if (argc != 6)
|
|
show_usage_and_exit(&config);
|
|
exit(mg_edit_passwords(argv[2], argv[3], argv[4], argv[5]) ==
|
|
MG_SUCCESS ? EXIT_SUCCESS : EXIT_FAILURE);
|
|
}
|
|
|
|
/* Show usage if -h or --help options are specified */
|
|
if (argc == 2 && (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help")))
|
|
show_usage_and_exit(&config);
|
|
|
|
/* Update config based on command line arguments */
|
|
process_command_line_arguments(&config, argv);
|
|
|
|
/* Setup signal handler: quit on Ctrl-C */
|
|
#ifndef _WIN32
|
|
(void) signal(SIGCHLD, signal_handler);
|
|
#endif /* _WIN32 */
|
|
(void) signal(SIGTERM, signal_handler);
|
|
(void) signal(SIGINT, signal_handler);
|
|
|
|
/* Start Mongoose */
|
|
if ((ctx = mg_start(&config)) == NULL) {
|
|
(void) printf("%s\n", "Cannot initialize Mongoose context");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
(void) printf("Mongoose %s started on port(s) %s "
|
|
"with web root [%s]\n",
|
|
mg_version(), config.listening_ports, config.document_root);
|
|
|
|
fflush(stdout);
|
|
while (exit_flag == 0)
|
|
sleep(1);
|
|
|
|
(void) printf("Exiting on signal %d, "
|
|
"waiting for all threads to finish...", exit_flag);
|
|
fflush(stdout);
|
|
mg_stop(ctx);
|
|
(void) printf("%s", " done.\n");
|
|
|
|
return (EXIT_SUCCESS);
|
|
}
|