ctdb-common: Add command line processing abstraction
[metze/samba/wip.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                         third_party/socket_wrapper:third_party/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                         lib/async_req:lib/async_req
38                         buildtools:buildtools third_party/waf:third_party/waf''')
39
40 manpages_binary = [
41     'ctdb.1',
42     'ctdbd.1',
43     'ltdbtool.1',
44     'ping_pong.1'
45 ]
46
47 manpages_misc = [
48     'ctdb_diagnostics.1',
49     'ctdbd_wrapper.1',
50     'onnode.1',
51     'ctdb-script.options.5',
52     'ctdb.sysconfig.5',
53     'ctdbd.conf.5',
54     'ctdb.7',
55     'ctdb-statistics.7',
56     'ctdb-tunables.7',
57 ]
58
59 manpages_etcd = [
60     'ctdb-etcd.7',
61 ]
62
63 manpages_ceph = [
64     'ctdb_mutex_ceph_rados_helper.7',
65 ]
66
67
68 def set_options(opt):
69     opt.PRIVATE_EXTENSION_DEFAULT('ctdb')
70
71     opt.RECURSE('lib/replace')
72
73     opt.RECURSE('lib/util')
74
75     opt.RECURSE('lib/talloc')
76     opt.RECURSE('lib/tevent')
77     opt.RECURSE('lib/tdb')
78
79     opt.add_option('--enable-infiniband',
80                    help=("Turn on infiniband support (default=no)"),
81                    action="store_true", dest='ctdb_infiniband', default=False)
82     opt.add_option('--enable-pmda',
83                    help=("Turn on PCP pmda support (default=no)"),
84                    action="store_true", dest='ctdb_pmda', default=False)
85     opt.add_option('--enable-etcd-reclock',
86                    help=("Enable etcd recovery lock helper (default=no)"),
87                    action="store_true", dest='ctdb_etcd_reclock', default=False)
88     opt.add_option('--enable-ceph-reclock',
89                    help=("Enable Ceph CTDB recovery lock helper (default=no)"),
90                    action="store_true", dest='ctdb_ceph_reclock', default=False)
91
92     opt.add_option('--with-logdir',
93                    help=("Path to log directory"),
94                    action="store", dest='ctdb_logdir', default=None)
95     opt.add_option('--with-socketpath',
96                    help=("path to CTDB daemon socket"),
97                    action="store", dest='ctdb_sockpath', default=None)
98
99
100 def configure(conf):
101
102     # No need to build python bindings for talloc/tevent/tdb
103     if conf.IN_LAUNCH_DIR():
104         conf.env.standalone_ctdb = True
105         Options.options.disable_python = True
106
107     conf.RECURSE('lib/replace')
108
109     if conf.env.standalone_ctdb:
110         conf.SAMBA_CHECK_PERL(mandatory=True)
111
112         conf.SAMBA_CHECK_PYTHON(mandatory=False, version=(2,5,0))
113         conf.SAMBA_CHECK_PYTHON_HEADERS(mandatory=False)
114
115     if conf.CHECK_FOR_THIRD_PARTY():
116         conf.RECURSE('third_party/popt')
117         if conf.env.standalone_ctdb or conf.CONFIG_GET('ENABLE_SELFTEST'):
118             conf.RECURSE('third_party/socket_wrapper')
119             conf.env.SOCKET_WRAPPER_SO_PATH = conf.CONFIG_GET('LIBSOCKET_WRAPPER_SO_PATH')
120     else:
121         if not conf.CHECK_POPT():
122             raise Utils.WafError('popt development packages have not been found\nIf third_party is installed, check that it is in the proper place.')
123         else:
124             conf.define('USING_SYSTEM_POPT', 1)
125         conf.env.SOCKET_WRAPPER_SO_PATH = ''
126
127
128         if conf.env.standalone_ctdb or conf.CONFIG_GET('ENABLE_SELFTEST'):
129             if not conf.CHECK_SOCKET_WRAPPER():
130                 raise Utils.WafError('socket_wrapper package has not been found.\nIf third_party is installed, check that it is in the proper place.')
131             else:
132                 conf.define('USING_SYSTEM_SOCKET_WRAPPER', 1)
133                 conf.env.SOCKET_WRAPPER_SO_PATH = conf.CONFIG_GET('LIBSOCKET_WRAPPER_SO_PATH')
134
135     conf.RECURSE('lib/util')
136
137     conf.RECURSE('lib/talloc')
138     conf.RECURSE('lib/tevent')
139     conf.RECURSE('lib/tdb')
140
141     conf.CHECK_HEADERS('sched.h')
142     conf.CHECK_HEADERS('procinfo.h')
143     if sys.platform.startswith('aix') and not conf.CHECK_FUNCS('thread_setsched'):
144         Logs.error('Need thread_setsched() on AIX')
145         sys.exit(1)
146     elif not conf.CHECK_FUNCS('sched_setscheduler'):
147         Logs.error('Need sched_setscheduler()')
148         sys.exit(1)
149     conf.CHECK_FUNCS('mlockall')
150
151     if not conf.CHECK_VARIABLE('ETIME', headers='errno.h'):
152         conf.DEFINE('ETIME', 'ETIMEDOUT')
153
154     if sys.platform.startswith('linux'):
155         conf.SET_TARGET_TYPE('pcap', 'EMPTY')
156     else:
157         if not conf.CHECK_HEADERS('pcap.h'):
158             Logs.error('Need libpcap')
159             sys.exit(1)
160         if not conf.CHECK_FUNCS_IN('pcap_open_live', 'pcap', headers='pcap.h'):
161             Logs.error('Need libpcap')
162             sys.exit(1)
163
164     if not conf.CHECK_FUNCS_IN('backtrace backtrace_symbols', 'execinfo',
165                                checklibc=True, headers='execinfo.h'):
166         Logs.error('backtrace support not available')
167
168     have_pmda = False
169     if Options.options.ctdb_pmda:
170         pmda_support = True
171
172         if not conf.CHECK_HEADERS('pcp/pmapi.h pcp/impl.h pcp/pmda.h',
173                                   together=True):
174             pmda_support = False
175         if not conf.CHECK_FUNCS_IN('pmProgname', 'pcp'):
176             pmda_support = False
177         if not conf.CHECK_FUNCS_IN('pmdaDaemon', 'pcp_pmda'):
178             pmda_support = False
179         if pmda_support:
180             conf.CHECK_TYPE_IN('__pmID_int', 'pcp/pmapi.h pcp/impl.h')
181             have_pmda = True
182         else:
183             Logs.error("PMDA support not available")
184             sys.exit(1)
185     if have_pmda:
186         Logs.info('Building with PMDA support')
187         conf.define('HAVE_PMDA', 1)
188         conf.env.CTDB_PMDADIR = os.path.join(conf.env.LOCALSTATEDIR,
189                                              'lib/pcp/pmdas/ctdb')
190
191     have_infiniband = False
192     if Options.options.ctdb_infiniband:
193         ib_support = True
194
195         if not conf.CHECK_HEADERS('infiniband/verbs.h rdma/rdma_cma.h'):
196             ib_support = False
197         if not conf.CHECK_FUNCS_IN('ibv_create_qp', 'ibverbs'):
198             ib_support = False
199         if not conf.CHECK_FUNCS_IN('rdma_connect', 'rdmacm'):
200             ib_support = False
201         if ib_support:
202             have_infiniband = True
203         else:
204             Logs.error("Infiniband support not available")
205             sys.exit(1)
206     if have_infiniband:
207         Logs.info('Building with Infiniband support')
208         conf.define('HAVE_INFINIBAND', 1)
209         conf.define('USE_INFINIBAND', 1)
210
211     have_etcd_reclock = False
212     if Options.options.ctdb_etcd_reclock:
213         try:
214             conf.check_python_module('etcd')
215             have_etcd_reclock = True
216         except:
217             Logs.error('etcd support not available')
218             sys.exit(1)
219     if have_etcd_reclock:
220         Logs.info('Building with etcd support')
221     conf.env.etcd_reclock = have_etcd_reclock
222
223     if Options.options.ctdb_ceph_reclock:
224         if (conf.CHECK_HEADERS('rados/librados.h', False, False, 'rados') and
225                                         conf.CHECK_LIB('rados', shlib=True)):
226             Logs.info('Building with Ceph librados recovery lock support')
227             conf.define('HAVE_LIBRADOS', 1)
228         else:
229             Logs.error("Missing librados for Ceph recovery lock support")
230             sys.exit(1)
231
232     conf.env.CTDB_BINDIR = os.path.join(conf.env.EXEC_PREFIX, 'bin')
233     conf.env.CTDB_ETCDIR = os.path.join(conf.env.SYSCONFDIR, 'ctdb')
234     conf.env.CTDB_VARDIR = os.path.join(conf.env.LOCALSTATEDIR, 'lib/ctdb')
235     conf.env.CTDB_RUNDIR = os.path.join(conf.env.LOCALSTATEDIR, 'run/ctdb')
236     conf.env.CTDB_HELPER_BINDIR = os.path.join(conf.env.LIBEXECDIR, 'ctdb')
237
238     if Options.options.ctdb_logdir:
239         conf.env.CTDB_LOGDIR = Options.options.ctdb_logdir
240     else:
241         conf.env.CTDB_LOGDIR = os.path.join(conf.env.LOCALSTATEDIR, 'log')
242
243     if Options.options.ctdb_sockpath:
244         conf.env.CTDB_SOCKPATH = Options.options.ctdb_sockpath
245     else:
246         conf.env.CTDB_SOCKPATH = os.path.join(conf.env.CTDB_RUNDIR,
247                                               'ctdbd.socket')
248     conf.define('CTDB_SOCKET', conf.env.CTDB_SOCKPATH)
249
250     conf.ADD_CFLAGS('''-DCTDB_HELPER_BINDIR=\"%s\"
251                        -DLOGDIR=\"%s\"
252                        -DCTDB_ETCDIR=\"%s\"
253                        -DCTDB_VARDIR=\"%s\"
254                        -DCTDB_RUNDIR=\"%s\"''' % (
255                     conf.env.CTDB_HELPER_BINDIR,
256                     conf.env.CTDB_LOGDIR,
257                     conf.env.CTDB_ETCDIR,
258                     conf.env.CTDB_VARDIR,
259                     conf.env.CTDB_RUNDIR))
260
261     conf.env.CTDB_TEST_DATADIR = os.path.join(conf.env.EXEC_PREFIX,
262                                               'share/ctdb/tests')
263     conf.env.CTDB_TEST_LIBEXECDIR = os.path.join(conf.env.LIBEXECDIR, 'ctdb/tests')
264
265     # Allow unified compilation and separate compilation of utilities
266     # to find includes
267     if not conf.env.standalone_ctdb:
268         conf.ADD_EXTRA_INCLUDES('#include/public #ctdb/include #ctdb')
269     else:
270         if srcdir == '.':
271             # Building from tarball
272             conf.ADD_EXTRA_INCLUDES('#include')
273         else:
274             # Building standalone CTDB from within Samba tree
275             conf.ADD_EXTRA_INCLUDES('#ctdb/include')
276             conf.ADD_EXTRA_INCLUDES('#ctdb')
277         conf.ADD_EXTRA_INCLUDES('#lib #lib/replace')
278
279         conf.DEFINE('HAVE_CONFIG_H', 1, add_to_cflags=True)
280         conf.DEFINE('SAMBA_UTIL_CORE_ONLY', 1, add_to_cflags=True)
281         conf.SAMBA_CONFIG_H()
282
283     if 'XSLTPROC_MANPAGES' in conf.env and conf.env['XSLTPROC_MANPAGES']:
284         conf.env.ctdb_generate_manpages = True
285     else:
286         conf.env.ctdb_generate_manpages = False
287
288         Logs.info("xsltproc unavailable, checking for pre-built manpages")
289         conf.env.ctdb_prebuilt_manpages = []
290         manpages = manpages_binary + manpages_misc
291         if conf.env.etcd_reclock:
292             manpages += manpages_etcd
293         if conf.env.HAVE_LIBRADOS:
294             manpages += manpages_ceph
295         for m in manpages:
296             if os.path.exists(os.path.join("doc", m)):
297                 Logs.info("  %s: yes" % (m))
298                 conf.env.ctdb_prebuilt_manpages.append(m)
299             else:
300                 Logs.info("  %s: no" % (m))
301
302 def gen_ctdb_version(task):
303     fp = file(task.outputs[0].bldpath(task.env), 'w')
304     fp.write('/* This file is auto-generated from waf */\n')
305     fp.write('#include "version.h"\n')
306     fp.write('\n')
307     fp.write('#define CTDB_VERSION_STRING "%s"\n' % VERSION)
308     fp.close()
309
310
311 def build(bld):
312     if bld.env.standalone_ctdb:
313         # enable building of public headers in the build tree
314         bld.env.build_public_headers = 'include/public'
315
316     if bld.env.standalone_ctdb:
317         bld.SAMBA_MKVERSION('version.h', '%s/VERSION' % vdir)
318
319     t = bld.SAMBA_GENERATOR('ctdb-version-header',
320                             target='include/ctdb_version.h',
321                             rule=gen_ctdb_version,
322                             dep_vars=['VERSION'])
323     t.env.VERSION = VERSION
324
325     bld.env.PKGCONFIGDIR = '${LIBDIR}/pkgconfig'
326
327     bld.RECURSE('lib/replace')
328     if bld.CHECK_FOR_THIRD_PARTY():
329         bld.RECURSE('third_party/popt')
330         if bld.env.standalone_ctdb or bld.CONFIG_GET('SOCKET_WRAPPER'):
331             bld.RECURSE('third_party/socket_wrapper')
332
333     bld.RECURSE('lib/tdb_wrap')
334     bld.RECURSE('lib/util')
335     bld.RECURSE('lib/async_req')
336
337     bld.RECURSE('lib/talloc')
338     bld.RECURSE('lib/tevent')
339     bld.RECURSE('lib/tdb')
340
341     if bld.env.standalone_ctdb:
342         # If a combined build is implemented, CTDB will want to
343         # build against samba-util rather than samba-util-core.
344         # Similarly, other Samba subsystems expect samba-util.  So,
345         # for a standalone build, just define a fake samba-util
346         # subsystem that pulls in samba-util-core.
347         bld.SAMBA_SUBSYSTEM('samba-util',
348                             source='',
349                             deps='samba-util-core')
350
351     bld.SAMBA_SUBSYSTEM('ctdb-tcp',
352                         source=bld.SUBDIR('tcp',
353                                           'tcp_connect.c tcp_init.c tcp_io.c'),
354                         includes='include',
355                         deps='replace tdb talloc tevent')
356
357     ib_deps = ''
358     if bld.env.HAVE_INFINIBAND:
359         bld.SAMBA_SUBSYSTEM('ctdb-ib',
360                             source=bld.SUBDIR('ib',
361                                               '''ibwrapper.c ibw_ctdb.c
362                                                  ibw_ctdb_init.c'''),
363                             includes='include',
364                             deps='replace talloc tevent tdb')
365         ib_deps = ' ctdb-ib rdmacm ibverbs'
366
367     if sys.platform.startswith('linux'):
368         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_linux.c')
369     elif sys.platform.startswith('aix'):
370         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_aix.c')
371     elif sys.platform.startswith('freebsd'):
372         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_freebsd.c')
373     elif sys.platform.startswith('gnukfreebsd'):
374         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_kfreebsd.c')
375     elif sys.platform == 'gnu':
376         CTDB_SYSTEM_SRC = bld.SUBDIR('common', 'system_gnu.c')
377     else:
378         Logs.error("Platform %s not supported" % sys.platform)
379
380     bld.SAMBA_SUBSYSTEM('ctdb-system',
381                         source=bld.SUBDIR('common',
382                                           'system_common.c system_util.c') +
383                                CTDB_SYSTEM_SRC,
384                         includes='include',
385                         deps='replace talloc tevent tdb pcap samba-util')
386
387     bld.SAMBA_SUBSYSTEM('ctdb-common',
388                         source=bld.SUBDIR('common',
389                                           '''ctdb_io.c ctdb_util.c ctdb_ltdb.c
390                                              sock_io.c'''),
391                         includes='include',
392                         deps='''replace popt talloc tevent tdb popt ctdb-system
393                                 ctdb-protocol-util''')
394
395     bld.SAMBA_SUBSYSTEM('ctdb-util',
396                         source=bld.SUBDIR('common',
397                                           '''db_hash.c srvid.c reqid.c
398                                              pkt_read.c pkt_write.c comm.c
399                                              logging.c rb_tree.c tunable.c
400                                              pidfile.c run_proc.c
401                                              hash_count.c run_event.c
402                                              sock_client.c version.c
403                                              cmdline.c
404                                           '''),
405                         deps='''samba-util sys_rw tevent-util
406                                 replace talloc tevent tdb popt''')
407
408     bld.SAMBA_SUBSYSTEM('ctdb-protocol',
409                         source=bld.SUBDIR('protocol',
410                                           '''protocol_header.c protocol_packet.c
411                                              protocol_basic.c protocol_types.c
412                                              protocol_call.c
413                                              protocol_message.c
414                                              protocol_control.c
415                                              protocol_keepalive.c
416                                              protocol_tunnel.c
417                                              protocol_client.c
418                                              protocol_debug.c
419                                              protocol_event.c
420                                              protocol_sock.c'''),
421                         includes='include',
422                         deps='replace talloc tdb')
423
424     bld.SAMBA_SUBSYSTEM('ctdb-protocol-util',
425                         source='protocol/protocol_util.c',
426                         deps='replace talloc tdb')
427
428     bld.SAMBA_SUBSYSTEM('ctdb-client',
429                         source=bld.SUBDIR('client',
430                                           '''client_connect.c client_call.c
431                                              client_message.c client_control.c
432                                              client_message_sync.c
433                                              client_control_sync.c
434                                              client_db.c client_util.c
435                                              client_event.c client_tunnel.c
436                                           '''),
437                         includes='include',
438                         deps='replace talloc tevent tdb tdb-wrap')
439
440     bld.SAMBA_SUBSYSTEM('ctdb-server-util',
441                         source=bld.SUBDIR('common',
442                                           '''sock_daemon.c'''),
443                         deps='''samba-util ctdb-util tevent-util
444                                 LIBASYNC_REQ replace talloc tevent''')
445
446     bld.SAMBA_SUBSYSTEM('ctdb-ipalloc',
447                         source=bld.SUBDIR('server',
448                                           '''ipalloc_deterministic.c
449                                              ipalloc_nondeterministic.c
450                                              ipalloc_lcp2.c
451                                              ipalloc_common.c
452                                              ipalloc.c
453                                           '''),
454                         includes='include',
455                         deps='ctdb-protocol-util replace talloc tevent')
456
457     bld.SAMBA_BINARY('ctdbd',
458                      source='server/ctdbd.c ' +
459                                bld.SUBDIR('server',
460                                           '''ctdb_daemon.c ctdb_recoverd.c
461                                              ctdb_recover.c ctdb_freeze.c
462                                              ctdb_tunables.c ctdb_monitor.c
463                                              ctdb_server.c ctdb_control.c
464                                              ctdb_call.c ctdb_ltdb_server.c
465                                              ctdb_traverse.c eventscript.c
466                                              ctdb_takeover.c
467                                              ctdb_persistent.c ctdb_keepalive.c
468                                              ctdb_cluster_mutex.c
469                                              ctdb_logging.c
470                                              ctdb_uptime.c
471                                              ctdb_vacuum.c ctdb_banning.c
472                                              ctdb_statistics.c
473                                              ctdb_update_record.c
474                                              ctdb_lock.c ctdb_fork.c
475                                              ctdb_tunnel.c ctdb_client.c'''),
476                      includes='include',
477                      deps='''ctdb-common ctdb-system ctdb-protocol
478                              ctdb-tcp ctdb-util replace sys_rw popt
479                              talloc tevent tdb-wrap tdb talloc_report''' +
480                           ib_deps,
481                      install_path='${SBINDIR}',
482                      manpages='ctdbd.1')
483
484     bld.SAMBA_BINARY('ctdb',
485                      source='tools/ctdb.c',
486                      deps='''ctdb-client ctdb-protocol ctdb-protocol-util
487                              ctdb-util ctdb-system samba-util sys_rw popt''',
488                      install_path='${BINDIR}',
489                      manpages='ctdb.1')
490
491     bld.SAMBA_BINARY('ctdb_killtcp',
492                      source='tools/ctdb_killtcp.c',
493                      deps='''ctdb-protocol-util ctdb-util ctdb-system
494                              samba-util replace''',
495                      install_path='${CTDB_HELPER_BINDIR}')
496
497     bld.SAMBA_BINARY('ctdb_event',
498                      source='tools/ctdb_event.c',
499                      deps='''ctdb-client ctdb-protocol ctdb-protocol-util
500                              ctdb-util ctdb-system samba-util replace''',
501                      install_path='${CTDB_HELPER_BINDIR}')
502
503     bld.SAMBA_BINARY('ltdbtool',
504                      source='tools/ltdbtool.c',
505                      includes='include',
506                      deps='tdb',
507                      install_path='${BINDIR}',
508                      manpages='ltdbtool.1')
509
510     bld.SAMBA_BINARY('ctdb_eventd',
511                      source='server/ctdb_eventd.c',
512                      deps='''ctdb-server-util ctdb-protocol ctdb-protocol-util
513                              ctdb-util samba-util replace popt''',
514                      install_path='${CTDB_HELPER_BINDIR}')
515
516     bld.SAMBA_BINARY('ctdb_lock_helper',
517                      source='server/ctdb_lock_helper.c',
518                      deps='''samba-util sys_rw ctdb-system tevent-util
519                              talloc tevent tdb''',
520                      includes='include',
521                      install_path='${CTDB_HELPER_BINDIR}')
522
523     bld.SAMBA_BINARY('ctdb_recovery_helper',
524                      source='server/ctdb_recovery_helper.c',
525                      deps='''ctdb-client ctdb-protocol ctdb-util
526                              samba-util sys_rw replace tdb''',
527                      install_path='${CTDB_HELPER_BINDIR}')
528
529     bld.SAMBA_BINARY('ctdb_takeover_helper',
530                      source='server/ctdb_takeover_helper.c',
531                      deps='''ctdb-client ctdb-protocol ctdb-util
532                              samba-util sys_rw replace ctdb-ipalloc popt''',
533                      install_path='${CTDB_HELPER_BINDIR}')
534
535     bld.SAMBA_BINARY('ctdb_mutex_fcntl_helper',
536                      source='server/ctdb_mutex_fcntl_helper.c',
537                      deps='sys_rw ctdb-system',
538                      includes='include',
539                      install_path='${CTDB_HELPER_BINDIR}')
540
541     bld.SAMBA_GENERATOR('ctdb-smnotify-h',
542                         source='utils/smnotify/smnotify.x',
543                         target='utils/smnotify/smnotify.h',
544                         rule='rpcgen -h ${SRC} > ${TGT}')
545
546     xdr_buf_hack = 'sed -e "s@^\([ \t]*register int32_t \*buf\);@\\1 = buf;@"'
547
548     bld.SAMBA_GENERATOR('ctdb-smnotify-x',
549                         source='utils/smnotify/smnotify.x',
550                         target='utils/smnotify/gen_xdr.c',
551                         rule='rpcgen -c ${SRC} | ' + xdr_buf_hack + ' > ${TGT}')
552
553     bld.SAMBA_GENERATOR('ctdb-smnotify-c',
554                         source='utils/smnotify/smnotify.x',
555                         target='utils/smnotify/gen_smnotify.c',
556                         rule='rpcgen -l ${SRC} > ${TGT}')
557
558     bld.SAMBA_BINARY('smnotify',
559                      source=bld.SUBDIR('utils/smnotify',
560                                        'smnotify.c gen_smnotify.c gen_xdr.c'),
561                      deps='ctdb-smnotify-h ctdb-smnotify-c ctdb-smnotify-x popt tirpc',
562                      includes='utils utils/smnotify',
563                      install_path='${CTDB_HELPER_BINDIR}')
564
565     bld.SAMBA_BINARY('ping_pong',
566                      source='utils/ping_pong/ping_pong.c',
567                      deps='',
568                      install_path='${BINDIR}',
569                      manpages='ping_pong.1')
570
571     if bld.env.HAVE_PMDA:
572         bld.SAMBA_BINARY('pmdactdb',
573                          source='utils/pmda/pmda_ctdb.c',
574                          includes='include',
575                          deps='''ctdb-client ctdb-protocol ctdb-util
576                                  samba-util pcp_pmda pcp''',
577                          install_path='${CTDB_PMDADIR}')
578         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Install',
579                           destname='Install')
580         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/Remove',
581                           destname='Remove')
582         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/pmns',
583                           destname='pmns')
584         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/domain.h',
585                           destname='domain.h')
586         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/help',
587                           destname='help')
588         bld.INSTALL_FILES('${CTDB_PMDADIR}', 'utils/pmda/README',
589                           destname='README')
590
591     if bld.env.HAVE_LIBRADOS:
592         bld.SAMBA_BINARY('ctdb_mutex_ceph_rados_helper',
593                          source='utils/ceph/ctdb_mutex_ceph_rados_helper.c',
594                          deps='talloc tevent rados',
595                          includes='include',
596                          install_path='${CTDB_HELPER_BINDIR}')
597
598     sed_expr1 = 's|/usr/local/var/lib/ctdb|%s|g'  % (bld.env.CTDB_VARDIR)
599     sed_expr2 = 's|/usr/local/etc/ctdb|%s|g'      % (bld.env.CTDB_ETCDIR)
600     sed_expr3 = 's|/usr/local/var/log|%s|g'       % (bld.env.CTDB_LOGDIR)
601     sed_expr4 = 's|/usr/local/var/run/ctdb|%s|g'  % (bld.env.CTDB_RUNDIR)
602     sed_expr5 = 's|/usr/local/sbin|%s|g'          % (bld.env.SBINDIR)
603     sed_expr6 = 's|/usr/local/libexec/ctdb|%s|g'  % (bld.env.CTDB_HELPER_BINDIR)
604     sed_expr7 = 's|/usr/local/bin|%s|g'           % (bld.env.BINDIR)
605     sed_cmdline = '-e "%s" -e "%s" -e "%s" -e "%s" -e "%s" -e "%s" -e "%s"' % \
606                        (sed_expr1, sed_expr2, sed_expr3, sed_expr4, sed_expr5,
607                         sed_expr6, sed_expr7)
608
609     manpages_extra = manpages_misc
610     if bld.env.etcd_reclock:
611         manpages_extra += manpages_etcd
612     if bld.env.HAVE_LIBRADOS:
613         manpages_extra += manpages_ceph
614     for f in manpages_binary + manpages_extra:
615         x = '%s.xml' % (f)
616         bld.SAMBA_GENERATOR(x,
617                             source=os.path.join('doc', x),
618                             target=x,
619                             rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
620
621     if bld.env.ctdb_generate_manpages:
622         bld.MANPAGES(' '.join(manpages_extra), True)
623     else:
624         for m in bld.env.ctdb_prebuilt_manpages:
625             bld.SAMBA_GENERATOR(m,
626                                 source=os.path.join("doc", m),
627                                 target=m,
628                                 rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
629             bld.INSTALL_FILES('${MANDIR}/man%s' % m[-1], m)
630
631     bld.SAMBA_GENERATOR('ctdb-onnode',
632                         source='tools/onnode',
633                         target='onnode',
634                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
635     bld.INSTALL_FILES('${BINDIR}', 'onnode',
636                       destname='onnode', chmod=0755)
637
638     bld.SAMBA_GENERATOR('ctdb-diagnostics',
639                         source='tools/ctdb_diagnostics',
640                         target='ctdb_diagnostics',
641                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
642     bld.INSTALL_FILES('${BINDIR}', 'ctdb_diagnostics',
643                       destname='ctdb_diagnostics', chmod=0755)
644
645     if bld.env.etcd_reclock:
646         bld.SAMBA_GENERATOR('ctdb-etcd-lock',
647                             source='utils/etcd/ctdb_etcd_lock',
648                             target='ctdb_etcd_lock',
649                             rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
650         bld.INSTALL_FILES('${CTDB_HELPER_BINDIR}', 'ctdb_etcd_lock',
651                           destname='ctdb_etcd_lock', chmod=0744)
652
653     bld.SAMBA_GENERATOR('ctdb-natgw',
654                         source='tools/ctdb_natgw',
655                         target='ctdb_natgw',
656                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
657     bld.INSTALL_FILES('${CTDB_HELPER_BINDIR}', 'ctdb_natgw',
658                       destname='ctdb_natgw', chmod=0755)
659
660     bld.SAMBA_GENERATOR('ctdb-lvs',
661                         source='tools/ctdb_lvs',
662                         target='ctdb_lvs',
663                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
664     bld.INSTALL_FILES('${CTDB_HELPER_BINDIR}', 'ctdb_lvs',
665                       destname='ctdb_lvs', chmod=0755)
666
667     bld.SAMBA_GENERATOR('ctdbd-wrapper',
668                         source='config/ctdbd_wrapper',
669                         target='ctdbd_wrapper',
670                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
671     bld.INSTALL_FILES('${SBINDIR}', 'ctdbd_wrapper',
672                       destname='ctdbd_wrapper', chmod=0755)
673
674     def SUBDIR_MODE_callback(arg, dirname, fnames):
675         for f in fnames:
676             fl = os.path.join(dirname, f)
677             if os.path.isdir(fl) or os.path.islink(fl):
678                 continue
679             mode = os.lstat(fl).st_mode & 0777
680             if arg['trim_path']:
681                 fl = samba_utils.os_path_relpath(fl, arg['trim_path'])
682             arg['file_list'].append([fl, mode])
683
684     def SUBDIR_MODE(path, trim_path=None):
685         pd = {'trim_path': trim_path, 'file_list': []}
686         os.path.walk(path, SUBDIR_MODE_callback, pd)
687         return pd['file_list']
688
689     etc_subdirs = [
690         'events.d',
691         'nfs-checks.d'
692     ]
693
694     if bld.env.standalone_ctdb:
695         configdir = 'config'
696     else:
697         configdir = 'ctdb/config'
698
699     for t in etc_subdirs:
700         files = SUBDIR_MODE('%s/%s' % (configdir, t), trim_path=configdir)
701         for fmode in files:
702             bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % fmode[0],
703                               destname=fmode[0], chmod=fmode[1])
704
705     bld.SAMBA_GENERATOR('ctdb-functions',
706                         source='config/functions',
707                         target='functions',
708                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
709     bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'functions', destname='functions')
710
711     etc_scripts = [
712         'ctdb-crash-cleanup.sh',
713         'debug-hung-script.sh',
714         'debug_locks.sh',
715         'nfs-linux-kernel-callout',
716         'notify.sh',
717         'statd-callout'
718     ]
719
720     for t in etc_scripts:
721         bld.INSTALL_FILES(bld.env.CTDB_ETCDIR, 'config/%s' % t,
722                           destname=t, chmod=0755)
723
724     bld.SAMBA_GENERATOR('ctdb-sudoers',
725                         source='config/ctdb.sudoers',
726                         target='ctdb.sudoers',
727                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
728     bld.INSTALL_FILES('${SYSCONFDIR}/sudoers.d', 'ctdb.sudoers',
729                       destname='ctdb')
730
731     bld.INSTALL_FILES('${CTDB_ETCDIR}/notify.d', 'config/notify.d.README',
732                       destname='README')
733
734     bld.install_dir(bld.env.CTDB_LOGDIR)
735     bld.install_dir(bld.env.CTDB_RUNDIR)
736     bld.install_dir(bld.env.CTDB_VARDIR)
737
738     # Unit tests
739     ctdb_unit_tests = [
740         'db_hash_test',
741         'srvid_test',
742         'pkt_read_test',
743         'pkt_write_test',
744         'comm_test',
745         'comm_server_test',
746         'comm_client_test',
747         'pidfile_test',
748         'run_proc_test',
749         'sock_daemon_test',
750         'sock_io_test',
751         'hash_count_test',
752         'run_event_test',
753         'cmdline_test'
754     ]
755
756     for target in ctdb_unit_tests:
757         src = 'tests/src/' + target + '.c'
758
759         bld.SAMBA_BINARY(target,
760                          source=src,
761                          deps='''talloc tevent tdb tevent-util popt
762                                  LIBASYNC_REQ samba-util sys_rw''',
763                          install_path='${CTDB_TEST_LIBEXECDIR}')
764
765     bld.SAMBA_BINARY('reqid_test',
766                      source='tests/src/reqid_test.c',
767                      deps='samba-util',
768                      install_path='${CTDB_TEST_LIBEXECDIR}')
769
770     bld.SAMBA_BINARY('rb_test',
771                      source='tests/src/rb_test.c',
772                      deps='samba-util talloc',
773                      install_path='${CTDB_TEST_LIBEXECDIR}')
774
775     bld.SAMBA_BINARY('ctdb_packet_parse',
776                      source='tests/src/ctdb_packet_parse.c',
777                      deps='talloc tevent tdb ctdb-protocol',
778                      install_path='${CTDB_TEST_LIBEXECDIR}')
779
780     bld.SAMBA_BINARY('porting_tests',
781                      source='tests/src/porting_tests.c',
782                      deps='samba-util ctdb-system popt',
783                      install_path='${CTDB_TEST_LIBEXECDIR}')
784
785     bld.SAMBA_SUBSYSTEM('protocol-tests-common',
786                         source=bld.SUBDIR('tests/src',
787                                           '''protocol_common.c
788                                              protocol_common_ctdb.c
789                                              protocol_common_event.c
790                                           '''),
791                         includes='include',
792                         deps='replace popt talloc tevent tdb')
793
794     ctdb_protocol_tests = [
795             'protocol_basic_test',
796             'protocol_types_test',
797             'protocol_ctdb_test',
798             'protocol_event_test',
799             'protocol_util_test',
800             'protocol_types_compat_test',
801             'protocol_ctdb_compat_test',
802     ]
803
804     for target in ctdb_protocol_tests:
805         src = 'tests/src/' + target + '.c'
806
807         bld.SAMBA_BINARY(target,
808                          source=src,
809                          deps='''protocol-tests-common
810                                  samba-util talloc tdb''',
811                          install_path='${CTDB_TEST_LIBEXECDIR}')
812
813     bld.SAMBA_SUBSYSTEM('ctdb-tests-common',
814                         source=bld.SUBDIR('tests/src',
815                                           'cluster_wait.c test_options.c'),
816                         includes='include',
817                         deps='replace popt talloc tevent tdb')
818
819     # Test binaries
820     ctdb_tests = [
821         'g_lock_loop',
822         'message_ring',
823         'fetch_ring',
824         'fetch_loop',
825         'fetch_loop_key',
826         'fetch_readonly',
827         'fetch_readonly_loop',
828         'transaction_loop',
829         'update_record',
830         'update_record_persistent',
831         'lock_tdb',
832         'dummy_client',
833         'tunnel_test',
834         'tunnel_cmd',
835     ]
836
837     for target in ctdb_tests:
838         src = 'tests/src/' + target + '.c'
839
840         bld.SAMBA_BINARY(target,
841                          source=src,
842                          includes='include',
843                          deps='''ctdb-client ctdb-protocol ctdb-util
844                                  samba-util ctdb-tests-common''',
845                          install_path='${CTDB_TEST_LIBEXECDIR}')
846
847     bld.SAMBA_BINARY('ctdb_takeover_tests',
848                      source='''tests/src/ctdb_takeover_tests.c
849                                tests/src/ipalloc_read_known_ips.c''',
850                      deps='''replace popt tdb tevent talloc ctdb-system
851                              samba-util tdb-wrap talloc_report
852                              ctdb-ipalloc ctdb-protocol ctdb-util''',
853                      includes='include',
854                      install_path='${CTDB_TEST_LIBEXECDIR}')
855
856     bld.SAMBA_BINARY('fake_ctdbd',
857                      source='''tests/src/fake_ctdbd.c
858                                tests/src/ipalloc_read_known_ips.c''',
859                      deps='''ctdb-util ctdb-protocol ctdb-protocol-util
860                              ctdb-system samba-util tevent-util
861                              LIBASYNC_REQ popt''',
862                      install_path='${CTDB_TEST_LIBEXECDIR}')
863
864     if bld.env.HAVE_INFINIBAND:
865         bld.SAMBA_BINARY('ibwrapper_test',
866                          source='ib/ibwrapper_test.c',
867                          includes='include',
868                          deps='replace talloc ctdb-common sys_rw' +
869                               ib_deps,
870                          install_path='${CTDB_TEST_LIBEXECDIR}')
871
872     if bld.env.HAVE_ROBUST_MUTEXES and sys.platform.startswith('linux'):
873         bld.SAMBA_BINARY('test_mutex_raw',
874                          source='tests/src/test_mutex_raw.c',
875                          deps='pthread',
876                          install_path='${CTDB_TEST_LIBEXECDIR}')
877
878     test_subdirs = [
879         'complex',
880         'cunit',
881         'eventd',
882         'eventscripts',
883         'onnode',
884         'shellcheck',
885         'simple',
886         'takeover',
887         'takeover_helper',
888         'tool'
889     ]
890
891     if bld.env.standalone_ctdb:
892         testdir = 'tests'
893     else:
894         testdir = 'ctdb/tests'
895
896     for t in test_subdirs:
897         files = SUBDIR_MODE('%s/%s' % (testdir, t), trim_path=testdir)
898         for fmode in files:
899             bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR, 'tests/%s' % fmode[0],
900                               destname=fmode[0], chmod=fmode[1])
901
902     # Install tests/scripts directory without test_wrap
903     test_scripts = [
904         'common.sh',
905         'integration.bash',
906         'unit.sh'
907     ]
908
909     for t in test_scripts:
910         bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR,
911                           os.path.join('tests/scripts', t),
912                           destname=os.path.join('scripts', t))
913
914     sed_expr = 's@^TEST_SCRIPTS_DIR=.*@&\\nexport TEST_BIN_DIR=\"%s\"@' % (
915                bld.env.CTDB_TEST_LIBEXECDIR)
916     bld.SAMBA_GENERATOR('ctdb-test-wrap',
917                         source='tests/scripts/test_wrap',
918                         target='test_wrap',
919                         rule='sed -e "%s" ${SRC} > ${TGT}' % sed_expr)
920     bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR+"/scripts", 'test_wrap',
921                       destname='test_wrap', chmod=0755)
922
923     bld.SAMBA_GENERATOR('ctdb-test-script-install-paths',
924                         source='tests/scripts/script_install_paths.sh',
925                         target='script_install_paths.sh',
926                         rule='sed %s ${SRC} > ${TGT}' % (sed_cmdline))
927     bld.INSTALL_FILES(bld.env.CTDB_TEST_DATADIR+"/scripts",
928                       'script_install_paths.sh',
929                       destname='script_install_paths.sh', chmod=0644)
930
931     sed_expr1 = 's@^\(export %s\)=.*@\\1=%s\\nexport %s=\"%s\"@''' % (
932                     'CTDB_TEST_DIR', bld.env.CTDB_TEST_DATADIR,
933                     'TEST_BIN_DIR', bld.env.CTDB_TEST_LIBEXECDIR)
934     sed_expr2 = 's@^\(export CTDB_TESTS_ARE_INSTALLED\)=false@\\1=true@'
935     bld.SAMBA_GENERATOR('ctdb-test-runner',
936                         source='tests/run_tests.sh',
937                         target='ctdb_run_tests.sh',
938                         rule='sed -e "%s" -e "%s" ${SRC} > ${TGT}' % (
939                              sed_expr1, sed_expr2))
940     bld.INSTALL_FILES('${BINDIR}', 'ctdb_run_tests.sh',
941                       destname='ctdb_run_tests', chmod=0755)
942     bld.symlink_as(os.path.join(bld.env.BINDIR, 'ctdb_run_cluster_tests'),
943                    'ctdb_run_tests')
944
945
946 def testonly(ctx):
947     cmd = 'tests/run_tests.sh -V tests/var'
948     ret = samba_utils.RUN_COMMAND(cmd)
949     if ret != 0:
950         print('tests exited with exit status %d' % ret)
951         sys.exit(ret)
952
953
954 def test(ctx):
955     import Scripting
956     Scripting.commands.append('build')
957     Scripting.commands.append('testonly')
958
959
960 def autotest(ctx):
961     env = samba_utils.LOAD_ENVIRONMENT()
962     cmd = 'tests/run_tests.sh -e -S %s -C' % env.SOCKET_WRAPPER_SO_PATH
963     ret = samba_utils.RUN_COMMAND(cmd)
964     if ret != 0:
965         print('autotest exited with exit status %d' % ret)
966         sys.exit(ret)
967
968
969 def show_version(ctx):
970     print VERSION
971
972
973 def manpages(ctx):
974     BASE_URL = 'http://docbook.sourceforge.net/release/xsl/current'
975     MAN_XSL = '%s/manpages/docbook.xsl' % BASE_URL
976     HTML_XSL = '%s/html/docbook.xsl' % BASE_URL
977     CMD_TEMPLATE = 'xsltproc --xinclude -o %s --nonet %s %s'
978     manpages = manpages_binary + manpages_misc + manpages_etcd + manpages_ceph
979     for t in manpages:
980         cmd = CMD_TEMPLATE % ('doc/%s' % t, MAN_XSL, 'doc/%s.xml' % t)
981         ret = samba_utils.RUN_COMMAND(cmd)
982         if ret != 0:
983             print('Command %s failed with exit status %d' % (cmd, ret))
984             sys.exit(ret)
985
986         cmd = CMD_TEMPLATE % ('doc/%s.html' % t, HTML_XSL, 'doc/%s.xml' % t)
987         ret = samba_utils.RUN_COMMAND(cmd)
988         if ret != 0:
989             print('Command %s failed with exit status %d' % (cmd, ret))
990             sys.exit(ret)
991
992
993 def distonly(ctx):
994     samba_dist.DIST_FILES('VERSION:VERSION', extend=True)
995
996     distfile = file('.distversion', 'w')
997     for field in version.vcs_fields:
998         distfile.write('%s=%s\n' % (field, str(version.vcs_fields[field])))
999     distfile.close()
1000     samba_dist.DIST_FILES('ctdb/.distversion:.distversion', extend=True)
1001
1002     t = 'ctdb.spec'
1003     sed_expr1 = 's/@VERSION@/%s/g' % VERSION
1004     sed_expr2 = 's/@RELEASE@/%s/g' % '1'
1005     cmd = 'sed -e "%s" -e "%s" packaging/RPM/ctdb.spec.in > %s' % (
1006         sed_expr1, sed_expr2, t)
1007     ret = samba_utils.RUN_COMMAND(cmd)
1008     if ret != 0:
1009         print('Command "%s" failed with exit status %d' % (cmd, ret))
1010         sys.exit(ret)
1011     samba_dist.DIST_FILES('ctdb/%s:%s' % (t, t), extend=True)
1012
1013     manpages = manpages_binary + manpages_misc + manpages_etcd + manpages_ceph
1014     for t in manpages:
1015         samba_dist.DIST_FILES('ctdb/doc/%s:doc/%s' % (t, t), extend=True)
1016         samba_dist.DIST_FILES('ctdb/doc/%s.html:doc/%s.html' % (t, t),
1017                               extend=True)
1018
1019     samba_dist.dist()
1020
1021
1022 def dist():
1023     import Scripting
1024     Scripting.commands.append('manpages')
1025     Scripting.commands.append('distonly')
1026
1027
1028 def rpmonly(ctx):
1029     opts = os.getenv('RPM_OPTIONS') or ''
1030     cmd = 'rpmbuild -ta --clean --rmsource %s ctdb-%s.tar.gz' % (opts, VERSION)
1031     ret = samba_utils.RUN_COMMAND(cmd)
1032     if ret != 0:
1033         print('rpmbuild exited with exit status %d' % ret)
1034         sys.exit(ret)
1035
1036
1037 def rpm(ctx):
1038     import Scripting
1039     Scripting.commands.append('manpages')
1040     Scripting.commands.append('distonly')
1041     Scripting.commands.append('rpmonly')
1042
1043
1044 def ctags(ctx):
1045     "build 'tags' file using ctags"
1046     import Utils
1047     source_root = os.path.dirname(Utils.g_module.root_path)
1048     cmd = 'ctags $(find %s -name "*.[ch]")' % source_root
1049     print("Running: %s" % cmd)
1050     ret = samba_utils.RUN_COMMAND(cmd)
1051     if ret != 0:
1052         print('ctags failed with exit status %d' % ret)
1053         sys.exit(ret)