Unha pequena porta traseira en Flask ou como controlar un ordenador nunha rede local

Ola Habr!

Recentemente vin unha versión descargada do fluxo de programación "Como crear a túa propia aplicación web en Flask". E decidín consolidar os meus coñecementos nalgún proxecto. Durante moito tempo non sabía que escribir e xurdiume a idea: "Por que non facer unha mini-porta traseira en Flask?"

As primeiras opcións para implementacións e capacidades da porta traseira apareceron inmediatamente na miña cabeza. Pero decidín facer inmediatamente unha lista de capacidades de porta traseira:

1. Saber como abrir sitios web
2. Ter acceso á liña de comandos
3. Poder abrir programas, fotos, vídeos

Polo tanto, o primeiro punto é moi sinxelo de implementar usando o módulo navegador web. Decidín implementar o segundo punto usando o módulo os. E o terceiro tamén é a través do módulo os, pero vou usar "ligazóns" (máis sobre iso máis adiante).

Escribindo un servidor

Entón, *drumroll* todo o código do servidor:

from flask import Flask, request
import webbrowser
import os
import re

app = Flask(__name__)
@app.route('/mycomp', methods=['POST'])
def hell():
    json_string = request.json
    if json_string['command'] == 'test':
        return 'The server is running and waiting for commands...'
    if json_string['command'] == 'openweb':
        webbrowser.open(url='https://www.'+json_string['data'], new=0)
        return 'Site opening ' + json_string['data'] + '...'
    if json_string['command'] == 'shell':
        os.system(json_string['data'])
        return 'Command execution ' + json_string['data'] + '...'
    if json_string['command'] == 'link':
        links = open('links.txt', 'r')
        for i in range(int(json_string['data'])):
            link = links.readline()
        os.system(link.split('>')[0])
        return 'Launch ' + link.split('>')[1]
if __name__ == '__main__':
    app.run(host='0.0.0.0')

Xa botei todo o código, é hora de explicar a esencia.

Todo o código execútase no ordenador local no porto 5000. Para interactuar co servidor, debemos enviar unha solicitude JSON POST.

Estrutura da solicitude JSON:

{‘command’:  ‘comecommand’, ‘data’: ‘somedata’}

Ben, ten sentido que "comando" sexa o comando que queremos executar. E "datos" son os argumentos do comando.

Podes escribir e enviar solicitudes JSON para interactuar co servidor manualmente (as solicitudes axudarache). Ou pode escribir un cliente de consola.

Escribindo un cliente

Código:

import requests

logo = ['nn',
        '******      ********',
        '*******     *********',
        '**    **    **     **',
        '**    **    **     **      Written on Python',
        '*******     **     **',
        '********    **     **',
        '**     **   **     **      Author: ROBOTD4',
        '**     **   **     **',
        '**     **   **     **',
        '********    *********',
        '*******     ********',
        'nn']

p = ''
iport = '192.168.1.2:5000'
host = 'http://' + iport + '/mycomp'

def test():
    dict = {'command': 'test', 'data': 0}
    r = requests.post(host, json=dict)
    if r.status_code == 200:
        print (r.content.decode('utf-8'))

def start():
    for i in logo:
        print(i)

start()
test()

while True:
    command = input('>')
    if command == '':
        continue
    a = command.split()
    if command == 'test':
        dict = {'command': 'test', 'data': 0}
        r = requests.post(host, json=dict)
        if r.status_code == 200:
            print (r.content.decode('utf-8'))
    if a[0] == 'shell':
        for i in range(1, len(a)):
            p = p + a[i] + ' '
        dict = {'command': 'shell', 'data': p}
        r = requests.post(host, json=dict)
        if r.status_code == 200:
            print (r.content.decode('utf-8'))
        p = ''
    if a[0] == 'link':
        if len(a) > 1:
            dict = {'command': 'link', 'data': int(a[1])}
            r = requests.post(host, json=dict)
            if r.status_code == 200:
                print (r.content.decode('utf-8'))
        else:
            print('Комманда не содержит аргументов!')
    if a[0] == 'openweb':
            if len(a) > 1:
                dict = {'command': 'openweb', 'data': a[1]}
                r = requests.post(host, json=dict)
                if r.status_code == 200:
                    print (r.content.decode('utf-8'))
            else:
                print('Комманда не содержит аргументов!')
    if a[0] == 'set':
        if a[1] == 'host':
            ip = a[2] + ':5000'
    if command == 'quit':
        break

Explicacións:

En primeiro lugar, impórtase o módulo de solicitudes (para interactuar co servidor). A continuación móstranse descricións das funcións de inicio e proba. E despois o ciclo no que sucede a maxia. Liches o código? Así entendes o significado da maxia que ocorre no ciclo. Introduza o comando - execútase. Shell: comandos para a liña de comandos (a lóxica está fóra de escala).

Proba: comproba se o servidor está en execución (porta traseira)
Ligazón: uso dun "atallo"
Openweb: abrir un sitio web
Saír - saír do cliente
Establecer: configura a ip do teu ordenador na rede local

E agora máis sobre a ligazón.

Hai un ficheiro link.txt xunto ao servidor. Contén ligazóns (ruta completa) a ficheiros (vídeos, fotos, programas).

A estrutura é a seguinte:

полный_путь>описание
полный_путь>описание

Total

Temos un servidor de porta traseira para controlar un ordenador nunha rede local (dentro dunha rede wi-fi). Tecnicamente, podemos executar o cliente desde calquera dispositivo que teña un intérprete de Python.

PS Engadín o comando set para que se a un ordenador da rede local se lle asigna unha IP diferente, pódese cambiar directamente no cliente.

Fonte: www.habr.com