Subversion Repositories Kolibri OS

Rev

Rev 7422 | Go to most recent revision | Blame | Last modification | View Log | Download | RSS feed

  1. //===================================================//
  2. //                                                   //
  3. //                       HTTP                        //
  4. //                                                   //
  5. //===================================================//
  6.  
  7. struct _http
  8. {
  9.     dword url;
  10.     dword transfer;
  11.     dword content_length;
  12.     dword content_received;
  13.     dword status_code;
  14.     dword receive_result;
  15.     dword content_pointer;
  16.     char redirect_url[4096*3];
  17.     char finaladress[4096*3];
  18.  
  19.     dword get();
  20.     void free();
  21.     void receive();
  22.     bool handle_redirect();
  23. };
  24.  
  25. dword _http::get(dword _url)
  26. {
  27.     url = _url;
  28.     http_get stdcall (url, 0, 0, #accept_language);
  29.     transfer = EAX;
  30.     return transfer;
  31. }
  32.  
  33. void _http::free()
  34. {
  35.     http_free stdcall (transfer);
  36.     transfer=0;
  37. }
  38.  
  39. void _http::receive()
  40. {
  41.     http_receive stdcall (transfer);
  42.     receive_result = EAX;
  43.  
  44.     EDI = transfer;
  45.     if (!EAX) {
  46.         status_code = EDI.http_msg.status;
  47.         content_pointer = EDI.http_msg.content_ptr;
  48.     }
  49.     content_length = EDI.http_msg.content_length;
  50.     content_received = EDI.http_msg.content_received;
  51.  
  52.  }
  53.  
  54. bool _http::handle_redirect()
  55. {
  56.         dword redirect;
  57.     http_find_header_field stdcall (transfer, "location\0");
  58.     if (EAX!=0) {
  59.         ESI = EAX;
  60.         EDI = #redirect_url;
  61.         do {
  62.             $lodsb;
  63.             $stosb;
  64.         } while (AL != 0) && (AL != 13) && (AL != 10);
  65.         DSBYTE[EDI-1]='\0';
  66.         get_absolute_url(#finaladress, url, #redirect_url);
  67.         strcpy(url, #finaladress);
  68.         return true;
  69.     }
  70.     return false;
  71. }
  72.  
  73. //===================================================//
  74. //                                                   //
  75. //                    DOWNLOADER                     //
  76. //                                                   //
  77. //===================================================//
  78.  
  79.  
  80. enum {
  81.         STATE_NOT_STARTED,
  82.         STATE_IN_PROGRESS,
  83.         STATE_COMPLETED
  84. };
  85.  
  86. struct DOWNLOADER {
  87.         _http httpd;
  88.         dword bufpointer, bufsize, url;
  89.         int state;
  90.         dword Start();
  91.         void Stop();
  92.         bool MonitorProgress();
  93. };
  94.  
  95. dword DOWNLOADER::Start(dword _url)
  96. {
  97.         url = _url;
  98.         state = STATE_IN_PROGRESS;
  99.         httpd.get(url);
  100.         if (!httpd.transfer) Stop();
  101.         return httpd.transfer;
  102. }
  103.  
  104. void DOWNLOADER::Stop()
  105. {
  106.         state = STATE_NOT_STARTED;
  107.         if (httpd.transfer!=0)
  108.         {
  109.                 EAX = httpd.transfer;
  110.                 EAX = EAX.http_msg.content_ptr;         // get pointer to data
  111.                 $push EAX                                                       // save it on the stack
  112.                 http_free stdcall (httpd.transfer);     // abort connection
  113.                 $pop  EAX                                                      
  114.                 free(EAX);                                              // free data
  115.                 httpd.transfer=0;
  116.                 bufsize = 0;
  117.                 bufpointer = free(bufpointer);
  118.         }
  119.         httpd.content_received = httpd.content_length = 0;
  120. }
  121.  
  122. bool DOWNLOADER::MonitorProgress()
  123. {
  124.         if (httpd.transfer <= 0) return false;
  125.         httpd.receive();
  126.         if (!httpd.content_length) httpd.content_length = httpd.content_received * 20;
  127.  
  128.         if (httpd.receive_result == 0) {
  129.                 if (httpd.status_code >= 300) && (httpd.status_code < 400)
  130.                 {
  131.                         httpd.handle_redirect();
  132.                         strcpy(url, httpd.url);
  133.                         get_absolute_url(#httpd.finaladress, url, #httpd.redirect_url);
  134.                         Stop();
  135.                         Start(#httpd.finaladress);
  136.                         url = #httpd.finaladress;
  137.                         return false;
  138.                 }
  139.                 state = STATE_COMPLETED;
  140.                 bufpointer = httpd.content_pointer;
  141.                 bufsize = httpd.content_received;
  142.                 httpd.free();
  143.         }
  144.         return true;
  145. }
  146.  
  147.  
  148. /*=====================================
  149. ==                                   ==
  150. ==         CHECK PATH TYPE           ==
  151. ==                                   ==
  152. =====================================*/
  153.  
  154.  
  155. int check_is_the_adress_local(dword _in)
  156. {
  157.         if (ESBYTE[_in]!='/') return false;
  158.         _in++;
  159.         if(!strncmp(_in,"rd/",3)) return true;
  160.         if(!strncmp(_in,"fd/",3)) return true;
  161.         if(!strncmp(_in,"hd",2)) return true;
  162.         if(!strncmp(_in,"bd",2)) return true;
  163.         if(!strncmp(_in,"cd",2)) return true;
  164.         if(!strncmp(_in,"sys/",4)) return true;
  165.         if(!strncmp(_in,"tmp/",4)) return true;
  166.         if(!strncmp(_in,"usbhd",5)) return true;
  167.         if(!strncmp(_in,"kolibrios",9)) return true;
  168.         return false;
  169. }
  170.  
  171. int check_is_the_url_absolute(dword _in)
  172. {
  173.         if(!strncmp(_in,"ftp:",4)) return true;
  174.         if(!strncmp(_in,"http:",5)) return true;
  175.         if(!strncmp(_in,"https:",6)) return true;
  176.         if(!strncmp(_in,"mailto:",7)) return true;
  177.         if(!strncmp(_in,"tel:",4)) return true;
  178.         if(!strncmp(_in,"#",1)) return true;
  179.         if(check_is_the_adress_local(_in)) return true;
  180.         return false;
  181. }
  182.  
  183. void get_absolute_url(dword _rez, _base, _new)
  184. {
  185.         int i;
  186.         //case: ./valera.html
  187.         if (!strncmp(_new,"./", 2)) _new+=2;
  188.         //case: http://site.name
  189.         if (check_is_the_url_absolute(_new)) {
  190.                 strcpy(_rez, _new);
  191.                 goto _GET_ABSOLUTE_URL_END;
  192.         }
  193.         //case: /valera.html
  194.         if (ESBYTE[_new] == '/') //remove everything after site domain name
  195.         {
  196.                 strcpy(_rez, _base);
  197.                 i = strchr(_rez+8,'/');
  198.                 if (i<1) i=strlen(_rez)+_rez;
  199.                 strcpy(i, _new);
  200.                 goto _GET_ABSOLUTE_URL_END;
  201.         }      
  202.         //case: ../../valera.html
  203.         strcpy(_rez, _base);
  204.         ESBYTE[ strrchr(_rez,'/') + _rez -1 ] = '\0'; //remove name
  205.         while (!strncmp(_new,"../",3))
  206.         {
  207.                 _new += 3;
  208.                 ESBYTE[ strrchr(_rez,'/') + _rez -1 ] = '\0';  
  209.         }
  210.         //case: valera.html
  211.         chrcat(_rez, '/');
  212.         strcat(_rez, _new);
  213.         _GET_ABSOLUTE_URL_END:
  214.         while (i=strstr(_rez, "&amp;")) strcpy(i+1, i+5);      
  215. }
  216.