librex 0.0.16 → 0.0.17

Sign up to get free protection for your applications and to get access to all the features.
Files changed (434) hide show
  1. data/lib/rex/LICENSE +29 -0
  2. data/lib/rex/arch.rb +103 -0
  3. data/lib/rex/arch/sparc.rb +75 -0
  4. data/lib/rex/arch/sparc.rb.ut.rb +18 -0
  5. data/lib/rex/arch/x86.rb +513 -0
  6. data/lib/rex/arch/x86.rb.ut.rb +93 -0
  7. data/lib/rex/assembly/nasm.rb +104 -0
  8. data/lib/rex/assembly/nasm.rb.ut.rb +22 -0
  9. data/lib/rex/codepage.map +104 -0
  10. data/lib/rex/compat.rb +311 -0
  11. data/lib/rex/constants.rb +113 -0
  12. data/lib/rex/elfparsey.rb +11 -0
  13. data/lib/rex/elfparsey/elf.rb +123 -0
  14. data/lib/rex/elfparsey/elfbase.rb +258 -0
  15. data/lib/rex/elfparsey/exceptions.rb +27 -0
  16. data/lib/rex/elfscan.rb +12 -0
  17. data/lib/rex/elfscan/scanner.rb +207 -0
  18. data/lib/rex/elfscan/search.rb +46 -0
  19. data/lib/rex/encoder/alpha2.rb +31 -0
  20. data/lib/rex/encoder/alpha2/alpha_mixed.rb +68 -0
  21. data/lib/rex/encoder/alpha2/alpha_upper.rb +79 -0
  22. data/lib/rex/encoder/alpha2/generic.rb +114 -0
  23. data/lib/rex/encoder/alpha2/unicode_mixed.rb +117 -0
  24. data/lib/rex/encoder/alpha2/unicode_upper.rb +129 -0
  25. data/lib/rex/encoder/ndr.rb +89 -0
  26. data/lib/rex/encoder/ndr.rb.ut.rb +44 -0
  27. data/lib/rex/encoder/nonalpha.rb +61 -0
  28. data/lib/rex/encoder/nonupper.rb +64 -0
  29. data/lib/rex/encoder/xdr.rb +106 -0
  30. data/lib/rex/encoder/xdr.rb.ut.rb +29 -0
  31. data/lib/rex/encoder/xor.rb +69 -0
  32. data/lib/rex/encoder/xor/dword.rb +13 -0
  33. data/lib/rex/encoder/xor/dword_additive.rb +13 -0
  34. data/lib/rex/encoders/xor_dword.rb +35 -0
  35. data/lib/rex/encoders/xor_dword_additive.rb +53 -0
  36. data/lib/rex/encoders/xor_dword_additive.rb.ut.rb +12 -0
  37. data/lib/rex/encoding/xor.rb +20 -0
  38. data/lib/rex/encoding/xor.rb.ts.rb +14 -0
  39. data/lib/rex/encoding/xor/byte.rb +15 -0
  40. data/lib/rex/encoding/xor/byte.rb.ut.rb +21 -0
  41. data/lib/rex/encoding/xor/dword.rb +21 -0
  42. data/lib/rex/encoding/xor/dword.rb.ut.rb +15 -0
  43. data/lib/rex/encoding/xor/dword_additive.rb +92 -0
  44. data/lib/rex/encoding/xor/dword_additive.rb.ut.rb +15 -0
  45. data/lib/rex/encoding/xor/exceptions.rb +17 -0
  46. data/lib/rex/encoding/xor/generic.rb +146 -0
  47. data/lib/rex/encoding/xor/generic.rb.ut.rb +120 -0
  48. data/lib/rex/encoding/xor/qword.rb +15 -0
  49. data/lib/rex/encoding/xor/word.rb +21 -0
  50. data/lib/rex/encoding/xor/word.rb.ut.rb +13 -0
  51. data/lib/rex/exceptions.rb +275 -0
  52. data/lib/rex/exceptions.rb.ut.rb +44 -0
  53. data/lib/rex/exploitation/cmdstager.rb +9 -0
  54. data/lib/rex/exploitation/cmdstager/base.rb +175 -0
  55. data/lib/rex/exploitation/cmdstager/debug_asm.rb +142 -0
  56. data/lib/rex/exploitation/cmdstager/debug_write.rb +136 -0
  57. data/lib/rex/exploitation/cmdstager/tftp.rb +63 -0
  58. data/lib/rex/exploitation/cmdstager/vbs.rb +128 -0
  59. data/lib/rex/exploitation/egghunter.rb +277 -0
  60. data/lib/rex/exploitation/egghunter.rb.ut.rb +25 -0
  61. data/lib/rex/exploitation/encryptjs.rb +77 -0
  62. data/lib/rex/exploitation/heaplib.js.b64 +331 -0
  63. data/lib/rex/exploitation/heaplib.rb +94 -0
  64. data/lib/rex/exploitation/javascriptosdetect.rb +897 -0
  65. data/lib/rex/exploitation/obfuscatejs.rb +335 -0
  66. data/lib/rex/exploitation/omelet.rb +320 -0
  67. data/lib/rex/exploitation/omelet.rb.ut.rb +13 -0
  68. data/lib/rex/exploitation/opcodedb.rb +818 -0
  69. data/lib/rex/exploitation/opcodedb.rb.ut.rb +279 -0
  70. data/lib/rex/exploitation/seh.rb +92 -0
  71. data/lib/rex/exploitation/seh.rb.ut.rb +19 -0
  72. data/lib/rex/file.rb +112 -0
  73. data/lib/rex/file.rb.ut.rb +16 -0
  74. data/lib/rex/image_source.rb +12 -0
  75. data/lib/rex/image_source/disk.rb +60 -0
  76. data/lib/rex/image_source/image_source.rb +46 -0
  77. data/lib/rex/image_source/memory.rb +37 -0
  78. data/lib/rex/io/bidirectional_pipe.rb +157 -0
  79. data/lib/rex/io/datagram_abstraction.rb +35 -0
  80. data/lib/rex/io/ring_buffer.rb +364 -0
  81. data/lib/rex/io/stream.rb +319 -0
  82. data/lib/rex/io/stream_abstraction.rb +197 -0
  83. data/lib/rex/io/stream_server.rb +211 -0
  84. data/lib/rex/job_container.rb +187 -0
  85. data/lib/rex/logging.rb +4 -0
  86. data/lib/rex/logging/log_dispatcher.rb +179 -0
  87. data/lib/rex/logging/log_sink.rb +42 -0
  88. data/lib/rex/logging/sinks/flatfile.rb +55 -0
  89. data/lib/rex/logging/sinks/stderr.rb +43 -0
  90. data/lib/rex/machparsey.rb +9 -0
  91. data/lib/rex/machparsey/exceptions.rb +34 -0
  92. data/lib/rex/machparsey/mach.rb +209 -0
  93. data/lib/rex/machparsey/machbase.rb +408 -0
  94. data/lib/rex/machscan.rb +9 -0
  95. data/lib/rex/machscan/scanner.rb +217 -0
  96. data/lib/rex/mime.rb +9 -0
  97. data/lib/rex/mime/header.rb +77 -0
  98. data/lib/rex/mime/message.rb +144 -0
  99. data/lib/rex/mime/part.rb +20 -0
  100. data/lib/rex/nop/opty2.rb +108 -0
  101. data/lib/rex/nop/opty2.rb.ut.rb +23 -0
  102. data/lib/rex/nop/opty2_tables.rb +300 -0
  103. data/lib/rex/ole.rb +205 -0
  104. data/lib/rex/ole/clsid.rb +47 -0
  105. data/lib/rex/ole/difat.rb +141 -0
  106. data/lib/rex/ole/directory.rb +231 -0
  107. data/lib/rex/ole/direntry.rb +240 -0
  108. data/lib/rex/ole/docs/dependencies.txt +8 -0
  109. data/lib/rex/ole/docs/references.txt +1 -0
  110. data/lib/rex/ole/fat.rb +99 -0
  111. data/lib/rex/ole/header.rb +204 -0
  112. data/lib/rex/ole/minifat.rb +77 -0
  113. data/lib/rex/ole/propset.rb +144 -0
  114. data/lib/rex/ole/samples/create_ole.rb +27 -0
  115. data/lib/rex/ole/samples/dir.rb +35 -0
  116. data/lib/rex/ole/samples/dump_stream.rb +34 -0
  117. data/lib/rex/ole/samples/ole_info.rb +23 -0
  118. data/lib/rex/ole/storage.rb +395 -0
  119. data/lib/rex/ole/stream.rb +53 -0
  120. data/lib/rex/ole/substorage.rb +49 -0
  121. data/lib/rex/ole/util.rb +157 -0
  122. data/lib/rex/parser/arguments.rb +97 -0
  123. data/lib/rex/parser/arguments.rb.ut.rb +67 -0
  124. data/lib/rex/parser/ini.rb +185 -0
  125. data/lib/rex/parser/ini.rb.ut.rb +29 -0
  126. data/lib/rex/parser/ip360_aspl_xml.rb +102 -0
  127. data/lib/rex/parser/ip360_xml.rb +97 -0
  128. data/lib/rex/parser/nessus_xml.rb +118 -0
  129. data/lib/rex/parser/netsparker_xml.rb +94 -0
  130. data/lib/rex/parser/nexpose_xml.rb +136 -0
  131. data/lib/rex/parser/nmap_xml.rb +137 -0
  132. data/lib/rex/parser/retina_xml.rb +109 -0
  133. data/lib/rex/payloads.rb +1 -0
  134. data/lib/rex/payloads/win32.rb +2 -0
  135. data/lib/rex/payloads/win32/common.rb +26 -0
  136. data/lib/rex/payloads/win32/kernel.rb +53 -0
  137. data/lib/rex/payloads/win32/kernel/common.rb +54 -0
  138. data/lib/rex/payloads/win32/kernel/migration.rb +12 -0
  139. data/lib/rex/payloads/win32/kernel/recovery.rb +50 -0
  140. data/lib/rex/payloads/win32/kernel/stager.rb +194 -0
  141. data/lib/rex/peparsey.rb +12 -0
  142. data/lib/rex/peparsey/exceptions.rb +32 -0
  143. data/lib/rex/peparsey/pe.rb +212 -0
  144. data/lib/rex/peparsey/pe_memdump.rb +63 -0
  145. data/lib/rex/peparsey/pebase.rb +1680 -0
  146. data/lib/rex/peparsey/section.rb +136 -0
  147. data/lib/rex/pescan.rb +13 -0
  148. data/lib/rex/pescan/analyze.rb +309 -0
  149. data/lib/rex/pescan/scanner.rb +206 -0
  150. data/lib/rex/pescan/search.rb +56 -0
  151. data/lib/rex/platforms.rb +1 -0
  152. data/lib/rex/platforms/windows.rb +51 -0
  153. data/lib/rex/poly.rb +132 -0
  154. data/lib/rex/poly/block.rb +477 -0
  155. data/lib/rex/poly/register.rb +100 -0
  156. data/lib/rex/poly/register/x86.rb +40 -0
  157. data/lib/rex/post.rb +8 -0
  158. data/lib/rex/post/dir.rb +51 -0
  159. data/lib/rex/post/file.rb +172 -0
  160. data/lib/rex/post/file_stat.rb +220 -0
  161. data/lib/rex/post/gen.pl +13 -0
  162. data/lib/rex/post/io.rb +182 -0
  163. data/lib/rex/post/meterpreter.rb +4 -0
  164. data/lib/rex/post/meterpreter/channel.rb +445 -0
  165. data/lib/rex/post/meterpreter/channel_container.rb +54 -0
  166. data/lib/rex/post/meterpreter/channels/pool.rb +160 -0
  167. data/lib/rex/post/meterpreter/channels/pools/file.rb +62 -0
  168. data/lib/rex/post/meterpreter/channels/pools/stream_pool.rb +103 -0
  169. data/lib/rex/post/meterpreter/channels/stream.rb +87 -0
  170. data/lib/rex/post/meterpreter/client.rb +364 -0
  171. data/lib/rex/post/meterpreter/client_core.rb +274 -0
  172. data/lib/rex/post/meterpreter/dependencies.rb +3 -0
  173. data/lib/rex/post/meterpreter/extension.rb +32 -0
  174. data/lib/rex/post/meterpreter/extensions/espia/espia.rb +58 -0
  175. data/lib/rex/post/meterpreter/extensions/espia/tlv.rb +16 -0
  176. data/lib/rex/post/meterpreter/extensions/incognito/incognito.rb +94 -0
  177. data/lib/rex/post/meterpreter/extensions/incognito/tlv.rb +21 -0
  178. data/lib/rex/post/meterpreter/extensions/networkpug/networkpug.rb +57 -0
  179. data/lib/rex/post/meterpreter/extensions/networkpug/tlv.rb +15 -0
  180. data/lib/rex/post/meterpreter/extensions/priv/fs.rb +118 -0
  181. data/lib/rex/post/meterpreter/extensions/priv/passwd.rb +61 -0
  182. data/lib/rex/post/meterpreter/extensions/priv/priv.rb +111 -0
  183. data/lib/rex/post/meterpreter/extensions/priv/tlv.rb +28 -0
  184. data/lib/rex/post/meterpreter/extensions/sniffer/sniffer.rb +101 -0
  185. data/lib/rex/post/meterpreter/extensions/sniffer/tlv.rb +26 -0
  186. data/lib/rex/post/meterpreter/extensions/stdapi/constants.rb +333 -0
  187. data/lib/rex/post/meterpreter/extensions/stdapi/fs/dir.rb +282 -0
  188. data/lib/rex/post/meterpreter/extensions/stdapi/fs/file.rb +266 -0
  189. data/lib/rex/post/meterpreter/extensions/stdapi/fs/file_stat.rb +103 -0
  190. data/lib/rex/post/meterpreter/extensions/stdapi/fs/io.rb +48 -0
  191. data/lib/rex/post/meterpreter/extensions/stdapi/net/config.rb +144 -0
  192. data/lib/rex/post/meterpreter/extensions/stdapi/net/interface.rb +73 -0
  193. data/lib/rex/post/meterpreter/extensions/stdapi/net/route.rb +56 -0
  194. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket.rb +137 -0
  195. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_client_channel.rb +180 -0
  196. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_server_channel.rb +167 -0
  197. data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/udp_channel.rb +208 -0
  198. data/lib/rex/post/meterpreter/extensions/stdapi/railgun.rb.ts.rb +6 -0
  199. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb +38106 -0
  200. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb.ut.rb +31 -0
  201. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb +47 -0
  202. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb.ut.rb +36 -0
  203. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_advapi32.rb +1818 -0
  204. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_iphlpapi.rb +96 -0
  205. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_kernel32.rb +3848 -0
  206. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_netapi32.rb +26 -0
  207. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ntdll.rb +153 -0
  208. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_shell32.rb +21 -0
  209. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_user32.rb +3169 -0
  210. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ws2_32.rb +599 -0
  211. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll.rb +318 -0
  212. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb +100 -0
  213. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb.ut.rb +42 -0
  214. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb +148 -0
  215. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb.ut.rb +127 -0
  216. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/multicall.rb +309 -0
  217. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb +204 -0
  218. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/tlv.rb +51 -0
  219. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/util.rb +630 -0
  220. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb +75 -0
  221. data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb.ut.rb +103 -0
  222. data/lib/rex/post/meterpreter/extensions/stdapi/stdapi.rb +149 -0
  223. data/lib/rex/post/meterpreter/extensions/stdapi/sys/config.rb +97 -0
  224. data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log.rb +192 -0
  225. data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log_subsystem/event_record.rb +41 -0
  226. data/lib/rex/post/meterpreter/extensions/stdapi/sys/power.rb +61 -0
  227. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process.rb +370 -0
  228. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/image.rb +129 -0
  229. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/io.rb +55 -0
  230. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/memory.rb +336 -0
  231. data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/thread.rb +141 -0
  232. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry.rb +279 -0
  233. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_key.rb +193 -0
  234. data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_value.rb +102 -0
  235. data/lib/rex/post/meterpreter/extensions/stdapi/sys/thread.rb +180 -0
  236. data/lib/rex/post/meterpreter/extensions/stdapi/tlv.rb +211 -0
  237. data/lib/rex/post/meterpreter/extensions/stdapi/ui.rb +227 -0
  238. data/lib/rex/post/meterpreter/extensions/stdapi/webcam/webcam.rb +63 -0
  239. data/lib/rex/post/meterpreter/inbound_packet_handler.rb +30 -0
  240. data/lib/rex/post/meterpreter/object_aliases.rb +83 -0
  241. data/lib/rex/post/meterpreter/packet.rb +688 -0
  242. data/lib/rex/post/meterpreter/packet_dispatcher.rb +431 -0
  243. data/lib/rex/post/meterpreter/packet_parser.rb +94 -0
  244. data/lib/rex/post/meterpreter/packet_response_waiter.rb +83 -0
  245. data/lib/rex/post/meterpreter/ui/console.rb +137 -0
  246. data/lib/rex/post/meterpreter/ui/console/command_dispatcher.rb +62 -0
  247. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/core.rb +730 -0
  248. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/espia.rb +108 -0
  249. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/incognito.rb +241 -0
  250. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/networkpug.rb +231 -0
  251. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv.rb +61 -0
  252. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/elevate.rb +98 -0
  253. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/passwd.rb +51 -0
  254. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/timestomp.rb +132 -0
  255. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/sniffer.rb +187 -0
  256. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi.rb +65 -0
  257. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/fs.rb +442 -0
  258. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/net.rb +298 -0
  259. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/sys.rb +486 -0
  260. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/ui.rb +315 -0
  261. data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/webcam.rb +157 -0
  262. data/lib/rex/post/meterpreter/ui/console/interactive_channel.rb +95 -0
  263. data/lib/rex/post/permission.rb +26 -0
  264. data/lib/rex/post/process.rb +57 -0
  265. data/lib/rex/post/thread.rb +57 -0
  266. data/lib/rex/post/ui.rb +52 -0
  267. data/lib/rex/proto.rb +13 -0
  268. data/lib/rex/proto.rb.ts.rb +8 -0
  269. data/lib/rex/proto/dcerpc.rb +6 -0
  270. data/lib/rex/proto/dcerpc.rb.ts.rb +9 -0
  271. data/lib/rex/proto/dcerpc/client.rb +361 -0
  272. data/lib/rex/proto/dcerpc/client.rb.ut.rb +491 -0
  273. data/lib/rex/proto/dcerpc/exceptions.rb +150 -0
  274. data/lib/rex/proto/dcerpc/handle.rb +47 -0
  275. data/lib/rex/proto/dcerpc/handle.rb.ut.rb +85 -0
  276. data/lib/rex/proto/dcerpc/ndr.rb +72 -0
  277. data/lib/rex/proto/dcerpc/ndr.rb.ut.rb +41 -0
  278. data/lib/rex/proto/dcerpc/packet.rb +253 -0
  279. data/lib/rex/proto/dcerpc/packet.rb.ut.rb +56 -0
  280. data/lib/rex/proto/dcerpc/response.rb +187 -0
  281. data/lib/rex/proto/dcerpc/response.rb.ut.rb +15 -0
  282. data/lib/rex/proto/dcerpc/uuid.rb +84 -0
  283. data/lib/rex/proto/dcerpc/uuid.rb.ut.rb +46 -0
  284. data/lib/rex/proto/dhcp.rb +7 -0
  285. data/lib/rex/proto/dhcp/constants.rb +33 -0
  286. data/lib/rex/proto/dhcp/server.rb +292 -0
  287. data/lib/rex/proto/drda.rb +5 -0
  288. data/lib/rex/proto/drda.rb.ts.rb +17 -0
  289. data/lib/rex/proto/drda/constants.rb +49 -0
  290. data/lib/rex/proto/drda/constants.rb.ut.rb +23 -0
  291. data/lib/rex/proto/drda/packet.rb +252 -0
  292. data/lib/rex/proto/drda/packet.rb.ut.rb +109 -0
  293. data/lib/rex/proto/drda/utils.rb +123 -0
  294. data/lib/rex/proto/drda/utils.rb.ut.rb +84 -0
  295. data/lib/rex/proto/http.rb +5 -0
  296. data/lib/rex/proto/http.rb.ts.rb +12 -0
  297. data/lib/rex/proto/http/client.rb +821 -0
  298. data/lib/rex/proto/http/client.rb.ut.rb +95 -0
  299. data/lib/rex/proto/http/handler.rb +46 -0
  300. data/lib/rex/proto/http/handler/erb.rb +128 -0
  301. data/lib/rex/proto/http/handler/erb.rb.ut.rb +21 -0
  302. data/lib/rex/proto/http/handler/erb.rb.ut.rb.rhtml +1 -0
  303. data/lib/rex/proto/http/handler/proc.rb +60 -0
  304. data/lib/rex/proto/http/handler/proc.rb.ut.rb +24 -0
  305. data/lib/rex/proto/http/header.rb +161 -0
  306. data/lib/rex/proto/http/header.rb.ut.rb +46 -0
  307. data/lib/rex/proto/http/packet.rb +407 -0
  308. data/lib/rex/proto/http/packet.rb.ut.rb +165 -0
  309. data/lib/rex/proto/http/request.rb +356 -0
  310. data/lib/rex/proto/http/request.rb.ut.rb +214 -0
  311. data/lib/rex/proto/http/response.rb +90 -0
  312. data/lib/rex/proto/http/response.rb.ut.rb +149 -0
  313. data/lib/rex/proto/http/server.rb +369 -0
  314. data/lib/rex/proto/http/server.rb.ut.rb +79 -0
  315. data/lib/rex/proto/ntlm.rb +7 -0
  316. data/lib/rex/proto/ntlm.rb.ut.rb +177 -0
  317. data/lib/rex/proto/ntlm/base.rb +326 -0
  318. data/lib/rex/proto/ntlm/constants.rb +74 -0
  319. data/lib/rex/proto/ntlm/crypt.rb +415 -0
  320. data/lib/rex/proto/ntlm/exceptions.rb +16 -0
  321. data/lib/rex/proto/ntlm/message.rb +536 -0
  322. data/lib/rex/proto/ntlm/utils.rb +764 -0
  323. data/lib/rex/proto/proxy/socks4a.rb +440 -0
  324. data/lib/rex/proto/rfb.rb +19 -0
  325. data/lib/rex/proto/rfb.rb.ut.rb +37 -0
  326. data/lib/rex/proto/rfb/cipher.rb +84 -0
  327. data/lib/rex/proto/rfb/client.rb +207 -0
  328. data/lib/rex/proto/rfb/constants.rb +52 -0
  329. data/lib/rex/proto/smb.rb +7 -0
  330. data/lib/rex/proto/smb.rb.ts.rb +8 -0
  331. data/lib/rex/proto/smb/client.rb +1952 -0
  332. data/lib/rex/proto/smb/client.rb.ut.rb +223 -0
  333. data/lib/rex/proto/smb/constants.rb +1047 -0
  334. data/lib/rex/proto/smb/constants.rb.ut.rb +18 -0
  335. data/lib/rex/proto/smb/crypt.rb +36 -0
  336. data/lib/rex/proto/smb/evasions.rb +66 -0
  337. data/lib/rex/proto/smb/exceptions.rb +858 -0
  338. data/lib/rex/proto/smb/simpleclient.rb +306 -0
  339. data/lib/rex/proto/smb/simpleclient.rb.ut.rb +128 -0
  340. data/lib/rex/proto/smb/utils.rb +103 -0
  341. data/lib/rex/proto/smb/utils.rb.ut.rb +20 -0
  342. data/lib/rex/proto/sunrpc.rb +1 -0
  343. data/lib/rex/proto/sunrpc/client.rb +195 -0
  344. data/lib/rex/proto/tftp.rb +12 -0
  345. data/lib/rex/proto/tftp/constants.rb +39 -0
  346. data/lib/rex/proto/tftp/server.rb +497 -0
  347. data/lib/rex/proto/tftp/server.rb.ut.rb +28 -0
  348. data/lib/rex/script.rb +42 -0
  349. data/lib/rex/script/base.rb +59 -0
  350. data/lib/rex/script/meterpreter.rb +15 -0
  351. data/lib/rex/script/shell.rb +9 -0
  352. data/lib/rex/service.rb +48 -0
  353. data/lib/rex/service_manager.rb +141 -0
  354. data/lib/rex/service_manager.rb.ut.rb +32 -0
  355. data/lib/rex/services/local_relay.rb +423 -0
  356. data/lib/rex/socket.rb +684 -0
  357. data/lib/rex/socket.rb.ut.rb +107 -0
  358. data/lib/rex/socket/comm.rb +119 -0
  359. data/lib/rex/socket/comm/local.rb +412 -0
  360. data/lib/rex/socket/comm/local.rb.ut.rb +75 -0
  361. data/lib/rex/socket/ip.rb +130 -0
  362. data/lib/rex/socket/parameters.rb +345 -0
  363. data/lib/rex/socket/parameters.rb.ut.rb +51 -0
  364. data/lib/rex/socket/range_walker.rb +346 -0
  365. data/lib/rex/socket/range_walker.rb.ut.rb +55 -0
  366. data/lib/rex/socket/ssl_tcp.rb +184 -0
  367. data/lib/rex/socket/ssl_tcp.rb.ut.rb +39 -0
  368. data/lib/rex/socket/ssl_tcp_server.rb +122 -0
  369. data/lib/rex/socket/ssl_tcp_server.rb.ut.rb +61 -0
  370. data/lib/rex/socket/subnet_walker.rb +75 -0
  371. data/lib/rex/socket/subnet_walker.rb.ut.rb +28 -0
  372. data/lib/rex/socket/switch_board.rb +278 -0
  373. data/lib/rex/socket/switch_board.rb.ut.rb +52 -0
  374. data/lib/rex/socket/tcp.rb +76 -0
  375. data/lib/rex/socket/tcp.rb.ut.rb +64 -0
  376. data/lib/rex/socket/tcp_server.rb +66 -0
  377. data/lib/rex/socket/tcp_server.rb.ut.rb +44 -0
  378. data/lib/rex/socket/udp.rb +164 -0
  379. data/lib/rex/socket/udp.rb.ut.rb +44 -0
  380. data/lib/rex/struct2.rb +5 -0
  381. data/lib/rex/struct2/c_struct.rb +181 -0
  382. data/lib/rex/struct2/c_struct_template.rb +39 -0
  383. data/lib/rex/struct2/constant.rb +26 -0
  384. data/lib/rex/struct2/element.rb +44 -0
  385. data/lib/rex/struct2/generic.rb +73 -0
  386. data/lib/rex/struct2/restraint.rb +54 -0
  387. data/lib/rex/struct2/s_string.rb +72 -0
  388. data/lib/rex/struct2/s_struct.rb +111 -0
  389. data/lib/rex/sync.rb +6 -0
  390. data/lib/rex/sync/event.rb +94 -0
  391. data/lib/rex/sync/read_write_lock.rb +176 -0
  392. data/lib/rex/sync/ref.rb +57 -0
  393. data/lib/rex/sync/thread_safe.rb +82 -0
  394. data/lib/rex/test.rb +35 -0
  395. data/lib/rex/text.rb +1163 -0
  396. data/lib/rex/text.rb.ut.rb +190 -0
  397. data/lib/rex/thread_factory.rb +42 -0
  398. data/lib/rex/time.rb +65 -0
  399. data/lib/rex/transformer.rb +115 -0
  400. data/lib/rex/transformer.rb.ut.rb +38 -0
  401. data/lib/rex/ui.rb +21 -0
  402. data/lib/rex/ui/interactive.rb +298 -0
  403. data/lib/rex/ui/output.rb +78 -0
  404. data/lib/rex/ui/output/none.rb +18 -0
  405. data/lib/rex/ui/progress_tracker.rb +96 -0
  406. data/lib/rex/ui/subscriber.rb +149 -0
  407. data/lib/rex/ui/text/color.rb +97 -0
  408. data/lib/rex/ui/text/color.rb.ut.rb +18 -0
  409. data/lib/rex/ui/text/dispatcher_shell.rb +467 -0
  410. data/lib/rex/ui/text/input.rb +117 -0
  411. data/lib/rex/ui/text/input/buffer.rb +75 -0
  412. data/lib/rex/ui/text/input/readline.rb +129 -0
  413. data/lib/rex/ui/text/input/socket.rb +95 -0
  414. data/lib/rex/ui/text/input/stdio.rb +45 -0
  415. data/lib/rex/ui/text/irb_shell.rb +57 -0
  416. data/lib/rex/ui/text/output.rb +80 -0
  417. data/lib/rex/ui/text/output/buffer.rb +61 -0
  418. data/lib/rex/ui/text/output/file.rb +43 -0
  419. data/lib/rex/ui/text/output/socket.rb +43 -0
  420. data/lib/rex/ui/text/output/stdio.rb +40 -0
  421. data/lib/rex/ui/text/progress_tracker.rb +56 -0
  422. data/lib/rex/ui/text/progress_tracker.rb.ut.rb +34 -0
  423. data/lib/rex/ui/text/shell.rb +328 -0
  424. data/lib/rex/ui/text/table.rb +279 -0
  425. data/lib/rex/ui/text/table.rb.ut.rb +55 -0
  426. data/lib/rex/zip.rb +93 -0
  427. data/lib/rex/zip/archive.rb +184 -0
  428. data/lib/rex/zip/blocks.rb +182 -0
  429. data/lib/rex/zip/entry.rb +104 -0
  430. data/lib/rex/zip/samples/comment.rb +32 -0
  431. data/lib/rex/zip/samples/mkwar.rb +138 -0
  432. data/lib/rex/zip/samples/mkzip.rb +19 -0
  433. data/lib/rex/zip/samples/recursive.rb +58 -0
  434. metadata +434 -1
@@ -0,0 +1,57 @@
1
+ module Rex
2
+ module Ui
3
+ module Text
4
+
5
+ ###
6
+ #
7
+ # This class wraps the creation of an IRB shell.
8
+ #
9
+ ###
10
+ class IrbShell
11
+
12
+ @@IrbInitialized = false
13
+
14
+ def initialize(binding)
15
+ @binding_ctx = binding
16
+ end
17
+
18
+ #
19
+ # Runs the IRB shell until completion. The binding parameter initializes
20
+ # IRB to the appropriate binding context.
21
+ #
22
+ def run
23
+ # Initialize IRB by setting up its internal configuration hash and
24
+ # stuff.
25
+ if (@@IrbInitialized == false)
26
+ load('irb.rb')
27
+
28
+ IRB.setup(nil)
29
+ IRB.conf[:PROMPT_MODE] = :SIMPLE
30
+
31
+ @@IrbInitialized = true
32
+ end
33
+
34
+ # Create a new IRB instance
35
+ irb = IRB::Irb.new(IRB::WorkSpace.new(@binding_ctx))
36
+
37
+ # Set the primary irb context so that exit and other intrinsic
38
+ # commands will work.
39
+ IRB.conf[:MAIN_CONTEXT] = irb.context
40
+
41
+ # Trap interrupt
42
+ old_sigint = trap("SIGINT") do
43
+ irb.signal_handle
44
+ end
45
+
46
+ # Keep processing input until the cows come home...
47
+ catch(:IRB_EXIT) do
48
+ irb.eval_input
49
+ end
50
+
51
+ trap("SIGINT", old_sigint)
52
+ end
53
+
54
+ end
55
+ end
56
+ end
57
+ end
@@ -0,0 +1,80 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This class implements text-based output but is not
10
+ # tied to an output medium.
11
+ #
12
+ ###
13
+ class Output < Rex::Ui::Output
14
+
15
+ require 'rex/ui/text/output/stdio'
16
+ require 'rex/ui/text/output/socket'
17
+ require 'rex/ui/text/output/buffer'
18
+ require 'rex/ui/text/output/file'
19
+ require 'rex/ui/text/color'
20
+
21
+ include Rex::Ui::Text::Color
22
+
23
+ def initialize
24
+ @config = {
25
+ :color => :auto, # true, false, :auto
26
+ }
27
+ super
28
+ end
29
+ attr_reader :config
30
+
31
+ def disable_color
32
+ @config[:color] = false
33
+ end
34
+
35
+ def enable_color
36
+ @config[:color] = true
37
+ end
38
+
39
+ def auto_color
40
+ @config[:color] = :auto
41
+ end
42
+
43
+ def update_prompt(prompt = nil)
44
+ return if prompt.nil?
45
+ substitute_colors(prompt, true)
46
+ end
47
+
48
+ def print_error(msg = '')
49
+ print_line("%bld%red[-]%clr #{msg}")
50
+ end
51
+
52
+ def print_good(msg = '')
53
+ print_line("%bld%grn[+]%clr #{msg}")
54
+ end
55
+
56
+ def print_debug(msg = '')
57
+ print_line("%bld%cya[!]%clr #{msg}")
58
+ end
59
+
60
+ def print_status(msg = '')
61
+ print_line("%bld%blu[*]%clr #{msg}")
62
+ end
63
+
64
+ def print_line(msg = '')
65
+ print(msg + "\n")
66
+ end
67
+
68
+ def print(msg = '')
69
+ print_raw(substitute_colors(msg))
70
+ end
71
+
72
+ def reset
73
+ end
74
+
75
+ end
76
+
77
+ end
78
+ end
79
+ end
80
+
@@ -0,0 +1,61 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This class implements output against a buffer.
10
+ #
11
+ ###
12
+ class Output::Buffer < Rex::Ui::Text::Output
13
+
14
+ #
15
+ # Initializes an output buffer.
16
+ #
17
+ def initialize
18
+ self.buf = ''
19
+ end
20
+
21
+ def supports_color?
22
+ false
23
+ end
24
+
25
+ #
26
+ # Appends the supplied message to the output buffer.
27
+ #
28
+ def print_raw(msg = '')
29
+ self.buf += msg || ''
30
+
31
+ msg
32
+ end
33
+
34
+
35
+ #
36
+ # Read everything out of the buffer and reset it
37
+ #
38
+ def dump_buffer
39
+ self.buf ||= ''
40
+ buffer = self.buf.dup
41
+ reset()
42
+ buffer
43
+ end
44
+
45
+ #
46
+ # Reset the buffer to an empty string.
47
+ #
48
+ def reset
49
+ self.buf = ''
50
+ end
51
+
52
+ #
53
+ # The underlying buffer state.
54
+ #
55
+ attr_accessor :buf
56
+
57
+ end
58
+
59
+ end
60
+ end
61
+ end
@@ -0,0 +1,43 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This class implements output against a file
10
+ #
11
+ ###
12
+ class Output::File < Rex::Ui::Text::Output
13
+
14
+ attr_accessor :fd
15
+
16
+ def initialize(path)
17
+ self.fd = ::File.open(path, "wb")
18
+ end
19
+
20
+ def supports_color?
21
+ false
22
+ end
23
+
24
+ #
25
+ # Prints the supplied message to file output.
26
+ #
27
+ def print_raw(msg = '')
28
+ return if not self.fd
29
+ self.fd.write(msg)
30
+ self.fd.flush
31
+ msg
32
+ end
33
+
34
+ def close
35
+ self.fd.close if self.fd
36
+ self.fd = nil
37
+ end
38
+ end
39
+
40
+ end
41
+ end
42
+ end
43
+
@@ -0,0 +1,43 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This class implements the output interface against a socket.
10
+ #
11
+ ###
12
+ class Output::Socket < Rex::Ui::Text::Output
13
+
14
+ def initialize(sock)
15
+ @sock = sock
16
+ super()
17
+ end
18
+
19
+ def supports_color?
20
+ case config[:color]
21
+ when true
22
+ # Allow color if the user forces it on
23
+ return true
24
+ else
25
+ false
26
+ end
27
+ end
28
+
29
+ #
30
+ # Prints the supplied message to the socket.
31
+ #
32
+ def print_raw(msg = '')
33
+ @sock.write(msg)
34
+ @sock.flush
35
+
36
+ msg
37
+ end
38
+ end
39
+
40
+ end
41
+ end
42
+ end
43
+
@@ -0,0 +1,40 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This class implements output against standard out.
10
+ #
11
+ ###
12
+ class Output::Stdio < Rex::Ui::Text::Output
13
+
14
+ def supports_color?
15
+ case config[:color]
16
+ when true
17
+ return true
18
+ when false
19
+ return false
20
+ else # auto
21
+ term = Rex::Compat.getenv('TERM')
22
+ return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
23
+ end
24
+ end
25
+
26
+ #
27
+ # Prints the supplied message to standard output.
28
+ #
29
+ def print_raw(msg = '')
30
+ $stdout.print(msg)
31
+ $stdout.flush
32
+
33
+ msg
34
+ end
35
+ end
36
+
37
+ end
38
+ end
39
+ end
40
+
@@ -0,0 +1,56 @@
1
+ require 'rex/ui/progress_tracker'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # This module implements progress tracking against a text interface.
10
+ #
11
+ ###
12
+ class ProgressTracker < Rex::Ui::ProgressTracker
13
+
14
+ def initialize(output)
15
+ self.output = output
16
+ self.start = 0
17
+ self.stop = 0
18
+ self.pos = 0
19
+ end
20
+
21
+ #
22
+ # Passes a generic status message that isn't necessarily associated
23
+ # with a step event.
24
+ #
25
+ def status(msg = '')
26
+ output.print_status(msg)
27
+ end
28
+
29
+ #
30
+ # Updates the status associated with the current step.
31
+ #
32
+ def step_status(msg = '')
33
+ output.print_status("#{pos}: #{msg}") if (msg and msg.length > 0)
34
+ end
35
+
36
+ #
37
+ # An error occurred that may result in aborting the progress.
38
+ #
39
+ def error(msg = '')
40
+ output.print_error(msg)
41
+ end
42
+
43
+ #
44
+ # Progress has been aborted, the reason is supplied in msg.
45
+ #
46
+ def abort(msg = '')
47
+ output.print_error("fatal: #{msg}")
48
+ end
49
+
50
+ attr_accessor :output
51
+
52
+ end
53
+
54
+ end
55
+ end
56
+ end
@@ -0,0 +1,34 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ $:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
4
+
5
+ require 'test/unit'
6
+ require 'rex/ui'
7
+
8
+ class Rex::Ui::Text::ProgressTracker::UnitTest < Test::Unit::TestCase
9
+
10
+ def test_stuff
11
+ output = Rex::Ui::Text::Output::Buffer.new
12
+ pt = Rex::Ui::Text::ProgressTracker.new(output)
13
+
14
+ pt.range = 1..10
15
+
16
+ assert_equal(1, pt.start)
17
+ assert_equal(10, pt.stop)
18
+
19
+ pt.start = 2
20
+ assert_equal(2, pt.start)
21
+ pt.stop = 9
22
+ assert_equal(9, pt.stop)
23
+ assert_equal(2, pt.pos)
24
+ assert_equal('', output.buf)
25
+ assert_equal(3, pt.step)
26
+ assert_equal(4, pt.step("test"))
27
+ assert_equal("[*] 4: test\n", output.buf)
28
+ output.reset
29
+ assert_equal("[-] bad\n", pt.error("bad"))
30
+ output.reset
31
+ assert_equal("[-] fatal: bad\n", pt.abort("bad"))
32
+ end
33
+
34
+ end
@@ -0,0 +1,328 @@
1
+ require 'rex/ui'
2
+
3
+ module Rex
4
+ module Ui
5
+ module Text
6
+
7
+ ###
8
+ #
9
+ # The shell class provides a command-prompt style interface in a
10
+ # generic fashion.
11
+ #
12
+ ###
13
+ module Shell
14
+
15
+ ###
16
+ #
17
+ # This module is meant to be mixed into an input medium class instance as a
18
+ # means of extending it to display a prompt before each call to gets.
19
+ #
20
+ ###
21
+ module InputShell
22
+ attr_accessor :prompt, :output
23
+
24
+ def pgets
25
+ output.print(prompt)
26
+ output.flush
27
+
28
+ output.prompting
29
+ buf = gets
30
+ output.prompting(false)
31
+
32
+ buf
33
+ end
34
+ end
35
+
36
+ #
37
+ # Initializes a shell that has a prompt and can be interacted with.
38
+ #
39
+ def initialize(prompt, prompt_char = '>', histfile = nil)
40
+ # Set the stop flag to false
41
+ self.stop_flag = false
42
+ self.disable_output = false
43
+ self.stop_count = 0
44
+
45
+ # Initialize the prompt
46
+ self.init_prompt = prompt
47
+ self.prompt_char = prompt_char
48
+
49
+ self.histfile = histfile
50
+ self.hist_last_saved = 0
51
+ end
52
+
53
+ def init_tab_complete
54
+ if (self.input and self.input.supports_readline)
55
+ self.input = Input::Readline.new(lambda { |str| tab_complete(str) })
56
+ if Readline::HISTORY.length == 0 and histfile and File.exists?(histfile)
57
+ File.readlines(histfile).each { |e|
58
+ Readline::HISTORY << e.chomp
59
+ }
60
+ self.hist_last_saved = Readline::HISTORY.length
61
+ end
62
+ self.input.output = self.output
63
+ update_prompt(input.prompt)
64
+ end
65
+ end
66
+
67
+ #
68
+ # Initializes the user interface input/output classes.
69
+ #
70
+ def init_ui(in_input = nil, in_output = nil)
71
+ # Initialize the input and output methods
72
+ self.input = in_input
73
+ self.output = in_output
74
+
75
+ if (self.input)
76
+ # Extend the input medium as an input shell if the input medium
77
+ # isn't intrinsicly a shell.
78
+ if (self.input.intrinsic_shell? == false)
79
+ self.input.extend(InputShell)
80
+ end
81
+
82
+ self.input.output = self.output
83
+ end
84
+ update_prompt('')
85
+ end
86
+
87
+ #
88
+ # Resets the user interface handles.
89
+ #
90
+ def reset_ui
91
+ init_ui
92
+ end
93
+
94
+ #
95
+ # Sets the log source that should be used for logging input and output.
96
+ #
97
+ def set_log_source(log_source)
98
+ self.log_source = log_source
99
+ end
100
+
101
+ #
102
+ # Unsets the log source so that logging becomes disabled.
103
+ #
104
+ def unset_log_source
105
+ set_log_source(nil)
106
+ end
107
+
108
+ #
109
+ # Performs tab completion on the supplied string.
110
+ #
111
+ def tab_complete(str)
112
+ return tab_complete_proc(str) if (tab_complete_proc)
113
+ end
114
+
115
+ #
116
+ # Run the command processing loop.
117
+ #
118
+ def run(&block)
119
+
120
+ begin
121
+
122
+ while true
123
+ # If the stop flag was set or we've hit EOF, break out
124
+ break if (self.stop_flag or self.stop_count > 1)
125
+
126
+ init_tab_complete
127
+ line = input.pgets
128
+ log_output(input.prompt)
129
+
130
+ # If a block was passed in, pass the line to it. If it returns true,
131
+ # break out of the shell loop.
132
+ if (block)
133
+ break if (line == nil or block.call(line))
134
+ elsif(input.eof? or line == nil)
135
+ # If you have sessions active, this will give you a shot to exit gravefully
136
+ # If you really are ambitious, 2 eofs will kick this out
137
+ self.stop_count += 1
138
+ next if(self.stop_count > 1)
139
+ run_single("quit")
140
+ else
141
+ # Otherwise, call what should be an overriden instance method to
142
+ # process the line.
143
+ ret = run_single(line)
144
+ # don't bother saving lines that couldn't be found as a
145
+ # command, create the file if it doesn't exist
146
+ if ret and self.histfile
147
+ File.open(self.histfile, "a+") { |f|
148
+ f.puts(line)
149
+ }
150
+ end
151
+ self.stop_count = 0
152
+ end
153
+
154
+ end
155
+ # Prevent accidental console quits
156
+ rescue ::Interrupt
157
+ output.print("Interrupt: use the 'exit' command to quit\n")
158
+ retry
159
+ end
160
+ end
161
+
162
+ #
163
+ # Stop processing user input.
164
+ #
165
+ def stop
166
+ self.stop_flag = true
167
+ end
168
+
169
+ #
170
+ # Checks to see if the shell has stopped.
171
+ #
172
+ def stopped?
173
+ self.stop_flag
174
+ end
175
+
176
+ #
177
+ # Change the input prompt.
178
+ #
179
+ def update_prompt(prompt = nil, new_prompt_char = nil)
180
+ if (self.input)
181
+ if (prompt)
182
+ new_prompt = self.init_prompt + ' ' + prompt + prompt_char + ' '
183
+ else
184
+ new_prompt = self.prompt || ''
185
+ end
186
+
187
+ # Save the prompt before any substitutions
188
+ self.prompt = new_prompt
189
+
190
+ # Set the actual prompt to the saved prompt with any substitutions
191
+ # or updates from our output driver, be they color or whatever
192
+ self.input.prompt = self.output.update_prompt(new_prompt)
193
+ self.prompt_char = new_prompt_char if (new_prompt_char)
194
+ end
195
+ end
196
+
197
+ #
198
+ # Output shortcuts
199
+ #
200
+
201
+ #
202
+ # Prints an error message to the output handle.
203
+ #
204
+ def print_error(msg='')
205
+ return if (output.nil?)
206
+
207
+ self.on_print_proc.call(msg) if self.on_print_proc
208
+ # Errors are not subject to disabled output
209
+ log_output(output.print_error(msg))
210
+ end
211
+
212
+ #
213
+ # Prints a status message to the output handle.
214
+ #
215
+ def print_status(msg='')
216
+ return if (disable_output == true)
217
+
218
+ self.on_print_proc.call(msg) if self.on_print_proc
219
+ log_output(output.print_status(msg))
220
+ end
221
+
222
+ #
223
+ # Prints a good message to the output handle.
224
+ #
225
+ def print_good(msg='')
226
+ return if (disable_output == true)
227
+
228
+ self.on_print_proc.call(msg) if self.on_print_proc
229
+ log_output(output.print_good(msg))
230
+ end
231
+
232
+ #
233
+ # Prints a line of text to the output handle.
234
+ #
235
+ def print_line(msg='')
236
+ return if (disable_output == true)
237
+
238
+ self.on_print_proc.call(msg) if self.on_print_proc
239
+ log_output(output.print_line(msg))
240
+ end
241
+
242
+ #
243
+ # Prints a raw message to the output handle.
244
+ #
245
+ def print(msg='')
246
+ return if (disable_output == true)
247
+ self.on_print_proc.call(msg) if self.on_print_proc
248
+ log_output(output.print(msg))
249
+ end
250
+
251
+ #
252
+ # Whether or not output has been disabled.
253
+ #
254
+ attr_accessor :disable_output
255
+ #
256
+ # The input handle to read user input from.
257
+ #
258
+ attr_reader :input
259
+ #
260
+ # The output handle to write output to.
261
+ #
262
+ attr_reader :output
263
+
264
+ attr_accessor :on_command_proc
265
+ attr_accessor :on_print_proc
266
+
267
+ protected
268
+
269
+ #
270
+ # Parse a line into an array of arguments.
271
+ #
272
+ def parse_line(line)
273
+ log_input(line)
274
+
275
+ line.gsub!(/(\r|\n)/, '')
276
+
277
+ begin
278
+ return args = Rex::Parser::Arguments.from_s(line)
279
+ rescue ::ArgumentError
280
+ print_error("Parse error: #{$!}")
281
+ end
282
+
283
+ return []
284
+ end
285
+
286
+ #
287
+ # Print the prompt, but do not log it.
288
+ #
289
+ def _print_prompt(prompt)
290
+ output.print(prompt)
291
+ end
292
+
293
+ #
294
+ # Writes the supplied input to the log source if one has been registered.
295
+ #
296
+ def log_input(buf)
297
+ rlog(buf, log_source) if (log_source)
298
+ end
299
+
300
+ #
301
+ # Writes the supplied output to the log source if one has been registered.
302
+ #
303
+ def log_output(buf)
304
+ rlog(buf, log_source) if (log_source)
305
+ end
306
+
307
+ attr_writer :input, :output # :nodoc:
308
+ attr_accessor :stop_flag, :init_prompt # :nodoc:
309
+ attr_accessor :prompt # :nodoc:
310
+ attr_accessor :prompt_char, :tab_complete_proc # :nodoc:
311
+ attr_accessor :histfile # :nodoc:
312
+ attr_accessor :hist_last_saved # the number of history lines when last saved/loaded
313
+ attr_accessor :log_source, :stop_count # :nodoc:
314
+
315
+ end
316
+
317
+ ###
318
+ #
319
+ # Pseudo-shell interface that simply includes the Shell mixin.
320
+ #
321
+ ###
322
+ class PseudoShell
323
+ include Shell
324
+ end
325
+
326
+
327
+ end end end
328
+