summaryrefslogtreecommitdiffstats
path: root/func/minion/utils.py
blob: acdf61344ece69deff82c505595c761502803b9c (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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
"""
Copyright 2007, Red Hat, Inc
see AUTHORS

This software may be freely redistributed under the terms of the GNU
general public license.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
"""

import os
import socket
import string
import sys
import time
import traceback
import xmlrpclib
import glob

import codes
from func import certs
from func.config import read_config
from certmaster.commonconfig import MinionConfig
from func import logger

# "localhost" is a lame hostname to use for a key, so try to get
# a more meaningful hostname. We do this by connecting to the certmaster
# and seeing what interface/ip it uses to make that connection, and looking
# up the hostname for that. 
def get_hostname():

    # FIXME: this code ignores http proxies (which granted, we don't
    #      support elsewhere either. It also hardcodes the port number
    #      for the certmaster for now
    hostname = None
    hostname = socket.gethostname()
    try:
        ip = socket.gethostbyname(hostname)
    except:
        return hostname
    if ip != "127.0.0.1":
        return hostname


    config_file = '/etc/certmaster/minion.conf'
    minion_config = read_config(config_file, MinionConfig)

    server = minion_config.certmaster
    port = 51235

    try:
        s = socket.socket()
        s.settimeout(5)
        s.connect((server, port))
        (intf, port) = s.getsockname()
        hostname = socket.gethostbyaddr(intf)[0]
        s.close()
    except:
        s.close()
        raise

    return hostname
    


# this is kind of handy, so keep it around for now
# but we really need to fix out server side logging and error
# reporting so we don't need it
def trace_me():
    x = traceback.extract_stack()
    bar = string.join(traceback.format_list(x))
    return bar


def daemonize(pidfile=None):
    """
    Daemonize this process with the UNIX double-fork trick.
    Writes the new PID to the provided file name if not None.
    """

    print pidfile
    pid = os.fork()
    if pid > 0:
        sys.exit(0)
    os.setsid()
    os.umask(0)
    pid = os.fork()


    if pid > 0:
        if pidfile is not None:
            open(pidfile, "w").write(str(pid))
        sys.exit(0)

def get_acls_from_config(acldir='/etc/func/minion-acl.d'):
    """
    takes a dir of .acl files
    returns a dict of hostname+hash =  [methods, to, run]
    
    """
    
    acls = {}
    if not os.path.exists(acldir):
        print 'acl dir does not exist: %s' % acldir
        return acls
    
    # get the set of files
    acl_glob = '%s/*.acl' % acldir
    files = glob.glob(acl_glob)
    
    for acl_file in files:
        
        try:
            fo = open(acl_file, 'r')
        except (IOError, OSError), e:
            print 'cannot open acl config file: %s - %s' % (acl_file, e)
            continue
    
        for line in fo.readlines():
            if line.startswith('#'): continue
            if line.strip() == '': continue
            line = line.replace('\n', '')
            (host, methods) = line.split('=')
            host = host.strip().lower()
            methods = methods.strip()
            methods = methods.replace(',',' ')
            methods = methods.split()
            if not acls.has_key(host):
                acls[host] = []
            acls[host].extend(methods)
    
    return acls