ooxml_crypt 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (264) hide show
  1. checksums.yaml +7 -0
  2. data/Gemfile +4 -0
  3. data/LICENSE.txt +21 -0
  4. data/README.md +58 -0
  5. data/Rakefile +12 -0
  6. data/bin/console +15 -0
  7. data/bin/setup +8 -0
  8. data/ext/ooxml_crypt/extconf.rb +18 -0
  9. data/ext/ooxml_crypt/ooxml_crypt.c +27 -0
  10. data/ext/ooxml_crypt/ooxml_crypt.h +7 -0
  11. data/lib/ooxml_crypt/version.rb +5 -0
  12. data/lib/ooxml_crypt.rb +75 -0
  13. data/vendor/cybozulib/.github/workflows/main.yml +12 -0
  14. data/vendor/cybozulib/.gitignore +5 -0
  15. data/vendor/cybozulib/CMakeLists.txt +6 -0
  16. data/vendor/cybozulib/COPYRIGHT +27 -0
  17. data/vendor/cybozulib/Makefile +26 -0
  18. data/vendor/cybozulib/bin/libeay32.dll +0 -0
  19. data/vendor/cybozulib/bin/libmecab.dll +0 -0
  20. data/vendor/cybozulib/bin/ssleay32.dll +0 -0
  21. data/vendor/cybozulib/common.mk +116 -0
  22. data/vendor/cybozulib/common.props +25 -0
  23. data/vendor/cybozulib/cybozulib.sln +286 -0
  24. data/vendor/cybozulib/debug.props +14 -0
  25. data/vendor/cybozulib/include/cybozu/array.hpp +197 -0
  26. data/vendor/cybozulib/include/cybozu/atoi.hpp +238 -0
  27. data/vendor/cybozulib/include/cybozu/atomic.hpp +146 -0
  28. data/vendor/cybozulib/include/cybozu/base64.hpp +210 -0
  29. data/vendor/cybozulib/include/cybozu/benchmark.hpp +212 -0
  30. data/vendor/cybozulib/include/cybozu/bfd.hpp +105 -0
  31. data/vendor/cybozulib/include/cybozu/bit_operation.hpp +139 -0
  32. data/vendor/cybozulib/include/cybozu/bitvector.hpp +358 -0
  33. data/vendor/cybozulib/include/cybozu/condition_variable.hpp +113 -0
  34. data/vendor/cybozulib/include/cybozu/condition_variable_cs.hpp +74 -0
  35. data/vendor/cybozulib/include/cybozu/config.hpp +392 -0
  36. data/vendor/cybozulib/include/cybozu/critical_section.hpp +60 -0
  37. data/vendor/cybozulib/include/cybozu/crypto.hpp +321 -0
  38. data/vendor/cybozulib/include/cybozu/csucvector.hpp +624 -0
  39. data/vendor/cybozulib/include/cybozu/csv.hpp +294 -0
  40. data/vendor/cybozulib/include/cybozu/data_type.hpp +27 -0
  41. data/vendor/cybozulib/include/cybozu/endian.hpp +224 -0
  42. data/vendor/cybozulib/include/cybozu/env.hpp +63 -0
  43. data/vendor/cybozulib/include/cybozu/event.hpp +122 -0
  44. data/vendor/cybozulib/include/cybozu/exception.hpp +253 -0
  45. data/vendor/cybozulib/include/cybozu/file.hpp +626 -0
  46. data/vendor/cybozulib/include/cybozu/fmindex.hpp +291 -0
  47. data/vendor/cybozulib/include/cybozu/format.hpp +93 -0
  48. data/vendor/cybozulib/include/cybozu/frequency.hpp +264 -0
  49. data/vendor/cybozulib/include/cybozu/hash.hpp +67 -0
  50. data/vendor/cybozulib/include/cybozu/inttype.hpp +174 -0
  51. data/vendor/cybozulib/include/cybozu/itoa.hpp +336 -0
  52. data/vendor/cybozulib/include/cybozu/json.hpp +120 -0
  53. data/vendor/cybozulib/include/cybozu/line_stream.hpp +149 -0
  54. data/vendor/cybozulib/include/cybozu/link_libeay32.hpp +21 -0
  55. data/vendor/cybozulib/include/cybozu/link_mpir.hpp +18 -0
  56. data/vendor/cybozulib/include/cybozu/link_ssleay32.hpp +19 -0
  57. data/vendor/cybozulib/include/cybozu/log.hpp +237 -0
  58. data/vendor/cybozulib/include/cybozu/minixml.hpp +452 -0
  59. data/vendor/cybozulib/include/cybozu/mmap.hpp +143 -0
  60. data/vendor/cybozulib/include/cybozu/mutex.hpp +144 -0
  61. data/vendor/cybozulib/include/cybozu/nlp/mecab.hpp +96 -0
  62. data/vendor/cybozulib/include/cybozu/nlp/plsi.hpp +315 -0
  63. data/vendor/cybozulib/include/cybozu/nlp/random.hpp +74 -0
  64. data/vendor/cybozulib/include/cybozu/nlp/sparse.hpp +529 -0
  65. data/vendor/cybozulib/include/cybozu/nlp/svd.hpp +486 -0
  66. data/vendor/cybozulib/include/cybozu/nlp/tfidf.hpp +226 -0
  67. data/vendor/cybozulib/include/cybozu/nlp/top_score.hpp +75 -0
  68. data/vendor/cybozulib/include/cybozu/option.hpp +743 -0
  69. data/vendor/cybozulib/include/cybozu/parallel.hpp +88 -0
  70. data/vendor/cybozulib/include/cybozu/pcg.hpp +72 -0
  71. data/vendor/cybozulib/include/cybozu/process.hpp +324 -0
  72. data/vendor/cybozulib/include/cybozu/quit_signal_handler.hpp +66 -0
  73. data/vendor/cybozulib/include/cybozu/random_generator.hpp +144 -0
  74. data/vendor/cybozulib/include/cybozu/regex.hpp +463 -0
  75. data/vendor/cybozulib/include/cybozu/select8.hpp +279 -0
  76. data/vendor/cybozulib/include/cybozu/serializer.hpp +363 -0
  77. data/vendor/cybozulib/include/cybozu/sha1.hpp +209 -0
  78. data/vendor/cybozulib/include/cybozu/sha2.hpp +506 -0
  79. data/vendor/cybozulib/include/cybozu/siphash.hpp +105 -0
  80. data/vendor/cybozulib/include/cybozu/socket.hpp +785 -0
  81. data/vendor/cybozulib/include/cybozu/ssl.hpp +203 -0
  82. data/vendor/cybozulib/include/cybozu/stacktrace.hpp +291 -0
  83. data/vendor/cybozulib/include/cybozu/stream.hpp +269 -0
  84. data/vendor/cybozulib/include/cybozu/string.hpp +1746 -0
  85. data/vendor/cybozulib/include/cybozu/string_operation.hpp +365 -0
  86. data/vendor/cybozulib/include/cybozu/sucvector.hpp +378 -0
  87. data/vendor/cybozulib/include/cybozu/test.hpp +373 -0
  88. data/vendor/cybozulib/include/cybozu/thread.hpp +229 -0
  89. data/vendor/cybozulib/include/cybozu/time.hpp +281 -0
  90. data/vendor/cybozulib/include/cybozu/tls.hpp +115 -0
  91. data/vendor/cybozulib/include/cybozu/unordered_map.hpp +13 -0
  92. data/vendor/cybozulib/include/cybozu/unordered_set.hpp +13 -0
  93. data/vendor/cybozulib/include/cybozu/v128.hpp +376 -0
  94. data/vendor/cybozulib/include/cybozu/wavelet_matrix.hpp +345 -0
  95. data/vendor/cybozulib/include/cybozu/xorshift.hpp +189 -0
  96. data/vendor/cybozulib/include/cybozu/zlib.hpp +325 -0
  97. data/vendor/cybozulib/include/sais.hxx +364 -0
  98. data/vendor/cybozulib/misc/make_select8tbl.cpp +26 -0
  99. data/vendor/cybozulib/mk.bat +37 -0
  100. data/vendor/cybozulib/readme.md +29 -0
  101. data/vendor/cybozulib/release.props +12 -0
  102. data/vendor/cybozulib/sample/Makefile +30 -0
  103. data/vendor/cybozulib/sample/csucvector_smpl.cpp +42 -0
  104. data/vendor/cybozulib/sample/data/svd/org/test1.S +4 -0
  105. data/vendor/cybozulib/sample/data/svd/org/test1.U +4 -0
  106. data/vendor/cybozulib/sample/data/svd/org/test1.V +6 -0
  107. data/vendor/cybozulib/sample/data/svd/test1 +4 -0
  108. data/vendor/cybozulib/sample/data/svd/test2 +4 -0
  109. data/vendor/cybozulib/sample/desymbol.cpp +127 -0
  110. data/vendor/cybozulib/sample/exception_smpl.cpp +46 -0
  111. data/vendor/cybozulib/sample/fmindex_smpl.cpp +231 -0
  112. data/vendor/cybozulib/sample/log_smpl.cpp +19 -0
  113. data/vendor/cybozulib/sample/mecab_smpl.cpp +37 -0
  114. data/vendor/cybozulib/sample/option2_smpl.cpp +68 -0
  115. data/vendor/cybozulib/sample/option_smpl.cpp +42 -0
  116. data/vendor/cybozulib/sample/plsi_smpl.cpp +207 -0
  117. data/vendor/cybozulib/sample/proj/exception_smpl.vcproj +184 -0
  118. data/vendor/cybozulib/sample/proj/mecab_smpl.vcproj +184 -0
  119. data/vendor/cybozulib/sample/proj/ssl_smpl/ssl_smpl.vcxproj +85 -0
  120. data/vendor/cybozulib/sample/proj/ssl_smpl.vcproj +347 -0
  121. data/vendor/cybozulib/sample/proj/stacktrace_smpl/stacktrace_smpl.vcxproj +85 -0
  122. data/vendor/cybozulib/sample/proj/svd_smpl.vcproj +184 -0
  123. data/vendor/cybozulib/sample/quit_signal_handler.cpp +30 -0
  124. data/vendor/cybozulib/sample/serializer_smpl.cpp +196 -0
  125. data/vendor/cybozulib/sample/socket_smpl.cpp +82 -0
  126. data/vendor/cybozulib/sample/ssl_smpl.cpp +39 -0
  127. data/vendor/cybozulib/sample/stacktrace_smpl.cpp +52 -0
  128. data/vendor/cybozulib/sample/svd_bench_smpl.cpp +143 -0
  129. data/vendor/cybozulib/sample/svd_smpl.cpp +94 -0
  130. data/vendor/cybozulib/sample/wm_bench_smpl.cpp +182 -0
  131. data/vendor/cybozulib/sample/zlib_smpl.cpp +41 -0
  132. data/vendor/cybozulib/src/Makefile +8 -0
  133. data/vendor/cybozulib/src/base/Makefile +19 -0
  134. data/vendor/cybozulib/test/Makefile +12 -0
  135. data/vendor/cybozulib/test/base/Makefile +37 -0
  136. data/vendor/cybozulib/test/base/array_test.cpp +173 -0
  137. data/vendor/cybozulib/test/base/atoi_test.cpp +774 -0
  138. data/vendor/cybozulib/test/base/atomic_test.cpp +49 -0
  139. data/vendor/cybozulib/test/base/base64_test.cpp +113 -0
  140. data/vendor/cybozulib/test/base/bit_operation_test.cpp +134 -0
  141. data/vendor/cybozulib/test/base/bitvector_test.cpp +204 -0
  142. data/vendor/cybozulib/test/base/condition_variable_cs_test.cpp +92 -0
  143. data/vendor/cybozulib/test/base/condition_variable_test.cpp +88 -0
  144. data/vendor/cybozulib/test/base/config_test.cpp +236 -0
  145. data/vendor/cybozulib/test/base/crypto_test.cpp +122 -0
  146. data/vendor/cybozulib/test/base/csucvector_test.cpp +63 -0
  147. data/vendor/cybozulib/test/base/csv_test.cpp +182 -0
  148. data/vendor/cybozulib/test/base/data/a.xml +26 -0
  149. data/vendor/cybozulib/test/base/endian_test.cpp +56 -0
  150. data/vendor/cybozulib/test/base/env_test.cpp +22 -0
  151. data/vendor/cybozulib/test/base/event_test.cpp +41 -0
  152. data/vendor/cybozulib/test/base/file_test.cpp +233 -0
  153. data/vendor/cybozulib/test/base/fmindex_test.cpp +118 -0
  154. data/vendor/cybozulib/test/base/format_test.cpp +12 -0
  155. data/vendor/cybozulib/test/base/frequency_test.cpp +104 -0
  156. data/vendor/cybozulib/test/base/itoa_test.cpp +522 -0
  157. data/vendor/cybozulib/test/base/line_stream_test.cpp +208 -0
  158. data/vendor/cybozulib/test/base/mecab_test.cpp +41 -0
  159. data/vendor/cybozulib/test/base/minixml_test.cpp +103 -0
  160. data/vendor/cybozulib/test/base/mmap_test.cpp +15 -0
  161. data/vendor/cybozulib/test/base/option_test.cpp +487 -0
  162. data/vendor/cybozulib/test/base/parallel_test.cpp +48 -0
  163. data/vendor/cybozulib/test/base/proj/array_test/array_test.vcxproj +86 -0
  164. data/vendor/cybozulib/test/base/proj/atoi_test/atoi_test.vcxproj +86 -0
  165. data/vendor/cybozulib/test/base/proj/atomic_test/atomic_test.vcxproj +86 -0
  166. data/vendor/cybozulib/test/base/proj/base64_test/base64_test.vcxproj +86 -0
  167. data/vendor/cybozulib/test/base/proj/condition_variable_cs_test/condition_variable_cs_test.vcxproj +86 -0
  168. data/vendor/cybozulib/test/base/proj/condition_variable_test/condition_variable_test.vcxproj +86 -0
  169. data/vendor/cybozulib/test/base/proj/config_test/config_test.vcxproj +86 -0
  170. data/vendor/cybozulib/test/base/proj/csv_test/csv_test.vcxproj +86 -0
  171. data/vendor/cybozulib/test/base/proj/endian_test/endian_test.vcxproj +86 -0
  172. data/vendor/cybozulib/test/base/proj/env_test/env_test.vcxproj +86 -0
  173. data/vendor/cybozulib/test/base/proj/event_test/event_test.vcxproj +86 -0
  174. data/vendor/cybozulib/test/base/proj/file_test/file_test.vcxproj +86 -0
  175. data/vendor/cybozulib/test/base/proj/itoa_test/itoa_test.vcxproj +86 -0
  176. data/vendor/cybozulib/test/base/proj/mecab_test/mecab_test.vcxproj +88 -0
  177. data/vendor/cybozulib/test/base/proj/minixml_test/minixml_test.vcxproj +86 -0
  178. data/vendor/cybozulib/test/base/proj/mmap_test/mmap_test.vcxproj +86 -0
  179. data/vendor/cybozulib/test/base/proj/serializer_test/serializer_test.vcxproj +86 -0
  180. data/vendor/cybozulib/test/base/proj/sha1_test/sha1_test.vcxproj +86 -0
  181. data/vendor/cybozulib/test/base/proj/stream_test/stream_test.vcxproj +86 -0
  182. data/vendor/cybozulib/test/base/proj/string_operation_test/string_operation_test.vcxproj +86 -0
  183. data/vendor/cybozulib/test/base/proj/string_test/string_test.vcxproj +86 -0
  184. data/vendor/cybozulib/test/base/proj/thread_test/thread_test.vcxproj +86 -0
  185. data/vendor/cybozulib/test/base/proj/time_test/time_test.vcxproj +86 -0
  186. data/vendor/cybozulib/test/base/proj/tls_test/tls_test.vcxproj +86 -0
  187. data/vendor/cybozulib/test/base/proj/zlib_test/zlib_test.vcxproj +86 -0
  188. data/vendor/cybozulib/test/base/random_generator_test.cpp +28 -0
  189. data/vendor/cybozulib/test/base/regex_test.cpp +74 -0
  190. data/vendor/cybozulib/test/base/serializer_test.cpp +483 -0
  191. data/vendor/cybozulib/test/base/sha1_test.cpp +61 -0
  192. data/vendor/cybozulib/test/base/sha2_test.cpp +191 -0
  193. data/vendor/cybozulib/test/base/siphash_test.cpp +33 -0
  194. data/vendor/cybozulib/test/base/socket_test.cpp +76 -0
  195. data/vendor/cybozulib/test/base/stream_test.cpp +101 -0
  196. data/vendor/cybozulib/test/base/string_operation_test.cpp +340 -0
  197. data/vendor/cybozulib/test/base/string_test.cpp +1705 -0
  198. data/vendor/cybozulib/test/base/sucvector_test.cpp +312 -0
  199. data/vendor/cybozulib/test/base/thread_test.cpp +62 -0
  200. data/vendor/cybozulib/test/base/time_test.cpp +164 -0
  201. data/vendor/cybozulib/test/base/tls_test.cpp +50 -0
  202. data/vendor/cybozulib/test/base/wavelet_matrix_test.cpp +145 -0
  203. data/vendor/cybozulib/test/base/zlib_test.cpp +371 -0
  204. data/vendor/cybozulib/test/nlp/Makefile +27 -0
  205. data/vendor/cybozulib/test/nlp/proj/random_test.vcproj +184 -0
  206. data/vendor/cybozulib/test/nlp/proj/sparse_test.vcproj +184 -0
  207. data/vendor/cybozulib/test/nlp/proj/svd_test.vcproj +184 -0
  208. data/vendor/cybozulib/test/nlp/random_test.cpp +62 -0
  209. data/vendor/cybozulib/test/nlp/sparse_test.cpp +347 -0
  210. data/vendor/cybozulib/test/nlp/svd_test.cpp +234 -0
  211. data/vendor/cybozulib/test/nlp/top_score_test.cpp +40 -0
  212. data/vendor/cybozulib/tool/create_vcproj.py +186 -0
  213. data/vendor/cybozulib/tool/vcproj_tmpl.py +185 -0
  214. data/vendor/msoffice/COPYRIGHT +27 -0
  215. data/vendor/msoffice/Makefile +29 -0
  216. data/vendor/msoffice/bin/64/msoc.dll +0 -0
  217. data/vendor/msoffice/bin/64/msocsample.exe +0 -0
  218. data/vendor/msoffice/bin/64/msoffice-crypt.exe +0 -0
  219. data/vendor/msoffice/bin/msoc.dll +0 -0
  220. data/vendor/msoffice/bin/msocsample.exe +0 -0
  221. data/vendor/msoffice/bin/msoffice-crypt.exe +0 -0
  222. data/vendor/msoffice/common.mk +71 -0
  223. data/vendor/msoffice/common.props +26 -0
  224. data/vendor/msoffice/debug.props +14 -0
  225. data/vendor/msoffice/include/attack.hpp +211 -0
  226. data/vendor/msoffice/include/cfb.hpp +777 -0
  227. data/vendor/msoffice/include/crypto_util.hpp +450 -0
  228. data/vendor/msoffice/include/custom_sha1.hpp +342 -0
  229. data/vendor/msoffice/include/decode.hpp +240 -0
  230. data/vendor/msoffice/include/encode.hpp +221 -0
  231. data/vendor/msoffice/include/make_dataspace.hpp +316 -0
  232. data/vendor/msoffice/include/msoc.h +129 -0
  233. data/vendor/msoffice/include/resource.hpp +7 -0
  234. data/vendor/msoffice/include/standard_encryption.hpp +145 -0
  235. data/vendor/msoffice/include/uint32vec.hpp +179 -0
  236. data/vendor/msoffice/include/util.hpp +212 -0
  237. data/vendor/msoffice/lib/.emptydir +0 -0
  238. data/vendor/msoffice/misc/decrypt-xls.vbs +46 -0
  239. data/vendor/msoffice/mk.bat +1 -0
  240. data/vendor/msoffice/mkdll.bat +3 -0
  241. data/vendor/msoffice/msoc.def +13 -0
  242. data/vendor/msoffice/msocsample.py +178 -0
  243. data/vendor/msoffice/msoffice12.sln +31 -0
  244. data/vendor/msoffice/readme.md +110 -0
  245. data/vendor/msoffice/release.props +28 -0
  246. data/vendor/msoffice/src/Makefile +19 -0
  247. data/vendor/msoffice/src/attack.cpp +124 -0
  248. data/vendor/msoffice/src/cfb_test.cpp +77 -0
  249. data/vendor/msoffice/src/minisample.c +54 -0
  250. data/vendor/msoffice/src/msocdll.cpp +276 -0
  251. data/vendor/msoffice/src/msocsample.c +136 -0
  252. data/vendor/msoffice/src/msoffice-crypt.cpp +219 -0
  253. data/vendor/msoffice/src/proj/attack/attack.vcxproj +88 -0
  254. data/vendor/msoffice/src/proj/main/msoffice-crypt.vcxproj +88 -0
  255. data/vendor/msoffice/src/sha1.cpp +234 -0
  256. data/vendor/msoffice/test/Makefile +20 -0
  257. data/vendor/msoffice/test/cfb_test.cpp +74 -0
  258. data/vendor/msoffice/test/hash_test.cpp +59 -0
  259. data/vendor/msoffice/test/proj/cfb/cfb_test.vcxproj +90 -0
  260. data/vendor/msoffice/test/proj/hash/hash_test.vcxproj +90 -0
  261. data/vendor/msoffice/test/sampl.bat +8 -0
  262. data/vendor/msoffice/test_all.py +46 -0
  263. data/vendor/update +4 -0
  264. metadata +351 -0
@@ -0,0 +1,145 @@
1
+ #include <cybozu/test.hpp>
2
+ #include <cybozu/wavelet_matrix.hpp>
3
+ #include <algorithm>
4
+ #include <cybozu/xorshift.hpp>
5
+ #include <cybozu/benchmark.hpp>
6
+
7
+ template<class T>
8
+ size_t selectC(const T *a, size_t N, uint32_t val, size_t n)
9
+ {
10
+ n++;
11
+ for (size_t i = 0; i < N; i++) {
12
+ if (a[i] == val) n--;
13
+ if (n == 0) return i;
14
+ }
15
+ return cybozu::NotFound;
16
+ }
17
+
18
+ template<class T>
19
+ size_t rankC(const T *a, size_t N, uint32_t val)
20
+ {
21
+ return std::count(a, a + N, val);
22
+ }
23
+
24
+ template<class T>
25
+ size_t rankLtC(const T *a, size_t N, uint32_t val)
26
+ {
27
+ size_t ret = 0;
28
+ for (size_t i = 0; i < N; i++) {
29
+ if (a[i] < val) ret++;
30
+ }
31
+ return ret;
32
+ }
33
+
34
+ CYBOZU_TEST_AUTO(mini)
35
+ {
36
+ std::vector<uint32_t> v;
37
+ const uint32_t vTbl[] = { 4, 7, 6, 5, 3, 2, 1, 0, 1, 4, 1, 7 };
38
+ const size_t vn = CYBOZU_NUM_OF_ARRAY(vTbl);
39
+ for (size_t i = 0; i < vn; i++) {
40
+ v.push_back(vTbl[i]);
41
+ }
42
+ cybozu::WaveletMatrix wm;
43
+ wm.init(v, 3);
44
+ for (size_t i = 0; i < vn; i++) {
45
+ CYBOZU_TEST_EQUAL(wm.get(i), vTbl[i]);
46
+ }
47
+
48
+ for (uint32_t val = 0; val <= 7; val++) {
49
+ for (size_t pos = 0; pos < vn; pos++) {
50
+ size_t a = wm.rank(val, pos);
51
+ size_t b = rankC(vTbl, pos, val);
52
+ CYBOZU_TEST_EQUAL(a, b);
53
+ a = wm.rankLt(val, pos);
54
+ b = rankLtC(vTbl, pos, val);
55
+ CYBOZU_TEST_EQUAL(a, b);
56
+ }
57
+ }
58
+ for (int r = 0; r < 3; r++) {
59
+ for (int val = 0; val < 8; val++) {
60
+ size_t a = wm.select(val, r);
61
+ size_t b = selectC(vTbl, vn, val, r);
62
+ CYBOZU_TEST_EQUAL(a, b);
63
+ }
64
+ }
65
+ }
66
+
67
+ void testSub(const cybozu::WaveletMatrix& wm, const std::vector<uint32_t>& v, uint32_t maxVal, size_t valBitLen)
68
+ {
69
+ const size_t vn = v.size();
70
+ for (size_t i = 0; i < vn; i++) {
71
+ CYBOZU_TEST_EQUAL(wm.get(i), v[i]);
72
+ uint32_t val;
73
+ uint64_t p1 = wm.get(&val, i);
74
+ uint64_t p2 = wm.rank(v[i], i);
75
+ CYBOZU_TEST_EQUAL(val, v[i]);
76
+ CYBOZU_TEST_EQUAL(p1, p2);
77
+ }
78
+ for (uint32_t val = 0; val < maxVal; val++) {
79
+ for (size_t pos = 0; pos < vn; pos++) {
80
+ size_t a = wm.rank(val, pos);
81
+ size_t b = std::count(v.begin(), v.begin() + pos, val);
82
+ CYBOZU_TEST_EQUAL(a, b);
83
+ }
84
+ CYBOZU_TEST_EQUAL(wm.rank(val, vn), (uint64_t)std::count(v.begin(), v.end(), val));
85
+ }
86
+ for (size_t r = 0; r < valBitLen; r++) {
87
+ for (uint32_t val = 0; val < maxVal; val++) {
88
+ size_t a = wm.select(val, r);
89
+ size_t b = selectC(&v[0], vn, val, r);
90
+ CYBOZU_TEST_EQUAL(a, b);
91
+ }
92
+ }
93
+ }
94
+
95
+ CYBOZU_TEST_AUTO(large_load_save)
96
+ {
97
+ cybozu::XorShift rg;
98
+ std::vector<uint32_t> v;
99
+ const size_t vn = 5000;
100
+ const size_t valBitLen = 8;
101
+ const uint32_t maxVal = 1 << valBitLen;
102
+ v.resize(vn);
103
+ for (size_t i = 0; i < vn; i++) {
104
+ v[i] = rg() % maxVal;
105
+ }
106
+ cybozu::WaveletMatrix wm;
107
+ wm.init(v, valBitLen);
108
+ testSub(wm, v, maxVal, valBitLen);
109
+ std::string data;
110
+ {
111
+ std::ostringstream os;
112
+ wm.save(os);
113
+ data = os.str();
114
+ }
115
+ {
116
+ std::istringstream is(data);
117
+ cybozu::WaveletMatrix wm2;
118
+ wm2.load(is);
119
+ testSub(wm2, v, maxVal, valBitLen);
120
+ }
121
+ #ifndef NDEUBG
122
+ size_t x = 0;
123
+ cybozu::CpuClock clk;
124
+ for (uint32_t val = 0; val < maxVal; val++) {
125
+ clk.begin();
126
+ for (size_t pos = 0; pos < vn; pos++) {
127
+ x += wm.rank(val, pos);
128
+ }
129
+ clk.end();
130
+ }
131
+ printf("wm.rank %08x %5.2fKclk\n", (int)x, clk.getClock() / double(clk.getCount() * vn) * 1e-3);
132
+ x = 0;
133
+ const int N = 3;
134
+ for (int i = 0; i < N; i++) {
135
+ for (size_t r = 0; r < valBitLen; r++) {
136
+ clk.begin();
137
+ for (uint32_t val = 0; val < maxVal; val++) {
138
+ x += wm.select(val, r);
139
+ }
140
+ clk.end();
141
+ }
142
+ }
143
+ printf("wm.select %08x %5.2fKclk\n", (int)x, clk.getClock() / double(clk.getCount() * maxVal) * 1e-3);
144
+ #endif
145
+ }
@@ -0,0 +1,371 @@
1
+ #define CYBOZU_TEST_DISABLE_AUTO_RUN
2
+ #include <cybozu/test.hpp>
3
+ #include <cybozu/zlib.hpp>
4
+ #include <cybozu/file.hpp>
5
+ #include <cybozu/stream.hpp>
6
+ #include <cybozu/mmap.hpp>
7
+ #include <cybozu/serializer.hpp>
8
+ #include <cybozu/nlp/sparse.hpp>
9
+ #include <sstream>
10
+ #include <fstream>
11
+ #include <string>
12
+ #include <vector>
13
+ #include <map>
14
+
15
+ const char *g_testFile = "zlib_test.log";
16
+ const char *g_extraFile = 0;
17
+ const char *g_compressFile = 0;
18
+
19
+ CYBOZU_TEST_AUTO(test1_deflate)
20
+ {
21
+ typedef cybozu::ZlibCompressorT<std::stringstream> Compressor;
22
+ typedef cybozu::ZlibDecompressorT<std::stringstream> Decompressor;
23
+ const std::string in = "this is a pen";
24
+
25
+ std::stringstream os;
26
+ {
27
+ Compressor comp(os);
28
+ comp.write(in.c_str(), in.size());
29
+ comp.flush();
30
+ }
31
+ const std::string enc = os.str();
32
+
33
+ const char encTbl[] = "\x78\x9c\x2b\xc9\xc8\x2c\x56\x00\xa2\x44\x85\x82\xd4\x3c\x00\x21\x0c\x04\x99";
34
+
35
+ const std::string encOk(encTbl, encTbl + sizeof(encTbl) - 1);
36
+ CYBOZU_TEST_EQUAL(enc, encOk);
37
+
38
+ std::stringstream is;
39
+ is << encOk;
40
+
41
+ {
42
+ Decompressor dec(is);
43
+ char buf[2048];
44
+ std::string out;
45
+ for (;;) {
46
+ size_t size = dec.readSome(buf, sizeof(buf));
47
+ if (size == 0) break;
48
+ out.append(buf, buf + size);
49
+ }
50
+ CYBOZU_TEST_EQUAL(in, out);
51
+ }
52
+ }
53
+
54
+ CYBOZU_TEST_AUTO(test2_deflate)
55
+ {
56
+ typedef cybozu::ZlibCompressorT<std::stringstream, 1> Compressor;
57
+ typedef cybozu::ZlibDecompressorT<std::stringstream, 1> Decompressor;
58
+ const std::string in = "this is a pen";
59
+
60
+ std::stringstream os;
61
+ {
62
+ Compressor comp(os);
63
+ for (size_t i = 0; i < in.size(); i++) {
64
+ comp.write(&in[i], 1);
65
+ }
66
+ comp.flush();
67
+ }
68
+ const std::string enc = os.str();
69
+
70
+ const char encTbl[] = "\x78\x9c\x2b\xc9\xc8\x2c\x56\x00\xa2\x44\x85\x82\xd4\x3c\x00\x21\x0c\x04\x99";
71
+
72
+ const std::string encOk(encTbl, encTbl + sizeof(encTbl) - 1);
73
+ CYBOZU_TEST_EQUAL(enc, encOk);
74
+
75
+ std::stringstream is;
76
+ is << encOk;
77
+
78
+ {
79
+ Decompressor dec(is);
80
+ char buf[2048];
81
+ std::string out;
82
+ for (;;) {
83
+ size_t size = dec.readSome(buf, sizeof(buf));
84
+ if (size == 0) break;
85
+ out.append(buf, buf + size);
86
+ }
87
+ CYBOZU_TEST_EQUAL(in, out);
88
+ }
89
+ }
90
+
91
+ CYBOZU_TEST_AUTO(enc_and_dec)
92
+ {
93
+ std::string body = "From: cybozu\r\n"
94
+ "To: cybozu\r\n"
95
+ "\r\n"
96
+ "hello with zlib compressed\r\n"
97
+ ".\r\n";
98
+ std::stringstream ss;
99
+ cybozu::ZlibCompressorT<std::stringstream> comp(ss);
100
+ comp.write(&body[0], body.size());
101
+ comp.flush();
102
+
103
+ std::string enc = ss.str();
104
+
105
+ cybozu::StringInputStream ims(enc);
106
+
107
+ cybozu::ZlibDecompressorT<cybozu::StringInputStream> dec(ims);
108
+ char buf[4096];
109
+ size_t size = dec.readSome(buf, sizeof(buf));
110
+ std::string decStr(buf, buf + size);
111
+ CYBOZU_TEST_EQUAL(body, decStr);
112
+ }
113
+
114
+ CYBOZU_TEST_AUTO(output_gzip1)
115
+ {
116
+ std::string str = "Only a test, test, test, test, test, test, test, test!\n";
117
+ std::stringstream ss;
118
+ cybozu::ZlibCompressorT<std::stringstream> comp(ss, true);
119
+ comp.write(&str[0], str.size());
120
+ comp.flush();
121
+ std::string enc = ss.str();
122
+
123
+ {
124
+ cybozu::ZlibDecompressorT<std::stringstream> dec(ss, true);
125
+ char buf[4096];
126
+ size_t size = dec.readSome(buf, sizeof(buf));
127
+ std::string decStr(buf, buf + size);
128
+ CYBOZU_TEST_EQUAL(decStr, str);
129
+ }
130
+
131
+ {
132
+ cybozu::StringInputStream is(enc);
133
+ cybozu::ZlibDecompressorT<cybozu::StringInputStream> dec(is, true);
134
+ char buf[4096];
135
+ size_t size = dec.readSome(buf, sizeof(buf));
136
+ std::string decStr(buf, buf + size);
137
+ CYBOZU_TEST_EQUAL(decStr, str);
138
+ }
139
+ }
140
+
141
+ #ifdef _MSC_VER
142
+ #pragma warning(disable : 4309)
143
+ #endif
144
+
145
+ CYBOZU_TEST_AUTO(output_gzip2)
146
+ {
147
+ const uint8_t textBufTbl[] = {
148
+ 0x23, 0x69, 0x6E, 0x63, 0x6C, 0x75, 0x64, 0x65, 0x20, 0x3C, 0x73, 0x74, 0x64, 0x69, 0x6F, 0x2E,
149
+ 0x68, 0x3E, 0x0A, 0x0A, 0x69, 0x6E, 0x74, 0x20, 0x6D, 0x61, 0x69, 0x6E, 0x28, 0x29, 0x0A, 0x7B,
150
+ 0x0A, 0x09, 0x70, 0x75, 0x74, 0x73, 0x28, 0x22, 0x68, 0x65, 0x6C, 0x6C, 0x6F, 0x22, 0x29, 0x3B,
151
+ 0x0A, 0x09, 0x72, 0x65, 0x74, 0x75, 0x72, 0x6E, 0x20, 0x30, 0x3B, 0x0A, 0x7D, 0x0A, 0x0A,
152
+ };
153
+
154
+ const uint8_t encBufTbl[] = {
155
+ 0x1F, 0x8B, 0x08, 0x08, 0xA4, 0x7E, 0x20, 0x4B, 0x00, 0x03, 0x74, 0x2E, 0x63, 0x70, 0x70, 0x00,
156
+ 0x53, 0xCE, 0xCC, 0x4B, 0xCE, 0x29, 0x4D, 0x49, 0x55, 0xB0, 0x29, 0x2E, 0x49, 0xC9, 0xCC, 0xD7,
157
+ 0xCB, 0xB0, 0xE3, 0xE2, 0xCA, 0xCC, 0x2B, 0x51, 0xC8, 0x4D, 0xCC, 0xCC, 0xD3, 0xD0, 0xE4, 0xAA,
158
+ 0xE6, 0xE2, 0x2C, 0x28, 0x2D, 0x29, 0xD6, 0x50, 0xCA, 0x48, 0xCD, 0xC9, 0xC9, 0x57, 0xD2, 0xB4,
159
+ 0xE6, 0xE2, 0x2C, 0x4A, 0x2D, 0x29, 0x2D, 0xCA, 0x53, 0x30, 0xB0, 0xE6, 0xAA, 0xE5, 0xE2, 0x02,
160
+ 0x00, 0x48, 0xAB, 0x48, 0x61, 0x3F, 0x00, 0x00, 0x00,
161
+ };
162
+ const char *const textBuf = cybozu::cast<const char*>(textBufTbl);
163
+ const char *const encBuf = cybozu::cast<const char*>(encBufTbl);
164
+ const std::string text(textBuf, textBuf + sizeof(textBufTbl));
165
+
166
+ {
167
+ std::stringstream ss;
168
+ ss.write(encBuf, sizeof(encBufTbl));
169
+ cybozu::ZlibDecompressorT<std::stringstream> dec(ss, true);
170
+ char buf[4096];
171
+ size_t size = dec.readSome(buf, sizeof(buf));
172
+ std::string decStr(buf, buf + size);
173
+ CYBOZU_TEST_EQUAL(decStr, text);
174
+ }
175
+ }
176
+
177
+ template<class Map>
178
+ void compareMap(const Map& x, const Map& y)
179
+ {
180
+ CYBOZU_TEST_EQUAL(x.size(), y.size());
181
+ for (typename Map::const_iterator i = x.begin(), ie = x.end(), j = y.begin(); i != ie; ++i, ++j) {
182
+ CYBOZU_TEST_EQUAL(i->first, j->first);
183
+ CYBOZU_TEST_EQUAL(i->second, j->second);
184
+ }
185
+ }
186
+
187
+ CYBOZU_TEST_AUTO(serializer_with_zlib)
188
+ {
189
+ typedef std::map<int, double> Map;
190
+ Map src;
191
+ for (int i = 0; i < 100; i++) {
192
+ src[i * i] = (i + i * i) / 3.42;
193
+ }
194
+ std::stringstream ss;
195
+ {
196
+ cybozu::ZlibCompressorT<std::stringstream> enc(ss);
197
+ cybozu::save(enc, src);
198
+ }
199
+ {
200
+ cybozu::ZlibDecompressorT<std::stringstream> dec(ss);
201
+ Map dst;
202
+ cybozu::load(dst, dec);
203
+ compareMap(src, dst);
204
+ }
205
+ {
206
+ std::ofstream ofs(g_testFile, std::ios::binary);
207
+ cybozu::ZlibCompressorT<std::ofstream> enc(ofs);
208
+ cybozu::save(enc, src);
209
+ }
210
+ {
211
+ std::ifstream ifs(g_testFile, std::ios::binary);
212
+ cybozu::ZlibDecompressorT<std::ifstream> dec(ifs);
213
+ Map dst;
214
+ cybozu::load(dst, dec);
215
+ compareMap(src, dst);
216
+ }
217
+ }
218
+
219
+ CYBOZU_TEST_AUTO(sparse_with_zlib)
220
+ {
221
+ typedef cybozu::nlp::SparseVector<double> Vec;
222
+ Vec x;
223
+ const struct {
224
+ int pos;
225
+ double val;
226
+ } tbl[] = {
227
+ { 3, 1.2 }, { 5, 9.4 }, { 8, 123.4 }, { 999, -324.0 },
228
+ };
229
+ for (size_t i = 0; i < CYBOZU_NUM_OF_ARRAY(tbl); i++) {
230
+ x.push_back(tbl[i].pos, tbl[i].val);
231
+ }
232
+ {
233
+ std::ofstream ofs(g_testFile, std::ios::binary);
234
+ cybozu::ZlibCompressorT<std::ofstream> enc(ofs);
235
+ cybozu::save(enc, x);
236
+ }
237
+ {
238
+ Vec y;
239
+ std::ifstream ifs(g_testFile, std::ios::binary);
240
+ cybozu::ZlibDecompressorT<std::ifstream> dec(ifs);
241
+ cybozu::load(y, dec);
242
+ // CYBOZU_TEST_ASSERT(x == y);
243
+ CYBOZU_TEST_EQUAL(x.size(), y.size());
244
+ Vec::const_iterator xi = x.begin();
245
+ Vec::const_iterator yi = y.begin();
246
+ while (xi != x.end()) {
247
+ CYBOZU_TEST_EQUAL(xi->pos(), yi->pos());
248
+ CYBOZU_TEST_EQUAL(xi->val(), yi->val());
249
+ ++xi;
250
+ ++yi;
251
+ }
252
+ CYBOZU_TEST_ASSERT(yi == y.end());
253
+ }
254
+ }
255
+
256
+ CYBOZU_TEST_AUTO(random)
257
+ {
258
+ std::vector<char> out;
259
+ std::string in;
260
+ std::vector<char> out2;
261
+
262
+ for (size_t inSize = 12000; inSize < 14000; inSize += 3) {
263
+ in.resize(inSize);
264
+ out.resize(inSize + 100);
265
+ for (size_t i = 0; i < inSize; i++) {
266
+ in[i] = char((3 * i * i + i) % 1905);
267
+ }
268
+ cybozu::MemoryOutputStream os(out.data(), out.size());
269
+ {
270
+ cybozu::ZlibCompressorT<cybozu::MemoryOutputStream> enc(os, false, Z_DEFAULT_COMPRESSION);
271
+ enc.write(in.data(), inSize);
272
+ enc.flush();
273
+ out.resize(os.getPos());
274
+ }
275
+ {
276
+ std::string ok;
277
+ ok.resize(in.size());
278
+ size_t okSize = cybozu::ZlibCompress(&ok[0], ok.size(), in.data(), in.size());
279
+ CYBOZU_TEST_EQUAL(okSize, out.size());
280
+ CYBOZU_TEST_ASSERT(memcmp(ok.data(), out.data(), okSize) == 0);
281
+ }
282
+ {
283
+ std::string dec;
284
+ dec.resize(in.size() + 100);
285
+ size_t decSize = cybozu::ZlibUncompress(&dec[0], dec.size(), out.data(), out.size());
286
+ dec.resize(decSize);
287
+ CYBOZU_TEST_EQUAL(decSize, in.size());
288
+ CYBOZU_TEST_EQUAL(dec, in);
289
+ }
290
+ out2.resize(in.size() + 100);
291
+ size_t outSize = out.size();
292
+ {
293
+ cybozu::MemoryInputStream is(out.data(), outSize);
294
+ cybozu::ZlibDecompressorT<cybozu::MemoryInputStream> dec(is);
295
+ char *const top = (char*)out2.data();
296
+ size_t pos = 0;
297
+ for (;;) {
298
+ size_t readSize = dec.readSome(top + pos, out2.size() - pos);
299
+ if (readSize == 0) break;
300
+ pos += readSize;
301
+ }
302
+ CYBOZU_TEST_ASSERT(dec.isEmpty());
303
+ CYBOZU_TEST_ASSERT(memcmp(in.data(), out2.data(), pos) == 0);
304
+ }
305
+ }
306
+ }
307
+
308
+ CYBOZU_TEST_AUTO(extra)
309
+ {
310
+ if (g_extraFile == 0) return;
311
+ cybozu::Mmap f(g_extraFile);
312
+ printf("extraFile=%s\n", g_extraFile);
313
+ std::string comp;
314
+ std::string decomp;
315
+ {
316
+ cybozu::MemoryInputStream is(f.get(), f.size());
317
+ cybozu::StringOutputStream os(comp);
318
+ cybozu::ZlibCompressorT<cybozu::StringOutputStream> enc(os);
319
+ enc.write(f.get(), f.size());
320
+ }
321
+ if (g_compressFile) {
322
+ std::ofstream ofs(g_compressFile, std::ios::binary);
323
+ ofs.write(comp.data(), comp.size());
324
+ }
325
+
326
+ {
327
+ std::string ok;
328
+ ok.resize(f.size());
329
+ size_t okSize = cybozu::ZlibCompress(&ok[0], ok.size(), f.get(), f.size());
330
+ ok.resize(okSize);
331
+ CYBOZU_TEST_EQUAL(okSize, comp.size());
332
+ CYBOZU_TEST_EQUAL(ok, comp);
333
+ }
334
+
335
+ const size_t decompSize = f.size() + 100;
336
+ decomp.resize(decompSize);
337
+ {
338
+ cybozu::StringInputStream is(comp);
339
+ cybozu::ZlibDecompressorT<cybozu::StringInputStream> dec(is);
340
+ char *top = &decomp[0];
341
+ size_t pos = 0;
342
+ for (;;) {
343
+ size_t readSize = dec.readSome(top + pos, decompSize - pos);
344
+ if (readSize == 0) break;
345
+ pos += readSize;
346
+ }
347
+ CYBOZU_TEST_ASSERT(dec.isEmpty());
348
+ decomp.resize(pos);
349
+ CYBOZU_TEST_EQUAL(pos, f.size());
350
+ CYBOZU_TEST_ASSERT(memcmp(f.get(), decomp.data(), f.size()) == 0);
351
+ }
352
+ {
353
+ std::string ok;
354
+ ok.resize(decomp.size() + 100);
355
+ size_t okSize = cybozu::ZlibUncompress(&ok[0], ok.size(), comp.data(), comp.size());
356
+ ok.resize(okSize);
357
+ CYBOZU_TEST_EQUAL(okSize, decomp.size());
358
+ CYBOZU_TEST_ASSERT(ok == decomp);
359
+ }
360
+ }
361
+
362
+ int main(int argc, char *argv[])
363
+ {
364
+ if (argc >= 2) {
365
+ g_extraFile = argv[1];
366
+ }
367
+ if (argc == 3) {
368
+ g_compressFile = argv[2];
369
+ }
370
+ return cybozu::test::autoRun.run(argc, argv);
371
+ }
@@ -0,0 +1,27 @@
1
+ include ../../common.mk
2
+
3
+ TARGET=$(TEST_FILE)
4
+ #LIBS=$(TOPDIR)lib/libcybozu_base$(OBJSUF).a
5
+ LIBS=
6
+
7
+ SRC=$(wildcard *.cpp)
8
+ ifneq ($(HAS_EIGEN),1)
9
+ SRC:=$(filter-out svd_test.cpp,$(SRC))
10
+ endif
11
+
12
+ all: $(TARGET)
13
+
14
+ test: $(TARGET)
15
+ @$(UNIT_TEST)
16
+
17
+ $(OBJDIR):
18
+ @$(MKDIR) $(OBJDIR)
19
+
20
+ clean:
21
+ $(CLEAN)
22
+
23
+ $(LIBS):
24
+ $(MAKE) -C ../../src
25
+
26
+ -include $(DEPEND_FILE)
27
+