tup_client.py

# This script can be imported from a python build script to access the
# tup.config variables.
#
# Usage:
#
# import tup_client
# value = tup_client.config_var('FOO')
#
# This will store the value of CONFIG_FOO into 'value'. If the variable does
# not exist, 'value' will be None. In either case, a dependency will be created
# on the @(FOO) variable.
import os
import mmap
import struct
import string

tup_vardict_map = None
tup_vardict_num = None
tup_entry_size = 0

def config_var(key):
    global tup_vardict_map
    global tup_vardict_num
    global tup_entry_size

    try:
        f = open("@tup@/" + key, "r");
    except IOError:
        pass

    if tup_vardict_num is None:
        fd = os.getenv('tup_vardict');
        if fd is None:
            raise Exception('tup error: This python module can only be used as a sub-process from tup')

        if int(fd) < 0 or os.fstat(int(fd)).st_size == 0:
            tup_vardict_num = 0
        else:
            tup_vardict_map = mmap.mmap(int(fd), 0, mmap.MAP_PRIVATE, mmap.PROT_READ);
            tup_vardict_num = struct.unpack("@i", tup_vardict_map.read(4))[0];
            tup_entry_size = (tup_vardict_num + 1) * 4

    left = -1
    right = tup_vardict_num

    while(True):
        cur = (right - left) / 2
        if(cur <= 0):
            return None
        cur += left
        if(cur >= tup_vardict_num):
            return None
        # +1 for the number of entries at the start of the file
        tup_vardict_map.seek((cur+1) * 4)
        offset = struct.unpack("@i", tup_vardict_map.read(4))[0];
        tup_vardict_map.seek(offset + tup_entry_size)
        keylen = len(key)
        bytesleft = keylen

        p = '\0'
        while(bytesleft > 0):
            p = tup_vardict_map.read_byte()
            k = key[keylen - bytesleft]
            if(p == '=' or p < k):
                left = cur
                break
            elif(p > k):
                right = cur
                break
            bytesleft -= 1

        if(bytesleft == 0):
            p = tup_vardict_map.read_byte()
            if(p == '='):
                rc = []
                while(1):
                    p = tup_vardict_map.read_byte()
                    if(p == '\0'):
                        return string.join(rc, "")
                    rc.append(p)
            else:
                right = cur