mirrors/PocketMine-MP
1
0
Fork 0
mirror of https://github.com/pmmp/PocketMine-MP.git synced 2025-04-23 00:55:57 +00:00
Code Issues Projects Releases Wiki Activity

Multiple RCON clients per thread, allow external connections

Browse Source
This commit is contained in:
Shoghi Cervantes Pueyo 2013-04-22 16:28:24 +02:00
parent d08807abaf
commit e81793174b
1 changed files with 102 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

179
src/utils/RCON.php
View File

@ -30,7 +30,7 @@ the Free Software Foundation, either version 3 of the License, or
class RCON{
private $socket, $password, $workers, $threads;
public function __construct($password, $port = 19132, $threads = 4){
public function __construct($password, $port = 19132, $interface = "0.0.0.0", $threads = 4, $clientsPerThread = 5){
$this->workers = array();
$this->password = (string) $password;
console("[INFO] Starting remote control listener");
@ -39,14 +39,15 @@ class RCON{
return;
}
$this->threads = (int) max(1, $threads);
$this->socket = socket_create_listen((int) $port);
if($this->socket === false){
$this->clientsPerThread = (int) max(1, $clientsPerThread);
$this->socket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
if($this->socket === false or !socket_bind($this->socket, $interface, (int) $port) or !socket_listen($this->socket)){
console("[ERROR] RCON can't be started: ".socket_strerror(socket_last_error()));
return;
}
@socket_set_nonblock($this->socket);
for($n = 0; $n < $this->threads; ++$n){
$this->workers[$n] = new RCONInstance($this->socket, $this->password);
$this->workers[$n] = new RCONInstance($this->socket, $this->password, $this->clientsPerThread);
}
@socket_getsockname($this->socket, $addr, $port);
console("[INFO] RCON running on $addr:$port");
@ -65,7 +66,7 @@ class RCON{
public function check(){
for($n = 0; $n < $this->threads; ++$n){
if($this->workers[$n]->isTerminated() === true){
$this->workers[$n] = new RCONInstance($this->socket, $this->password);
$this->workers[$n] = new RCONInstance($this->socket, $this->password, $this->clientsPerThread);
}elseif($this->workers[$n]->isWaiting()){
if($this->workers[$n]->response !== ""){
console($this->workers[$n]->response);
@ -87,107 +88,131 @@ class RCONInstance extends Thread{
private $socket;
private $password;
private $status;
private $client;
private $maxClients;
public function __construct($socket, $password){
public function __construct($socket, $password, $maxClients = 5){
$this->stop = false;
$this->cmd = "";
$this->response = "";
$this->socket = $socket;
$this->password = $password;
$this->status = 0;
$this->maxClients = (int) $maxClients;
for($n = 0; $n < $this->maxClients; ++$n){
$this->{"client".$n} = null;
$this->{"status".$n} = 0;
}
$this->status = array();
$this->start();
}
private function writePacket($requestID, $packetType, $payload){
return socket_write($this->client, Utils::writeLInt(strlen($payload)).Utils::writeLInt((int) $requestID).Utils::writeLInt((int) $packetType).($payload === "" ? "\x00":$payload)."\x00");
private function writePacket($client, $requestID, $packetType, $payload){
return socket_write($client, Utils::writeLInt(strlen($payload)).Utils::writeLInt((int) $requestID).Utils::writeLInt((int) $packetType).($payload === "" ? "\x00":$payload)."\x00");
}
private function readPacket(&$size, &$requestID, &$packetType, &$payload){
@socket_set_nonblock($this->client);
while(true){
usleep(1);
$d = socket_read($this->client, 4);
if($this->stop === true){
return false;
}elseif($d === false){
continue;
}elseif($d === ""){
return false;
}
break;
private function readPacket($client, &$size, &$requestID, &$packetType, &$payload){
@socket_set_nonblock($client);
$d = socket_read($client, 4);
if($this->stop === true){
return false;
}elseif($d === false){
return null;
}elseif($d === ""){
return false;
}
@socket_set_block($this->client);
@socket_set_block($client);
$size = Utils::readLInt($d);
if($size < 0){
return false;
}
$requestID = Utils::readLInt(socket_read($this->client, 4));
$packetType = Utils::readLInt(socket_read($this->client, 4));
$payload = rtrim(socket_read($this->client, $size + 2)); //Strip two null bytes
$requestID = Utils::readLInt(socket_read($client, 4));
$packetType = Utils::readLInt(socket_read($client, 4));
$payload = rtrim(socket_read($client, $size + 2)); //Strip two null bytes
return true;
}
public function close(){
public function close(){
$this->stop = true;
$this->status = -1;
}
public function run(){
while($this->stop !== true){
usleep(1000);
if(($this->client = socket_accept($this->socket)) !== false){
socket_set_block($this->client);
socket_set_option($this->client, SOL_SOCKET, SO_KEEPALIVE, 1);
while($this->status !== -1 and $this->stop !== true){
if($this->readPacket($size, $requestID, $packetType, $payload) === false){
$this->status = -1;
usleep(1);
if(($client = socket_accept($this->socket)) !== false){
socket_set_block($client);
socket_set_option($client, SOL_SOCKET, SO_KEEPALIVE, 1);
$done = false;
for($n = 0; $n < $this->maxClients; ++$n){
if($this->{"client".$n} === null){
$this->{"client".$n} = $client;
$this->{"status".$n} = 0;
$done = true;
break;
}
switch($packetType){
case 3: //Login
if($this->status !== 0){
$this->status = -1;
continue;
}
if($payload === $this->password){
@socket_getpeername($this->client, $addr, $port);
$this->response = "[INFO] Successful Rcon connection from: /$addr:$port";
$this->wait();
$this->response = "";
$this->writePacket($requestID, 2, "");
$this->status = 1;
}else{
$this->status = -1;
$this->writePacket(-1, 2, "");
continue;
}
break;
case 2: //Command
if($this->status !== 1){
$this->status = -1;
continue;
}
if(strlen($payload) > 0){
$this->cmd = ltrim($payload);
$this->wait();
$this->writePacket($requestID, 0, str_replace("\n", "\r\n", trim($this->response)));
$this->response = "";
$this->cmd = "";
}
break;
}
usleep(1);
}
@socket_set_option($this->client, SOL_SOCKET, SO_LINGER, array("l_onoff" => 1, "l_linger" => 1));
@socket_shutdown($this->client, 2);
@socket_set_block($this->client);
@socket_read($this->client, 1);
@socket_close($this->client);
$this->status = 0;
if($done === false){
@socket_close($client);
}
}
for($n = 0; $n < $this->maxClients; ++$n){
$client = &$this->{"client".$n};
if($client !== null){
if($this->{"status".$n} !== -1 and $this->stop !== true){
$p = $this->readPacket($client, $size, $requestID, $packetType, $payload);
if($p === false){
$this->{"status".$n} = -1;
continue;
}elseif($p === null){
continue;
}
switch($packetType){
case 3: //Login
if($this->{"status".$n} !== 0){
$this->{"status".$n} = -1;
continue;
}
if($payload === $this->password){
@socket_getpeername($client, $addr, $port);
$this->response = "[INFO] Successful Rcon connection from: /$addr:$port";
$this->wait();
$this->response = "";
$this->writePacket($client, $requestID, 2, "");
$this->{"status".$n} = 1;
}else{
$this->{"status".$n} = -1;
$this->writePacket($client, -1, 2, "");
continue;
}
break;
case 2: //Command
if($this->{"status".$n} !== 1){
$this->{"status".$n} = -1;
continue;
}
if(strlen($payload) > 0){
$this->cmd = ltrim($payload);
$this->wait();
$this->writePacket($client, $requestID, 0, str_replace("\n", "\r\n", trim($this->response)));
$this->response = "";
$this->cmd = "";
}
break;
}
usleep(1);
}else{
@socket_set_option($client, SOL_SOCKET, SO_LINGER, array("l_onoff" => 1, "l_linger" => 1));
@socket_shutdown($client, 2);
@socket_set_block($client);
@socket_read($client, 1);
@socket_close($client);
$this->{"status".$n} = 0;
$this->{"client".$n} = null;
}
}
}
}
unset($this->client, $this->socket, $this->cmd, $this->response, $this->stop, $this->status);
unset($this->socket, $this->cmd, $this->response, $this->stop);
exit(0);
}
}