Subversion Repositories Tewi

Rev

Rev 51 | Rev 70 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
2 nishi 1 
/* $Id: main.c 62 2024-09-18 20:02:26Z nishi $ */
2 
 
16 nishi 3 
#define SOURCE
4 
 
43 nishi 5 
#include "../config.h"
6 
 
3 nishi 7 
#include <stdio.h>
8 
#include <stdbool.h>
9 
#include <string.h>
16 nishi 10 
#include <signal.h>
3 nishi 11 
 
43 nishi 12 
#ifndef NO_SSL
6 nishi 13 
#include <openssl/opensslv.h>
43 nishi 14 
#endif
6 nishi 15 
 
3 nishi 16 
#include <cm_log.h>
62 nishi 17 
#include <cm_string.h>
3 nishi 18 
 
4 nishi 19 
#include "tw_config.h"
8 nishi 20 
#include "tw_server.h"
3 nishi 21 
#include "tw_version.h"
22 
 
51 nishi 23 
#ifdef __MINGW32__
24 
#include <windows.h>
25 
#endif
26 
 
3 nishi 27 
extern bool cm_do_log;
18 nishi 28 
extern struct tw_config config;
62 nishi 29 
extern FILE* logfile;
3 nishi 30 
 
18 nishi 31 
char tw_server[2048];
32 
 
62 nishi 33 
int startup(int argc, char** argv);
34 
 
35 
#ifdef SERVICE
36 
SERVICE_STATUS status;
37 
SERVICE_STATUS_HANDLE status_handle;
38 
 
39 
void WINAPI servhandler(DWORD control){
40 
	switch(control){
41 
		case SERVICE_CONTROL_STOP:
42 
		case SERVICE_CONTROL_SHUTDOWN:
43 
			status.dwCurrentState = SERVICE_STOP_PENDING;
44 
			break;
45 
	}
46 
	SetServiceStatus(status_handle, &status);
47 
}
48 
 
49 
void WINAPI servmain(DWORD argc, LPSTR* argv){
50 
	logfile = fopen(PREFIX "/logs/tewi.log", "a");
51 
	if(logfile == NULL) logfile = stderr;
52 
	status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
53 
	status.dwCurrentState = SERVICE_START_PENDING;
54 
	status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
55 
	status.dwWin32ExitCode = NO_ERROR;
56 
	status.dwServiceSpecificExitCode = 0;
57 
	status.dwCheckPoint = 0;
58 
	status.dwWaitHint = 0;
59 
	status_handle = RegisterServiceCtrlHandler("Tewi HTTPd", servhandler);
60 
	if(status_handle == NULL) return;
61 
	if(SetServiceStatus(status_handle, &status) == 0) return;
62 
	int st = startup(argc, argv);
63 
	if(st != -1){
64 
		status.dwWin32ExitCode = NO_ERROR;
65 
		status.dwServiceSpecificExitCode = st;
66 
		status.dwCurrentState = SERVICE_STOPPED;
67 
		SetServiceStatus(status_handle, &status);
68 
		return;
69 
	}
70 
	status.dwCurrentState = SERVICE_RUNNING;
71 
	SetServiceStatus(status_handle, &status);
72 
	tw_server_loop();
73 
	status.dwCurrentState = SERVICE_STOPPED;
74 
	SetServiceStatus(status_handle, &status);
75 
}
76 
#endif
77 
 
3 nishi 78 
int main(int argc, char** argv) {
62 nishi 79 
	logfile = stderr;
80 
#ifdef SERVICE
81 
	SERVICE_TABLE_ENTRY table[] = {{"Tewi HTTPd", servmain}, {NULL, NULL}};
82 
	StartServiceCtrlDispatcher(table);
83 
#else
84 
	int st = startup(argc, argv);
85 
	if(st != -1) return st;
86 
	tw_server_loop();
87 
#endif
88 
}
89 
 
90 
int startup(int argc, char** argv){
3 nishi 91 
	int i;
18 nishi 92 
	const char* confpath = PREFIX "/etc/tewi.conf";
62 nishi 93 
	if(argv != NULL){
94 
		for(i = 1; i < argc; i++) {
95 
			if(argv[i][0] == '-') {
96 
				if(strcmp(argv[i], "--verbose") == 0 || strcmp(argv[i], "-v") == 0) {
97 
					if(!cm_do_log) {
98 
						cm_do_log = true;
99 
	#ifndef NO_SSL
100 
						cm_log("", "This is Tewi HTTPd, version %s, using %s", tw_get_version(), OPENSSL_VERSION_TEXT);
101 
	#else
102 
						cm_log("", "This is Tewi HTTPd, version %s", tw_get_version());
103 
	#endif
104 
					} else {
105 
						cm_do_log = true;
106 
					}
107 
				} else if(strcmp(argv[i], "--config") == 0 || strcmp(argv[i], "-C") == 0) {
108 
					i++;
109 
					if(argv[i] == NULL) {
110 
						fprintf(stderr, "Missing argument\n");
111 
						return 1;
112 
					}
113 
					confpath = argv[i];
114 
				} else if(strcmp(argv[i], "--version") == 0 || strcmp(argv[i], "-V") == 0) {
115 
					printf("Tewi HTTPd Tewi/%s\n", tw_get_version());
116 
					printf("Under public domain.\n");
117 
					printf("Original by 2024 Nishi\n");
118 
					printf("\n");
119 
					printf("Usage: %s [--config|-C config] [--verbose|-v] [--version|-V]\n", argv[0]);
120 
					printf("--config  | -C config     : Specify config\n");
121 
					printf("--verbose | -v            : Verbose mode\n");
122 
					printf("--version | -V            : Version information\n");
123 
					return 0;
3 nishi 124 
				} else {
62 nishi 125 
					fprintf(stderr, "Unknown option: %s\n", argv[i]);
4 nishi 126 
					return 1;
127 
				}
3 nishi 128 
			}
129 
		}
130 
	}
6 nishi 131 
	tw_config_init();
18 nishi 132 
	if(tw_config_read(confpath) != 0) {
4 nishi 133 
		fprintf(stderr, "Could not read the config\n");
134 
		return 1;
135 
	}
8 nishi 136 
	if(tw_server_init() != 0) {
137 
		fprintf(stderr, "Could not initialize the server\n");
138 
		return 1;
139 
	}
18 nishi 140 
	sprintf(tw_server, "Tewi/%s (%s)%s", tw_get_version(), tw_get_platform(), config.extension == NULL ? "" : config.extension);
62 nishi 141 
	char* r = cm_strcat(tw_server, " running...");
142 
	cm_force_log(r);
143 
	free(r);
16 nishi 144 
#ifndef __MINGW32__
145 
	signal(SIGCHLD, SIG_IGN);
50 nishi 146 
#else
147 
	SetConsoleTitle(tw_server);
16 nishi 148 
#endif
62 nishi 149 
	return -1;
3 nishi 150 
}