graphene/LibOS/buildglibc.py

#!/usr/bin/python


import sys, os, string, subprocess, shutil, fileinput, multiprocessing, re


def replaceAll(fd,searchExp,replaceExp):
    for line in fileinput.input(fd, inplace=1):
        if searchExp in line:
            line = line.replace(searchExp,replaceExp)
        sys.stdout.write(line)

def prependText(filename, text) :
    data = ""
    with open(filename, 'r') as original:
        data = original.read()
    with open(filename, 'w') as modified:
        modified.write(text)
        modified.write(data)

def appendText(filename, text) :
    with open(filename, "a") as myfile:
        myfile.write(text)



try:
    home = os.getcwd()
    glibc = "glibc-2.17"
    glibcParent = "" # glibc parent directory
    glibcDir = ""    # glibc dir (ex. glibc-2.17)
    buildDir = "build"
    installDir = "/usr/local/graphene"
    commandStr = ""
    commandOutput = ""

    debug_flags = ""
    if len(sys.argv) > 1 and sys.argv[1] == 'debug':
        debug_flags = "-g"



    #########################################
    #### get the locations of directories ###
    #########################################

    iput = raw_input('use {0} as the source of GNU libc? ([y]/n):'.format(glibc)).lower()
    if not iput == 'y' and not iput == '' :
        glibc = raw_input('enter the glibc source to install with: ')

    iput = raw_input('{0} contains glibc code to compile? ([y]/n): '.format(glibc)).lower()
    if not iput == 'y' and not iput == '':
        glibc = raw_input('directory containing glibc code to compile: ')
    if os.path.isdir(glibc) :
        glibc = os.path.abspath(glibc)
        glibcParent,glibcDir = os.path.split(glibc)
        print '{0} + {1}'.format(glibcParent, glibcDir)

    iput = raw_input('use {0} as the directory to build glibc in? ([y]/n): '.format(buildDir)).lower()
    if not iput == 'y' and not iput == '':
        buildDir = raw_input('the directory to build glibc in:  ')
    buildDir = os.path.abspath(buildDir)
    print 'using build dir: {0}'.format(buildDir)
    if os.path.isdir(buildDir) :
        clean = raw_input('clean build (delete {0}, rerun configure, etc.)? ([y]/n): '.format(buildDir))
        if clean == 'y' or clean == '':
            shutil.rmtree(buildDir)
            os.makedirs(buildDir)
        else :
            print 'Then just go to {0} and type make...'.format(buildDir)
            exit(0)
    else :
        os.makedirs(buildDir)

    iput = raw_input('use {0} as the directory to install glibc in? ([y]/n): '.format(installDir)).lower()
    if not iput == 'y' and not iput == '':
        installDir = raw_input('the directory to install glibc in:  ')
    installDir = os.path.abspath(installDir)
    print 'using install dir: {0}'.format(installDir)



    ################################
    #### doctor glibc's Makefile ###
    ################################

    os.chdir(buildDir)

    cflags= '{0} -O2 -U_FORTIFY_SOURCE -fno-stack-protector'.format(debug_flags)
    disabled_features = { 'nscd' }
    extra_flags = '--with-tls --enable-add-ons=nptl --without-selinux {0}'.format(' '.join(['--disable-' + f for f in disabled_features]))

    ##    configure
    commandStr = r'CFLAGS="{2}" {0}/configure --prefix={1} {3} | tee configure.out'.format(glibc, installDir, cflags, extra_flags)
    print commandStr
    commandOutput = subprocess.call(commandStr, shell=True)

    ##    Enable parallel builds
    numCPUs = multiprocessing.cpu_count()
    ##    Don't use up all the cores!
    numCPUs = numCPUs - 1
    if numCPUs == 0:
        numCPUs = 1
    replaceAll('Makefile', r'# PARALLELMFLAGS = -j4', r'PARALLELMFLAGS = -j{0}'.format(numCPUs))


    link_binaries = [ ( 'elf',    'ld-linux-x86-64.so.2' ),
                      ( 'nptl',   'libpthread.so.0' ),
                      ( 'nptl_db','libthread_db.so.1' ),
                      ( 'dlfcn',  'libdl.so.2' ),
                      ( 'csu',    'crt1.o' ),
                      ( 'csu',    'crti.o' ),
                      ( 'csu',    'crtn.o' ),
                      ( 'libos',  'liblibos.so.1' ) ]

    for (dir, bin) in link_binaries:
        print bin + ' -> ' + dir + '/' + bin
        os.symlink(dir + '/' + bin, bin)


    print '\n\n\nNow type \'make\' in \'{0}\'\n\n'.format(buildDir)

except:
    print 'uh-oh: {0}'.format(sys.exc_info()[0])