| #! /usr/bin/env python |
| # encoding: utf-8 |
| |
| ''' |
| |
| When using this tool, the wscript will look like: |
| |
| def options(opt): |
| opt.tool_options('ndnx') |
| |
| def configure(conf): |
| conf.load('compiler_c ndnx') |
| |
| def build(bld): |
| bld(source='main.cpp', target='app', use='NDNX') |
| |
| Options are generated, in order to specify the location of ndnx includes/libraries. |
| |
| |
| ''' |
| import sys, re |
| from waflib import Utils, Logs, Errors, Options, ConfigSet |
| from waflib.Configure import conf |
| |
| NDNX_DIR=['/usr','/usr/local','/opt/local','/sw'] |
| NDNX_VERSION_FILE='ndn/ndn.h' |
| NDNX_VERSION_CODE=''' |
| #include <ndn/ndn.h> |
| #include <stdio.h> |
| int main() { printf ("%d.%d.%d", ((NDN_API_VERSION/100000) % 100), ((NDN_API_VERSION/1000) % 100), (NDN_API_VERSION % 1000)); return 0; } |
| ''' |
| |
| @conf |
| def __ndnx_get_version_file(self,dir): |
| # Logs.pprint ('CYAN', ' + %s/%s/%s' % (dir, 'include', NDNX_VERSION_FILE)) |
| try: |
| return self.root.find_dir(dir).find_node('%s/%s' % ('include', NDNX_VERSION_FILE)) |
| except: |
| return None |
| @conf |
| def ndnx_get_version(self,dir): |
| val=self.check_cc(fragment=NDNX_VERSION_CODE,includes=['%s/%s' % (dir, 'include')],execute=True,define_ret = True, mandatory=True) |
| return val |
| @conf |
| def ndnx_get_root(self,*k,**kw): |
| root=Options.options.ndnx_dir or (k and k[0]) or kw.get('path',None) |
| |
| if root: |
| if self.__ndnx_get_version_file(root): |
| return root |
| self.fatal('NDNx not found in %s'%root) |
| |
| for dir in NDNX_DIR: |
| if self.__ndnx_get_version_file(dir): |
| return dir |
| self.fatal('NDNx not found, please provide a --ndnx argument (see help)') |
| |
| @conf |
| def check_openssl(self,*k,**kw): |
| root = k and k[0] or kw.get('path',None) or Options.options.openssl |
| mandatory = kw.get('mandatory', True) |
| var = kw.get('var', 'SSL') |
| |
| CODE = """ |
| #include <openssl/crypto.h> |
| #include <stdio.h> |
| |
| int main(int argc, char **argv) { |
| (void)argc; |
| printf ("%s", argv[0]); |
| |
| return 0; |
| } |
| """ |
| if root: |
| testApp = self.check_cc (lib=['ssl', 'crypto'], |
| header_name='openssl/crypto.h', |
| define_name='HAVE_%s' % var, |
| uselib_store=var, |
| mandatory = mandatory, |
| cflags="-I%s/include" % root, |
| linkflags="-L%s/lib" % root, |
| execute = True, fragment = CODE, define_ret = True) |
| else: |
| testApp = libcrypto = self.check_cc (lib=['ssl', 'crypto'], |
| header_name='openssl/crypto.h', |
| define_name='HAVE_%s' % var, |
| uselib_store=var, |
| mandatory = mandatory, |
| execute = True, fragment = CODE, define_ret = True) |
| |
| if not testApp: |
| return |
| |
| self.start_msg ('Checking if selected openssl matches NDNx') |
| |
| ndn_var = kw.get('ndn_var', "NDNX") |
| if Utils.unversioned_sys_platform () == "darwin": |
| def otool (binary): |
| p = Utils.subprocess.Popen (['/usr/bin/otool', '-L', binary], |
| stdout = Utils.subprocess.PIPE, ) |
| for line in p.communicate()[0].split ('\n'): |
| if re.match ('.*/libcrypto\..*', line): |
| return line |
| |
| selected_crypto = otool (testApp) |
| ndnd_crypto = otool ('%s/bin/ndnd' % self.env['%s_ROOT' % ndn_var]) |
| |
| if ndnd_crypto != selected_crypto: |
| self.fatal ("Selected openssl does not match used to compile NDNx (%s != %s)" % |
| (selected_crypto.strip (), ndnd_crypto.strip ())) |
| self.end_msg (True) |
| |
| elif Utils.unversioned_sys_platform () == "linux" or Utils.unversioned_sys_platform () == "freebsd": |
| def ldd (binary): |
| p = Utils.subprocess.Popen (['/usr/bin/ldd', binary], |
| stdout = Utils.subprocess.PIPE, ) |
| for line in p.communicate()[0].split ('\n'): |
| if re.match ('libcrypto\..*', line): |
| return line |
| |
| selected_crypto = ldd (testApp) |
| ndnd_crypto = ldd ('%s/bin/ndnd' % self.env['%s_ROOT' % ndn_var]) |
| |
| if ndnd_crypto != selected_crypto: |
| self.fatal ("Selected openssl does not match used to compile NDNx (%s != %s)" % |
| (selected_crypto.strip (), ndnd_crypto.strip ())) |
| self.end_msg (True) |
| else: |
| self.end_msg ("Don't know how to check", 'YELLOW') |
| |
| @conf |
| def check_ndnx(self,*k,**kw): |
| if not self.env['CC']: |
| self.fatal('load a c compiler first, conf.load("compiler_c")') |
| |
| var=kw.get('uselib_store', 'NDNX') |
| self.start_msg('Checking for NDNx') |
| root = self.ndnx_get_root(*k,**kw); |
| self.env.NDNX_VERSION=self.ndnx_get_version(root) |
| |
| self.env['INCLUDES_%s' % var]= '%s/%s' % (root, "include"); |
| self.env['LIB_%s' % var] = "ndn" |
| self.env['LIBPATH_%s' % var] = '%s/%s' % (root, "lib") |
| |
| self.env['%s_ROOT' % var] = root |
| |
| self.end_msg("%s in %s " % (self.env.NDNX_VERSION, root)) |
| if Logs.verbose: |
| Logs.pprint('CYAN',' NDNx include : %s'%self.env['INCLUDES_%s' % var]) |
| Logs.pprint('CYAN',' NDNx lib : %s'%self.env['LIB_%s' % var]) |
| Logs.pprint('CYAN',' NDNx libpath : %s'%self.env['LIBPATH_%s' % var]) |
| |
| def options(opt): |
| """ |
| NDNx options |
| """ |
| ndnopt = opt.add_option_group("NDNx Options") |
| ndnopt.add_option('--ndnx',type='string',default=None,dest='ndnx_dir',help='''path to where NDNx is installed, e.g. /usr/local''') |
| ndnopt.add_option('--openssl',type='string',default='',dest='openssl',help='''path to openssl, should be the same NDNx is compiled against''') |