ctdb-recovery: Include lib/util/time.h instead of samba_util.h
[samba.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     conf.env.CTDB_HELPER_BINDIR = os.path.join(conf.env.LIBEXECDIR, 'ctdb')
185
186     if Options.options.ctdb_logdir:
187         conf.env.CTDB_LOGDIR = Options.options.ctdb_logdir
188     else:
189         conf.env.CTDB_LOGDIR = os.path.join(conf.env.LOCALSTATEDIR, 'log')
190
191     if Options.options.ctdb_sockpath:
192         conf.env.CTDB_SOCKPATH = Options.options.ctdb_sockpath
193     else:
194         conf.env.CTDB_SOCKPATH = os.path.join(conf.env.CTDB_RUNDIR,
195                                               'ctdbd.socket')
196     conf.define('CTDB_SOCKET', conf.env.CTDB_SOCKPATH)
197
198     conf.ADD_CFLAGS('''-DCTDB_HELPER_BINDIR=\"%s\"
199                        -DLOGDIR=\"%s\"
200                        -DCTDB_ETCDIR=\"%s\"
201                        -DCTDB_VARDIR=\"%s\"
202                        -DCTDB_RUNDIR=\"%s\"''' % (
203                     conf.env.CTDB_HELPER_BINDIR,
204                     conf.env.CTDB_LOGDIR,
205                     conf.env.CTDB_ETCDIR,
206                     conf.env.CTDB_VARDIR,
207                     conf.env.CTDB_RUNDIR))
208
209     conf.env.CTDB_TEST_DATADIR = os.path.join(conf.env.EXEC_PREFIX,
210                                               'share/ctdb-tests')
211     conf.env.CTDB_TEST_LIBDIR = os.path.join(conf.env.LIBDIR, 'ctdb-tests')
212
213     # Allow unified compilation and separate compilation of utilities
214     # to find includes
215     if not conf.env.standalone_ctdb:
216         conf.ADD_EXTRA_INCLUDES('#include/public #ctdb/include #ctdb')
217     else:
218         if srcdir == '.':
219             # Building from tarball
220             conf.ADD_EXTRA_INCLUDES('#include')
221         else:
222             # Building standalone CTDB from within Samba tree
223             conf.ADD_EXTRA_INCLUDES('#ctdb/include')
224             conf.ADD_EXTRA_INCLUDES('#ctdb')
225         conf.ADD_EXTRA_INCLUDES('#lib #lib/replace')
226
227         conf.DEFINE('HAVE_CONFIG_H', 1, add_to_cflags=True)
228         conf.DEFINE('SAMBA_UTIL_CORE_ONLY', 1, add_to_cflags=True)
229         conf.SAMBA_CONFIG_H()
230
231     if 'XSLTPROC_MANPAGES' in conf.env and conf.env['XSLTPROC_MANPAGES']:
232         conf.env.ctdb_generate_manpages = True
233     else:
234         conf.env.ctdb_generate_manpages = False
235
236         Logs.info("xsltproc unavailable, checking for pre-built manpages")
237         conf.env.ctdb_prebuilt_manpages = []
238         for m in manpages:
239             if os.path.exists(os.path.join("doc", m)):
240                 Logs.info("  %s: yes" % (m))
241                 conf.env.ctdb_prebuilt_manpages.append(m)
242             else:
243                 Logs.info("  %s: no" % (m))
244
245 def build(bld):
246     if bld.env.standalone_ctdb:
247         # enable building of public headers in the build tree
248         bld.env.build_public_headers = 'include/public'
249
250     version_h = samba_utils.os_path_relpath(os.path.join(Options.launch_dir,
251                                                          "version.h"),
252                                             bld.curdir)
253
254     if bld.env.standalone_ctdb:
255         ctdb_mkversion = '../packaging/mkversion.sh'
256         t = bld.SAMBA_GENERATOR('ctdb-version-header',
257                                 target='include/ctdb_version.h',
258                                 rule='%s ${TGT} %s' % (ctdb_mkversion, VERSION),
259                                 dep_vars=['VERSION'])
260         t.env.VERSION = VERSION
261
262         t = bld.SAMBA_GENERATOR('ctdb-samba-version-header',
263                                 target=version_h,
264                                 rule='printf "#include \\"ctdb_version.h\\" \\n#define SAMBA_VERSION_STRING CTDB_VERSION_STRING\\n" > ${TGT}',
265                                 dep_vars=['VERSION'])
266         t.env.VERSION = VERSION
267     else:
268         t = bld.SAMBA_GENERATOR('ctdb-samba-version-header',
269                                 target='include/ctdb_version.h',
270                                 rule='printf "#include \\"%s\\" \\n#define CTDB_VERSION_STRING SAMBA_VERSION_STRING\\n" > ${TGT}' % version_h,
271                                 dep_vars=['VERSION'])
272         t.env.VERSION = VERSION
273
274     bld.env.PKGCONFIGDIR = '${LIBDIR}/pkgconfig'
275
276     bld.RECURSE('lib/replace')
277     if bld.CHECK_FOR_THIRD_PARTY():
278         bld.RECURSE('third_party/popt')
279
280     bld.RECURSE('lib/tdb_wrap')
281     bld.RECURSE('lib/util')
282
283     bld.RECURSE('lib/talloc')
284     bld.RECURSE('lib/tevent')
285     bld.RECURSE('lib/tdb')
286     if bld.env.standalone_ctdb or bld.CONFIG_GET('SOCKET_WRAPPER'):
287         bld.RECURSE('lib/socket_wrapper')
288
289     if bld.env.standalone_ctdb:
290         # If a combined build is implemented, CTDB will want to
291         # build against samba-util rather than samba-util-core.
292         # Similarly, other Samba subsystems expect samba-util.  So,
293         # for a standalone build, just define a fake samba-util
294         # subsystem that pulls in samba-util-core.
295         bld.SAMBA_SUBSYSTEM('samba-util',
296                             source='',
297                             deps='samba-util-core')
298
299     bld.SAMBA_SUBSYSTEM('ctdb-tcp',
300                         source=bld.SUBDIR('tcp',
301                                           'tcp_connect.c tcp_init.c tcp_io.c'),
302                         includes='include',
303                         deps='replace tdb talloc tevent')
304
305     ib_deps = ''
306     if bld.env.HAVE_INFINIBAND:
307         bld.SAMBA_SUBSYSTEM('ctdb-ib',
308                             source=bld.SUBDIR('ib',
309                                               '''ibwrapper.c ibw_ctdb.c
310                                                  ibw_ctdb_init.c'''),
311                             includes='include',
312                             deps='replace talloc tevent tdb')
313         ib_deps = ' ctdb-ib rdmacm ibverbs'
314
315     if sys.platform.startswith('linux'):
316         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_linux.c')
317     elif sys.platform.startswith('aix'):
318         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_aix.c')
319     elif sys.platform.startswith('freebsd'):
320         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_freebsd.c')
321     elif sys.platform == 'kfreebsd':
322         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_kfreebsd.c')
323     elif sys.platform == 'gnu':
324         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_gnu.c')
325     else:
326         Logs.error("Platform %s not supported" % sys.platform)
327
328     bld.SAMBA_SUBSYSTEM('ctdb-system',
329                         source=bld.SUBDIR('common',
330                                           'system_common.c system_util.c') +
331                                CTDB_SYSTEM_SRC,
332                         includes='include',
333                         deps='replace talloc tevent tdb pcap')
334
335     bld.SAMBA_SUBSYSTEM('ctdb-common',
336                         source=bld.SUBDIR('common',
337                                           '''ctdb_io.c ctdb_util.c ctdb_ltdb.c
338                                              cmdline.c rb_tree.c'''),
339                         includes='include',
340                         deps='replace popt talloc tevent tdb popt ctdb-system')
341
342     bld.SAMBA_SUBSYSTEM('ctdb-util',
343                         source=bld.SUBDIR('common',
344                                           '''db_hash.c srvid.c reqid.c
345                                              pkt_read.c pkt_write.c comm.c
346                                              logging.c'''),
347                         deps='replace talloc tevent tdb tevent-unix-util')
348
349     bld.SAMBA_SUBSYSTEM('ctdb-protocol',
350                         source=bld.SUBDIR('protocol',
351                                           '''protocol_header.c protocol_packet.c
352                                              protocol_types.c protocol_call.c
353                                              protocol_message.c
354                                              protocol_control.c
355                                              protocol_client.c
356                                              protocol_util.c'''),
357                         includes='include',
358                         deps='replace talloc tdb')
359
360     bld.SAMBA_SUBSYSTEM('ctdb-client',
361                         source=bld.SUBDIR('client', 'ctdb_client.c'),
362                         includes='include',
363                         deps='''replace popt talloc tevent tdb
364                                 samba-util tdb-wrap ctdb-util''')
365
366     bld.SAMBA_SUBSYSTEM('ctdb-client2',
367                         source=bld.SUBDIR('client',
368                                           '''client_connect.c client_call.c
369                                              client_message.c client_control.c
370                                              client_message_sync.c
371                                              client_control_sync.c
372                                              client_db.c client_util.c
373                                           '''),
374                         includes='include',
375                         deps='replace talloc tevent tdb tdb-wrap')
376
377     bld.SAMBA_SUBSYSTEM('ctdb-ipalloc',
378                         source=bld.SUBDIR('server',
379                                           '''ipalloc_deterministic.c
380                                              ipalloc_nondeterministic.c
381                                              ipalloc_lcp2.c
382                                              ipalloc_common.c
383                                              ipalloc.c
384                                           '''),
385                         includes='include',
386                         deps='ctdb-protocol replace talloc tevent')
387
388     bld.SAMBA_SUBSYSTEM('ctdb-server',
389                         source='server/ctdbd.c ' +
390                                bld.SUBDIR('server',
391                                           '''ctdb_daemon.c ctdb_recoverd.c
392                                              ctdb_recover.c ctdb_freeze.c
393                                              ctdb_tunables.c ctdb_monitor.c
394                                              ctdb_server.c ctdb_control.c
395                                              ctdb_call.c ctdb_ltdb_server.c
396                                              ctdb_traverse.c eventscript.c
397                                              ctdb_takeover.c ctdb_serverids.c
398                                              ctdb_persistent.c ctdb_keepalive.c
399                                              ctdb_logging.c
400                                              ctdb_logging_syslog.c
401                                              ctdb_logging_file.c
402                                              ctdb_uptime.c
403                                              ctdb_vacuum.c ctdb_banning.c
404                                              ctdb_statistics.c
405                                              ctdb_update_record.c
406                                              ctdb_lock.c ctdb_fork.c'''),
407                         includes='include',
408                         deps='ctdb-ipalloc replace popt talloc tevent tdb talloc_report')
409
410     bld.SAMBA_BINARY('ctdbd',
411                      source='',
412                      deps='''ctdb-server ctdb-client ctdb-common
413                              ctdb-system ctdb-tcp ctdb-util''' +
414                           ib_deps,
415                      install_path='${SBINDIR}',
416                      manpages='ctdbd.1')
417
418     bld.SAMBA_BINARY('ctdb',
419                      source='tools/ctdb.c',
420                      deps='ctdb-client ctdb-common ctdb-system ctdb-util',
421                      includes='include',
422                      install_path='${BINDIR}',
423                      manpages='ctdb.1')
424
425     bld.SAMBA_BINARY('ltdbtool',
426                      source='tools/ltdbtool.c',
427                      includes='include',
428                      deps='tdb',
429                      install_path='${BINDIR}',
430                      manpages='ltdbtool.1')
431
432     bld.SAMBA_BINARY('ctdb_lock_helper',
433                      source='server/ctdb_lock_helper.c',
434                      deps='samba-util ctdb-system talloc tdb',
435                      includes='include',
436                      install_path='${CTDB_HELPER_BINDIR}')
437
438     bld.SAMBA_BINARY('ctdb_event_helper',
439                      source='server/ctdb_event_helper.c',
440                      includes='include',
441                      deps='samba-util ctdb-system replace tdb',
442                      install_path='${CTDB_HELPER_BINDIR}')
443
444     bld.SAMBA_BINARY('ctdb_recovery_helper',
445                      source='server/ctdb_recovery_helper.c',
446                      deps='''ctdb-client2 ctdb-protocol ctdb-util
447                              samba-util replace tdb''',
448                      install_path='${CTDB_HELPER_BINDIR}')
449
450     bld.SAMBA_GENERATOR('ctdb-smnotify-h',
451                         source='utils/smnotify/smnotify.x',
452                         target='utils/smnotify/smnotify.h',
453                         rule='rpcgen -h ${SRC} > ${TGT}')
454
455     xdr_buf_hack = 'sed -e "s@^\([ \t]*register int32_t \*buf\);@\\1 = buf;@"'
456
457     bld.SAMBA_GENERATOR('ctdb-smnotify-x',
458                         source='utils/smnotify/smnotify.x',
459                         target='utils/smnotify/gen_xdr.c',
460                         rule='rpcgen -c ${SRC} | ' + xdr_buf_hack + ' > ${TGT}')
461
462     bld.SAMBA_GENERATOR('ctdb-smnotify-c',
463                         source='utils/smnotify/smnotify.x',
464                         target='utils/smnotify/gen_smnotify.c',
465                         rule='rpcgen -l ${SRC} > ${TGT}')
466
467     bld.SAMBA_BINARY('smnotify',
468                      source=bld.SUBDIR('utils/smnotify',
469                                        'smnotify.c gen_smnotify.c gen_xdr.c'),
470                      deps='ctdb-smnotify-h ctdb-smnotify-c ctdb-smnotify-x popt',
471                      includes='utils utils/smnotify',
472                      install_path='${CTDB_HELPER_BINDIR}')
473
474     bld.SAMBA_BINARY('ping_pong',
475                      source='utils/ping_pong/ping_pong.c',
476                      deps='',
477                      install_path='${BINDIR}',
478                      manpages='ping_pong.1')
479
480     if bld.env.HAVE_PMDA:
481         bld.SAMBA_BINARY('pmdactdb',
482                          source='utils/pmda/pmda_ctdb.c',
483                          includes='include',
484                          deps='ctdb-client ctdb-common pcp_pmda pcp',
485                          install_path='${CTDB_PMDADIR}')
486         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Install',
487                           destname='Install')
488         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Remove',
489                           destname='Remove')
490         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/pmns',
491                           destname='pmns')
492         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/domain.h',
493                           destname='domain.h')
494         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/help',
495                           destname='help')
496         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/README',
497                           destname='README')
498
499     sed_expr1 = 's|/usr/local/var/lib/ctdb|%s|g'  % (bld.env.CTDB_VARDIR)
500     sed_expr2 = 's|/usr/local/etc/ctdb|%s|g'      % (bld.env.CTDB_ETCDIR)
501     sed_expr3 = 's|/usr/local/var/log|%s|g'       % (bld.env.CTDB_LOGDIR)
502     sed_expr4 = 's|/usr/local/var/run/ctdb|%s|g'  % (bld.env.CTDB_RUNDIR)
503     sed_expr5 = 's|/usr/local/sbin|%s|g'          % (bld.env.SBINDIR)
504     sed_expr6 = 's|/usr/local/libexec/ctdb|%s|g'  % (bld.env.CTDB_HELPER_BINDIR)
505     sed_cmdline = '-e "%s" -e "%s" -e "%s" -e "%s" -e "%s" -e "%s"' % \
506                        (sed_expr1, sed_expr2, sed_expr3, sed_expr4, sed_expr5,
507                         sed_expr6)
508
509     for f in manpages:
510         x = '%s.xml' % (f)
511         bld.SAMBA_GENERATOR(x,
512                             source=os.path.join('doc', x),
513                             target=x,
514                             rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
515
516     if bld.env.ctdb_generate_manpages:
517         bld.MANPAGES('''onnode.1 ctdbd_wrapper.1 ctdbd.conf.5
518                         ctdb.7 ctdb-statistics.7 ctdb-tunables.7''',
519                       True)
520     else:
521         for m in bld.env.ctdb_prebuilt_manpages:
522             bld.SAMBA_GENERATOR(m,
523                                 source=os.path.join("doc", m),
524                                 target=m,
525                                 rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
526             bld.INSTALL_FILES('${MANDIR}/man%s' % m[-1], m)
527
528     bld.SAMBA_GENERATOR('ctdb-onnode',
529                         source='tools/onnode',
530                         target='onnode',
531                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
532     bld.INSTALL_FILES('${BINDIR}', 'onnode',
533                       destname='onnode', chmod=0755)
534
535     bld.SAMBA_GENERATOR('ctdb-diagnostics',
536                         source='tools/ctdb_diagnostics',
537                         target='ctdb_diagnostics',
538                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
539     bld.INSTALL_FILES('${BINDIR}', 'ctdb_diagnostics',
540                       destname='ctdb_diagnostics', chmod=0755)
541
542     bld.SAMBA_GENERATOR('ctdb-natgw',
543                         source='tools/ctdb_natgw',
544                         target='ctdb_natgw',
545                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
546     bld.INSTALL_FILES('${CTDB_HELPER_BINDIR}', 'ctdb_natgw',
547                       destname='ctdb_natgw', chmod=0755)
548
549     bld.SAMBA_GENERATOR('ctdbd-wrapper',
550                         source='config/ctdbd_wrapper',
551                         target='ctdbd_wrapper',
552                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
553     bld.INSTALL_FILES('${SBINDIR}', 'ctdbd_wrapper',
554                       destname='ctdbd_wrapper', chmod=0755)
555
556     def SUBDIR_MODE_callback(arg, dirname, fnames):
557         for f in fnames:
558             fl = os.path.join(dirname, f)
559             if os.path.isdir(fl) or os.path.islink(fl):
560                 continue
561             mode = os.lstat(fl).st_mode & 0777
562             if arg['trim_path']:
563                 fl = samba_utils.os_path_relpath(fl, arg['trim_path'])
564             arg['file_list'].append([fl, mode])
565
566     def SUBDIR_MODE(path, trim_path=None):
567         pd = {'trim_path': trim_path, 'file_list': []}
568         os.path.walk(path, SUBDIR_MODE_callback, pd)
569         return pd['file_list']
570
571     etc_subdirs = [
572         'events.d',
573         'nfs-checks.d'
574     ]
575
576     if bld.env.standalone_ctdb:
577         configdir = 'config'
578     else:
579         configdir = 'ctdb/config'
580
581     for t in etc_subdirs:
582         files = SUBDIR_MODE('%s/%s' % (configdir, t), trim_path=configdir)
583         for fmode in files:
584             bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % fmode[0],
585                               destname=fmode[0], chmod=fmode[1])
586
587     bld.SAMBA_GENERATOR('ctdb-functions',
588                         source='config/functions',
589                         target='functions',
590                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
591     bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'functions', destname='functions')
592
593     etc_scripts = [
594         'ctdb-crash-cleanup.sh',
595         'debug-hung-script.sh',
596         'debug_locks.sh',
597         'gcore_trace.sh',
598         'nfs-linux-kernel-callout',
599         'notify.sh',
600         'statd-callout'
601     ]
602
603     for t in etc_scripts:
604         bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % t,
605                           destname=t, chmod=0755)
606
607     bld.SAMBA_GENERATOR('ctdb-sudoers',
608                         source='config/ctdb.sudoers',
609                         target='ctdb.sudoers',
610                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
611     bld.INSTALL_FILES('${SYSCONFDIR}/sudoers.d', 'ctdb.sudoers',
612                       destname='ctdb')
613
614     bld.INSTALL_FILES('${CTDB_ETCDIR}/notify.d', 'config/notify.d.README',
615                       destname='README')
616
617     bld.install_dir(bld.env.CTDB_LOGDIR)
618     bld.install_dir(bld.env.CTDB_RUNDIR)
619     bld.install_dir(bld.env.CTDB_VARDIR)
620
621     # Unit tests
622     ctdb_unit_tests = [
623         'db_hash_test',
624         'srvid_test',
625         'pkt_read_test',
626         'pkt_write_test',
627         'comm_test',
628         'comm_server_test',
629         'comm_client_test',
630         'protocol_types_test',
631         'protocol_client_test',
632     ]
633
634     for target in ctdb_unit_tests:
635         src = 'tests/src/' + target + '.c'
636
637         bld.SAMBA_BINARY(target,
638                          source=src,
639                          deps='talloc tevent tdb tevent-unix-util',
640                          install_path='${CTDB_TEST_LIBDIR}')
641
642     bld.SAMBA_BINARY('reqid_test',
643                      source='tests/src/reqid_test.c',
644                      deps='samba-util',
645                      install_path='${CTDB_TEST_LIBDIR}')
646
647     # Test binaries
648     ctdb_tests = [
649         'rb_test',
650         'ctdb_bench',
651         'ctdb_fetch',
652         'ctdb_fetch_one',
653         'ctdb_fetch_readonly_once',
654         'ctdb_fetch_readonly_loop',
655         'ctdb_trackingdb_test',
656         'ctdb_update_record',
657         'ctdb_update_record_persistent',
658         'ctdb_store',
659         'ctdb_traverse',
660         'ctdb_randrec',
661         'ctdb_persistent',
662         'ctdb_porting_tests',
663         'ctdb_transaction',
664         'ctdb_lock_tdb'
665     ]
666
667     for target in ctdb_tests:
668         src = 'tests/src/' + target + '.c'
669
670         bld.SAMBA_BINARY(target,
671                          source=src,
672                          includes='include',
673                          deps='ctdb-client ctdb-common ctdb-util',
674                          install_path='${CTDB_TEST_LIBDIR}')
675
676     bld.SAMBA_BINARY('ctdb_takeover_tests',
677                      source='tests/src/ctdb_takeover_tests.c',
678                      deps='''replace popt tdb tevent talloc ctdb-system
679                              samba-util tdb-wrap talloc_report
680                              ctdb-protocol''' +
681                           ib_deps,
682                      includes='include',
683                      install_path='${CTDB_TEST_LIBDIR}')
684
685     bld.SAMBA_BINARY('ctdb_functest',
686                      source='tests/src/ctdb_functest.c',
687                      deps='''replace tdb tevent talloc popt ctdb-system
688                              samba-util tdb-wrap''',
689                      includes='include',
690                      install_path='${CTDB_TEST_LIBDIR}')
691
692     bld.SAMBA_BINARY('ctdb_stubtest',
693                      source='tests/src/ctdb_test.c',
694                      deps='''replace tdb tevent talloc popt ctdb-system
695                              samba-util tdb-wrap''',
696                      includes='include',
697                      install_path='${CTDB_TEST_LIBDIR}')
698
699     if bld.env.HAVE_INFINIBAND:
700         bld.SAMBA_BINARY('ibwrapper_test',
701                          source='ib/ibwrapper_test.c',
702                          includes='include',
703                          deps='replace talloc ctdb-client ctdb-common' +
704                               ib_deps,
705                          install_path='${CTDB_TEST_LIBDIR}')
706
707     test_subdirs = [
708         'complex',
709         'cunit',
710         'events.d',
711         'eventscripts',
712         'onnode',
713         'simple',
714         'takeover',
715         'tool'
716     ]
717
718     for t in test_subdirs:
719         files = SUBDIR_MODE('tests/%s' % t, trim_path='tests')
720         for fmode in files:
721             bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR, 'tests/%s' % fmode[0],
722                               destname=fmode[0], chmod=fmode[1])
723
724     # Install tests/scripts directory without test_wrap
725     test_scripts = [
726         'common.sh',
727         'integration.bash',
728         'unit.sh'
729     ]
730
731     for t in test_scripts:
732         bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR,
733                           os.path.join('tests/scripts', t),
734                           destname=os.path.join('scripts', t))
735
736     sed_expr = 's@^TEST_SCRIPTS_DIR=.*@&\\nexport TEST_BIN_DIR=\"%s\"@' % (
737                bld.env.CTDB_TEST_LIBDIR)
738     bld.SAMBA_GENERATOR('ctdb-test-wrap',
739                         source='tests/scripts/test_wrap',
740                         target='test_wrap',
741                         rule='sed -e "%s" ${SRC} > ${TGT}' % sed_expr)
742     bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR+"/scripts", 'test_wrap',
743                       destname='test_wrap', chmod=0755)
744
745     sed_expr1 = 's@^test_dir=.*@test_dir=%s\\nexport TEST_BIN_DIR=\"%s\"@' % (
746                 bld.env.CTDB_TEST_DATADIR, bld.env.CTDB_TEST_LIBDIR)
747     sed_expr2 = 's@^\(export CTDB_TESTS_ARE_INSTALLED\)=false@\\1=true@'
748     bld.SAMBA_GENERATOR('ctdb-test-runner',
749                         source='tests/run_tests.sh',
750                         target='ctdb_run_tests.sh',
751                         rule='sed -e "%s" -e "%s" ${SRC} > ${TGT}' % (
752                              sed_expr1, sed_expr2))
753     bld.INSTALL_FILES('${BINDIR}', 'ctdb_run_tests.sh',
754                       destname='ctdb_run_tests', chmod=0755)
755     bld.symlink_as(os.path.join(bld.env.BINDIR, 'ctdb_run_cluster_tests'),
756                    'ctdb_run_tests')
757
758     test_eventscript_links = [
759         'events.d',
760         'functions',
761         'nfs-checks.d',
762         'nfs-linux-kernel-callout',
763         'statd-callout'
764     ]
765
766     test_link_dir = os.path.join(bld.env.CTDB_TEST_DATADIR,
767                                  'eventscripts/etc-ctdb')
768     for t in test_eventscript_links:
769         bld.symlink_as(os.path.join(test_link_dir, t),
770                        os.path.join(bld.env.CTDB_ETCDIR, t))
771
772     # Tests that use onnode need to overwrite link to in-tree
773     # functions file when installed
774     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'onnode/functions'),
775                    os.path.join(bld.env.CTDB_ETCDIR, 'functions'))
776     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'simple/functions'),
777                    os.path.join(bld.env.CTDB_ETCDIR, 'functions'))
778
779     # Need a link to nodes file because $CTDB_BASE is overridden
780     bld.symlink_as(os.path.join(bld.env.CTDB_TEST_DATADIR, 'simple/nodes'),
781                    os.path.join(bld.env.CTDB_ETCDIR, 'nodes'))
782
783
784 def testonly(ctx):
785     cmd = 'tests/run_tests.sh -V tests/var'
786     ret = samba_utils.RUN_COMMAND(cmd)
787     if ret != 0:
788         print('tests exited with exit status %d' % ret)
789         sys.exit(ret)
790
791
792 def test(ctx):
793     import Scripting
794     Scripting.commands.append('build')
795     Scripting.commands.append('testonly')
796
797
798 def autotest(ctx):
799     ld = 'LD_PRELOAD=%s/bin/shared/libsocket-wrapper.so' % os.getcwd()
800     cmd = '%s tests/run_tests.sh -e -S -C' % ld
801     ret = samba_utils.RUN_COMMAND(cmd)
802     if ret != 0:
803         print('autotest exited with exit status %d' % ret)
804         sys.exit(ret)
805
806
807 def show_version(ctx):
808     print VERSION
809
810
811 def dist():
812     samba_dist.DIST_FILES('VERSION:VERSION', extend=True)
813
814     t = 'include/ctdb_version.h'
815     cmd = 'packaging/mkversion.sh %s %s' % (t, VERSION)
816     ret = samba_utils.RUN_COMMAND(cmd)
817     if ret != 0:
818         print('Command "%s" failed with exit status %d' % (cmd, ret))
819         sys.exit(ret)
820     samba_dist.DIST_FILES('ctdb/%s:%s' % (t, t), extend=True)
821
822     t = 'ctdb.spec'
823     sed_expr1 = 's/@VERSION@/%s/g' % VERSION
824     sed_expr2 = 's/@RELEASE@/%s/g' % '1'
825     cmd = 'sed -e "%s" -e "%s" packaging/RPM/ctdb.spec.in > %s' % (
826         sed_expr1, sed_expr2, t)
827     ret = samba_utils.RUN_COMMAND(cmd)
828     if ret != 0:
829         print('Command "%s" failed with exit status %d' % (cmd, ret))
830         sys.exit(ret)
831     samba_dist.DIST_FILES('ctdb/%s:%s' % (t, t), extend=True)
832
833     cmd = 'make -C doc'
834     ret = samba_utils.RUN_COMMAND(cmd)
835     if ret != 0:
836         print('Command "%s" failed with exit status %d' % (cmd, ret))
837         sys.exit(ret)
838     for t in manpages:
839         samba_dist.DIST_FILES('ctdb/doc/%s:doc/%s' % (t, t), extend=True)
840         samba_dist.DIST_FILES('ctdb/doc/%s.html:doc/%s.html' % (t, t),
841                               extend=True)
842
843     samba_dist.dist()
844
845
846 def rpmonly(ctx):
847     opts = os.getenv('RPM_OPTIONS') or ''
848     cmd = 'rpmbuild -ta --clean --rmsource %s ctdb-%s.tar.gz' % (opts, VERSION)
849     ret = samba_utils.RUN_COMMAND(cmd)
850     if ret != 0:
851         print('rpmbuild exited with exit status %d' % ret)
852         sys.exit(ret)
853
854
855 def rpm(ctx):
856     import Scripting
857     Scripting.commands.append('dist')
858     Scripting.commands.append('rpmonly')
859
860
861 def ctags(ctx):
862     "build 'tags' file using ctags"
863     import Utils
864     source_root = os.path.dirname(Utils.g_module.root_path)
865     cmd = 'ctags $(find %s -name "*.[ch]")' % source_root
866     print("Running: %s" % cmd)
867     ret = samba_utils.RUN_COMMAND(cmd)
868     if ret != 0:
869         print('ctags failed with exit status %d' % ret)
870         sys.exit(ret)