thinkphp socket to support sso (php socket 不稳定，不建议使用)

86565656

cli code

#!/bin/bash
filepath=$(cd "$(dirname "$0")"; pwd)
cd $filepath
#running status:sso.lock,stop status:ssostop.lock
TMPFILE=$filepath/sso.lock
STOPFILE=$filepath/ssostop.lock
runCLI(){
touch $TMPFILE
echo "sso server process started.\n"
logname="sso_"$(date +%Y%m%d);
su www-data -c "php index.php sso start 1>>./Log/$logname.log 2>&1"
rm -f $STOPFILE;
}
restartCLI(){
stopCLI
echo "Waiting for stop sso process."
sleep 1
echo "Exit prev sso process."
runCLI &   
}
stopCLI(){
if [ -f $TMPFILE ];then
touch $STOPFILE
fi
rm -f ${TMPFILE};
logname="sso_"$(date +%Y%m%d);
su www-data -c "php index.php sso stop 1>>./Log/$logname.log 2>&1"
echo "Sent stop sso server signal."
}
if [ "$1" = "-stop" ] ;then
stopCLI
elif [ "$1" = "-start" -o "$1" = "" ] ;then
if [ -f $TMPFILE ];then
echo "Another process is running."
else
runCLI &
fi
elif [ "$1" = "-restart" ] ;then
restartCLI
else
echo "Please select option -start -restart -stop"
fi


socket server code

<?php
namespace Admin\Action;
use Think\Action;
class SsoAction extends Action {
private $sock;
private $clients = array();
// 停止socket。kill 后，有可能socket临时文件还是没删除。会导致同一个端口无法启动
//check port status:netstat -ano |grep port
//search pid: lsof -i:port
//kill pid
//rm -f sso.lock
public function run($taskType)
{
switch ($taskType) {
case 'start':
$this->_start();
break;
default:
echo datetime() . " unknow task type!" . PHP_EOL;
break;
}
}
public function index() {
echo datetime() . " server msg: ". $_SERVER['argv'][1] . " " . $_SERVER['argv'][2] . PHP_EOL;
$this->run($_SERVER['argv'][2]);
}
private function _start()
{
set_time_limit(0);
ob_implicit_flush();
if (($this->sock = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)) === false) {
echo datetime() . " socket_create() failed: reason:" . socket_strerror(socket_last_error()) . "\n";
exit;
}
// start
if (socket_bind($this->sock, C('SSO_ADDRESS'), C('SSO_PORT')) === false) {
echo datetime() . " socket_bind failed, reason:" . socket_strerror(socket_last_error($sock)) . PHP_EOL;
exit;
}
// listen
if (socket_listen($this->sock, 5) === false) {
echo datetime() . " socket_listen() failed: reason: " . socket_strerror(socket_last_error($sock)) . PHP_EOL;
exit;
}
echo datetime() . " start socket server success!" . PHP_EOL;
do {
// accept socket msg failed
if (($msgsock = socket_accept($this->sock)) === false) {
echo datetime() . " socket_accepty() failed :reason:" . socket_strerror(socket_last_error($this->sock)) . PHP_EOL;
break;
}
//read client message
$buf = socket_read($msgsock, 1024);
$data = json_decode($buf, true);
$userID = $data['userID'];
if ($data['msg'] == 'logout') {
// clear session table
$this->_logout($userID);
} elseif ($data['msg'] == "check") {
// check cache with user ID
$back = S($userID) ? "success" : "fail";
echo datetime() . " no cache of {$userID} " . PHP_EOL;
if (false === socket_write($msgsock, $back)) {
echo datetime() . " socket_write() failed reason:" . socket_strerror(socket_last_error($this->sock)) . PHP_EOL;
}
}
socket_close($msgsock);
} while (true);
}
private function _logout($userID) {
if (!$userID) return false;
S($userID, null);
echo datetime() . " userID:{$userID} logout by client. " . PHP_EOL;
}
}
?>


socket client code

<?php
namespace Admin\Action;
use Think\Action;
use Org\Util\Socket;
class SsoClientAction extends Action {
private $SocketClient;
/**
* loop check sso server auth status
*/
public function run()
{
$result = true;
while($result===true) {
$this->_connect();
$result = $this->send();
if (!$result) {
// logout
session_destroy();
}
sleep(3);
}
//$this->_connect();
//$result = $this->send();
}
private function _connect()
{
$config = array('persistent'=> false,
'host'=> C("SSO_ADDRESS"),
'protocol'=> 'tcp',
'port'=> C("SSO_PORT"),
'timeout'=> 5);
$this->SocketClient = new Socket($config);
$this->SocketClient->connect();
}
/**
* @param msg check or logout string
*/
public function send($msg="check")
{
$userID = $_SESSION[C('USER_AUTH_KEY')];
if (!$userID) $msg = "logout";
$arr = array("msg"=>$msg, $userID=>$userID);
$data = json_encode($arr);
$result = $this->SocketClient->write($data);
//echo datetime() . " send msg:{$msg} \r\n"  . PHP_EOL;
$response = $this->SocketClient->read();
echo datetime() . " " . $response . " from sso server \r\n" . PHP_EOL;
$this->SocketClient->disconnect();
if ($response == "logout" || $response == "fail") {
echo datetime . " logout or fail \r\n" . PHP_EOL;
session_destroy();
return false;
}
return true;
}
}
?>