fumiama/simple-http-server
1
0
Fork 0
mirror of https://github.com/fumiama/simple-http-server.git synced 2026-06-05 00:30:23 +08:00
Code Activity

fix: hostname check

Browse Source
This commit is contained in:
源文雨
2023-03-11 11:54:10 +08:00
parent 9044c52cee
commit 5953fde6be
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server.c
View File

@@ -156,19 +156,23 @@ static void accept_request(void *cli) {
} }
} }
int content_length = 0; int content_length = 0;
int host_chk_passed = !(uintptr_t)hostnameport;
cgi &= ((st.st_mode & S_IXUSR) || (st.st_mode & S_IXGRP) || (st.st_mode & S_IXOTH)); cgi &= ((st.st_mode & S_IXUSR) || (st.st_mode & S_IXGRP) || (st.st_mode & S_IXOTH));
while((numchars > 0) && strcmp("\n", buf)) { while((numchars > 0) && strcmp("\n", buf)) {
numchars = get_line(client, buf, sizeof(buf)); numchars = get_line(client, buf, sizeof(buf));
if(!content_length && !strncasecmp(buf, "Content-Length: ", 16)) { if(!content_length && !strncasecmp(buf, "Content-Length: ", 16)) {
content_length = atoi(buf + 16); content_length = atoi(buf + 16);
} }
else if(hostnameport && !strncasecmp(buf, "Host: ", 6)) { else if(!host_chk_passed && !strncasecmp(buf, "Host: ", 6)) {
if(strncasecmp(buf+6, hostnameport, strlen(hostnameport))) { if(strncasecmp(buf+6, hostnameport, strlen(hostnameport))) {
forbidden(client); forbidden(client);
goto DISCARD_AND_CLOSE; host_chk_passed = 0;
break;
} }
host_chk_passed = 1;
} }
} }
if(!host_chk_passed) break;
if(method_type == POST && content_length == -1) bad_request(client); if(method_type == POST && content_length == -1) bad_request(client);
else if(!cgi) serve_file(client, path); else if(!cgi) serve_file(client, path);
else { else {