Diff for /parser3/src/main/pa_common.C between versions 1.92 and 1.140

version 1.92, 2001/11/23 12:56:37 version 1.140, 2002/12/26 11:45:37
Line 1 Line 1
 /** @file  /** @file
         Parser: commonly functions.          Parser: commonly functions.
   
         Copyright(c) 2001 ArtLebedev Group(http://www.artlebedev.com)          Copyright(c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com)
         Author: Alexander Petrosyan <paf@design.ru>(http://paf.design.ru)          Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
   
         $Id$  
 */  */
   
   static const char* IDENT_COMMON_C="$Date$"; 
   
 #include "pa_common.h"  #include "pa_common.h"
 #include "pa_types.h"  
 #include "pa_exception.h"  #include "pa_exception.h"
 #include "pa_pool.h"  
 #include "pa_globals.h"  #include "pa_globals.h"
 #include "pa_value.h"  
 #include "pa_hash.h"  #include "pa_hash.h"
 #include "pa_string.h"  #include "pa_vstring.h"
   #include "pa_vdate.h"
   
   #ifdef WIN32
   #       include <windows.h>
   #else
   #       define closesocket close
   #endif
   
   // some maybe-undefined constants
   
 #ifndef _O_TEXT  #ifndef _O_TEXT
 #       define _O_TEXT 0  #       define _O_TEXT 0
Line 22 Line 28
 #ifndef _O_BINARY  #ifndef _O_BINARY
 #       define _O_BINARY 0  #       define _O_BINARY 0
 #endif  #endif
 #ifndef O_TRUNC  
 #       define O_TRUNC 0  #ifdef HAVE_FTRUNCATE
   #       define PA_O_TRUNC 0
   #else
   #       ifdef _O_TRUNC
   #               define PA_O_TRUNC _O_TRUNC
   #       else
   #               error you must have either ftruncate function or _O_TRUNC bit declared
   #       endif
 #endif  #endif
   
 static char *strnchr(char *buf, size_t size, char c) {  // locking constants
         for(; size-->0; buf++) {  
                 if(*buf==c)  
                         return buf;  
         }  
   
         return 0;  #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
   
   #define DEFAULT_USER_AGENT "parser3"
   
   
   void fix_line_breaks(char* buf, size_t& size) {
           if(size==0)
                   return;
   
 void fix_line_breaks(char *buf, size_t& size) {  
         //_asm int 3;          //_asm int 3;
         const char * const eob=buf+size;          const char* const eob=buf+size;
         char *dest=buf;          char* dest=buf;
         // fix DOS: \r\n -> \n          // fix DOS: \r\n -> \n
         // fix Macintosh: \r -> \n          // fix Macintosh: \r -> \n
         char *bol=buf;          char* bol=buf;
         while(char *eol=strnchr(bol, eob -bol, '\r')) {          while(char* eol=(char*)memchr(bol, '\r', eob -bol)) {
                 size_t len=eol-bol;                  size_t len=eol-bol;
                 if(dest!=bol)                  if(dest!=bol)
                         memcpy(dest, bol, len);                          memcpy(dest, bol, len); 
                 dest+=len;                  dest+=len;
                 *dest++='\n';                  *dest++='\n'; 
   
                 if(&eol[1]<eob && eol[1]=='\n') { // \r,\n = DOS                  if(&eol[1]<eob && eol[1]=='\n') { // \r, \n = DOS
                         bol=eol+2;                          bol=eol+2;
                         size--;                          size--; 
                 } else // \r,not \n = Macintosh                  } else // \r, not \n = Macintosh
                         bol=eol+1;                          bol=eol+1;
         }          }
         // last piece without \r, including terminating 0          // last piece without \r, including terminating 0
         if(dest!=bol)          if(dest!=bol)
                 memcpy(dest, bol, eob-bol);                  memcpy(dest, bol, eob-bol); 
 }  }
   
 char *file_read_text(Pool& pool, const String& file_spec, bool fail_on_read_problem) {  char* file_read_text(Pool& pool, const String& file_spec, 
                                            bool fail_on_read_problem,
                                            Hash *params, Hash** out_fields) {
         void *result;  size_t size;          void *result;  size_t size;
         return file_read(pool, file_spec, result, size, true, fail_on_read_problem)?(char *)result:0;          return file_read(pool, file_spec, result, size, true, params, out_fields, fail_on_read_problem)?(char *)result:0;
   }
   
   //http request stuff
   /* ************************ http stuff *********************** */
   
   static bool set_addr(struct sockaddr_in *addr, const char* host, const short port){
       memset(addr, 0, sizeof(*addr)); 
       addr->sin_family=AF_INET;
       addr->sin_port=htons(port); 
       if(host) {
                   if(struct hostent *hostIP=gethostbyname(host)) 
                           memcpy(&addr->sin_addr, hostIP->h_addr, hostIP->h_length); 
                   else
                           return false;
       } else 
                   addr->sin_addr.s_addr=INADDR_ANY;
       return true;
   }
   
   static void http_read_response(String& response, int sock){
           bool check_status=true;
           ssize_t EOLat=0;
           while(true) {
                   char *buf=(char *)response.pool().malloc(MAX_STRING); 
                   ssize_t size=recv(sock, buf, MAX_STRING, 0); 
                   if(size<=0)
                           break;
                   response.APPEND_TAINTED(buf, size, "remote HTTP server response", 0); 
                   if(check_status && (EOLat=response.pos("\r\n", 2))>=0) { // checking status in first response
                           check_status=false;
   
                           const String& status_line=response.mid(0, (size_t)EOLat);
                           Array astatus(response.pool()); 
                           size_t pos_after_ref=0; status_line.split(astatus, &pos_after_ref, " ", 1); 
                           const String* status_code=astatus.get_string(1)
   
                           if(!status_code || *status_code!="200")
                                   throw Exception("http.status",
                                           status_code,
                                           "invalid HTTP response status");
                   }
       }
           if(check_status)
                   throw Exception("http.response",
                           0,
                           "bad response from host - no status found (size=%lu)", response.size()); 
   }
   
   /* ********************** request *************************** */
   
   #if defined(SIGALRM) && defined(HAVE_SIGSETJMP) && defined(HAVE_SIGLONGJMP)
   #       define WE_CAN_USE_ALARM
   #endif
   
   #ifdef WE_CAN_USE_ALARM
   static sigjmp_buf timeout_env;
   static void timeout_handler(int sig){
       siglongjmp(timeout_env, 1); 
   }
   #endif
   
   static void http_request(String& response,
                                                           const String *origin_string, 
                                                           const char* host, int port, 
                                                           const char* request, 
                                                           int timeout){
           if(!host)
                   throw Exception("http.host", 
                           origin_string, 
                           "zero hostname");  //never
   
   #ifdef WE_CAN_USE_ALARM
       signal(SIGALRM, timeout_handler); 
   #endif
           int sock=-1;
           try {
   #ifdef WE_CAN_USE_ALARM
                   if(sigsetjmp(timeout_env, 1))
                           throw Exception("http.timeout", 
                                   origin_string, 
                                   "timeout occured while retrieving document"); 
                   else {
                           alarm(timeout); 
   #endif
                           struct sockaddr_in dest;
                   
                   if(!set_addr(&dest, host, port))
                                   throw Exception("http.host", 
                                           origin_string, 
                                           "can not resolve hostname \"%s\"", host); 
                           
                           if((sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/))<0)
                                   throw Exception("http.connect", 
                                           origin_string, 
                                           "can not make socket: %s (%d)", strerror(errno), errno); 
                           if(connect(sock, (struct sockaddr *)&dest, sizeof(dest)))
                                   throw Exception("http.connect", 
                                           origin_string, 
                                           "can not connect to host \"%s\": %s (%d)", host, strerror(errno), errno); 
                           size_t request_size=strlen(request);
                           if(send(sock, request, request_size, 0)!=(ssize_t)request_size)
                                   throw Exception("http.connect", 
                                           origin_string, 
                                           "error sending request: %s (%d)", strerror(errno), errno); 
   
                           http_read_response(response, sock); 
                           if(sock>=0) 
                                   closesocket(sock); 
   #ifdef WE_CAN_USE_ALARM
                   }
   #endif
           } catch(...) {
                   if(sock>=0) 
                           closesocket(sock); 
   #ifdef WE_CAN_USE_ALARM
                   alarm(0)
   #endif
                   /*re*/throw;
           }
   }
   
   #ifndef DOXYGEN
   struct Http_pass_header_info {
           String* request;
           bool user_agent_specified;
   };
   #endif
   static void http_pass_header(const Hash::Key& key, Hash::Val *value, void *info)
   {
           Http_pass_header_info& i=*static_cast<Http_pass_header_info *>(info);
           Pool& pool=i.request->pool();
       
       *(i.request)<<key<<": "
                   << attributed_meaning_to_string(*static_cast<Value *>(value), String::UL_HTTP_HEADER, false)
                   <<"\n"; 
           
           if(key.change_case(pool, String::CC_UPPER)=="USER-AGENT")
                   i.user_agent_specified=true;
   }
   static void file_read_http(Pool& pool, const String& file_spec, 
                                           void*& data, size_t& data_size, 
                                           Hash *options=0, Hash** out_fields=0) {
           char host[MAX_STRING]; 
           const char* uri; 
           int port;
           const char* method="GET"; 
           int timeout=2;
           Value *vheaders=0;
   
           String connect_string(pool);
           // not in ^sql{... UL_SQL ...} spirit, but closer to ^file::load one
           connect_string.append(file_spec, String::UL_URI); // tainted pieces -> URI pieces
   
           char* connect_string_cstr=connect_string.cstr(String::UL_UNSPECIFIED); 
           if(strncmp(connect_string_cstr, "http://", 7)!=0)
                   throw Exception(0, 
                           &connect_string, 
                           "does not start with http://"); //never
           connect_string_cstr+=7;
   
           strncpy(host, connect_string_cstr, sizeof(host)-1);  host[sizeof(host)-1]=0;
           char* host_uri=lsplit(host, '/'); 
           uri=host_uri?connect_string_cstr+(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=static_cast<Value *>(options->get(*http_method_name))) {
                           valid_options++;
                           method=vmethod->as_string().cstr(); 
                   }
                   if(Value *vtimeout=static_cast<Value *>(options->get(*http_timeout_name))) {
                           valid_options++;
                           timeout=vtimeout->as_int(); 
                   }
                   if(vheaders=static_cast<Value *>(options->get(*http_headers_name))) {
                           valid_options++;
                   }
                   if(valid_options!=options->size())
                           throw Exception("parser.runtime",
                                   0,
                                   "invalid option passed");
           } 
   
           //making request
           String request(pool);
           request<< method <<" "<< uri <<" HTTP/1.0\nHost: "<< host<<"\n"; 
           bool user_agent_specified=false;
           if(vheaders && !vheaders->is_string()) { // allow empty
                   if(Hash *headers=vheaders->get_hash(&connect_string)) {
                           Http_pass_header_info info={&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(pool); 
           http_request(response,
                   &connect_string, host, port, request.cstr(String::UL_UNSPECIFIED), timeout); 
           
           //processing results    
           int pos=response.pos("\r\n\r\n", 4); 
           if(pos<1){
                   throw Exception("http.response", 
                           &connect_string,
                           "bad response from host - no headers found"); 
           }
           String header_block=response.mid(0, pos); 
           String body=response.mid(pos+4, response.size()); 
           
           Array aheaders(pool); 
           Hash& headers=*new(pool) Hash(pool); 
           size_t pos_after_ref=0;
           header_block.split(aheaders, &pos_after_ref, "\r\n", 2); 
           
           //processing status code
           const String *status_line=aheaders.get_string(0)
           if(!status_line){
                   throw Exception("http.response", 
                           &connect_string,
                           "bad response from host - no status line "); 
           }
           //processing headers
           for(int i=1;i<aheaders.size();i++) {
                   if(const String *line=aheaders.get_string(i)) {
                           pos=line->pos(": ", 2); 
                           if(pos<1)
                                   throw Exception("http.response", 
                                           &connect_string,
                                           "bad response from host - bad header \"%s\"", line->cstr()); 
                                   
                           headers.put(
                                   line->mid(0, pos).change_case(pool, String::CC_UPPER), 
                                   new(pool) VString(line->mid(pos+2, line->size()))); 
                   } else
                           throw Exception("http.response", 
                                   &connect_string, 
                                   "bad response from host - bad headers \"%s\"", header_block.cstr()); 
           }
   
           // output response
           data=body.cstr(); data_size=body.size();
           if(out_fields)
                   *out_fields=&headers;
   }
   
   #ifndef DOXYGEN
   struct File_read_action_info {
           void **data; size_t *data_size;
   }; 
   #endif
   static void file_read_action(Pool& pool, 
                                                            struct stat& finfo, 
                                                            int f, 
                                                            const String& file_spec, const char* fname, bool as_text, 
                                                            void *context) {
           File_read_action_info& info=*static_cast<File_read_action_info *>(context); 
           if(size_t to_read_size=(size_t)finfo.st_size) { 
                   *info.data=pool.malloc(to_read_size+(as_text?1:0), 3); 
                   *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)
                           throw Exception(0, 
                                   &file_spec, 
                                   "read failed: actually read %lu bytes count not in [0..%lu] valid range", 
                                           *info.data_size, to_read_size); 
           } else { // empty file
                   if(as_text) {
                           *info.data=pool.malloc(1)
                           *(char*)(*info.data)=0;
                   } else 
                           *info.data=0;
                   *info.data_size=0;
                   return;
           }
 }  }
 bool file_read(Pool& pool, const String& file_spec,   bool file_read(Pool& pool, const String& file_spec, 
                            void*& data, size_t& data_size, bool as_text,                             void*& data, size_t& data_size, 
                            bool fail_on_read_problem,                             bool as_text, Hash *params, Hash** out_fields, 
                            size_t offset, size_t limit) {                             bool fail_on_read_problem) {
         const char *fname=file_spec.cstr(String::UL_FILE_SPEC);          bool result;
 //printf("file_read(%s)\n", fname);          if(file_spec.starts_with("http://", 7)) {
                   // fail on read problem
                   file_read_http(pool, file_spec, data, data_size, params, out_fields); 
                   result=true;
           } else {
                   File_read_action_info info={&data, &data_size}; 
                   result=file_read_action_under_lock(pool, file_spec, 
                           "read", file_read_action, &info, 
                           as_text, fail_on_read_problem); 
           }
   
           if(result && as_text) {
                   // UTF-8 signature: EF BB BF
                   if(data_size>=3) {
                           char *in=(char *)data;
                           if((in[0] == '\xEF') && (in[1] == '\xBB') &&
                                   (in[2] == '\xBF')) {
                                   data=in+3; data_size-=3;// skip prefix
                           }
                   }
   
                   fix_line_breaks((char *)(data), data_size); 
                   // note: after fixing
                   ((char*&)(data))[data_size]=0;
           }
   
           return result;
   }
   
   bool file_read_action_under_lock(Pool& pool, const String& file_spec, 
                                   const char* action_name, File_read_action action, void *context, 
                                   bool as_text, 
                                   bool fail_on_read_problem) {
           const char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
         int f;          int f;
     struct stat finfo;  
   
         // first open, next stat:          // first open, next stat:
         // directory update of NTFS hard links performed on open.          // directory update of NTFS hard links performed on open.
         // ex:           // ex: 
         //   a.html:^test[] and b.html hardlink to a.html          //   a.html:^test[] and b.html hardlink to a.html
         //   user inserts ! before ^test in a.html          //   user inserts ! before ^test in a.html
         //   directory entry of b.html in NTFS not updated at once,          //   directory entry of b.html in NTFS not updated at once, 
         //   they delay update till open, so we would receive "!^test[" string          //   they delay update till open, so we would receive "!^test[" string
         //   if would do stat, next open.          //   if would do stat, next open.
     if(          // later: it seems, even this does not help sometimes
                 (f=open(fname, O_RDONLY|(as_text?_O_BINARY/*_O_TEXT*/:_O_BINARY)))>=0 &&       if((f=open(fname, O_RDONLY|(as_text?_O_TEXT:_O_BINARY)))>=0) {
                 stat(fname, &finfo)==0) {                  try {
                 /*if(exclusive)                          if(lock_shared_blocking(f)!=0)
                         flock(f, LOCK_EX);*/                                  throw Exception("file.lock", 
                 size_t max_size=limit?min(offset+limit, finfo.st_size)-offset:finfo.st_size;                                                  &file_spec, 
                 int read_size;                                                  "shared lock failed: %s (%d), actual filename '%s'", 
                 if(!max_size) { // eof                                                          strerror(errno), errno, fname); 
                         if(as_text) {  
                                 data=pool.malloc(1);                          struct stat finfo;
                                 *(char*)data=0;                          if(stat(fname, &finfo)!=0)
                         } else                                   throw Exception("file.missing", // hardly possible: we just opened it OK
                                 data=0;                                          &file_spec, 
                         read_size=0;                                          "stat failed: %s (%d), actual filename '%s'", 
                 } else {                                                  strerror(errno), errno, fname); 
                         data=pool.malloc(max_size+(as_text?1:0), 3);  
                         if(offset)  #ifdef PA_SAFE_MODE
                                 lseek(f, offset, SEEK_SET);                          if(finfo.st_uid/*foreign?*/!=geteuid()
                         read_size=read(f, data, max_size);                                  && finfo.st_gid/*foreign?*/!=getegid())
                 }                                  throw Exception("parser.runtime", 
                 /*if(exclusive)                                          &file_spec, 
                         flock(f, LOCK_UN);*/                                          "parser is in safe mode: reading files of foreign group and user disabled [recompile parser with --disable-safe-mode configure option], actual filename '%s'", 
                 close(f);                                                  fname); 
                 if(!max_size) // eof  #endif
                         return true;  
   
                 if(read_size<0 || read_size>max_size)                          action(pool, finfo, f, file_spec, fname, as_text, context); 
                         throw Exception(0, 0,                   } catch(...) {
                                 &file_spec,                           unlock(f);close(f); 
                                 "read failed: actually read %d bytes count not in [0..%lu] valid range",                           if(fail_on_read_problem)
                                         read_size, (unsigned long)max_size); //never                                  /*re*/throw;
                           return false;                   
                   } 
   
                 data_size=read_size;                              unlock(f);close(f); 
                 if(as_text) {  
                         fix_line_breaks((char *)data, data_size);  
                         // note: after fixing  
                         ((char*&)data)[data_size]=0;  
                 }  
                 return true;                  return true;
     }      } else {
         if(fail_on_read_problem)                  if(fail_on_read_problem)
                 throw Exception(0, 0,                           throw Exception(errno==EACCES?"file.access":errno==ENOENT?"file.missing":0, 
                         &file_spec,                                   &file_spec, 
                         "read failed: %s (%d), actual filename '%s'",                                   "%s failed: %s (%d), actual filename '%s'", 
                                 strerror(errno), errno, fname);                                          action_name, strerror(errno), errno, fname); 
     return false;                  return false;
           }
 }  }
   
 static void create_dir_for_file(const String& file_spec) {  static void create_dir_for_file(const String& file_spec) {
         size_t pos_after=1;          size_t pos_after=1;
         int pos_before;          int pos_before;
         while((pos_before=file_spec.pos("/", 1, pos_after))>=0) {          while((pos_before=file_spec.pos("/", 1, pos_after))>=0) {
                 mkdir(file_spec.mid(0, pos_before).cstr(String::UL_FILE_SPEC), 0775);                  mkdir(file_spec.mid(0, pos_before).cstr(String::UL_FILE_SPEC), 0775); 
                 pos_after=pos_before+1;                  pos_after=pos_before+1;
         }          }
 }  }
   
 void file_write(Pool& pool,   bool file_write_action_under_lock(
                                 const String& file_spec,                                   const String& file_spec, 
                                 const void *data, size_t size,                                   const char* action_name, File_write_action action, void *context, 
                                 bool as_text,                                  bool as_text, 
                                 bool do_append/*,                                   bool do_append, 
                                 bool exclusive*/) {                                  bool do_block, 
         const char *fname=file_spec.cstr(String::UL_FILE_SPEC);                                  bool fail_on_lock_problem) {
           const char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
         int f;          int f;
         if(access(fname, W_OK)!=0) // no          if(access(fname, W_OK)!=0) // no
                 create_dir_for_file(file_spec);                  create_dir_for_file(file_spec); 
   
         if((f=open(fname,           if((f=open(fname, 
                 O_CREAT|O_RDWR                  O_CREAT|O_RDWR
                 |(as_text?_O_TEXT:_O_BINARY)                  |(as_text?_O_TEXT:_O_BINARY)
                 |(do_append?O_APPEND:O_TRUNC), 0664))>=0) {                  |(do_append?O_APPEND:PA_O_TRUNC), 0664))>=0) {
                 /*if(exclusive)                  if((do_block?lock_exclusive_blocking(f):lock_exclusive_nonblocking(f))!=0) {
                         flock(f, LOCK_EX);*/                          Exception e("file.lock", 
                                   &file_spec, 
                                   "shared lock failed: %s (%d), actual filename '%s'", 
                                   strerror(errno), errno, fname); 
                           close(f); 
                           if(fail_on_lock_problem)
                                   throw e;
                           return false;
                   }
   
                   try {
                           action(f, context); 
                   } catch(...) {
   #ifdef HAVE_FTRUNCATE
                           if(!do_append)
                                   ftruncate(f, lseek(f, 0, SEEK_CUR)); // one can not use O_TRUNC, read lower
   #endif
                           unlock(f);close(f); 
                           /*re*/throw;
                   }
                                   
                 if(size) write(f, data, size);  #ifdef HAVE_FTRUNCATE
 #if O_TRUNC==0                  if(!do_append)
                 ftruncate(f, size);                          ftruncate(f, lseek(f, 0, SEEK_CUR)); // O_TRUNC truncates even exclusevely write-locked file [thanks to Igor Milyakov <virtan@rotabanner.com> for discovering]
 #endif  #endif
                 /*if(exclusive)                  unlock(f);close(f); 
                         flock(f, LOCK_UN);*/                  return true;
                 close(f);  
         } else          } else
                 throw Exception(0, 0,                   throw Exception(errno==EACCES?"file.access":0, 
                         &file_spec,                           &file_spec, 
                         "write failed: %s (%d), actual filename '%s'",                           "%s failed: %s (%d), actual filename '%s'", 
                                 strerror(errno), errno, fname);                                  action_name, strerror(errno), errno, fname); 
           // here should be nothing, see rethrow above
   }
   
   #ifndef DOXYGEN
   struct File_write_action_info {
           const void *data; size_t size;
   }; 
   #endif
   static void file_write_action(int f, void *context) {
           File_write_action_info& info=*static_cast<File_write_action_info *>(context); 
           if(info.size) {
                   int written=write(f, info.data, info.size); 
                   if(written<0)
                           throw Exception(0, 
                                   0, 
                                   "write failed: %s (%d)",  strerror(errno), errno); 
           }
   }
   void file_write(
                                   const String& file_spec, 
                                   const void *data, size_t size, 
                                   bool as_text, 
                                   bool do_append) {
           File_write_action_info info={data, size}; 
           file_write_action_under_lock(
                                   file_spec, 
                                   "write", file_write_action, &info, 
                                   as_text, 
                                   do_append); 
 }  }
   
 // throws nothing! [this is required in file_move & file_delete]  // throws nothing! [this is required in file_move & file_delete]
 static void rmdir(const String& file_spec, size_t pos_after) {  static void rmdir(const String& file_spec, size_t pos_after) {
         int pos_before;          int pos_before;
         if((pos_before=file_spec.pos("/", 1, pos_after))>=0)          if((pos_before=file_spec.pos("/", 1, pos_after))>=0)
                 rmdir(file_spec, pos_before+1);                  rmdir(file_spec, pos_before+1); 
                   
         rmdir(file_spec.mid(0, pos_after-1/* / */).cstr(String::UL_FILE_SPEC));          rmdir(file_spec.mid(0, pos_after-1/* / */).cstr(String::UL_FILE_SPEC)); 
 }  }
 void file_delete(Pool& pool, const String& file_spec) {  bool file_delete(const String& file_spec, bool fail_on_read_problem) {
         const char *fname=file_spec.cstr(String::UL_FILE_SPEC);          const char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
         if(unlink(fname)!=0)          if(unlink(fname)!=0)
                 throw Exception(0, 0,                   if(fail_on_read_problem)
                         &file_spec,                           throw Exception(errno==EACCES?"file.access":errno==ENOENT?"file.missing":0, 
                         "unlink failed: %s (%d), actual filename '%s'",                                   &file_spec, 
                                 strerror(errno), errno, fname);                                  "unlink failed: %s (%d), actual filename '%s'", 
                                           strerror(errno), errno, fname); 
                   else
                           return false;
   
         rmdir(file_spec, 1);          rmdir(file_spec, 1); 
           return true;
 }  }
 void file_move(Pool& pool, const String& old_spec, const String& new_spec) {  void file_move(const String& old_spec, const String& new_spec) {
         const char *old_spec_cstr=old_spec.cstr(String::UL_FILE_SPEC);          const char* old_spec_cstr=old_spec.cstr(String::UL_FILE_SPEC); 
         const char *new_spec_cstr=new_spec.cstr(String::UL_FILE_SPEC);          const char* new_spec_cstr=new_spec.cstr(String::UL_FILE_SPEC); 
                   
         create_dir_for_file(new_spec);          create_dir_for_file(new_spec); 
   
         if(rename(old_spec_cstr, new_spec_cstr)!=0)          if(rename(old_spec_cstr, new_spec_cstr)!=0)
                 throw Exception(0, 0,                   throw Exception(errno==EACCES?"file.access":errno==ENOENT?"file.missing":0, 
                         &old_spec,                           &old_spec, 
                         "rename failed: %s (%d), actual filename '%s' to '%s'",                           "rename failed: %s (%d), actual filename '%s' to '%s'", 
                                 strerror(errno), errno, old_spec_cstr, new_spec_cstr);                                  strerror(errno), errno, old_spec_cstr, new_spec_cstr); 
   
         rmdir(old_spec, 1);          rmdir(old_spec, 1); 
 }  }
   
   
   bool entry_exists(const char* fname, struct stat *afinfo) {
           struct stat lfinfo;
           bool result=stat(fname, &lfinfo)==0;
           if(afinfo)
                   *afinfo=lfinfo;
           return result;
   }
   
   bool entry_exists(const String& file_spec) {
           const char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
           return entry_exists(fname, 0); 
   }
   
 static bool entry_readable(const String& file_spec, bool need_dir) {  static bool entry_readable(const String& file_spec, bool need_dir) {
     const char *fname=file_spec.cstr(String::UL_FILE_SPEC);      char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
           if(need_dir) {
                   size_t size=strlen(fname); 
                   while(size) {
                           char c=fname[size-1]; 
                           if(c=='/' || c=='\\')
                                   fname[--size]=0;
                           else
                                   break;
                   }
           }
         struct stat finfo;          struct stat finfo;
         if(access(fname, R_OK)==0 && stat(fname, &finfo)==0) {          if(access(fname, R_OK)==0 && entry_exists(fname, &finfo)) {
                 bool is_dir=finfo.st_mode&S_IFDIR != 0;                  bool is_dir=(finfo.st_mode&S_IFDIR) != 0;
                 return is_dir==need_dir;                  return is_dir==need_dir;
         }          }
         return false;          return false;
 }  }
 bool file_readable(const String& file_spec) {  bool file_readable(const String& file_spec) {
         return entry_readable(file_spec, false);          return entry_readable(file_spec, false); 
 }  }
 bool dir_readable(const String& file_spec) {  bool dir_readable(const String& file_spec) {
         return entry_readable(file_spec, true);          return entry_readable(file_spec, true); 
 }  }
 String *file_readable(const String& path, const String& name) {  String *file_readable(const String& path, const String& name) {
         String *result=new(path.pool()) String(path);          String *result=new(path.pool()) String(path); 
         *result << "/";          *result << "/"; 
         *result << name;          *result << name;
         return file_readable(*result)?result:0;          return file_readable(*result)?result:0;
 }  }
Line 231  bool file_executable(const String& file_ Line 676  bool file_executable(const String& file_
   
 bool file_stat(const String& file_spec,   bool file_stat(const String& file_spec, 
                            size_t& rsize,                              size_t& rsize, 
                            time_t& ratime,                             time_t& ratime, 
                            time_t& rmtime,                             time_t& rmtime, 
                            time_t& rctime,                             time_t& rctime, 
                            bool fail_on_read_problem) {                             bool fail_on_read_problem) {
         Pool& pool=file_spec.pool();          Pool& pool=file_spec.pool(); 
         const char *fname=file_spec.cstr(String::UL_FILE_SPEC);          const char* fname=file_spec.cstr(String::UL_FILE_SPEC); 
     struct stat finfo;      struct stat finfo;
         if(stat(fname, &finfo)!=0)          if(stat(fname, &finfo)!=0)
                 if(fail_on_read_problem)                  if(fail_on_read_problem)
                         throw Exception(0, 0,                           throw Exception("file.missing", 
                                 &file_spec,                                   &file_spec, 
                                 "getting file size failed: %s (%d), real filename '%s'",                                   "getting file size failed: %s (%d), real filename '%s'", 
                                         strerror(errno), errno, fname);                                          strerror(errno), errno, fname); 
                 else                  else
                         return false;                          return false;
         rsize=finfo.st_size;          rsize=finfo.st_size;
Line 253  bool file_stat(const String& file_spec, Line 698  bool file_stat(const String& file_spec,
         return true;          return true;
 }  }
   
 char *getrow(char **row_ref, char delim) {  char* getrow(char* *row_ref, char delim) {
     char *result=*row_ref;      char* result=*row_ref;
     if(result) {      if(result) {
                 *row_ref=strchr(result, delim);                  *row_ref=strchr(result, delim); 
                 if(*row_ref)                   if(*row_ref) 
                         *((*row_ref)++)=0;                           *((*row_ref)++)=0; 
                 else if(!*result)                   else if(!*result) 
Line 265  char *getrow(char **row_ref, char delim) Line 710  char *getrow(char **row_ref, char delim)
     return result;      return result;
 }  }
   
 char *lsplit(char *string, char delim) {  char* lsplit(char* string, char delim) {
     if(string) {      if(string) {
                 char *v=strchr(string, delim);                  char* v=strchr(string, delim); 
                 if(v) {                  if(v) {
                         *v=0;                          *v=0;
                         return v+1;                          return v+1;
Line 276  char *lsplit(char *string, char delim) { Line 721  char *lsplit(char *string, char delim) {
     return 0;      return 0;
 }  }
   
 char *lsplit(char **string_ref, char delim) {  char* lsplit(char* *string_ref, char delim) {
     char *result=*string_ref;      char* result=*string_ref;
         char *next=lsplit(*string_ref, delim);          char* next=lsplit(*string_ref, delim); 
     *string_ref=next;      *string_ref=next;
     return result;      return result;
 }  }
   
 char *rsplit(char *string, char delim) {  char* rsplit(char* string, char delim) {
     if(string) {      if(string) {
                 char *v=strrchr(string, delim);                  char* v=strrchr(string, delim); 
                 if(v) {                  if(v) {
                         *v=0;                          *v=0;
                         return v+1;                          return v+1;
Line 295  char *rsplit(char *string, char delim) { Line 740  char *rsplit(char *string, char delim) {
 }  }
   
 /// @todo less stupid type detection  /// @todo less stupid type detection
 char *format(Pool& pool, double value, char *fmt) {  char* format(Pool& pool, double value, char* fmt) {
         char *result=(char *)pool.malloc(MAX_NUMBER, 4);          char local_buf[MAX_NUMBER]; 
           size_t size;
           
         if(fmt)          if(fmt)
                 if(strpbrk(fmt, "diouxX"))                  if(strpbrk(fmt, "diouxX"))
                         if(strpbrk(fmt, "ouxX"))                          if(strpbrk(fmt, "ouxX"))
                                 snprintf(result, MAX_NUMBER, fmt, (uint)value );                                  size=snprintf(local_buf, sizeof(local_buf), fmt, (uint)value); 
                         else                          else
                                 snprintf(result, MAX_NUMBER, fmt, (int)value );                                  size=snprintf(local_buf, sizeof(local_buf), fmt, (int)value); 
                 else                  else
                         snprintf(result, MAX_NUMBER, fmt, value);                          size=snprintf(local_buf, sizeof(local_buf), fmt, value); 
         else          else
                 snprintf(result, MAX_NUMBER, "%d", (int)value);                  size=snprintf(local_buf, sizeof(local_buf), "%d", (int)value); 
                   
         return result;          char* pool_buf=(char *)pool.malloc(size+1, 4); 
           memcpy(pool_buf, local_buf, size+1); 
           return pool_buf;
 }  }
   
 size_t stdout_write(const void *buf, size_t size) {  size_t stdout_write(const void *buf, size_t size) {
 #ifdef WIN32  #ifdef WIN32
         do{          do{
                 int chunk_written=fwrite(buf, 1, min(8*0x400, size), stdout);                  int chunk_written=fwrite(buf, 1, min(8*0x400, size), stdout); 
                 if(chunk_written<=0)                  if(chunk_written<=0)
                         break;                          break;
                 size-=chunk_written;                  size-=chunk_written;
                 buf=((const char*)buf)+chunk_written;                  buf=((const char*)buf)+chunk_written;
         } while(size>0);          } while(size>0); 
   
         return size;          return size;
 #else  #else
         return fwrite(buf, 1, size, stdout);          return fwrite(buf, 1, size, stdout); 
 #endif  #endif
 }  }
   
 char *unescape_chars(Pool& pool, const char *cp, int len) {  char* unescape_chars(Pool& pool, const char* cp, int len) {
         char *s=(char *)pool.malloc(len + 1, 5);          char* s=(char *)pool.malloc(len + 1, 5); 
         enum EscapeState {          enum EscapeState {
                 EscapeRest,                   EscapeRest, 
                 EscapeFirst,                   EscapeFirst, 
Line 338  char *unescape_chars(Pool& pool, const c Line 787  char *unescape_chars(Pool& pool, const c
         int srcPos=0;          int srcPos=0;
         int dstPos=0;          int dstPos=0;
         while(srcPos < len) {          while(srcPos < len) {
                 int ch=cp[srcPos];                  int ch=cp[srcPos]; 
                 switch(escapeState) {                  switch(escapeState) {
                         case EscapeRest:                          case EscapeRest:
                         if(ch=='%') {                          if(ch=='%') {
                                 escapeState=EscapeFirst;                                  escapeState=EscapeFirst;
                         } else if(ch=='+') {                          } else if(ch=='+') {
                                 s[dstPos++]=' ';                                  s[dstPos++]=' '; 
                         } else {                          } else {
                                 s[dstPos++]=ch;                                   s[dstPos++]=ch; 
                         }                          }
Line 354  char *unescape_chars(Pool& pool, const c Line 803  char *unescape_chars(Pool& pool, const c
                         escapeState=EscapeSecond;                          escapeState=EscapeSecond;
                         break;                          break;
                         case EscapeSecond:                          case EscapeSecond:
                         escapedValue +=hex_value[ch];                          escapedValue +=hex_value[ch]; 
                         s[dstPos++]=escapedValue;                          s[dstPos++]=escapedValue;
                         escapeState=EscapeRest;                          escapeState=EscapeRest;
                         break;                          break;
                 }                  }
                 srcPos++;                  srcPos++; 
         }          }
         s[dstPos]=0;          s[dstPos]=0;
         return s;          return s;
 }  }
   
 /// used by attributed_meaning_to_string / append_attribute_subattribute  
 struct Attributed_meaning_info {  
         String *header; // header line being constructed  
         String::Untaint_lang lang; // language in which to append to that line  
 };  
 static void append_attribute_subattribute(const Hash::Key& akey, Hash::Val *avalue,   
                                                                                   void *info) {  
         if(akey==VALUE_NAME)  
                 return;  
   
         Attributed_meaning_info& ami=*static_cast<Attributed_meaning_info *>(info);  
   
         // ...; charset=windows1251  
         *ami.header << "; ";  
         ami.header->append(akey, ami.lang);  
         *ami.header << "=";  
         ami.header->append(static_cast<Value *>(avalue)->as_string(), ami.lang);  
 }  
 const String& attributed_meaning_to_string(Value& meaning,   
                                                                                    String::Untaint_lang lang) {  
         String &result=*new(meaning.pool()) String(meaning.pool());  
         if(Hash *hash=meaning.get_hash(0)) {  
                 // $value(value) $subattribute(subattribute value)  
                 if(Value *value=static_cast<Value *>(hash->get(*value_name)))  
                         result.append(value->as_string(), lang, true);  
   
                 Attributed_meaning_info attributed_meaning_info={  
                         &result,  
                         lang  
                 };  
                 hash->for_each(append_attribute_subattribute, &attributed_meaning_info);  
         } else // result value  
                 result.append(meaning.as_string(), lang, true);  
   
         return result;  
 }  
   
 #ifdef WIN32  #ifdef WIN32
 void back_slashes_to_slashes(char *s) {  void back_slashes_to_slashes(char* s) {
         if(s)          if(s)
                 for(; *s; s++)                  for(; *s; s++)
                         if(*s=='\\')                          if(*s=='\\')
                                 *s='/';                                  *s='/'; 
 }  }
 /*  /*
 void slashes_to_back_slashes(char *s) {  void slashes_to_back_slashes(char* s) {
         if(s)          if(s)
                 for(; *s; s++)                  for(; *s; s++)
                         if(*s=='/')                          if(*s=='/')
                                 *s='\\';                                  *s='\\'; 
 }  }
 */  */
 #endif  #endif
   
 bool StrEqNc(const char *s1, const char *s2, bool strict) {  bool StrEqNc(const char* s1, const char* s2, bool strict) {
         while(true) {          while(true) {
                 if(!(*s1)) {                  if(!(*s1)) {
                         if(!(*s2))                          if(!(*s2))
Line 433  bool StrEqNc(const char *s1, const char Line 845  bool StrEqNc(const char *s1, const char
                                 return false;                                  return false;
                 } else if((*s1) !=(*s2))                  } else if((*s1) !=(*s2))
                         return false;                          return false;
                 s1++;                  s1++; 
                 s2++;                  s2++; 
         }          }
 }  }
   
 static bool isLeap(int year) {  static bool isLeap(int year) {
     return !(      return !(
              (year % 4) || ((year % 400) && !(year % 100))               (year % 4) || ((year % 400) && !(year % 100))
             );              ); 
 }  }
   
 int getMonthDays(int year, int month) {  int getMonthDays(int year, int month) {
     int monthDays[]={      int monthDays[]={
         31,          31, 
         isLeap(year) ? 29 : 28,          isLeap(year) ? 29 : 28, 
         31,          31, 
         30,          30, 
         31,          31, 
         30,          30, 
         31,          31, 
         31,          31, 
         30,          30, 
         31,          31, 
         30,          30, 
         31          31
     };      }; 
     return monthDays[month];      return monthDays[month]; 
 }  }
   
 void remove_crlf(char *start, char *end) {  void remove_crlf(char* start, char* end) {
         for(char *p=start; p<end; p++)          for(char* p=start; p<end; p++)
                 switch(*p) {                  switch(*p) {
                         case '\n': *p='|'; break;                          case '\n': *p='|';  break;
                         case '\r': *p=' '; break;                          case '\r': *p=' ';  break;
                 }                  }
 }  }
   
   
 /// must be last in this file  /// must be last in this file
 #undef vsnprintf  #undef vsnprintf
 int __vsnprintf(char *b, size_t s, const char *f, va_list l) {  int __vsnprintf(char* b, size_t s, const char* f, va_list l) {
         if(!s)          if(!s)
                 return 0;                  return 0;
   
Line 488  int __vsnprintf(char *b, size_t s, const Line 900  int __vsnprintf(char *b, size_t s, const
   
           if the number of bytes to write exceeds buffer, then count bytes are written and –1 is returned            if the number of bytes to write exceeds buffer, then count bytes are written and –1 is returned
         */          */
         r=_vsnprintf(b, s, f, l);          r=_vsnprintf(b, s, f, l); 
         if(r<0)           if(r<0) 
                 r=s;                  r=s;
 #else  #else
         r=vsnprintf(b, s, f, l);          r=vsnprintf(b, s, f, l); 
         /*          /*
         solaris:           solaris: 
         man vsnprintf          man vsnprintf
Line 513  int __vsnprintf(char *b, size_t s, const Line 925  int __vsnprintf(char *b, size_t s, const
         return r;          return r;
 }  }
   
 int __snprintf(char *b, size_t s, const char *f, ...) {  int __snprintf(char* b, size_t s, const char* f, ...) {
         va_list l;          va_list l;
     va_start(l, f);      va_start(l, f); 
     int r=__vsnprintf(b, s, f, l);      int r=__vsnprintf(b, s, f, l); 
     va_end(l);      va_end(l); 
         return r;          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
   }
   
   
   // attributed meaning
   
   static size_t date_attribute(const VDate& vdate, char *buf, size_t buf_size) {
       const char month_names[12][4]={
                   "Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"};
       const char days[7][4]={
                   "Sun","Mon","Tue","Wed","Thu","Fri","Sat"};
   
           time_t when=vdate.get_time();
           struct tm *tms=gmtime(&when);
           if(!tms)
                   throw Exception(0,
                           0,
                           "bad time in attribute value (seconds from epoch=%ld)", when);
           return snprintf(buf, MAX_STRING, "%s, %.2d-%s-%.4d %.2d:%.2d:%.2d GMT", 
                   days[tms->tm_wday],
                   tms->tm_mday,month_names[tms->tm_mon],tms->tm_year+1900,
                   tms->tm_hour,tms->tm_min,tms->tm_sec);
   }
   static void append_attribute_meaning(String& result,
                                                                            Value& value, String::Untaint_lang lang, bool forced) {
           if(const String *string=value.get_string())
                   result.append(string->join_chains(result.pool(), 0), lang, forced);
           else
                   if(Value *vdate=value.as(VDATE_TYPE, false)) {
                           char *buf=(char *)result.malloc(MAX_STRING);
                           size_t size=date_attribute(*static_cast<VDate *>(vdate), 
                                   buf, MAX_STRING);
   
                           result.APPEND_CLEAN(buf, size, "converted from date", 0);
                   } else
                           throw Exception("parser.runtime",
                                   &result,
                                   "trying to append here neither string nor date (%s)",
                                           value.type());
   }
   #ifndef DOXYGEN
   struct Attributed_meaning_info {
           String *header; // header line being constructed
           String::Untaint_lang lang; // language in which to append to that line
           bool forced; // do they force that lang?
   };
   #endif
   static void append_attribute_subattribute(const Hash::Key& akey, Hash::Val *avalue, 
                                                                                     void *info) {
           if(akey==VALUE_NAME)
                   return;
   
           Attributed_meaning_info& ami=*static_cast<Attributed_meaning_info *>(info);
   
           // ...; charset=windows1251
           *ami.header << "; ";
           ami.header->append(akey, ami.lang, ami.forced);
           *ami.header << "=";
           append_attribute_meaning(*ami.header, *static_cast<Value *>(avalue), ami.lang, ami.forced);
   }
   const String& attributed_meaning_to_string(Value& meaning, 
                                                                                      String::Untaint_lang lang, bool forced) {
           String &result=*new(meaning.pool()) String(meaning.pool());
           if(Hash *hash=meaning.get_hash(0)) {
                   // $value(value) $subattribute(subattribute value)
                   if(Value *value=static_cast<Value *>(hash->get(*value_name)))
                           append_attribute_meaning(result, *value, lang, forced);
   
                   Attributed_meaning_info attributed_meaning_info={&result, lang};
                   hash->for_each(append_attribute_subattribute, &attributed_meaning_info);
           } else // result value
                   append_attribute_meaning(result, meaning, lang, forced);
   
           return result;
   }

Removed from v.1.92  
changed lines
  Added in v.1.140


E-mail: