--- parser3/src/main/pa_common.C 2003/04/08 16:11:54 1.143.2.21.2.25 +++ parser3/src/main/pa_common.C 2003/11/20 16:34:26 1.168 @@ -5,7 +5,7 @@ Author: Alexandr Petrosian (http://paf.design.ru) */ -static const char* IDENT_COMMON_C="$Date: 2003/04/08 16:11:54 $"; +static const char * const IDENT_COMMON_C="$Date: 2003/11/20 16:34:26 $"; #include "pa_common.h" #include "pa_exception.h" @@ -19,6 +19,8 @@ static const char* IDENT_COMMON_C="$Date #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 @@ -62,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" @@ -133,6 +79,7 @@ const String file_status_name(FILE_STATU #define HTTP_HEADERS_NAME "headers" #define HTTP_ANY_STATUS_NAME "any-status" #define HTTP_CHARSET_NAME "charset" +#define HTTP_TABLES_NAME "tables" // statics @@ -141,6 +88,7 @@ static const String http_timeout_name(HT 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 @@ -168,9 +116,10 @@ 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(Request_charsets& charsets, @@ -185,6 +134,9 @@ char* file_read_text(Request_charsets& c #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; @@ -208,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; @@ -246,7 +198,11 @@ static void timeout_handler(int sig){ static int http_request(String& response, const char* host, int port, const char* request, - int timeout, + int +#ifdef PA_USE_ALARM + timeout +#endif + , bool fail_on_status_ne_200) { if(!host) throw Exception("http.host", @@ -325,7 +281,7 @@ static void http_pass_header(HashStringV Http_pass_header_info *info) { *info->request <charsets->source(), String::CC_UPPER)=="USER-AGENT") info->user_agent_specified=true; @@ -333,7 +289,7 @@ static void http_pass_header(HashStringV static Charset* detect_charset(Charset& source_charset, const String& content_type_value) { - const StringBody 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" @@ -345,7 +301,7 @@ static Charset* detect_charset(Charset& bool quoted=open_quote_pos==charset_begin; if(quoted) charset_begin++; // skip opening '"' - size_t charset_end=0; + 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) @@ -355,7 +311,7 @@ static Charset* detect_charset(Charset& if(delim_pos!=STRING_NOT_FOUND) charset_end=delim_pos; } - const StringBody CHARSET_NAME_BODY= + const String::Body CHARSET_NAME_BODY= CONTENT_TYPE_VALUE.mid(charset_begin, charset_end); return &charsets.get(CHARSET_NAME_BODY); @@ -371,6 +327,7 @@ struct File_read_http_result { HashStringValue* headers; }; #endif +/// @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) { @@ -384,25 +341,6 @@ static File_read_http_result file_read_h Value* vheaders=0; Charset *asked_remote_charset=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; - if(options) { int valid_options=0; if(Value* vmethod=options->get(http_method_name)) { @@ -413,7 +351,7 @@ static File_read_http_result file_read_h valid_options++; timeout=vtimeout->as_int(); } - if(vheaders=options->get(http_headers_name)) { + if((vheaders=options->get(http_headers_name))) { valid_options++; } if(Value* vany_status=options->get(http_any_status_name)) { @@ -434,51 +372,73 @@ static File_read_http_result file_read_h 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={&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 "\n"; - request<<"\n"; - - //sending request - String response; + //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); - request_cstr=request.cstr(String::L_UNSPECIFIED); + 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, false}; + 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; - // recode pieces 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); + 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; 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); + size_t pos=response.pos(CRLF CRLF, 4); if(pos==STRING_NOT_FOUND || pos<1) { throw Exception("http.response", &connect_string, @@ -489,8 +449,12 @@ static File_read_http_result file_read_h 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(); @@ -502,17 +466,42 @@ static File_read_http_result file_read_h throw Exception("http.response", &connect_string, "bad response from host - bad header \"%s\"", line.cstr()); - const StringBody HEADER_NAME= + 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") + 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; + // todo: do that only if ^file::load[text // output response String::C real_body=Charset::transcode( String::C(raw_body.cstrm()/*must be modifiable*/, raw_body.length()), @@ -533,10 +522,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)]; @@ -545,7 +534,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) { @@ -560,7 +549,7 @@ static void file_read_action( 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}; + File_read_result result={false, 0, 0, 0}; #ifdef PA_HTTP if(file_spec.starts_with("http://")) { // fail on read problem @@ -571,6 +560,11 @@ File_read_result file_read(Request_chars 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, @@ -583,7 +577,7 @@ File_read_result file_read(Request_chars // UTF-8 signature: EF BB BF if(result.length>=3) { char *in=(char *)result.str; - if(strncmp(in, "\xEB\xBB\xBF", 3)==0) { + if(strncmp(in, "\xEF\xBB\xBF", 3)==0) { result.str=in+3; result.length-=3;// skip prefix } } @@ -597,7 +591,7 @@ File_read_result file_read(Request_chars } #ifdef PA_SAFE_MODE -void check_safe_mode(stat finfo, const String& file_spec, const char* fname) { +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", @@ -631,7 +625,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'", @@ -650,13 +644,13 @@ bool file_read_action_under_lock(const S 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) @@ -693,7 +687,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'", @@ -711,7 +705,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; } @@ -719,7 +713,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, @@ -765,10 +759,10 @@ static void rmdir(const String& file_spe rmdir(file_spec.mid(0, pos_after-1/* / */).cstr(String::L_FILE_SPEC)); } -bool file_delete(const String& file_spec, bool fail_on_read_problem) { +bool file_delete(const String& file_spec, bool fail_on_problem) { const char* fname=file_spec.cstr(String::L_FILE_SPEC); if(unlink(fname)!=0) - if(fail_on_read_problem) + if(fail_on_problem) throw Exception(errno==EACCES?"file.access":errno==ENOENT?"file.missing":0, &file_spec, "unlink failed: %s (%d), actual filename '%s'", @@ -1084,7 +1078,7 @@ int __vsnprintf(char* b, size_t s, const if(r<0) r=0; - else if(r>s) + else if((size_t)r>s) r=s; #endif b[r]=0; @@ -1098,19 +1092,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 -} - -