[BACK]Return to pa_http.C CVS log [TXT] [DIR] Up to [parser3project] / parser3 / src / main

Diff for /parser3/src/main/pa_http.C between versions 1.64 and 1.89

version 1.64, 2015/04/06 22:27:26 version 1.89, 2020/10/12 21:15:07
Line 1 Line 1
 /** @file  /** @file
         Parser: http support functions.          Parser: http support functions.
   
         Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com)          Copyright (c) 2001-2017 Art. Lebedev Studio (http://www.artlebedev.com)
         Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)          Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
  */   */
   
 #include "pa_http.h"  #include "pa_http.h"
 #include "pa_common.h"  #include "pa_common.h"
   #include "pa_base64.h"
 #include "pa_charsets.h"  #include "pa_charsets.h"
 #include "pa_request_charsets.h"  #include "pa_request_charsets.h"
 #include "pa_request.h"  #include "pa_request.h"
Line 17  volatile const char * IDENT_PA_HTTP_C="$ Line 18  volatile const char * IDENT_PA_HTTP_C="$
   
 #ifdef _MSC_VER  #ifdef _MSC_VER
 #include <windows.h>  #include <windows.h>
   #define socklen_t int
 #else  #else
 #define closesocket close  #define closesocket close
 #endif  #endif
Line 32  volatile const char * IDENT_PA_HTTP_C="$ Line 34  volatile const char * IDENT_PA_HTTP_C="$
 #define HTTP_ANY_STATUS_NAME    "any-status"  #define HTTP_ANY_STATUS_NAME    "any-status"
 #define HTTP_OMIT_POST_CHARSET_NAME     "omit-post-charset"     // ^file::load[...;http://...;$.method[post]] by default adds charset to content-type  #define HTTP_OMIT_POST_CHARSET_NAME     "omit-post-charset"     // ^file::load[...;http://...;$.method[post]] by default adds charset to content-type
   
 #define HTTP_TABLES_NAME "tables"  
   
 #define HTTP_USER "user"  #define HTTP_USER "user"
 #define HTTP_PASSWORD "password"  #define HTTP_PASSWORD "password"
   
   #define HTTP_USER_AGENT "user-agent"
 #define DEFAULT_USER_AGENT "parser3"  #define DEFAULT_USER_AGENT "parser3"
   
 #ifndef INADDR_NONE  #ifndef INADDR_NONE
Line 48  volatile const char * IDENT_PA_HTTP_C="$ Line 49  volatile const char * IDENT_PA_HTTP_C="$
   
 // helpers  // helpers
   
   bool HTTP_Headers::add_header(const char *line){
           const char *value=strchr(line, ':');
   
           if(value && value != line){ // we need only headers, not the response code
                   Header header(str_upper(line, value-line), String::Body(value+1).trim(String::TRIM_BOTH, " \t\n\r"));
   
                   if(header.name == String::Body(HTTP_CONTENT_TYPE_UPPER) && content_type.is_empty())
                           content_type=header.value;
   
                   if(header.name == String::Body("CONTENT-LENGTH") && content_length==0)
                           content_length=pa_atoul(header.value.cstr(), 10);
   
                   headers+=header;
   
                   return true;
           }
           return false;
   }
   
 class Cookies_table_template_columns: public ArrayString {  class Cookies_table_template_columns: public ArrayString {
 public:  public:
         Cookies_table_template_columns() {          Cookies_table_template_columns() {
Line 68  static bool set_addr(struct sockaddr_in Line 88  static bool set_addr(struct sockaddr_in
         addr->sin_family=AF_INET;          addr->sin_family=AF_INET;
         addr->sin_port=htons(port);           addr->sin_port=htons(port); 
         if(host) {          if(host) {
                 ulong packed_ip=inet_addr(host);                  struct hostent *hostIP=gethostbyname(host);
                 if(packed_ip!=INADDR_NONE)                  if(hostIP && hostIP->h_addrtype == AF_INET){
                         memcpy(&addr->sin_addr, &packed_ip, sizeof(packed_ip));                           memcpy(&addr->sin_addr, hostIP->h_addr, hostIP->h_length);
                 else {                          return true;
                         struct hostent *hostIP=gethostbyname(host);                  }
                         if(hostIP)           }
                                 memcpy(&addr->sin_addr, hostIP->h_addr, hostIP->h_length);           return false;
                         else  
                                 return false;  
                 }   
         } else   
                 addr->sin_addr.s_addr=INADDR_ANY;  
         return true;  
 }  
   
 size_t guess_content_length(char* buf) {  
         char* ptr;  
         if((ptr=strstr(buf, "Content-Length:"))) // Apache  
                 goto found;  
         if((ptr=strstr(buf, "content-length:"))) // Parser 3 before 3.4.0  
                 goto found;  
         if((ptr=strstr(buf, "Content-length:"))) // maybe 1  
                 goto found;  
         if((ptr=strstr(buf, "CONTENT-LENGTH:"))) // maybe 2  
                 goto found;  
         return 0;  
 found:  
         char *error_pos;  
         size_t result=(size_t)strtol(ptr+15/*strlen("Content-Length:")*/, &error_pos, 0);  
           
         const size_t reasonable_initial_max=0x400*0x400*10 /*10M*/;  
         if(result>reasonable_initial_max) // sanity check  
                 return reasonable_initial_max;  
         return 0;//result;  
 }  }
   
 static int http_read_response(char*& response, size_t& response_size, int sock, bool fail_on_status_ne_200) {  class HTTP_response : public PA_Allocated {
         int result=0;  public:
         // fetching some to local buffer, guessing on possible Content-Length          char *buf;
         response_size=0x400*20; // initial size if Content-Length could not be determined                 size_t length;
         const size_t preview_size=0x400*20;          size_t buf_size;
         char preview_buf[preview_size+1/*terminator*/];  // 20K buffer to preview headers          size_t body_offset;
         ssize_t received_size=recv(sock, preview_buf, preview_size, 0);   
         if(received_size==0)          HTTP_Headers headers;
                 goto done;          const String &url;
         if(received_size<0) {  
                 if(int no=pa_socks_errno())          HTTP_response(const String& aurl) : buf(NULL), length(0), buf_size(0), body_offset(0), url(aurl){}
                         throw Exception("http.timeout",   
                                 0,           void resize(size_t size){
                                 "error receiving response header: %s (%d)", pa_socks_strerr(no), no);                   buf_size=size;
                 goto done;                  buf=(char *)pa_realloc(buf, size + 1);
         }  
         // terminator [helps futher string searches]  
         preview_buf[received_size]=0;   
         // checking status  
         if(char* EOLat=strstr(preview_buf, "\n")) {   
                 const String status_line(pa_strdup(preview_buf, EOLat-preview_buf));  
                 ArrayString astatus;   
                 size_t pos_after=0;  
                 status_line.split(astatus, pos_after, " ");   
                 const String& status_code=*astatus.get(astatus.count()>1?1:0);  
                 result=status_code.as_int();   
   
                 if(fail_on_status_ne_200 && result!=200)  
                         throw Exception("http.status",  
                                 &status_code,  
                                 "invalid HTTP response status");  
         }          }
         // detecting response_size  
         {          bool read(int sock, size_t size){
                 if(size_t content_length=guess_content_length(preview_buf))                  if(length+size>buf_size)
                         response_size=preview_size+content_length; // a little more than needed, will adjust response_size by actual received size later                          resize(buf_size*2 + size);
                   ssize_t received_size=recv(sock, buf + length, size, 0);
                   if(received_size==0)
                           return false;
                   if(received_size<0) {
                           if(int no=pa_socks_errno())
                                   throw Exception("http.timeout", &url, "error receiving response body: %s (%d)", pa_socks_strerr(no), no);
                           return false;
                   }
                   length+=received_size;
                   buf[length]='\0';
                   return true;
         }          }
   
         // [gcc is happier this way, see goto above]          size_t first_line(){
         {                  char *header=strchr(buf, '\n');
                 // allocating initial buf                  if(!header)
                 response=(char*)pa_malloc_atomic(response_size+1/*terminator*/); // just setting memory block type                          return false;
                 char* ptr=response;  
                 size_t todo_size=response_size;                  return header-buf;
                 // coping part of already received body          }
                 memcpy(ptr, preview_buf, received_size);  
                 ptr+=received_size;          const char *status_code(char *status_line, int &result){
                 todo_size-=received_size;                                 char* status_start = strchr(status_line, ' ');
   
                 // we use terminator byte for two purposes here:                  if(!(status_start++))
                 // 1. we return there zero always, not knowing: maybe they would want to create String form $file.body?                          return status_line;
                 //     invariant: all Strings should have zero-terminated buffers  
                 // 2. we use that out-of-size byte to detect if our Content-Length guess was wrong                  char* status_end=strchr(status_start, ' ');
                 //    when recv gets more than we expected  
                 //    a) we know that the Content-Length guess was wrong                  if(!status_end)
                 //    b) we have space to put the first byte of extra data                          return status_line;
                 //    c) we use less code to detect normal situation: on last while-cycle recv expected to just return 0  
                 while(true) {                  if(status_end==status_start)
                         received_size=recv(sock, ptr, todo_size+1/*there is always a place for terminator*/, 0);                           return status_line;
                         if(received_size==0) {  
                                 response_size-=todo_size; // in case we received less than expected, cut down the reported size                  const char *result_str=pa_strdup(status_start, status_end-status_start);
                   result=pa_atoui(result_str, 10);
                   return result_str;
           }
   
           bool body_start(){
                   char *p=buf;
                   while((p=strchr(p, '\n'))) {
                           if(p[1]=='\r' && p[2]=='\n'){  // \r\n\r\n
                                   *p='\0';
                                   body_offset=p-buf+3;
                                   return true;
                           }
                           if(p[1]=='\n') { // \n\n
                                   *p='\0';
                                   body_offset=p-buf+2;
                                   return true;
                           }
                           p++;
                   }
                   return false;
           }
   
           void parse_headers(){
                   const String header_block(buf, String::L_TAINTED);
                   
                   ArrayString aheaders;
                   header_block.split(aheaders, 0, "\n");
   
                   Array_iterator<const String*> i(aheaders);
                   i.next(); // skipping status
                   for(;i.has_next();){
                           const char *line=i.next()->cstr();
                           if(!headers.add_header(line))
                                   throw Exception("http.response", &url, "bad response from host - bad header \"%s\"", line);
                   }
           }
   
           int read_response(int sock, bool fail_on_status_ne_200);
   };
   
   enum HTTP_response_state {
           HTTP_STATUS_CODE,
           HTTP_HEADERS,
           HTTP_BODY
   };
   
   int HTTP_response::read_response(int sock, bool fail_on_status_ne_200) {
           HTTP_response_state state=HTTP_STATUS_CODE;
           int result=0;
   
           size_t chunk_size=0x400*16;
           resize(2*chunk_size);
   
           while(read(sock, chunk_size)){
                   switch(state){
                           case HTTP_STATUS_CODE: {
                                   size_t status_size=first_line();
                                   if(!status_size)
                                           break;
   
                                   const char *status=status_code(pa_strdup(buf, status_size), result);
   
                                   if(!result || fail_on_status_ne_200 && result!=200)
                                           throw Exception("http.status", status ? new String(status) : &String::Empty, "invalid HTTP response status");
   
                                   state=HTTP_HEADERS;
                           }
   
                           case HTTP_HEADERS: {
                                   if(!body_start())
                                           break;
   
                                   parse_headers();
   
                                   size_t content_length=check_file_size(headers.content_length, url);
                                   if(content_length>0 && (content_length + body_offset) > length){
                                           resize(content_length + body_offset + 0x400*64);
                                   }
   
                                   state=HTTP_BODY;
                                 break;                                  break;
                         }                          }
                         if(received_size<0) {  
                                 if(int no=pa_socks_errno())                          case HTTP_BODY: {
                                         throw Exception("http.timeout",                                   chunk_size=0x400*64;
                                                 0,   
                                                 "error receiving response body: %s (%d)", pa_socks_strerr(no), no);   
                                 break;                                  break;
                         }                          }
                         // they've touched the terminator?  
                         if((size_t)received_size>todo_size)  
                         {  
                                 // that means that our guessed response_size was not big enough  
                                 const size_t grow_chunk_size=0x400*0x400; // 1M  
                                 response_size+=grow_chunk_size;  
                                 size_t ptr_offset=ptr-response;  
                                 response=(char*)pa_realloc(response, response_size+1/*terminator*/);  
                                 ptr=response+ptr_offset;  
                                 todo_size+=grow_chunk_size;  
                         }  
                         // can't do this before realloc: we need <todo_size check  
                         ptr+=received_size;  
                         todo_size-=received_size;  
                 }                  }
         }          }
 done:  
         if(result)          if(state==HTTP_STATUS_CODE)
         {                  throw Exception("http.response", &url, "bad response from host - no status found (size=%u)", length);
                 response[response_size]=0;  
                 return result;          if(state==HTTP_HEADERS){
                   parse_headers();
                   body_offset=length;
         }          }
         else  
                 throw Exception("http.response",          return result;
                         0,  
                         "bad response from host - no status found (size=%u)", response_size);   
 }  }
   
 /* ********************** request *************************** */  /* ********************** request *************************** */
Line 215  static void timeout_handler(int /*sig*/) Line 267  static void timeout_handler(int /*sig*/)
 }  }
 #endif  #endif
   
 static int http_request(char*& response, size_t& response_size,  static int http_request(HTTP_response& response, const char* host, short port, const char* request, size_t request_size, int timeout_secs, bool fail_on_status_ne_200) {
                         const char* host, short port,   
                         const char* request, size_t request_size,  
                         int timeout_secs,  
                         bool fail_on_status_ne_200) {  
         if(!host)          if(!host)
                 throw Exception("http.host",                   throw Exception("http.host", 0, "zero hostname");  //never
                         0,   
                         "zero hostname");  //never  
   
         volatile // to prevent makeing it register variable, because it will be clobbered by longjmp [thanks gcc warning]          volatile // to prevent makeing it register variable, because it will be clobbered by longjmp [thanks gcc warning]
                 int sock=-1;                  int sock=-1;
Line 237  static int http_request(char*& response, Line 283  static int http_request(char*& response,
                 // rewritten simplier [athough duplicating closesocket code]                  // rewritten simplier [athough duplicating closesocket code]
                 if(sock>=0)                   if(sock>=0) 
                         closesocket(sock);                           closesocket(sock); 
                 throw Exception("http.timeout",                   throw Exception("http.timeout", 0, "timeout occurred while retrieving document"); 
                         0,   
                         "timeout occured while retrieving document");   
                 return 0; // never                  return 0; // never
         } else {          } else {
                 alarm(timeout_secs);                   alarm(timeout_secs); 
Line 249  static int http_request(char*& response, Line 293  static int http_request(char*& response,
                         struct sockaddr_in dest;                          struct sockaddr_in dest;
                                   
                         if(!set_addr(&dest, host, port))                          if(!set_addr(&dest, host, port))
                                 throw Exception("http.host",                                   throw Exception("http.host", 0, "can not resolve hostname \"%s\"", host); 
                                         0,   
                                         "can not resolve hostname \"%s\"", host);   
                                                   
                         if((sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/))<0) {                          if((sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/))<0) {
                                 int no=pa_socks_errno();                                  int no=pa_socks_errno();
                                 throw Exception("http.connect",                                   throw Exception("http.connect", 0, "can not make socket: %s (%d)", pa_socks_strerr(no), no); 
                                         0,   
                                         "can not make socket: %s (%d)", pa_socks_strerr(no), no);   
                         }                          }
   
                         // To enable SO_DONTLINGER (that is, disable SO_LINGER)                           // To enable SO_DONTLINGER (that is, disable SO_LINGER) 
Line 276  static int http_request(char*& response, Line 316  static int http_request(char*& response,
   
                         if(connect(sock, (struct sockaddr *)&dest, sizeof(dest))) {                          if(connect(sock, (struct sockaddr *)&dest, sizeof(dest))) {
                                 int no=pa_socks_errno();                                  int no=pa_socks_errno();
                                 throw Exception("http.connect",                                   throw Exception("http.connect", 0, "can not connect to host \"%s\": %s (%d)", host, pa_socks_strerr(no), no);
                                         0,   
                                         "can not connect to host \"%s\": %s (%d)", host, pa_socks_strerr(no), no);   
                         }                          }
   
                         if(send(sock, request, request_size, 0)!=(ssize_t)request_size) {                          if(send(sock, request, request_size, 0)!=(ssize_t)request_size) {
                                 int no=pa_socks_errno();                                  int no=pa_socks_errno();
                                 throw Exception("http.timeout",                                   throw Exception("http.timeout", 0, "error sending request: %s (%d)", pa_socks_strerr(no), no);
                                         0,   
                                         "error sending request: %s (%d)", pa_socks_strerr(no), no);   
                         }                          }
   
                         result=http_read_response(response, response_size, sock, fail_on_status_ne_200);                           result=response.read_response(sock, fail_on_status_ne_200);
                         closesocket(sock);                           closesocket(sock);
 #ifdef PA_USE_ALARM  #ifdef PA_USE_ALARM
                         alarm(0);                           alarm(0);
 #endif  #endif
                         return result;                          return result;
                 } catch(...) {                  } catch(...) {
 #ifdef PA_USE_ALARM  #ifdef PA_USE_ALARM
                         alarm(0);                           alarm(0);
 #endif  #endif
                         if(sock>=0)                           if(sock>=0)
                                 closesocket(sock);                                   closesocket(sock);
                         rethrow;                          rethrow;
                 }                  }
 #ifdef PA_USE_ALARM  #ifdef PA_USE_ALARM
Line 373  static const String* basic_authorization Line 409  static const String* basic_authorization
                 combined<<pass;                  combined<<pass;
                   
         String* result=new String("Basic ");          String* result=new String("Basic ");
         *result<<pa_base64_encode(combined.cstr(), combined.length());          *result<<pa_base64_encode(combined.cstr(), combined.length(), Base64Options(false /*no wrap*/));
         return result;          return result;
 }  }
   
 static void form_string_value2string(  static void form_string_value2string(HashStringValue::key_type key, const String& value, String& result) {
                                         HashStringValue::key_type key,   
                                         const String& value,   
                                         String& result)   
 {  
         result << String(key, String::L_URI) << "=" << String(value, String::L_URI) << "&";          result << String(key, String::L_URI) << "=" << String(value, String::L_URI) << "&";
 }  }
   
Line 397  struct Form_table_value2string_info { Line 429  struct Form_table_value2string_info {
 static void form_table_value2string(Table::element_type row, Form_table_value2string_info* info) {  static void form_table_value2string(Table::element_type row, Form_table_value2string_info* info) {
         form_string_value2string(info->key, *row->get(0), info->result);          form_string_value2string(info->key, *row->get(0), info->result);
 }  }
 static void form_value2string(  
                                         HashStringValue::key_type key,   static void form_value2string(HashStringValue::key_type key, HashStringValue::value_type value, String* result) {
                                         HashStringValue::value_type value,   
                                         String* result)   
 {  
         if(const String* svalue=value->get_string())          if(const String* svalue=value->get_string())
                 form_string_value2string(key, *svalue, *result);                  form_string_value2string(key, *svalue, *result);
         else if(Table* tvalue=value->get_table()) {          else if(Table* tvalue=value->get_table()) {
                 Form_table_value2string_info info(key, *result);                  Form_table_value2string_info info(key, *result);
                 tvalue->for_each(form_table_value2string, &info);                  tvalue->for_each(form_table_value2string, &info);
         } else          } else
                 throw Exception(PARSER_RUNTIME,                  throw Exception(PARSER_RUNTIME, new String(key, String::L_TAINTED),
                         new String(key, String::L_TAINTED),  
                         "is %s, " HTTP_FORM_NAME " option value can be string or table only (file is allowed for $." HTTP_METHOD_NAME "[POST] + $." HTTP_FORM_ENCTYPE_NAME "[" HTTP_CONTENT_TYPE_MULTIPART_FORMDATA "])", value->type());                          "is %s, " HTTP_FORM_NAME " option value can be string or table only (file is allowed for $." HTTP_METHOD_NAME "[POST] + $." HTTP_FORM_ENCTYPE_NAME "[" HTTP_CONTENT_TYPE_MULTIPART_FORMDATA "])", value->type());
 }  }
   
Line 462  struct FormPart { Line 490  struct FormPart {
         }          }
 };  };
   
 static void form_part_boundary_header(FormPart& part, String::Body name, const char* file_name=0){  static void form_part_boundary_header(FormPart& part, String::Body name, const char* file_name=0) {
         *part.string << "--" << part.boundary          *part.string << "--" << part.boundary << CRLF CONTENT_DISPOSITION_CAPITALIZED ": form-data; name=\"" << name << "\"";
                                 << CRLF CONTENT_DISPOSITION_CAPITALIZED ": form-data; name=\""   
                                 << name  
                                 << "\"";  
         if(file_name){          if(file_name){
                 if(strcmp(file_name, NONAME_DAT)!=0)                  if(strcmp(file_name, NONAME_DAT)!=0)
                         *part.string << "; filename=\"" << file_name << "\"";                          *part.string << "; filename=\"" << file_name << "\"";
Line 475  static void form_part_boundary_header(Fo Line 500  static void form_part_boundary_header(Fo
         *part.string << CRLF CRLF;          *part.string << CRLF CRLF;
 }  }
   
 static void form_string_value2part(  static void form_string_value2part(HashStringValue::key_type key, const String& value, FormPart& part) {
                                 HashStringValue::key_type key,  
                                 const String& value,  
                                 FormPart& part)  
 {  
         form_part_boundary_header(part, key);          form_part_boundary_header(part, key);
         *part.string << value << CRLF;          *part.string << value << CRLF;
 }  }
   
 static void form_file_value2part(  static void form_file_value2part(HashStringValue::key_type key, VFile& vfile, FormPart& part) {
                                 HashStringValue::key_type key,  
                                 VFile& vfile,    
                                 FormPart& part)  
 {  
         form_part_boundary_header(part, key, vfile.fields().get(name_name)->as_string().cstr());          form_part_boundary_header(part, key, vfile.fields().get(name_name)->as_string().cstr());
         part.blocks+=FormPart::BinaryBlock(part.string, part.r);          part.blocks+=FormPart::BinaryBlock(part.string, part.r);
         part.blocks+=FormPart::BinaryBlock(vfile.value_ptr(), vfile.value_size());          part.blocks+=FormPart::BinaryBlock(vfile.value_ptr(), vfile.value_size());
Line 500  static void form_table_value2part(Table: Line 517  static void form_table_value2part(Table:
         form_string_value2part(part->info->key, *row->get(0), *part);          form_string_value2part(part->info->key, *row->get(0), *part);
 }  }
   
 static void form_value2part(  static void form_value2part(HashStringValue::key_type key, HashStringValue::value_type value, FormPart& part) {
                                 HashStringValue::key_type key,  
                                 HashStringValue::value_type value,  
                                 FormPart& part)  
 {  
         if(const String* svalue=value->get_string())          if(const String* svalue=value->get_string())
                 form_string_value2part(key, *svalue, part);                  form_string_value2part(key, *svalue, part);
         else if(Table* tvalue=value->get_table()) {          else if(Table* tvalue=value->get_table()) {
Line 514  static void form_value2part( Line 527  static void form_value2part(
         } else if(VFile* vfile=static_cast<VFile *>(value->as("file"))){          } else if(VFile* vfile=static_cast<VFile *>(value->as("file"))){
                 form_file_value2part(key, *vfile, part);                  form_file_value2part(key, *vfile, part);
         } else          } else
                 throw Exception(PARSER_RUNTIME,                  throw Exception(PARSER_RUNTIME, new String(key, String::L_TAINTED), "is %s, " HTTP_FORM_NAME " option value can be string, table or file only", value->type());
                         new String(key, String::L_TAINTED),  
                         "is %s, " HTTP_FORM_NAME " option value can be string, table or file only", value->type());  
 }  }
   
 const char* pa_form2string_multipart(HashStringValue& form, Request& r, const char* boundary, size_t& post_size){  const char* pa_form2string_multipart(HashStringValue& form, Request& r, const char* boundary, size_t& post_size){
Line 527  const char* pa_form2string_multipart(Has Line 538  const char* pa_form2string_multipart(Has
         return formpart.post(post_size);          return formpart.post(post_size);
 }  }
   
 static void find_headers_end(char* p,  
                 char*& headers_end_at,  
                 char*& raw_body)  
 {  
         raw_body=p;  
         // \n\n  
         // \r\n\r\n  
         while((p=strchr(p, '\n'))) {  
                 headers_end_at=++p; // \n>.<  
                 if(*p=='\r')  // \r\n>\r?<\n  
                         p++;  
                 if(*p=='\n') { // \r\n\r>\n?<  
                         raw_body=p+1;  
                         return;                   
                 }  
         }  
         headers_end_at=0;  
 }  
   
 // Set-Cookie: name=value; Domain=docs.foo.com; Path=/accounts; Expires=Wed, 13-Jan-2021 22:23:01 GMT; Secure; HttpOnly  // Set-Cookie: name=value; Domain=docs.foo.com; Path=/accounts; Expires=Wed, 13-Jan-2021 22:23:01 GMT; Secure; HttpOnly
 static ArrayString* parse_cookie(Request& r, const String& cookie) {  static ArrayString* parse_cookie(Request& r, const String& cookie) {
         char *current=pa_strdup(cookie.cstr());          char *current=pa_strdup(cookie.cstr());
Line 622  Table* parse_cookies(Request& r, Table * Line 614  Table* parse_cookies(Request& r, Table *
         return &result;          return &result;
 }  }
   
   void tables_update(HashStringValue& tables, const String::Body name, const String& value){
           Table *table;
           if(Value *valready=tables.get(name)) {
                   // second+ appearence
                   table=valready->get_table();
           } else {
                   // first appearence
                   Table::columns_type columns=new ArrayString(1);
                   *columns+=new String("value");
                   table=new Table(columns);
                   tables.put(name, new VTable(table));
           }
           // this string becomes next row
           ArrayString& row=*new ArrayString(1);
           row+=&value;
           *table+=&row;
   }
   
 /// @todo build .cookies field. use ^file.tables.SET-COOKIES.menu{ for now  /// @todo build .cookies field. use ^file.tables.SET-COOKIES.menu{ for now
 File_read_http_result pa_internal_file_read_http(Request& r,  File_read_http_result pa_internal_file_read_http(Request& r, const String& file_spec, bool as_text, HashStringValue *options, bool transcode_text_result) {
                                                 const String& file_spec,  
                                                 bool as_text,  
                                                 HashStringValue *options,  
                                                 bool transcode_text_result) {  
         File_read_http_result result;          File_read_http_result result;
         char host[MAX_STRING];          char host[MAX_STRING];
           const char *idna_host;
         const char* uri;           const char* uri; 
         short port=80;          short port=80;
         const char* method="GET";          const char* method="GET";
Line 641  File_read_http_result pa_internal_file_r Line 648  File_read_http_result pa_internal_file_r
         Value* vheaders=0;          Value* vheaders=0;
         Value* vcookies=0;          Value* vcookies=0;
         Value* vbody=0;          Value* vbody=0;
         Charset *asked_remote_charset=0;          Charset* asked_remote_charset=0;
         Charset* real_remote_charset=0;          Charset* real_remote_charset=0;
         const char* user_cstr=0;          const char* user_cstr=0;
         const char* password_cstr=0;          const char* password_cstr=0;
Line 686  File_read_http_result pa_internal_file_r Line 693  File_read_http_result pa_internal_file_r
                         omit_post_charset=vomit_post_charset->as_bool();                          omit_post_charset=vomit_post_charset->as_bool();
                 }                  }
                 if(Value* vcharset_name=options->get(PA_CHARSET_NAME)) {                  if(Value* vcharset_name=options->get(PA_CHARSET_NAME)) {
                         asked_remote_charset=&charsets.get(vcharset_name->as_string().change_case(r.charsets.source(), String::CC_UPPER));                          asked_remote_charset=&pa_charsets.get(vcharset_name->as_string());
                 }                   } 
                 if(Value* vresponse_charset_name=options->get(PA_RESPONSE_CHARSET_NAME)) {                  if(Value* vresponse_charset_name=options->get(PA_RESPONSE_CHARSET_NAME)) {
                         valid_options++;                          valid_options++;
                         real_remote_charset=&charsets.get(vresponse_charset_name->as_string().change_case(r.charsets.source(), String::CC_UPPER));                          real_remote_charset=&pa_charsets.get(vresponse_charset_name->as_string());
                 }                   } 
                 if(Value* vuser=options->get(HTTP_USER)) {                  if(Value* vuser=options->get(HTTP_USER)) {
                         valid_options++;                          valid_options++;
Line 709  File_read_http_result pa_internal_file_r Line 716  File_read_http_result pa_internal_file_r
   
         if(encode){          if(encode){
                 if(method_is_get)                  if(method_is_get)
                         throw Exception(PARSER_RUNTIME,                          throw Exception(PARSER_RUNTIME, 0, "you can not use $." HTTP_FORM_ENCTYPE_NAME " option with method GET");
                                 0,  
                                 "you can not use $." HTTP_FORM_ENCTYPE_NAME " option with method GET");  
   
                 multipart=strcasecmp(encode, HTTP_CONTENT_TYPE_MULTIPART_FORMDATA)==0;                  multipart=strcasecmp(encode, HTTP_CONTENT_TYPE_MULTIPART_FORMDATA)==0;
   
                 if(!multipart && strcasecmp(encode, HTTP_CONTENT_TYPE_FORM_URLENCODED)!=0)                  if(!multipart && strcasecmp(encode, HTTP_CONTENT_TYPE_FORM_URLENCODED)!=0)
                         throw Exception(PARSER_RUNTIME,                          throw Exception(PARSER_RUNTIME, 0, "$." HTTP_FORM_ENCTYPE_NAME " option value can be " HTTP_CONTENT_TYPE_FORM_URLENCODED " or " HTTP_CONTENT_TYPE_MULTIPART_FORMDATA " only");
                                 0,  
                                 "$." HTTP_FORM_ENCTYPE_NAME " option value can be " HTTP_CONTENT_TYPE_FORM_URLENCODED " or " HTTP_CONTENT_TYPE_MULTIPART_FORMDATA " only");  
         }          }
   
         if(vbody){          if(vbody){
                 if(method_is_get)                  if(method_is_get)
                         throw Exception(PARSER_RUNTIME,                          throw Exception(PARSER_RUNTIME, 0, "you can not use $." HTTP_BODY_NAME " option with method GET");
                                 0,  
                                 "you can not use $." HTTP_BODY_NAME " option with method GET");  
   
                 if(form)                  if(form)
                         throw Exception(PARSER_RUNTIME,                          throw Exception(PARSER_RUNTIME, 0, "you can not use options $." HTTP_BODY_NAME " and $." HTTP_FORM_NAME " together");
                                 0,  
                                 "you can not use options $." HTTP_BODY_NAME " and $." HTTP_FORM_NAME " together");  
         }          }
   
         //preparing request          //preparing request
Line 746  File_read_http_result pa_internal_file_r Line 745  File_read_http_result pa_internal_file_r
   
                 const char* current=connect_string_cstr;                  const char* current=connect_string_cstr;
                 if(strncmp(current, "http://", 7)!=0)                  if(strncmp(current, "http://", 7)!=0)
                         throw Exception(PARSER_RUNTIME,                           throw Exception(PARSER_RUNTIME, &connect_string, "does not start with http://"); //never
                                 &connect_string,   
                                 "does not start with http://"); //never  
                 current+=7;                  current+=7;
   
                 strncpy(host, current, sizeof(host)-1);  host[sizeof(host)-1]=0;                  strncpy(host, current, sizeof(host)-1);  host[sizeof(host)-1]=0;
Line 763  File_read_http_result pa_internal_file_r Line 760  File_read_http_result pa_internal_file_r
                                 throw Exception(PARSER_RUNTIME, &connect_string, "invalid port number '%s'", port_cstr);                                  throw Exception(PARSER_RUNTIME, &connect_string, "invalid port number '%s'", port_cstr);
                 }                  }
   
                   idna_host=pa_idna_encode(host, r.charsets.source());
   
                 // making request head                  // making request head
                 String head;                  String head;
                 head << method << " " << uri;                  head << method << " " << uri;
                 if(method_is_get && form)                  if(method_is_get && form)
                         head << (strchr(uri, '?')!=0?"&":"?") << pa_form2string(*form, r.charsets);                          head << (strchr(uri, '?')!=0?"&":"?") << pa_form2string(*form, r.charsets);
   
                 head <<" HTTP/1.0" CRLF "Host: "<< host;                  head <<" HTTP/1.0" CRLF "Host: "<< idna_host;
                 if (port != 80)                  if (port != 80)
                         head << ":" << port_cstr;                          head << ":" << port_cstr;
                 head << CRLF;                  head << CRLF;
   
                 char* boundary=0;                  char* boundary= multipart ? get_uuid_boundary() : 0;
   
                 if(multipart){  
                         uuid uuid=get_uuid();  
                         const int boundary_bufsize=10+32+1/*for zero-teminator*/+1/*for faulty snprintfs*/;  
                         boundary=new(PointerFreeGC) char[boundary_bufsize];  
                         snprintf(boundary, boundary_bufsize,  
                                 "----------%08X%04X%04X%02X%02X%02X%02X%02X%02X%02X%02X",  
                                 uuid.time_low, uuid.time_mid, uuid.time_hi_and_version,  
                                 uuid.clock_seq >> 8, uuid.clock_seq & 0xFF,  
                                 uuid.node[0], uuid.node[1], uuid.node[2],  
                                 uuid.node[3], uuid.node[4], uuid.node[5]);  
                 }  
   
                 String user_headers;                  String user_headers;
                 bool user_agent_specified=false;                  bool user_agent_specified=false;
Line 802  File_read_http_result pa_internal_file_r Line 789  File_read_http_result pa_internal_file_r
                                         &content_type_url_encoded};                                          &content_type_url_encoded};
                                 headers->for_each<Http_pass_header_info*>(http_pass_header, &info);                                   headers->for_each<Http_pass_header_info*>(http_pass_header, &info); 
                         } else                          } else
                                 throw Exception(PARSER_RUNTIME,                                   throw Exception(PARSER_RUNTIME, 0, "headers param must be hash"); 
                                         0,  
                                         "headers param must be hash");   
                 };                  };
   
                 const char* request_body=0;                  const char* request_body=0;
Line 830  File_read_http_result pa_internal_file_r Line 815  File_read_http_result pa_internal_file_r
                                 request_body=vbody->as_string().untaint_and_transcode_cstr(String::L_URI, &(r.charsets));                                  request_body=vbody->as_string().untaint_and_transcode_cstr(String::L_URI, &(r.charsets));
                         } else {                          } else {
                                 // content-type != application/x-www-form-urlencoded -> transcode only, don't url-encode!                                  // content-type != application/x-www-form-urlencoded -> transcode only, don't url-encode!
                                 request_body=Charset::transcode(                                  const String &sbody=vbody->as_string();
                                         String::C(vbody->as_string().cstr(), vbody->as_string().length()),                                  request_body=Charset::transcode(String::C(sbody.cstr(), sbody.length()), r.charsets.source(), *asked_remote_charset).str;
                                         r.charsets.source(),  
                                         *asked_remote_charset  
                                 );  
                         }                          }
                         post_size=strlen(request_body);                          post_size=strlen(request_body);
                 }                  }
Line 849  File_read_http_result pa_internal_file_r Line 831  File_read_http_result pa_internal_file_r
                         head << "User-Agent: " DEFAULT_USER_AGENT CRLF;                          head << "User-Agent: " DEFAULT_USER_AGENT CRLF;
   
                 if(form && !method_is_get && content_type_specified) // POST + form + content-type was specified                  if(form && !method_is_get && content_type_specified) // POST + form + content-type was specified
                         throw Exception(PARSER_RUNTIME,                          throw Exception(PARSER_RUNTIME, 0, "$.content-type can't be specified with method POST"); 
                                 0,  
                                 "$.content-type can't be specified with method POST");   
   
                 if(vcookies && !vcookies->is_string()){ // allow empty                  if(vcookies && !vcookies->is_string()){ // allow empty
                         if(HashStringValue* cookies=vcookies->get_hash()) {                          if(HashStringValue* cookies=vcookies->get_hash()) {
Line 860  File_read_http_result pa_internal_file_r Line 840  File_read_http_result pa_internal_file_r
                                 cookies->for_each<Http_pass_header_info*>(http_pass_cookie, &info);                                   cookies->for_each<Http_pass_header_info*>(http_pass_cookie, &info); 
                                 head << CRLF;                                  head << CRLF;
                         } else                          } else
                                 throw Exception(PARSER_RUNTIME,                                   throw Exception(PARSER_RUNTIME, 0, "cookies param must be hash");
                                         0,  
                                         "cookies param must be hash");  
                 }                  }
   
                 if(request_body)                  if(request_body)
Line 885  File_read_http_result pa_internal_file_r Line 863  File_read_http_result pa_internal_file_r
                 }                  }
         }          }
                   
         char* response;  
         size_t response_size;          HTTP_response response(connect_string);
   
         // sending request          // sending request
         int status_code=http_request(response, response_size,          int status_code=http_request(response, idna_host, port, request, request_size, timeout_secs, fail_on_status_ne_200);
                 host, port, request, request_size,  
                 timeout_secs, fail_on_status_ne_200);           // processing results
                   char* raw_body=response.buf + response.body_offset;
         // processing results             size_t raw_body_size=response.length - response.body_offset;
         char* raw_body; size_t raw_body_size;  
         char* headers_end_at;  
         find_headers_end(response,   
                 headers_end_at,  
                 raw_body);  
         raw_body_size=response_size-(raw_body-response);  
           
         result.headers=new HashStringValue;          result.headers=new HashStringValue;
         VHash* vtables=new VHash;          VHash* vtables=new VHash;
         result.headers->put(HTTP_TABLES_NAME, vtables);          result.headers->put("tables", vtables);
   
         if(headers_end_at) {          if (!real_remote_charset && !response.headers.content_type.is_empty())
                 *headers_end_at=0;                  real_remote_charset=detect_charset(response.headers.content_type.cstr());
                 const String header_block(String::C(response, headers_end_at-response), String::L_TAINTED);  
                   
                 ArrayString aheaders;  
                 HashStringValue& tables=vtables->hash();  
   
                 size_t pos_after=0;          if(as_text)
                 header_block.split(aheaders, pos_after, "\n");                   real_remote_charset=pa_charsets.checkBOM(raw_body, raw_body_size, real_remote_charset);
                   
                 // processing headers  
                 size_t aheaders_count=aheaders.count();  
                 for(size_t i=1; i<aheaders_count; i++) {  
                         const String& line=*aheaders.get(i);  
                         size_t pos=line.pos(':');   
                         if(pos==STRING_NOT_FOUND || pos<1)  
                                 throw Exception("http.response",   
                                         &connect_string,  
                                         "bad response from host - bad header \"%s\"", line.cstr());  
                         const String::Body HEADER_NAME=line.mid(0, pos).change_case(r.charsets.source(), String::CC_UPPER);  
                         const String& HEADER_VALUE=line.mid(pos+1, line.length()).trim(String::TRIM_BOTH, " \t\r");  
                         if(as_text && HEADER_NAME==HTTP_CONTENT_TYPE_UPPER && !real_remote_charset)  
                                 real_remote_charset=detect_charset(HEADER_VALUE.cstr());  
   
                         // 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));          if (!real_remote_charset)
                 }                  real_remote_charset=asked_remote_charset; // never null
   
                 // filling $.cookies          for(Array_iterator<HTTP_Headers::Header> i(response.headers.headers); i.has_next(); ){
                 if(Value *vcookies=(Value *)tables.get("SET-COOKIE"))                  HTTP_Headers::Header header=i.next();
                         result.headers->put(HTTP_COOKIES_NAME, new VTable(parse_cookies(r, vcookies->get_table())));  
         }  
   
         if(as_text && raw_body_size>=3 && strncmp(raw_body, "\xEF\xBB\xBF", 3)==0){                  header.transcode(*real_remote_charset, r.charsets.source());
                 // skip UTF-8 signature (BOM code)  
                 raw_body+=3;                  String &header_value=*new String(header.value, String::L_TAINTED);
                 raw_body_size-=3;  
                 if(!real_remote_charset)                  tables_update(vtables->hash(), header.name, header_value);
                         real_remote_charset=&UTF8_charset;                  result.headers->put(header.name, new VString(header_value));
         }          }
   
           // filling $.cookies
           if(vcookies=vtables->hash().get("SET-COOKIE"))
                   result.headers->put(HTTP_COOKIES_NAME, new VTable(parse_cookies(r, vcookies->get_table())));
   
         // output response          // output response
         String::C real_body=String::C(raw_body, raw_body_size);          String::C real_body=String::C(raw_body, raw_body_size);
   
         if(as_text && transcode_text_result && raw_body_size) { // raw_body_size must be checked because transcode returns CONST string in case length==0, which contradicts hacking few lines below          if(as_text && transcode_text_result && raw_body_size) { // raw_body_size must be checked because transcode returns CONST string in case length==0, which contradicts hacking few lines below
                 // defaulting to used-asked charset [it's never empty!]  
                 if(!real_remote_charset)  
                         real_remote_charset=asked_remote_charset;  
   
                 real_body=Charset::transcode(real_body, *real_remote_charset, r.charsets.source());                  real_body=Charset::transcode(real_body, *real_remote_charset, r.charsets.source());
   
         }          }
   
         result.str=const_cast<char *>(real_body.str); // hacking a little          result.str=const_cast<char *>(real_body.str); // hacking a little
Line 990  File_read_http_result pa_internal_file_r Line 918  File_read_http_result pa_internal_file_r
   
         return result;          return result;
 }  }
   
   /* ********************** httpd *************************** */
   
   class HTTPD_request : public HTTP_response {
   public:
           const char *method;
           const char *uri;
   
           HTTPD_request() : HTTP_response(String::Empty), method(NULL), uri(NULL){};
   
           const char *extract_method(char *method_line){
                   char* uri_start = strchr(method_line, ' ');
   
                   if(!uri_start || uri_start == method_line)
                           return NULL;
   
                   char* uri_end=strchr(uri_start+1, ' ');
   
                   if(!uri_end || uri_end == uri_start+1)
                           return NULL;
   
                   uri=pa_strdup(uri_start+1, uri_end-uri_start-1);
                   return str_upper(method_line, uri_start-method_line);
           }
   
           void read_header(int);
           size_t read_post(int, char *, size_t);
   };
   
   enum HTTPD_request_state {
           HTTPD_METHOD,
           HTTPD_HEADERS
   };
   
   void HTTPD_request::read_header(int sock) {
           enum HTTPD_request_state state = HTTPD_METHOD;
   
           size_t chunk_size = 0x400*4;
           resize(chunk_size);
   
           while(read(sock, chunk_size)){
                   switch(state){
                           case HTTPD_METHOD: {
                                   size_t method_size = first_line();
                                   if(!method_size)
                                           break;
   
                                   char *method_line = pa_strdup(buf, method_size);
                                   method = extract_method(method_line);
   
                                   if(!method || strcmp(method, "GET") && strcmp(method, "HEAD") && strcmp(method, "POST") && strcmp(method, "PUT") && strcmp(method, "DELETE"))
                                           throw Exception("httpd.method", new String(method ? method : method_line), "invalid request method");
                                   state = HTTPD_HEADERS;
                           }
   
                           case HTTPD_HEADERS: {
                                   if(!body_start())
                                           break;
   
                                   parse_headers();
                                   return;
                           }
                   }
           }
   
           if(state == HTTPD_METHOD)
                   throw Exception("httpd.request", 0, "bad request from host - no method found (size=%u)", length);
   
           if(state == HTTPD_HEADERS){
                   parse_headers();
                   body_offset=length;
           }
   }
   
   size_t HTTPD_request::read_post(int sock, char *body, size_t max_bytes) {
           size_t total_read = min(length - body_offset, max_bytes);
           memcpy(body, buf, total_read);
   
           while (total_read < max_bytes){
                   ssize_t received_size = recv(sock, buf + total_read, max_bytes - total_read, 0);
                   if(received_size == 0)
                           return total_read;
                   if(received_size < 0) {
                           if(int no = pa_socks_errno())
                                   throw Exception("httpd.timeout", &url, "error receiving request body: %s (%d)", pa_socks_strerr(no), no);
                           return total_read;
                   }
                   total_read += received_size;
           }
           return total_read;
   }
   
   /* ********************************************************** */
   
   Array<HTTP_Headers::Header> &HTTPD_Connection::headers() {
           return request->headers.headers;
   }
   
   const char *HTTPD_Connection::method() {
           return request->method;
   }
   
   const char *HTTPD_Connection::uri() {
           return request->uri;
   }
   
   const char *HTTPD_Connection::content_type() {
           return request->headers.content_type.cstr();
   }
   
   uint64_t HTTPD_Connection::content_length(){
           return request->headers.content_length;
   }
   
   void HTTPD_Connection::read_header(){
           request = new HTTPD_request();
           request->read_header(sock);
   }
   
   size_t HTTPD_Connection::read_post(char *body, size_t max_bytes) {
           return request->read_post(sock, body, max_bytes);
   }
   
   static int sock_on = 1;
   
   int HTTPD_Server::bind(const char *host_port){
           struct sockaddr_in me;
   
           const char *port = strchr(host_port, ':');
           const char *host = NULL;
           if(port && port > host_port){
                   host = pa_strdup(host_port, port - host_port);
                   port += 1;
           } else {
                   port = host_port;
           }
   
           if(!set_addr(&me, host, pa_atoui(port, 10))){
                   if (host)
                           throw Exception("httpd.bind", 0, "can not resolve hostname \"%s\"", host);
                   me.sin_addr.s_addr=INADDR_ANY;
           }
   
           int sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP/*0*/);
   
           if(sock < 0){
                   int no=pa_socks_errno();
                   throw Exception("httpd.bind", 0, "can not make socket: %s (%d)", pa_socks_strerr(no), no);
           }
   
           if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char *)&sock_on, sizeof(sock_on)) ||
               setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (char *)&sock_on, sizeof(sock_on)) ||
               ::bind(sock, (struct sockaddr*)&me, sizeof(me)) ||
               listen(sock, 16)) {
                   closesocket(sock);
                   int no = pa_socks_errno();
                   throw Exception("httpd.bind", 0, "can not bind socket: %s (%d)", pa_socks_strerr(no), no);
           }
           return sock;
   }
   
   static int ready(int fd,int operation,int timeout_value){
           struct timeval timeout = {0, timeout_value * 1000};
           fd_set fds;
           FD_ZERO(&fds);
           FD_SET(fd, &fds);
           switch (operation){
                   case 0: return select(fd + 1, &fds, NULL, NULL, &timeout)>0;  /* read */
                   case 1: return select(fd + 1, NULL, &fds, NULL, &timeout)>0;  /* write */
                   default: return select(fd + 1, &fds, &fds, NULL, &timeout)>0;  /* both */
           }
   }
   
   HTTPD_Connection *HTTPD_Server::accept(int sock, int timeout_value) {
           int ready = ::ready(sock, 0, timeout_value);
           if (ready < 0) {
                   int no=pa_socks_errno();
                   if(no == EINTR)
                           return NULL;
                   throw Exception("httpd.accept", 0, "error waiting for connection: %s (%d)", pa_socks_strerr(no), no);
           }
           if (ready == 0) {
                   /* Timeout */
                   return NULL;
           }
   
           struct sockaddr_in addr;
           socklen_t sock_addr_len = sizeof(struct sockaddr_in);
           memset(&addr, 0, sock_addr_len);
   
           int csock = ::accept(sock, (struct sockaddr *)&addr, &sock_addr_len);
           if(csock == -1){
                   int no=pa_socks_errno();
                   throw Exception("httpd.accept", 0, "error accepting connection: %s (%d)", pa_socks_strerr(no), no);
           }
   
           return new HTTPD_Connection(csock, pa_strdup(inet_ntoa(addr.sin_addr)));
   }
   
Removed from v.1.64  
changed lines
  Added in v.1.89

E-mail: