summaryrefslogtreecommitdiffstats
path: root/scripts/fork.py
blob: 8d1e85586bb14ab1a7e6a31da30be41ef45e39ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import irssi
import os
import time
import sys

child_pid = 0

def sig_pidwait(pid, status):
    if child_pid != pid:
        print 'pidwait dont know',pid
        return

    if os.WIFSIGNALED(status):
        print '%d killed by signal' % pid
    elif os.WIFEXITED(status):
        print '%d exited(%d)' % (pid, os.WEXITSTATUS(status))

    irssi.signal_remove('pidwait')

def read_child(fd, condition, out):
    data = os.read(fd, 512)
    if not data:
        return False
    out.write(data) 
    return True

def childfunc():
    """ do your stuff """
    for i in xrange(30):
        print 'ME CHILD', i
        time.sleep(1)    


def cmd_forkoff(data, server, witem):
    global child_pid
   
    rs, ws = os.pipe()
    re, we = os.pipe()

    pid = os.fork()
    if pid > 0:
        #parent
        child_pid = pid
        irssi.pidwait_add(pid)
        print 'forked off',pid
        irssi.signal_add('pidwait', sig_pidwait)

        #redirect child output
        irssi.io_add_watch(rs, read_child, sys.stdout)
        irssi.io_add_watch(re, read_child, sys.stderr)

    else:
        #child
        sys.stdout = os.fdopen(ws, 'w', 0)
        sys.stderr = os.fdopen(we, 'w', 0)

        childfunc()

        sys.stdout.close()
        sys.stderr.close()
        os._exit(5)

irssi.command_bind('forkoff', cmd_forkoff)