我怎样才能 use/authenticate msfrpc 与 python3.x？

Question

已编辑：下面的代码有效，并对更改进行了注释。如前所述，使用 python3 必须在字符串文字前加上 "b" 以生成字节类型的实例而不是 Unicode str 类型。

我正在尝试在 Python 3 中使用 msfrpc（在 Python 2 中编写），但遇到身份验证错误。我正在使用的代码如下；查看代码中的注释以了解我所做的更改。

该程序在 python2 中成功运行（当使用 httplib 而不是 http.client 时），与使用 python3.[=15 时看到的身份验证交换似乎相同=]

import msgpack
import http.client #Changed from httplib

class Msfrpc:
  class MsfError(Exception):
    def __init__(self,msg):
      self.msg = msg
    def __str__(self):
      return repr(self.msg)

  class MsfAuthError(MsfError):
    def __init__(self,msg):
      self.msg = msg

  def __init__(self,opts=[]):
    self.host = opts.get('host') or "127.0.0.1"
    self.port = opts.get('port') or 55552
    self.uri = opts.get('uri') or "/api/"
    self.ssl = opts.get('ssl') or False
    self.authenticated = False
    self.token = False
    self.headers = {"Content-type" : "binary/message-pack" }
    if self.ssl:
      self.client = http.client.HTTPSConnection(self.host,self.port)  #Changed httplib -> http.client
    else:
      self.client = http.client.HTTPConnection(self.host,self.port)        #Changed httplib -> http.client

  def encode(self,data):
    return msgpack.packb(data)
  def decode(self,data):
    return msgpack.unpackb(data)

   def call(self,meth,opts = []):
    if meth != "auth.login":
      if not self.authenticated:
        raise self.MsfAuthError("MsfRPC: Not Authenticated")

    if meth != "auth.login":
      opts.insert(0,self.token)

    opts.insert(0,meth)
    params = self.encode(opts)
    self.client.request("POST",self.uri,params,self.headers)
    resp = self.client.getresponse()
    return self.decode(resp.read()) 

  def login(self,user,password):
    ret = self.call('auth.login',[user,password])
    if ret.get(b'result') == b'success': #Added b
      self.authenticated = True
      self.token = ret.get(b'token') #Added b
      return True
    else:
      raise self.MsfAuthError("MsfRPC: Authentication failed")

if __name__ == '__main__':

  # Create a new instance of the Msfrpc client with the default options
  client = Msfrpc({})

  # Login to the msfmsg server using the password "abc123"
  client.login('msf','abc123')

  # Get a list of the exploits from the server
  mod = client.call('module.exploits')

  # Grab the first item from the modules value of the returned dict
  print ("Compatible payloads for : %s\n" % mod[b'modules'][0]) #Added () #Added b

  # Get the list of compatible payloads for the first option
  ret = client.call('module.compatible_payloads',[mod[b'modules'][0]]) #Added b
  for i in (ret.get(b'payloads')): #Added b
    print ("\t%s" % i)  #Added ()

当运行程序时，结果为：

root@kali:~/Dropbox/PythonStuff/python-nmap-test# python3 test3.py
Traceback (most recent call last):
  File "test3.py", line 20, in <module>
    client.login('msf','abc123')
  File "/usr/local/lib/python3.7/dist-packages/msfrpc.py", line 64, in login  
    raise self.MsfAuthError("MsfRPC: Authentication failed")
msfrpc.MsfAuthError: 'MsfRPC: Authentication failed'

奇怪的是，运行 tcpdump 在程序执行时显示身份验证成功并授予令牌：

当程序以 python2 成功执行时，身份验证交换看起来相同，但如果有人觉得发布程序的数据包捕获（运行 in python2）成功完成会有所帮助回答问题我可以轻松添加。

Answer 1

对此没有太大兴趣，但如果其他人有问题，我会发布答案；解决方案非常简单，我注意到上面发布的代码中的更改。

问题是 msfrpc 最初是在 Python 2 中编写的，因此来自 Metasploit 主机的 msgpack RPC 响应中 'b' 的前缀被忽略，但在 [=24] 中是必需的=] 3表示字面量应该变成字节字面量而不是字符串类型。

有关详细信息，请查看下面的 link；一旦我阅读它，答案就很明显了： https://timothybramlett.com/Strings_Bytes_and_Unicode_in_Python_2_and_3.html

上面的代码工作正常，但真正更好的解决方案是使用 Dan McInerney 编写的 msfrpc 模块，它使用 'requests' 包。

https://github.com/DanMcInerney/msfrpc

我怎样才能 use/authenticate msfrpc 与 python3.x？

How can I use/authenticate msfrpc with python3.x?

python-3.x

metasploit

python-2to3

http.client