|
|
| version 1.85, 2020/10/10 09:05:42 | version 1.90, 2020/10/12 21:55:17 |
|---|---|
| Line 18 volatile const char * IDENT_PA_HTTP_C="$ | Line 18 volatile const char * IDENT_PA_HTTP_C="$ |
| #ifdef _MSC_VER | #ifdef _MSC_VER |
| #include <windows.h> | #include <windows.h> |
| #define socklen_t int | |
| #else | #else |
| #define closesocket close | #define closesocket close |
| #endif | #endif |
| Line 130 public: | Line 131 public: |
| } | } |
| size_t first_line(){ | size_t first_line(){ |
| char *headers=strchr(buf, '\n'); | char *header=strchr(buf, '\n'); |
| if(!headers) | if(!header) |
| return false; | return false; |
| return headers-buf; | return header-buf; |
| } | } |
| const char *status_code(char *status_line, int &result){ | const char *status_code(char *status_line, int &result){ |
| Line 189 public: | Line 190 public: |
| } | } |
| } | } |
| int read_response(int sock, bool fail_on_status_ne_200); | |
| }; | }; |
| enum HTTP_response_state { | enum HTTP_response_state { |
| Line 197 enum HTTP_response_state { | Line 199 enum HTTP_response_state { |
| HTTP_BODY | HTTP_BODY |
| }; | }; |
| static int http_read_response(HTTP_response& response, int sock, bool fail_on_status_ne_200) { | int HTTP_response::read_response(int sock, bool fail_on_status_ne_200) { |
| HTTP_response_state state=HTTP_STATUS_CODE; | HTTP_response_state state=HTTP_STATUS_CODE; |
| int result=0; | int result=0; |
| size_t chunk_size=0x400*16; | size_t chunk_size=0x400*16; |
| response.resize(2*chunk_size); | resize(2*chunk_size); |
| while(response.read(sock, chunk_size)){ | while(read(sock, chunk_size)){ |
| switch(state){ | switch(state){ |
| case HTTP_STATUS_CODE: { | case HTTP_STATUS_CODE: { |
| size_t status_size=response.first_line(); | size_t status_size=first_line(); |
| if(!status_size) | if(!status_size) |
| break; | break; |
| const char *status=response.status_code(pa_strdup(response.buf, status_size), result); | const char *status=status_code(pa_strdup(buf, status_size), result); |
| if(!result || fail_on_status_ne_200 && result!=200) | if(!result || fail_on_status_ne_200 && result!=200) |
| throw Exception("http.status", status ? new String(status) : &String::Empty, "invalid HTTP response status"); | throw Exception("http.status", status ? new String(status) : &String::Empty, "invalid HTTP response status"); |
| Line 220 static int http_read_response(HTTP_respo | Line 222 static int http_read_response(HTTP_respo |
| } | } |
| case HTTP_HEADERS: { | case HTTP_HEADERS: { |
| if(!response.body_start()) | if(!body_start()) |
| break; | break; |
| response.parse_headers(); | parse_headers(); |
| size_t content_length=check_file_size(response.headers.content_length, response.url); | size_t content_length=check_file_size(headers.content_length, url); |
| if(content_length>0 && (content_length + response.body_offset) > response.length){ | if(content_length>0 && (content_length + body_offset) > length){ |
| response.resize(content_length + response.body_offset + 0x400*64); | resize(content_length + body_offset + 0x400*64); |
| } | } |
| state=HTTP_BODY; | state=HTTP_BODY; |
| Line 242 static int http_read_response(HTTP_respo | Line 244 static int http_read_response(HTTP_respo |
| } | } |
| if(state==HTTP_STATUS_CODE) | if(state==HTTP_STATUS_CODE) |
| throw Exception("http.response", &response.url, "bad response from host - no status found (size=%u)", response.length); | throw Exception("http.response", &url, "bad response from host - no status found (size=%u)", length); |
| if(state==HTTP_HEADERS){ | if(state==HTTP_HEADERS){ |
| response.parse_headers(); | parse_headers(); |
| response.body_offset=response.length; | body_offset=length; |
| } | } |
| return result; | return result; |
| Line 322 static int http_request(HTTP_response& r | Line 324 static int http_request(HTTP_response& r |
| throw Exception("http.timeout", 0, "error sending request: %s (%d)", pa_socks_strerr(no), no); | throw Exception("http.timeout", 0, "error sending request: %s (%d)", pa_socks_strerr(no), no); |
| } | } |
| result=http_read_response(response, sock, fail_on_status_ne_200); | result=response.read_response(sock, fail_on_status_ne_200); |
| closesocket(sock); | closesocket(sock); |
| #ifdef PA_USE_ALARM | #ifdef PA_USE_ALARM |
| alarm(0); | alarm(0); |
| Line 896 File_read_http_result pa_internal_file_r | Line 898 File_read_http_result pa_internal_file_r |
| } | } |
| // filling $.cookies | // filling $.cookies |
| if(Value *vcookies=vtables->hash().get("SET-COOKIE")) | if(vcookies=vtables->hash().get("SET-COOKIE")) |
| result.headers->put(HTTP_COOKIES_NAME, new VTable(parse_cookies(r, vcookies->get_table()))); | result.headers->put(HTTP_COOKIES_NAME, new VTable(parse_cookies(r, vcookies->get_table()))); |
| // output response | // output response |
| Line 941 public: | Line 943 public: |
| return str_upper(method_line, uri_start-method_line); | return str_upper(method_line, uri_start-method_line); |
| } | } |
| void read_header(int sock); | void read_header(int); |
| size_t read_post(int, char *, size_t); | |
| }; | }; |
| enum HTTPD_request_state { | enum HTTPD_request_state { |
| Line 989 void HTTPD_request::read_header(int sock | Line 992 void HTTPD_request::read_header(int sock |
| } | } |
| } | } |
| size_t HTTPD_request::read_post(int sock, char *body, size_t max_bytes) { | |
| size_t total_read = min(length - body_offset, max_bytes); | |
| memcpy(body, buf, total_read); | |
| while (total_read < max_bytes){ | |
| ssize_t received_size = recv(sock, buf + total_read, max_bytes - total_read, 0); | |
| if(received_size == 0) | |
| return total_read; | |
| if(received_size < 0) { | |
| if(int no = pa_socks_errno()) | |
| throw Exception("httpd.timeout", &url, "error receiving request body: %s (%d)", pa_socks_strerr(no), no); | |
| return total_read; | |
| } | |
| total_read += received_size; | |
| } | |
| return total_read; | |
| } | |
| /* ********************************************************** */ | /* ********************************************************** */ |
| Array<HTTP_Headers::Header> &HTTPD_Connection::headers() { | Array<HTTP_Headers::Header> &HTTPD_Connection::headers() { |
| Line 1016 void HTTPD_Connection::read_header(){ | Line 1037 void HTTPD_Connection::read_header(){ |
| request->read_header(sock); | request->read_header(sock); |
| } | } |
| size_t HTTPD_Connection::read_post(char *body, size_t max_bytes) { | |
| return request->read_post(sock, body, max_bytes); | |
| } | |
| size_t HTTPD_Connection::send_body(const void *buf, size_t size) { | |
| if(send(sock, buf, size, 0) != (ssize_t)size) { | |
| int no=pa_socks_errno(); | |
| throw Exception("httpd.timeout", 0, "error sending response: %s (%d)", pa_socks_strerr(no), no); | |
| } | |
| return size; | |
| } | |
| static int sock_on = 1; | static int sock_on = 1; |
| int HTTPD_Server::bind(const char *host, int port){ | int HTTPD_Server::bind(const char *host_port){ |
| struct sockaddr_in me; | struct sockaddr_in me; |
| if(!set_addr(&me, host, port)){ | const char *port = strchr(host_port, ':'); |
| const char *host = NULL; | |
| if(port && port > host_port){ | |
| host = pa_strdup(host_port, port - host_port); | |
| port += 1; | |
| } else { | |
| port = host_port; | |
| } | |
| if(!set_addr(&me, host, pa_atoui(port, 10))){ | |
| if (host) | if (host) |
| throw Exception("httpd.bind", 0, "can not resolve hostname \"%s\"", host); | throw Exception("httpd.bind", 0, "can not resolve hostname \"%s\"", host); |
| me.sin_addr.s_addr=INADDR_ANY; | me.sin_addr.s_addr=INADDR_ANY; |
| Line 1038 int HTTPD_Server::bind(const char *host, | Line 1080 int HTTPD_Server::bind(const char *host, |
| setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (char *)&sock_on, sizeof(sock_on)) || | setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (char *)&sock_on, sizeof(sock_on)) || |
| ::bind(sock, (struct sockaddr*)&me, sizeof(me)) || | ::bind(sock, (struct sockaddr*)&me, sizeof(me)) || |
| listen(sock, 16)) { | listen(sock, 16)) { |
| close(sock); | closesocket(sock); |
| int no = pa_socks_errno(); | int no = pa_socks_errno(); |
| throw Exception("httpd.bind", 0, "can not bind socket: %s (%d)", pa_socks_strerr(no), no); | throw Exception("httpd.bind", 0, "can not bind socket: %s (%d)", pa_socks_strerr(no), no); |
| } | } |
| return sock; | return sock; |
| } | } |
| int ready(int fd,int operation,int timeout_value){ | static int ready(int fd,int operation,int timeout_value){ |
| struct timeval timeout = {0, timeout_value * 1000}; | struct timeval timeout = {0, timeout_value * 1000}; |
| fd_set fds; | fd_set fds; |
| FD_ZERO(&fds); | FD_ZERO(&fds); |
| Line 1057 int ready(int fd,int operation,int timeo | Line 1099 int ready(int fd,int operation,int timeo |
| } | } |
| } | } |
| HTTPD_Connection *HTTPD_Server::accept(int sock, int timeout_value) { | HTTPD_Connection *HTTPD_Server::accept(int sock, int timeout_value) { |
| int ready = ::ready(sock, 0, timeout_value); | int ready = ::ready(sock, 0, timeout_value); |
| if (ready < 0) { | if (ready < 0) { |
| int no=pa_socks_errno(); | int no=pa_socks_errno(); |
| Line 1073 HTTPD_Connection *HTTPD_Server::accept(i | Line 1113 HTTPD_Connection *HTTPD_Server::accept(i |
| } | } |
| struct sockaddr_in addr; | struct sockaddr_in addr; |
| unsigned int sock_addr_len = sizeof(struct sockaddr_in); | socklen_t sock_addr_len = sizeof(struct sockaddr_in); |
| memset(&addr, 0, sock_addr_len); | memset(&addr, 0, sock_addr_len); |
| int csock = ::accept(sock, (struct sockaddr *)&addr, &sock_addr_len); | int csock = ::accept(sock, (struct sockaddr *)&addr, &sock_addr_len); |