Og.

From 962918612a1fcddb56acc3589120c317fa37904c Mon Sep 17 00:00:00 2001 From: noah Date: Wed, 3 Mar 2004 09:31:18 +0000 Subject: Works so well that it's almost useful. git-svn-id: http://pexpect.svn.sourceforge.net/svnroot/pexpect/trunk@217 656d521f-e311-0410-88e0-e7920216d269 --- pexpect/examples/bd_client_web.py | 101 +++++++++++++++--------- pexpect/examples/bd_serv.py | 159 ++++++++++++++++++++++++++++++-------- 2 files changed, 191 insertions(+), 69 deletions(-) diff --git a/pexpect/examples/bd_client_web.py b/pexpect/examples/bd_client_web.py index 5b5e9c6..10a0bd0 100755 --- a/pexpect/examples/bd_client_web.py +++ b/pexpect/examples/bd_client_web.py @@ -11,45 +11,74 @@ import re import socket ROOTPATH="/tmp" - -def page (result = ''): - """Return the main form""" - return """ - - Og. - + -

-%s
-

%(SHELL_OUTPUT)s

+ - """ % (result) +""" +def page (result = ''): + """Return the main form""" + return CGISH_HTML % {'SHELL_OUTPUT':result} def bd_client (command, host='localhost', port = 1666): HOST = 'localhost' # The remote host @@ -57,22 +86,22 @@ def bd_client (command, host='localhost', port = 1666): s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect((HOST, PORT)) s.send(command) - data = s.recv(1024) + data = s.recv (1920) s.close() return data #fout = file ('/tmp/log2','w') #fout.write (command) #fout.write ('\n') - s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - s.connect((host, port)) - s.send(command) - data = s.recv(1024) +# s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) +# s.connect((host, port)) +# s.send(command) +# data = s.recv(1024) #fout.write (data) #fout.write ('\n') - s.close() +# s.close() #fout.close() - return data +# return data #def link (matchobject): # """Used in re.sub calls to replace a matched object with an HTML link.""" diff --git a/pexpect/examples/bd_serv.py b/pexpect/examples/bd_serv.py index 260a5d6..d88f487 100755 --- a/pexpect/examples/bd_serv.py +++ b/pexpect/examples/bd_serv.py @@ -1,36 +1,129 @@ #!/usr/bin/env python -import socket, pexpect -import time - -p = pexpect.spawn ('bash') -time.sleep (0.1) -p.expect ('\$') -time.sleep (0.1) - -HOST = '' # Symbolic name meaning the local host -PORT = 1666 # Arbitrary non-privileged port -s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) -s.bind((HOST, PORT)) -print 'Listen' -s.listen(1) -print 'Accept' -while 1: - conn, addr = s.accept() - print 'Connected by', addr - data = conn.recv(1024) - print 'RECEIVED:' - print data -# if not data: -# break - time.sleep (0.1) - p.sendline (data) - time.sleep (0.1) - p.expect ('\$') - time.sleep (0.1) - response = p.before + p.after - print 'RESPONSE:' - print response - conn.send(response) +import socket, pexpect, ANSI +import time, sys, os + +# Clearly having the password on the command line is not a good idea, but +# then this entire enterprise is probably not the most security concious thing +# I've ever built. This should be considered an experimental tool. +# USER = sys.argv[1] +# PASSWORD = sys.argv[2] +# PORT = sys.argv[3] + +def daemonize (stdin='/dev/null', stdout='/dev/null', stderr='/dev/null'): + '''This forks the current process into a daemon. + Almost none of this is necessary (or advisable) if your daemon + is being started by inetd. In that case, stdin, stdout and stderr are + all set up for you to refer to the network connection, and the fork()s + and session manipulation should not be done (to avoid confusing inetd). + Only the chdir() and umask() steps remain as useful. + + References: + UNIX Programming FAQ + 1.7 How do I get my program to act like a daemon? + http://www.erlenstar.demon.co.uk/unix/faq_2.html#SEC16 + + Advanced Programming in the Unix Environment + W. Richard Stevens, 1992, Addison-Wesley, ISBN 0-201-56317-7. + + The stdin, stdout, and stderr arguments are file names that + will be opened and be used to replace the standard file descriptors + in sys.stdin, sys.stdout, and sys.stderr. + These arguments are optional and default to /dev/null. + Note that stderr is opened unbuffered, so + if it shares a file with stdout then interleaved output + may not appear in the order that you expect. + ''' + + # Do first fork. + try: + pid = os.fork() + if pid > 0: + sys.exit(0) # Exit first parent. + except OSError, e: + sys.stderr.write ("fork #1 failed: (%d) %s\n" % (e.errno, e.strerror) ) + sys.exit(1) + + # Decouple from parent environment. + os.chdir("/") + os.umask(0) + os.setsid() + + # Do second fork. + try: + pid = os.fork() + if pid > 0: + sys.exit(0) # Exit second parent. + except OSError, e: + sys.stderr.write ("fork #2 failed: (%d) %s\n" % (e.errno, e.strerror) ) + sys.exit(1) + + # Now I am a daemon! -conn.close() + # Redirect standard file descriptors. + si = open(stdin, 'r') + so = open(stdout, 'a+') + se = open(stderr, 'a+', 0) + os.dup2(si.fileno(), sys.stdin.fileno()) + os.dup2(so.fileno(), sys.stdout.fileno()) + os.dup2(se.fileno(), sys.stderr.fileno()) + + # I now return as the daemon + return 0 + +def main (): + USER = sys.argv[1] + PASSWORD = sys.argv[2] + PORT = int(sys.argv[3]) + + #daemonize () + #daemonize('/dev/null','/tmp/daemon.log','/tmp/daemon.log') + + sys.stdout.write ('Daemon started with pid %d\n' % os.getpid() ) + + vs = ANSI.ANSI (24,80) + p = pexpect.spawn ('ssh %(USER)s@localhost'%locals(), timeout=9) + p.expect ('assword') + p.sendline (PASSWORD) + time.sleep (0.2) + #p.sendline ('stty -echo') + #time.sleep (0.2) + p.sendline ('export PS1="HAON "') + time.sleep (0.2) + p.expect (pexpect.TIMEOUT) + print p.before + vs.process (p.before) + HOST = '' # Symbolic name meaning the local host + s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + s.bind((HOST, PORT)) + print 'Listen' + s.listen(1) + print 'Accept' + while 1: + conn, addr = s.accept() + print 'Connected by', addr + data = conn.recv(1024) + print data + + if data == 'exit': + p.sendline (exit) + break + if not data == "pass": + p.sendline (data) + time.sleep (0.1) + p.expect (['HAON ',pexpect.TIMEOUT]) + #response = p.before + sh_response = p.before.replace ('\r', '') + vs.process_list (sh_response) + response = str (vs) + print response + sent = conn.send(response) + if sent < len (response): + print "Sent is too short" + + +if __name__ == "__main__": +# daemonize('/dev/null','/tmp/daemon.log','/tmp/daemon.log') + main() + + -- cgit v1.2.1