ot_btc_mainnet.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2022 The MMGen Project <mmgen@tuta.io>
  5. """
  6. test.objtest_py_d.ot_btc_mainnet: BTC mainnet test vectors for MMGen data objects
  7. """
  8. from mmgen.obj import *
  9. from mmgen.addrlist import AddrIdxList
  10. from mmgen.seedsplit import *
  11. from mmgen.key import *
  12. from .ot_common import *
  13. from mmgen.protocol import init_proto
  14. proto = init_proto('btc',need_amt=True)
  15. tw_pfx = proto.base_coin.lower() + ':'
  16. ssm = str(SeedShareCount.max_val)
  17. privkey = PrivKey(proto=proto,s=bytes.fromhex('deadbeef'*8),compressed=True,pubkey_type='std')
  18. tests = {
  19. 'Int': {
  20. 'arg1': 'n',
  21. 'bad': ('1L',0.0,'0.0','1.0',1.0,'s',1.1,'1.1'),
  22. 'good': (
  23. ('0',0),('-1',-1),('7',7),-1,0,1,9999999,
  24. {'n':'0x0','base':16,'ret':0},
  25. {'n':'0x1','base':16,'ret':1},
  26. {'n':'0xf','base':16,'ret':15},
  27. {'n':'0xff','base':16,'ret':255},
  28. )
  29. },
  30. 'AddrIdx': {
  31. 'arg1': 'n',
  32. 'bad': ('s',1.1,10000000,-1,0),
  33. 'good': (('7',7),(1,1),(9999999,9999999))
  34. },
  35. 'SeedShareIdx': {
  36. 'arg1': 'n',
  37. 'bad': ('s',1.1,1025,-1,0),
  38. 'good': (('7',7),(1,1),(1024,1024))
  39. },
  40. 'SeedShareCount': {
  41. 'arg1': 'n',
  42. 'bad': ('s',2.1,1025,-1,0,1),
  43. 'good': (('7',7),(2,2),(1024,1024))
  44. },
  45. 'MasterShareIdx': {
  46. 'arg1': 'n',
  47. 'bad': ('s',1.1,1025,-1,0),
  48. 'good': (('7',7),(1,1),(1024,1024))
  49. },
  50. 'AddrIdxList': {
  51. 'arg1': 'fmt_str',
  52. 'bad': ('x','5,9,1-2-3','8,-11','66,3-2'),
  53. 'good': (
  54. ('3,2,2',[2,3]),
  55. ('101,1,3,5,2-7,99',[1,2,3,4,5,6,7,99,101]),
  56. ({'idx_list':AddrIdxList('1-5')},[1,2,3,4,5])
  57. )
  58. },
  59. 'SubSeedIdxRange': {
  60. 'bad': (33,'x','-11','66,3','0','3-2','8000000','100000000',(1,2,3)),
  61. 'good': (
  62. ('3',(3,3)),
  63. ((3,5),(3,5)),
  64. ('1-2',(1,2)),
  65. (str(SubSeedList.dfl_len),(SubSeedList.dfl_len,SubSeedList.dfl_len)),
  66. (str(SubSeedIdxRange.max_idx),(SubSeedIdxRange.max_idx,SubSeedIdxRange.max_idx)),
  67. )
  68. },
  69. 'BTCAmt': {
  70. 'arg1': 'num',
  71. 'bad': ('-3.2','0.123456789',123,'123L','22000000',20999999.12345678,
  72. {'num':'1','from_decimal':True},
  73. {'num':1,'from_decimal':True},
  74. ),
  75. 'good': (('20999999.12345678',Decimal('20999999.12345678')),
  76. {'num':Decimal('1.23456789623456789'),'from_decimal':True,'ret':Decimal('1.23456790')}, # rounding
  77. {'num':Decimal('1.234'),'from_decimal':True,'ret':Decimal('1.234')},
  78. {'num':Decimal('0.0'),'from_decimal':True,'ret':Decimal('0')},
  79. # emulate network fee estimation:
  80. # BTC/kB tx_fee_adj tx size
  81. { 'num':Decimal('0.00053249') * Decimal('0.9') * 109 / 1024 , # ≈53 sat/byte
  82. 'from_decimal':True,
  83. 'ret':Decimal('0.00005101') },
  84. { 'num':Decimal('0.00003249') * Decimal('1.1') * 109 / 1024 , # ≈3 sat/byte
  85. 'from_decimal':True,
  86. 'ret':Decimal('0.00000380') },
  87. { 'num':Decimal('0.00011249') * Decimal('1.0') * 221 / 1024 , # ≈11 sat/byte
  88. 'from_decimal':True,
  89. 'ret':Decimal('0.00002428') },
  90. {'num':1234,'from_unit':'satoshi','ret':Decimal('0.00001234')},
  91. )
  92. },
  93. 'CoinAddr': {
  94. 'arg1': 'addr',
  95. 'good': (
  96. {'addr':'1MjjELEy6EJwk8fSNfpS8b5teFRo4X5fZr', 'proto':proto},
  97. {'addr':'32GiSWo9zJQgkCmjAaLRrbPwXhKry2jHhj', 'proto':proto},
  98. ),
  99. 'bad': (
  100. {'addr':1, 'proto':proto},
  101. {'addr':'x', 'proto':proto},
  102. {'addr':'я', 'proto':proto},
  103. ),
  104. },
  105. 'SeedID': {
  106. 'arg1': 'sid',
  107. 'bad': (
  108. {'sid':'я'},
  109. {'sid':'F00F00'},
  110. {'sid':'xF00F00x'},
  111. {'sid':1},
  112. {'sid':'F00BAA123'},
  113. {'sid':'f00baa12'},
  114. 'я',r32,'abc'
  115. ),
  116. 'good': (
  117. {'sid':'F00BAA12'},
  118. {'seed': Seed(r16), 'ret': SeedID(seed=Seed(r16))},
  119. {'sid': Seed(r16).sid, 'ret': SeedID(seed=Seed(r16))}
  120. )
  121. },
  122. 'SubSeedIdx': {
  123. 'arg1': 's',
  124. 'bad': (33,'x','я','1x',200,'1ss','L','s','200LS','30ll','s100',str(SubSeedIdxRange.max_idx+1),'0'),
  125. 'good': (('1','1L'),('1s','1S'),'20S','30L',('300l','300L'),('200','200L'),str(SubSeedIdxRange.max_idx)+'S')
  126. },
  127. 'MMGenID': {
  128. 'arg1': 'id_str',
  129. 'bad': (
  130. {'id_str':'x', 'proto':proto},
  131. {'id_str':1, 'proto':proto},
  132. {'id_str':'f00f00f', 'proto':proto},
  133. {'id_str':'a:b', 'proto':proto},
  134. {'id_str':'x:L:3', 'proto':proto},
  135. {'id_str':'F00BAA12', 'proto':proto},
  136. {'id_str':'F00BAA12:Z:99', 'proto':proto},
  137. ),
  138. 'good': (
  139. {'id_str':'F00BAA12:99', 'proto':proto, 'ret':'F00BAA12:L:99'},
  140. {'id_str':'F00BAA12:L:99', 'proto':proto},
  141. {'id_str':'F00BAA12:S:99', 'proto':proto},
  142. ),
  143. },
  144. 'TwMMGenID': {
  145. 'arg1': 'id_str',
  146. 'bad': (
  147. {'id_str':'x', 'proto':proto},
  148. {'id_str':'я', 'proto':proto},
  149. {'id_str':'я:я', 'proto':proto},
  150. {'id_str':1, 'proto':proto},
  151. {'id_str':'f00f00f', 'proto':proto},
  152. {'id_str':'a:b', 'proto':proto},
  153. {'id_str':'x:L:3', 'proto':proto},
  154. {'id_str':'F00BAA12:0', 'proto':proto},
  155. {'id_str':'F00BAA12:Z:99', 'proto':proto},
  156. {'id_str':tw_pfx, 'proto':proto},
  157. {'id_str':tw_pfx+'я', 'proto':proto},
  158. ),
  159. 'good': (
  160. {'id_str':tw_pfx+'x', 'proto':proto},
  161. {'id_str':'F00BAA12:99', 'proto':proto, 'ret':'F00BAA12:L:99'},
  162. {'id_str':'F00BAA12:L:99', 'proto':proto},
  163. {'id_str':'F00BAA12:S:9999999', 'proto':proto},
  164. ),
  165. },
  166. 'TwLabel': {
  167. 'arg1': 'proto',
  168. 'exc_name': 'BadTwLabel',
  169. 'bad': (
  170. {'text':'', 'proto':proto},
  171. {'text':'x x', 'proto':proto},
  172. {'text':'x я', 'proto':proto},
  173. {'text':'я:я', 'proto':proto},
  174. {'text':1, 'proto':proto},
  175. {'text':'f00f00f', 'proto':proto},
  176. {'text':'a:b', 'proto':proto},
  177. {'text':'x:L:3', 'proto':proto},
  178. {'text':'F00BAA12:0 x', 'proto':proto},
  179. {'text':'F00BAA12:Z:99', 'proto':proto},
  180. {'text':tw_pfx+' x', 'proto':proto},
  181. {'text':tw_pfx+'я x', 'proto':proto},
  182. {'text':utf8_ctrl[:40], 'proto':proto},
  183. {'text':'F00BAA12:S:1 ' + utf8_ctrl[:40], 'proto':proto, 'exc_name': 'BadTwComment'},
  184. ),
  185. 'good': (
  186. {'text':'F00BAA12:99 a comment', 'proto':proto, 'ret':'F00BAA12:L:99 a comment'},
  187. {'text':'F00BAA12:L:99 a comment', 'proto':proto},
  188. {'text': 'F00BAA12:L:99 comment (UTF-8) α', 'proto':proto},
  189. {'text':'F00BAA12:S:9999999 comment', 'proto':proto},
  190. {'text':tw_pfx+'x comment', 'proto':proto},
  191. ),
  192. },
  193. 'MMGenTxID': {
  194. 'arg1': 's',
  195. 'bad': (1,[],'\0','\1','я','g','gg','FF','f00','F00F0012'),
  196. 'good': ('DEADBE','F00BAA')
  197. },
  198. 'CoinTxID':{
  199. 'arg1': 's',
  200. 'bad': (1,[],'\0','\1','я','g','gg','FF','f00','F00F0012',r16.hex(),r32.hex()+'ee'),
  201. 'good': (r32.hex(),)
  202. },
  203. 'WifKey': {
  204. 'arg1': 'proto',
  205. 'bad': (
  206. {'proto':proto, 'wif':1},
  207. {'proto':proto, 'wif':[]},
  208. {'proto':proto, 'wif':'\0'},
  209. {'proto':proto, 'wif':'\1'},
  210. {'proto':proto, 'wif':'я'},
  211. {'proto':proto, 'wif':'g'},
  212. {'proto':proto, 'wif':'gg'},
  213. {'proto':proto, 'wif':'FF'},
  214. {'proto':proto, 'wif':'f00'},
  215. {'proto':proto, 'wif':r16.hex()},
  216. {'proto':proto, 'wif':'2MspvWFjBbkv2wzQGqhxJUYPCk3Y2jMaxLN'},
  217. ),
  218. 'good': (
  219. {'proto':proto, 'wif':'5KXEpVzjWreTcQoG5hX357s1969MUKNLuSfcszF6yu84kpsNZKb', 'ret_idx':1},
  220. {'proto':proto, 'wif':'KwWr9rDh8KK5TtDa3HLChEvQXNYcUXpwhRFUPc5uSNnMtqNKLFhk', 'ret_idx':1},
  221. )
  222. },
  223. 'PubKey': {
  224. 'arg1': 's',
  225. 'bad': (
  226. {'s':1, 'compressed':True },
  227. {'s':'F00BAA12','compressed':False},
  228. ),
  229. 'good': ( # TODO: add real pubkeys
  230. {'s':bytes.fromhex('deadbeef'),'compressed':True},
  231. )
  232. },
  233. 'PrivKey': {
  234. 'arg1': 'proto',
  235. 'bad': (
  236. {'proto':proto, 'wif':1},
  237. {'proto':proto, 'wif':'1'},
  238. {'proto':proto, 'wif':'cMsqcmDYZP1LdKgqRh9L4ZRU9br28yvdmTPwW2YQwVSN9aQiMAoR'},
  239. {'proto':proto, 's':r32,'wif':'5KXEpVzjWreTcQoG5hX357s1969MUKNLuSfcszF6yu84kpsNZKb'},
  240. {'proto':proto, 'pubkey_type':'std'},
  241. {'proto':proto, 's':r32},
  242. {'proto':proto, 's':r32,'compressed':'yes'},
  243. {'proto':proto, 's':r32,'compressed':'yes','pubkey_type':'std'},
  244. {'proto':proto, 's':r32,'compressed':True,'pubkey_type':'nonstd'},
  245. {'proto':proto, 's':r32+b'x','compressed':True,'pubkey_type':'std'}
  246. ),
  247. 'good': (
  248. {'proto':proto, 'wif':'5KXEpVzjWreTcQoG5hX357s1969MUKNLuSfcszF6yu84kpsNZKb',
  249. 'ret':bytes.fromhex('e0aef965b905a2fedf907151df8e0a6bac832aa697801c51f58bd2ecb4fd381c')},
  250. {'proto':proto, 'wif':'KwWr9rDh8KK5TtDa3HLChEvQXNYcUXpwhRFUPc5uSNnMtqNKLFhk',
  251. 'ret':bytes.fromhex('08d0ed83b64b68d56fa064be48e2385060ed205be2b1e63cd56d218038c3a05f')},
  252. {'proto':proto, 's':r32,'compressed':False,'pubkey_type':'std','ret':r32},
  253. {'proto':proto, 's':r32,'compressed':True,'pubkey_type':'std','ret':r32}
  254. )
  255. },
  256. 'AddrListID': { # a rather pointless test, but do it anyway
  257. 'arg1': 'sid',
  258. 'bad': (
  259. {'sid':SeedID(sid='F00BAA12'),'mmtype':'Z','ret':'F00BAA12:Z'},
  260. ),
  261. 'good': (
  262. {'sid':SeedID(sid='F00BAA12'),'mmtype':proto.addr_type(id_str='S'),'ret':'F00BAA12:S'},
  263. {'sid':SeedID(sid='F00BAA12'),'mmtype':proto.addr_type(id_str='L'),'ret':'F00BAA12:L'},
  264. )
  265. },
  266. 'MMGenWalletLabel': {
  267. 'arg1': 's',
  268. 'bad': (utf8_text[:49],utf8_combining[:48],utf8_ctrl[:48],gr_uc_w_ctrl),
  269. 'good': (utf8_text[:48],)
  270. },
  271. 'TwComment': {
  272. 'exc_name': 'BadTwComment',
  273. 'arg1': 's',
  274. 'bad': ( utf8_combining[:40],
  275. utf8_ctrl[:40],
  276. text_jp[:41],
  277. text_zh[:41],
  278. gr_uc_w_ctrl,
  279. utf8_text[:81] ),
  280. 'good': ( utf8_text[:80],
  281. (ru_uc + gr_uc + utf8_text)[:80],
  282. text_jp[:40],
  283. text_zh[:40] )
  284. },
  285. 'MMGenTxLabel':{
  286. 'arg1': 's',
  287. 'bad': (utf8_text[:73],utf8_combining[:72],utf8_ctrl[:72],gr_uc_w_ctrl),
  288. 'good': (utf8_text[:72],)
  289. },
  290. 'MMGenPWIDString': { # forbidden = list(u' :/\\')
  291. 'arg1': 's',
  292. 'bad': ('foo/','foo:','foo:\\'),
  293. 'good': ('qwerty@яяя',)
  294. },
  295. 'MMGenAddrType': {
  296. 'arg1': 'proto',
  297. 'bad': (
  298. {'proto':proto, 'id_str':'U', 'ret':'L'},
  299. {'proto':proto, 'id_str':'z', 'ret':'L'},
  300. {'proto':proto, 'id_str':'xx', 'ret':'C'},
  301. {'proto':proto, 'id_str':'dogecoin', 'ret':'C'},
  302. ),
  303. 'good': (
  304. {'proto':proto, 'id_str':'legacy', 'ret':'L'},
  305. {'proto':proto, 'id_str':'L', 'ret':'L'},
  306. {'proto':proto, 'id_str':'compressed','ret':'C'},
  307. {'proto':proto, 'id_str':'C', 'ret':'C'},
  308. {'proto':proto, 'id_str':'segwit', 'ret':'S'},
  309. {'proto':proto, 'id_str':'S', 'ret':'S'},
  310. {'proto':proto, 'id_str':'bech32', 'ret':'B'},
  311. {'proto':proto, 'id_str':'B', 'ret':'B'}
  312. )
  313. },
  314. 'MMGenPasswordType': {
  315. 'arg1': 'proto',
  316. 'bad': (
  317. {'proto':proto, 'id_str':'U', 'ret':'L'},
  318. {'proto':proto, 'id_str':'z', 'ret':'L'},
  319. {'proto':proto, 'id_str':'я', 'ret':'C'},
  320. {'proto':proto, 'id_str':1, 'ret':'C'},
  321. {'proto':proto, 'id_str':'passw0rd', 'ret':'C'},
  322. ),
  323. 'good': (
  324. {'proto':proto, 'id_str':'password', 'ret':'P'},
  325. {'proto':proto, 'id_str':'P', 'ret':'P'},
  326. )
  327. },
  328. 'SeedSplitSpecifier': {
  329. 'arg1': 's',
  330. 'bad': ('M','αβ:2',1,'0:1','1:1','2:1','3:2','1:2000','abc:0:2'),
  331. 'good': (
  332. ('1:2','2:2','alice:2:2','αβ:2:2','1:'+ssm,ssm+':'+ssm)
  333. )
  334. },
  335. 'IPPort': {
  336. 'arg1': 's',
  337. 'bad': (
  338. '.0.0.0:0',
  339. '127.0.0.1',
  340. '127.0.0.1:',
  341. '20',
  342. ':20',
  343. '127.0.0.256:0',
  344. '127.0.0.1:65536',
  345. '127.0.0.01:60000',
  346. '0.0.0.0:00',
  347. ),
  348. 'good': (
  349. (
  350. '0.0.0.0:0',
  351. '0.0.0.1:0',
  352. '0.0.1.0:0',
  353. '0.0.1.1:0',
  354. '0.1.0.0:0',
  355. '1.0.0.0:0',
  356. '10.0.0.1:0',
  357. '127.0.0.1:65535',
  358. '255.0.0.0:400',
  359. '255.255.255.255:65535',
  360. )
  361. )
  362. },
  363. }