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

Last change on this file since 189 was 189, checked in by Nishi, on Sep 29, 2024 at 9:37:00 AM

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