source: Main/trunk/Server/main.c@ 188

Last change on this file since 188 was 187, checked in by Nishi, on Sep 28, 2024 at 9:07:34 AM

kinda works
  • Property svn:keywords set to Id
File size: 8.5 KB
Line 
1/* $Id: main.c 187 2024-09-28 00:07:34Z nishi $ */
2
3#define SOURCE
4
5#include "../config.h"
6
7#include <stdio.h>
8#include <stdbool.h>
9#include <string.h>
10#include <signal.h>
11#include <stdlib.h>
12
13#ifndef NO_SSL
14#include <openssl/opensslv.h>
15#endif
16
17#include <cm_log.h>
18#include <cm_string.h>
19
20#include "tw_config.h"
21#include "tw_server.h"
22#include "tw_version.h"
23
24#ifdef __MINGW32__
25#include <windows.h>
26#endif
27
28#ifdef _PSP
29#include <pspkernel.h>
30#include <pspdebug.h>
31#include <pspsdk.h>
32#include <psputility.h>
33#include <pspctrl.h>
34#include <pspnet_apctl.h>
35#include <pspwlan.h>
36
37PSP_MODULE_INFO("Tewi HTTPd", PSP_MODULE_USER, 1, 1);
38PSP_MAIN_THREAD_ATTR(PSP_THREAD_ATTR_USER);
39
40#define printf(...) pspDebugScreenPrintf(__VA_ARGS__)
41#define STDERR_LOG(...) pspDebugScreenPrintf(__VA_ARGS__)
42#else
43#define STDERR_LOG(...) fprintf(stderr, __VA_ARGS__)
44#endif
45
46extern bool cm_do_log;
47extern struct tw_config config;
48extern FILE* logfile;
49
50char tw_server[2048];
51
52int startup(int argc, char** argv);
53
54#ifdef SERVICE
55SERVICE_STATUS status;
56SERVICE_STATUS_HANDLE status_handle;
57
58void WINAPI servhandler(DWORD control) {
59 switch(control) {
60 case SERVICE_CONTROL_STOP:
61 case SERVICE_CONTROL_SHUTDOWN:
62 status.dwCurrentState = SERVICE_STOP_PENDING;
63 break;
64 }
65 SetServiceStatus(status_handle, &status);
66}
67
68void WINAPI servmain(DWORD argc, LPSTR* argv) {
69 logfile = fopen(PREFIX "/logs/tewi.log", "a");
70 if(logfile == NULL) logfile = stderr;
71 status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
72 status.dwCurrentState = SERVICE_START_PENDING;
73 status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
74 status.dwWin32ExitCode = NO_ERROR;
75 status.dwServiceSpecificExitCode = 0;
76 status.dwCheckPoint = 0;
77 status.dwWaitHint = 0;
78 status_handle = RegisterServiceCtrlHandler("Tewi HTTPd", servhandler);
79 if(status_handle == NULL) return;
80 if(SetServiceStatus(status_handle, &status) == 0) return;
81 int st = startup(argc, argv);
82 if(st != -1) {
83 status.dwWin32ExitCode = NO_ERROR;
84 status.dwServiceSpecificExitCode = st;
85 status.dwCurrentState = SERVICE_STOPPED;
86 SetServiceStatus(status_handle, &status);
87 return;
88 }
89 status.dwCurrentState = SERVICE_RUNNING;
90 SetServiceStatus(status_handle, &status);
91 tw_server_loop();
92 status.dwCurrentState = SERVICE_STOPPED;
93 SetServiceStatus(status_handle, &status);
94}
95#endif
96
97int running = 1;
98#ifdef _PSP
99
100int psp_exit_callback(int arg1, int arg2, void* arg3) { running = 0; }
101
102int psp_callback_thread(SceSize args, void* argp) {
103 int cid;
104 cid = sceKernelCreateCallback("Exit Call Back", psp_exit_callback, NULL);
105 sceKernelRegisterExitCallback(cid);
106 sceKernelSleepThreadCB();
107 return 0;
108}
109#endif
110
111int main(int argc, char** argv) {
112 logfile = stderr;
113#ifdef SERVICE
114 SERVICE_TABLE_ENTRY table[] = {{"Tewi HTTPd", servmain}, {NULL, NULL}};
115 StartServiceCtrlDispatcher(table);
116#else
117#ifdef _PSP
118 pspDebugScreenInit();
119 pspDebugScreenSetXY(0, 0);
120 printf("PSP Bootstrap, Tewi/%s\n", tw_get_version());
121 int thid = sceKernelCreateThread("update_thread", psp_callback_thread, 0x11, 0xfa0, 0, NULL);
122 if(thid >= 0) {
123 sceKernelStartThread(thid, 0, NULL);
124 } else {
125 printf("Failed to start thread\n");
126 while(running) sceKernelDelayThread(50 * 1000);
127 sceKernelExitGame();
128 }
129 sceCtrlSetSamplingCycle(0);
130 sceCtrlSetSamplingMode(PSP_CTRL_MODE_ANALOG);
131 sceUtilityLoadNetModule(PSP_NET_MODULE_COMMON);
132 sceUtilityLoadNetModule(PSP_NET_MODULE_INET);
133 if(pspSdkInetInit()) {
134 printf("Could not init the network\n");
135 while(running) sceKernelDelayThread(50 * 1000);
136 sceKernelExitGame();
137 } else {
138 printf("Network initialization successful\n");
139 }
140 if(sceWlanGetSwitchState() != 1) {
141 printf("Turn the Wi-Fi switch on\n");
142 while(sceWlanGetSwitchState() != 1) {
143 sceKernelDelayThread(1000 * 1000);
144 }
145 } else {
146 printf("Wi-Fi is turned on\n");
147 }
148 int i;
149 int choice[100];
150 int incr = 0;
151 int last = 0;
152 int cur = 0;
153 for(i = 1; i < 100; i++) {
154 choice[i - 1] = 0;
155 netData name;
156 netData data;
157 if(sceUtilityCheckNetParam(i) != 0) continue;
158 choice[incr++] = i;
159 pspDebugScreenSetXY(0, 1 + 3 + incr - 1);
160 if(incr == 1) printf("> ");
161 pspDebugScreenSetXY(2, 1 + 3 + incr - 1);
162 sceUtilityGetNetParam(i, 0, &name);
163 sceUtilityGetNetParam(i, 1, &data);
164 printf("SSID=%s", data.asString);
165 sceUtilityGetNetParam(i, 4, &data);
166 if(data.asString[0]) {
167 sceUtilityGetNetParam(i, 5, &data);
168 printf(" IPADDR=%s\n", data.asString);
169 } else {
170 printf(" DHCP\n");
171 }
172 }
173 int press = 0;
174 while(1) {
175 if(!running) {
176 sceKernelExitGame();
177 }
178 SceCtrlData c;
179 sceCtrlReadBufferPositive(&c, 1);
180 press = 0;
181 if(c.Buttons & PSP_CTRL_DOWN) {
182 if(cur < incr - 1) {
183 cur++;
184 }
185 press = 1;
186 } else if(c.Buttons & PSP_CTRL_UP) {
187 if(cur > 0) {
188 cur--;
189 }
190 press = -1;
191 } else if(c.Buttons & PSP_CTRL_START) {
192 break;
193 }
194 if(last != cur) {
195 pspDebugScreenSetXY(0, 1 + 3 + last);
196 printf(" ");
197 pspDebugScreenSetXY(0, 1 + 3 + cur);
198 printf("> ");
199 last = cur;
200 }
201 if(press != 0) {
202 while(1) {
203 SceCtrlData c;
204 sceCtrlReadBufferPositive(&c, 1);
205 if(press == 1) {
206 if(!(c.Buttons & PSP_CTRL_DOWN)) break;
207 } else if(press == -1) {
208 if(!(c.Buttons & PSP_CTRL_UP)) break;
209 }
210 }
211 }
212 }
213 pspDebugScreenSetXY(0, 1 + 3 + incr + 1);
214 int err = sceNetApctlConnect(choice[cur]);
215 if(err != 0) {
216 printf("Apctl initialization failure\n");
217 while(running) sceKernelDelayThread(50 * 1000);
218 sceKernelExitGame();
219 } else {
220 printf("Apctl initialization successful\n");
221 }
222 printf("Apctl connecting\n");
223 while(1) {
224 int state;
225 err = sceNetApctlGetState(&state);
226 if(err != 0) {
227 printf("Apctl getting status failure\n");
228 while(running) sceKernelDelayThread(50 * 1000);
229 sceKernelExitGame();
230 }
231 if(state == 4) {
232 break;
233 }
234 sceKernelDelayThread(50 * 1000);
235 }
236 union SceNetApctlInfo info;
237 if(sceNetApctlGetInfo(8, &info) != 0) {
238 printf("Got an unknown IP\n");
239 while(running) sceKernelDelayThread(50 * 1000);
240 sceKernelExitGame();
241 }
242 printf("Connected, My IP is %s\n", info.ip);
243#elif defined(__PPU__)
244 printf("PS3 Bootstrap, Tewi/%s\n", tw_get_version());
245 netInitialize();
246#endif
247 int st = startup(argc, argv);
248 if(st != -1) {
249#ifdef _PSP
250 printf("Error code %d\n", st);
251 while(running) sceKernelDelayThread(50 * 1000);
252 sceKernelExitGame();
253#else
254 return st;
255#endif
256 }
257 tw_server_loop();
258#endif
259#ifdef _PSP
260 sceKernelExitGame();
261#endif
262 return 0;
263}
264
265int startup(int argc, char** argv) {
266 int i;
267 const char* confpath = PREFIX "/etc/tewi.conf";
268 if(argv != NULL) {
269 for(i = 1; i < argc; i++) {
270 if(argv[i][0] == '-') {
271 if(strcmp(argv[i], "--verbose") == 0 || strcmp(argv[i], "-v") == 0) {
272 if(!cm_do_log) {
273 cm_do_log = true;
274#ifndef NO_SSL
275 cm_log("", "This is Tewi HTTPd, version %s, using %s", tw_get_version(), OPENSSL_VERSION_TEXT);
276#else
277 cm_log("", "This is Tewi HTTPd, version %s", tw_get_version());
278#endif
279 } else {
280 cm_do_log = true;
281 }
282 } else if(strcmp(argv[i], "--config") == 0 || strcmp(argv[i], "-C") == 0) {
283 i++;
284 if(argv[i] == NULL) {
285 STDERR_LOG("Missing argument\n");
286 return 1;
287 }
288 confpath = argv[i];
289#ifndef _PSP
290 } else if(strcmp(argv[i], "--logfile") == 0 || strcmp(argv[i], "-l") == 0) {
291 i++;
292 if(argv[i] == NULL) {
293 STDERR_LOG("Missing argument\n");
294 return 1;
295 }
296 if(logfile != NULL && logfile != stderr) {
297 fclose(logfile);
298 }
299 logfile = fopen(argv[i], "a");
300 if(logfile == NULL) {
301 STDERR_LOG("Failed to open logfile\n");
302 return 1;
303 }
304#endif
305 } else if(strcmp(argv[i], "--version") == 0 || strcmp(argv[i], "-V") == 0) {
306 printf("Tewi HTTPd Tewi/%s\n", tw_get_version());
307 printf("Under public domain.\n");
308 printf("Original by 2024 Nishi\n");
309 printf("\n");
310 printf("Usage: %s [--config|-C config] [--verbose|-v] [--version|-V]\n", argv[0]);
311 printf("--config | -C config : Specify config\n");
312#ifndef _PSP
313 printf("--logfile | -l logfile : Specify logfile\n");
314#endif
315 printf("--verbose | -v : Verbose mode\n");
316 printf("--version | -V : Version information\n");
317 return 0;
318 } else {
319 STDERR_LOG("Unknown option: %s\n", argv[i]);
320 return 1;
321 }
322 }
323 }
324 }
325 tw_config_init();
326 if(tw_config_read(confpath) != 0) {
327 STDERR_LOG("Could not read the config\n");
328 return 1;
329 }
330 if(tw_server_init() != 0) {
331 STDERR_LOG("Could not initialize the server\n");
332 return 1;
333 }
334 sprintf(tw_server, "Tewi/%s (%s)%s", tw_get_version(), tw_get_platform(), config.extension == NULL ? "" : config.extension);
335 char* r = cm_strcat(tw_server, " running...");
336 cm_force_log(r);
337 free(r);
338#ifndef __MINGW32__
339 signal(SIGCHLD, SIG_IGN);
340 signal(SIGPIPE, SIG_IGN);
341#else
342 SetConsoleTitle(tw_server);
343#endif
344 return -1;
345}
Note: See TracBrowser for help on using the repository browser.