nwrap: Don't leak memory from gethostbyname*() functions
[obnox/samba/samba-obnox.git] / ctdb / wscript
1 #!/usr/bin/env python
2
3 APPNAME = 'ctdb'
4
5 blddir = 'bin'
6
7 import sys, os
8
9 # find the buildtools directory
10 srcdir = '.'
11 while not os.path.exists(srcdir+'/buildtools') and len(srcdir.split('/')) < 5:
12     srcdir = srcdir + '/..'
13 sys.path.insert(0, srcdir + '/buildtools/wafsamba')
14
15 import wafsamba, samba_dist, Options, Logs, Utils
16 import samba_utils, samba_version
17
18 env = samba_utils.LOAD_ENVIRONMENT()
19 if os.path.isfile('./VERSION'):
20     vdir = '.'
21 elif os.path.isfile('../VERSION'):
22     vdir = '..'
23 else:
24     Logs.error("VERSION file not found")
25
26 version = samba_version.samba_version_file('%s/VERSION' % vdir, vdir, env)
27 VERSION = version.STRING.replace('-', '.')
28
29 default_prefix = Options.default_prefix = '/usr/local'
30
31 samba_dist.DIST_DIRS('''ctdb:. lib/replace:lib/replace lib/talloc:lib/talloc
32                         lib/tevent:lib/tevent lib/tdb:lib/tdb
33                         lib/socket_wrapper:lib/socket_wrapper
34                         third_party/popt:third_party/popt
35                         lib/util:lib/util lib/tdb_wrap:lib/tdb_wrap
36                         lib/ccan:lib/ccan libcli/util:libcli/util
37                         buildtools:buildtools third_party/waf:third_party/waf''')
38
39 manpages = [
40     'ctdb.1',
41     'ctdb.7',
42     'ctdbd.1',
43     'ctdbd.conf.5',
44     'ctdbd_wrapper.1',
45     'ctdb-statistics.7',
46     'ctdb-tunables.7',
47     'ltdbtool.1',
48     'onnode.1',
49     'ping_pong.1'
50 ]
51
52
53 def set_options(opt):
54     opt.PRIVATE_EXTENSION_DEFAULT('ctdb')
55
56     opt.RECURSE('lib/replace')
57
58     opt.RECURSE('lib/util')
59
60     opt.RECURSE('lib/talloc')
61     opt.RECURSE('lib/tevent')
62     opt.RECURSE('lib/tdb')
63
64     opt.add_option('--enable-infiniband',
65                    help=("Turn on infiniband support (default=no)"),
66                    action="store_true", dest='ctdb_infiniband', default=False)
67     opt.add_option('--enable-pmda',
68                    help=("Turn on PCP pmda support (default=no)"),
69                    action="store_true", dest='ctdb_pmda', default=False)
70
71     opt.add_option('--with-logdir',
72                    help=("Path to log directory"),
73                    action="store", dest='ctdb_logdir', default=None)
74     opt.add_option('--with-socketpath',
75                    help=("path to CTDB daemon socket"),
76                    action="store", dest='ctdb_sockpath', default=None)
77
78
79 def configure(conf):
80
81     # CTDB relies on nested event loops
82     conf.env.TEVENT_DEPRECATED = 1
83
84     # No need to build python bindings for talloc/tevent/tdb
85     if conf.IN_LAUNCH_DIR():
86         conf.env.standalone_ctdb = True
87         Options.options.disable_python = True
88
89     conf.RECURSE('lib/replace')
90
91     if conf.env.standalone_ctdb:
92         conf.SAMBA_CHECK_PERL(mandatory=True)
93
94         conf.SAMBA_CHECK_PYTHON(mandatory=True, version=(2,5,0))
95         conf.SAMBA_CHECK_PYTHON_HEADERS(mandatory=True)
96
97     if conf.CHECK_FOR_THIRD_PARTY():
98         conf.RECURSE('third_party/popt')
99     else:
100         if not conf.CHECK_POPT():
101             raise Utils.WafError('popt development packages have not been found\nIf third_party is installed, check that it is in the proper place.')
102         else:
103             conf.define('USING_SYSTEM_POPT', 1)
104
105     conf.RECURSE('lib/util')
106
107     conf.RECURSE('lib/talloc')
108     conf.RECURSE('lib/tevent')
109     conf.RECURSE('lib/tdb')
110     if conf.env.standalone_ctdb or conf.CONFIG_GET('ENABLE_SELFTEST'):
111         conf.RECURSE('lib/socket_wrapper')
112
113     conf.CHECK_HEADERS('sched.h')
114     conf.CHECK_HEADERS('procinfo.h')
115     if sys.platform.startswith('aix') and not conf.CHECK_FUNCS('thread_setsched'):
116         Logs.error('Need thread_setsched() on AIX')
117         sys.exit(1)
118     elif not conf.CHECK_FUNCS('sched_setscheduler'):
119         Logs.error('Need sched_setscheduler()')
120         sys.exit(1)
121     conf.CHECK_FUNCS('mlockall')
122
123     if not conf.CHECK_VARIABLE('ETIME', headers='errno.h'):
124         conf.DEFINE('ETIME', 'ETIMEDOUT')
125
126     if sys.platform.startswith('linux'):
127         conf.SET_TARGET_TYPE('pcap', 'EMPTY')
128     else:
129         if not conf.CHECK_HEADERS('pcap.h'):
130             Logs.error('Need libpcap')
131             sys.exit(1)
132         if not conf.CHECK_FUNCS_IN('pcap_open_live', 'pcap', headers='pcap.h'):
133             Logs.error('Need libpcap')
134             sys.exit(1)
135
136     if not conf.CHECK_FUNCS_IN('backtrace backtrace_symbols', 'execinfo',
137                                checklibc=True, headers='execinfo.h'):
138         Logs.error('backtrace support not available')
139
140     have_pmda = False
141     if Options.options.ctdb_pmda:
142         pmda_support = True
143
144         if not conf.CHECK_HEADERS('pcp/pmapi.h pcp/impl.h pcp/pmda.h',
145                                   together=True):
146             pmda_support = False
147         if not conf.CHECK_FUNCS_IN('pmProgname', 'pcp'):
148             pmda_support = False
149         if not conf.CHECK_FUNCS_IN('pmdaDaemon', 'pcp_pmda'):
150             pmda_support = False
151         if pmda_support:
152             have_pmda = True
153         else:
154             Logs.error("PMDA support not available")
155     if have_pmda:
156         Logs.info('Building with PMDA support')
157         conf.define('HAVE_PMDA', 1)
158         conf.env.CTDB_PMDADIR = os.path.join(conf.env.LOCALSTATEDIR,
159                                              'lib/pcp/pmdas/ctdb')
160
161     have_infiniband = False
162     if Options.options.ctdb_infiniband:
163         ib_support = True
164
165         if not conf.CHECK_HEADERS('infiniband/verbs.h rdma/rdma_cma.h'):
166             ib_support = False
167         if not conf.CHECK_FUNCS_IN('ibv_create_qp', 'ibverbs'):
168             ib_support = False
169         if not conf.CHECK_FUNCS_IN('rdma_connect', 'rdmacm'):
170             ib_support = False
171         if ib_support:
172             have_infiniband = True
173         else:
174             Logs.error("Infiniband support not available")
175     if have_infiniband:
176         Logs.info('Building with Infiniband support')
177         conf.define('HAVE_INFINIBAND', 1)
178         conf.define('USE_INFINIBAND', 1)
179
180     conf.env.CTDB_BINDIR = os.path.join(conf.env.EXEC_PREFIX, 'bin')
181     conf.env.CTDB_ETCDIR = os.path.join(conf.env.SYSCONFDIR, 'ctdb')
182     conf.env.CTDB_VARDIR = os.path.join(conf.env.LOCALSTATEDIR, 'lib/ctdb')
183     conf.env.CTDB_RUNDIR = os.path.join(conf.env.LOCALSTATEDIR, 'run/ctdb')
184
185     if Options.options.ctdb_logdir:
186         conf.env.CTDB_LOGDIR = Options.options.ctdb_logdir
187     else:
188         conf.env.CTDB_LOGDIR = os.path.join(conf.env.LOCALSTATEDIR, 'log')
189
190     if Options.options.ctdb_sockpath:
191         conf.env.CTDB_SOCKPATH = Options.options.ctdb_sockpath
192     else:
193         conf.env.CTDB_SOCKPATH = os.path.join(conf.env.CTDB_RUNDIR,
194                                               'ctdbd.socket')
195     conf.define('CTDB_SOCKET', conf.env.CTDB_SOCKPATH)
196
197     conf.ADD_CFLAGS('''-DCTDB_HELPER_BINDIR=\"%s\"
198                        -DLOGDIR=\"%s\"
199                        -DCTDB_ETCDIR=\"%s\"
200                        -DCTDB_VARDIR=\"%s\"
201                        -DCTDB_RUNDIR=\"%s\"''' % (
202                     conf.env.CTDB_BINDIR,
203                     conf.env.CTDB_LOGDIR,
204                     conf.env.CTDB_ETCDIR,
205                     conf.env.CTDB_VARDIR,
206                     conf.env.CTDB_RUNDIR))
207
208     conf.env.CTDB_TEST_DATADIR = os.path.join(conf.env.EXEC_PREFIX,
209                                               'share/ctdb-tests')
210     conf.env.CTDB_TEST_LIBDIR = os.path.join(conf.env.LIBDIR, 'ctdb-tests')
211
212     # Allow unified compilation and separate compilation of utilities
213     # to find includes
214     if not conf.env.standalone_ctdb:
215         conf.ADD_EXTRA_INCLUDES('#include/public #ctdb/include #ctdb')
216     else:
217         if srcdir == '.':
218             # Building from tarball
219             conf.ADD_EXTRA_INCLUDES('#include')
220         else:
221             # Building standalone CTDB from within Samba tree
222             conf.ADD_EXTRA_INCLUDES('#ctdb/include')
223             conf.ADD_EXTRA_INCLUDES('#ctdb')
224         conf.ADD_EXTRA_INCLUDES('#lib #lib/replace')
225
226         conf.DEFINE('HAVE_CONFIG_H', 1, add_to_cflags=True)
227         conf.DEFINE('SAMBA_UTIL_CORE_ONLY', 1, add_to_cflags=True)
228         conf.SAMBA_CONFIG_H()
229
230     if 'XSLTPROC_MANPAGES' in conf.env and conf.env['XSLTPROC_MANPAGES']:
231         conf.env.ctdb_generate_manpages = True
232     else:
233         conf.env.ctdb_generate_manpages = False
234
235         Logs.info("xsltproc unavailable, checking for pre-built manpages")
236         conf.env.ctdb_prebuilt_manpages = []
237         for m in manpages:
238             if os.path.exists(os.path.join("doc", m)):
239                 Logs.info("  %s: yes" % (m))
240                 conf.env.ctdb_prebuilt_manpages.append(m)
241             else:
242                 Logs.info("  %s: no" % (m))
243
244 def build(bld):
245     if bld.env.standalone_ctdb:
246         # enable building of public headers in the build tree
247         bld.env.build_public_headers = 'include/public'
248
249     version_h = samba_utils.os_path_relpath(os.path.join(Options.launch_dir,
250                                                          "version.h"),
251                                             bld.curdir)
252
253     if bld.env.standalone_ctdb:
254         ctdb_mkversion = '../packaging/mkversion.sh'
255         t = bld.SAMBA_GENERATOR('ctdb-version-header',
256                                 target='include/ctdb_version.h',
257                                 rule='%s ${TGT} %s' % (ctdb_mkversion, VERSION),
258                                 dep_vars=['VERSION'])
259         t.env.VERSION = VERSION
260
261         t = bld.SAMBA_GENERATOR('ctdb-samba-version-header',
262                                 target=version_h,
263                                 rule='printf "#include \\"ctdb_version.h\\" \\n#define SAMBA_VERSION_STRING CTDB_VERSION_STRING\\n" > ${TGT}',
264                                 dep_vars=['VERSION'])
265         t.env.VERSION = VERSION
266     else:
267         t = bld.SAMBA_GENERATOR('ctdb-samba-version-header',
268                                 target='include/ctdb_version.h',
269                                 rule='printf "#include \\"%s\\" \\n#define CTDB_VERSION_STRING SAMBA_VERSION_STRING\\n" > ${TGT}' % version_h,
270                                 dep_vars=['VERSION'])
271         t.env.VERSION = VERSION
272
273     bld.env.PKGCONFIGDIR = '${LIBDIR}/pkgconfig'
274
275     bld.RECURSE('lib/replace')
276     if bld.CHECK_FOR_THIRD_PARTY():
277         bld.RECURSE('third_party/popt')
278
279     bld.RECURSE('lib/tdb_wrap')
280     bld.RECURSE('lib/util')
281
282     bld.RECURSE('lib/talloc')
283     bld.RECURSE('lib/tevent')
284     bld.RECURSE('lib/tdb')
285     if bld.env.standalone_ctdb or bld.CONFIG_GET('SOCKET_WRAPPER'):
286         bld.RECURSE('lib/socket_wrapper')
287
288     if bld.env.standalone_ctdb:
289         # If a combined build is implemented, CTDB will want to
290         # build against samba-util rather than samba-util-core.
291         # Similarly, other Samba subsystems expect samba-util.  So,
292         # for a standalone build, just define a fake samba-util
293         # subsystem that pulls in samba-util-core.
294         bld.SAMBA_SUBSYSTEM('samba-util',
295                             source='',
296                             deps='samba-util-core')
297
298     bld.SAMBA_SUBSYSTEM('ctdb-tcp',
299                         source=bld.SUBDIR('tcp',
300                                           'tcp_connect.c tcp_init.c tcp_io.c'),
301                         includes='include',
302                         deps='replace tdb talloc tevent')
303
304     ib_deps = ''
305     if bld.env.HAVE_INFINIBAND:
306         bld.SAMBA_SUBSYSTEM('ctdb-ib',
307                             source=bld.SUBDIR('ib',
308                                               '''ibwrapper.c ibw_ctdb.c
309                                                  ibw_ctdb_init.c'''),
310                             includes='include',
311                             deps='replace talloc tevent')
312         ib_deps = ' ctdb-ib rdmacm ibverbs'
313
314     if sys.platform.startswith('linux'):
315         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_linux.c')
316     elif sys.platform.startswith('aix'):
317         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_aix.c')
318     elif sys.platform.startswith('freebsd'):
319         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_freebsd.c')
320     elif sys.platform == 'kfreebsd':
321         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_kfreebsd.c')
322     elif sys.platform == 'gnu':
323         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_gnu.c')
324     else:
325         Logs.error("Platform %s not supported" % sys.platform)
326
327     bld.SAMBA_SUBSYSTEM('ctdb-system',
328                         source=bld.SUBDIR('common',
329                                           'system_common.c system_util.c') +
330                                CTDB_SYSTEM_SRC,
331                         includes='include',
332                         deps='replace talloc tevent tdb pcap')
333
334     bld.SAMBA_SUBSYSTEM('ctdb-common',
335                         source=bld.SUBDIR('common',
336                                           '''ctdb_io.c ctdb_util.c ctdb_ltdb.c
337                                              cmdline.c rb_tree.c'''),
338                         includes='include',
339                         deps='replace popt talloc tevent tdb popt ctdb-system')
340
341     bld.SAMBA_SUBSYSTEM('ctdb-util',
342                         source=bld.SUBDIR('common',
343                                           '''db_hash.c srvid.c reqid.c
344                                              pkt_read.c pkt_write.c comm.c
345                                              logging.c'''),
346                         deps='replace talloc tevent tdb tevent-unix-util')
347
348     bld.SAMBA_SUBSYSTEM('ctdb-protocol',
349                         source=bld.SUBDIR('protocol',
350                                           '''protocol_header.c protocol_packet.c
351                                              protocol_types.c protocol_call.c
352                                              protocol_message.c
353                                              protocol_control.c
354                                              protocol_client.c
355                                              protocol_util.c'''),
356                         includes='include',
357                         deps='replace talloc tdb')
358
359     bld.SAMBA_SUBSYSTEM('ctdb-client',
360                         source=bld.SUBDIR('client', 'ctdb_client.c'),
361                         includes='include',
362                         deps='''replace popt talloc tevent tdb
363                                 samba-util tdb-wrap ctdb-util''')
364
365     bld.SAMBA_SUBSYSTEM('ctdb-client2',
366                         source=bld.SUBDIR('client',
367                                           '''client_connect.c client_call.c
368                                              client_message.c client_control.c
369                                              client_message_sync.c
370                                              client_control_sync.c
371                                              client_db.c client_util.c
372                                           '''),
373                         includes='include',
374                         deps='replace talloc tevent tdb tdb-wrap')
375
376     bld.SAMBA_SUBSYSTEM('ctdb-server',
377                         source='server/ctdbd.c ' +
378                                bld.SUBDIR('server',
379                                           '''ctdb_daemon.c ctdb_recoverd.c
380                                              ctdb_recover.c ctdb_freeze.c
381                                              ctdb_tunables.c ctdb_monitor.c
382                                              ctdb_server.c ctdb_control.c
383                                              ctdb_call.c ctdb_ltdb_server.c
384                                              ctdb_traverse.c eventscript.c
385                                              ctdb_takeover.c ctdb_serverids.c
386                                              ctdb_persistent.c ctdb_keepalive.c
387                                              ctdb_logging.c
388                                              ctdb_logging_syslog.c
389                                              ctdb_logging_file.c
390                                              ctdb_uptime.c
391                                              ctdb_vacuum.c ctdb_banning.c
392                                              ctdb_statistics.c
393                                              ctdb_update_record.c
394                                              ctdb_lock.c ctdb_fork.c'''),
395                         includes='include',
396                         deps='replace popt talloc tevent tdb talloc_report')
397
398     bld.SAMBA_BINARY('ctdbd',
399                      source='',
400                      deps='''ctdb-server ctdb-client ctdb-common
401                              ctdb-system ctdb-tcp ctdb-util''' +
402                           ib_deps,
403                      install_path='${SBINDIR}',
404                      manpages='ctdbd.1')
405
406     bld.SAMBA_BINARY('ctdb',
407                      source='tools/ctdb.c',
408                      deps='ctdb-client ctdb-common ctdb-system ctdb-util',
409                      includes='include',
410                      install_path='${BINDIR}',
411                      manpages='ctdb.1')
412
413     bld.SAMBA_BINARY('ltdbtool',
414                      source='tools/ltdbtool.c',
415                      includes='include',
416                      deps='tdb',
417                      install_path='${BINDIR}',
418                      manpages='ltdbtool.1')
419
420     bld.SAMBA_BINARY('ctdb_lock_helper',
421                      source='server/ctdb_lock_helper.c',
422                      deps='samba-util ctdb-system talloc tdb',
423                      includes='include',
424                      install_path='${BINDIR}')
425
426     bld.SAMBA_BINARY('ctdb_event_helper',
427                      source='server/ctdb_event_helper.c',
428                      includes='include',
429                      deps='samba-util ctdb-system replace tdb',
430                      install_path='${BINDIR}')
431
432     bld.SAMBA_BINARY('ctdb_recovery_helper',
433                      source='server/ctdb_recovery_helper.c',
434                      deps='''ctdb-client2 ctdb-protocol ctdb-util
435                              samba-util replace tdb''',
436                      install_path='${BINDIR}')
437
438     bld.SAMBA_GENERATOR('ctdb-smnotify-h',
439                         source='utils/smnotify/smnotify.x',
440                         target='utils/smnotify/smnotify.h',
441                         rule='rpcgen -h ${SRC} > ${TGT}')
442
443     xdr_buf_hack = 'sed -e "s@^\([ \t]*register int32_t \*buf\);@\\1 = buf;@"'
444
445     bld.SAMBA_GENERATOR('ctdb-smnotify-x',
446                         source='utils/smnotify/smnotify.x',
447                         target='utils/smnotify/gen_xdr.c',
448                         rule='rpcgen -c ${SRC} | ' + xdr_buf_hack + ' > ${TGT}')
449
450     bld.SAMBA_GENERATOR('ctdb-smnotify-c',
451                         source='utils/smnotify/smnotify.x',
452                         target='utils/smnotify/gen_smnotify.c',
453                         rule='rpcgen -l ${SRC} > ${TGT}')
454
455     bld.SAMBA_BINARY('smnotify',
456                      source=bld.SUBDIR('utils/smnotify',
457                                        'smnotify.c gen_smnotify.c gen_xdr.c'),
458                      deps='ctdb-smnotify-h ctdb-smnotify-c ctdb-smnotify-x popt',
459                      includes='utils utils/smnotify',
460                      install_path='${BINDIR}')
461
462     bld.SAMBA_BINARY('ping_pong',
463                      source='utils/ping_pong/ping_pong.c',
464                      deps='',
465                      install_path='${BINDIR}',
466                      manpages='ping_pong.1')
467
468     if bld.env.HAVE_PMDA:
469         bld.SAMBA_BINARY('pmdactdb',
470                          source='utils/pmda/pmda_ctdb.c',
471                          includes='include',
472                          deps='ctdb-client ctdb-common pcp_pmda pcp',
473                          install_path='${CTDB_PMDADIR}')
474         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Install',
475                           destname='Install')
476         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Remove',
477                           destname='Remove')
478         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/pmns',
479                           destname='pmns')
480         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/domain.h',
481                           destname='domain.h')
482         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/help',
483                           destname='help')
484         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/README',
485                           destname='README')
486
487     sed_expr1 = 's|/usr/local/var/lib/ctdb|%s|g'  % (bld.env.CTDB_VARDIR)
488     sed_expr2 = 's|/usr/local/etc/ctdb|%s|g'      % (bld.env.CTDB_ETCDIR)
489     sed_expr3 = 's|/usr/local/var/log|%s|g'       % (bld.env.CTDB_LOGDIR)
490     sed_expr4 = 's|/usr/local/var/run/ctdb|%s|g'  % (bld.env.CTDB_RUNDIR)
491     sed_expr5 = 's|/usr/local/sbin|%s|g'          % (bld.env.SBINDIR)
492     sed_cmdline = '-e "%s" -e "%s" -e "%s" -e "%s" -e "%s"' % \
493                        (sed_expr1, sed_expr2, sed_expr3, sed_expr4, sed_expr5)
494
495     for f in manpages:
496         x = '%s.xml' % (f)
497         bld.SAMBA_GENERATOR(x,
498                             source=os.path.join('doc', x),
499                             target=x,
500                             rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
501
502     if bld.env.ctdb_generate_manpages:
503         bld.MANPAGES('''onnode.1 ctdbd_wrapper.1 ctdbd.conf.5
504                         ctdb.7 ctdb-statistics.7 ctdb-tunables.7''',
505                       True)
506     else:
507         for m in bld.env.ctdb_prebuilt_manpages:
508             bld.SAMBA_GENERATOR(m,
509                                 source=os.path.join("doc", m),
510                                 target=m,
511                                 rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
512             bld.INSTALL_FILES('${MANDIR}/man%s' % m[-1], m)
513
514     bld.SAMBA_GENERATOR('ctdb-onnode',
515                         source='tools/onnode',
516                         target='onnode',
517                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
518     bld.INSTALL_FILES('${BINDIR}', 'onnode',
519                       destname='onnode', chmod=0755)
520
521     bld.SAMBA_GENERATOR('ctdb-diagnostics',
522                         source='tools/ctdb_diagnostics',
523                         target='ctdb_diagnostics',
524                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
525     bld.INSTALL_FILES('${BINDIR}', 'ctdb_diagnostics',
526                       destname='ctdb_diagnostics', chmod=0755)
527
528     bld.SAMBA_GENERATOR('ctdbd-wrapper',
529                         source='config/ctdbd_wrapper',
530                         target='ctdbd_wrapper',
531                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
532     bld.INSTALL_FILES('${SBINDIR}', 'ctdbd_wrapper',
533                       destname='ctdbd_wrapper', chmod=0755)
534
535     def SUBDIR_MODE_callback(arg, dirname, fnames):
536         for f in fnames:
537             fl = os.path.join(dirname, f)
538             if os.path.isdir(fl) or os.path.islink(fl):
539                 continue
540             mode = os.lstat(fl).st_mode & 0777
541             if arg['trim_path']:
542                 fl = samba_utils.os_path_relpath(fl, arg['trim_path'])
543             arg['file_list'].append([fl, mode])
544
545     def SUBDIR_MODE(path, trim_path=None):
546         pd = {'trim_path': trim_path, 'file_list': []}
547         os.path.walk(path, SUBDIR_MODE_callback, pd)
548         return pd['file_list']
549
550     etc_subdirs = [
551         'events.d',
552         'nfs-checks.d'
553     ]
554
555     if bld.env.standalone_ctdb:
556         configdir = 'config'
557     else:
558         configdir = 'ctdb/config'
559
560     for t in etc_subdirs:
561         files = SUBDIR_MODE('%s/%s' % (configdir, t), trim_path=configdir)
562         for fmode in files:
563             bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % fmode[0],
564                               destname=fmode[0], chmod=fmode[1])
565
566     bld.SAMBA_GENERATOR('ctdb-functions',
567                         source='config/functions',
568                         target='functions',
569                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
570     bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'functions', destname='functions')
571
572     etc_scripts = [
573         'ctdb-crash-cleanup.sh',
574         'debug-hung-script.sh',
575         'debug_locks.sh',
576         'gcore_trace.sh',
577         'nfs-linux-kernel-callout',
578         'notify.sh',
579         'statd-callout'
580     ]
581
582     for t in etc_scripts:
583         bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % t,
584                           destname=t, chmod=0755)
585
586     bld.SAMBA_GENERATOR('ctdb-sudoers',
587                         source='config/ctdb.sudoers',
588                         target='ctdb.sudoers',
589                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
590     bld.INSTALL_FILES('${SYSCONFDIR}/sudoers.d', 'ctdb.sudoers',
591                       destname='ctdb')
592
593     bld.INSTALL_FILES('${CTDB_ETCDIR}/notify.d', 'config/notify.d.README',
594                       destname='README')
595
596     bld.install_dir(bld.env.CTDB_LOGDIR)
597     bld.install_dir(bld.env.CTDB_RUNDIR)
598     bld.install_dir(bld.env.CTDB_VARDIR)
599
600     sed_expr = 's/@PACKAGE_VERSION@/%s/g' % VERSION
601     t = bld.SAMBA_GENERATOR('ctdb-pc',
602                             source='ctdb.pc.in',
603                             target='ctdb.pc',
604                             rule='sed -e "%s" ${SRC} > ${TGT}' % sed_expr,
605                             dep_vars=['VERSION'])
606     t.env.VERSION = VERSION
607     bld.INSTALL_FILES('${LIBDIR}/pkgconfig', 'ctdb.pc')
608
609     # Unit tests
610     ctdb_unit_tests = [
611         'db_hash_test',
612         'srvid_test',
613         'pkt_read_test',
614         'pkt_write_test',
615         'comm_test',
616         'comm_server_test',
617         'comm_client_test',
618         'protocol_types_test',
619         'protocol_client_test',
620     ]
621
622     for target in ctdb_unit_tests:
623         src = 'tests/src/' + target + '.c'
624
625         bld.SAMBA_BINARY(target,
626                          source=src,
627                          deps='talloc tevent tdb tevent-unix-util',
628                          install_path='${CTDB_TEST_LIBDIR}')
629
630     bld.SAMBA_BINARY('reqid_test',
631                      source='tests/src/reqid_test.c',
632                      deps='samba-util',
633                      install_path='${CTDB_TEST_LIBDIR}')
634
635     # Test binaries
636     ctdb_tests = [
637         'rb_test',
638         'ctdb_bench',
639         'ctdb_fetch',
640         'ctdb_fetch_one',
641         'ctdb_fetch_readonly_once',
642         'ctdb_fetch_readonly_loop',
643         'ctdb_trackingdb_test',
644         'ctdb_update_record',
645         'ctdb_update_record_persistent',
646         'ctdb_store',
647         'ctdb_traverse',
648         'ctdb_randrec',
649         'ctdb_persistent',
650         'ctdb_porting_tests',
651         'ctdb_transaction',
652         'ctdb_lock_tdb'
653     ]
654
655     for target in ctdb_tests:
656         src = 'tests/src/' + target + '.c'
657
658         bld.SAMBA_BINARY(target,
659                          source=src,
660                          includes='include',
661                          deps='ctdb-client ctdb-common ctdb-util',
662                          install_path='${CTDB_TEST_LIBDIR}')
663
664     bld.SAMBA_BINARY('ctdb_takeover_tests',
665                      source='tests/src/ctdb_takeover_tests.c',
666                      deps='''replace popt tdb tevent talloc ctdb-system
667                              samba-util tdb-wrap talloc_report''' +
668                           ib_deps,
669                      includes='include',
670                      install_path='${CTDB_TEST_LIBDIR}')
671
672     bld.SAMBA_BINARY('ctdb_functest',
673                      source='tests/src/ctdb_functest.c',
674                      deps='''replace tdb tevent talloc popt ctdb-system
675                              samba-util tdb-wrap''',
676                      includes='include',
677                      install_path='${CTDB_TEST_LIBDIR}')
678
679     bld.SAMBA_BINARY('ctdb_stubtest',
680                      source='tests/src/ctdb_test.c',
681                      deps='''replace tdb tevent talloc popt ctdb-system
682                              samba-util tdb-wrap''',
683                      includes='include',
684                      install_path='${CTDB_TEST_LIBDIR}')
685
686     if bld.env.HAVE_INFINIBAND:
687         bld.SAMBA_BINARY('ibwrapper_test',
688                          source='ib/ibwrapper_test.c',
689                          includes='include',
690                          deps='replace talloc ctdb-client ctdb-common' +
691                               ib_deps,
692                          install_path='${CTDB_TEST_LIBDIR}')
693
694     test_subdirs = [
695         'complex',
696         'cunit',
697         'events.d',
698         'eventscripts',
699         'onnode',
700         'simple',
701         'takeover',
702         'tool'
703     ]
704
705     for t in test_subdirs:
706         files = SUBDIR_MODE('tests/%s' % t, trim_path='tests')
707         for fmode in files:
708             bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR, 'tests/%s' % fmode[0],
709                               destname=fmode[0], chmod=fmode[1])
710
711     # Install tests/scripts directory without test_wrap
712     test_scripts = [
713         'common.sh',
714         'integration.bash',
715         'unit.sh'
716     ]
717
718     for t in test_scripts:
719         bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR,
720                           os.path.join('tests/scripts', t),
721                           destname=os.path.join('scripts', t))
722
723     sed_expr = 's@^TEST_SCRIPTS_DIR=.*@&\\nexport TEST_BIN_DIR=\"%s\"@' % (
724                bld.env.CTDB_TEST_LIBDIR)
725     bld.SAMBA_GENERATOR('ctdb-test-wrap',
726                         source='tests/scripts/test_wrap',
727                         target='test_wrap',
728                         rule='sed -e "%s" ${SRC} > ${TGT}' % sed_expr)
729     bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR+"/scripts", 'test_wrap',
730                       destname='test_wrap', chmod=0755)
731
732     sed_expr1 = 's@^test_dir=.*@test_dir=%s\\nexport TEST_BIN_DIR=\"%s\"@' % (
733                 bld.env.CTDB_TEST_DATADIR, bld.env.CTDB_TEST_LIBDIR)
734     sed_expr2 = 's@^\(export CTDB_TESTS_ARE_INSTALLED\)=false@\\1=true@'
735     bld.SAMBA_GENERATOR('ctdb-test-runner',
736                         source='tests/run_tests.sh',
737                         target='ctdb_run_tests.sh',
738                         rule='sed -e "%s" -e "%s" ${SRC} > ${TGT}' % (
739                              sed_expr1, sed_expr2))
740     bld.INSTALL_FILES('${BINDIR}', 'ctdb_run_tests.sh',
741                       destname='ctdb_run_tests', chmod=0755)
742     bld.symlink_as(os.path.join(bld.env.BINDIR, 'ctdb_run_cluster_tests'),
743                    'ctdb_run_tests')
744
745     test_eventscript_links = [
746         'events.d',
747         'functions',
748         'nfs-checks.d',
749         'nfs-linux-kernel-callout',
750         'statd-callout'
751     ]
752
753     test_link_dir = os.path.join(bld.env.CTDB_TEST_DATADIR,
754                                  'eventscripts/etc-ctdb')
755     for t in test_eventscript_links:
756         bld.symlink_as(os.path.join(test_link_dir, t),
757                        os.path.join(bld.env.CTDB_ETCDIR, t))
758
759     # Tests that use onnode need to overwrite link to in-tree
760     # functions file when installed
761     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'onnode/functions'),
762                    os.path.join(bld.env.CTDB_ETCDIR, 'functions'))
763     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'simple/functions'),
764                    os.path.join(bld.env.CTDB_ETCDIR, 'functions'))
765
766     # Need a link to nodes file because $CTDB_BASE is overridden
767     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'simple/nodes'),
768                    os.path.join(bld.env.CTDB_ETCDIR, 'nodes'))
769
770
771 def testonly(ctx):
772     cmd = 'tests/run_tests.sh -V tests/var'
773     ret = samba_utils.RUN_COMMAND(cmd)
774     if ret != 0:
775         print('tests exited with exit status %d' % ret)
776         sys.exit(ret)
777
778
779 def test(ctx):
780     import Scripting
781     Scripting.commands.append('build')
782     Scripting.commands.append('testonly')
783
784
785 def autotest(ctx):
786     ld = 'LD_PRELOAD=%s/bin/shared/libsocket-wrapper.so' % os.getcwd()
787     cmd = '%s tests/run_tests.sh -e -S -C' % ld
788     ret = samba_utils.RUN_COMMAND(cmd)
789     if ret != 0:
790         print('autotest exited with exit status %d' % ret)
791         sys.exit(ret)
792
793
794 def show_version(ctx):
795     print VERSION
796
797
798 def dist():
799     samba_dist.DIST_FILES('VERSION:VERSION', extend=True)
800
801     t = 'include/ctdb_version.h'
802     cmd = 'packaging/mkversion.sh %s %s' % (t, VERSION)
803     ret = samba_utils.RUN_COMMAND(cmd)
804     if ret != 0:
805         print('Command "%s" failed with exit status %d' % (cmd, ret))
806         sys.exit(ret)
807     samba_dist.DIST_FILES('ctdb/%s:%s' % (t, t), extend=True)
808
809     t = 'ctdb.spec'
810     sed_expr1 = 's/@VERSION@/%s/g' % VERSION
811     sed_expr2 = 's/@RELEASE@/%s/g' % '1'
812     cmd = 'sed -e "%s" -e "%s" packaging/RPM/ctdb.spec.in > %s' % (
813         sed_expr1, sed_expr2, t)
814     ret = samba_utils.RUN_COMMAND(cmd)
815     if ret != 0:
816         print('Command "%s" failed with exit status %d' % (cmd, ret))
817         sys.exit(ret)
818     samba_dist.DIST_FILES('ctdb/%s:%s' % (t, t), extend=True)
819
820     cmd = 'make -C doc'
821     ret = samba_utils.RUN_COMMAND(cmd)
822     if ret != 0:
823         print('Command "%s" failed with exit status %d' % (cmd, ret))
824         sys.exit(ret)
825     for t in manpages:
826         samba_dist.DIST_FILES('ctdb/doc/%s:doc/%s' % (t, t), extend=True)
827         samba_dist.DIST_FILES('ctdb/doc/%s.html:doc/%s.html' % (t, t),
828                               extend=True)
829
830     samba_dist.dist()
831
832
833 def rpmonly(ctx):
834     opts = os.getenv('RPM_OPTIONS') or ''
835     cmd = 'rpmbuild -ta --clean --rmsource %s ctdb-%s.tar.gz' % (opts, VERSION)
836     ret = samba_utils.RUN_COMMAND(cmd)
837     if ret != 0:
838         print('rpmbuild exited with exit status %d' % ret)
839         sys.exit(ret)
840
841
842 def rpm(ctx):
843     import Scripting
844     Scripting.commands.append('dist')
845     Scripting.commands.append('rpmonly')
846
847
848 def ctags(ctx):
849     "build 'tags' file using ctags"
850     import Utils
851     source_root = os.path.dirname(Utils.g_module.root_path)
852     cmd = 'ctags $(find %s -name "*.[ch]")' % source_root
853     print("Running: %s" % cmd)
854     ret = samba_utils.RUN_COMMAND(cmd)
855     if ret != 0:
856         print('ctags failed with exit status %d' % ret)
857         sys.exit(ret)