add options in conf section FIND

optimize request keys from server add tmp clear function optimize mirror checks
Kingston-kms · Feb 20, 2021 · 30bb35f · 30bb35f
1 parent 58c8e3e
commit 30bb35f
Show file tree

Hide file tree

Showing 8 changed files with 81 additions and 14 deletions.
diff --git a/inc/classes/Mirror.class.php b/inc/classes/Mirror.class.php
@@ -93,6 +93,7 @@ static public function test_key()
 
         $test_mirrors = [];
         foreach (static::$ESET['mirror'] as $mirror) {
+
             Tools::download_file(
                 [
                     CURLOPT_USERPWD => static::$key[0] . ":" . static::$key[1],
@@ -118,7 +119,8 @@ static public function test_key()
             if ($version) {
                 $maxVersion = $version > $maxVersion ? $version : $maxVersion;
                 $sameMirrors[] = ['host' => $mirror, 'db_version' => $version];
-            }
+            } else
+                return false;
         }
 
         static::$mirrors = array_filter($sameMirrors, function ($v, $k) use ($maxVersion) {
@@ -416,6 +418,7 @@ static protected function single_download($download_files, $onlyCheck = false, $
                     static::$total_downloads += $header['size_download'];
                     break;
                 } else {
+                    if ($onlyCheck) @unlink(static::$tmp_update_file);
                     @unlink($out);
                 }
             }
@@ -431,7 +434,7 @@ static protected function download($download_files, $onlyCheck = false, $checked
     {
         Log::write_log(Language::t("Running %s", __METHOD__), 5, static::$version);
 
-        switch (function_exists('curl_multi_init') && Config::get('CONNECTION')['use_multidownload']) {
+        switch (function_exists('curl_multi_init') && (bool)(Config::get('CONNECTION')['use_multidownload']) && !$onlyCheck) {
             case true:
                 static::multiple_download($download_files, $onlyCheck, $checkedMirror);
                 break;
@@ -533,6 +536,7 @@ static public function set_key($key)
     static public function destruct()
     {
         Log::write_log(Language::t("Running %s", __METHOD__), 5, static::$version);
+
         static::$total_downloads = 0;
         static::$version = null;
         static::$source_update_file = null;
@@ -543,6 +547,7 @@ static public function destruct()
         static::$unAuthorized = false;
     }
 
+
     /**
      * @param $folder
      * @return int

diff --git a/inc/classes/Nod32ms.class.php b/inc/classes/Nod32ms.class.php
@@ -166,8 +166,9 @@ private function validate_key($key)
     {
         Log::write_log(Language::t("Running %s", __METHOD__), 5, Mirror::$version);
         $result = explode(":", $key);
-        Log::write_log(Language::t("Validating key [%s:%s]", $result[0], $result[1]), 4, Mirror::$version);
+
         if ($this->key_exists_in_file($result[0], $result[1], static::$key_invalid_file)) return false;
+        Log::write_log(Language::t("Validating key [%s:%s] for version %s", $result[0], $result[1], Mirror::$version), 4, Mirror::$version);
 
         Mirror::set_key(array($result[0], $result[1]));
         $ret = Mirror::test_key();
@@ -401,9 +402,26 @@ private function get_key_from_server()
 
         if ($FIND['use_server']) {
             try {
-                $key = file_get_contents($FIND['server_url']);
+                $ch = curl_init();
+                $opt = [
+
+                    CURLOPT_URL => $FIND['server_url'],
+                    CURLOPT_SSL_VERIFYPEER => false,
+                    CURLOPT_RETURNTRANSFER => 1
+                ];
+                if (strlen($FIND['user']) > 0 ) {
+                    $opt[CURLOPT_POST] = true;
+                    $opt[CURLOPT_POSTFIELDS] = "user=" . $FIND['user'];
+                }
+                curl_setopt_array($ch, $opt);
+                $key = curl_exec($ch);
+                if ($key == false) {
+                    Log::write_log(Language::t("Error %s", curl_error($ch)), 5, Mirror::$version);
+                    return false;
+                }
+
                 $key = json_decode($key, true);
-                if ($this->validate_key($key['username'] . ':' . $key['password'])) {
+                if ($key && $this->validate_key($key['username'] . ':' . $key['password'])) {
                     static::$foundValidKey = true;
                     return true;
                 }
@@ -423,7 +441,11 @@ private function find_keys()
         Log::write_log(Language::t("Running %s", __METHOD__), 5, Mirror::$version);
         $FIND = Config::get('FIND');
 
-        if ($this->get_key_from_server()) return true;
+        $attempts = 0;
+        while ($attempts < $FIND['number_attempts']) {
+            if ($this->get_key_from_server()) return true;
+            $attempts++;
+        }
 
         if ($FIND['auto'] != 1)
             return null;
@@ -595,15 +617,17 @@ private function run_script()
         $total_size = array();
         $total_downloads = array();
         $average_speed = array();
-        //$web_dir = Config::get('SCRIPT')['web_dir'];
+
 
         foreach ($DIRECTORIES as $version => $dir) {
             if (Config::upd_version_is_set($version) == '1') {
 
                 Log::write_log(Language::t("Init Mirror for version %s in %s", $version, $dir['name']), 5, $version);
                 Mirror::init($version, $dir);
+
                 static::$foundValidKey = false;
                 $this->read_keys();
+
                 if (static::$foundValidKey == false) {
                     $this->find_keys();
 
@@ -652,6 +676,11 @@ private function run_script()
             }
         }
 
+        foreach (glob(Tools::ds(TMP_PATH, '*')) as $folder) {
+            static::clear_tmp($folder);
+            @rmdir($folder);
+        }
+
         Log::write_log(Language::t("Total size for all databases: %s", Tools::bytesToSize1024(array_sum($total_size))), 3);
 
         if (array_sum($total_downloads) > 0)
@@ -663,6 +692,27 @@ private function run_script()
         if (Config::get('SCRIPT')['generate_html'] == '1') $this->generate_html();
     }
 
+    private function clear_tmp($path)
+    {
+        try {
+            $iterator = new DirectoryIterator($path);
+            foreach ( $iterator as $fileinfo) {
+                if($fileinfo->isDot())continue;
+                if($fileinfo->isDir()){
+                    if(static::clear_tmp($fileinfo->getPathname()))
+                        @rmdir($fileinfo->getPathname());
+                }
+                if($fileinfo->isFile()){
+                    @unlink($fileinfo->getPathname());
+                }
+            }
+        } catch ( Exception $e ){
+            // write log
+            return false;
+        }
+        return true;
+    }
+
     /**
      * @param $old_version
      * @param $new_version

diff --git a/inc/classes/Tools.class.php b/inc/classes/Tools.class.php
@@ -126,7 +126,7 @@ static public function ping(array $options, $hostname, $port = 80, $file = NULL)
     static public function bytesToSize1024($bytes, $precision = 2)
     {
         $unit = ['Bytes', 'KBytes', 'MBytes', 'GBytes', 'TBytes', 'PBytes', 'EBytes'];
-        return @round($bytes / pow(1024, ($i = floor(log($bytes, 1024)))), $precision) . ' ' . $unit[intval($i)];
+        return $bytes > 0 ? @round($bytes / pow(1024, ($i = floor(log($bytes, 1024)))), $precision) . ' ' . $unit[intval($i)] :  '0 ' . $unit[intval(0)];
     }
 
     /**

diff --git a/langpacks/en.lng b/langpacks/en.lng
@@ -76,7 +76,7 @@ Average speed for all databases: %s/s
 n/a
 Generating html...
 Testing key [%s:%s]
-Validating key [%s:%s]
+Validating key [%s:%s] for version %s
 The script has been stopped!
 Checking mirror %s with key [%s:%s]
 Extracting file %s to %s
@@ -90,4 +90,4 @@ Running %s
 Your version of database is relevant %s
 Mirror %s active
 Mirror %s inactive
-Try next mirror %s
+Try next mirror %s
diff --git a/langpacks/ru.lng b/langpacks/ru.lng
@@ -76,7 +76,7 @@ ESET NOD32 сервер обновления
 н/д
 Генерация html...
 Проверка ключа [%s:%s]
-Валидация ключа [%s:%s]
+Валидация ключа [%s:%s] для версии %s
 Скрипт будет остановлен!
 Проверка зеркала %s с ключом [%s:%s]
 Распаковка файла %s в %s

diff --git a/langpacks/ukr.lng b/langpacks/ukr.lng
@@ -76,7 +76,7 @@ ESET NOD32 сервер оновлення
 н/д
 Створення html...
 Перевірка ключа [%s:%s]
-Валідація ключа [%s:%s]
+Валідація ключа [%s:%s] для версії %s
 Скрипт буде зупинен!
 Перевірка дзеркала %s з ключом [%s:%s]
 Розпаковка файлу %s в %s
@@ -90,4 +90,4 @@ Unrar не є виконуваним за шляхом %s
 Версія вашої бази %s актуальна
 Дзеркало %s доступне
 Дзеркало %s недоступне
-Спроба завантаження з наступного дзеркала %s
+Спроба завантаження з наступного дзеркала %s
diff --git a/nod32ms.conf.eng b/nod32ms.conf.eng
@@ -160,7 +160,7 @@ auto = 1
 ; Telegram channel: https://t.me/nod32trialKeys
 
 ; This parameter is responsible for the ability to take keys from the key server
-;  Trial keys, official, it is possible to take a unique key with the ability to automatically update the key
+; Trial keys, official, it is possible to take a unique key with the ability to automatically update the key
 ; If the option is enabled, then the key search does not occur.
 use_server = 1
 
@@ -169,6 +169,12 @@ use_server = 1
 ;
 server_url = "https://api.nod32-trial-keys.site/api/single-license"
 
+; User email which registered on site nod32-trial-keys.site to request licenses
+user = ''
+
+; Number of attempts to request new key if previous invalid
+number_attempts = 5
+
 ; ###########################
 ; end
 

diff --git a/nod32ms.conf.rus b/nod32ms.conf.rus
@@ -169,6 +169,12 @@ use_server = 1
 ;
 server_url = "https://api.nod32-trial-keys.site/api/single-license"
 
+;  Email юзера, отправившего запрос лицензий на сайте nod32-trial-keys.site
+user = ''
+
+; Количество попыток запросить новый ключ, если предыдущий невалидный
+number_attempts = 5
+
 ; ###########################
 ; конец блока