Last active
December 31, 2019 19:37
-
-
Save EnisBerk/d805b7ff4cc09923e001142e1ffb2c22 to your computer and use it in GitHub Desktop.
open ssh connection to colab notebook
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import random, string, urllib.request, json, getpass | |
#Generate root password | |
password = ''.join(random.choice(string.ascii_letters + string.digits) for i in range(20)) | |
#Download ngrok | |
! wget -q -c -nc https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip | |
! unzip -qq -n ngrok-stable-linux-amd64.zip | |
#Setup sshd | |
! apt-get install -qq -o=Dpkg::Use-Pty=0 openssh-server pwgen > /dev/null | |
#Set root password | |
! echo root:$password | chpasswd | |
! mkdir -p /var/run/sshd | |
! echo "PermitRootLogin yes" >> /etc/ssh/sshd_config | |
! echo "PasswordAuthentication yes" >> /etc/ssh/sshd_config | |
! echo "LD_LIBRARY_PATH=/usr/lib64-nvidia" >> /root/.bashrc | |
! echo "export LD_LIBRARY_PATH" >> /root/.bashrc | |
#Run sshd | |
get_ipython().system_raw('/usr/sbin/sshd -D &') | |
#Ask token | |
print("Copy authtoken from https://dashboard.ngrok.com/auth") | |
authtoken = getpass.getpass() | |
while True: | |
try: | |
#Create tunnel | |
get_ipython().system_raw('./ngrok authtoken $authtoken && ./ngrok tcp 22 &') | |
#Get public address and print connect command | |
with urllib.request.urlopen('http://localhost:4040/api/tunnels') as response: | |
data = json.loads(response.read().decode()) | |
(host, port) = data['tunnels'][0]['public_url'][6:].split(':') | |
print(f'SSH command: ssh -p{port} root@{host}') | |
#Print root password | |
print(f'Root password: {password}') | |
except: | |
continue | |
break |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment