build: improved exception handling for systems without rpath
[nivanova/samba-autobuild/.git] / buildtools / wafsamba / samba_conftests.py
1 # a set of config tests that use the samba_autoconf functions
2 # to test for commonly needed configuration options
3 import os, Build, shutil, Utils
4 from Configure import conf
5
6 @conf
7 def CHECK_ICONV(conf, define='HAVE_NATIVE_ICONV'):
8     '''check if the iconv library is installed
9        optionally pass a define'''
10     if conf.CHECK_FUNCS_IN('iconv_open', 'iconv', checklibc=True, headers='iconv.h'):
11         conf.DEFINE(define, 1)
12         return True
13     return False
14
15
16 @conf
17 def CHECK_LARGEFILE(conf, define='HAVE_LARGEFILE'):
18     '''see what we need for largefile support'''
19     if conf.CHECK_CODE('return !(sizeof(off_t) >= 8)',
20                        define,
21                        execute=True,
22                        msg='Checking for large file support'):
23         return True
24     if conf.CHECK_CODE('return !(sizeof(off_t) >= 8)',
25                        define,
26                        execute=True,
27                        cflags='-D_FILE_OFFSET_BITS=64',
28                        msg='Checking for -D_FILE_OFFSET_BITS=64'):
29         conf.DEFINE('_FILE_OFFSET_BITS', 64)
30         return True
31     return False
32
33
34 @conf
35 def CHECK_C_PROTOTYPE(conf, function, prototype, define, headers=None):
36     '''verify that a C prototype matches the one on the current system'''
37     if not conf.CHECK_DECLS(function, headers=headers):
38         return False
39     return conf.CHECK_CODE('%s; void *_x = (void *)%s' % (prototype, function),
40                            define=define,
41                            local_include=False,
42                            headers=headers,
43                            msg='Checking C prototype for %s' % function)
44
45
46 @conf
47 def CHECK_CHARSET_EXISTS(conf, charset, outcharset='UCS2-LE', libs=None, headers=None, define=None):
48     '''check that a named charset is able to be used with iconv_open() for conversion
49     to a target charset
50     '''
51     msg = 'Checking if can we convert from %s to %s' % (charset, outcharset)
52     if define is None:
53         define = 'HAVE_CHARSET_%s' % charset.upper().replace('-','_')
54     return conf.CHECK_CODE('''
55                            iconv_t cd = iconv_open("%s", "%s");
56                            if (cd == 0 || cd == (iconv_t)-1) {
57                              return -1;
58                              }
59                              return 0;
60                              ''' % (charset, outcharset),
61                            define=define,
62                            execute=True,
63                            libs=libs,
64                            msg=msg,
65                            headers=headers)
66
67
68
69 # this one is quite complex, and should probably be broken up
70 # into several parts. I'd quite like to create a set of CHECK_COMPOUND()
71 # functions that make writing complex compound tests like this much easier
72 @conf
73 def CHECK_RPATH_SUPPORT(conf):
74     '''see if the platform supports rpath for libraries'''
75     k = 0
76     while k < 10000:
77         dir = os.path.join(conf.blddir, '.conf_check_%d' % k)
78         try:
79             shutil.rmtree(dir)
80         except OSError:
81             pass
82         try:
83             os.stat(dir)
84         except:
85             break
86         k += 1
87
88     try:
89         os.makedirs(dir)
90     except:
91         conf.fatal('cannot create a configuration test folder %r' % dir)
92
93     try:
94         os.stat(dir)
95     except:
96         conf.fatal('cannot use the configuration test folder %r' % dir)
97
98     bdir = os.path.join(dir, 'testbuild')
99     if not os.path.exists(bdir):
100         os.makedirs(bdir)
101
102     env = conf.env
103
104     subdir = os.path.join(dir, "libdir")
105
106     os.makedirs(subdir)
107
108     dest = open(os.path.join(subdir, 'lib1.c'), 'w')
109     dest.write('int lib_func(void) { return 42; }\n')
110     dest.close()
111
112     dest = open(os.path.join(dir, 'main.c'), 'w')
113     dest.write('int main(void) {return !(lib_func() == 42);}\n')
114     dest.close()
115
116     back = os.path.abspath('.')
117
118     bld = Build.BuildContext()
119     bld.log = conf.log
120     bld.all_envs.update(conf.all_envs)
121     bld.all_envs['default'] = env
122     bld.lst_variants = bld.all_envs.keys()
123     bld.load_dirs(dir, bdir)
124
125     os.chdir(dir)
126
127     bld.rescan(bld.srcnode)
128
129     bld(features='cc cshlib',
130         source='libdir/lib1.c',
131         target='libdir/lib1',
132         name='lib1')
133
134     o = bld(features='cc cprogram',
135             source='main.c',
136             target='prog1',
137             uselib_local='lib1',
138             rpath=os.path.join(bdir, 'default/libdir'))
139
140     # compile the program
141     try:
142         bld.compile()
143     except:
144         conf.check_message('rpath support', '', False)
145         return False
146
147     # chdir before returning
148     os.chdir(back)
149
150     # path for execution
151     lastprog = o.link_task.outputs[0].abspath(env)
152
153     # we need to run the program, try to get its result
154     args = []
155     proc = Utils.pproc.Popen([lastprog] + args, stdout=Utils.pproc.PIPE, stderr=Utils.pproc.PIPE)
156     (out, err) = proc.communicate()
157     w = conf.log.write
158     w(str(out))
159     w('\n')
160     w(str(err))
161     w('\nreturncode %r\n' % proc.returncode)
162     ret = (proc.returncode == 0)
163
164     conf.check_message('rpath support', '', ret)
165     return ret