|
|
| version 1.186, 2004/04/06 14:18:00 | version 1.204.6.6, 2005/11/22 11:38:44 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: commonly functions. | Parser: commonly functions. |
| Copyright(c) 2001-2004 ArtLebedev Group (http://www.artlebedev.com) | Copyright(c) 2001-2005 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) |
| * BASE64 part | |
| * Authors: Michael Zucchi <notzed@ximian.com> | |
| * Jeffrey Stedfast <fejj@ximian.com> | |
| * | |
| * Copyright 2000-2004 Ximian, Inc. (www.ximian.com) | |
| * | |
| * This program is free software; you can redistribute it and/or modify | |
| * it under the terms of the GNU General Public License as published by | |
| * the Free Software Foundation; either version 2 of the License, or | |
| * (at your option) any later version. | |
| * | |
| * This program is distributed in the hope that it will be useful, | |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| * GNU General Public License for more details. | |
| * | |
| * You should have received a copy of the GNU General Public License | |
| * along with this program; if not, write to the Free Software | |
| * Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA. | |
| * | |
| */ | */ |
| static const char * const IDENT_COMMON_C="$Date$"; | static const char * const IDENT_COMMON_C="$Date$"; |
| Line 21 static const char * const IDENT_COMMON_C | Line 42 static const char * const IDENT_COMMON_C |
| #include "pa_vint.h" | #include "pa_vint.h" |
| #include "pa_vhash.h" | #include "pa_vhash.h" |
| #include "pa_vtable.h" | #include "pa_vtable.h" |
| #include "pa_socks.h" | |
| #ifdef CYGWIN | #ifdef CYGWIN |
| #define _GNU_H_WINDOWS32_SOCKETS | #define _GNU_H_WINDOWS32_SOCKETS |
| Line 130 char* file_read_text(Request_charsets& c | Line 152 char* file_read_text(Request_charsets& c |
| return file.success?file.str:0; | return file.success?file.str:0; |
| } | } |
| /// these options were handled but not checked elsewhere, now check them | |
| static int get_valid_file_options_count(HashStringValue& options) | |
| { | |
| int result=0; | |
| if(options.get(PA_SQL_LIMIT_NAME)) | |
| result++; | |
| if(options.get(PA_SQL_OFFSET_NAME)) | |
| result++; | |
| if(options.get(PA_COLUMN_SEPARATOR_NAME)) | |
| result++; | |
| if(options.get(PA_COLUMN_ENCLOSER_NAME)) | |
| result++; | |
| return result; | |
| } | |
| //http request stuff | //http request stuff |
| #ifdef PA_HTTP | #ifdef PA_HTTP |
| Line 156 static bool set_addr(struct sockaddr_in | Line 193 static bool set_addr(struct sockaddr_in |
| return true; | return true; |
| } | } |
| static int http_read_response(char*& response, size_t& response_size, int sock, bool fail_on_status_ne_200){ | size_t guess_content_length(char* buf) { |
| response=(char*)pa_malloc_atomic(1/*terminator*/); // setting memory block type | char* ptr; |
| response[(response_size=0)]=0; | if((ptr=strstr(buf, "Content-Length:"))) // Apache |
| goto found; | |
| if((ptr=strstr(buf, "content-length:"))) // Parser 3 | |
| goto found; | |
| if((ptr=strstr(buf, "Content-length:"))) // maybe 1 | |
| goto found; | |
| if((ptr=strstr(buf, "CONTENT-LENGTH:"))) // maybe 2 | |
| goto found; | |
| return 0; | |
| found: | |
| char *error_pos; | |
| size_t result=(size_t)strtol(ptr+15/*strlen("CONTENT-LENGTH:")*/, &error_pos, 0); | |
| const size_t reasonable_initial_max=0x400*0x400*10 /*10M*/; | |
| if(result>reasonable_initial_max) // sanity check | |
| return reasonable_initial_max; | |
| return 0;//result; | |
| } | |
| static int http_read_response(char*& response, size_t& response_size, int sock, bool fail_on_status_ne_200) { | |
| int result=0; | int result=0; |
| char* EOLat=0; | // fetching some to local buffer, guessing on possible content-length |
| while(true) { | response_size=0x400*20; // initial size if content-length could not be determined |
| char buf[MAX_STRING*10]; | const size_t preview_size=0x400*20; |
| ssize_t received_size=recv(sock, buf, sizeof(buf), 0); | char preview_buf[preview_size+1/*terminator*/]; // 20K buffer to preview headers |
| if(received_size<=0) | ssize_t received_size=recv(sock, preview_buf, preview_size, 0); |
| break; | if(received_size==0) |
| response=(char*)pa_realloc(response, response_size+received_size+1/*terminator*/); | goto done; |
| memcpy(response+response_size, buf, received_size); | if(received_size<0) { |
| response_size+=received_size; | if(int no=pa_socks_errno()) |
| response[response_size]=0; | throw Exception("http.timeout", |
| 0, | |
| "error receiving response header: %s (%d)", pa_socks_strerr(no), no); | |
| goto done; | |
| } | |
| // detecting response_size | |
| { | |
| preview_buf[received_size]=0; // terminator | |
| if(size_t content_length=guess_content_length(preview_buf)) | |
| response_size=preview_size+content_length; // a little more than needed, will adjust response_size by actual received size later | |
| } | |
| if(!result && (EOLat=strstr(response, CRLF))) { // checking status in first response | // [gcc is happier this way, see goto above] |
| const String status_line(pa_strdup(response, EOLat-response)); | { |
| ArrayString astatus; | // allocating initial buf |
| size_t pos_after=0; | response=(char*)pa_malloc_atomic(response_size+1/*terminator*/); // just setting memory block type |
| status_line.split(astatus, pos_after, " "); | char* ptr=response; |
| const String& status_code=*astatus.get(astatus.count()>1?1:0); | size_t todo_size=response_size; |
| result=status_code.as_int(); | // coping part of already received body |
| memcpy(ptr, preview_buf, received_size); | |
| if(fail_on_status_ne_200 && result!=200) | ptr+=received_size; |
| throw Exception("http.status", | todo_size-=received_size; |
| &status_code, | |
| "invalid HTTP response status"); | // we use terminator byte for two purposes here: |
| // 1. we return there zero always, not knowing: maybe they would want to create String form $file.body? | |
| // invariant: all Strings should have zero-terminated buffers | |
| // 2. we use that out-of-size byte to detect if our content-length guess was wrong | |
| // when recv gets more than we expected | |
| // a) we know that the content-length guess was wrong | |
| // b) we have space to put the first byte of extra data | |
| // c) we use less code to detect normal situation: on last while-cycle recv expected to just return 0 | |
| while(true) { | |
| received_size=recv(sock, ptr, todo_size+1/*there is always a place for terminator*/, 0); | |
| if(received_size==0) { | |
| response_size-=todo_size; // in case we received less than expected, cut down the reported size | |
| break; | |
| } | |
| if(received_size<0) { | |
| if(int no=pa_socks_errno()) | |
| throw Exception("http.timeout", | |
| 0, | |
| "error receiving response body: %s (%d)", pa_socks_strerr(no), no); | |
| break; | |
| } | |
| // they've touched the terminator? | |
| if((size_t)received_size>todo_size) | |
| { | |
| // that means that our guessed response_size was not big enough | |
| const size_t grow_chunk_size=0x400*0x400; // 1M | |
| response_size+=grow_chunk_size; | |
| size_t ptr_offset=ptr-response; | |
| response=(char*)pa_realloc(response, response_size+1/*terminator*/); | |
| ptr=response+ptr_offset; | |
| todo_size+=grow_chunk_size; | |
| } | |
| // can't do this before realloc: we need <todo_size check | |
| ptr+=received_size; | |
| todo_size-=received_size; | |
| char* EOLat=0; | |
| if(!result && (EOLat=strstr(response, "\n"))) { // checking status in first response | |
| const String status_line(pa_strdup(response, EOLat-response)); | |
| ArrayString astatus; | |
| size_t pos_after=0; | |
| status_line.split(astatus, pos_after, " "); | |
| const String& status_code=*astatus.get(astatus.count()>1?1:0); | |
| result=status_code.as_int(); | |
| if(fail_on_status_ne_200 && result!=200) | |
| throw Exception("http.status", | |
| &status_code, | |
| "invalid HTTP response status"); | |
| } | |
| } | } |
| } | } |
| done: | |
| if(result) | if(result) |
| { | |
| response[response_size]=0; | |
| return result; | return result; |
| } | |
| else | else |
| throw Exception("http.response", | throw Exception("http.response", |
| 0, | 0, |
| Line 201 static int http_read_response(char*& res | Line 320 static int http_read_response(char*& res |
| #ifdef PA_USE_ALARM | #ifdef PA_USE_ALARM |
| static sigjmp_buf timeout_env; | static sigjmp_buf timeout_env; |
| static void timeout_handler(int sig){ | static void timeout_handler(int /*sig*/){ |
| siglongjmp(timeout_env, 1); | siglongjmp(timeout_env, 1); |
| } | } |
| #endif | #endif |
| static int http_request(char*& response, size_t& response_size, | static int http_request(char*& response, size_t& response_size, |
| const char* host, int port, | const char* host, short port, |
| const char* request, | const char* request, |
| int | int timeout_secs, |
| #ifdef PA_USE_ALARM | |
| timeout | |
| #endif | |
| , | |
| bool fail_on_status_ne_200) { | bool fail_on_status_ne_200) { |
| if(!host) | if(!host) |
| throw Exception("http.host", | throw Exception("http.host", |
| 0, | 0, |
| "zero hostname"); //never | "zero hostname"); //never |
| volatile // to prevent makeing it register variable, because it will be clobbered by longjmp [thanks gcc warning] | |
| int sock=-1; | |
| #ifdef PA_USE_ALARM | #ifdef PA_USE_ALARM |
| signal(SIGALRM, timeout_handler); | signal(SIGALRM, timeout_handler); |
| #endif | #endif |
| int sock=-1; | |
| #ifdef PA_USE_ALARM | #ifdef PA_USE_ALARM |
| if(sigsetjmp(timeout_env, 1)) { | if(sigsetjmp(timeout_env, 1)) { |
| // stupid gcc [2.95.4] generated bad code | // stupid gcc [2.95.4] generated bad code |
| // which failed to handle sigsetjmp+throw: crashed inside of pre-throw code. | // which failed to handle sigsetjmp+throw: crashed inside of pre-throw code. |
| // rewritten simplier [though duplicating closesocket code] | // rewritten simplier [athough duplicating closesocket code] |
| if(sock>=0) | if(sock>=0) |
| closesocket(sock); | closesocket(sock); |
| throw Exception("http.timeout", | throw Exception("http.timeout", |
| origin_string, | 0, |
| "timeout occured while retrieving document"); | "timeout occured while retrieving document"); |
| return 0; // never | return 0; // never |
| } else { | } else { |
| alarm(timeout); | alarm(timeout_secs); |
| #endif | #endif |
| try { | try { |
| int result; | int result; |
| Line 247 static int http_request(char*& response, | Line 363 static int http_request(char*& response, |
| 0, | 0, |
| "can not resolve hostname \"%s\"", host); | "can not resolve hostname \"%s\"", host); |
| if((sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/))<0) | if((sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/))<0) { |
| int no=pa_socks_errno(); | |
| throw Exception("http.connect", | throw Exception("http.connect", |
| 0, | 0, |
| "can not make socket: %s (%d)", strerror(errno), errno); | "can not make socket: %s (%d)", pa_socks_strerr(no), no); |
| if(connect(sock, (struct sockaddr *)&dest, sizeof(dest))) | } |
| // To enable SO_DONTLINGER (that is, disable SO_LINGER) | |
| // l_onoff should be set to zero and setsockopt should be called | |
| linger dont_linger={0,0}; | |
| setsockopt(sock, SOL_SOCKET, SO_LINGER, (const char *)&dont_linger, sizeof(dont_linger)); | |
| #ifdef WIN32 | |
| // SO_*TIMEO can be defined in .h but not implemlemented in protocol, | |
| // failing subsequently with Option not supported by protocol (99) message | |
| // could not suppress that, so leaving this only for win32 | |
| int timeout_ms=timeout_secs*1000; | |
| setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, (const char*)&timeout_ms, sizeof(timeout_ms)); | |
| setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout_ms, sizeof(timeout_ms)); | |
| #endif | |
| if(connect(sock, (struct sockaddr *)&dest, sizeof(dest))) { | |
| int no=pa_socks_errno(); | |
| throw Exception("http.connect", | throw Exception("http.connect", |
| 0, | 0, |
| "can not connect to host \"%s\": %s (%d)", host, strerror(errno), errno); | "can not connect to host \"%s\": %s (%d)", host, pa_socks_strerr(no), no); |
| } | |
| size_t request_size=strlen(request); | size_t request_size=strlen(request); |
| if(send(sock, request, request_size, 0)!=(ssize_t)request_size) | if(send(sock, request, request_size, 0)!=(ssize_t)request_size) { |
| throw Exception("http.connect", | int no=pa_socks_errno(); |
| throw Exception("http.timeout", | |
| 0, | 0, |
| "error sending request: %s (%d)", strerror(errno), errno); | "error sending request: %s (%d)", pa_socks_strerr(no), no); |
| } | |
| result=http_read_response(response, response_size, sock, fail_on_status_ne_200); | result=http_read_response(response, response_size, sock, fail_on_status_ne_200); |
| closesocket(sock); | closesocket(sock); |
| Line 342 static const String* basic_authorization | Line 479 static const String* basic_authorization |
| if(pass) | if(pass) |
| combined<<pass; | combined<<pass; |
| String* result=new String("Basic "); *result<<pa_base64(combined.cstr(), combined.length()); | String* result=new String("Basic "); *result<<pa_base64_encode(combined.cstr(), combined.length()); |
| return result; | return result; |
| } | } |
| Line 393 struct File_read_http_result { | Line 530 struct File_read_http_result { |
| HashStringValue* headers; | HashStringValue* headers; |
| }; | }; |
| #endif | #endif |
| static void find_headers_end(char* p, | |
| char*& headers_end_at, | |
| char*& raw_body) | |
| { | |
| raw_body=p; | |
| // \n\n | |
| // \r\n\r\n | |
| while((p=strchr(p, '\n'))) { | |
| headers_end_at=++p; // \n>.< | |
| if(*p=='\r') // \r\n>\r?<\n | |
| p++; | |
| if(*p=='\n') { // \r\n\r>\n?< | |
| raw_body=p+1; | |
| return; | |
| } | |
| } | |
| headers_end_at=0; | |
| } | |
| /// @todo build .cookies field. use ^file.tables.SET-COOKIES.menu{ for now | /// @todo build .cookies field. use ^file.tables.SET-COOKIES.menu{ for now |
| static File_read_http_result file_read_http(Request_charsets& charsets, | static File_read_http_result file_read_http(Request_charsets& charsets, |
| const String& file_spec, | const String& file_spec, |
| Line 401 static File_read_http_result file_read_h | Line 557 static File_read_http_result file_read_h |
| File_read_http_result result; | File_read_http_result result; |
| char host[MAX_STRING]; | char host[MAX_STRING]; |
| const char* uri; | const char* uri; |
| int port; | short port; |
| const char* method="GET"; bool method_is_get; | const char* method="GET"; bool method_is_get; |
| HashStringValue* form=0; | HashStringValue* form=0; |
| const char* body_cstr=0; | const char* body_cstr=0; |
| int timeout=2; | int timeout_secs=2; |
| bool fail_on_status_ne_200=true; | bool fail_on_status_ne_200=true; |
| Value* vheaders=0; | Value* vheaders=0; |
| Charset *asked_remote_charset=0; | Charset *asked_remote_charset=0; |
| Line 413 static File_read_http_result file_read_h | Line 569 static File_read_http_result file_read_h |
| const char* password_cstr=0; | const char* password_cstr=0; |
| if(options) { | if(options) { |
| int valid_options=0; | int valid_options=get_valid_file_options_count(*options); |
| if(Value* vmethod=options->get(HTTP_METHOD_NAME)) { | if(Value* vmethod=options->get(HTTP_METHOD_NAME)) { |
| valid_options++; | valid_options++; |
| method=vmethod->as_string().cstr(); | method=vmethod->as_string().cstr(); |
| Line 428 static File_read_http_result file_read_h | Line 585 static File_read_http_result file_read_h |
| } | } |
| if(Value* vtimeout=options->get(HTTP_TIMEOUT_NAME)) { | if(Value* vtimeout=options->get(HTTP_TIMEOUT_NAME)) { |
| valid_options++; | valid_options++; |
| timeout=vtimeout->as_int(); | timeout_secs=vtimeout->as_int(); |
| } | } |
| if((vheaders=options->get(HTTP_HEADERS_NAME))) { | if((vheaders=options->get(HTTP_HEADERS_NAME))) { |
| valid_options++; | valid_options++; |
| Line 489 static File_read_http_result file_read_h | Line 646 static File_read_http_result file_read_h |
| uri=host_uri?current+(host_uri-1-host):"/"; | uri=host_uri?current+(host_uri-1-host):"/"; |
| char* port_cstr=lsplit(host, ':'); | char* port_cstr=lsplit(host, ':'); |
| char* error_pos=0; | char* error_pos=0; |
| port=port_cstr?strtol(port_cstr, &error_pos, 0):80; | port=port_cstr?(short)strtol(port_cstr, &error_pos, 0):80; |
| if(strchr(uri, '?') && form) | if(strchr(uri, '?') && form) |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| Line 560 static File_read_http_result file_read_h | Line 717 static File_read_http_result file_read_h |
| size_t response_size; | size_t response_size; |
| int status_code=http_request(response, response_size, | int status_code=http_request(response, response_size, |
| host, port, request_head_and_body.cstr(), | host, port, request_head_and_body.cstr(), |
| timeout, fail_on_status_ne_200); | timeout_secs, fail_on_status_ne_200); |
| //processing results | //processing results |
| char* raw_body; size_t raw_body_size; | char* raw_body; size_t raw_body_size; |
| char* headers_end_at=strstr(response, CRLF CRLF /*change '4' below along!*/); | char* headers_end_at; |
| if(headers_end_at) { | find_headers_end(response, |
| raw_body=headers_end_at+4; | headers_end_at, |
| raw_body_size=response_size-(raw_body-response); | raw_body); |
| } else | raw_body_size=response_size-(raw_body-response); |
| throw Exception("http.response", | |
| &connect_string, | |
| "bad response from host - no headers found"); | |
| *headers_end_at=0; | |
| const String header_block(response, headers_end_at-response, true); | |
| ArrayString aheaders; | |
| result.headers=new HashStringValue; | result.headers=new HashStringValue; |
| VHash* vtables=new VHash; | VHash* vtables=new VHash; |
| result.headers->put(HTTP_TABLES_NAME, vtables); | result.headers->put(HTTP_TABLES_NAME, vtables); |
| HashStringValue& tables=vtables->hash(); | |
| size_t pos_after=0; | |
| header_block.split(aheaders, pos_after, CRLF); | |
| //processing headers | |
| size_t aheaders_count=aheaders.count(); | |
| Charset* real_remote_charset=0; // undetected, yet | Charset* real_remote_charset=0; // undetected, yet |
| for(size_t i=1; i<aheaders_count; i++) { | |
| const String& line=*aheaders.get(i); | if(headers_end_at) { |
| size_t pos=line.pos(": ", 2); | *headers_end_at=0; |
| if(pos==STRING_NOT_FOUND || pos<1) | const String header_block(String::C(response, headers_end_at-response), true); |
| throw Exception("http.response", | |
| &connect_string, | ArrayString aheaders; |
| "bad response from host - bad header \"%s\"", line.cstr()); | HashStringValue& tables=vtables->hash(); |
| const String::Body HEADER_NAME= | |
| line.mid(0, pos).change_case(charsets.source(), String::CC_UPPER); | size_t pos_after=0; |
| const String& header_value=line.mid(pos+2, line.length()); | header_block.split(aheaders, pos_after, "\n"); |
| if(HEADER_NAME=="CONTENT-TYPE") | |
| real_remote_charset=detect_charset(charsets.source(), header_value); | //processing headers |
| size_t aheaders_count=aheaders.count(); | |
| // tables | for(size_t i=1; i<aheaders_count; i++) { |
| { | const String& line=*aheaders.get(i); |
| Value *valready=(Value *)tables.get(HEADER_NAME); | size_t pos=line.pos(':'); |
| bool existed=valready!=0; | if(pos==STRING_NOT_FOUND || pos<1) |
| Table *table; | throw Exception("http.response", |
| if(existed) { | &connect_string, |
| // second+ appearence | "bad response from host - bad header \"%s\"", line.cstr()); |
| table=valready->get_table(); | const String::Body HEADER_NAME= |
| } else { | line.mid(0, pos).change_case(charsets.source(), String::CC_UPPER); |
| // first appearence | const String& header_value=line.mid(pos+1, line.length()).trim(String::TRIM_BOTH, " \t\r"); |
| Table::columns_type columns =new ArrayString(1); | if(as_text && HEADER_NAME=="CONTENT-TYPE") |
| *columns+=new String("value"); | real_remote_charset=detect_charset(charsets.source(), header_value); |
| table=new Table(columns); | |
| // tables | |
| { | |
| Value *valready=(Value *)tables.get(HEADER_NAME); | |
| bool existed=valready!=0; | |
| Table *table; | |
| if(existed) { | |
| // second+ appearence | |
| table=valready->get_table(); | |
| } else { | |
| // first appearence | |
| Table::columns_type columns =new ArrayString(1); | |
| *columns+=new String("value"); | |
| table=new Table(columns); | |
| } | |
| // this string becomes next row | |
| ArrayString& row=*new ArrayString(1); | |
| row+=&header_value; | |
| *table+=&row; | |
| // not existed before? add it | |
| if(!existed) | |
| tables.put(HEADER_NAME, new VTable(table)); | |
| } | } |
| // this string becomes next row | |
| ArrayString& row=*new ArrayString(1); | |
| row+=&header_value; | |
| *table+=&row; | |
| // not existed before? add it | |
| if(!existed) | |
| tables.put(HEADER_NAME, new VTable(table)); | |
| } | |
| result.headers->put(HEADER_NAME, new VString(header_value)); | result.headers->put(HEADER_NAME, new VString(header_value)); |
| } | } |
| // defaulting to used-asked charset [it's never empty!] | } |
| if(!real_remote_charset) | |
| real_remote_charset=asked_remote_charset; | |
| // output response | // output response |
| String::C real_body=String::C(raw_body, raw_body_size); | String::C real_body=String::C(raw_body, raw_body_size); |
| if(as_text && raw_body_size) // must be checked because transcode returns CONST string in case length==0, which contradicts hacking few lines below | if(as_text && raw_body_size) { // must be checked because transcode returns CONST string in case length==0, which contradicts hacking few lines below |
| // defaulting to used-asked charset [it's never empty!] | |
| if(!real_remote_charset) | |
| real_remote_charset=asked_remote_charset; | |
| real_body=Charset::transcode(real_body, *real_remote_charset, charsets.source()); | real_body=Charset::transcode(real_body, *real_remote_charset, charsets.source()); |
| } | |
| result.str=const_cast<char *>(real_body.str); // hacking a little | result.str=const_cast<char *>(real_body.str); // hacking a little |
| result.length=real_body.length; | result.length=real_body.length; |
| Line 646 static File_read_http_result file_read_h | Line 804 static File_read_http_result file_read_h |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| struct File_read_action_info { | struct File_read_action_info { |
| char **data; size_t *data_size; | char **data; size_t *data_size; |
| char* buf; size_t offset; size_t count; | |
| }; | }; |
| #endif | #endif |
| static void file_read_action( | static void file_read_action( |
| Line 654 static void file_read_action( | Line 813 static void file_read_action( |
| const String& file_spec, const char* /*fname*/, bool as_text, | const String& file_spec, const char* /*fname*/, bool as_text, |
| void *context) { | void *context) { |
| File_read_action_info& info=*static_cast<File_read_action_info *>(context); | File_read_action_info& info=*static_cast<File_read_action_info *>(context); |
| if(size_t to_read_size=(size_t)finfo.st_size) { | size_t to_read_size=info.count; |
| *info.data=new(PointerFreeGC) char[to_read_size+(as_text?1:0)]; | if(!to_read_size) |
| to_read_size=(size_t)finfo.st_size; | |
| assert( !(info.buf && as_text) ); | |
| if(to_read_size) { | |
| if(info.offset) | |
| lseek(f, info.offset, SEEK_SET); | |
| *info.data=info.buf | |
| ? info.buf | |
| : new(PointerFreeGC) char[to_read_size+(as_text?1:0)]; | |
| *info.data_size=(size_t)read(f, *info.data, to_read_size); | *info.data_size=(size_t)read(f, *info.data, to_read_size); |
| if(ssize_t(*info.data_size)<0 || *info.data_size>to_read_size) | if(ssize_t(*info.data_size)<0 || *info.data_size>to_read_size) |
| Line 664 static void file_read_action( | Line 831 static void file_read_action( |
| "read failed: actually read %u bytes count not in [0..%u] valid range", | "read failed: actually read %u bytes count not in [0..%u] valid range", |
| *info.data_size, to_read_size); | *info.data_size, to_read_size); |
| } else { // empty file | } else { // empty file |
| if(as_text) { | // for both, text and binary: for text we need that terminator, for binary we need nonzero pointer to be able to save such files |
| *info.data=new(PointerFreeGC) char[1]; | *info.data=new(PointerFreeGC) char[1]; |
| *(char*)(*info.data)=0; | *(char*)(*info.data)=0; |
| } else | |
| *info.data=0; | |
| *info.data_size=0; | *info.data_size=0; |
| return; | return; |
| } | } |
| } | } |
| File_read_result file_read(Request_charsets& charsets, const String& file_spec, | File_read_result file_read(Request_charsets& charsets, const String& file_spec, |
| bool as_text, HashStringValue *params, | bool as_text, HashStringValue *params, |
| bool fail_on_read_problem) { | bool fail_on_read_problem, |
| char* buf, size_t offset, size_t count) { | |
| File_read_result result={false, 0, 0, 0}; | File_read_result result={false, 0, 0, 0}; |
| #ifdef PA_HTTP | #ifdef PA_HTTP |
| if(file_spec.starts_with("http://")) { | if(file_spec.starts_with("http://")) { |
| if(offset || count) | |
| throw Exception("parser.runtime", | |
| 0, | |
| "offset and load options are not supported for HTTP:// file load"); | |
| // fail on read problem | // fail on read problem |
| File_read_http_result http=file_read_http(charsets, file_spec, as_text, params); | File_read_http_result http=file_read_http(charsets, file_spec, as_text, params); |
| result.success=true; | result.success=true; |
| Line 687 File_read_result file_read(Request_chars | Line 858 File_read_result file_read(Request_chars |
| result.headers=http.headers; | result.headers=http.headers; |
| } else { | } else { |
| #endif | #endif |
| if(params && params->count()) | if(params) { |
| throw Exception("parser.runtime", | int valid_options=get_valid_file_options_count(*params); |
| 0, | if(valid_options!=params->count()) |
| "invalid option passed"); | throw Exception("parser.runtime", |
| 0, | |
| "invalid option passed"); | |
| } | |
| File_read_action_info info={&result.str, &result.length}; | File_read_action_info info={&result.str, &result.length, |
| buf, offset, count}; | |
| result.success=file_read_action_under_lock(file_spec, | result.success=file_read_action_under_lock(file_spec, |
| "read", file_read_action, &info, | "read", file_read_action, &info, |
| as_text, fail_on_read_problem); | as_text, fail_on_read_problem); |
| Line 787 bool file_read_action_under_lock(const S | Line 962 bool file_read_action_under_lock(const S |
| } | } |
| } | } |
| static void create_dir_for_file(const String& file_spec) { | void create_dir_for_file(const String& file_spec) { |
| size_t pos_after=1; | size_t pos_after=1; |
| size_t pos_before; | size_t pos_before; |
| while((pos_before=file_spec.pos('/', pos_after))!=STRING_NOT_FOUND) { | while((pos_before=file_spec.pos('/', pos_after))!=STRING_NOT_FOUND) { |
| Line 1047 const char* format(double value, char* f | Line 1222 const char* format(double value, char* f |
| size_t stdout_write(const void *buf, size_t size) { | size_t stdout_write(const void *buf, size_t size) { |
| #ifdef WIN32 | #ifdef WIN32 |
| size_t to_write = size; | |
| do{ | do{ |
| int chunk_written=fwrite(buf, 1, min((size_t)8*0x400, size), stdout); | int chunk_written=fwrite(buf, 1, min((size_t)8*0x400, size), stdout); |
| if(chunk_written<=0) | if(chunk_written<=0) |
| Line 1055 size_t stdout_write(const void *buf, siz | Line 1231 size_t stdout_write(const void *buf, siz |
| buf=((const char*)buf)+chunk_written; | buf=((const char*)buf)+chunk_written; |
| } while(size>0); | } while(size>0); |
| return size; | return to_write-size; |
| #else | #else |
| return fwrite(buf, 1, size, stdout); | return fwrite(buf, 1, size, stdout); |
| #endif | #endif |
| Line 1068 char* unescape_chars(const char* cp, int | Line 1244 char* unescape_chars(const char* cp, int |
| EscapeFirst, | EscapeFirst, |
| EscapeSecond | EscapeSecond |
| } escapeState=EscapeRest; | } escapeState=EscapeRest; |
| int escapedValue=0; | uchar escapedValue=0; |
| int srcPos=0; | int srcPos=0; |
| int dstPos=0; | int dstPos=0; |
| while(srcPos < len) { | while(srcPos < len) { |
| int ch=cp[srcPos]; | uchar ch=(uchar)cp[srcPos]; |
| switch(escapeState) { | switch(escapeState) { |
| case EscapeRest: | case EscapeRest: |
| if(ch=='%') { | if(ch=='%') { |
| Line 1084 char* unescape_chars(const char* cp, int | Line 1260 char* unescape_chars(const char* cp, int |
| } | } |
| break; | break; |
| case EscapeFirst: | case EscapeFirst: |
| escapedValue=hex_value[ch] << 4; | escapedValue=(uchar)(hex_value[ch] << 4); |
| escapeState=EscapeSecond; | escapeState=EscapeSecond; |
| break; | break; |
| case EscapeSecond: | case EscapeSecond: |
| Line 1125 bool StrEqNc(const char* s1, const char* | Line 1301 bool StrEqNc(const char* s1, const char* |
| return !strict; | return !strict; |
| } else if(!(*s2)) | } else if(!(*s2)) |
| return !strict; | return !strict; |
| if(isalpha(*s1)) { | if(isalpha((unsigned char)*s1)) { |
| if(tolower(*s1) !=tolower(*s2)) | if(tolower((unsigned char)*s1) !=tolower((unsigned char)*s2)) |
| return false; | return false; |
| } else if((*s1) !=(*s2)) | } else if((*s1) !=(*s2)) |
| return false; | return false; |
| Line 1383 g_mime_utils_base64_encode_close (const | Line 1559 g_mime_utils_base64_encode_close (const |
| return (outptr - out); | return (outptr - out); |
| } | } |
| char* pa_base64(const char *in, size_t len) | static unsigned char gmime_base64_rank[256] = { |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255, 62,255,255,255, 63, | |
| 52, 53, 54, 55, 56, 57, 58, 59, 60, 61,255,255,255, 0,255,255, | |
| 255, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, | |
| 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,255,255,255,255,255, | |
| 255, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, | |
| 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255, | |
| }; | |
| /** | |
| * g_mime_utils_base64_decode_step: | |
| * @in: input stream | |
| * @inlen: max length of data to decode | |
| * @out: output stream | |
| * @state: holds the number of bits that are stored in @save | |
| * @save: leftover bits that have not yet been decoded | |
| * | |
| * Decodes a chunk of base64 encoded data. | |
| * | |
| * Returns the number of bytes decoded (which have been dumped in @out). | |
| **/ | |
| size_t | |
| g_mime_utils_base64_decode_step (const unsigned char *in, size_t inlen, unsigned char *out, int *state, int *save) | |
| { | |
| const register unsigned char *inptr; | |
| register unsigned char *outptr; | |
| const unsigned char *inend; | |
| register guint32 saved; | |
| unsigned char c; | |
| int i; | |
| inend = in + inlen; | |
| outptr = out; | |
| /* convert 4 base64 bytes to 3 normal bytes */ | |
| saved = *save; | |
| i = *state; | |
| inptr = in; | |
| while (inptr < inend) { | |
| c = gmime_base64_rank[*inptr++]; | |
| if (c != 0xff) { | |
| saved = (saved << 6) | c; | |
| i++; | |
| if (i == 4) { | |
| *outptr++ = saved >> 16; | |
| *outptr++ = saved >> 8; | |
| *outptr++ = saved; | |
| i = 0; | |
| } | |
| } | |
| } | |
| *save = saved; | |
| *state = i; | |
| /* quick scan back for '=' on the end somewhere */ | |
| /* fortunately we can drop 1 output char for each trailing = (upto 2) */ | |
| i = 2; | |
| while (inptr > in && i) { | |
| inptr--; | |
| if (gmime_base64_rank[*inptr] != 0xff) { | |
| if (*inptr == '=' && outptr > out) | |
| outptr--; | |
| i--; | |
| } | |
| } | |
| /* if i != 0 then there is a truncation error! */ | |
| return (outptr - out); | |
| } | |
| char* pa_base64_encode(const char *in, size_t in_size) | |
| { | { |
| /* wont go to more than 2x size (overly conservative) */ | /* wont go to more than 2x size (overly conservative) */ |
| char* result=new(PointerFreeGC) char[len * 2 + 6]; | char* result=new(PointerFreeGC) char[in_size * 2 + 6]; |
| int state=0; | int state=0; |
| int save=0; | int save=0; |
| #ifndef NDEBUG | #ifndef NDEBUG |
| size_t filled= | size_t filled= |
| #endif | #endif |
| g_mime_utils_base64_encode_close ((const unsigned char*)in, len, | g_mime_utils_base64_encode_close ((const unsigned char*)in, in_size, |
| (unsigned char*)result, &state, &save); | (unsigned char*)result, &state, &save); |
| assert(filled <= len * 2 + 6); | assert(filled <= in_size * 2 + 6); |
| return result; | return result; |
| } | } |
| void pa_base64_decode(const char *in, size_t in_size, char*& result, size_t& result_size) | |
| { | |
| /* wont go to more than had (overly conservative) */ | |
| result=new(PointerFreeGC) char[in_size+1/*terminator*/]; | |
| int state=0; | |
| int save=0; | |
| result_size= | |
| g_mime_utils_base64_decode_step ((const unsigned char*)in, in_size, | |
| (unsigned char*)result, &state, &save); | |
| assert(result_size <= in_size); | |
| result[result_size]=0; // for text files | |
| } |