dns mitm, update requirements

Author: rev1si0n

README.md

@@ -311,8 +311,8 @@ profile.port = 代理服务器端口
 profile.password = "代理服务器登录密码"
 profile.login = "代理服务器登录用户"
 
-# 注意目前无法通过TCP代理转发UDP流量，所以APP发出的任何UDP流量都是不经过代理的！
-# 因为这些原因，我们提供了一个选项让你可以选择是否只允许 TCP 流量
+# socks5 模式支持 udp 代理，但是 http 代理并不支持
+# 这会导致 udp 流量逃逸泄露你的真实IP地址，所以提供这个可选项
 # 当 drop_udp 为 True 时，应用的 UDP 流量将会被屏蔽，默认为 False
 profile.drop_udp = False
 

lamda/__init__.py

@@ -3,4 +3,4 @@
 # Distributed under MIT license.
 # See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
 __version__ = "3.0"
-__build__ = 0
+__build__ = 10

setup.py

@@ -7,9 +7,10 @@
 setuptools.setup(
     name            = "lamda",
     version         = "{}.{}".format(__version__, __build__),
-    url             = "https://github.com/rev1si0n",
+    description     = "Android reverse engineering & automation framework",
+    url             = "https://github.com/rev1si0n/lamda",
     author          = "rev1si0n",
-    python_requires = ">=3.6,<4.0",
+    python_requires = ">=3.6,<=3.10",
     zip_safe        = False,
     extras_require  = {
         "frida": ["frida>=15.0.0,<16.0.0,!=15.1.15,!=15.1.16,!=15.1.17"],
@@ -18,11 +19,20 @@
         ],
     },
     install_requires= [
-        "grpcio-tools>=1.35.0,<=1.40.0",
-        "grpc-interceptor>=0.13.0,<0.14.0",
-        "grpcio>=1.35.0,<=1.40.0",
+        "grpcio-tools>=1.35.0,<1.48.0",
+        "grpc-interceptor>=0.13.0,<0.14.2",
+        "grpcio>=1.35.0,<1.48.0",
         "asn1crypto>=1.0.0",
     ],
+    classifiers = [
+        "Environment :: Console",
+        "Intended Audience :: Developers",
+        "Intended Audience :: Information Technology",
+        "Intended Audience :: Science/Research",
+        "Programming Language :: Python :: 3",
+        "Operating System :: Android",
+        "Topic :: Security",
+    ],
     package_data    = {
         "lamda": ["*.py", "*.proto"],
         "lamda.google.protobuf.compiler": ["*.proto"],

tools/README.md

@@ -47,7 +47,7 @@ test.pem    # 用于加密客户端与服务端通信的证书
 
 ## startmitm.py
 
-启动中间人，这将会完全自动的在设备上开启全局的中间人代理，你就可以截获大部分APP的 http/s 流量。
+启动中间人，这将会完全自动的在设备上开启全局的中间人代理，你就可以截获大部分APP的 http/s 流量，当然，也包括 DNS 请求。
 
 首先确保当前电脑与设备在同一个网段，192.168.1.2 为运行了 lamda 的手机设备。
 其次，确保你已在命令行验证 mitmproxy 已安装成功（在命令行输入 `mitmdump` 进行验证）。
@@ -82,6 +82,31 @@ python3 -u startmitm.py localhost
 
 按下一次 `CONTROL` + `C` 退出脚本。
 
+### DNS 中间人
+
+截获 DNS 请求需要确保 mitmproxy 的版本 >= 8.1.0，且需要以**管理员**或者**root**身份运行脚本。
+```bash
+python3 -u startmitm.py 192.168.1.2 --set dns_server=true
+```
+即可。
+
+这些DNS请求默认会从本机发出，你也可以将这些 DNS 请求转发到指定的上游DNS服务器例如 `1.1.1.1`。
+```bash
+python3 -u startmitm.py 192.168.1.2 --set dns_server=true --set dns_mode=reverse:1.1.1.1
+```
+
+hook 脚本的方法名称定义有一些变化，正常 http 请求为 `response()`，截获 DNS 时需要使用 `dns_response()`。
+
+```python
+def response(flow):
+    print (flow, type(flow))
+
+def dns_response(flow):
+    print (flow, type(flow))
+```
+
+具体请查看 mitmproxy 的文档。
+
 ## ssh.sh
 
 连接入手机上的 shell 终端。

tools/startmitm.py

@@ -8,6 +8,8 @@
 
 from socket import *
 from random import randint
+from packaging.version import parse as ver
+from mitmproxy.version import VERSION
 from lamda.client import *
 
 
@@ -35,6 +37,9 @@ def get_default_interface_ip():
 host = sys.argv[1]
 if ":" in host:
     host, pkgName = host.split(":")
+if "dns_server=true" in sys.argv and ver(VERSION)<ver("8.1.0"):
+    print (time.ctime(), "dns_server needs mitmproxy>=8.1.0")
+    sys.exit(1)
 certfile = os.environ.get("CERTIFICATE", None)
 d = Device(host, certificate=certfile)
 
@@ -77,7 +82,8 @@ def get_default_interface_ip():
 # 初始化 proxy 配置
 profile = GproxyProfile()
 profile.type = GproxyType.HTTP_CONNECT
-profile.nameserver = "114.114.114.114"
+if "dns_server=true" in sys.argv:
+    profile.nameserver = host
 profile.drop_udp = True
 # http 代理不支持 udp
 #profile.udp_proxy = False
@@ -90,6 +96,11 @@ def get_default_interface_ip():
 
 servercmd = []
 servercmd.append("mitmweb")
+# 默认监听的是 127.0.0.1，改为全局
+servercmd.append("--set")
+servercmd.append("dns_listen_host=0.0.0.0")
+servercmd.append("--set")
+servercmd.append("dns_listen_port=53")
 servercmd.append("--ssl-insecure")
 # 随机 web-port
 servercmd.append("--web-port")