php socket实现的聊天室代码分享

<div class="jb51code">
<pre class="brush: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,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],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 ) . "
  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();

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!