Module: Rex::Compat
- Defined in:
- lib/rex/compat.rb
Overview
This class provides os-specific functionality
Constant Summary collapse
- STD_INPUT_HANDLE =
-10
- STD_OUTPUT_HANDLE =
-11
- STD_ERROR_HANDLE =
-12
- GENERIC_READ =
0x80000000
- GENERIC_WRITE =
0x40000000
- GENERIC_EXECUTE =
0x20000000
- FILE_SHARE_READ =
0x00000001
- FILE_SHARE_WRITE =
0x00000002
- OPEN_EXISTING =
0x00000003
- ENABLE_LINE_INPUT =
2
- ENABLE_ECHO_INPUT =
4
- ENABLE_PROCESSED_INPUT =
1
- @@is_windows =
Platform detection
@@is_cygwin = @@is_macosx = @@is_linux = @@is_bsdi = @@is_freebsd = @@is_netbsd = @@is_openbsd = @@is_java = false
- @@loaded_win32api =
false
- @@loaded_tempfile =
false
- @@loaded_fileutils =
false
Class Method Summary collapse
- .cygwin_to_win32(path) ⇒ Object
- .getenv(var) ⇒ Object
- .is_bsdi ⇒ Object
- .is_cygwin ⇒ Object
- .is_freebsd ⇒ Object
- .is_java ⇒ Object
- .is_linux ⇒ Object
- .is_macosx ⇒ Object
- .is_netbsd ⇒ Object
- .is_openbsd ⇒ Object
- .is_windows ⇒ Object
- .is_wow64 ⇒ Object
- .open_browser(url = 'http://metasploit.com/') ⇒ Object
- .open_email(addr) ⇒ Object
- .open_file(url = '') ⇒ Object
-
.pipe ⇒ Object
Platform independent socket pair.
- .play_sound(path) ⇒ Object
- .setenv(var, val) ⇒ Object
-
.temp_copy(path) ⇒ Object
Copy a file to a temporary path.
-
.temp_delete(tp) ⇒ Object
Delete an opened temporary file.
-
.win32_ruby_path ⇒ Object
Obtain the path to our interpreter.
-
.win32_winexec(cmd) ⇒ Object
Call WinExec (equiv to system(“cmd &”)).
Class Method Details
.cygwin_to_win32(path) ⇒ Object
95 96 97 98 99 100 101 102 103 104 |
# File 'lib/rex/compat.rb', line 95 def self.cygwin_to_win32(path) if(path !~ /^\/cygdrive/) return ::IO.popen("cygpath -w #{path}", "rb").read.strip end dir = path.split("/") dir.shift dir.shift dir[0] = dir[0] + ":" dir.join("\\") end |
.getenv(var) ⇒ Object
179 180 181 182 183 184 185 186 187 188 189 |
# File 'lib/rex/compat.rb', line 179 def self.getenv(var) if (is_windows and @@loaded_win32api) f = Win32API.new("kernel32", "GetEnvironmentVariable", ["P", "P", "I"], "I") buff = "\x00" * 16384 sz = f.call(var, buff, buff.length) return nil if sz == 0 buff[0,sz] else ENV[var] end end |
.is_bsdi ⇒ Object
58 59 60 61 |
# File 'lib/rex/compat.rb', line 58 def self.is_bsdi return @@is_bsdi if @@is_bsdi @@is_bsdi = (RUBY_PLATFORM =~ /bsdi/i) ? true : false end |
.is_cygwin ⇒ Object
43 44 45 46 |
# File 'lib/rex/compat.rb', line 43 def self.is_cygwin return @@is_cygwin if @@is_cygwin @@is_cygwin = (RUBY_PLATFORM =~ /cygwin/) ? true : false end |
.is_freebsd ⇒ Object
68 69 70 71 |
# File 'lib/rex/compat.rb', line 68 def self.is_freebsd return @@is_freebsd if @@is_freebsd @@is_freebsd = (RUBY_PLATFORM =~ /freebsd/) ? true : false end |
.is_java ⇒ Object
78 79 80 81 |
# File 'lib/rex/compat.rb', line 78 def self.is_java return @@is_java if @@is_java @@is_java = (RUBY_PLATFORM =~ /java/) ? true : false end |
.is_linux ⇒ Object
53 54 55 56 |
# File 'lib/rex/compat.rb', line 53 def self.is_linux return @@is_linux if @@is_linux @@is_linux = (RUBY_PLATFORM =~ /linux/) ? true : false end |
.is_macosx ⇒ Object
48 49 50 51 |
# File 'lib/rex/compat.rb', line 48 def self.is_macosx return @@is_macosx if @@is_macosx @@is_macosx = (RUBY_PLATFORM =~ /darwin/) ? true : false end |
.is_netbsd ⇒ Object
63 64 65 66 |
# File 'lib/rex/compat.rb', line 63 def self.is_netbsd return @@is_netbsd if @@is_netbsd @@is_netbsd = (RUBY_PLATFORM =~ /netbsd/) ? true : false end |
.is_openbsd ⇒ Object
73 74 75 76 |
# File 'lib/rex/compat.rb', line 73 def self.is_openbsd return @@is_openbsd if @@is_openbsd @@is_openbsd = (RUBY_PLATFORM =~ /openbsd/) ? true : false end |
.is_windows ⇒ Object
38 39 40 41 |
# File 'lib/rex/compat.rb', line 38 def self.is_windows return @@is_windows if @@is_windows @@is_windows = (RUBY_PLATFORM =~ /mswin32|mingw32/) ? true : false end |
.is_wow64 ⇒ Object
83 84 85 86 87 88 89 90 91 92 93 |
# File 'lib/rex/compat.rb', line 83 def self.is_wow64 return false if not is_windows is64 = false begin buff = "\x00" * 4 Win32API.new("kernel32","IsWow64Process",['L','P'],'L').call(-1, buff) is64 = (buff.unpack("V")[0]) == 1 ? true : false rescue ::Exception end is64 end |
.open_browser(url = 'http://metasploit.com/') ⇒ Object
116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 |
# File 'lib/rex/compat.rb', line 116 def self.open_browser(url='http://metasploit.com/') case RUBY_PLATFORM when /cygwin/ if(url[0,1] == "/") self.open_file(url) end return if not @@loaded_win32api Win32API.new("shell32.dll", "ShellExecute", ["PPPPPL"], "L").call(nil, "open", url, nil, nil, 0) when /mswin32|mingw/ return if not @@loaded_win32api Win32API.new("shell32.dll", "ShellExecute", ["PPPPPL"], "L").call(nil, "open", url, nil, nil, 0) when /darwin/ system("open #{url}") else # Search through the PATH variable (if it exists) and chose a browser # We are making an assumption about the nature of "PATH" so tread lightly if defined? ENV['PATH'] # "sensible-browser" opens the "default" browser in Ubuntu and others, so try that first # but also provide fallbacks ['sensible-browser', 'firefox', 'opera', 'chromium-browser', 'konqueror'].each do |browser| ENV['PATH'].split(':').each do |path| # Does the browser exists? if File.exists?("#{path}/#{browser}") system("#{browser} #{url} &") return end end end end # If nothing else worked, default to firefox system("firefox #{url} &") end end |
.open_email(addr) ⇒ Object
151 152 153 154 155 156 157 158 159 160 161 |
# File 'lib/rex/compat.rb', line 151 def self.open_email(addr) case RUBY_PLATFORM when /mswin32|cygwin/ return if not @@loaded_win32api Win32API.new("shell32.dll", "ShellExecute", ["PPPPPL"], "L").call(nil, "open", "mailto:"+addr, nil, nil, 0) when /darwin/ system("open mailto:#{addr}") else # ? end end |
.open_file(url = '') ⇒ Object
106 107 108 109 110 111 112 113 114 |
# File 'lib/rex/compat.rb', line 106 def self.open_file(url='') case RUBY_PLATFORM when /cygwin/ path = self.cygwin_to_win32(url) system(["cmd", "cmd"], "/c", "explorer", path) else self.open_browser(url) end end |
.pipe ⇒ Object
Platform independent socket pair
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 |
# File 'lib/rex/compat.rb', line 226 def self.pipe if (! is_windows()) # Standard pipes should be fine return ::IO.pipe end # Create a socket connection for Windows serv = nil port = 1024 while (! serv and port < 65535) begin serv = TCPServer.new('127.0.0.1', (port += 1)) rescue ::Exception end end pipe1 = TCPSocket.new('127.0.0.1', port) # Accept the forked child pipe2 = serv.accept # Shutdown the server serv.close return [pipe1, pipe2] end |
.play_sound(path) ⇒ Object
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 |
# File 'lib/rex/compat.rb', line 163 def self.play_sound(path) case RUBY_PLATFORM when /cygwin/ path = self.cygwin_to_win32(path) return if not @@loaded_win32api Win32API.new("winmm.dll", "sndPlaySoundA", ["SI"], "I").call(path, 0x20000) when /mswin32/ return if not @@loaded_win32api Win32API.new("winmm.dll", "sndPlaySoundA", ["SI"], "I").call(path, 0x20000) when /darwin/ system("afplay #{path} >/dev/null 2>&1") else system("aplay #{path} >/dev/null 2>&1") end end |
.setenv(var, val) ⇒ Object
191 192 193 194 195 196 197 198 |
# File 'lib/rex/compat.rb', line 191 def self.setenv(var,val) if (is_windows and @@loaded_win32api) f = Win32API.new("kernel32", "SetEnvironmentVariable", ["P", "P"], "I") f.call(var, val + "\x00") else ENV[var]= val end end |
.temp_copy(path) ⇒ Object
Copy a file to a temporary path
259 260 261 262 263 264 265 266 267 268 |
# File 'lib/rex/compat.rb', line 259 def self.temp_copy(path) raise RuntimeError,"missing Tempfile" if not @@loaded_tempfile fd = File.open(path, "rb") tp = Tempfile.new("msftemp") tp.binmode tp.write(fd.read(File.size(path))) tp.close fd.close tp end |
.temp_delete(tp) ⇒ Object
Delete an opened temporary file
274 275 276 277 278 279 280 |
# File 'lib/rex/compat.rb', line 274 def self.temp_delete(tp) raise RuntimeError,"missing FileUtils" if not @@loaded_fileutils begin FileUtils.rm(tp.path) rescue end end |
.win32_ruby_path ⇒ Object
Obtain the path to our interpreter
204 205 206 207 208 209 210 211 212 |
# File 'lib/rex/compat.rb', line 204 def self.win32_ruby_path return nil if ! (is_windows and @@loaded_win32api) gmh = Win32API.new("kernel32", "GetModuleHandle", ["P"], "L") gmf = Win32API.new("kernel32", "GetModuleFileName", ["LPL"], "L") mod = gmh.call(nil) inf = "\x00" * 1024 gmf.call(mod, inf, 1024) inf.unpack("Z*")[0] end |
.win32_winexec(cmd) ⇒ Object
Call WinExec (equiv to system(“cmd &”))
217 218 219 220 221 |
# File 'lib/rex/compat.rb', line 217 def self.win32_winexec(cmd) return nil if ! (is_windows and @@loaded_win32api) exe = Win32API.new("kernel32", "WinExec", ["PL"], "L") exe.call(cmd, 0) end |