Method: Rex::IO::Stream#get

Defined in:
lib/rex/io/stream.rb

#get(timeout = nil, ltimeout = def_read_loop_timeout, opts = {}) ⇒ Object

This method reads as much data as it can from the wire given a maximum timeout.



211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
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
254
255
256
257
258
259
260
261
262
 
# File 'lib/rex/io/stream.rb', line 211

def get(timeout = nil, ltimeout = def_read_loop_timeout, opts = {})
  # For those people who are used to being able to use a negative timeout!
  if (timeout and timeout.to_i < 0)
    timeout = nil
  end

  # No data in the first place? bust.
  if (has_read_data?(timeout) == false)
    return nil
  end

  buf = ""
  lps = 0
  eof = false

  # Keep looping until there is no more data to be gotten..
  while (has_read_data?(ltimeout) == true)
    # Catch EOF errors so that we can handle them properly.
    begin
      temp = read(def_block_size)
    rescue EOFError
      eof = true
    end

    # If we read zero bytes and we had data, then we've hit EOF
    if (temp and temp.length == 0)
      eof = true
    end

    # If we reached EOF and there are no bytes in the buffer we've been
    # reading into, then throw an EOF error.
    if (eof)
      # If we've already read at least some data, then it's time to
      # break out and let it be processed before throwing an EOFError.
      if (buf.length > 0)
        break
      else
        raise EOFError
      end
    end

    break if (temp == nil or temp.empty? == true)

    buf += temp
    lps += 1

    break if (lps >= def_max_loops)
  end

  # Return the entire buffer we read in
  return buf
end