reverse_ssh.py

import argparse
import thread
import subprocess
from socketIO_client import SocketIO, LoggingNamespace

def main():
    global commandArgs
    global robotID
    global appServerSocketIO

    parser = argparse.ArgumentParser(description='start robot control program')

    parser.add_argument('robot_id')
    parser.add_argument('--reverse-ssh-key-file', default='/home/pi/reverse_ssh_key1.pem')
    parser.add_argument('--reverse-ssh-host', default='ubuntu@52.52.204.174')

    commandArgs = parser.parse_args()
    print commandArgs
    robotID = commandArgs.robot_id


    print "connecting to app server socket.io"
    appServerSocketIO = SocketIO('letsrobot.tv', 8022, LoggingNamespace)
    print "finished connecting to app server"
   
    appServerSocketIO.on('connect', onHandleAppServerConnect)
    appServerSocketIO.on('reconnect', onHandleAppServerReconnect)
    appServerSocketIO.on('disconnect', onHandleAppServerDisconnect)
    
    appServerSocketIO.on('reverse_ssh_8872381747239', startReverseSshProcess)
    appServerSocketIO.on('end_reverse_ssh_8872381747239', endReverseSshProcess)


    while True:
        appServerSocketIO.wait(seconds=1)



def startReverseSshProcess(*args):
    thread.start_new_thread(handleStartReverseSshProcess, args)

   
def endReverseSshProcess(*args):
    thread.start_new_thread(handleEndReverseSshProcess, args)



def handleStartReverseSshProcess(args):
    print "starting reverse ssh"
    appServerSocketIO.emit("reverse_ssh_info", "starting")

    returnCode = subprocess.call(["/usr/bin/ssh",
                                  "-X",
                                  "-i", commandArgs.reverse_ssh_key_file,
                                  "-N",
                                  "-R", "2222:localhost:22",
                                  commandArgs.reverse_ssh_host])

    appServerSocketIO.emit("reverse_ssh_info", "return code: " + str(returnCode))
    print "reverse ssh process has exited with code", str(returnCode)

    
def handleEndReverseSshProcess(args):
    print "handling end reverse ssh process"
    resultCode = subprocess.call(["killall", "ssh"])
    print "result code of killall ssh:", resultCode



def onHandleAppServerConnect(*args):
    print
    print "chat socket.io connect"
    print
    identifyRobotID()


def onHandleAppServerReconnect(*args):
    print
    print "app server socket.io reconnect"
    print
    identifyRobotID()    
    

def onHandleAppServerDisconnect(*args):
    print
    print "app server socket.io disconnect"
    print


    
def identifyRobotID():
    """tells the server which robot is using the connection"""
    print "sending identify robot id messages"
    appServerSocketIO.emit('identify_robot_id', robotID);


    

      
main()