This commit is contained in:
Daniil Gentili 2017-03-24 21:13:23 +01:00
commit 7971237fe1
18 changed files with 152 additions and 91 deletions

View File

@ -36,6 +36,7 @@ class API extends APIFactory
$this->API->v = $this->API->getV();
\danog\MadelineProto\Logger::log(['MadelineProto is ready!'], Logger::NOTICE);
}
/*
public function __sleep()
{

View File

@ -105,7 +105,8 @@ class Connection
case 'https':
try {
fclose($this->sock);
} catch (\danog\MadelineProto\Exception $e) { ; }
} catch (\danog\MadelineProto\Exception $e) {
}
break;
case 'udp':
throw new Exception("Connection: This protocol wasn't implemented yet.");

View File

@ -24,9 +24,11 @@ class DataCenter
public $dclist = [];
public $settings = [];
public function __sleep() {
public function __sleep()
{
return ['sockets', 'curdc', 'dclist', 'settings'];
}
public function __construct(&$dclist, &$settings)
{
$this->dclist = &$dclist;
@ -71,13 +73,18 @@ class DataCenter
\danog\MadelineProto\Logger::log(['Connecting to DC '.$dc_number.' ('.$test.' server, '.$ipv6.', '.$this->settings[$dc_number]['protocol'].')...'], \danog\MadelineProto\Logger::VERBOSE);
$this->sockets[$dc_number] = new Connection($address, $port, $this->settings[$dc_number]['protocol'], $this->settings[$dc_number]['timeout']);
return true;
}
public function get_dcs() {
public function get_dcs()
{
$test = $this->settings[2]['test_mode'] ? 'test' : 'main';
$ipv6 = $this->settings[2]['ipv6'] ? 'ipv6' : 'ipv4';
return array_keys($this->dclist[$test][$ipv6]);
}
public function &__get($name)
{
return $this->sockets[$this->curdc]->{$name};

View File

@ -14,13 +14,14 @@ namespace danog\MadelineProto;
class Exception extends \Exception
{
public function __construct($message = null, $code = 0, Exception $previous = null) {
public function __construct($message = null, $code = 0, Exception $previous = null)
{
parent::__construct($message, $code, $previous);
if (\danog\MadelineProto\Logger::$constructed && $this->file !== __FILE__) {
\danog\MadelineProto\Logger::log([$message.' in '.basename($this->file).':'.$this->line], \danog\MadelineProto\Logger::FATAL_ERROR);
}
}
/**
* ExceptionErrorHandler.
*

View File

@ -29,8 +29,6 @@ class Logger
const ERROR = 1;
const FATAL_ERROR = 0;
/*
* Constructor function
* Accepts various logger modes:
@ -53,7 +51,6 @@ class Logger
public static function log($params, $level = self::NOTICE)
{
if (!self::$constructed) {
throw new Exception("The constructor function wasn't called! Please call the constructor function before using this method.");
}

View File

@ -100,17 +100,22 @@ class MTProto
$this->v = $this->getV();
$this->should_serialize = true;
}
public function setup_threads() {
if ($this->threads = $this->run_workers = class_exists('\Pool') && php_sapi_name() == "cli" && $this->settings['threading']['allow_threading']) {
public function setup_threads()
{
if ($this->threads = $this->run_workers = class_exists('\Pool') && php_sapi_name() == 'cli' && $this->settings['threading']['allow_threading']) {
\danog\MadelineProto\Logger::log(['THREADING IS ENABLED'], \danog\MadelineProto\Logger::NOTICE);
$this->start_threads();
}
}
public function start_threads() {
public function start_threads()
{
if ($this->threads) {
$dcs = $this->datacenter->get_dcs();
if (!isset($this->reader_pool)) $this->reader_pool = new \Pool(count($dcs));
if (!isset($this->reader_pool)) {
$this->reader_pool = new \Pool(count($dcs));
}
foreach ($dcs as $dc) {
if (!isset($this->readers[$dc])) {
$this->readers[$dc] = new \danog\MadelineProto\Threads\SocketReader($this, $dc);
@ -125,14 +130,22 @@ class MTProto
}
}
}
public function __sleep() {
public function __sleep()
{
$t = get_object_vars($this);
if (isset($t['reader_pool'])) unset($t['reader_pool']);
if (isset($t['reader_pool'])) {
unset($t['reader_pool']);
}
return array_keys($t);
}
public function __wakeup()
{
if (debug_backtrace()[0]['file'] === __DIR__.'/Threads/SocketReader.php' || (debug_backtrace()[0]['file'] === __FILE__ && debug_backtrace()[0]['line'] === 117)) return;
if (debug_backtrace()[0]['file'] === __DIR__.'/Threads/SocketReader.php' || (debug_backtrace()[0]['file'] === __FILE__ && debug_backtrace()[0]['line'] === 117)) {
return;
}
$this->bigint = PHP_INT_SIZE < 8;
$this->setup_logger();
if (!isset($this->v) || $this->v !== $this->getV()) {
@ -147,13 +160,16 @@ class MTProto
$this->get_updates_difference();
}
}
public function __destruct() {
public function __destruct()
{
if (isset($this->reader_pool)) {
$this->run_workers = false;
\danog\MadelineProto\Logger::log(['Shutting down reader pool...'], Logger::NOTICE);
$this->reader_pool->shutdown();
}
}
public function parse_settings($settings)
{
// Detect ipv6
@ -304,7 +320,7 @@ Slv8kg9qv1m6XHVQY3PnEw+QQtqSIXklHwIDAQAB
],
'threading' => [
'allow_threading' => false, // Should I use threading, if it is enabled?
'handler_workers' => 10 // How many workers should every message handler pool of each socket reader have
'handler_workers' => 10, // How many workers should every message handler pool of each socket reader have
],
'pwr' => ['pwr' => false, 'db_token' => false, 'strict' => false],
];
@ -333,7 +349,6 @@ Slv8kg9qv1m6XHVQY3PnEw+QQtqSIXklHwIDAQAB
$this->should_serialize = true;
}
public function setup_logger()
{
\danog\MadelineProto\Logger::constructor($this->settings['logger']['logger'], $this->settings['logger']['logger_param'], isset($this->authorization['user']) ? (isset($this->authorization['user']['username']) ? $this->authorization['user']['username'] : $this->authorization['user']['id']) : '', isset($this->settings['logger']['logger_level']) ? $this->settings['logger']['logger_level'] : Logger::VERBOSE);
@ -360,11 +375,15 @@ Slv8kg9qv1m6XHVQY3PnEw+QQtqSIXklHwIDAQAB
public function connect_to_all_dcs()
{
foreach ($old = $this->datacenter->get_dcs() as $new_dc) {
if (!isset($this->datacenter->sockets[$new_dc])) $this->datacenter->dc_connect($new_dc);
if (!isset($this->datacenter->sockets[$new_dc])) {
$this->datacenter->dc_connect($new_dc);
}
}
$this->setup_threads();
$this->init_authorization();
if ($old !== $this->datacenter->get_dcs()) $this->connect_to_all_dcs();
if ($old !== $this->datacenter->get_dcs()) {
$this->connect_to_all_dcs();
}
}
// Creates authorization keys
@ -393,9 +412,13 @@ Slv8kg9qv1m6XHVQY3PnEw+QQtqSIXklHwIDAQAB
}
}
}
public function sync_authorization($authorized_dc) {
public function sync_authorization($authorized_dc)
{
foreach ($this->datacenter->sockets as $new_dc => &$socket) {
if ($new_dc === $authorized_dc) continue;
if ($new_dc === $authorized_dc) {
continue;
}
\danog\MadelineProto\Logger::log(['Copying authorization from dc '.$authorized_dc.' to dc '.$new_dc.'...'], Logger::VERBOSE);
$this->should_serialize = true;
$exported_authorization = $this->method_call('auth.exportAuthorization', ['dc_id' => $new_dc], ['datacenter' => $authorized_dc]);

View File

@ -24,6 +24,7 @@ trait AckHandler
\danog\MadelineProto\Logger::log(["WARNING: Couldn't find message id ".$message_id.' in the array of outgoing messages. Maybe try to increase its size?'], \danog\MadelineProto\Logger::WARNING);
var_dump($message_id);
var_dump(debug_backtrace()[0]['file'], debug_backtrace()[0]['line']);
return false;
}

View File

@ -68,10 +68,9 @@ trait AuthKeyHandler
if (!isset($this->key->keydata['fp'])) {
$this->key = new \danog\MadelineProto\RSA($this->settings['authorization']['rsa_key']);
}
if (in_array($this->key->keydata['fp'], $ResPQ['server_public_key_fingerprints'])) throw new \danog\MadelineProto\SecurityException("Couldn't find our key in the server_public_key_fingerprints vector.");
if (in_array($this->key->keydata['fp'], $ResPQ['server_public_key_fingerprints'])) {
throw new \danog\MadelineProto\SecurityException("Couldn't find our key in the server_public_key_fingerprints vector.");
}
$pq_bytes = $ResPQ['pq'];
$server_nonce = $ResPQ['server_nonce'];
@ -512,7 +511,6 @@ trait AuthKeyHandler
private $temp_requested_calls = [];
private $calls = [];
public function accept_call($params)
{
$dh_config = $this->get_dh_config();
@ -530,6 +528,7 @@ trait AuthKeyHandler
$this->check_G($g_b, $dh_config['p']);
$this->handle_pending_updates();
}
public function request_call($user)
{
$user = $this->get_info($user)['InputUser'];
@ -549,8 +548,6 @@ trait AuthKeyHandler
return $res['phone_call']['id'];
}
public function complete_call($params)
{
if ($this->call_status($params['id']) !== 1) {
@ -573,7 +570,6 @@ trait AuthKeyHandler
$this->handle_pending_updates();
}
public function call_status($id)
{
if (isset($this->calls[$id])) {

View File

@ -25,7 +25,9 @@ trait CallHandler
if (!is_array($aargs)) {
throw new \danog\MadelineProto\Exception("Additonal arguments aren't an array.");
}
if (!isset($aargs['datacenter'])) throw new \danog\MadelineProto\Exception("No datacenter provided");
if (!isset($aargs['datacenter'])) {
throw new \danog\MadelineProto\Exception('No datacenter provided');
}
$args = $this->botAPI_to_MTProto($args);
$serialized = $this->serialize_method($method, $args);
$content_related = $this->content_related($method);
@ -66,7 +68,6 @@ trait CallHandler
$this->recv_message($aargs['datacenter']); // This method receives data from the socket, and parses stuff
$only_updates = $this->handle_messages($aargs['datacenter']); // This method receives data from the socket, and parses stuff
//}
} catch (\danog\MadelineProto\Exception $e) {
if ($e->getMessage() === 'I had to recreate the temporary authorization key') {
continue 2;
@ -149,13 +150,16 @@ var_dump($this->datacenter->sockets[$aargs['datacenter']]->outgoing_messages);
if (!is_array($args)) {
throw new \danog\MadelineProto\Exception("Arguments aren't an array.");
}
if (!isset($aargs['datacenter'])) throw new \danog\MadelineProto\Exception("No datacenter provided");
if (!isset($aargs['datacenter'])) {
throw new \danog\MadelineProto\Exception('No datacenter provided');
}
for ($count = 1; $count <= $this->settings['max_tries']['query']; $count++) {
try {
\danog\MadelineProto\Logger::log([$object === 'msgs_ack' ? 'ack '.$args['msg_ids'][0] : 'Sending object (try number '.$count.' for '.$object.')...'], \danog\MadelineProto\Logger::ULTRA_VERBOSE);
$int_message_id = $this->send_message($this->serialize_object(['type' => $object], $args), $this->content_related($object), $aargs);
if ($object !== 'msgs_ack') $this->datacenter->sockets[$aargs['datacenter']]->outgoing_messages[$int_message_id]['content'] = ['method' => $object, 'args' => $args];
if ($object !== 'msgs_ack') {
$this->datacenter->sockets[$aargs['datacenter']]->outgoing_messages[$int_message_id]['content'] = ['method' => $object, 'args' => $args];
}
} catch (Exception $e) {
\danog\MadelineProto\Logger::log(['An error occurred while calling object '.$object.': '.$e->getMessage().' in '.$e->getFile().':'.$e->getLine().'. Recreating connection and retrying to call object...'], \danog\MadelineProto\Logger::WARNING);
$this->datacenter->sockets[$aargs['datacenter']]->close_and_reopen();

View File

@ -86,10 +86,17 @@ trait MsgIdHandler
return strrev($message_id->toBytes());
}
public function get_max_id($datacenter, $incoming) {
public function get_max_id($datacenter, $incoming)
{
$keys = array_keys($this->datacenter->sockets[$datacenter]->{$incoming ? 'incoming_messages' : 'outgoing_messages'});
if (empty($keys)) return $this->zero;
array_walk($keys, function (&$value, $key) { $value = is_integer($value) ? new \phpseclib\Math\BigInteger($value) : new \phpseclib\Math\BigInteger(strrev($value), 256); });
if (empty($keys)) {
return $this->zero;
}
array_walk($keys, function (&$value, $key) {
$value = is_int($value) ? new \phpseclib\Math\BigInteger($value) : new \phpseclib\Math\BigInteger(strrev($value), 256);
});
return \phpseclib\Math\BigInteger::max(...$keys);
}
}

View File

@ -49,7 +49,6 @@ trait ResponseHandler
foreach ($msg_ids as $msg_id) {
$cur_info = 0;
if (!in_array($msg_id, $this->datacenter->sockets[$datacenter]->incoming_messages)) {
$msg_id = new \phpseclib\Math\BigInteger(strrev($msg_id), 256);
if ((new \phpseclib\Math\BigInteger(time() + $this->datacenter->sockets[$datacenter]->time_delta + 30))->bitwise_leftShift(32)->compare($msg_id) < 0) {
$cur_info |= 3;
@ -287,15 +286,16 @@ trait ResponseHandler
unset($this->datacenter->sockets[$datacenter]->incoming_messages[$current_msg_id]);
}
}
return $only_updates;
}
public function handle_rpc_error($server_answer, &$datacenter) {
public function handle_rpc_error($server_answer, &$datacenter)
{
switch ($server_answer['error_code']) {
case 303:
$this->datacenter->curdc = $datacenter = preg_replace('/[^0-9]+/', '', $server_answer['error_message']);
throw new \danog\MadelineProto\Exception('Received request to switch to DC '.$this->datacenter->curdc);
case 401:
switch ($server_answer['error_message']) {
case 'USER_DEACTIVATED':
@ -326,6 +326,7 @@ trait ResponseHandler
break;
}
}
public function handle_pending_updates()
{
\danog\MadelineProto\Logger::log(['Parsing pending updates...'], \danog\MadelineProto\Logger::VERBOSE);

View File

@ -22,6 +22,7 @@ trait SeqNoHandler
$in = $content_related ? 1 : 0;
$value = $this->datacenter->sockets[$datacenter]->session_out_seq_no;
$this->datacenter->sockets[$datacenter]->session_out_seq_no += $in;
return ($value * 2) + $in;
}
@ -30,6 +31,7 @@ trait SeqNoHandler
$in = $content_related ? 1 : 0;
$value = $this->datacenter->sockets[$datacenter]->session_in_seq_no;
$this->datacenter->sockets[$datacenter]->session_in_seq_no += $in;
return ($value * 2) + $in;
}

View File

@ -39,6 +39,7 @@ trait AuthKeyHandler
$this->notify_layer($params['id']);
$this->handle_pending_updates();
}
public function request_secret_chat($user)
{
$user = $this->get_info($user)['InputUser'];
@ -82,7 +83,6 @@ trait AuthKeyHandler
$this->handle_pending_updates();
}
public function notify_layer($chat)
{
$this->method_call('messages.sendEncryptedService', ['peer' => $chat, 'message' => ['_' => 'decryptedMessageService', 'action' => ['_' => 'decryptedMessageActionNotifyLayer', 'layer' => $this->encrypted_layer]]], ['datacenter' => $this->datacenter->curdc]);

View File

@ -244,6 +244,7 @@ trait TL
if (strlen($object) !== 8) {
throw new Exception('Given value is not 8 bytes long');
}
return $object;
}
@ -256,16 +257,19 @@ trait TL
if (strlen($object) !== 16) {
throw new Exception('Given value is not 16 bytes long');
}
return (string) $object;
case 'int256':
if (strlen($object) !== 32) {
throw new Exception('Given value is not 32 bytes long');
}
return (string) $object;
case 'int512':
if (strlen($object) !== 64) {
throw new Exception('Given value is not 64 bytes long');
}
return (string) $object;
case 'double':
return \danog\PHP\Struct::pack('<d', $object);
@ -561,5 +565,4 @@ trait TL
return $x;
}
}

View File

@ -28,22 +28,26 @@ class SocketHandler extends \Threaded implements \Collectable
*/
public function run()
{
require_once(__DIR__.'/../SecurityException.php');
require_once(__DIR__.'/../RPCErrorException.php');
require_once(__DIR__.'/../ResponseException.php');
require_once(__DIR__.'/../TL/Conversion/Exception.php');
require_once(__DIR__.'/../TL/Exception.php');
require_once(__DIR__.'/../NothingInTheSocketException.php');
require_once(__DIR__.'/../Exception.php');
require_once __DIR__.'/../SecurityException.php';
require_once __DIR__.'/../RPCErrorException.php';
require_once __DIR__.'/../ResponseException.php';
require_once __DIR__.'/../TL/Conversion/Exception.php';
require_once __DIR__.'/../TL/Exception.php';
require_once __DIR__.'/../NothingInTheSocketException.php';
require_once __DIR__.'/../Exception.php';
$this->API->handle_messages($current);
$this->setGarbage();
}
private $garbage = false;
public function setGarbage():void {
public function setGarbage():void
{
$this->garbage = true;
}
public function isGarbage():bool {
public function isGarbage():bool
{
return $this->garbage;
}
}

View File

@ -21,29 +21,37 @@ class SocketReader extends \Threaded implements \Collectable
{
$this->API = $me;
$this->current = $current;
}
public function __sleep() {
public function __sleep()
{
return ['current', 'API', 'garbage'];
}
public function __destruct() {
public function __destruct()
{
\danog\MadelineProto\Logger::log(['Shutting down handler pool for DC '.$this->current], \danog\MadelineProto\Logger::NOTICE);
if (isset($this->handler_pool)) $this->handler_pool->shutdown();
if (isset($this->handler_pool)) {
$this->handler_pool->shutdown();
}
}
/**
* Reading connection and receiving message from server. Check the CRC32.
*/
public function run()
{
require_once(__DIR__.'/../SecurityException.php');
require_once(__DIR__.'/../RPCErrorException.php');
require_once(__DIR__.'/../ResponseException.php');
require_once(__DIR__.'/../TL/Conversion/Exception.php');
require_once(__DIR__.'/../TL/Exception.php');
require_once(__DIR__.'/../NothingInTheSocketException.php');
require_once(__DIR__.'/../Exception.php');
require_once __DIR__.'/../SecurityException.php';
require_once __DIR__.'/../RPCErrorException.php';
require_once __DIR__.'/../ResponseException.php';
require_once __DIR__.'/../TL/Conversion/Exception.php';
require_once __DIR__.'/../TL/Exception.php';
require_once __DIR__.'/../NothingInTheSocketException.php';
require_once __DIR__.'/../Exception.php';
var_dump($this->API->settings['threading']);
if (!isset($this->handler_pool)) $this->handler_pool = new \Pool(2);
if (!isset($this->handler_pool)) {
$this->handler_pool = new \Pool(2);
}
var_dump($this->API->settings['threading']);
@ -51,16 +59,21 @@ var_dump($this->API->settings['threading']);
try {
$this->API->recv_message($this->current);
$this->handler_pool->submit(new SocketHandler($this->API, $this->current));
} catch (\danog\MadelineProto\NothingInTheSocketException $e) { ; }
} catch (\danog\MadelineProto\NothingInTheSocketException $e) {
}
}
$this->setGarbage();
}
public $garbage = false;
public function setGarbage():void {
public function setGarbage():void
{
$this->garbage = true;
}
public function isGarbage():bool {
public function isGarbage():bool
{
return $this->garbage;
}
}