← Back to branch summary

~ntt-pf-lab/nova/monkey_patch_notification

« back to all changes in this revision

Viewing changes to vendor/Twisted-10.0.0/twisted/runner/inetd.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/Twisted-10.0.0/twisted/runner/inetd.py
 
1
# Copyright (c) 2001-2004 Twisted Matrix Laboratories.
 
2
# See LICENSE for details.
 
3
 
 
4
# 
 
5
 
 
6
"""
 
7
Twisted inetd.
 
8
 
 
9
Maintainer: Andrew Bennetts
 
10
 
 
11
Future Plans: Bugfixes.  Specifically for UDP and Sun-RPC, which don't work
 
12
correctly yet.
 
13
"""
 
14
 
 
15
import os
 
16
 
 
17
from twisted.internet import process, reactor, fdesc
 
18
from twisted.internet.protocol import Protocol, ServerFactory
 
19
from twisted.protocols import wire
 
20
 
 
21
# A dict of known 'internal' services (i.e. those that don't involve spawning
 
22
# another process.
 
23
internalProtocols = {
 
24
    'echo': wire.Echo,
 
25
    'chargen': wire.Chargen,
 
26
    'discard': wire.Discard,
 
27
    'daytime': wire.Daytime,
 
28
    'time': wire.Time,
 
29
}
 
30
            
 
31
 
 
32
class InetdProtocol(Protocol):
 
33
    """Forks a child process on connectionMade, passing the socket as fd 0."""
 
34
    def connectionMade(self):
 
35
        sockFD = self.transport.fileno()
 
36
        childFDs = {0: sockFD, 1: sockFD}
 
37
        if self.factory.stderrFile:
 
38
            childFDs[2] = self.factory.stderrFile.fileno()
 
39
 
 
40
        # processes run by inetd expect blocking sockets
 
41
        # FIXME: maybe this should be done in process.py?  are other uses of
 
42
        #        Process possibly affected by this?
 
43
        fdesc.setBlocking(sockFD)
 
44
        if childFDs.has_key(2):
 
45
            fdesc.setBlocking(childFDs[2])
 
46
 
 
47
        service = self.factory.service
 
48
        uid = service.user
 
49
        gid = service.group
 
50
 
 
51
        # don't tell Process to change our UID/GID if it's what we
 
52
        # already are
 
53
        if uid == os.getuid():
 
54
            uid = None
 
55
        if gid == os.getgid():
 
56
            gid = None
 
57
 
 
58
        process.Process(None, service.program, service.programArgs, os.environ,
 
59
                        None, None, uid, gid, childFDs)
 
60
 
 
61
        reactor.removeReader(self.transport)
 
62
        reactor.removeWriter(self.transport)
 
63
                        
 
64
 
 
65
class InetdFactory(ServerFactory):
 
66
    protocol = InetdProtocol
 
67
    stderrFile = None
 
68
    
 
69
    def __init__(self, service):
 
70
        self.service = service