Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 1 | #! /usr/bin/env python |
| 2 | # encoding: utf-8 |
| 3 | |
| 4 | ''' |
| 5 | |
| 6 | When using this tool, the wscript will look like: |
| 7 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 8 | def options(opt): |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 9 | opt.load('compiler_cxx cryptopp') |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 10 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 11 | def configure(conf): |
| 12 | conf.load('compiler_cxx cryptopp') |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 13 | conf.check_cryptopp() |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 14 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 15 | def build(bld): |
| 16 | bld(source='main.cpp', target='app', use='CRYPTOPP') |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 17 | |
| 18 | Options are generated, in order to specify the location of cryptopp includes/libraries. |
| 19 | |
| 20 | |
| 21 | ''' |
| 22 | import sys |
| 23 | import re |
| 24 | from waflib import Utils,Logs,Errors |
| 25 | from waflib.Configure import conf |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 26 | CRYPTOPP_DIR = ['/usr', '/usr/local', '/opt/local', '/sw', '/usr/local/ndn', '/opt/ndn'] |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 27 | CRYPTOPP_VERSION_FILE = 'config.h' |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 28 | |
Alexander Afanasyev | 00f855e | 2015-05-28 14:38:43 -0700 | [diff] [blame] | 29 | CRYPTOPP_CHECK_FRAGMENT = ''' |
| 30 | #include "../../src/security/cryptopp.hpp" |
| 31 | #include <iostream> |
| 32 | |
| 33 | int |
| 34 | main() |
| 35 | { |
| 36 | using namespace CryptoPP; |
| 37 | |
| 38 | std::string buffer = "Lorem ipsum dolor sit amet, consectetur adipiscing elit."; |
| 39 | SHA256 hash; |
| 40 | StringSource(buffer, true, new HashFilter(hash, new FileSink(std::cout))); |
| 41 | StringSource(reinterpret_cast<const uint8_t*>(buffer.c_str()), buffer.size(), |
| 42 | true, new HashFilter(hash, new FileSink(std::cout))); |
Alexander Afanasyev | 00f855e | 2015-05-28 14:38:43 -0700 | [diff] [blame] | 43 | } |
| 44 | ''' |
| 45 | |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 46 | def options(opt): |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 47 | opt.add_option('--with-cryptopp', type='string', default=None, dest='cryptopp_dir', |
| 48 | help='''Path to where CryptoPP is installed, e.g., /usr/local''') |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 49 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 50 | @conf |
| 51 | def __cryptopp_get_version_file(self, dir): |
| 52 | try: |
| 53 | return self.root.find_dir(dir).find_node('%s/%s' % ('include/cryptopp', |
| 54 | CRYPTOPP_VERSION_FILE)) |
| 55 | except: |
| 56 | return None |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 57 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 58 | @conf |
| 59 | def __cryptopp_find_root_and_version_file(self, *k, **kw): |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 60 | root = k and k[0] or kw.get('path', self.options.cryptopp_dir) |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 61 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 62 | file = self.__cryptopp_get_version_file(root) |
| 63 | if root and file: |
| 64 | return (root, file) |
| 65 | for dir in CRYPTOPP_DIR: |
| 66 | file = self.__cryptopp_get_version_file(dir) |
| 67 | if file: |
| 68 | return (dir, file) |
Alexander Afanasyev | a1ae0a1 | 2014-01-28 15:21:02 -0800 | [diff] [blame] | 69 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 70 | if root: |
| 71 | self.fatal('CryptoPP not found in %s' % root) |
| 72 | else: |
Mathias Gibbens | 16f24c5 | 2014-05-06 17:12:27 -0700 | [diff] [blame] | 73 | self.fatal('CryptoPP not found, please provide a --with-cryptopp=PATH argument (see help)') |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 74 | |
| 75 | @conf |
| 76 | def check_cryptopp(self, *k, **kw): |
| 77 | if not self.env['CXX']: |
| 78 | self.fatal('Load a c++ compiler first, e.g., conf.load("compiler_cxx")') |
| 79 | |
Alexander Afanasyev | fff47d6 | 2014-05-11 19:24:46 -0700 | [diff] [blame] | 80 | var = kw.get('uselib_store', 'CRYPTOPP') |
Alexander Afanasyev | 1160baa | 2014-04-10 18:50:29 -0700 | [diff] [blame] | 81 | mandatory = kw.get('mandatory', True) |
| 82 | |
Alexander Afanasyev | fff47d6 | 2014-05-11 19:24:46 -0700 | [diff] [blame] | 83 | use = kw.get('use', 'PTHREAD') |
| 84 | |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 85 | self.start_msg('Checking Crypto++ lib') |
| 86 | (root, file) = self.__cryptopp_find_root_and_version_file(*k, **kw) |
| 87 | |
| 88 | try: |
| 89 | txt = file.read() |
Alexander Afanasyev | 3b5f4ea | 2015-12-23 22:47:20 -0800 | [diff] [blame^] | 90 | re_version = re.compile('^#define\\s+CRYPTOPP_VERSION\\s+([0-9]+)', re.M) |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 91 | match = re_version.search(txt) |
| 92 | |
| 93 | if match: |
| 94 | self.env.CRYPTOPP_VERSION = match.group(1) |
Alexander Afanasyev | 3b5f4ea | 2015-12-23 22:47:20 -0800 | [diff] [blame^] | 95 | v = int(self.env.CRYPTOPP_VERSION) |
| 96 | (major, minor, patch) = (int(v / 100), int(v % 100 / 10), int(v % 10)) |
| 97 | self.end_msg("%d.%d.%d" % (major, minor, patch)) |
Alexander Afanasyev | 5e1288e | 2014-03-28 11:11:48 -0700 | [diff] [blame] | 98 | else: |
| 99 | self.fatal('CryptoPP files are present, but are not recognizable') |
| 100 | except: |
| 101 | self.fatal('CryptoPP not found or is not usable') |
| 102 | |
Alexander Afanasyev | 00f855e | 2015-05-28 14:38:43 -0700 | [diff] [blame] | 103 | isLibWorking = False |
| 104 | for defines in ['', 'CRYPTOPP_DISABLE_ASM']: |
| 105 | try: |
| 106 | self.check_cxx(msg='Checking if CryptoPP library works', |
| 107 | fragment=CRYPTOPP_CHECK_FRAGMENT, |
| 108 | lib='cryptopp', |
| 109 | includes="%s/include" % root, |
| 110 | libpath="%s/lib" % root, |
| 111 | mandatory=True, |
| 112 | use=use, |
| 113 | defines=defines, |
| 114 | uselib_store=var) |
| 115 | isLibWorking = True |
| 116 | break |
| 117 | except: |
| 118 | # try another flags |
| 119 | pass |
| 120 | |
| 121 | if mandatory and not isLibWorking: |
| 122 | self.fatal('CryptoPP is present, but is not usable') |