使用python做内网穿透小工具实现花生壳功能，一键生成内网穿透端口映射

本文是基于https://gitee.com/stlswm/transponder 这位大佬的源码的基础上，做的小工具
测试系统：
外网：linux
内网：windows
python：python3.6（需要安装flask、psutil）
功能介绍：使用flask，实现可一键添加端口映射，可以删除单端口，可删除全部端口映射

需要准备的东西：
1：如上的内网穿透工具：git clone https://gitee.com/stlswm/transponder.git
2：外网服务器 （需要安装 Nginx），开放端口 9000-9999
3：域名，需要泛域名，基本上是添加一个@.domain.com（没有的话可以使用ip+端口来访问）

外网服务器：
把 outer_server 中的二进制linux可执行文件main移动到随意文件夹，文件地址替换python中mainPath变量，

python文件

from flask import Flask
from flask import request
import os
import time
import random
import sys
import string
import psutil
import re
app = Flask(__name__)

# 文件存放地址
mainPath="/root/go/src/transponder/outer_server/"
# ng配置文件存放地址
ngConfigPath="/www/server/panel/vhost/nginx/"
# 你的域名
domain="domain.com"


#添加端口映射
@app.route('/add')
def add():
    # 设置端口信息
    if os.path.exists("./port.txt") == False:
        pf=open("./port.txt",'w')
        pf.write("9000")
        pf.close
        port=9000
    else:
        portFile = open("./port.txt",'r')
        port=portFile.read()
        portFile.close()
        
    serverPort=int(port)+1
    
    # 端口信息存会文件中
    portFile = open("./port.txt",'w')
    portFile.write(str(serverPort))
    portFile.close()
    
    fileName="main" + str(serverPort)
    
    # 操作服务端 打包添加配置文件
    os.mkdir(mainPath+fileName+"s")
    os.system('cd '+mainPath+';cp -r main ./'+fileName+"s;cd "+fileName+"s;mv main "+fileName)
    fp = open(mainPath+fileName+"s/outer.config.json",'w')
    psd=random.randint(10000000000,90000000000)
    fp.write('{"InnerServerAddress": "tcp://0.0.0.0:'+str(serverPort)+'","OuterServerAddress": "unix:///var/run/'+fileName+'.sock","AuthKey": "'+str(psd)+'"}')
    fp.close()
    #操作ng  添加ng配置文件
    f=open(ngConfigPath+fileName+".conf","w")
    head=generate_random_str(6)
    f.write("server {listen 80;server_name  "+head+"."+domain+";access_log  /var/log/www.abc.com.access.log;error_log  /var/log/www.abc.com.error.log;location / {proxy_pass http://unix:/var/run/"+fileName+".sock:/;proxy_redirect     off;proxy_set_header   Host             $host;proxy_set_header   X-Real-IP        $remote_addr;proxy_set_header   X-Forwarded-For  $proxy_add_x_forwarded_for;proxy_next_upstream error timeout invalid_header http_500 http_502 http_503 http_504;proxy_max_temp_file_size 0;proxy_connect_timeout      90;proxy_send_timeout         90;proxy_read_timeout         90;proxy_buffer_size          4k;proxy_buffers              4 32k;proxy_busy_buffers_size    64k;proxy_temp_file_write_size 64k;}}")
    f.close()
    os.system("nginx -s reload")
    os.system("cd "+mainPath+fileName+"s;nohup ./"+fileName+" &")
    
    f=open("./portAll.txt","a")
    f.write(","+fileName)
    f.close()
    
    return "
启动成功
服务器：119.45.136.232:"+str(serverPort)+"
客户端域名：http://"+head+"."+domain+"
秘钥："+str(psd)

#删除对应的外网端口
@app.route('/kill')
def kill():
    port = request.args.get("port")
    fileName="main"+str(port)
    pid=processinfo(fileName)
    if pid==False:
        return "没有找改进程"
    os.system("kill -9 "+str(pid))
    os.system("cd "+mainPath+";rm -rf "+fileName+"s")
    os.system("cd /run;rm  "+fileName+".sock")
    os.system("cd "+ngConfigPath+";rm "+fileName+".conf;nginx -s reload")
    return "操作成功！"
    
#删除全部端口映射
@app.route('/killAll')
def killAll():
    if os.path.exists("./portAll.txt") == False:
        return "无端口映射"
    portFile = open("./portAll.txt",'r')
    portAllStr=portFile.read()
    portAllArr=portAllStr.split(",")
    myStr="
操作成功
"
    for v in portAllArr:
        if len(v) > 0:
            fileName=v
            pid=processinfo(fileName)
            if pid==False:
                myStr=myStr+"未找到进程："+fileName+"
"
            else :
                myStr=myStr+"成功停止进程："+fileName+"
"
                os.system("kill -9 "+str(pid))
            os.system("cd "+mainPath+";rm -rf "+fileName+"s")
            os.system("cd /run;rm "+fileName+".sock")
            os.system("cd "+ngConfigPath+";rm "+fileName+".conf;nginx -s reload")
    f=open("./portAll.txt","w")
    f.write("")
    f.close()
    f=open("./port.txt","w")
    f.write("9000")
    f.close()
    return myStr
    

# 获取进程ID
def processinfo(processName):
    pids = psutil.pids()
    for pid in pids:
        # print(pid)
        p = psutil.Process(pid)
        # print(p.name)
        if p.name() == processName:
            # print(pid)
            return pid  # 如果找到该进程则打印它的PID，返回true
    return False  # 没有找到该进程，返回false



# 随机生成字符串
def generate_random_str(randomlength):    
    '''    
    string.digits = 0123456789    
    string.ascii_letters = 26个小写,26个大写    
    '''    
    str_list = random.sample(string.digits + string.ascii_letters,randomlength)    				
    random_str = ''.join(str_list)    
    return random_str

if __name__ == '__main__':
    app.run(host="0.0.0.0",port="8001")

内网：

对应git文件中的inner_server
请求：http://外网ip:8001/add
1：获取到服务器端口和客户端请求域名和秘钥，
2：对应的替换到config里的内容，配置好ProxyAddress的内网端口，
4：windows双击打开inner_server中的main.exe linux: ./main
3：使用域名就能访问到内网