web/Obj/sac-a-push.phh

   1 <?php
   2
   3 /*
   4  *  brisk - spush/sac-a-push.phh
   5  *
   6  *  Copyright (C) 2012 Matteo Nastasi
   7  *                          mailto: nastasi@alternativeoutput.it
   8  *                                  matteo.nastasi@milug.org
   9  *                          web: http://www.alternativeoutput.it
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful, but
  17  * WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABLILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19  * General Public License for more details. You should have received a
  20  * copy of the GNU General Public License along with this program; if
  21  * not, write to the Free Software Foundation, Inc, 59 Temple Place -
  22  * Suite 330, Boston, MA 02111-1307, USA.
  23  *
  24  */
  25
  26 define('SITE_PREFIX', '/brisk/');
  27 define('SITE_PREFIX_LEN', 7);
  28
  29 function spu_process_info($stream_info, $method, &$header, &$get, &$post, &$cookie)
  30 {
  31     $check_post = FALSE;
  32     $header = array();
  33     $get = array();
  34     $post = array();
  35     foreach(preg_split("/(\r?\n)/", $stream_info) as $line) {
  36         // printf("LINE: [%s]\n", $line);
  37         if ($check_post) {
  38             if (!isset($header['The-Request'])) {
  39                 return FALSE;
  40             }
  41             $req = explode(" ", $header['The-Request']);
  42             $method = $req[0];
  43
  44             if (isset($header['Cookie'])) {
  45                 $cookies = explode(";", $header['Cookie']);
  46                 for ($i = 0 ; $i < count($cookies) ; $i++) {
  47                     $nameval = explode("=", trim($cookies[$i]));
  48                     if (count($nameval) != 2) {
  49                         printf("WARNING: malformat cookie element [%s]\n", $cookies[$i]);
  50                         continue;
  51                     }
  52                     $cookie[$nameval[0]] = urldecode($nameval[1]);
  53                 }
  54             }
  55             // GET params management
  56             $get_vars = explode('?', $req[1], 2);
  57             $path =   $get_vars[0];
  58             if (count($get_vars) > 1) {
  59                 $a = explode('&', $get_vars[1]);
  60                 printf("A COUNT: [%s] %d\n", $a[0], count($a));
  61                 for ($i = 0 ; $i < count($a) ; $i++) {
  62                     $b = explode('=', $a[$i]);
  63                     $get[$b[0]] = urldecode($b[1]);
  64                 }
  65             }
  66             // POST params management
  67             if ($req[0] == 'POST') {
  68                 if ($header['Content-Type'] != 'application/x-www-form-urlencoded'
  69                     || !isset($header['Content-Length'])) {
  70                     return FALSE;
  71                 }
  72                 $post_len = mb_strlen($line, "latin1");
  73                 $a = explode('&', $line);
  74                 for ($i = 0 ; $i < count($a) ; $i++) {
  75                     $b = explode('=', $a[$i]);
  76                     $post[$b[0]] = urldecode($b[1]);
  77                 }
  78                 printf("INFO: postlen: %d\n", $post_len);
  79             }
  80             break;
  81         }
  82         if ($line == "") {
  83             $check_post = TRUE;
  84             continue;
  85         }
  86         $split = explode(":", $line, 2);
  87         $header[$split[0]] = $split[1];
  88     }
  89     return $path;
  90 }
  91
  92 function gpcs_var($name, $get, $post, $cookie)
  93 {
  94     if (isset($GLOBALS[$name]))
  95         return FALSE;
  96     else if (isset($cookie[$name]))
  97         return ($cookie[$name]);
  98     else if (isset($post[$name]))
  99         return ($post[$name]);
 100     else if (isset($get[$name]))
 101         return ($get[$name]);
 102
 103     return FALSE;
 104 }
 105
 106 function headers_render($header, $len)
 107 {
 108     $s = "";
 109     if (isset($header['Location'])) {
 110         return sprintf("HTTP/1.1 302 OK\r\nLocation: %s\r\n\r\n", $header['Location']);
 111     }
 112     else {
 113         $s .= "HTTP/1.1 200 OK\r\n";
 114     }
 115     if (!isset($header['Date']))
 116         $s .= sprintf("Date: %s\r\n", date(DATE_RFC822));
 117     if (!isset($header['Connection']))
 118         $s .= "Connection: close\r\n";
 119     if (!isset($header['Content-Type']))
 120         $s .= "Content-Type: text/html\r\n";
 121     foreach($header as $key => $value) {
 122         $s .= sprintf("%s: %s\r\n", $key, $value);
 123     }
 124     if ($len == -1) {
 125         $s .= "Cache-Control: no-cache, must-revalidate\r\n";
 126         $s .= "Expires: Mon, 26 Jul 1997 05:00:00 GMT\r\n";
 127         $s .= "Content-Encoding: chunked\r\n";
 128         $s .= "Transfer-Encoding: chunked\r\n";
 129     }
 130     else if ($len > 0) {
 131         $s .= sprintf("Content-Length: %d\r\n", $len);
 132     }
 133     $s .= "\r\n";
 134
 135     return ($s);
 136 }
 137
 138 /*
 139  *  Caching system using ob php system to cache old style pages
 140  *  to a var and than send it with more calm
 141  */
 142
 143 function shutta()
 144 {
 145   log_rd2("SHUTTA [".connection_status()."] !");
 146 }
 147
 148 register_shutdown_function('shutta');
 149
 150 /*
 151  *  MAIN
 152  */
 153
 154 function chunked_content($content)
 155 {
 156     $content_l = mb_strlen($content, "ASCII");
 157
 158     return (sprintf("%X\r\n%s\r\n", $content_l, $content));
 159 }
 160
 161 function chunked_fini()
 162 {
 163     return sprintf("0\r\n");
 164 }
 165
 166 class Sac_a_push {
 167     static $fixed_fd = 2;
 168
 169     var $file_socket;
 170     var $unix_socket;
 171     var $socks;
 172     var $s2u;
 173     var $pages_flush;
 174
 175     var $list;
 176     var $in;
 177
 178     var $debug;
 179     var $blocking_mode;
 180
 181     var $app;
 182     var $bin5;
 183
 184     var $curtime;
 185
 186     var $rndstr;
 187     var $main_loop;
 188
 189     function Sac_a_push()
 190     {
 191     }
 192
 193     // Sac_a_push::create("/tmp/brisk.sock", 0, 0
 194
 195     static function create(&$app, $sockname, $debug, $blocking_mode)
 196     {
 197         $thiz = new Sac_a_push();
 198
 199         $thiz->app = $app;
 200         $thiz->file_socket = $sockname;
 201         $thiz->unix_socket = "unix://$sockname";
 202         $thiz->debug = $debug;
 203         $thiz->socks = array();
 204         $thiz->s2u  = array();
 205         $thiz->pages_flush = array();
 206
 207         $thiz->blocking_mode = 0; // 0 for non-blocking
 208
 209         $thiz->rndstr = "";
 210         for ($i = 0 ; $i < 4096 ; $i++) {
 211             $thiz->rndstr .= chr(mt_rand(65, 90));
 212         }
 213
 214         if (file_exists($thiz->file_socket)) {
 215             unlink($thiz->file_socket);
 216         }
 217
 218         $old_umask = umask(0);
 219         if (($thiz->list = stream_socket_server($thiz->unix_socket, $err, $errs)) === FALSE) {
 220             return (FALSE);
 221         }
 222         umask($old_umask);
 223         stream_set_blocking($thiz->list, $thiz->blocking_mode); # Set the stream to non-blocking
 224
 225         if (($thiz->in = fopen("php://stdin", "r")) === FALSE) {
 226             return(FALSE);
 227         }
 228
 229         $thiz->main_loop = FALSE;
 230
 231         return ($thiz);
 232     }
 233
 234     function socks_set($sock, $user)
 235     {
 236         $id = intval($sock);
 237
 238         $this->s2u[$id]   = $user;
 239         $this->socks[$id] = $sock;
 240     }
 241
 242     function socks_unset($sock)
 243     {
 244         $id = intval($sock);
 245
 246         unset($this->s2u[$id]);
 247         unset($this->socks[$id]);
 248     }
 249
 250     function pgflush_try_add(&$new_socket, $tout, $header_out, $content)
 251     {
 252         $pgflush = new PageFlush($new_socket, $this->curtime, $tout, $header_out, $content);
 253
 254         if ($pgflush->try_flush($this->curtime) == FALSE) {
 255             // Add $pgflush to the pgflush array
 256             $this->pgflush_add($pgflush);
 257         }
 258     }
 259
 260     function pgflush_add($pgflush)
 261     {
 262         array_push($this->pages_flush, $pgflush);
 263     }
 264
 265     function garbage_manager($force)
 266     {
 267         $this->app->garbage_manager($force);
 268
 269         foreach ($this->socks as $k => $sock) {
 270             if ($this->s2u[intval($sock)]->sess == '') {
 271                 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
 272                     $this->s2u[intval($sock)]->rd_socket_set(NULL);
 273                 }
 274                 unset($this->socks[intval($sock)]);
 275                 unset($this->s2u[intval($sock)]);
 276                 fclose($sock);
 277                 printf("CLOSE ON GARBAGE MANAGER\n");
 278             }
 279         }
 280     }
 281
 282     function run()
 283     {
 284         if ($this->main_loop) {
 285             return (FALSE);
 286         }
 287
 288         $this->main_loop = TRUE;
 289
 290         while ($this->main_loop) {
 291             $this->curtime = time();
 292             printf("IN LOOP: Current opened: %d  pages_flush: %d - ", count($this->socks), count($this->pages_flush));
 293
 294             /* Prepare the read array */
 295             /* // when we manage it ... */
 296             /* if ($shutdown)  */
 297             /*     $read   = array_merge(array("$in" => $in), $socks); */
 298             /* else */
 299             $read   = array_merge(array(intval($this->list) => $this->list, intval($this->in) => $this->in),
 300                                   $this->socks);
 301
 302             if ($this->debug > 1) {
 303                 printf("PRE_SELECT\n");
 304                 print_r($read);
 305             }
 306             $write  = NULL;
 307             $except = NULL;
 308             $num_changed_sockets = stream_select($read, $write, $except, 0, 100000);
 309
 310             if ($num_changed_sockets == 0) {
 311                 printf(" no data in 5 secs ");
 312             }
 313             else if ($num_changed_sockets > 0) {
 314                 printf("num sock %d num_of_socket: %d\n", $num_changed_sockets, count($read));
 315                 if ($this->debug > 1) {
 316                     print_r($read);
 317                 }
 318                 /* At least at one of the sockets something interesting happened */
 319                 foreach ($read as $i => $sock) {
 320                     /* is_resource check is required because there is the possibility that
 321                        during new request an old connection is closed */
 322                     if (!is_resource($sock)) {
 323                         continue;
 324                     }
 325                     if ($sock === $this->list) {
 326                         printf("NUOVA CONNEX\n");
 327                         if (($new_unix = stream_socket_accept($this->list)) == FALSE) {
 328                             printf("SOCKET_ACCEPT FAILED\n");
 329                             continue;
 330                         }
 331                         $stream_info = "";
 332                         $method      = "";
 333                         $get         = array();
 334                         $post        = array();
 335                         $cookie      = array();
 336                         if (($new_socket = ancillary_getstream($new_unix, $stream_info)) !== FALSE) {
 337                             printf("NEW_SOCKET: %d\n", intval($new_socket));
 338                             stream_set_blocking($new_socket, $this->blocking_mode); // Set the stream to non-blocking
 339                             printf("RECEIVED HEADER:\n%s", $stream_info);
 340                             $path = spu_process_info($stream_info, $method, $header, $get, $post, $cookie);
 341                             printf("PATH: [%s]\n", $path);
 342                             printf("M: %s\nHEADER:\n", $method);
 343                             print_r($header);
 344                             printf("GET:\n");
 345                             print_r($get);
 346                             printf("POST:\n");
 347                             print_r($post);
 348                             printf("COOKIE:\n");
 349                             print_r($cookie);
 350
 351                             $addr = stream_socket_get_name($new_socket, TRUE);
 352                             $header_out = array();
 353
 354                             $subs = SITE_PREFIX."briskin5/";
 355                             $subs_l = strlen($subs);
 356                             $rret = FALSE;
 357                             if (!strncmp($path, SITE_PREFIX, SITE_PREFIX_LEN)) {
 358                                 $rret = $this->app->request_mgr($this, $header_out, $new_socket, substr($path, SITE_PREFIX_LEN), $addr, $get, $post, $cookie);
 359                             }
 360                             if ($rret == FALSE) {
 361                                 // FIXME: manage 404 !!!
 362                                 printf("TODO: fix unknown page\n");
 363                             }
 364                             printf("number of sockets after %d\n", count($this->socks));
 365                         }
 366                         else {
 367                             printf("WARNING: ancillary_getstream failed\n");
 368                         }
 369                     }
 370                     else {
 371                         $buf = fread($sock, 512);
 372                         if ($buf == FALSE || strlen($buf) == 0) {
 373                             if ($buf == FALSE) {
 374                                 printf("ERROR READING\n");
 375                             }
 376                             if ($sock === $this->list) {
 377                                 printf("Arrivati %d bytes da list\n", strlen($buf));
 378                             }
 379                             else if ($sock === $this->in) {
 380                                 printf("Arrivati %d bytes da stdin\n", strlen($buf));
 381                             }
 382                             else {
 383                                 // $user_a[$s2u[intval($sock)]]->disable();
 384                                 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
 385                                     $this->s2u[intval($sock)]->rd_socket_set(NULL);
 386                                 }
 387                                 unset($this->socks[intval($sock)]);
 388                                 unset($this->s2u[intval($sock)]);
 389                             }
 390                             fclose($sock);
 391                             printf("CLOSE ON READ\n");
 392
 393                             if ($this->debug > 1) {
 394                                 printf("post unset\n");
 395                                 print_r($this->socks);
 396                             }
 397                         }
 398                         else {
 399                             if ($this->debug > 1) {
 400                                 print_r($read);
 401                             }
 402                             if ($sock === $this->list) {
 403                                 printf("Arrivati %d bytes da list\n", strlen($buf));
 404                             }
 405                             else if ($sock === $this->in) {
 406                                 printf("Arrivati %d bytes da stdin\n", strlen($buf));
 407                             }
 408                             else {
 409                                 $key = array_search("$sock", $this->socks);
 410                                 printf("Arrivati %d bytes dalla socket n. %d\n", strlen($buf), $key);
 411                             }
 412                         }
 413                     }
 414                 }
 415             }
 416
 417             $this->garbage_manager(FALSE);
 418
 419             /* manage unfinished pages */
 420             foreach ($this->pages_flush as $k => $pgflush) {
 421                 if ($pgflush->try_flush($this->curtime) == TRUE) {
 422                     unset($this->pages_flush[$k]);
 423                 }
 424             }
 425
 426             /*
 427                $response:                        raw stream data not sent
 428                $content:                         html consistent data (<script bla bla>)
 429                $user->stream_keepalive($w_ping)  ping srv->cli OR srv->cli + cli->srv if $w_ping == TRUE
 430             */
 431
 432             /* manage open streaming */
 433             foreach ($this->socks as $k => $sock) {
 434                 if (isset($this->s2u[intval($sock)])) {
 435                     $user = $this->s2u[intval($sock)];
 436                     $response = $user->rd_cache_get();
 437                     $do_ping = FALSE;
 438                     if (($this->curtime - $user->lacc) > (EXPIRE_TIME_RD / 3)) {
 439                         $do_ping = TRUE;
 440                     }
 441                     else {
 442                         $user->ping_req = FALSE;
 443                     }
 444
 445                     if ($response == "") {
 446                         $content = "";
 447                         $user->stream_main($content, $get, $post, $cookie);
 448                         printf("[%s] [%d] [%d]\n", $user->name, $user->lacc, $this->curtime);
 449                         if ($do_ping && $user->ping_req == FALSE) {
 450                             $content .= $user->stream_keepalive(TRUE);
 451                             $user->ping_req = TRUE;
 452                         }
 453                         else if ($content == "" && $user->rd_kalive_is_expired($this->curtime)) {
 454                             $content = $user->stream_keepalive(FALSE);
 455                         }
 456                         if ($content != "") {
 457                             $response = chunked_content($content);
 458                         }
 459                     }
 460
 461                     if ($response != "") {
 462                         // echo "SPIA: [".substr($response, 0, 60)."...]\n";
 463                         echo "SPIA: [".$response."]\n";
 464                         $response_l = mb_strlen($response, "ASCII");
 465                         $wret = @fwrite($sock, $response);
 466                         if ($wret < $response_l) {
 467                             printf("TROUBLE WITH FWRITE: %d\n", $wret);
 468                             $user->rd_cache_set(mb_substr($response, $wret, $response_l - $wret, "ASCII"));
 469                         }
 470                         else {
 471                             $user->rd_cache_set("");
 472                         }
 473                         fflush($sock);
 474                         $user->rd_kalive_reset($this->curtime);
 475                     }
 476
 477                     // close socket after a while to prevent client memory consumption
 478                     if ($user->rd_endtime_is_expired($this->curtime)) {
 479                         if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
 480                             $this->s2u[intval($sock)]->rd_socket_set(NULL);
 481                         }
 482                         unset($this->socks[intval($sock)]);
 483                         unset($this->s2u[intval($sock)]);
 484                         fclose($sock);
 485                         printf("CLOSE ON LOOP\n");
 486                     }
 487                 }
 488             }  // foreach ($this->socks...
 489             printf("\n");
 490         }  // while (...
 491     }  // function run(...
 492 }
 493
 494 ?>