trunk/freehttpd/main.c

#include "freehttpd.h"
#include <errno.h>
#include <getopt.h>
#include <magic.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static struct option const long_options[] = {
    { "help",    no_argument,       NULL, 'h' },
    { "version", no_argument,       NULL, 'v' },
    { "config",  required_argument, NULL, 'c' },
    { NULL,      0,                 NULL, 0   }
};

static const char *const short_options = "hvc:";

struct config
{
    const char *progname;
    const char *config_file;
};

static struct config config = { 0 };

static void
usage (void)
{
    printf ("Usage: %s [OPTIONS]\n", config.progname);
    printf ("Options:\n");
    printf ("  -h, --help     Print this help message\n");
    printf ("  -v, --version  Print version information\n");
    printf ("  -c, --config   Specify configuration file\n");
}

static void
version (void)
{
    printf ("freehttpd 0.0.0\n");
}

static bool
validate_settings (void)
{
    return true;
}

static void
initialize (const char *progname)
{
    config.progname = progname;
    config.config_file = "freehttpd.conf";
}

int
main (int argc, char **argv)
{
    initialize (argv[0]);

    while (true)
        {
            int option_index = 0;
            int c = getopt_long (argc, argv, short_options, long_options,
                                 &option_index);

            if (c == -1)
                break;

            switch (c)
                {
                case 'h':
                    usage ();
                    exit (EXIT_SUCCESS);

                case 'v':
                    version ();
                    exit (EXIT_SUCCESS);

                case 'c':
                    config.config_file = optarg;
                    break;

                case '?':
                default:
                    exit (EXIT_FAILURE);
                }
        }

    if (!validate_settings ())
        exit (EXIT_FAILURE);

    magic_t magic = magic_open (MAGIC_MIME);
    unsigned int port = 8080;
    size_t uri_len = 1024;
    ecode_t code = E_OK;

    if (magic == NULL)
        {
            fprintf (stderr, "%s: failed to initialize magic\n",
                     config.progname);
            exit (EXIT_FAILURE);
        }

    if (magic_load (magic, NULL) != 0)
        {
            fprintf (stderr, "%s: failed to load magic database\n",
                     config.progname);
            magic_close (magic);
            exit (EXIT_FAILURE);
        }

    freehttpd_t *freehttpd = freehttpd_init (magic);

    code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_PORT, &port);

    if (code != E_OK)
        goto error;

    code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_ADDR, NULL);

    if (code != E_OK)
        goto error;

    code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_MAX_URI_LEN, &uri_len);

    if (code != E_OK)
        goto error;

    char *docroot = getenv ("FREEHTTPD_DOCROOT");
    code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_DOCROOT,
                             docroot ? docroot : "/var/www");

    if (code != E_OK)
        goto error;

    code = freehttpd_start (freehttpd);

    if (code != E_OK)
        {
        error:
            fprintf (stderr, "%s: failed to start: %d: %s\n", config.progname,
                     code, strerror (errno));
            freehttpd_free (freehttpd);
            exit (EXIT_FAILURE);
        }

    freehttpd_free (freehttpd);
    magic_close (magic);
    return 0;
}
1	#include "freehttpd.h"
2	#include <errno.h>
3	#include <getopt.h>
4	#include <magic.h>
5	#include <stdbool.h>
6	#include <stdio.h>
7	#include <stdlib.h>
8	#include <string.h>
9
10	static struct option const long_options[] = {
11	{ "help", no_argument, NULL, 'h' },
12	{ "version", no_argument, NULL, 'v' },
13	{ "config", required_argument, NULL, 'c' },
14	{ NULL, 0, NULL, 0 }
15	};
16
17	static const char *const short_options = "hvc:";
18
19	struct config
20	{
21	const char *progname;
22	const char *config_file;
23	};
24
25	static struct config config = { 0 };
26
27	static void
28	usage (void)
29	{
30	printf ("Usage: %s [OPTIONS]\n", config.progname);
31	printf ("Options:\n");
32	printf (" -h, --help Print this help message\n");
33	printf (" -v, --version Print version information\n");
34	printf (" -c, --config Specify configuration file\n");
35	}
36
37	static void
38	version (void)
39	{
40	printf ("freehttpd 0.0.0\n");
41	}
42
43	static bool
44	validate_settings (void)
45	{
46	return true;
47	}
48
49	static void
50	initialize (const char *progname)
51	{
52	config.progname = progname;
53	config.config_file = "freehttpd.conf";
54	}
55
56	int
57	main (int argc, char **argv)
58	{
59	initialize (argv[0]);
60
61	while (true)
62	{
63	int option_index = 0;
64	int c = getopt_long (argc, argv, short_options, long_options,
65	&option_index);
66
67	if (c == -1)
68	break;
69
70	switch (c)
71	{
72	case 'h':
73	usage ();
74	exit (EXIT_SUCCESS);
75
76	case 'v':
77	version ();
78	exit (EXIT_SUCCESS);
79
80	case 'c':
81	config.config_file = optarg;
82	break;
83
84	case '?':
85	default:
86	exit (EXIT_FAILURE);
87	}
88	}
89
90	if (!validate_settings ())
91	exit (EXIT_FAILURE);
92
93	magic_t magic = magic_open (MAGIC_MIME);
94	unsigned int port = 8080;
95	size_t uri_len = 1024;
96	ecode_t code = E_OK;
97
98	if (magic == NULL)
99	{
100	fprintf (stderr, "%s: failed to initialize magic\n",
101	config.progname);
102	exit (EXIT_FAILURE);
103	}
104
105	if (magic_load (magic, NULL) != 0)
106	{
107	fprintf (stderr, "%s: failed to load magic database\n",
108	config.progname);
109	magic_close (magic);
110	exit (EXIT_FAILURE);
111	}
112
113	freehttpd_t *freehttpd = freehttpd_init (magic);
114
115	code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_PORT, &port);
116
117	if (code != E_OK)
118	goto error;
119
120	code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_ADDR, NULL);
121
122	if (code != E_OK)
123	goto error;
124
125	code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_MAX_URI_LEN, &uri_len);
126
127	if (code != E_OK)
128	goto error;
129
130	char *docroot = getenv ("FREEHTTPD_DOCROOT");
131	code = freehttpd_setopt (freehttpd, FREEHTTPD_CONFIG_DOCROOT,
132	docroot ? docroot : "/var/www");
133
134	if (code != E_OK)
135	goto error;
136
137	code = freehttpd_start (freehttpd);
138
139	if (code != E_OK)
140	{
141	error:
142	fprintf (stderr, "%s: failed to start: %d: %s\n", config.progname,
143	code, strerror (errno));
144	freehttpd_free (freehttpd);
145	exit (EXIT_FAILURE);
146	}
147
148	freehttpd_free (freehttpd);
149	magic_close (magic);
150	return 0;
151	}