--- parser3/src/main/pa_common.C 2003/04/02 10:10:00 1.143.2.21.2.20 +++ parser3/src/main/pa_common.C 2003/11/06 14:11:14 1.163 @@ -5,18 +5,22 @@ Author: Alexandr Petrosian (http://paf.design.ru) */ -static const char* IDENT_COMMON_C="$Date: 2003/04/02 10:10:00 $"; +static const char* IDENT_COMMON_C="$Date: 2003/11/06 14:11:14 $"; #include "pa_common.h" #include "pa_exception.h" #include "pa_hash.h" #include "pa_globals.h" +#include "pa_request_charsets.h" +#include "pa_charsets.h" #define PA_HTTP #ifdef PA_HTTP #include "pa_vstring.h" #include "pa_vint.h" +#include "pa_vhash.h" +#include "pa_vtable.h" #ifdef CYGWIN #define _GNU_H_WINDOWS32_SOCKETS @@ -60,62 +64,6 @@ int PASCAL closesocket(SOCKET); # endif #endif -// locking constants -//#define PA_DEBUG_NO_LOCKING - -#ifdef PA_DEBUG_NO_LOCKING - -#ifdef HAVE_FLOCK - -static int lock_shared_blocking(int fd) { return flock(fd, LOCK_SH); } -static int lock_exclusive_blocking(int fd) { return flock(fd, LOCK_EX); } -static int lock_exclusive_nonblocking(int fd) { return flock(fd, LOCK_EX || LOCK_NB); } -static int unlock(int fd) { return flock(fd, LOCK_UN); } - -#else -#ifdef HAVE__LOCKING - -#define FLOCK(operation) lseek(fd, 0, SEEK_SET); return _locking(fd, operation, 1) -static int lock_shared_blocking(int fd) { FLOCK(_LK_LOCK); } -static int lock_exclusive_blocking(int fd) { FLOCK(_LK_LOCK); } -static int lock_exclusive_nonblocking(int fd) { FLOCK(_LK_NBLCK); } -static int unlock(int fd) { FLOCK(_LK_UNLCK); } - -#else -#ifdef HAVE_FCNTL - -#define FLOCK(cmd, arg) struct flock ls={arg, SEEK_SET}; return fcntl(fd, cmd, &ls) -static int lock_shared_blocking(int fd) { FLOCK(F_SETLKW, F_RDLCK); } -static int lock_exclusive_blocking(int fd) { FLOCK(F_SETLKW, F_WRLCK); } -static int lock_exclusive_nonblocking(int fd) { FLOCK(F_SETLK, F_RDLCK); } -static int unlock(int fd) { FLOCK(F_SETLK, F_UNLCK); } - -#else -#ifdef HAVE_LOCKF - -#define FLOCK(fd, operation) lseek(fd, 0, SEEK_SET); return lockf(fd, operation, 1) -static int lock_shared_blocking(int fd) { FLOCK(F_LOCK); } // on intel solaris man doesn't have doc on shared blocking -static int lock_exclusive_blocking(int fd) { FLOCK(F_LOCK); } -static int lock_exclusive_nonblocking(int fd) { FLOCK(F_TLOCK); } -static int unlock(int fd) { FLOCK(F_TLOCK); } - -#else - -#error unable to find file locking func - -#endif -#endif -#endif -#endif - -#else -static int lock_shared_blocking(int fd) { return 0; } -static int lock_exclusive_blocking(int fd) { return 0; } -static int lock_exclusive_nonblocking(int fd) { return 0; } -static int unlock(int fd) { return 0; } - -#endif - // defines for globals #define FILE_STATUS_NAME "status" @@ -130,6 +78,8 @@ const String file_status_name(FILE_STATU #define HTTP_TIMEOUT_NAME "timeout" #define HTTP_HEADERS_NAME "headers" #define HTTP_ANY_STATUS_NAME "any-status" +#define HTTP_CHARSET_NAME "charset" +#define HTTP_TABLES_NAME "tables" // statics @@ -137,6 +87,8 @@ static const String http_method_name(HTT static const String http_timeout_name(HTTP_TIMEOUT_NAME); static const String http_headers_name(HTTP_HEADERS_NAME); static const String http_any_status_name(HTTP_ANY_STATUS_NAME); +static const String http_charset_name(HTTP_CHARSET_NAME); +static const String http_tables_name(HTTP_TABLES_NAME); // defines @@ -164,23 +116,27 @@ void fix_line_breaks(char *str, size_t& } else // \r, not \n = Macintosh bol=eol+1; } - // last piece without \r, including terminating 0 + // last piece without \r if(dest!=bol) memcpy(dest, bol, eob-bol); + str[length]=0; // terminating } -char* file_read_text(Charset& charset, - const String& file_spec, - bool fail_on_read_problem, - HashStringValue* params/*, HashStringValue* * out_fields*/) { +char* file_read_text(Request_charsets& charsets, + const String& file_spec, + bool fail_on_read_problem, + HashStringValue* params/*, HashStringValue* * out_fields*/) { File_read_result file= - file_read(charset, file_spec, true, params, fail_on_read_problem); + file_read(charsets, file_spec, true, params, fail_on_read_problem); return file.success?file.str:0; } #ifdef PA_HTTP //http request stuff +#undef CRLF +#define CRLF "\r\n" + static bool set_addr(struct sockaddr_in *addr, const char* host, const short port){ memset(addr, 0, sizeof(*addr)); addr->sin_family=AF_INET; @@ -204,7 +160,7 @@ static int http_read_response(String& re if(size<=0) break; response.append_strdup(buf, size, String::L_TAINTED); - if(!result && (EOLat=response.pos("\r\n", 2))!=STRING_NOT_FOUND) { // checking status in first response + if(!result && (EOLat=response.pos(CRLF, 2))!=STRING_NOT_FOUND) { // checking status in first response const String& status_line=response.mid(0, (size_t)EOLat); ArrayString astatus; size_t pos_after=0; @@ -229,18 +185,17 @@ static int http_read_response(String& re /* ********************** request *************************** */ #if defined(SIGALRM) && defined(HAVE_SIGSETJMP) && defined(HAVE_SIGLONGJMP) -# define WE_CAN_USE_ALARM +# define PA_USE_ALARM #endif -#ifdef WE_CAN_USE_ALARM +#ifdef PA_USE_ALARM static sigjmp_buf timeout_env; static void timeout_handler(int sig){ siglongjmp(timeout_env, 1); } #endif -static int http_request( - String& response, +static int http_request(String& response, const char* host, int port, const char* request, int timeout, @@ -250,23 +205,29 @@ static int http_request( 0, "zero hostname"); //never -#ifdef WE_CAN_USE_ALARM - signal(SIGALRM, timeout_handler); +#ifdef PA_USE_ALARM + signal(SIGALRM, timeout_handler); #endif int sock=-1; - try { - int result; -#ifdef WE_CAN_USE_ALARM - if(sigsetjmp(timeout_env, 1)) - throw Exception("http.timeout", - 0, - "timeout occured while retrieving document"); - else { - alarm(timeout); +#ifdef PA_USE_ALARM + if(sigsetjmp(timeout_env, 1)) { + // stupid gcc [2.95.4] generated bad code + // which failed to handle sigsetjmp+throw: crashed inside of pre-throw code. + // rewritten simplier [though duplicating closesocket code] + if(sock>=0) + closesocket(sock); + throw Exception("http.timeout", + origin_string, + "timeout occured while retrieving document"); + return 0; // never + } else { + alarm(timeout); #endif + try { + int result; struct sockaddr_in dest; - if(!set_addr(&dest, host, port)) + if(!set_addr(&dest, host, port)) throw Exception("http.host", 0, "can not resolve hostname \"%s\"", host); @@ -287,44 +248,84 @@ static int http_request( result=http_read_response(response, sock, fail_on_status_ne_200); closesocket(sock); -#ifdef WE_CAN_USE_ALARM +#ifdef PA_USE_ALARM alarm(0); - } #endif - return result; - } catch(...) { - if(sock>=0) - closesocket(sock); -#ifdef WE_CAN_USE_ALARM - alarm(0); + return result; + } catch(...) { +#ifdef PA_USE_ALARM + alarm(0); #endif - rethrow; + if(sock>=0) + closesocket(sock); + rethrow; + } +#ifdef PA_USE_ALARM } +#endif } #ifndef DOXYGEN struct Http_pass_header_info { - Charset* charset; + Request_charsets* charsets; String* request; bool user_agent_specified; }; #endif -static void http_pass_header(HashStringValue::key_type key, HashStringValue::value_type value, +static void http_pass_header(HashStringValue::key_type key, + HashStringValue::value_type value, Http_pass_header_info *info) { *info->request <charset, String::CC_UPPER)=="USER-AGENT") - info->user_agent_specified=true; + if(String(key, String::L_TAINTED).change_case(info->charsets->source(), String::CC_UPPER)=="USER-AGENT") + info->user_agent_specified=true; +} + + +static Charset* detect_charset(Charset& source_charset, const String& content_type_value) { + const String::Body CONTENT_TYPE_VALUE= + content_type_value.change_case(source_charset, String::CC_UPPER); + // content-type: xxx/xxx; source_charset=WE-NEED-THIS + // content-type: xxx/xxx; source_charset="WE-NEED-THIS" + // content-type: xxx/xxx; source_charset="WE-NEED-THIS"; + size_t before_charseteq_pos=CONTENT_TYPE_VALUE.pos("CHARSET="); + if(before_charseteq_pos!=STRING_NOT_FOUND) { + size_t charset_begin=before_charseteq_pos+8/*CHARSET="*/; + size_t open_quote_pos=CONTENT_TYPE_VALUE.pos('"', charset_begin); + bool quoted=open_quote_pos==charset_begin; + if(quoted) + charset_begin++; // skip opening '"' + size_t charset_end=CONTENT_TYPE_VALUE.length(); + if(quoted) { + size_t close_quote_pos=CONTENT_TYPE_VALUE.pos('"', charset_begin); + if(close_quote_pos!=STRING_NOT_FOUND) + charset_end=close_quote_pos; + } else { + size_t delim_pos=CONTENT_TYPE_VALUE.pos(';', charset_begin); + if(delim_pos!=STRING_NOT_FOUND) + charset_end=delim_pos; + } + const String::Body CHARSET_NAME_BODY= + CONTENT_TYPE_VALUE.mid(charset_begin, charset_end); + + return &charsets.get(CHARSET_NAME_BODY); + } + + return 0; } + + #ifndef DOXYGEN struct File_read_http_result { char *str; size_t length; HashStringValue* headers; }; #endif -static File_read_http_result file_read_http(Charset& charset, const String& file_spec, +/// @todo build .cookies field. use ^file.tables.SET-COOKIES.menu{ for now +static File_read_http_result file_read_http(Request_charsets& charsets, + const String& file_spec, HashStringValue *options=0) { File_read_http_result result; char host[MAX_STRING]; @@ -334,25 +335,7 @@ static File_read_http_result file_read_h int timeout=2; bool fail_on_status_ne_200=true; Value* vheaders=0; - - String& connect_string=*new String; - // not in ^sql{... L_SQL ...} spirit, but closer to ^file::load one - connect_string.append(file_spec, String::L_URI); // tainted pieces -> URI pieces - - const char* connect_string_cstr=connect_string.cstr(String::L_UNSPECIFIED); - const char* current=connect_string_cstr; - if(strncmp(current, "http://", 7)!=0) - throw Exception(0, - &connect_string, - "does not start with http://"); //never - current+=7; - - strncpy(host, current, sizeof(host)-1); host[sizeof(host)-1]=0; - char* host_uri=lsplit(host, '/'); - uri=host_uri?current+(host_uri-1-host):"/"; - char* port_cstr=lsplit(host, ':'); - char* error_pos=0; - port=port_cstr?strtol(port_cstr, &error_pos, 0):80; + Charset *asked_remote_charset=0; if(options) { int valid_options=0; @@ -371,59 +354,107 @@ static File_read_http_result file_read_h valid_options++; fail_on_status_ne_200=!vany_status->as_bool(); } + if(Value* vcharset_name=options->get(http_charset_name)) { + valid_options++; + asked_remote_charset=&::charsets.get(vcharset_name->as_string(). + change_case(charsets.source(), String::CC_UPPER)); + } if(valid_options!=options->count()) throw Exception("parser.runtime", 0, "invalid option passed"); - } + } + if(!asked_remote_charset) // defaulting to $request:charset + asked_remote_charset=&charsets.source(); - //making request - String request; - if(method) - request<is_string()) { // allow empty - if(HashStringValue *headers=vheaders->get_hash()) { - Http_pass_header_info info={&charset, &request}; - headers->for_each(http_pass_header, &info); - user_agent_specified=info.user_agent_specified; - } else - throw Exception("parser.runtime", - &connect_string, - "headers param must be hash"); - }; - if(!user_agent_specified) // defaulting - request << "user-agent: " DEFAULT_USER_AGENT "\n"; - request<<"\n"; + //preparing request + String& connect_string=*new String; + // not in ^sql{... L_SQL ...} spirit, but closer to ^file::load one + connect_string.append(file_spec, String::L_URI); // tainted pieces -> URI pieces + + const char* request_cstr; + { + // influence URLencoding of tainted pieces to String::L_URI lang + Temp_client_charset temp(charsets, *asked_remote_charset); + + const char* connect_string_cstr=connect_string.cstr(String::L_UNSPECIFIED); + + const char* current=connect_string_cstr; + if(strncmp(current, "http://", 7)!=0) + throw Exception(0, + &connect_string, + "does not start with http://"); //never + current+=7; + + strncpy(host, current, sizeof(host)-1); host[sizeof(host)-1]=0; + char* host_uri=lsplit(host, '/'); + uri=host_uri?current+(host_uri-1-host):"/"; + char* port_cstr=lsplit(host, ':'); + char* error_pos=0; + port=port_cstr?strtol(port_cstr, &error_pos, 0):80; + + + //making request + String request; + if(method) + request<is_string()) { // allow empty + if(HashStringValue *headers=vheaders->get_hash()) { + Http_pass_header_info info={&charsets, &request}; + headers->for_each(http_pass_header, &info); + user_agent_specified=info.user_agent_specified; + } else + throw Exception("parser.runtime", + &connect_string, + "headers param must be hash"); + }; + if(!user_agent_specified) // defaulting + request << "user-agent: " DEFAULT_USER_AGENT CRLF; + request << CRLF; + + request_cstr=request.cstr(String::L_UNSPECIFIED); + } + // recode those pieces which are not in String::L_URI lang + // [those violating HTTP standard, but widly used] + request_cstr=Charset::transcode( + String::C(request_cstr, strlen(request_cstr)), + charsets.source(), + *asked_remote_charset); //sending request String response; - const char* request_cstr=request.cstr(String::L_UNSPECIFIED); int status_code=http_request(response, host, port, request_cstr, timeout, fail_on_status_ne_200); //processing results - size_t pos=response.pos("\r\n\r\n", 4); - if(pos==STRING_NOT_FOUND || pos<1){ + size_t pos=response.pos(CRLF CRLF, 4); + if(pos==STRING_NOT_FOUND || pos<1) { throw Exception("http.response", &connect_string, "bad response from host - no headers found"); } const String& header_block=response.mid(0, pos); - const String& body=response.mid(pos+4, response.length()); + const String& raw_body=response.mid(pos+4, response.length()); ArrayString aheaders; result.headers=new HashStringValue; + VHash* vtables=new VHash; + result.headers->put(http_tables_name, vtables); + HashStringValue& tables=vtables->hash(); + size_t pos_after=0; - header_block.split(aheaders, pos_after, "\r\n"); + header_block.split(aheaders, pos_after, CRLF); //processing headers size_t aheaders_count=aheaders.count(); + Charset* real_remote_charset=0; // undetected, yet for(size_t i=1; iput( - line.mid(0, pos).change_case(charset, String::CC_UPPER), - new VString(line.mid(pos+2, line.length()))); + const String::Body HEADER_NAME= + line.mid(0, pos).change_case(charsets.source(), String::CC_UPPER); + const String& header_value=line.mid(pos+2, line.length()); + if(HEADER_NAME=="CONTENT-TYPE") + real_remote_charset=detect_charset(charsets.source(), header_value); + + // 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)); + } + + 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 - result.str=body.cstrm(); - result.length=body.length(); + String::C real_body=Charset::transcode( + String::C(raw_body.cstrm()/*must be modifiable*/, raw_body.length()), + *real_remote_charset, + charsets.source()); + + result.str=const_cast(real_body.str); // hacking a little + result.length=real_body.length; result.headers->put(file_status_name, new VInt(status_code)); return result; } @@ -452,10 +517,10 @@ struct File_read_action_info { }; #endif static void file_read_action( - struct stat& finfo, - int f, - const String& file_spec, const char* fname, bool as_text, - void *context) { + struct stat& finfo, + int f, + const String& file_spec, const char* fname, bool as_text, + void *context) { File_read_action_info& info=*static_cast(context); if(size_t to_read_size=(size_t)finfo.st_size) { *info.data=new(PointerFreeGC) char[to_read_size+(as_text?1:0)]; @@ -464,7 +529,7 @@ static void file_read_action( if(ssize_t(*info.data_size)<0 || *info.data_size>to_read_size) throw Exception(0, &file_spec, - "read failed: actually read %lu bytes count not in [0..%lu] valid range", + "read failed: actually read %l bytes count not in [0..%lu] valid range", *info.data_size, to_read_size); } else { // empty file if(as_text) { @@ -476,20 +541,25 @@ static void file_read_action( return; } } -File_read_result file_read(Charset& charset, const String& file_spec, +File_read_result file_read(Request_charsets& charsets, const String& file_spec, bool as_text, HashStringValue *params, bool fail_on_read_problem) { File_read_result result={false}; #ifdef PA_HTTP if(file_spec.starts_with("http://")) { // fail on read problem - File_read_http_result http;//=file_read_http(charset, file_spec, params); + File_read_http_result http=file_read_http(charsets, file_spec, params); result.success=true; - result.str="HELLO";//http.str; - result.length=5;//http.length; - result.headers=new HashStringValue;//http.headers; + result.str=http.str; + result.length=http.length; + result.headers=http.headers; } else { #endif + if(params && params->count()) + throw Exception("parser.runtime", + 0, + "invalid option passed"); + File_read_action_info info={&result.str, &result.length}; result.success=file_read_action_under_lock(file_spec, "read", file_read_action, &info, @@ -502,8 +572,7 @@ File_read_result file_read(Charset& char // UTF-8 signature: EF BB BF if(result.length>=3) { char *in=(char *)result.str; - if((in[0] == '\xEF') && (in[1] == '\xBB') && - (in[2] == '\xBF')) { + if(strncmp(in, "\xEF\xBB\xBF", 3)==0) { result.str=in+3; result.length-=3;// skip prefix } } @@ -516,6 +585,23 @@ File_read_result file_read(Charset& char return result; } +#ifdef PA_SAFE_MODE +void check_safe_mode(struct stat finfo, const String& file_spec, const char* fname) { + if(finfo.st_uid/*foreign?*/!=geteuid() + && finfo.st_gid/*foreign?*/!=getegid()) + throw Exception("parser.runtime", + &file_spec, + "parser is in safe mode: " + "reading files of foreign group and user disabled " + "[recompile parser with --disable-safe-mode configure option], " + "actual filename '%s', " + "fuid(%d)!=euid(%d) or fgid(%d)!=egid(%d)", + fname, + finfo.st_uid, geteuid(), + finfo.st_gid, getegid()); +} +#endif + bool file_read_action_under_lock(const String& file_spec, const char* action_name, File_read_action action, void *context, bool as_text, @@ -534,7 +620,7 @@ bool file_read_action_under_lock(const S // later: it seems, even this does not help sometimes if((f=open(fname, O_RDONLY|(as_text?_O_TEXT:_O_BINARY)))>=0) { try { - if(lock_shared_blocking(f)!=0) + if(pa_lock_shared_blocking(f)!=0) throw Exception("file.lock", &file_spec, "shared lock failed: %s (%d), actual filename '%s'", @@ -548,23 +634,18 @@ bool file_read_action_under_lock(const S strerror(errno), errno, fname); #ifdef PA_SAFE_MODE - if(finfo.st_uid/*foreign?*/!=geteuid() - && finfo.st_gid/*foreign?*/!=getegid()) - throw Exception("parser.runtime", - &file_spec, - "parser is in safe mode: reading files of foreign group and user disabled [recompile parser with --disable-safe-mode configure option], actual filename '%s'", - fname); + check_safe_mode(finfo, file_spec, fname); #endif action(finfo, f, file_spec, fname, as_text, context); } catch(...) { - unlock(f);close(f); + pa_unlock(f);close(f); if(fail_on_read_problem) rethrow; return false; } - unlock(f);close(f); + pa_unlock(f);close(f); return true; } else { if(fail_on_read_problem) @@ -601,7 +682,7 @@ bool file_write_action_under_lock( O_CREAT|O_RDWR |(as_text?_O_TEXT:_O_BINARY) |(do_append?O_APPEND:PA_O_TRUNC), 0664))>=0) { - if((do_block?lock_exclusive_blocking(f):lock_exclusive_nonblocking(f))!=0) { + if((do_block?pa_lock_exclusive_blocking(f):pa_lock_exclusive_nonblocking(f))!=0) { Exception e("file.lock", &file_spec, "shared lock failed: %s (%d), actual filename '%s'", @@ -619,7 +700,7 @@ bool file_write_action_under_lock( if(!do_append) ftruncate(f, lseek(f, 0, SEEK_CUR)); // one can not use O_TRUNC, read lower #endif - unlock(f);close(f); + pa_unlock(f);close(f); rethrow; } @@ -627,7 +708,7 @@ bool file_write_action_under_lock( if(!do_append) ftruncate(f, lseek(f, 0, SEEK_CUR)); // O_TRUNC truncates even exclusevely write-locked file [thanks to Igor Milyakov for discovering] #endif - unlock(f);close(f); + pa_unlock(f);close(f); return true; } else throw Exception(errno==EACCES?"file.access":0, @@ -1006,19 +1087,3 @@ int __snprintf(char* b, size_t s, const va_end(l); return r; } - -int pa_sleep(unsigned long secs, unsigned long usecs) { - for (; usecs >= 1000000; ++secs, usecs -= 1000000); - -#ifdef WIN32 - Sleep(secs * 1000 + usecs / 1000); - return 0; -#else - struct timeval t; - t.tv_sec = secs; - t.tv_usec = usecs; - return (select(0, NULL, NULL, NULL, &t) == -1 ? errno : 0); -#endif -} - -