4 * brisk - spush/sac-a-push.phh
6 * Copyright (C) 2012 Matteo Nastasi
7 * mailto: nastasi@alternativeoutput.it
8 * matteo.nastasi@milug.org
9 * web: http://www.alternativeoutput.it
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.
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.
26 define('SITE_PREFIX', '/brisk/');
27 define('SITE_PREFIX_LEN', 7);
29 function spu_process_info($stream_info, $method, &$header, &$get, &$post, &$cookie)
35 foreach(preg_split("/(\r?\n)/", $stream_info) as $line) {
36 // printf("LINE: [%s]\n", $line);
38 if (!isset($header['The-Request'])) {
41 $req = explode(" ", $header['The-Request']);
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]);
52 $cookie[$nameval[0]] = urldecode($nameval[1]);
55 // GET params management
56 $get_vars = explode('?', $req[1], 2);
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]);
66 // POST params management
67 if ($req[0] == 'POST') {
68 if ($header['Content-Type'] != 'application/x-www-form-urlencoded'
69 || !isset($header['Content-Length'])) {
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]);
78 printf("INFO: postlen: %d\n", $post_len);
86 $split = explode(":", $line, 2);
87 $header[$split[0]] = $split[1];
92 function gpcs_var($name, $get, $post, $cookie)
94 if (isset($GLOBALS[$name]))
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]);
106 function headers_render($header, $len)
110 if (isset($header['cookies'])) {
111 $cookies = $header['cookies']->render();
112 unset($header['cookies']);
114 if (isset($header['Location'])) {
115 $s = sprintf("HTTP/1.1 302 OK\r\n%sLocation: %s\r\n", $cookies, $header['Location']);
118 $s = "HTTP/1.1 200 OK\r\n";
120 if (!isset($header['Date']))
121 $s .= sprintf("Date: %s\r\n", date(DATE_RFC822));
122 if (!isset($header['Connection']))
123 $s .= "Connection: close\r\n";
124 if (!isset($header['Content-Type']))
125 $s .= "Content-Type: text/html\r\n";
126 foreach($header as $key => $value) {
127 $s .= sprintf("%s: %s\r\n", $key, $value);
130 $s .= sprintf("Content-Length: %d\r\n", $len);
133 $s .= "Cache-Control: no-cache, must-revalidate\r\n";
134 $s .= "Expires: Mon, 26 Jul 1997 05:00:00 GMT\r\n";
135 if (!isset($header['Content-Encoding'])) {
136 $s .= "Content-Encoding: chunked\r\n";
138 $s .= "Transfer-Encoding: chunked\r\n";
148 * Caching system using ob php system to cache old style pages
149 * to a var and than send it with more calm
154 log_rd2("SHUTTA [".connection_status()."] !");
157 register_shutdown_function('shutta');
163 function chunked_content($zls, $content)
166 $cont_comp = $zls->compress_chunk($content);
169 $cont_comp = $content;
171 $cont_comp_l = mb_strlen($cont_comp, "ASCII");
172 printf("CHUNK: [%s]\n", $content);
174 return (sprintf("%X\r\n", $cont_comp_l).$cont_comp."\r\n");
177 function chunked_fini()
179 return sprintf("0\r\n");
182 function get_encoding($header)
185 if (isset($header['Accept-Encoding'])) {
186 $acc = explode(',', $header['Accept-Encoding']);
188 if (array_search('gzip', $acc) !== FALSE) {
191 else if (array_search('deflate', $acc) !== FALSE) {
201 // Set-Cookie: reg_fb_gate=deleted; Expires=Thu, 01-Jan-1970 00:00:01 GMT; Path=/; Domain=.foo.com; HttpOnly
202 // string $name [, string $value [, int $expire = 0 [, string $path [, string $domain [, bool $secure = false [, bool $httponly = false ]]]]]] )
205 $this->attr = array();
208 static function create($name)
210 $thiz = new Cookie();
212 $thiz->attr[$name] = "";
214 $argc = func_num_args();
215 for ($i = 1 ; $i < $argc ; $i++) {
216 $arg = func_get_arg($i);
219 $thiz->attr[$name] = urlencode($arg);
222 $thiz->attr['Expires'] = gmdate('D, d M Y H:i:s \G\M\T', $arg); // RFC 1211 format
225 $thiz->attr['Path'] = $arg;
228 $thiz->attr['Domain'] = $arg;
232 $thiz->attr['Secure'] = NULL;
237 $thiz->attr['HttpOnly'] = NULL;
253 foreach ($this->attr as $k => $v) {
255 $r .= sprintf("%s%s", ($isfirst ? "" : "; "), $k);
258 $r .= sprintf("%s%s=%s", ($isfirst ? "" : "; "), $k, $v);
273 $this->cookies = array();
278 if (($cookie = call_user_func_array("Cookie::create", func_get_args())) == FALSE)
281 array_push($this->cookies, $cookie);
289 foreach ($this->cookies as $cookie) {
290 $r .= $cookie->render();
299 static $fixed_fd = 2;
321 function Sac_a_push()
325 // Sac_a_push::create("/tmp/brisk.sock", 0, 0
327 static function create(&$app, $sockname, $debug, $blocking_mode)
329 $thiz = new Sac_a_push();
332 $thiz->file_socket = $sockname;
333 $thiz->unix_socket = "unix://$sockname";
334 $thiz->debug = $debug;
335 $thiz->socks = array();
336 $thiz->s2u = array();
337 $thiz->pages_flush = array();
339 $thiz->blocking_mode = 0; // 0 for non-blocking
342 for ($i = 0 ; $i < 4096 ; $i++) {
344 $thiz->rndstr .= "\n";
346 $thiz->rndstr .= chr(mt_rand(65, 90));
349 if (file_exists($thiz->file_socket)) {
350 unlink($thiz->file_socket);
353 $old_umask = umask(0);
354 if (($thiz->list = stream_socket_server($thiz->unix_socket, $err, $errs)) === FALSE) {
358 stream_set_blocking($thiz->list, $thiz->blocking_mode); # Set the stream to non-blocking
360 if (($thiz->in = fopen("php://stdin", "r")) === FALSE) {
364 $thiz->main_loop = FALSE;
369 function socks_set($sock, $user)
373 $this->s2u[$id] = $user;
374 $this->socks[$id] = $sock;
377 function socks_unset($sock)
381 unset($this->s2u[$id]);
382 unset($this->socks[$id]);
385 function pgflush_try_add($enc, &$new_socket, $tout, $header_out, $content)
387 $pgflush = new PageFlush($new_socket, $enc, $this->curtime, $tout, $header_out, $content);
389 if ($pgflush->try_flush($this->curtime) == FALSE) {
390 // Add $pgflush to the pgflush array
391 $this->pgflush_add($pgflush);
395 function pgflush_add($pgflush)
397 array_push($this->pages_flush, $pgflush);
400 function garbage_manager($force)
402 $this->app->garbage_manager($force);
404 foreach ($this->socks as $k => $sock) {
405 if ($this->s2u[intval($sock)]->sess == '') {
406 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
407 $this->s2u[intval($sock)]->rd_socket_set(NULL);
409 unset($this->socks[intval($sock)]);
410 unset($this->s2u[intval($sock)]);
412 printf("CLOSE ON GARBAGE MANAGER\n");
419 GLOBAL $DOCUMENT_ROOT, $HTTP_HOST, $G_with_splash;
421 if ($this->main_loop) {
425 $this->main_loop = TRUE;
427 while ($this->main_loop) {
428 $this->curtime = time();
429 printf("IN LOOP: Current opened: %d pages_flush: %d - ", count($this->socks), count($this->pages_flush));
431 /* Prepare the read array */
432 /* // when we manage it ... */
434 /* $read = array_merge(array("$in" => $in), $socks); */
436 $read = array_merge(array(intval($this->list) => $this->list, intval($this->in) => $this->in),
439 if ($this->debug > 1) {
440 printf("PRE_SELECT\n");
445 $num_changed_sockets = stream_select($read, $write, $except, 0, 500000);
447 if ($num_changed_sockets == 0) {
448 printf(" no data in 5 secs, splash [%d]\n", $G_with_splash);
450 else if ($num_changed_sockets > 0) {
451 printf("num sock %d num_of_socket: %d\n", $num_changed_sockets, count($read));
452 if ($this->debug > 1) {
455 /* At least at one of the sockets something interesting happened */
456 foreach ($read as $i => $sock) {
457 /* is_resource check is required because there is the possibility that
458 during new request an old connection is closed */
459 if (!is_resource($sock)) {
462 if ($sock === $this->list) {
463 printf("NUOVA CONNEX\n");
464 if (($new_unix = stream_socket_accept($this->list)) == FALSE) {
465 printf("SOCKET_ACCEPT FAILED\n");
473 if (($new_socket = ancillary_getstream($new_unix, $stream_info)) !== FALSE) {
474 printf("NEW_SOCKET: %d\n", intval($new_socket));
475 stream_set_blocking($new_socket, $this->blocking_mode); // Set the stream to non-blocking
476 printf("RECEIVED HEADER:\n%s", $stream_info);
477 $path = spu_process_info($stream_info, $method, $header, $get, $post, $cookie);
478 printf("PATH: [%s]\n", $path);
479 printf("M: %s\nHEADER:\n", $method);
487 $addr = stream_socket_get_name($new_socket, TRUE);
488 $header_out = array();
490 $enc = get_encoding($header);
492 $subs = SITE_PREFIX."briskin5/";
493 $subs_l = strlen($subs);
495 if (!strncmp($path, SITE_PREFIX, SITE_PREFIX_LEN)) {
496 $rret = $this->app->request_mgr($this, $enc, $header_out, $new_socket, substr($path, SITE_PREFIX_LEN), $addr, $get, $post, $cookie);
498 if ($rret == FALSE) {
499 // FIXME: manage 404 !!!
500 printf("TODO: fix unknown page\n");
502 printf("number of sockets after %d\n", count($this->socks));
505 printf("WARNING: ancillary_getstream failed\n");
509 $buf = fread($sock, 512);
510 // if socket is closed
511 if ($buf == FALSE || strlen($buf) == 0) {
513 printf("ERROR READING\n");
515 if ($sock === $this->list) {
516 printf("Arrivati %d bytes da list\n", strlen($buf));
519 else if ($sock === $this->in) {
520 printf("Arrivati %d bytes da stdin\n", strlen($buf));
524 // $user_a[$s2u[intval($sock)]]->disable();
525 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
526 $this->s2u[intval($sock)]->rd_socket_set(NULL);
528 unset($this->socks[intval($sock)]);
529 unset($this->s2u[intval($sock)]);
532 printf("CLOSE ON READ\n");
534 if ($this->debug > 1) {
535 printf("post unset\n");
536 print_r($this->socks);
540 if ($this->debug > 1) {
543 if ($sock === $this->list) {
544 printf("Arrivati %d bytes da list\n", strlen($buf));
546 else if ($sock === $this->in) {
547 printf("Arrivati %d bytes da stdin\n", strlen($buf));
549 if ($line == "reload") {
550 require("$DOCUMENT_ROOT/Etc/".BRISK_CONF);
554 $key = array_search("$sock", $this->socks);
555 printf("Arrivati %d bytes dalla socket n. %d\n", strlen($buf), $key);
562 $this->garbage_manager(FALSE);
564 /* manage unfinished pages */
565 foreach ($this->pages_flush as $k => $pgflush) {
566 if ($pgflush->try_flush($this->curtime) == TRUE) {
567 unset($this->pages_flush[$k]);
572 $response: raw stream data not sent
573 $content: html consistent data (<script bla bla>)
574 $user->stream_keepalive($w_ping) ping srv->cli OR srv->cli + cli->srv if $w_ping == TRUE
577 /* manage open streaming */
578 foreach ($this->socks as $k => $sock) {
579 if (isset($this->s2u[intval($sock)])) {
580 $user = $this->s2u[intval($sock)];
581 $response = $user->rd_cache_get();
583 if (($this->curtime - $user->lacc) > (EXPIRE_TIME_RD / 3)) {
587 $user->ping_req = FALSE;
590 if ($response == "") {
592 $user->stream_main($content, $get, $post, $cookie);
593 printf("[%s] [%d] [%d]\n", $user->name, $user->lacc, $this->curtime);
594 if ($do_ping && $user->ping_req == FALSE) {
595 $content .= $user->stream_keepalive(TRUE);
596 $user->ping_req = TRUE;
598 else if ($content == "" && $user->rd_kalive_is_expired($this->curtime)) {
599 $content = $user->stream_keepalive(FALSE);
601 if ($content != "") {
602 $response = chunked_content($user->rd_zls_get(), $content);
606 if ($response != "") {
607 // echo "SPIA: [".substr($response, 0, 60)."...]\n";
608 echo "SPIA: [".$response."]\n";
609 $response_l = mb_strlen($response, "ASCII");
610 $wret = @fwrite($sock, $response);
611 if ($wret < $response_l) {
612 printf("TROUBLE WITH FWRITE: %d\n", $wret);
613 $user->rd_cache_set(mb_substr($response, $wret, $response_l - $wret, "ASCII"));
616 $user->rd_cache_set("");
619 $user->rd_kalive_reset($this->curtime);
622 // close socket after a while to prevent client memory consumption
623 if ($user->rd_endtime_is_expired($this->curtime)) {
624 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
625 $this->s2u[intval($sock)]->rd_socket_set(NULL);
627 unset($this->socks[intval($sock)]);
628 unset($this->s2u[intval($sock)]);
630 printf("CLOSE ON LOOP\n");
633 } // foreach ($this->socks...
636 } // function run(...