站长论坛

标题: 用PHP的Socket建立自己的聊天室服务器 [打印本页]

作者: jsper 时间: 2008-9-14 15:31
标题: 用PHP的Socket建立自己的聊天室服务器
<?PHP
/**
* patServer
* PHP socket server base class
* Events that can be handled:
* * onStart
* * onConnect
* * onConnectionRefused
* * onClose
* * onShutdown
* * onReceiveData
*
* @version 1.1
* @author Stephan Schmidt <schst@php-tools.de>
* @package patServer
*/
class patServer{
/**
* information about the project
* @var array $systemVars
*/
var $systemVars = array(

"appName" => "patServer",
"appVersion" => "1.1",
"author" => array("Stephan Schmidt <schst@php-tools.de>", )
);

/**
* port to listen
* @var integer $port
*/
var $port = 10000;

/**
* domain to bind to
* @var string $domain
*/
var $domain = "localhost";

/**
* maximum amount of clients
* @var integer $maxClients
*/
var $maxClients = -1;

/**
* buffer size for socket_read
* @var integer $readBufferSize
*/
var $readBufferSize = 128;

/**
* end character for socket_read
* @var integer $readEndCharacter
*/
var $readEndCharacter = "\n";

/**
* maximum of backlog in queue
* @var integer $maxQueue
*/
var $maxQueue = 500;

/**
* debug mode
* @var boolean $debug
*/
var $debug = true;

/**
* debug mode
* @var string $debugMode
*/
var $debugMode = "text";

/**
* debug destination (filename or stdout)
* @var string $debugDest
*/
var $debugDest = "stdout";

/**
* empty array, used for socket_select
* @var array $null
*/
var $null = array();

/**
* all file descriptors are stored here
* @var array $clientFD
*/
var $clientFD = array();

/**
* needed to store client information
* @var array $clientInfo
*/
var $clientInfo = array();

/**
* needed to store server information
* @var array $serverInfo
*/
var $serverInfo = array();

/**
* amount of clients
* @var integer $clients
*/
var $clients = 0;

/**
* create a new socket server
*
* @access public
* @param string $domain domain to bind to
* @param integer $port port to listen to
*/
function patServer( $domain = "localhost", $port = 10000 ){
$this->domain = $domain;
$this->port = $port;

$this->serverInfo["domain"]       = $domain;
$this->serverInfo["port"]       = $port;
$this->serverInfo["servername"]    = $this->systemVars["appName"];
$this->serverInfo["serverversion"] = $this->systemVars["appVersion"];

set_time_limit( 0 );
}

/**
* set maximum amount of simultaneous connections
*
* @access public
* @param int $maxClients
*/
function setMaxClients( $maxClients ){
$this->maxClients = $maxClients;
}

/**
* set debug mode
*
* @access public
* @param mixed $debug [text|htmlfalse]
* @param string $dest destination of debug message (stdout to output or filename if log should be written)
*/
function setDebugMode( $debug, $dest = "stdout" ){
if( $debug === false ){
$this->debug = false;
return true;
}

$this->debug = true;
$this->debugMode = $debug;
$this->debugDest = $dest;
}

/**
* start the server
*
* @access public
* @param int $maxClients
*/
function start(){
$this->initFD = @socket_create( AF_INET, SOCK_STREAM, 0 );
if( !$this->initFD )
die( "patServer: Could not create socket." );

// adress may be reused
socket_setopt( $this->initFD, SOL_SOCKET, SO_REUSEADDR, 1 );

// bind the socket
if( !@socket_bind( $this->initFD, $this->domain, $this->port ) ){
@socket_close( $this->initFD );
die( "patServer: Could not bind socket to ".$this->domain." on port ".$this->port." ( ".$this->getLastSocketError( $this->initFd )." )." );
}

// listen on selected port
if( !@socket_listen( $this->initFD, $this->maxQueue ) )
die( "patServer: Could not listen ( ".$this->getLastSocketError( $this->initFd )." )." );

$this->sendDebugMessage( "Listening on port ".$this->port.". Server started at ".date( "H:i:s", time() ) );

// this allows the shutdown function to check whether the server is already shut down
$GLOBALS["_patServerStatus"] = "running";
// this ensures that the server will be sutdown correctly
register_shutdown_function( array( $this, "shutdown" ) );

if( method_exists( $this, "onStart" ) )
$this->onStart();

$this->serverInfo["started"] = time();
$this->serverInfo["status"] = "running";

while( true ){
$readFDs = array();
array_push( $readFDs, $this->initFD );

// fetch all clients that are awaiting connections
for( $i = 0; $i < count( $this->clientFD ); $i++ )
   if( isset( $this->clientFD[$i] ) )
   array_push( $readFDs, $this->clientFD[$i] );

// block and wait for data or new connection
$ready = @socket_select( $readFDs, $this->null, $this->null, NULL );

if( $ready === false ){
   $this->sendDebugMessage( "socket_select failed." );
   $this->shutdown();
}

// check for new connection
if( in_array( $this->initFD, $readFDs ) ){
   $newClient = $this->acceptConnection( $this->initFD );

   // check for maximum amount of connections
   if( $this->maxClients > 0 ){
   if( $this->clients > $this->maxClients ){
   $this->sendDebugMessage( "Too many connections." );

   if( method_exists( $this, "onConnectionRefused" ) )
      $this->onConnectionRefused( $newClient );

   $this->closeConnection( $newClient );
   }
   }

   if( --$ready <= 0 )
   continue;
}

// check all clients for incoming data
for( $i = 0; $i < count( $this->clientFD ); $i++ ){
   if( !isset( $this->clientFD[$i] ) )
   continue;

   if( in_array( $this->clientFD[$i], $readFDs ) ){
   $data = $this->readFromSocket( $i );

   // empty data => connection was closed
   if( !$data ){
   $this->sendDebugMessage( "Connection closed by peer" );
   $this->closeConnection( $i );
   }else{
   $this->sendDebugMessage( "Received ".trim( $data )." from ".$i );

   if( method_exists( $this, "onReceiveData" ) )
      $this->onReceiveData( $i, $data );
   }
   }
}
}
}

/**
* read from a socket
*
* @access private
* @param integer $clientId internal id of the client to read from
* @return string $data data that was read
*/
function readFromSocket( $clientId ){
// start with empty string
$data = "";

// read data from socket
while( $buf = socket_read( $this->clientFD[$clientId], $this->readBufferSize ) ){
$data .= $buf;

$endString = substr( $buf, - strlen( $this->readEndCharacter ) );
if( $endString == $this->readEndCharacter )
   break;
if( $buf == NULL )
   break;
}

if( $buf === false )
$this->sendDebugMessage( "Could not read from client ".$clientId." ( ".$this->getLastSocketError( $this->clientFD[$clientId] )." )." );

return $data;
}

/**
* accept a new connection
*
* @access public
* @param resource &$socket socket that received the new connection
* @return int $clientID internal ID of the client
*/
function acceptConnection( &$socket ){
for( $i = 0 ; $i <= count( $this->clientFD ); $i++ ){
if( !isset( $this->clientFD[$i] ) || $this->clientFD[$i] == NULL ){
   $this->clientFD[$i] = socket_accept( $socket );
   socket_setopt( $this->clientFD[$i], SOL_SOCKET, SO_REUSEADDR, 1 );
   $peer_host = "";
   $peer_port = "";
   socket_getpeername( $this->clientFD[$i], $peer_host, $peer_port );
   $this->clientInfo[$i] = array(
            "host" => $peer_host,
            "port" => $peer_port,
            "connectOn" => time()
         );
   $this->clients++;

   $this->sendDebugMessage( "New connection ( ".$i." ) from ".$peer_host." on port ".$peer_port );

   if( method_exists( $this, "onConnect" ) )
   $this->onConnect( $i );
   return $i;
}
}
}

/**
* check, whether a client is still connected
*
* @access public
* @param integer $id client id
* @return boolean $connected true if client is connected, false otherwise
*/
function isConnected( $id ){
if( !isset( $this->clientFD[$id] ) )
return false;
return true;
}

/**
* close connection to a client
*
* @access public
* @param int $clientID internal ID of the client
*/
function closeConnection( $id ){
if( !isset( $this->clientFD[$id] ) )
return false;

if( method_exists( $this, "onClose" ) )
$this->onClose( $id );

$this->sendDebugMessage( "Closed connection ( ".$id." ) from ".$this->clientInfo[$id]["host"]." on port ".$this->clientInfo[$id]["port"] );

@socket_close( $this->clientFD[$id] );
$this->clientFD[$id] = NULL;
unset( $this->clientInfo[$id] );
$this->clients--;
}

/**
* shutdown server
*
* @access public
*/
function shutDown(){
if( $GLOBALS["_patServerStatus"] != "running" )
exit;
$GLOBALS["_patServerStatus"] = "stopped";

if( method_exists( $this, "onShutdown" ) )
$this->onShutdown();

$maxFD = count( $this->clientFD );
for( $i = 0; $i < $maxFD; $i++ )
$this->closeConnection( $i );

@socket_close( $this->initFD );

$this->sendDebugMessage( "Shutdown server." );
exit;
}

/**
* get current amount of clients
*
* @access public
* @return int $clients amount of clients
*/
function getClients(){
return $this->clients;
}

/**
* send data to a client
*
* @access public
* @param int $clientId ID of the client
* @param string $data data to send
* @param boolean $debugData flag to indicate whether data that is written to socket should also be sent as debug message
*/
function sendData( $clientId, $data, $debugData = true ){
if( !isset( $this->clientFD[$clientId] ) || $this->clientFD[$clientId] == NULL )
return false;

if( $debugData )
$this->sendDebugMessage( "sending: \"" . $data . "\" to: $clientId" );

if( !@socket_write( $this->clientFD[$clientId], $data ) )
$this->sendDebugMessage( "Could not write '".$data."' client ".$clientId." ( ".$this->getLastSocketError( $this->clientFD[$clientId] )." )." );
}

/**
* send data to all clients
*
* @access public
* @param string $data data to send
* @param array $exclude client ids to exclude
*/
function broadcastData( $data, $exclude = array(), $debugData = true ){
if( !empty( $exclude ) && !is_array( $exclude ) )
$exclude = array( $exclude );

for( $i = 0; $i < count( $this->clientFD ); $i++ ){
if( isset( $this->clientFD[$i] ) && $this->clientFD[$i] != NULL && !in_array( $i, $exclude ) ){
   if( $debugData )
   $this->sendDebugMessage( "sending: \"" . $data . "\" to: $i" );

   if( !@socket_write( $this->clientFD[$i], $data ) )
   $this->sendDebugMessage( "Could not write '".$data."' client ".$i." ( ".$this->getLastSocketError( $this->clientFD[$i] )." )." );
}
}
}

/**
* get current information about a client
*
* @access public
* @param int $clientId ID of the client
* @return array $info information about the client
*/
function getClientInfo( $clientId ){
if( !isset( $this->clientFD[$clientId] ) || $this->clientFD[$clientId] == NULL )
return false;
return $this->clientInfo[$clientId];
}

/**
* send a debug message
*
* @access private
* @param string $msg message to debug
*/
function sendDebugMessage( $msg ){
if( !$this->debug )
return false;

$msg = date( "Y-m-d H:i:s", time() ) . " " . $msg;

switch( $this->debugMode ){
case "text":
   $msg = $msg."\n";
   break;
case "html":
   $msg = htmlspecialchars( $msg ) . "<br />\n";
   break;
}

if( $this->debugDest == "stdout" || empty( $this->debugDest ) ){
echo $msg;
flush();
return true;
}

error_log( $msg, 3, $this->debugDest );
return true;
}

/**
* return string for last socket error
*
* @access public
* @return string $error last error
*/
function getLastSocketError( &$fd ){
$lastError = socket_last_error( $fd );
return "msg: " . socket_strerror( $lastError ) . " / Code: ".$lastError;
}
function onReceiveData($ip,$data){
  $this->broadcastData( $data,array(), true );
}
}
$patServer = new patServer();
$patServer->start();
?>

欢迎光临站长论坛 (http://www.tzlink.com/bbs/)