1 import os

     2 import sys

     3 import string

     4 from Tkinter import *

     5 from ScrolledText import ScrolledText

     6 from Dialog import Dialog

     7 import signal

     8 

     9 BUFSIZE = 512

    10 

    11 class ShellWindow(ScrolledText):

    12 

    13     def __init__(self, master=None, shell=None, **cnf):

    14         if not shell:

    15             try:

    16                 shell = os.environ['SHELL']

    17             except KeyError:

    18                 shell = '/bin/sh'

    19             shell = shell + ' -i'

    20         args = string.split(shell)

    21         shell = args[0]

    22 

    23         apply(ScrolledText.__init__, (self, master), cnf)

    24         self.pos = '1.0'

    25         self.bind('<Return>', self.inputhandler)

    26         self.bind('<Control-c>', self.sigint)

    27         self.bind('<Control-t>', self.sigterm)

    28         self.bind('<Control-k>', self.sigkill)

    29         self.bind('<Control-d>', self.sendeof)

    30 

    31         self.pid, self.fromchild, self.tochild = spawn(shell, args)

    32         self.tk.createfilehandler(self.fromchild, READABLE,

    33                                   self.outputhandler)

    34 

    35     def outputhandler(self, file, mask):

    36         data = os.read(file, BUFSIZE)

    37         if not data:

    38             self.tk.deletefilehandler(file)

    39             pid, sts = os.waitpid(self.pid, 0)

    40             print 'pid', pid, 'status', sts

    41             self.pid = None

    42             detail = sts>>8

    43             cause = sts & 0xff

    44             if cause == 0:

    45                 msg = "exit status %d" % detail

    46             else:

    47                 msg = "killed by signal %d" % (cause & 0x7f)

    48                 if cause & 0x80:

    49                     msg = msg + " -- core dumped"

    50             Dialog(self.master,

    51                    text=msg,

    52                    title="Exit status",

    53                    bitmap='warning',

    54                    default=0,

    55                    strings=('OK',))

    56             return

    57         self.insert(END, data)

    58         self.pos = self.index("end - 1 char")

    59         self.yview_pickplace(END)

    60 

    61     def inputhandler(self, *args):

    62         if not self.pid:

    63             self.no_process()

    64             return "break"

    65         self.insert(END, "\n")

    66         line = self.get(self.pos, "end - 1 char")

    67         self.pos = self.index(END)

    68         os.write(self.tochild, line)

    69         return "break"

    70 

    71     def sendeof(self, *args):

    72         if not self.pid:

    73             self.no_process()

    74             return "break"

    75         os.close(self.tochild)

    76         return "break"

    77 

    78     def sendsig(self, sig):

    79         if not self.pid:

    80             self.no_process()

    81             return "break"

    82         os.kill(self.pid, sig)

    83         return "break"

    84 

    85     def sigint(self, *args):

    86         return self.sendsig(signal.SIGINT)

    87 

    88     def sigquit(self, *args):

    89         return self.sendsig(signal.SIGQUIT)

    90 

    91     def sigterm(self, *args):

    92         return self.sendsig(signal.SIGTERM)

    93 

    94     def sigkill(self, *args):

    95         return self.sendsig(signal.SIGKILL)

    96 

    97     def no_process(self):

    98         Dialog(self.master,

    99                text="No active process",

   100                title="No process",

   101                bitmap='error',

   102                default=0,

   103                strings=('OK',))

   104 

   105 MAXFD = 100     # Max number of file descriptors (os.getdtablesize()???)

   106 

   107 def spawn(prog, args):

   108     p2cread, p2cwrite = os.pipe()

   109     c2pread, c2pwrite = os.pipe()

   110     pid = os.fork()

   111     if pid == 0:

   112         # Child

   113         for i in 0, 1, 2:

   114             try:

   115                 os.close(i)

   116             except os.error:

   117                 pass

   118         if os.dup(p2cread) <> 0:

   119             sys.stderr.write('popen2: bad read dup\n')

   120         if os.dup(c2pwrite) <> 1:

   121             sys.stderr.write('popen2: bad write dup\n')

   122         if os.dup(c2pwrite) <> 2:

   123             sys.stderr.write('popen2: bad write dup\n')

   124         os.closerange(3, MAXFD)

   125         try:

   126             os.execvp(prog, args)

   127         finally:

   128             sys.stderr.write('execvp failed\n')

   129             os._exit(1)

   130     os.close(p2cread)

   131     os.close(c2pwrite)

   132     return pid, c2pread, p2cwrite

   133 

   134 def test():

   135     shell = string.join(sys.argv[1:])

   136     root = Tk()

   137     root.minsize(1, 1)

   138     if shell:

   139         w = ShellWindow(root, shell=shell)

   140     else:

   141         w = ShellWindow(root)

   142     w.pack(expand=1, fill=BOTH)

   143     w.focus_set()

   144     w.tk.mainloop()

   145 

   146 if __name__ == '__main__':

   147     test()