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)
109 if (isset($header['Location'])) {
110 return sprintf("HTTP/1.1 302 OK\r\nLocation: %s\r\n\r\n", $header['Location']);
113 $s .= "HTTP/1.1 200 OK\r\n";
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);
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";
131 $s .= sprintf("Content-Length: %d\r\n", $len);
139 * Caching system using ob php system to cache old style pages
140 * to a var and than send it with more calm
145 log_rd2("SHUTTA [".connection_status()."] !");
148 register_shutdown_function('shutta');
154 function chunked_content($content)
156 $content_l = mb_strlen($content, "ASCII");
158 return (sprintf("%X\r\n%s\r\n", $content_l, $content));
161 function chunked_fini()
163 return sprintf("0\r\n");
167 static $fixed_fd = 2;
189 function Sac_a_push()
193 // Sac_a_push::create("/tmp/brisk.sock", 0, 0
195 static function create(&$app, $sockname, $debug, $blocking_mode)
197 $thiz = new Sac_a_push();
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();
207 $thiz->blocking_mode = 0; // 0 for non-blocking
210 for ($i = 0 ; $i < 4096 ; $i++) {
211 $thiz->rndstr .= chr(mt_rand(65, 90));
214 if (file_exists($thiz->file_socket)) {
215 unlink($thiz->file_socket);
218 $old_umask = umask(0);
219 if (($thiz->list = stream_socket_server($thiz->unix_socket, $err, $errs)) === FALSE) {
223 stream_set_blocking($thiz->list, $thiz->blocking_mode); # Set the stream to non-blocking
225 if (($thiz->in = fopen("php://stdin", "r")) === FALSE) {
229 $thiz->main_loop = FALSE;
234 function socks_set($sock, $user)
238 $this->s2u[$id] = $user;
239 $this->socks[$id] = $sock;
242 function socks_unset($sock)
246 unset($this->s2u[$id]);
247 unset($this->socks[$id]);
250 function pgflush_try_add(&$new_socket, $tout, $header_out, $content)
252 $pgflush = new PageFlush($new_socket, $this->curtime, $tout, $header_out, $content);
254 if ($pgflush->try_flush($this->curtime) == FALSE) {
255 // Add $pgflush to the pgflush array
256 $this->pgflush_add($pgflush);
260 function pgflush_add($pgflush)
262 array_push($this->pages_flush, $pgflush);
265 function garbage_manager($force)
267 $this->app->garbage_manager($force);
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);
274 unset($this->socks[intval($sock)]);
275 unset($this->s2u[intval($sock)]);
277 printf("CLOSE ON GARBAGE MANAGER\n");
284 if ($this->main_loop) {
288 $this->main_loop = TRUE;
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));
294 /* Prepare the read array */
295 /* // when we manage it ... */
297 /* $read = array_merge(array("$in" => $in), $socks); */
299 $read = array_merge(array(intval($this->list) => $this->list, intval($this->in) => $this->in),
302 if ($this->debug > 1) {
303 printf("PRE_SELECT\n");
308 $num_changed_sockets = stream_select($read, $write, $except, 0, 100000);
310 if ($num_changed_sockets == 0) {
311 printf(" no data in 5 secs ");
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) {
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)) {
325 if ($sock === $this->list) {
326 printf("NUOVA CONNEX\n");
327 $new_unix = stream_socket_accept($this->list);
333 if (($new_socket = ancillary_getstream($new_unix, $stream_info)) !== FALSE) {
334 printf("NEW_SOCKET: %d\n", intval($new_socket));
335 stream_set_blocking($new_socket, $this->blocking_mode); // Set the stream to non-blocking
336 printf("RECEIVED HEADER:\n%s", $stream_info);
337 $path = spu_process_info($stream_info, $method, $header, $get, $post, $cookie);
338 printf("PATH: [%s]\n", $path);
339 printf("M: %s\nHEADER:\n", $method);
348 $addr = stream_socket_get_name($new_socket, TRUE);
349 $header_out = array();
351 $subs = SITE_PREFIX."briskin5/";
352 $subs_l = strlen($subs);
354 if (!strncmp($path, SITE_PREFIX, SITE_PREFIX_LEN)) {
355 $rret = $this->app->request_mgr($this, $header_out, $new_socket, substr($path, SITE_PREFIX_LEN), $addr, $get, $post, $cookie);
357 if ($rret == FALSE) {
358 // FIXME: manage 404 !!!
359 printf("TODO: fix unknown page\n");
361 printf("number of sockets after %d\n", count($this->socks));
364 printf("WARNING: ancillary_getstream failed\n");
368 if (($buf = fread($sock, 512)) === FALSE) {
369 printf("error read\n");
371 else if (strlen($buf) === 0) {
372 if ($sock === $this->list) {
373 printf("Arrivati %d bytes da list\n", strlen($buf));
375 else if ($sock === $this->in) {
376 printf("Arrivati %d bytes da stdin\n", strlen($buf));
379 // $user_a[$s2u[intval($sock)]]->disable();
380 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
381 $this->s2u[intval($sock)]->rd_socket_set(NULL);
383 unset($this->socks[intval($sock)]);
384 unset($this->s2u[intval($sock)]);
386 printf("CLOSE ON READ\n");
388 if ($this->debug > 1) {
389 printf("post unset\n");
390 print_r($this->socks);
394 if ($this->debug > 1) {
397 if ($sock === $this->list) {
398 printf("Arrivati %d bytes da list\n", strlen($buf));
400 else if ($sock === $this->in) {
401 printf("Arrivati %d bytes da stdin\n", strlen($buf));
404 $key = array_search("$sock", $this->socks);
405 printf("Arrivati %d bytes dalla socket n. %d\n", strlen($buf), $key);
412 $this->garbage_manager(FALSE);
414 /* manage unfinished pages */
415 foreach ($this->pages_flush as $k => $pgflush) {
416 if ($pgflush->try_flush($this->curtime) == TRUE) {
417 unset($this->pages_flush[$k]);
422 $response: raw stream data not sent
423 $content: html consistent data (<script bla bla>)
424 $user->stream_keepalive($w_ping) ping srv->cli OR srv->cli + cli->srv if $w_ping == TRUE
427 /* manage open streaming */
428 foreach ($this->socks as $k => $sock) {
429 if (isset($this->s2u[intval($sock)])) {
430 $user = $this->s2u[intval($sock)];
431 $response = $user->rd_cache_get();
433 if (($this->curtime - $user->lacc) > (EXPIRE_TIME_RD / 3)) {
437 $user->ping_req = FALSE;
440 if ($response == "") {
442 $user->stream_main($content, $get, $post, $cookie);
443 printf("[%s] [%d] [%d]\n", $user->name, $user->lacc, $this->curtime);
444 if ($do_ping && $user->ping_req == FALSE) {
445 $content .= $user->stream_keepalive(TRUE);
446 $user->ping_req = TRUE;
448 else if ($content == "" && $user->rd_kalive_is_expired($this->curtime)) {
449 $content = $user->stream_keepalive(FALSE);
451 if ($content != "") {
452 $response = chunked_content($content);
456 if ($response != "") {
457 // echo "SPIA: [".substr($response, 0, 60)."...]\n";
458 echo "SPIA: [".$response."]\n";
459 $response_l = mb_strlen($response, "ASCII");
460 $wret = @fwrite($sock, $response);
461 if ($wret < $response_l) {
462 printf("TROUBLE WITH FWRITE: %d\n", $wret);
463 $user->rd_cache_set(mb_substr($response, $wret, $response_l - $wret, "ASCII"));
466 $user->rd_cache_set("");
469 $user->rd_kalive_reset($this->curtime);
472 // close socket after a while to prevent client memory consumption
473 if ($user->rd_endtime_is_expired($this->curtime)) {
474 if ($this->s2u[intval($sock)]->rd_socket_get() != NULL) {
475 $this->s2u[intval($sock)]->rd_socket_set(NULL);
477 unset($this->socks[intval($sock)]);
478 unset($this->s2u[intval($sock)]);
480 printf("CLOSE ON LOOP\n");
483 } // foreach ($this->socks...
486 } // function run(...