librex 0.0.13 → 0.0.15

Sign up to get free protection for your applications and to get access to all the features.
Files changed (435) hide show
  1. data/README.markdown +1 -1
  2. data/Rakefile +1 -0
  3. metadata +3 -435
  4. data/lib/rex/LICENSE +0 -29
  5. data/lib/rex/arch.rb +0 -103
  6. data/lib/rex/arch/sparc.rb +0 -75
  7. data/lib/rex/arch/sparc.rb.ut.rb +0 -18
  8. data/lib/rex/arch/x86.rb +0 -513
  9. data/lib/rex/arch/x86.rb.ut.rb +0 -93
  10. data/lib/rex/assembly/nasm.rb +0 -104
  11. data/lib/rex/assembly/nasm.rb.ut.rb +0 -22
  12. data/lib/rex/codepage.map +0 -104
  13. data/lib/rex/compat.rb +0 -311
  14. data/lib/rex/constants.rb +0 -113
  15. data/lib/rex/elfparsey.rb +0 -11
  16. data/lib/rex/elfparsey/elf.rb +0 -123
  17. data/lib/rex/elfparsey/elfbase.rb +0 -258
  18. data/lib/rex/elfparsey/exceptions.rb +0 -27
  19. data/lib/rex/elfscan.rb +0 -12
  20. data/lib/rex/elfscan/scanner.rb +0 -207
  21. data/lib/rex/elfscan/search.rb +0 -46
  22. data/lib/rex/encoder/alpha2.rb +0 -31
  23. data/lib/rex/encoder/alpha2/alpha_mixed.rb +0 -68
  24. data/lib/rex/encoder/alpha2/alpha_upper.rb +0 -79
  25. data/lib/rex/encoder/alpha2/generic.rb +0 -114
  26. data/lib/rex/encoder/alpha2/unicode_mixed.rb +0 -117
  27. data/lib/rex/encoder/alpha2/unicode_upper.rb +0 -129
  28. data/lib/rex/encoder/ndr.rb +0 -89
  29. data/lib/rex/encoder/ndr.rb.ut.rb +0 -44
  30. data/lib/rex/encoder/nonalpha.rb +0 -61
  31. data/lib/rex/encoder/nonupper.rb +0 -64
  32. data/lib/rex/encoder/xdr.rb +0 -106
  33. data/lib/rex/encoder/xdr.rb.ut.rb +0 -29
  34. data/lib/rex/encoder/xor.rb +0 -69
  35. data/lib/rex/encoder/xor/dword.rb +0 -13
  36. data/lib/rex/encoder/xor/dword_additive.rb +0 -13
  37. data/lib/rex/encoders/xor_dword.rb +0 -35
  38. data/lib/rex/encoders/xor_dword_additive.rb +0 -53
  39. data/lib/rex/encoders/xor_dword_additive.rb.ut.rb +0 -12
  40. data/lib/rex/encoding/xor.rb +0 -20
  41. data/lib/rex/encoding/xor.rb.ts.rb +0 -14
  42. data/lib/rex/encoding/xor/byte.rb +0 -15
  43. data/lib/rex/encoding/xor/byte.rb.ut.rb +0 -21
  44. data/lib/rex/encoding/xor/dword.rb +0 -21
  45. data/lib/rex/encoding/xor/dword.rb.ut.rb +0 -15
  46. data/lib/rex/encoding/xor/dword_additive.rb +0 -92
  47. data/lib/rex/encoding/xor/dword_additive.rb.ut.rb +0 -15
  48. data/lib/rex/encoding/xor/exceptions.rb +0 -17
  49. data/lib/rex/encoding/xor/generic.rb +0 -146
  50. data/lib/rex/encoding/xor/generic.rb.ut.rb +0 -120
  51. data/lib/rex/encoding/xor/qword.rb +0 -15
  52. data/lib/rex/encoding/xor/word.rb +0 -21
  53. data/lib/rex/encoding/xor/word.rb.ut.rb +0 -13
  54. data/lib/rex/exceptions.rb +0 -275
  55. data/lib/rex/exceptions.rb.ut.rb +0 -44
  56. data/lib/rex/exploitation/cmdstager.rb +0 -9
  57. data/lib/rex/exploitation/cmdstager/base.rb +0 -175
  58. data/lib/rex/exploitation/cmdstager/debug_asm.rb +0 -142
  59. data/lib/rex/exploitation/cmdstager/debug_write.rb +0 -136
  60. data/lib/rex/exploitation/cmdstager/tftp.rb +0 -63
  61. data/lib/rex/exploitation/cmdstager/vbs.rb +0 -128
  62. data/lib/rex/exploitation/egghunter.rb +0 -277
  63. data/lib/rex/exploitation/egghunter.rb.ut.rb +0 -25
  64. data/lib/rex/exploitation/encryptjs.rb +0 -77
  65. data/lib/rex/exploitation/heaplib.js.b64 +0 -331
  66. data/lib/rex/exploitation/heaplib.rb +0 -94
  67. data/lib/rex/exploitation/javascriptosdetect.rb +0 -897
  68. data/lib/rex/exploitation/obfuscatejs.rb +0 -335
  69. data/lib/rex/exploitation/omelet.rb +0 -320
  70. data/lib/rex/exploitation/omelet.rb.ut.rb +0 -13
  71. data/lib/rex/exploitation/opcodedb.rb +0 -818
  72. data/lib/rex/exploitation/opcodedb.rb.ut.rb +0 -279
  73. data/lib/rex/exploitation/seh.rb +0 -92
  74. data/lib/rex/exploitation/seh.rb.ut.rb +0 -19
  75. data/lib/rex/file.rb +0 -112
  76. data/lib/rex/file.rb.ut.rb +0 -16
  77. data/lib/rex/image_source.rb +0 -12
  78. data/lib/rex/image_source/disk.rb +0 -60
  79. data/lib/rex/image_source/image_source.rb +0 -46
  80. data/lib/rex/image_source/memory.rb +0 -37
  81. data/lib/rex/io/bidirectional_pipe.rb +0 -157
  82. data/lib/rex/io/datagram_abstraction.rb +0 -35
  83. data/lib/rex/io/stream.rb +0 -319
  84. data/lib/rex/io/stream_abstraction.rb +0 -197
  85. data/lib/rex/io/stream_server.rb +0 -211
  86. data/lib/rex/job_container.rb +0 -187
  87. data/lib/rex/logging.rb +0 -4
  88. data/lib/rex/logging/log_dispatcher.rb +0 -179
  89. data/lib/rex/logging/log_sink.rb +0 -42
  90. data/lib/rex/logging/sinks/flatfile.rb +0 -55
  91. data/lib/rex/logging/sinks/stderr.rb +0 -43
  92. data/lib/rex/machparsey.rb +0 -9
  93. data/lib/rex/machparsey/exceptions.rb +0 -34
  94. data/lib/rex/machparsey/mach.rb +0 -209
  95. data/lib/rex/machparsey/machbase.rb +0 -408
  96. data/lib/rex/machscan.rb +0 -9
  97. data/lib/rex/machscan/scanner.rb +0 -217
  98. data/lib/rex/mime.rb +0 -9
  99. data/lib/rex/mime/header.rb +0 -77
  100. data/lib/rex/mime/message.rb +0 -144
  101. data/lib/rex/mime/part.rb +0 -20
  102. data/lib/rex/nop/opty2.rb +0 -108
  103. data/lib/rex/nop/opty2.rb.ut.rb +0 -23
  104. data/lib/rex/nop/opty2_tables.rb +0 -300
  105. data/lib/rex/ole.rb +0 -205
  106. data/lib/rex/ole/clsid.rb +0 -47
  107. data/lib/rex/ole/difat.rb +0 -141
  108. data/lib/rex/ole/directory.rb +0 -231
  109. data/lib/rex/ole/direntry.rb +0 -240
  110. data/lib/rex/ole/docs/dependencies.txt +0 -8
  111. data/lib/rex/ole/docs/references.txt +0 -1
  112. data/lib/rex/ole/fat.rb +0 -99
  113. data/lib/rex/ole/header.rb +0 -204
  114. data/lib/rex/ole/minifat.rb +0 -77
  115. data/lib/rex/ole/propset.rb +0 -144
  116. data/lib/rex/ole/samples/create_ole.rb +0 -27
  117. data/lib/rex/ole/samples/dir.rb +0 -35
  118. data/lib/rex/ole/samples/dump_stream.rb +0 -34
  119. data/lib/rex/ole/samples/ole_info.rb +0 -23
  120. data/lib/rex/ole/storage.rb +0 -395
  121. data/lib/rex/ole/stream.rb +0 -53
  122. data/lib/rex/ole/substorage.rb +0 -49
  123. data/lib/rex/ole/util.rb +0 -157
  124. data/lib/rex/parser/arguments.rb +0 -97
  125. data/lib/rex/parser/arguments.rb.ut.rb +0 -67
  126. data/lib/rex/parser/ini.rb +0 -185
  127. data/lib/rex/parser/ini.rb.ut.rb +0 -29
  128. data/lib/rex/parser/ip360_aspl_xml.rb +0 -102
  129. data/lib/rex/parser/ip360_xml.rb +0 -93
  130. data/lib/rex/parser/nessus_xml.rb +0 -118
  131. data/lib/rex/parser/netsparker_xml.rb +0 -94
  132. data/lib/rex/parser/nexpose_xml.rb +0 -131
  133. data/lib/rex/parser/nmap_xml.rb +0 -121
  134. data/lib/rex/parser/retina_xml.rb +0 -109
  135. data/lib/rex/payloads.rb +0 -1
  136. data/lib/rex/payloads/win32.rb +0 -2
  137. data/lib/rex/payloads/win32/common.rb +0 -26
  138. data/lib/rex/payloads/win32/kernel.rb +0 -53
  139. data/lib/rex/payloads/win32/kernel/common.rb +0 -54
  140. data/lib/rex/payloads/win32/kernel/migration.rb +0 -12
  141. data/lib/rex/payloads/win32/kernel/recovery.rb +0 -50
  142. data/lib/rex/payloads/win32/kernel/stager.rb +0 -194
  143. data/lib/rex/peparsey.rb +0 -12
  144. data/lib/rex/peparsey/exceptions.rb +0 -32
  145. data/lib/rex/peparsey/pe.rb +0 -212
  146. data/lib/rex/peparsey/pe_memdump.rb +0 -63
  147. data/lib/rex/peparsey/pebase.rb +0 -1680
  148. data/lib/rex/peparsey/section.rb +0 -136
  149. data/lib/rex/pescan.rb +0 -13
  150. data/lib/rex/pescan/analyze.rb +0 -309
  151. data/lib/rex/pescan/scanner.rb +0 -206
  152. data/lib/rex/pescan/search.rb +0 -56
  153. data/lib/rex/platforms.rb +0 -1
  154. data/lib/rex/platforms/windows.rb +0 -51
  155. data/lib/rex/poly.rb +0 -132
  156. data/lib/rex/poly/block.rb +0 -477
  157. data/lib/rex/poly/register.rb +0 -100
  158. data/lib/rex/poly/register/x86.rb +0 -40
  159. data/lib/rex/post.rb +0 -8
  160. data/lib/rex/post/dir.rb +0 -51
  161. data/lib/rex/post/file.rb +0 -172
  162. data/lib/rex/post/file_stat.rb +0 -220
  163. data/lib/rex/post/gen.pl +0 -13
  164. data/lib/rex/post/io.rb +0 -182
  165. data/lib/rex/post/meterpreter.rb +0 -4
  166. data/lib/rex/post/meterpreter/channel.rb +0 -445
  167. data/lib/rex/post/meterpreter/channel_container.rb +0 -54
  168. data/lib/rex/post/meterpreter/channels/pool.rb +0 -160
  169. data/lib/rex/post/meterpreter/channels/pools/file.rb +0 -62
  170. data/lib/rex/post/meterpreter/channels/pools/stream_pool.rb +0 -103
  171. data/lib/rex/post/meterpreter/channels/stream.rb +0 -87
  172. data/lib/rex/post/meterpreter/client.rb +0 -364
  173. data/lib/rex/post/meterpreter/client_core.rb +0 -274
  174. data/lib/rex/post/meterpreter/dependencies.rb +0 -3
  175. data/lib/rex/post/meterpreter/extension.rb +0 -32
  176. data/lib/rex/post/meterpreter/extensions/espia/espia.rb +0 -58
  177. data/lib/rex/post/meterpreter/extensions/espia/tlv.rb +0 -16
  178. data/lib/rex/post/meterpreter/extensions/incognito/incognito.rb +0 -94
  179. data/lib/rex/post/meterpreter/extensions/incognito/tlv.rb +0 -21
  180. data/lib/rex/post/meterpreter/extensions/networkpug/networkpug.rb +0 -57
  181. data/lib/rex/post/meterpreter/extensions/networkpug/tlv.rb +0 -15
  182. data/lib/rex/post/meterpreter/extensions/priv/fs.rb +0 -118
  183. data/lib/rex/post/meterpreter/extensions/priv/passwd.rb +0 -61
  184. data/lib/rex/post/meterpreter/extensions/priv/priv.rb +0 -111
  185. data/lib/rex/post/meterpreter/extensions/priv/tlv.rb +0 -28
  186. data/lib/rex/post/meterpreter/extensions/sniffer/sniffer.rb +0 -101
  187. data/lib/rex/post/meterpreter/extensions/sniffer/tlv.rb +0 -26
  188. data/lib/rex/post/meterpreter/extensions/stdapi/constants.rb +0 -333
  189. data/lib/rex/post/meterpreter/extensions/stdapi/fs/dir.rb +0 -282
  190. data/lib/rex/post/meterpreter/extensions/stdapi/fs/file.rb +0 -266
  191. data/lib/rex/post/meterpreter/extensions/stdapi/fs/file_stat.rb +0 -103
  192. data/lib/rex/post/meterpreter/extensions/stdapi/fs/io.rb +0 -48
  193. data/lib/rex/post/meterpreter/extensions/stdapi/net/config.rb +0 -144
  194. data/lib/rex/post/meterpreter/extensions/stdapi/net/interface.rb +0 -73
  195. data/lib/rex/post/meterpreter/extensions/stdapi/net/route.rb +0 -56
  196. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket.rb +0 -137
  197. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_client_channel.rb +0 -180
  198. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_server_channel.rb +0 -167
  199. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/udp_channel.rb +0 -208
  200. data/lib/rex/post/meterpreter/extensions/stdapi/railgun.rb.ts.rb +0 -6
  201. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb +0 -38106
  202. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb.ut.rb +0 -31
  203. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb +0 -47
  204. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb.ut.rb +0 -36
  205. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_advapi32.rb +0 -1818
  206. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_iphlpapi.rb +0 -96
  207. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_kernel32.rb +0 -3848
  208. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_netapi32.rb +0 -26
  209. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ntdll.rb +0 -153
  210. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_shell32.rb +0 -21
  211. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_user32.rb +0 -3169
  212. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ws2_32.rb +0 -599
  213. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll.rb +0 -318
  214. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb +0 -100
  215. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb.ut.rb +0 -42
  216. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb +0 -148
  217. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb.ut.rb +0 -127
  218. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/multicall.rb +0 -309
  219. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb +0 -204
  220. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/tlv.rb +0 -51
  221. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/util.rb +0 -630
  222. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb +0 -75
  223. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb.ut.rb +0 -103
  224. data/lib/rex/post/meterpreter/extensions/stdapi/stdapi.rb +0 -149
  225. data/lib/rex/post/meterpreter/extensions/stdapi/sys/config.rb +0 -97
  226. data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log.rb +0 -192
  227. data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log_subsystem/event_record.rb +0 -41
  228. data/lib/rex/post/meterpreter/extensions/stdapi/sys/power.rb +0 -61
  229. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process.rb +0 -370
  230. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/image.rb +0 -129
  231. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/io.rb +0 -55
  232. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/memory.rb +0 -336
  233. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/thread.rb +0 -141
  234. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry.rb +0 -279
  235. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_key.rb +0 -193
  236. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_value.rb +0 -102
  237. data/lib/rex/post/meterpreter/extensions/stdapi/sys/thread.rb +0 -180
  238. data/lib/rex/post/meterpreter/extensions/stdapi/tlv.rb +0 -211
  239. data/lib/rex/post/meterpreter/extensions/stdapi/ui.rb +0 -227
  240. data/lib/rex/post/meterpreter/extensions/stdapi/webcam/webcam.rb +0 -63
  241. data/lib/rex/post/meterpreter/inbound_packet_handler.rb +0 -30
  242. data/lib/rex/post/meterpreter/object_aliases.rb +0 -83
  243. data/lib/rex/post/meterpreter/packet.rb +0 -688
  244. data/lib/rex/post/meterpreter/packet_dispatcher.rb +0 -431
  245. data/lib/rex/post/meterpreter/packet_parser.rb +0 -94
  246. data/lib/rex/post/meterpreter/packet_response_waiter.rb +0 -83
  247. data/lib/rex/post/meterpreter/ui/console.rb +0 -137
  248. data/lib/rex/post/meterpreter/ui/console/command_dispatcher.rb +0 -62
  249. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/core.rb +0 -730
  250. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/espia.rb +0 -108
  251. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/incognito.rb +0 -241
  252. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/networkpug.rb +0 -231
  253. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv.rb +0 -61
  254. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/elevate.rb +0 -98
  255. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/passwd.rb +0 -51
  256. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/timestomp.rb +0 -132
  257. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/sniffer.rb +0 -187
  258. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi.rb +0 -65
  259. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/fs.rb +0 -442
  260. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/net.rb +0 -298
  261. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/sys.rb +0 -486
  262. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/ui.rb +0 -315
  263. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/webcam.rb +0 -157
  264. data/lib/rex/post/meterpreter/ui/console/interactive_channel.rb +0 -95
  265. data/lib/rex/post/permission.rb +0 -26
  266. data/lib/rex/post/process.rb +0 -57
  267. data/lib/rex/post/thread.rb +0 -57
  268. data/lib/rex/post/ui.rb +0 -52
  269. data/lib/rex/proto.rb +0 -13
  270. data/lib/rex/proto.rb.ts.rb +0 -8
  271. data/lib/rex/proto/dcerpc.rb +0 -6
  272. data/lib/rex/proto/dcerpc.rb.ts.rb +0 -9
  273. data/lib/rex/proto/dcerpc/client.rb +0 -361
  274. data/lib/rex/proto/dcerpc/client.rb.ut.rb +0 -491
  275. data/lib/rex/proto/dcerpc/exceptions.rb +0 -150
  276. data/lib/rex/proto/dcerpc/handle.rb +0 -47
  277. data/lib/rex/proto/dcerpc/handle.rb.ut.rb +0 -85
  278. data/lib/rex/proto/dcerpc/ndr.rb +0 -72
  279. data/lib/rex/proto/dcerpc/ndr.rb.ut.rb +0 -41
  280. data/lib/rex/proto/dcerpc/packet.rb +0 -253
  281. data/lib/rex/proto/dcerpc/packet.rb.ut.rb +0 -56
  282. data/lib/rex/proto/dcerpc/response.rb +0 -187
  283. data/lib/rex/proto/dcerpc/response.rb.ut.rb +0 -15
  284. data/lib/rex/proto/dcerpc/uuid.rb +0 -84
  285. data/lib/rex/proto/dcerpc/uuid.rb.ut.rb +0 -46
  286. data/lib/rex/proto/dhcp.rb +0 -7
  287. data/lib/rex/proto/dhcp/constants.rb +0 -33
  288. data/lib/rex/proto/dhcp/server.rb +0 -292
  289. data/lib/rex/proto/drda.rb +0 -5
  290. data/lib/rex/proto/drda.rb.ts.rb +0 -17
  291. data/lib/rex/proto/drda/constants.rb +0 -49
  292. data/lib/rex/proto/drda/constants.rb.ut.rb +0 -23
  293. data/lib/rex/proto/drda/packet.rb +0 -252
  294. data/lib/rex/proto/drda/packet.rb.ut.rb +0 -109
  295. data/lib/rex/proto/drda/utils.rb +0 -123
  296. data/lib/rex/proto/drda/utils.rb.ut.rb +0 -84
  297. data/lib/rex/proto/http.rb +0 -5
  298. data/lib/rex/proto/http.rb.ts.rb +0 -12
  299. data/lib/rex/proto/http/client.rb +0 -821
  300. data/lib/rex/proto/http/client.rb.ut.rb +0 -95
  301. data/lib/rex/proto/http/handler.rb +0 -46
  302. data/lib/rex/proto/http/handler/erb.rb +0 -128
  303. data/lib/rex/proto/http/handler/erb.rb.ut.rb +0 -21
  304. data/lib/rex/proto/http/handler/erb.rb.ut.rb.rhtml +0 -1
  305. data/lib/rex/proto/http/handler/proc.rb +0 -60
  306. data/lib/rex/proto/http/handler/proc.rb.ut.rb +0 -24
  307. data/lib/rex/proto/http/header.rb +0 -161
  308. data/lib/rex/proto/http/header.rb.ut.rb +0 -46
  309. data/lib/rex/proto/http/packet.rb +0 -407
  310. data/lib/rex/proto/http/packet.rb.ut.rb +0 -165
  311. data/lib/rex/proto/http/request.rb +0 -356
  312. data/lib/rex/proto/http/request.rb.ut.rb +0 -214
  313. data/lib/rex/proto/http/response.rb +0 -90
  314. data/lib/rex/proto/http/response.rb.ut.rb +0 -149
  315. data/lib/rex/proto/http/server.rb +0 -369
  316. data/lib/rex/proto/http/server.rb.ut.rb +0 -79
  317. data/lib/rex/proto/ntlm.rb +0 -7
  318. data/lib/rex/proto/ntlm.rb.ut.rb +0 -177
  319. data/lib/rex/proto/ntlm/base.rb +0 -326
  320. data/lib/rex/proto/ntlm/constants.rb +0 -74
  321. data/lib/rex/proto/ntlm/crypt.rb +0 -415
  322. data/lib/rex/proto/ntlm/exceptions.rb +0 -9
  323. data/lib/rex/proto/ntlm/message.rb +0 -533
  324. data/lib/rex/proto/ntlm/utils.rb +0 -763
  325. data/lib/rex/proto/proxy/socks4a.rb +0 -440
  326. data/lib/rex/proto/rfb.rb +0 -19
  327. data/lib/rex/proto/rfb.rb.ut.rb +0 -37
  328. data/lib/rex/proto/rfb/cipher.rb +0 -84
  329. data/lib/rex/proto/rfb/client.rb +0 -207
  330. data/lib/rex/proto/rfb/constants.rb +0 -52
  331. data/lib/rex/proto/smb.rb +0 -7
  332. data/lib/rex/proto/smb.rb.ts.rb +0 -8
  333. data/lib/rex/proto/smb/client.rb +0 -1952
  334. data/lib/rex/proto/smb/client.rb.ut.rb +0 -223
  335. data/lib/rex/proto/smb/constants.rb +0 -1047
  336. data/lib/rex/proto/smb/constants.rb.ut.rb +0 -18
  337. data/lib/rex/proto/smb/crypt.rb +0 -36
  338. data/lib/rex/proto/smb/evasions.rb +0 -66
  339. data/lib/rex/proto/smb/exceptions.rb +0 -858
  340. data/lib/rex/proto/smb/simpleclient.rb +0 -306
  341. data/lib/rex/proto/smb/simpleclient.rb.ut.rb +0 -128
  342. data/lib/rex/proto/smb/utils.rb +0 -103
  343. data/lib/rex/proto/smb/utils.rb.ut.rb +0 -20
  344. data/lib/rex/proto/sunrpc.rb +0 -1
  345. data/lib/rex/proto/sunrpc/client.rb +0 -195
  346. data/lib/rex/proto/tftp.rb +0 -12
  347. data/lib/rex/proto/tftp/constants.rb +0 -39
  348. data/lib/rex/proto/tftp/server.rb +0 -497
  349. data/lib/rex/proto/tftp/server.rb.ut.rb +0 -28
  350. data/lib/rex/script.rb +0 -42
  351. data/lib/rex/script/base.rb +0 -59
  352. data/lib/rex/script/meterpreter.rb +0 -15
  353. data/lib/rex/script/shell.rb +0 -9
  354. data/lib/rex/service.rb +0 -48
  355. data/lib/rex/service_manager.rb +0 -141
  356. data/lib/rex/service_manager.rb.ut.rb +0 -32
  357. data/lib/rex/services/local_relay.rb +0 -423
  358. data/lib/rex/socket.rb +0 -684
  359. data/lib/rex/socket.rb.ut.rb +0 -107
  360. data/lib/rex/socket/comm.rb +0 -119
  361. data/lib/rex/socket/comm/local.rb +0 -412
  362. data/lib/rex/socket/comm/local.rb.ut.rb +0 -75
  363. data/lib/rex/socket/ip.rb +0 -130
  364. data/lib/rex/socket/parameters.rb +0 -345
  365. data/lib/rex/socket/parameters.rb.ut.rb +0 -51
  366. data/lib/rex/socket/range_walker.rb +0 -346
  367. data/lib/rex/socket/range_walker.rb.ut.rb +0 -55
  368. data/lib/rex/socket/ssl_tcp.rb +0 -184
  369. data/lib/rex/socket/ssl_tcp.rb.ut.rb +0 -39
  370. data/lib/rex/socket/ssl_tcp_server.rb +0 -122
  371. data/lib/rex/socket/ssl_tcp_server.rb.ut.rb +0 -61
  372. data/lib/rex/socket/subnet_walker.rb +0 -75
  373. data/lib/rex/socket/subnet_walker.rb.ut.rb +0 -28
  374. data/lib/rex/socket/switch_board.rb +0 -278
  375. data/lib/rex/socket/switch_board.rb.ut.rb +0 -52
  376. data/lib/rex/socket/tcp.rb +0 -76
  377. data/lib/rex/socket/tcp.rb.ut.rb +0 -64
  378. data/lib/rex/socket/tcp_server.rb +0 -67
  379. data/lib/rex/socket/tcp_server.rb.ut.rb +0 -44
  380. data/lib/rex/socket/udp.rb +0 -164
  381. data/lib/rex/socket/udp.rb.ut.rb +0 -44
  382. data/lib/rex/struct2.rb +0 -5
  383. data/lib/rex/struct2/c_struct.rb +0 -181
  384. data/lib/rex/struct2/c_struct_template.rb +0 -39
  385. data/lib/rex/struct2/constant.rb +0 -26
  386. data/lib/rex/struct2/element.rb +0 -44
  387. data/lib/rex/struct2/generic.rb +0 -73
  388. data/lib/rex/struct2/restraint.rb +0 -54
  389. data/lib/rex/struct2/s_string.rb +0 -72
  390. data/lib/rex/struct2/s_struct.rb +0 -111
  391. data/lib/rex/sync.rb +0 -6
  392. data/lib/rex/sync/event.rb +0 -94
  393. data/lib/rex/sync/read_write_lock.rb +0 -176
  394. data/lib/rex/sync/ref.rb +0 -57
  395. data/lib/rex/sync/thread_safe.rb +0 -82
  396. data/lib/rex/test.rb +0 -35
  397. data/lib/rex/text.rb +0 -1149
  398. data/lib/rex/text.rb.ut.rb +0 -190
  399. data/lib/rex/thread_factory.rb +0 -42
  400. data/lib/rex/time.rb +0 -65
  401. data/lib/rex/transformer.rb +0 -115
  402. data/lib/rex/transformer.rb.ut.rb +0 -38
  403. data/lib/rex/ui.rb +0 -21
  404. data/lib/rex/ui/interactive.rb +0 -254
  405. data/lib/rex/ui/output.rb +0 -78
  406. data/lib/rex/ui/output/none.rb +0 -18
  407. data/lib/rex/ui/progress_tracker.rb +0 -96
  408. data/lib/rex/ui/subscriber.rb +0 -149
  409. data/lib/rex/ui/text/color.rb +0 -97
  410. data/lib/rex/ui/text/color.rb.ut.rb +0 -18
  411. data/lib/rex/ui/text/dispatcher_shell.rb +0 -467
  412. data/lib/rex/ui/text/input.rb +0 -117
  413. data/lib/rex/ui/text/input/buffer.rb +0 -75
  414. data/lib/rex/ui/text/input/readline.rb +0 -129
  415. data/lib/rex/ui/text/input/socket.rb +0 -95
  416. data/lib/rex/ui/text/input/stdio.rb +0 -45
  417. data/lib/rex/ui/text/irb_shell.rb +0 -57
  418. data/lib/rex/ui/text/output.rb +0 -80
  419. data/lib/rex/ui/text/output/buffer.rb +0 -61
  420. data/lib/rex/ui/text/output/file.rb +0 -43
  421. data/lib/rex/ui/text/output/socket.rb +0 -43
  422. data/lib/rex/ui/text/output/stdio.rb +0 -40
  423. data/lib/rex/ui/text/progress_tracker.rb +0 -56
  424. data/lib/rex/ui/text/progress_tracker.rb.ut.rb +0 -34
  425. data/lib/rex/ui/text/shell.rb +0 -328
  426. data/lib/rex/ui/text/table.rb +0 -279
  427. data/lib/rex/ui/text/table.rb.ut.rb +0 -55
  428. data/lib/rex/zip.rb +0 -93
  429. data/lib/rex/zip/archive.rb +0 -184
  430. data/lib/rex/zip/blocks.rb +0 -182
  431. data/lib/rex/zip/entry.rb +0 -104
  432. data/lib/rex/zip/samples/comment.rb +0 -32
  433. data/lib/rex/zip/samples/mkwar.rb +0 -138
  434. data/lib/rex/zip/samples/mkzip.rb +0 -19
  435. data/lib/rex/zip/samples/recursive.rb +0 -58
@@ -1,20 +0,0 @@
1
- #!/usr/bin/env ruby
2
-
3
- $:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
4
-
5
- require 'rex/test'
6
- require 'rex/proto/smb/utils'
7
-
8
- class Rex::Proto::SMB::Utils::UnitTest < Test::Unit::TestCase
9
-
10
- Klass = Rex::Proto::SMB::Utils
11
-
12
- def test_nbname
13
-
14
- nbdecoded = 'METASPLOITROCKS!'
15
- nbencoded = 'ENEFFEEBFDFAEMEPEJFEFCEPEDELFDCB'
16
-
17
- assert_equal(Klass.nbname_encode(nbdecoded), nbencoded )
18
- assert_equal(Klass.nbname_decode(nbencoded), nbdecoded )
19
- end
20
- end
@@ -1 +0,0 @@
1
- require 'rex/proto/sunrpc/client'
@@ -1,195 +0,0 @@
1
- require 'rex/socket'
2
- require 'rex/encoder/xdr'
3
-
4
- module Rex
5
- module Proto
6
- module SunRPC
7
-
8
- class RPCTimeout < ::Interrupt
9
- def initialize(msg = 'Operation timed out.')
10
- @msg = msg
11
- end
12
-
13
- def to_s
14
- @msg
15
- end
16
- end
17
-
18
- # XXX: CPORT!
19
- class Client
20
- AUTH_NULL = 0
21
- AUTH_UNIX = 1
22
-
23
- PMAP_PROG = 100000
24
- PMAP_VERS = 2
25
- PMAP_GETPORT = 3
26
-
27
- CALL = 0
28
-
29
- attr_accessor :rhost, :rport, :proto, :program, :version
30
- attr_accessor :pport, :call_sock, :timeout, :context
31
-
32
- attr_accessor :should_fragment
33
-
34
- def initialize(opts)
35
- self.rhost = opts[:rhost]
36
- self.rport = opts[:rport]
37
- self.program = opts[:program]
38
- self.version = opts[:version]
39
- self.timeout = opts[:timeout] || 20
40
- self.context = opts[:context] || {}
41
- self.proto = opts[:proto]
42
-
43
- if self.proto.downcase !~ /^(tcp|udp)$/
44
- raise ::Rex::ArgumentError, 'Protocol is not "tcp" or "udp"'
45
- end
46
-
47
- @pport = nil
48
-
49
- @auth_type = AUTH_NULL
50
- @auth_data = ''
51
-
52
- @call_sock = nil
53
- end
54
-
55
- # XXX: Add optional parameter to have proto be something else
56
- def create()
57
- proto_num = 0
58
- if @proto.eql?('tcp')
59
- proto_num = 6
60
- elsif @proto.eql?('udp')
61
- proto_num = 17
62
- end
63
-
64
- buf =
65
- Rex::Encoder::XDR.encode(CALL, 2, PMAP_PROG, PMAP_VERS, PMAP_GETPORT,
66
- @auth_type, [@auth_data, 400], AUTH_NULL, '',
67
- @program, @version, proto_num, 0)
68
-
69
- sock = make_rpc(@proto, @rhost, @rport)
70
- send_rpc(sock, buf)
71
- ret = recv_rpc(sock)
72
- close_rpc(sock)
73
-
74
- return ret
75
- end
76
-
77
- def call(procedure, buffer, maxwait = self.timeout)
78
- buf =
79
- Rex::Encoder::XDR.encode(CALL, 2, @program, @version, procedure,
80
- @auth_type, [@auth_data, 400], AUTH_NULL, '')+
81
- buffer
82
-
83
- if ! @call_sock
84
- @call_sock = make_rpc(@proto, @rhost, @pport)
85
- end
86
-
87
- send_rpc(@call_sock, buf)
88
- recv_rpc(@call_sock, maxwait)
89
- end
90
-
91
- def destroy
92
- close_rpc(@call_sock) if @call_sock
93
- @call_sock = nil
94
- end
95
-
96
- def authnull_create
97
- @auth_type = AUTH_NULL
98
- @auth_data = ''
99
- end
100
-
101
- def authunix_create(host, uid, gid, groupz)
102
- raise ::Rex::ArgumentError, 'Hostname length is too long' if host.length > 255
103
- # 10?
104
- raise ::Rex::ArgumentError, 'Too many groups' if groupz.length > 10
105
-
106
- @auth_type = AUTH_UNIX
107
- @auth_data =
108
- Rex::Encoder::XDR.encode(0, host, uid, gid, groupz) # XXX: TIME! GROUPZ?!
109
- end
110
-
111
- # XXX: Dirty, integrate some sort of request system into create/call?
112
- def portmap_req(host, port, rpc_vers, procedure, buffer)
113
- buf = Rex::Encoder::XDR.encode(CALL, 2, PMAP_PROG, rpc_vers, procedure,
114
- AUTH_NULL, '', AUTH_NULL, '') + buffer
115
-
116
- sock = make_rpc('tcp', host, port)
117
- send_rpc(sock, buf)
118
- ret = recv_rpc(sock)
119
- close_rpc(sock)
120
-
121
- return ret
122
- end
123
-
124
- private
125
- def make_rpc(proto, host, port)
126
- Rex::Socket.create(
127
- 'PeerHost' => host,
128
- 'PeerPort' => port,
129
- 'Proto' => proto,
130
- 'Timeout' => self.timeout,
131
- 'Context' => self.context
132
- )
133
- end
134
-
135
- def build_tcp(buf)
136
- if !self.should_fragment
137
- return Rex::Encoder::XDR.encode(0x80000000 | buf.length) + buf
138
- end
139
-
140
- str = buf.dup
141
-
142
- fragmented = ''
143
-
144
- while (str.size > 0)
145
- frag = str.slice!(0, rand(3) + 1)
146
- len = frag.size
147
- if str.size == 0
148
- len |= 0x80000000
149
- end
150
-
151
- fragmented += Rex::Encoder::XDR.encode(len) + frag
152
- end
153
-
154
- return fragmented
155
- end
156
-
157
- def send_rpc(sock, buf)
158
- buf = gen_xid() + buf
159
- if sock.type?.eql?('tcp')
160
- buf = build_tcp(buf)
161
- end
162
- sock.put(buf)
163
- end
164
-
165
- def recv_rpc(sock, maxwait=self.timeout)
166
-
167
- buf = nil
168
- begin
169
- Timeout.timeout(maxwait) { buf = sock.get }
170
- rescue ::Timeout
171
- end
172
-
173
- return nil if not buf
174
-
175
- buf.slice!(0..3)
176
- if sock.type?.eql?('tcp')
177
- buf.slice!(0..3)
178
- end
179
- return buf if buf.length > 1
180
- return nil
181
- end
182
-
183
- def close_rpc(sock)
184
- sock.close
185
- end
186
-
187
- def gen_xid
188
- return Rex::Encoder::XDR.encode(rand(0xffffffff) + 1)
189
- end
190
- end
191
-
192
- end
193
- end
194
- end
195
-
@@ -1,12 +0,0 @@
1
- # $Id: tftp.rb 9962 2010-08-06 17:21:22Z jduck $
2
- #
3
- # TFTP Server implementation according to:
4
- #
5
- # RFC1350, RFC2347, RFC2348, RFC2349
6
- #
7
- # written by jduck <jduck [at] metasploit.com>
8
- # thx to scriptjunkie for pointing out option extensions
9
- #
10
-
11
- require 'rex/proto/tftp/constants'
12
- require 'rex/proto/tftp/server'
@@ -1,39 +0,0 @@
1
- # $Id: constants.rb 9962 2010-08-06 17:21:22Z jduck $
2
- require 'rex/proto/tftp'
3
-
4
- module Rex
5
- module Proto
6
- module TFTP
7
-
8
- OPCODES = %w{ Unknown RRQ WRQ DATA ACK ERROR }
9
- OpRead = 1
10
- OpWrite = 2
11
- OpData = 3
12
- OpAck = 4
13
- OpError = 5
14
- OpOptAck = 6
15
-
16
- ERRCODES = [
17
- "Undefined",
18
- "File not found",
19
- "Access violation",
20
- "Disk full or allocation exceeded",
21
- "Illegal TFTP operation",
22
- "Unknown transfer ID",
23
- "File already exists",
24
- "No such user",
25
- "Failed option negotiation"
26
- ]
27
-
28
- ErrFileNotFound = 1
29
- ErrAccessViolation = 2
30
- ErrDiskFull = 3
31
- ErrIllegalOperation = 4
32
- ErrUnknownTransferId = 5
33
- ErrFileExists = 6
34
- ErrNoSuchUser = 7
35
- ErrFailedOptNegotiation = 8
36
-
37
- end
38
- end
39
- end
@@ -1,497 +0,0 @@
1
- # $Id: server.rb 11636 2011-01-25 02:24:37Z hdm $
2
- require 'rex/socket'
3
- require 'rex/proto/tftp'
4
-
5
- module Rex
6
- module Proto
7
- module TFTP
8
-
9
- #
10
- # Little util function
11
- #
12
- def self.get_string(data)
13
- idx = data.index("\x00")
14
- return nil if not idx
15
- ret = data.slice!(0, idx)
16
- # Slice off the nul byte.
17
- data.slice!(0,1)
18
- ret
19
- end
20
-
21
-
22
- ##
23
- #
24
- # TFTP Server class
25
- #
26
- ##
27
- class Server
28
-
29
- def initialize(port = 69, listen_host = '0.0.0.0', context = {})
30
- self.listen_host = listen_host
31
- self.listen_port = port
32
- self.context = context
33
- self.sock = nil
34
- @shutting_down = false
35
- @output_dir = nil
36
- @tftproot = nil
37
-
38
- self.files = []
39
- self.uploaded = []
40
- self.transfers = []
41
- end
42
-
43
-
44
- #
45
- # Start the TFTP server
46
- #
47
- def start
48
- self.sock = Rex::Socket::Udp.create(
49
- 'LocalHost' => listen_host,
50
- 'LocalPort' => listen_port,
51
- 'Context' => context
52
- )
53
-
54
- self.thread = Rex::ThreadFactory.spawn("TFTPServerMonitor", false) {
55
- monitor_socket
56
- }
57
- end
58
-
59
-
60
- #
61
- # Stop the TFTP server
62
- #
63
- def stop
64
- @shutting_down = true
65
-
66
- # Wait a maximum of 30 seconds for all transfers to finish.
67
- start = ::Time.now
68
- while (self.transfers.length > 0)
69
- ::IO.select(nil, nil, nil, 0.5)
70
- dur = ::Time.now - start
71
- break if (dur > 30)
72
- end
73
-
74
- self.files.clear
75
- self.thread.kill
76
- self.sock.close rescue nil # might be closed already
77
- end
78
-
79
-
80
- #
81
- # Register a filename and content for a client to request
82
- #
83
- def register_file(fn, content, once = false)
84
- self.files << {
85
- :name => fn,
86
- :data => content,
87
- :once => once
88
- }
89
- end
90
-
91
-
92
- #
93
- # Register an entire directory to serve files from
94
- #
95
- def set_tftproot(rootdir)
96
- @tftproot = rootdir if ::File.directory?(rootdir)
97
- end
98
-
99
-
100
- #
101
- # Register a directory to write uploaded files to
102
- #
103
- def set_output_dir(outdir)
104
- @output_dir = outdir if ::File.directory?(outdir)
105
- end
106
-
107
-
108
- #
109
- # Send an error packet w/the specified code and string
110
- #
111
- def send_error(from, num)
112
- if (num < 1 or num >= ERRCODES.length)
113
- # ignore..
114
- return
115
- end
116
- pkt = [OpError, num].pack('nn')
117
- pkt << ERRCODES[num]
118
- pkt << "\x00"
119
- send_packet(from, pkt)
120
- end
121
-
122
-
123
- #
124
- # Send a single packet to the specified host
125
- #
126
- def send_packet(from, pkt)
127
- self.sock.sendto(pkt, from[0], from[1])
128
- end
129
-
130
-
131
- #
132
- # Find the hash entry for a file that may be offered
133
- #
134
- def find_file(fname)
135
- # Files served via register_file() take precedence.
136
- self.files.each do |f|
137
- if (fname == f[:name])
138
- return f
139
- end
140
- end
141
-
142
- # Now, if we have a tftproot, see if it can serve from it
143
- if @tftproot
144
- return find_file_in_root(fname)
145
- end
146
-
147
- nil
148
- end
149
-
150
-
151
- #
152
- # Find the file in the specified tftp root and add a temporary
153
- # entry to the files hash.
154
- #
155
- def find_file_in_root(fname)
156
- fn = ::File.expand_path(::File.join(@tftproot, fname))
157
-
158
- # Don't allow directory traversal
159
- return nil if fn.index(@tftproot) != 0
160
-
161
- return nil if not ::File.file?(fn) or not ::File.readable?(fn)
162
-
163
- # Read the file contents, and register it as being served once
164
- data = data = ::File.open(fn, "rb") { |fd| fd.read(fd.stat.size) }
165
- register_file(fname, data, true)
166
-
167
- # Return the last file in the array
168
- return self.files[-1]
169
- end
170
-
171
-
172
- attr_accessor :listen_host, :listen_port, :context
173
- attr_accessor :sock, :files, :transfers, :uploaded
174
- attr_accessor :thread
175
-
176
- attr_accessor :incoming_file_hook
177
-
178
- protected
179
-
180
- def find_transfer(type, from, block)
181
- self.transfers.each do |tr|
182
- if (tr[:type] == type and tr[:from] == from and tr[:block] == block)
183
- return tr
184
- end
185
- end
186
- nil
187
- end
188
-
189
- def save_output(tr)
190
- self.uploaded << tr[:file]
191
-
192
- return incoming_file_hook.call(tr) if incoming_file_hook
193
-
194
- if @output_dir
195
- fn = tr[:file][:name].split(File::SEPARATOR)[-1]
196
- if fn
197
- fn = ::File.join(@output_dir, Rex::FileUtils.clean_path(fn))
198
- ::File.open(fn, "wb") { |fd|
199
- fd.write(tr[:file][:data])
200
- }
201
- end
202
- end
203
- end
204
-
205
-
206
- def check_retransmission(tr)
207
- elapsed = ::Time.now - tr[:last_sent]
208
- if (elapsed >= tr[:timeout])
209
- # max retries reached?
210
- if (tr[:retries] < 3)
211
- #if (tr[:type] == OpRead)
212
- # puts "[-] ack timed out, resending block"
213
- #else
214
- # puts "[-] block timed out, resending ack"
215
- #end
216
- tr[:last_sent] = nil
217
- tr[:retries] += 1
218
- else
219
- #puts "[-] maximum tries reached, terminating transfer"
220
- self.transfers.delete(tr)
221
- end
222
- end
223
- end
224
-
225
-
226
- #
227
- # See if there is anything to do.. If so, dispatch it.
228
- #
229
- def monitor_socket
230
- while true
231
- rds = [@sock]
232
- wds = []
233
- self.transfers.each do |tr|
234
- if (not tr[:last_sent])
235
- wds << @sock
236
- break
237
- end
238
- end
239
- eds = [@sock]
240
-
241
- r,w,e = ::IO.select(rds,wds,eds,1)
242
-
243
- if (r != nil and r[0] == self.sock)
244
- buf,host,port = self.sock.recvfrom(65535)
245
- # Lame compatabilitiy :-/
246
- from = [host, port]
247
- dispatch_request(from, buf)
248
- end
249
-
250
- #
251
- # Check to see if transfers need maintenance
252
- #
253
- self.transfers.each do |tr|
254
- # We handle RRQ and WRQ separately
255
- #
256
- if (tr[:type] == OpRead)
257
- # Are we awaiting an ack?
258
- if (tr[:last_sent])
259
- check_retransmission(tr)
260
- elsif (w != nil and w[0] == self.sock)
261
- # No ack waiting, send next block..
262
- chunk = tr[:file][:data].slice(tr[:offset], tr[:blksize])
263
- if (chunk and chunk.length >= 0)
264
- pkt = [OpData, tr[:block]].pack('nn')
265
- pkt << chunk
266
-
267
- send_packet(tr[:from], pkt)
268
- tr[:last_sent] = ::Time.now
269
-
270
- # If the file is a one-serve, mark it as started
271
- tr[:file][:started] = true if (tr[:file][:once])
272
- else
273
- # No more chunks.. transfer is most likely done.
274
- # However, we can only delete it once the last chunk has been
275
- # acked.
276
- end
277
- end
278
- else
279
- # Are we awaiting data?
280
- if (tr[:last_sent])
281
- check_retransmission(tr)
282
- elsif (w != nil and w[0] == self.sock)
283
- # Not waiting for data, send an ack..
284
- #puts "[*] sending ack for block %d" % [tr[:block]]
285
- pkt = [OpAck, tr[:block]].pack('nn')
286
-
287
- send_packet(tr[:from], pkt)
288
- tr[:last_sent] = ::Time.now
289
-
290
- # If we had a 0-511 byte chunk, we're done.
291
- if (tr[:last_size] and tr[:last_size] < tr[:blksize])
292
- #puts "[*] Transfer complete, saving output"
293
- save_output(tr)
294
- self.transfers.delete(tr)
295
- end
296
- end
297
- end
298
- end
299
- end
300
- end
301
-
302
-
303
- def next_block(tr)
304
- tr[:block] += 1
305
- tr[:last_sent] = nil
306
- tr[:retries] = 0
307
- end
308
-
309
-
310
- #
311
- # Dispatch a packet that we received
312
- #
313
- def dispatch_request(from, buf)
314
-
315
- op = buf.unpack('n')[0]
316
- buf.slice!(0,2)
317
-
318
- #XXX: todo - create call backs for status
319
- #start = "[*] TFTP - %s:%u - %s" % [from[0], from[1], OPCODES[op]]
320
-
321
- case op
322
- when OpRead
323
- # Process RRQ packets
324
- fn = TFTP::get_string(buf)
325
- mode = TFTP::get_string(buf).downcase
326
-
327
- #puts "%s %s %s" % [start, fn, mode]
328
-
329
- if (not @shutting_down) and (file = self.find_file(fn))
330
- if (file[:once] and file[:started])
331
- send_error(from, ErrFileNotFound)
332
- else
333
- transfer = {
334
- :type => OpRead,
335
- :from => from,
336
- :file => file,
337
- :block => 1,
338
- :blksize => 512,
339
- :offset => 0,
340
- :timeout => 3,
341
- :last_sent => nil,
342
- :retries => 0
343
- }
344
-
345
- process_options(from, buf, transfer)
346
-
347
- self.transfers << transfer
348
- end
349
- else
350
- #puts "[-] file not found!"
351
- send_error(from, ErrFileNotFound)
352
- end
353
-
354
- when OpWrite
355
- # Process WRQ packets
356
- fn = TFTP::get_string(buf)
357
- mode = TFTP::get_string(buf).downcase
358
-
359
- #puts "%s %s %s" % [start, fn, mode]
360
-
361
- if not @shutting_down
362
- transfer = {
363
- :type => OpWrite,
364
- :from => from,
365
- :file => { :name => fn, :data => '' },
366
- :block => 0, # WRQ starts at 0
367
- :blksize => 512,
368
- :timeout => 3,
369
- :last_sent => nil,
370
- :retries => 0
371
- }
372
-
373
- process_options(from, buf, transfer)
374
-
375
- self.transfers << transfer
376
- else
377
- send_error(from, ErrIllegalOperation)
378
- end
379
-
380
- when OpAck
381
- # Process ACK packets
382
- block = buf.unpack('n')[0]
383
-
384
- #puts "%s %d" % [start, block]
385
-
386
- tr = find_transfer(OpRead, from, block)
387
- if not tr
388
- # NOTE: some clients, such as pxelinux, send an ack for block 0.
389
- # To deal with this, we simply ignore it as we start with block 1.
390
- return if block == 0
391
-
392
- # If we didn't find it, send an error.
393
- send_error(from, ErrUnknownTransferId)
394
- else
395
- # acked! send the next block
396
- tr[:offset] += tr[:blksize]
397
- next_block(tr)
398
-
399
- # If the transfer is finished, delete it
400
- if (tr[:offset] > tr[:file][:data].length)
401
- #puts "[*] Transfer complete"
402
- self.transfers.delete(tr)
403
-
404
- # if the file is a one-serve, delete it from the files array
405
- if tr[:file][:once]
406
- #puts "[*] Removed one-serve file: #{tr[:file][:name]}"
407
- self.files.delete(tr[:file])
408
- end
409
- end
410
- end
411
-
412
- when OpData
413
- # Process Data packets
414
- block = buf.unpack('n')[0]
415
- data = buf.slice(2, buf.length)
416
-
417
- #puts "%s %d %d bytes" % [start, block, data.length]
418
-
419
- tr = find_transfer(OpWrite, from, (block-1))
420
- if not tr
421
- # If we didn't find it, send an error.
422
- send_error(from, ErrUnknownTransferId)
423
- else
424
- tr[:file][:data] << data
425
- tr[:last_size] = data.length
426
- next_block(tr)
427
-
428
- # Similar to RRQ transfers, we cannot detect that the
429
- # transfer finished here. We must do so after transmitting
430
- # the final ACK.
431
- end
432
-
433
- else
434
- # Other packets are unsupported
435
- #puts start
436
- send_error(from, ErrAccessViolation)
437
-
438
- end
439
- end
440
-
441
- def process_options(from, buf, tr)
442
- found = 0
443
- to_ack = []
444
- while buf.length >= 4
445
- opt = TFTP::get_string(buf)
446
- break if not opt
447
- val = TFTP::get_string(buf)
448
- break if not val
449
-
450
- found += 1
451
-
452
- # Is it one we support?
453
- opt.downcase!
454
-
455
- case opt
456
- when "blksize"
457
- val = val.to_i
458
- if val > 0
459
- tr[:blksize] = val
460
- to_ack << [ opt, val.to_s ]
461
- end
462
-
463
- when "timeout"
464
- val = val.to_i
465
- if val >= 1 and val <= 255
466
- tr[:timeout] = val
467
- to_ack << [ opt, val.to_s ]
468
- end
469
-
470
- when "tsize"
471
- if tr[:type] == OpRead
472
- len = tr[:file][:data].length
473
- else
474
- val = val.to_i
475
- len = val
476
- end
477
- to_ack << [ opt, len.to_s ]
478
-
479
- end
480
- end
481
-
482
- return if to_ack.length < 1
483
-
484
- # if we have anything to ack, do it
485
- data = [OpOptAck].pack('n')
486
- to_ack.each { |el|
487
- data << el[0] << "\x00" << el[1] << "\x00"
488
- }
489
-
490
- send_packet(from, data)
491
- end
492
-
493
- end
494
-
495
- end
496
- end
497
- end