librex 0.0.20 → 0.0.21
Sign up to get free protection for your applications and to get access to all the features.
- data/Rakefile +1 -0
- data/lib/rex/LICENSE +29 -0
- data/lib/rex/arch.rb +103 -0
- data/lib/rex/arch/sparc.rb +75 -0
- data/lib/rex/arch/sparc.rb.ut.rb +18 -0
- data/lib/rex/arch/x86.rb +513 -0
- data/lib/rex/arch/x86.rb.ut.rb +93 -0
- data/lib/rex/assembly/nasm.rb +104 -0
- data/lib/rex/assembly/nasm.rb.ut.rb +22 -0
- data/lib/rex/codepage.map +104 -0
- data/lib/rex/compat.rb +311 -0
- data/lib/rex/constants.rb +113 -0
- data/lib/rex/elfparsey.rb +11 -0
- data/lib/rex/elfparsey/elf.rb +123 -0
- data/lib/rex/elfparsey/elfbase.rb +258 -0
- data/lib/rex/elfparsey/exceptions.rb +27 -0
- data/lib/rex/elfscan.rb +12 -0
- data/lib/rex/elfscan/scanner.rb +207 -0
- data/lib/rex/elfscan/search.rb +46 -0
- data/lib/rex/encoder/alpha2.rb +31 -0
- data/lib/rex/encoder/alpha2/alpha_mixed.rb +68 -0
- data/lib/rex/encoder/alpha2/alpha_upper.rb +79 -0
- data/lib/rex/encoder/alpha2/generic.rb +114 -0
- data/lib/rex/encoder/alpha2/unicode_mixed.rb +117 -0
- data/lib/rex/encoder/alpha2/unicode_upper.rb +129 -0
- data/lib/rex/encoder/ndr.rb +89 -0
- data/lib/rex/encoder/ndr.rb.ut.rb +44 -0
- data/lib/rex/encoder/nonalpha.rb +61 -0
- data/lib/rex/encoder/nonupper.rb +64 -0
- data/lib/rex/encoder/xdr.rb +106 -0
- data/lib/rex/encoder/xdr.rb.ut.rb +29 -0
- data/lib/rex/encoder/xor.rb +69 -0
- data/lib/rex/encoder/xor/dword.rb +13 -0
- data/lib/rex/encoder/xor/dword_additive.rb +13 -0
- data/lib/rex/encoders/xor_dword.rb +35 -0
- data/lib/rex/encoders/xor_dword_additive.rb +53 -0
- data/lib/rex/encoders/xor_dword_additive.rb.ut.rb +12 -0
- data/lib/rex/encoding/xor.rb +20 -0
- data/lib/rex/encoding/xor.rb.ts.rb +14 -0
- data/lib/rex/encoding/xor/byte.rb +15 -0
- data/lib/rex/encoding/xor/byte.rb.ut.rb +21 -0
- data/lib/rex/encoding/xor/dword.rb +21 -0
- data/lib/rex/encoding/xor/dword.rb.ut.rb +15 -0
- data/lib/rex/encoding/xor/dword_additive.rb +92 -0
- data/lib/rex/encoding/xor/dword_additive.rb.ut.rb +15 -0
- data/lib/rex/encoding/xor/exceptions.rb +17 -0
- data/lib/rex/encoding/xor/generic.rb +146 -0
- data/lib/rex/encoding/xor/generic.rb.ut.rb +120 -0
- data/lib/rex/encoding/xor/qword.rb +15 -0
- data/lib/rex/encoding/xor/word.rb +21 -0
- data/lib/rex/encoding/xor/word.rb.ut.rb +13 -0
- data/lib/rex/exceptions.rb +275 -0
- data/lib/rex/exceptions.rb.ut.rb +44 -0
- data/lib/rex/exploitation/cmdstager.rb +9 -0
- data/lib/rex/exploitation/cmdstager/base.rb +175 -0
- data/lib/rex/exploitation/cmdstager/debug_asm.rb +142 -0
- data/lib/rex/exploitation/cmdstager/debug_write.rb +136 -0
- data/lib/rex/exploitation/cmdstager/tftp.rb +63 -0
- data/lib/rex/exploitation/cmdstager/vbs.rb +128 -0
- data/lib/rex/exploitation/egghunter.rb +277 -0
- data/lib/rex/exploitation/egghunter.rb.ut.rb +25 -0
- data/lib/rex/exploitation/encryptjs.rb +77 -0
- data/lib/rex/exploitation/heaplib.js.b64 +331 -0
- data/lib/rex/exploitation/heaplib.rb +94 -0
- data/lib/rex/exploitation/javascriptosdetect.rb +897 -0
- data/lib/rex/exploitation/obfuscatejs.rb +335 -0
- data/lib/rex/exploitation/omelet.rb +320 -0
- data/lib/rex/exploitation/omelet.rb.ut.rb +13 -0
- data/lib/rex/exploitation/opcodedb.rb +818 -0
- data/lib/rex/exploitation/opcodedb.rb.ut.rb +279 -0
- data/lib/rex/exploitation/seh.rb +92 -0
- data/lib/rex/exploitation/seh.rb.ut.rb +19 -0
- data/lib/rex/file.rb +112 -0
- data/lib/rex/file.rb.ut.rb +16 -0
- data/lib/rex/image_source.rb +12 -0
- data/lib/rex/image_source/disk.rb +60 -0
- data/lib/rex/image_source/image_source.rb +46 -0
- data/lib/rex/image_source/memory.rb +37 -0
- data/lib/rex/io/bidirectional_pipe.rb +157 -0
- data/lib/rex/io/datagram_abstraction.rb +35 -0
- data/lib/rex/io/ring_buffer.rb +364 -0
- data/lib/rex/io/stream.rb +319 -0
- data/lib/rex/io/stream_abstraction.rb +197 -0
- data/lib/rex/io/stream_server.rb +211 -0
- data/lib/rex/job_container.rb +187 -0
- data/lib/rex/logging.rb +4 -0
- data/lib/rex/logging/log_dispatcher.rb +179 -0
- data/lib/rex/logging/log_sink.rb +42 -0
- data/lib/rex/logging/sinks/flatfile.rb +55 -0
- data/lib/rex/logging/sinks/stderr.rb +43 -0
- data/lib/rex/machparsey.rb +9 -0
- data/lib/rex/machparsey/exceptions.rb +34 -0
- data/lib/rex/machparsey/mach.rb +209 -0
- data/lib/rex/machparsey/machbase.rb +408 -0
- data/lib/rex/machscan.rb +9 -0
- data/lib/rex/machscan/scanner.rb +217 -0
- data/lib/rex/mime.rb +9 -0
- data/lib/rex/mime/header.rb +77 -0
- data/lib/rex/mime/message.rb +144 -0
- data/lib/rex/mime/part.rb +20 -0
- data/lib/rex/nop/opty2.rb +108 -0
- data/lib/rex/nop/opty2.rb.ut.rb +23 -0
- data/lib/rex/nop/opty2_tables.rb +300 -0
- data/lib/rex/ole.rb +205 -0
- data/lib/rex/ole/clsid.rb +47 -0
- data/lib/rex/ole/difat.rb +141 -0
- data/lib/rex/ole/directory.rb +231 -0
- data/lib/rex/ole/direntry.rb +240 -0
- data/lib/rex/ole/docs/dependencies.txt +8 -0
- data/lib/rex/ole/docs/references.txt +1 -0
- data/lib/rex/ole/fat.rb +99 -0
- data/lib/rex/ole/header.rb +204 -0
- data/lib/rex/ole/minifat.rb +77 -0
- data/lib/rex/ole/propset.rb +144 -0
- data/lib/rex/ole/samples/create_ole.rb +27 -0
- data/lib/rex/ole/samples/dir.rb +35 -0
- data/lib/rex/ole/samples/dump_stream.rb +34 -0
- data/lib/rex/ole/samples/ole_info.rb +23 -0
- data/lib/rex/ole/storage.rb +395 -0
- data/lib/rex/ole/stream.rb +53 -0
- data/lib/rex/ole/substorage.rb +49 -0
- data/lib/rex/ole/util.rb +157 -0
- data/lib/rex/parser/arguments.rb +97 -0
- data/lib/rex/parser/arguments.rb.ut.rb +67 -0
- data/lib/rex/parser/ini.rb +185 -0
- data/lib/rex/parser/ini.rb.ut.rb +29 -0
- data/lib/rex/parser/ip360_aspl_xml.rb +102 -0
- data/lib/rex/parser/ip360_xml.rb +97 -0
- data/lib/rex/parser/nessus_xml.rb +118 -0
- data/lib/rex/parser/netsparker_xml.rb +94 -0
- data/lib/rex/parser/nexpose_xml.rb +136 -0
- data/lib/rex/parser/nmap_xml.rb +137 -0
- data/lib/rex/parser/retina_xml.rb +109 -0
- data/lib/rex/payloads.rb +1 -0
- data/lib/rex/payloads/win32.rb +2 -0
- data/lib/rex/payloads/win32/common.rb +26 -0
- data/lib/rex/payloads/win32/kernel.rb +53 -0
- data/lib/rex/payloads/win32/kernel/common.rb +54 -0
- data/lib/rex/payloads/win32/kernel/migration.rb +12 -0
- data/lib/rex/payloads/win32/kernel/recovery.rb +50 -0
- data/lib/rex/payloads/win32/kernel/stager.rb +194 -0
- data/lib/rex/peparsey.rb +12 -0
- data/lib/rex/peparsey/exceptions.rb +32 -0
- data/lib/rex/peparsey/pe.rb +212 -0
- data/lib/rex/peparsey/pe_memdump.rb +63 -0
- data/lib/rex/peparsey/pebase.rb +1680 -0
- data/lib/rex/peparsey/section.rb +136 -0
- data/lib/rex/pescan.rb +13 -0
- data/lib/rex/pescan/analyze.rb +309 -0
- data/lib/rex/pescan/scanner.rb +206 -0
- data/lib/rex/pescan/search.rb +56 -0
- data/lib/rex/platforms.rb +1 -0
- data/lib/rex/platforms/windows.rb +51 -0
- data/lib/rex/poly.rb +132 -0
- data/lib/rex/poly/block.rb +477 -0
- data/lib/rex/poly/register.rb +100 -0
- data/lib/rex/poly/register/x86.rb +40 -0
- data/lib/rex/post.rb +8 -0
- data/lib/rex/post/dir.rb +51 -0
- data/lib/rex/post/file.rb +172 -0
- data/lib/rex/post/file_stat.rb +220 -0
- data/lib/rex/post/gen.pl +13 -0
- data/lib/rex/post/io.rb +182 -0
- data/lib/rex/post/meterpreter.rb +4 -0
- data/lib/rex/post/meterpreter/channel.rb +445 -0
- data/lib/rex/post/meterpreter/channel_container.rb +54 -0
- data/lib/rex/post/meterpreter/channels/pool.rb +160 -0
- data/lib/rex/post/meterpreter/channels/pools/file.rb +62 -0
- data/lib/rex/post/meterpreter/channels/pools/stream_pool.rb +103 -0
- data/lib/rex/post/meterpreter/channels/stream.rb +87 -0
- data/lib/rex/post/meterpreter/client.rb +364 -0
- data/lib/rex/post/meterpreter/client_core.rb +274 -0
- data/lib/rex/post/meterpreter/dependencies.rb +3 -0
- data/lib/rex/post/meterpreter/extension.rb +32 -0
- data/lib/rex/post/meterpreter/extensions/espia/espia.rb +58 -0
- data/lib/rex/post/meterpreter/extensions/espia/tlv.rb +16 -0
- data/lib/rex/post/meterpreter/extensions/incognito/incognito.rb +94 -0
- data/lib/rex/post/meterpreter/extensions/incognito/tlv.rb +21 -0
- data/lib/rex/post/meterpreter/extensions/networkpug/networkpug.rb +57 -0
- data/lib/rex/post/meterpreter/extensions/networkpug/tlv.rb +15 -0
- data/lib/rex/post/meterpreter/extensions/priv/fs.rb +118 -0
- data/lib/rex/post/meterpreter/extensions/priv/passwd.rb +61 -0
- data/lib/rex/post/meterpreter/extensions/priv/priv.rb +111 -0
- data/lib/rex/post/meterpreter/extensions/priv/tlv.rb +28 -0
- data/lib/rex/post/meterpreter/extensions/sniffer/sniffer.rb +101 -0
- data/lib/rex/post/meterpreter/extensions/sniffer/tlv.rb +26 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/constants.rb +333 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/fs/dir.rb +282 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/fs/file.rb +266 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/fs/file_stat.rb +103 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/fs/io.rb +48 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/config.rb +144 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/interface.rb +73 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/route.rb +56 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/socket.rb +137 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_client_channel.rb +180 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/tcp_server_channel.rb +167 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/net/socket_subsystem/udp_channel.rb +208 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun.rb.ts.rb +6 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb +38106 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/api_constants.rb.ut.rb +31 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb +47 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/buffer_item.rb.ut.rb +36 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_advapi32.rb +1818 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_iphlpapi.rb +96 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_kernel32.rb +3848 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_netapi32.rb +26 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ntdll.rb +153 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_shell32.rb +21 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_user32.rb +3169 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ws2_32.rb +599 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll.rb +318 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb +100 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.rb.ut.rb +42 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb +148 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_helper.rb.ut.rb +127 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/multicall.rb +309 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb +204 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/tlv.rb +51 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/util.rb +630 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb +75 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb.ut.rb +103 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/stdapi.rb +149 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/config.rb +97 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log.rb +192 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log_subsystem/event_record.rb +41 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/power.rb +61 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/process.rb +370 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/image.rb +129 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/io.rb +55 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/memory.rb +336 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/thread.rb +141 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry.rb +279 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_key.rb +193 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_value.rb +102 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/sys/thread.rb +180 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/tlv.rb +211 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/ui.rb +227 -0
- data/lib/rex/post/meterpreter/extensions/stdapi/webcam/webcam.rb +63 -0
- data/lib/rex/post/meterpreter/inbound_packet_handler.rb +30 -0
- data/lib/rex/post/meterpreter/object_aliases.rb +83 -0
- data/lib/rex/post/meterpreter/packet.rb +688 -0
- data/lib/rex/post/meterpreter/packet_dispatcher.rb +431 -0
- data/lib/rex/post/meterpreter/packet_parser.rb +94 -0
- data/lib/rex/post/meterpreter/packet_response_waiter.rb +83 -0
- data/lib/rex/post/meterpreter/ui/console.rb +137 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher.rb +62 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/core.rb +730 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/espia.rb +108 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/incognito.rb +241 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/networkpug.rb +231 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv.rb +61 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/elevate.rb +98 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/passwd.rb +51 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/priv/timestomp.rb +132 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/sniffer.rb +187 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi.rb +65 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/fs.rb +442 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/net.rb +298 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/sys.rb +486 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/ui.rb +315 -0
- data/lib/rex/post/meterpreter/ui/console/command_dispatcher/stdapi/webcam.rb +157 -0
- data/lib/rex/post/meterpreter/ui/console/interactive_channel.rb +95 -0
- data/lib/rex/post/permission.rb +26 -0
- data/lib/rex/post/process.rb +57 -0
- data/lib/rex/post/thread.rb +57 -0
- data/lib/rex/post/ui.rb +52 -0
- data/lib/rex/proto.rb +13 -0
- data/lib/rex/proto.rb.ts.rb +8 -0
- data/lib/rex/proto/dcerpc.rb +6 -0
- data/lib/rex/proto/dcerpc.rb.ts.rb +9 -0
- data/lib/rex/proto/dcerpc/client.rb +361 -0
- data/lib/rex/proto/dcerpc/client.rb.ut.rb +491 -0
- data/lib/rex/proto/dcerpc/exceptions.rb +150 -0
- data/lib/rex/proto/dcerpc/handle.rb +47 -0
- data/lib/rex/proto/dcerpc/handle.rb.ut.rb +85 -0
- data/lib/rex/proto/dcerpc/ndr.rb +72 -0
- data/lib/rex/proto/dcerpc/ndr.rb.ut.rb +41 -0
- data/lib/rex/proto/dcerpc/packet.rb +253 -0
- data/lib/rex/proto/dcerpc/packet.rb.ut.rb +56 -0
- data/lib/rex/proto/dcerpc/response.rb +187 -0
- data/lib/rex/proto/dcerpc/response.rb.ut.rb +15 -0
- data/lib/rex/proto/dcerpc/uuid.rb +84 -0
- data/lib/rex/proto/dcerpc/uuid.rb.ut.rb +46 -0
- data/lib/rex/proto/dhcp.rb +7 -0
- data/lib/rex/proto/dhcp/constants.rb +33 -0
- data/lib/rex/proto/dhcp/server.rb +292 -0
- data/lib/rex/proto/drda.rb +5 -0
- data/lib/rex/proto/drda.rb.ts.rb +17 -0
- data/lib/rex/proto/drda/constants.rb +49 -0
- data/lib/rex/proto/drda/constants.rb.ut.rb +23 -0
- data/lib/rex/proto/drda/packet.rb +252 -0
- data/lib/rex/proto/drda/packet.rb.ut.rb +109 -0
- data/lib/rex/proto/drda/utils.rb +123 -0
- data/lib/rex/proto/drda/utils.rb.ut.rb +84 -0
- data/lib/rex/proto/http.rb +5 -0
- data/lib/rex/proto/http.rb.ts.rb +12 -0
- data/lib/rex/proto/http/client.rb +821 -0
- data/lib/rex/proto/http/client.rb.ut.rb +95 -0
- data/lib/rex/proto/http/handler.rb +46 -0
- data/lib/rex/proto/http/handler/erb.rb +128 -0
- data/lib/rex/proto/http/handler/erb.rb.ut.rb +21 -0
- data/lib/rex/proto/http/handler/erb.rb.ut.rb.rhtml +1 -0
- data/lib/rex/proto/http/handler/proc.rb +60 -0
- data/lib/rex/proto/http/handler/proc.rb.ut.rb +24 -0
- data/lib/rex/proto/http/header.rb +161 -0
- data/lib/rex/proto/http/header.rb.ut.rb +46 -0
- data/lib/rex/proto/http/packet.rb +407 -0
- data/lib/rex/proto/http/packet.rb.ut.rb +165 -0
- data/lib/rex/proto/http/request.rb +356 -0
- data/lib/rex/proto/http/request.rb.ut.rb +214 -0
- data/lib/rex/proto/http/response.rb +90 -0
- data/lib/rex/proto/http/response.rb.ut.rb +149 -0
- data/lib/rex/proto/http/server.rb +369 -0
- data/lib/rex/proto/http/server.rb.ut.rb +79 -0
- data/lib/rex/proto/ntlm.rb +7 -0
- data/lib/rex/proto/ntlm.rb.ut.rb +177 -0
- data/lib/rex/proto/ntlm/base.rb +326 -0
- data/lib/rex/proto/ntlm/constants.rb +74 -0
- data/lib/rex/proto/ntlm/crypt.rb +415 -0
- data/lib/rex/proto/ntlm/exceptions.rb +16 -0
- data/lib/rex/proto/ntlm/message.rb +536 -0
- data/lib/rex/proto/ntlm/utils.rb +764 -0
- data/lib/rex/proto/proxy/socks4a.rb +440 -0
- data/lib/rex/proto/rfb.rb +19 -0
- data/lib/rex/proto/rfb.rb.ut.rb +37 -0
- data/lib/rex/proto/rfb/cipher.rb +84 -0
- data/lib/rex/proto/rfb/client.rb +207 -0
- data/lib/rex/proto/rfb/constants.rb +52 -0
- data/lib/rex/proto/smb.rb +7 -0
- data/lib/rex/proto/smb.rb.ts.rb +8 -0
- data/lib/rex/proto/smb/client.rb +1952 -0
- data/lib/rex/proto/smb/client.rb.ut.rb +223 -0
- data/lib/rex/proto/smb/constants.rb +1047 -0
- data/lib/rex/proto/smb/constants.rb.ut.rb +18 -0
- data/lib/rex/proto/smb/crypt.rb +36 -0
- data/lib/rex/proto/smb/evasions.rb +66 -0
- data/lib/rex/proto/smb/exceptions.rb +858 -0
- data/lib/rex/proto/smb/simpleclient.rb +306 -0
- data/lib/rex/proto/smb/simpleclient.rb.ut.rb +128 -0
- data/lib/rex/proto/smb/utils.rb +103 -0
- data/lib/rex/proto/smb/utils.rb.ut.rb +20 -0
- data/lib/rex/proto/sunrpc.rb +1 -0
- data/lib/rex/proto/sunrpc/client.rb +195 -0
- data/lib/rex/proto/tftp.rb +12 -0
- data/lib/rex/proto/tftp/constants.rb +39 -0
- data/lib/rex/proto/tftp/server.rb +497 -0
- data/lib/rex/proto/tftp/server.rb.ut.rb +28 -0
- data/lib/rex/script.rb +42 -0
- data/lib/rex/script/base.rb +59 -0
- data/lib/rex/script/meterpreter.rb +15 -0
- data/lib/rex/script/shell.rb +9 -0
- data/lib/rex/service.rb +48 -0
- data/lib/rex/service_manager.rb +141 -0
- data/lib/rex/service_manager.rb.ut.rb +32 -0
- data/lib/rex/services/local_relay.rb +423 -0
- data/lib/rex/socket.rb +684 -0
- data/lib/rex/socket.rb.ut.rb +107 -0
- data/lib/rex/socket/comm.rb +119 -0
- data/lib/rex/socket/comm/local.rb +412 -0
- data/lib/rex/socket/comm/local.rb.ut.rb +75 -0
- data/lib/rex/socket/ip.rb +130 -0
- data/lib/rex/socket/parameters.rb +345 -0
- data/lib/rex/socket/parameters.rb.ut.rb +51 -0
- data/lib/rex/socket/range_walker.rb +346 -0
- data/lib/rex/socket/range_walker.rb.ut.rb +55 -0
- data/lib/rex/socket/ssl_tcp.rb +184 -0
- data/lib/rex/socket/ssl_tcp.rb.ut.rb +39 -0
- data/lib/rex/socket/ssl_tcp_server.rb +122 -0
- data/lib/rex/socket/ssl_tcp_server.rb.ut.rb +61 -0
- data/lib/rex/socket/subnet_walker.rb +75 -0
- data/lib/rex/socket/subnet_walker.rb.ut.rb +28 -0
- data/lib/rex/socket/switch_board.rb +278 -0
- data/lib/rex/socket/switch_board.rb.ut.rb +52 -0
- data/lib/rex/socket/tcp.rb +76 -0
- data/lib/rex/socket/tcp.rb.ut.rb +64 -0
- data/lib/rex/socket/tcp_server.rb +66 -0
- data/lib/rex/socket/tcp_server.rb.ut.rb +44 -0
- data/lib/rex/socket/udp.rb +164 -0
- data/lib/rex/socket/udp.rb.ut.rb +44 -0
- data/lib/rex/struct2.rb +5 -0
- data/lib/rex/struct2/c_struct.rb +181 -0
- data/lib/rex/struct2/c_struct_template.rb +39 -0
- data/lib/rex/struct2/constant.rb +26 -0
- data/lib/rex/struct2/element.rb +44 -0
- data/lib/rex/struct2/generic.rb +73 -0
- data/lib/rex/struct2/restraint.rb +54 -0
- data/lib/rex/struct2/s_string.rb +72 -0
- data/lib/rex/struct2/s_struct.rb +111 -0
- data/lib/rex/sync.rb +6 -0
- data/lib/rex/sync/event.rb +94 -0
- data/lib/rex/sync/read_write_lock.rb +176 -0
- data/lib/rex/sync/ref.rb +57 -0
- data/lib/rex/sync/thread_safe.rb +82 -0
- data/lib/rex/test.rb +35 -0
- data/lib/rex/text.rb +1163 -0
- data/lib/rex/text.rb.ut.rb +190 -0
- data/lib/rex/thread_factory.rb +42 -0
- data/lib/rex/time.rb +65 -0
- data/lib/rex/transformer.rb +115 -0
- data/lib/rex/transformer.rb.ut.rb +38 -0
- data/lib/rex/ui.rb +21 -0
- data/lib/rex/ui/interactive.rb +298 -0
- data/lib/rex/ui/output.rb +78 -0
- data/lib/rex/ui/output/none.rb +18 -0
- data/lib/rex/ui/progress_tracker.rb +96 -0
- data/lib/rex/ui/subscriber.rb +149 -0
- data/lib/rex/ui/text/color.rb +97 -0
- data/lib/rex/ui/text/color.rb.ut.rb +18 -0
- data/lib/rex/ui/text/dispatcher_shell.rb +467 -0
- data/lib/rex/ui/text/input.rb +117 -0
- data/lib/rex/ui/text/input/buffer.rb +75 -0
- data/lib/rex/ui/text/input/readline.rb +129 -0
- data/lib/rex/ui/text/input/socket.rb +95 -0
- data/lib/rex/ui/text/input/stdio.rb +45 -0
- data/lib/rex/ui/text/irb_shell.rb +57 -0
- data/lib/rex/ui/text/output.rb +80 -0
- data/lib/rex/ui/text/output/buffer.rb +61 -0
- data/lib/rex/ui/text/output/file.rb +43 -0
- data/lib/rex/ui/text/output/socket.rb +43 -0
- data/lib/rex/ui/text/output/stdio.rb +40 -0
- data/lib/rex/ui/text/progress_tracker.rb +56 -0
- data/lib/rex/ui/text/progress_tracker.rb.ut.rb +34 -0
- data/lib/rex/ui/text/shell.rb +328 -0
- data/lib/rex/ui/text/table.rb +279 -0
- data/lib/rex/ui/text/table.rb.ut.rb +55 -0
- data/lib/rex/zip.rb +93 -0
- data/lib/rex/zip/archive.rb +184 -0
- data/lib/rex/zip/blocks.rb +182 -0
- data/lib/rex/zip/entry.rb +104 -0
- data/lib/rex/zip/samples/comment.rb +32 -0
- data/lib/rex/zip/samples/mkwar.rb +138 -0
- data/lib/rex/zip/samples/mkzip.rb +19 -0
- data/lib/rex/zip/samples/recursive.rb +58 -0
- metadata +434 -1
@@ -0,0 +1,53 @@
|
|
1
|
+
##
|
2
|
+
# $Id: stream.rb 8457 2010-02-11 18:36:38Z jduck $
|
3
|
+
# Version: $Revision: 8457 $
|
4
|
+
##
|
5
|
+
|
6
|
+
##
|
7
|
+
# Rex::OLE - an OLE implementation
|
8
|
+
# written in 2010 by Joshua J. Drake <jduck [at] metasploit.com>
|
9
|
+
##
|
10
|
+
|
11
|
+
module Rex
|
12
|
+
module OLE
|
13
|
+
|
14
|
+
class Stream < DirEntry
|
15
|
+
|
16
|
+
def initialize(stg)
|
17
|
+
super
|
18
|
+
|
19
|
+
# for reading/writing from this
|
20
|
+
@offset = 0
|
21
|
+
@_mse = STGTY_STREAM
|
22
|
+
end
|
23
|
+
|
24
|
+
def close
|
25
|
+
@mode = nil
|
26
|
+
@offset = nil
|
27
|
+
end
|
28
|
+
|
29
|
+
def seek(offset)
|
30
|
+
@offset = offset
|
31
|
+
end
|
32
|
+
|
33
|
+
def read(len)
|
34
|
+
return nil if (not @data)
|
35
|
+
|
36
|
+
ret = @data[@offset, len]
|
37
|
+
@offset += len
|
38
|
+
ret
|
39
|
+
end
|
40
|
+
|
41
|
+
def <<(expr)
|
42
|
+
if (not @data)
|
43
|
+
@data = expr.dup
|
44
|
+
else
|
45
|
+
@data << expr
|
46
|
+
end
|
47
|
+
@_ulSize = @data.length
|
48
|
+
end
|
49
|
+
|
50
|
+
end
|
51
|
+
|
52
|
+
end
|
53
|
+
end
|
@@ -0,0 +1,49 @@
|
|
1
|
+
##
|
2
|
+
# $Id: substorage.rb 8457 2010-02-11 18:36:38Z jduck $
|
3
|
+
# Version: $Revision: 8457 $
|
4
|
+
##
|
5
|
+
|
6
|
+
##
|
7
|
+
# Rex::OLE - an OLE implementation
|
8
|
+
# written in 2010 by Joshua J. Drake <jduck [at] metasploit.com>
|
9
|
+
##
|
10
|
+
|
11
|
+
module Rex
|
12
|
+
module OLE
|
13
|
+
|
14
|
+
class SubStorage < DirEntry
|
15
|
+
|
16
|
+
def initialize(stg)
|
17
|
+
super
|
18
|
+
|
19
|
+
@_mse = STGTY_STORAGE
|
20
|
+
end
|
21
|
+
|
22
|
+
|
23
|
+
def close
|
24
|
+
end
|
25
|
+
|
26
|
+
|
27
|
+
# stream handling stuff
|
28
|
+
def create_stream(name, mode=STGM_WRITE)
|
29
|
+
@stg.create_stream(name, mode, self)
|
30
|
+
end
|
31
|
+
|
32
|
+
def open_stream(name, mode=STGM_READ)
|
33
|
+
@stg.open_stream(name, mode, self)
|
34
|
+
end
|
35
|
+
|
36
|
+
|
37
|
+
# storage handling stuff
|
38
|
+
def create_storage(name, mode=STGM_WRITE)
|
39
|
+
@stg.create_storage(name, mode, self)
|
40
|
+
end
|
41
|
+
|
42
|
+
def open_storage(name, mode=STGM_WRITE)
|
43
|
+
@stg.open_storage(name, mode, self)
|
44
|
+
end
|
45
|
+
|
46
|
+
end
|
47
|
+
|
48
|
+
end
|
49
|
+
end
|
data/lib/rex/ole/util.rb
ADDED
@@ -0,0 +1,157 @@
|
|
1
|
+
##
|
2
|
+
# $Id: util.rb 8457 2010-02-11 18:36:38Z jduck $
|
3
|
+
# Version: $Revision: 8457 $
|
4
|
+
##
|
5
|
+
|
6
|
+
##
|
7
|
+
# Rex::OLE - an OLE implementation
|
8
|
+
# written in 2010 by Joshua J. Drake <jduck [at] metasploit.com>
|
9
|
+
##
|
10
|
+
|
11
|
+
module Rex
|
12
|
+
module OLE
|
13
|
+
|
14
|
+
class Util
|
15
|
+
|
16
|
+
def self.Hexify32array(arr)
|
17
|
+
ret = ""
|
18
|
+
arr.each { |dw|
|
19
|
+
ret << " " if ret.length > 0
|
20
|
+
ret << "0x%08x" % dw
|
21
|
+
}
|
22
|
+
ret
|
23
|
+
end
|
24
|
+
|
25
|
+
def self.Printable(buf)
|
26
|
+
ret = ""
|
27
|
+
buf.unpack('C*').each { |byte|
|
28
|
+
ch = byte.chr
|
29
|
+
if (byte < 0x20 || byte > 0x7e)
|
30
|
+
ret << "\\x" + ch.unpack('H*')[0]
|
31
|
+
else
|
32
|
+
ret << ch
|
33
|
+
end
|
34
|
+
}
|
35
|
+
ret
|
36
|
+
end
|
37
|
+
|
38
|
+
|
39
|
+
def self.set_endian(endian)
|
40
|
+
@endian = endian
|
41
|
+
end
|
42
|
+
|
43
|
+
def self.get64(buf, offset)
|
44
|
+
@endian = LITTLE_ENDIAN if not @endian
|
45
|
+
if (@endian == LITTLE_ENDIAN)
|
46
|
+
arr = buf[offset,8].unpack('VV')
|
47
|
+
return (arr[0] + (arr[1] << 32))
|
48
|
+
else
|
49
|
+
arr = buf[offset,8].unpack('NN')
|
50
|
+
return ((arr[0] << 32) + arr[1])
|
51
|
+
end
|
52
|
+
end
|
53
|
+
|
54
|
+
def self.pack64(value)
|
55
|
+
@endian = LITTLE_ENDIAN if not @endian
|
56
|
+
arr = []
|
57
|
+
arr << (value & 0xffffffff)
|
58
|
+
arr << (value >> 32)
|
59
|
+
if (@endian == LITTLE_ENDIAN)
|
60
|
+
arr.pack('VV')
|
61
|
+
else
|
62
|
+
arr.reverse.pack('NN')
|
63
|
+
end
|
64
|
+
end
|
65
|
+
|
66
|
+
def self.get32(buf, offset)
|
67
|
+
@endian = LITTLE_ENDIAN if not @endian
|
68
|
+
if (@endian == LITTLE_ENDIAN)
|
69
|
+
buf[offset,4].unpack('V')[0]
|
70
|
+
else
|
71
|
+
buf[offset,4].unpack('N')[0]
|
72
|
+
end
|
73
|
+
end
|
74
|
+
|
75
|
+
def self.pack32(value)
|
76
|
+
@endian = LITTLE_ENDIAN if not @endian
|
77
|
+
if (@endian == LITTLE_ENDIAN)
|
78
|
+
[value].pack('V')
|
79
|
+
else
|
80
|
+
[value].pack('N')
|
81
|
+
end
|
82
|
+
end
|
83
|
+
|
84
|
+
def self.get32array(buf)
|
85
|
+
@endian = LITTLE_ENDIAN if not @endian
|
86
|
+
if (@endian == LITTLE_ENDIAN)
|
87
|
+
buf.unpack('V*')
|
88
|
+
else
|
89
|
+
buf.unpack('N*')
|
90
|
+
end
|
91
|
+
end
|
92
|
+
|
93
|
+
def self.pack32array(arr)
|
94
|
+
@endian = LITTLE_ENDIAN if not @endian
|
95
|
+
if (@endian == LITTLE_ENDIAN)
|
96
|
+
arr.pack('V*')
|
97
|
+
else
|
98
|
+
arr.pack('N*')
|
99
|
+
end
|
100
|
+
end
|
101
|
+
|
102
|
+
def self.get16(buf, offset)
|
103
|
+
@endian = LITTLE_ENDIAN if not @endian
|
104
|
+
if (@endian == LITTLE_ENDIAN)
|
105
|
+
buf[offset,2].unpack('v')[0]
|
106
|
+
else
|
107
|
+
buf[offset,2].unpack('n')[0]
|
108
|
+
end
|
109
|
+
end
|
110
|
+
|
111
|
+
def self.pack16(value)
|
112
|
+
@endian = LITTLE_ENDIAN if not @endian
|
113
|
+
if (@endian == LITTLE_ENDIAN)
|
114
|
+
[value].pack('v')
|
115
|
+
else
|
116
|
+
[value].pack('n')
|
117
|
+
end
|
118
|
+
end
|
119
|
+
|
120
|
+
def self.get8(buf, offset)
|
121
|
+
buf[offset,1].unpack('C')[0]
|
122
|
+
end
|
123
|
+
|
124
|
+
def self.pack8(value)
|
125
|
+
[value].pack('C')
|
126
|
+
end
|
127
|
+
|
128
|
+
|
129
|
+
def self.getUnicodeString(buf)
|
130
|
+
buf = buf.unpack('S*').pack('C*')
|
131
|
+
if (idx = buf.index(0x00.chr))
|
132
|
+
buf.slice!(idx, buf.length)
|
133
|
+
end
|
134
|
+
buf
|
135
|
+
end
|
136
|
+
|
137
|
+
def self.putUnicodeString(buf)
|
138
|
+
buf = buf.unpack('C*').pack('S*')
|
139
|
+
if (buf.length < 0x40)
|
140
|
+
buf << "\x00" * (0x40 - buf.length)
|
141
|
+
end
|
142
|
+
buf
|
143
|
+
end
|
144
|
+
|
145
|
+
|
146
|
+
def self.name_is_valid(name)
|
147
|
+
return nil if (name.length > 31)
|
148
|
+
(0..0x1f).to_a.each { |x|
|
149
|
+
return nil if (name.include?(x.chr))
|
150
|
+
}
|
151
|
+
return true
|
152
|
+
end
|
153
|
+
|
154
|
+
end
|
155
|
+
|
156
|
+
end
|
157
|
+
end
|
@@ -0,0 +1,97 @@
|
|
1
|
+
require 'shellwords'
|
2
|
+
|
3
|
+
module Rex
|
4
|
+
module Parser
|
5
|
+
|
6
|
+
###
|
7
|
+
#
|
8
|
+
# This class parses arguments in a getopt style format, kind of.
|
9
|
+
# Unfortunately, the default ruby getopt implementation will only
|
10
|
+
# work on ARGV, so we can't use it.
|
11
|
+
#
|
12
|
+
###
|
13
|
+
class Arguments
|
14
|
+
|
15
|
+
#
|
16
|
+
# Specifies that an option is expected to have an argument
|
17
|
+
#
|
18
|
+
HasArgument = (1 << 0)
|
19
|
+
|
20
|
+
#
|
21
|
+
# Initializes the format list with an array of formats like:
|
22
|
+
#
|
23
|
+
# Arguments.new(
|
24
|
+
# '-b' => [ false, "some text" ]
|
25
|
+
# )
|
26
|
+
#
|
27
|
+
def initialize(fmt)
|
28
|
+
self.fmt = fmt
|
29
|
+
end
|
30
|
+
|
31
|
+
#
|
32
|
+
# Takes a string and converts it into an array of arguments.
|
33
|
+
#
|
34
|
+
def self.from_s(str)
|
35
|
+
Shellwords.shellwords(str)
|
36
|
+
end
|
37
|
+
|
38
|
+
#
|
39
|
+
# Parses the supplied arguments into a set of options.
|
40
|
+
#
|
41
|
+
def parse(args, &block)
|
42
|
+
skip_next = false
|
43
|
+
|
44
|
+
args.each_with_index { |arg, idx|
|
45
|
+
if (skip_next == true)
|
46
|
+
skip_next = false
|
47
|
+
next
|
48
|
+
end
|
49
|
+
|
50
|
+
if (arg.match(/^-/))
|
51
|
+
cfs = arg[0..2]
|
52
|
+
|
53
|
+
fmt.each_pair { |fmtspec, val|
|
54
|
+
next if (fmtspec != cfs)
|
55
|
+
|
56
|
+
param = nil
|
57
|
+
|
58
|
+
if (val[0])
|
59
|
+
param = args[idx+1]
|
60
|
+
skip_next = true
|
61
|
+
end
|
62
|
+
|
63
|
+
yield fmtspec, idx, param
|
64
|
+
}
|
65
|
+
else
|
66
|
+
yield nil, idx, arg
|
67
|
+
end
|
68
|
+
}
|
69
|
+
end
|
70
|
+
|
71
|
+
#
|
72
|
+
# Returns usage information for this parsing context.
|
73
|
+
#
|
74
|
+
def usage
|
75
|
+
txt = "\nOPTIONS:\n\n"
|
76
|
+
|
77
|
+
fmt.sort.each { |entry|
|
78
|
+
fmtspec, val = entry
|
79
|
+
|
80
|
+
txt << " #{fmtspec}" + ((val[0] == true) ? " <opt> " : " ")
|
81
|
+
txt << val[1] + "\n"
|
82
|
+
}
|
83
|
+
|
84
|
+
txt << "\n"
|
85
|
+
|
86
|
+
return txt
|
87
|
+
end
|
88
|
+
def include?(search)
|
89
|
+
return fmt.include?(search)
|
90
|
+
end
|
91
|
+
|
92
|
+
attr_accessor :fmt # :nodoc:
|
93
|
+
|
94
|
+
end
|
95
|
+
|
96
|
+
end
|
97
|
+
end
|
@@ -0,0 +1,67 @@
|
|
1
|
+
#!/usr/bin/env ruby
|
2
|
+
|
3
|
+
$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
|
4
|
+
|
5
|
+
require 'test/unit'
|
6
|
+
require 'rex/parser/arguments'
|
7
|
+
|
8
|
+
class Rex::Parser::Arguments::UnitTest < Test::Unit::TestCase
|
9
|
+
|
10
|
+
def test_parse
|
11
|
+
args =
|
12
|
+
[
|
13
|
+
"-b",
|
14
|
+
"foo",
|
15
|
+
"-c",
|
16
|
+
"-f",
|
17
|
+
"-g",
|
18
|
+
"arg",
|
19
|
+
"none"
|
20
|
+
]
|
21
|
+
|
22
|
+
b = nil
|
23
|
+
c = false
|
24
|
+
f = false
|
25
|
+
g = nil
|
26
|
+
none = nil
|
27
|
+
|
28
|
+
Rex::Parser::Arguments.new(
|
29
|
+
'-b' => [ true, "bee" ],
|
30
|
+
'-c' => [ false, "cee" ],
|
31
|
+
'-f' => [ false, "eff" ],
|
32
|
+
'-g' => [ true, "gee" ]).parse(args) { |opt, idx, val|
|
33
|
+
case opt
|
34
|
+
when nil
|
35
|
+
none = val
|
36
|
+
when '-b'
|
37
|
+
b = val
|
38
|
+
when '-c'
|
39
|
+
c = true
|
40
|
+
when '-f'
|
41
|
+
f = true
|
42
|
+
when '-g'
|
43
|
+
g = val
|
44
|
+
end
|
45
|
+
}
|
46
|
+
|
47
|
+
assert_equal(b, "foo")
|
48
|
+
assert_equal(c, true)
|
49
|
+
assert_equal(f, true)
|
50
|
+
assert_equal(g, "arg")
|
51
|
+
assert_equal(none, "none")
|
52
|
+
end
|
53
|
+
|
54
|
+
def test_from_s
|
55
|
+
args = Rex::Parser::Arguments.from_s(
|
56
|
+
"this is a test \"of the emergency pimping\" system \\\"buh lee dat\\\" yup")
|
57
|
+
|
58
|
+
assert_equal(args[0], "this")
|
59
|
+
assert_equal(args[3], "test")
|
60
|
+
assert_equal(args[4], "of the emergency pimping")
|
61
|
+
assert_equal(args[5], "system")
|
62
|
+
assert_equal(args[6], "\"buh")
|
63
|
+
assert_equal(args[8], "dat\"")
|
64
|
+
assert_equal(args[9], "yup")
|
65
|
+
end
|
66
|
+
|
67
|
+
end
|
@@ -0,0 +1,185 @@
|
|
1
|
+
module Rex
|
2
|
+
module Parser
|
3
|
+
|
4
|
+
###
|
5
|
+
#
|
6
|
+
# This class parses the contents of an INI file.
|
7
|
+
#
|
8
|
+
###
|
9
|
+
class Ini < Hash
|
10
|
+
|
11
|
+
##
|
12
|
+
#
|
13
|
+
# Factories
|
14
|
+
#
|
15
|
+
##
|
16
|
+
|
17
|
+
#
|
18
|
+
# Creates a new class instance and reads in the contents of the supplied
|
19
|
+
# file path.
|
20
|
+
#
|
21
|
+
def self.from_file(path)
|
22
|
+
ini = Ini.new(path)
|
23
|
+
ini.from_file
|
24
|
+
return ini
|
25
|
+
end
|
26
|
+
|
27
|
+
#
|
28
|
+
# Creates a new class instance from the supplied string.
|
29
|
+
#
|
30
|
+
def self.from_s(str)
|
31
|
+
ini = Ini.new
|
32
|
+
ini.from_s(str)
|
33
|
+
return ini
|
34
|
+
end
|
35
|
+
|
36
|
+
#
|
37
|
+
# Initializes an ini instance and tries to read in the groups from the
|
38
|
+
# file if it exists.
|
39
|
+
#
|
40
|
+
def initialize(path = nil)
|
41
|
+
self.path = path
|
42
|
+
|
43
|
+
# Try to synchronize ourself with the file if we
|
44
|
+
# have one
|
45
|
+
begin
|
46
|
+
self.from_file if (self.path)
|
47
|
+
rescue
|
48
|
+
end
|
49
|
+
end
|
50
|
+
|
51
|
+
#
|
52
|
+
# Enumerates the groups hash keys.
|
53
|
+
#
|
54
|
+
def each_group(&block)
|
55
|
+
self.keys.each { |k|
|
56
|
+
yield
|
57
|
+
}
|
58
|
+
end
|
59
|
+
|
60
|
+
#
|
61
|
+
# Adds a group of the supplied name if it doesn't already exist.
|
62
|
+
#
|
63
|
+
def add_group(name = 'global', reset = true)
|
64
|
+
self[name] = {} if (reset == true)
|
65
|
+
self[name] = {} if (!self[name])
|
66
|
+
|
67
|
+
return self[name]
|
68
|
+
end
|
69
|
+
|
70
|
+
#
|
71
|
+
# Checks to see if name is a valid group.
|
72
|
+
#
|
73
|
+
def group?(name)
|
74
|
+
return (self[name] != nil)
|
75
|
+
end
|
76
|
+
|
77
|
+
##
|
78
|
+
#
|
79
|
+
# Serializers
|
80
|
+
#
|
81
|
+
##
|
82
|
+
|
83
|
+
#
|
84
|
+
# Reads in the groups from the supplied file path or the instance's file
|
85
|
+
# path.
|
86
|
+
#
|
87
|
+
def from_file(fpath = nil)
|
88
|
+
fpath = path if (!fpath)
|
89
|
+
|
90
|
+
read_groups(fpath)
|
91
|
+
end
|
92
|
+
|
93
|
+
#
|
94
|
+
# Reads in the groups from the supplied string.
|
95
|
+
#
|
96
|
+
def from_s(str)
|
97
|
+
read_groups_string(str.split("\n"))
|
98
|
+
end
|
99
|
+
|
100
|
+
#
|
101
|
+
# Writes the group settings to a file.
|
102
|
+
#
|
103
|
+
def to_file(tpath = nil)
|
104
|
+
tpath = path if (!tpath)
|
105
|
+
|
106
|
+
f = File.new(tpath, "w")
|
107
|
+
f.write(to_s)
|
108
|
+
f.close
|
109
|
+
end
|
110
|
+
|
111
|
+
#
|
112
|
+
# Converts the groups to a string.
|
113
|
+
#
|
114
|
+
def to_s
|
115
|
+
str = ''
|
116
|
+
keys.sort.each { |k|
|
117
|
+
str << "[#{k}]\n"
|
118
|
+
|
119
|
+
self[k].each_pair { |var, val|
|
120
|
+
str << "#{var}=#{val}\n"
|
121
|
+
}
|
122
|
+
|
123
|
+
str << "\n";
|
124
|
+
}
|
125
|
+
|
126
|
+
return str
|
127
|
+
end
|
128
|
+
|
129
|
+
attr_reader :path
|
130
|
+
|
131
|
+
protected
|
132
|
+
|
133
|
+
#
|
134
|
+
# Reads in the groups and their attributes from the supplied file
|
135
|
+
# path or from the instance's file path if one was set.
|
136
|
+
#
|
137
|
+
def read_groups(fpath) # :nodoc:
|
138
|
+
if (!fpath)
|
139
|
+
raise ArgumentError, "No file path specified.",
|
140
|
+
caller
|
141
|
+
end
|
142
|
+
|
143
|
+
# Read in the contents of the file
|
144
|
+
lines = ::IO.readlines(fpath)
|
145
|
+
|
146
|
+
# Now read the contents from the supplied string
|
147
|
+
read_groups_string(lines)
|
148
|
+
end
|
149
|
+
|
150
|
+
#
|
151
|
+
# Reads groups from the supplied string
|
152
|
+
#
|
153
|
+
def read_groups_string(str) # :nodoc:
|
154
|
+
# Reset the groups hash
|
155
|
+
self.clear
|
156
|
+
|
157
|
+
# The active group
|
158
|
+
active_group = nil
|
159
|
+
|
160
|
+
# Walk each line initializing the groups
|
161
|
+
str.each { |line|
|
162
|
+
next if (line.match(/^;/))
|
163
|
+
|
164
|
+
# Eliminate cr/lf
|
165
|
+
line.gsub!(/(\n|\r)/, '')
|
166
|
+
|
167
|
+
# Is it a group [bob]?
|
168
|
+
if (md = line.match(/^\[(.+?)\]/))
|
169
|
+
active_group = md[1]
|
170
|
+
self[md[1]] = {}
|
171
|
+
# Is it a VAR=VAL?
|
172
|
+
elsif (md = line.match(/^(.+?)=(.*)$/))
|
173
|
+
if (active_group)
|
174
|
+
self[active_group][md[1]] = md[2]
|
175
|
+
end
|
176
|
+
end
|
177
|
+
}
|
178
|
+
end
|
179
|
+
|
180
|
+
attr_writer :path # :nodoc:
|
181
|
+
|
182
|
+
end
|
183
|
+
|
184
|
+
end
|
185
|
+
end
|