Wykonuj polecenia powłoki w Pythonie

Obecnie studiuję testy penetracyjne i programowanie w języku Python. Chcę tylko wiedzieć, jak bym wykonał polecenie Linuksa w Pythonie. Polecenia, które chcę wykonać, to:

echo 1 > /proc/sys/net/ipv4/ip_forward
iptables -t nat -A PREROUTING -p tcp --destination-port 80 -j REDIRECT --to-port 8080

Jeśli po prostu printużyję w Pythonie i uruchomię go w terminalu, czy będzie on działał tak samo jak wykonywanie go tak, jakbyś pisał go sam i naciskał Enter?

Możesz użyć os.system()takiego:

import os
os.system('ls')

Lub w twoim przypadku:

os.system('echo 1 > /proc/sys/net/ipv4/ip_forward')
os.system('iptables -t nat -A PREROUTING -p tcp --destination-port 80 -j REDIRECT --to-port 8080')

Co więcej, możesz użyć połączenia podprocesu, jest to bezpieczniejsze, potężniejsze i prawdopodobnie szybsze:

from subprocess import call
call('echo "I like potatos"', shell=True)

Lub bez wywoływania powłoki:

call(['echo', 'I like potatos'])

Jeśli chcesz przechwycić dane wyjściowe, możesz to zrobić w następujący sposób:

import subprocess
cmd = ['echo', 'I like potatos']
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

o, e = proc.communicate()

print('Output: ' + o.decode('ascii'))
print('Error: '  + e.decode('ascii'))
print('code: ' + str(proc.returncode))

Ja bardzo polecam ustawienie timeoutin communicate, a także przechwytywać wyjątków można uzyskać podczas wywoływania go. Jest to bardzo podatny na błędy kod, więc należy spodziewać się, że wystąpią błędy i odpowiednio je obsłużyć.

https://docs.python.org/3/library/subprocess.html

Pierwsze polecenie po prostu zapisuje do pliku. Nie wykonałbyś tego jako polecenia powłoki, ponieważ pythonmoże czytać i zapisywać pliki bez pomocy powłoki:

with open('/proc/sys/net/ipv4/ip_forward', 'w') as f:
    f.write("1")

iptablesKomenda jest coś może chcesz wykonać na zewnątrz. Najlepszym sposobem na to jest użycie modułu podprocesu .

import subprocess
subprocess.check_call(['iptables', '-t', 'nat', '-A',
                       'PREROUTING', '-p', 'tcp', 
                       '--destination-port', '80',
                       '-j', 'REDIRECT', '--to-port', '8080'])

Zauważ, że ta metoda również nie używa powłoki, co jest niepotrzebnym narzutem.

Najszybszy sposób:

import os
os.system("your command here")

To nie jest najbardziej elastyczne podejście; jeśli potrzebujesz więcej kontroli nad procesem niż „uruchom go raz, do końca i blokuj, aż zakończy się”, powinieneś subprocesszamiast tego użyć modułu.

Zasadniczo lepiej używać powiązań pytona, gdy tylko jest to możliwe (między innymi lepsze przechwytywanie wyjątków).

W przypadku echopolecenia lepiej jest użyć Pythona do zapisu w pliku, jak sugeruje odpowiedź @ jordanm.

Do iptablespolecenia, może python-iptables( strona PyPI , GitHub strona z opisem i doc ) stanowiłaby co trzeba (nie sprawdzić konkretną komendę).

To sprawi, że będziesz zależał od zewnętrznej biblioteki lib, więc musisz rozważyć korzyści. Korzystanie z podprocesu działa, ale jeśli chcesz użyć danych wyjściowych, musisz je przeanalizować samodzielnie i poradzić sobie ze zmianami danych wyjściowych w przyszłych iptableswersjach.

Pythonowa wersja twojej powłoki. Uważaj, nie testowałem tego.

from subprocess import run

def bash(command):
        run(command.split())

>>> bash('find / -name null')
/dev/null
/sys/fs/selinux/null
/sys/devices/virtual/mem/null
/sys/class/mem/null
/usr/lib/kbd/consoletrans/null

Jeśli chcesz wykonać to polecenie w innym systemie po SSH, być może będziesz musiał użyć modułu o nazwie Paramiko. To jest naprawdę przydatne.

Jeśli wykonanie polecenia musi być wykonane z lokalnego komputera, pomocne będą funkcje modułu os.

Strona domowa: https://www.paramiko.org/

Rozwój: https://github.com/paramiko/paramiko