Class: GoogleSafeBrowsing::APIv2

Inherits:
Object
  • Object
show all
Defined in:
lib/google_safe_browsing/api_v2.rb

Overview

Main Interface for Module

Class Method Summary collapse

Class Method Details

.delay(delay_seconds) ⇒ Object

Can be used to force a delay into a script running updates

the return value of update

Parameters:

  • delay_seconds (Integer) —

    the number of seconds to delay, should be



64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/google_safe_browsing/api_v2.rb', line 64

def self.delay(delay_seconds)
  GoogleSafeBrowsing.logger.info \
    "Google told us to wait for #{delay_seconds} seconds"
  GoogleSafeBrowsing.logger.info 'We will wait....'
  start_time = Time.now
  until start_time + delay_seconds <= Time.now
      GoogleSafeBrowsing.logger.info \
        "#{(delay_seconds - (Time.now - start_time)).to_i}..."
      sleep(10)
  end
  GoogleSafeBrowsing.logger.info 'Thank you for being patient'
end

.lookup(url) ⇒ String?

Performs a lookup of the given url

Parameters:

  • url (String) —

    a url string to be looked up

Returns:

  • (String, nil) —

    the friendly list name if found, or ‘nil`



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/google_safe_browsing/api_v2.rb', line 27

def self.lookup(url)
  urls = Canonicalize.urls_for_lookup(url.force_encoding('ASCII-8BIT'))
  return nil if urls.empty?

  hashes = HashHelper.urls_to_hashes(urls)
  raw_hash_array = hashes.map { |h| h.to_s }

  full = FullHash.where(full_hash: raw_hash_array).first
  return GoogleSafeBrowsing.friendly_list_name(full.list) if full

  hits =  AddShavar.where(prefix: hashes.map { |h| h.prefix }).map { |s| [s.list, s.prefix] }
  safes = SubShavar.where(prefix: hashes.map { |h| h.prefix }).map { |s| [s.list, s.prefix] }

  reals = hits - safes

  if reals.any?
    full_hashes = HttpHelper.request_full_hashes(reals.map { |r| r[1] })

    # save hashes first
    # cannot return early because all FullHashes need to be saved
    hit_list = nil
    full_hashes.each do |hash|
      FullHash.create!(list: hash[:list],
                       add_chunk_number: hash[:add_chunk_num],
                       full_hash: hash[:full_hash])

      hit_list = hash[:list] if raw_hash_array.include?(hash[:full_hash])
    end
    return GoogleSafeBrowsing.friendly_list_name(hit_list)
  end
  nil
end

.update ⇒ Integer

Completes an update

Returns:

  • (Integer) —

    the number of seconds before this method should be called again



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
# File 'lib/google_safe_browsing/api_v2.rb', line 7

def self.update
  HttpHelper.get_keys unless GoogleSafeBrowsing.config.have_keys?

  data_response = HttpHelper.get_data

  to_do_array = ResponseHelper.parse_data_response(data_response.body)

  to_do_array[:lists].each do |list|
    to_do_array[:data_urls][list].each do |url|
      GoogleSafeBrowsing.logger.info "#{list} - #{url}\n"
      ResponseHelper.receive_data('http://' + url, list)
    end
  end
  to_do_array[:delay_seconds]
end