virustotal_api 0.5.4 → 0.5.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8bf33b0c496e55c74969e71e323aa64c664a4e7090db4f849601829143ee976a
-  data.tar.gz: 13eed4b64f8923c637e2155c3ece40f9bf55378c3f3a4cc7a321e29f58d8a357
+  metadata.gz: 4d61c90abdeb3a513e29da52a083c10044eace4aa1f51e6b758798abfee9137a
+  data.tar.gz: eae6f4130f49c3e5b6d17553aaf4c4e6725e7e74260ae2c0c250bc0130823c55
 SHA512:
-  metadata.gz: 168c48232321aecaa6ad6bf5d34d07eb417cbf82f4bcd9a69ed678bb1c28c23a0121a859151662d0e810ea1ca4d8cda2dc50a771d98eaf85f5c98f424b9c4458
-  data.tar.gz: '021498bfce536a34eadc04c7b12766ff605b1f3ae61163be4c88e72aba5e05f2ce4be692aaa601420bdc480327d81e6e5496fd4c293c60f6fdb11f9537d68363'
+  metadata.gz: ef0c2142c1e1ad07be96a88a035f7c906c22c67bb183e7c5906256a1ba83a03378eb985eb02fae68362bb9d9e1439fffef7c82eccfd9a3004144ac6640dd14f3
+  data.tar.gz: 2016ad82dadcc09fd03e3ca89cc324bcf7670a3a762a0ea1464569171e589e9d83052f7f95210406747f3d7b3b0ff93e8f67801a35c8d1d42bb482270b036717

data/.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,38 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**Desktop (please complete the following information):**
+ - OS: [e.g. iOS]
+ - Browser [e.g. chrome, safari]
+ - Version [e.g. 22]
+
+**Smartphone (please complete the following information):**
+ - Device: [e.g. iPhone6]
+ - OS: [e.g. iOS8.1]
+ - Browser [e.g. stock browser, safari]
+ - Version [e.g. 22]
+
+**Additional context**
+Add any other context about the problem here.

data/.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,20 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

data/.github/pull_request_template.md

@@ -0,0 +1,11 @@
+CHANGELOG: no-impact/Added/Changed/Deprecated/Removed/Fixed/Security
+
+## Summary
+
+## Testing
+
+1. _Describe how to test this change_
+
+## Other Information
+
+_Include screenshots, GIFs, and/or API responses as appropriate._

data/.github/workflows/ruby.yml

@@ -7,17 +7,56 @@ on:
     branches: [ master ]
 
 jobs:
-  Test:
+  Test-Ruby-2-5:
     runs-on: ubuntu-latest
-
     steps:
     - uses: actions/checkout@v2
     - name: Set up Ruby
     # To automatically get bug fixes and new Ruby versions for ruby/setup-ruby,
     # change this to (see https://github.com/ruby/setup-ruby#versioning):
-      uses: ruby/setup-ruby@v1
+      uses: ruby/setup-ruby@v1.70.1
       with:
         ruby-version: 2.5
+    - name: Show Ruby Version
+      run: ruby --version
+    - name: Install dependencies
+      run: bundle install
+    - name: Lint
+      run: bundle exec rake rubocop
+    - name: Run tests
+      run: bundle exec rake
+      
+  Test-Ruby-2-7:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Ruby
+    # To automatically get bug fixes and new Ruby versions for ruby/setup-ruby,
+    # change this to (see https://github.com/ruby/setup-ruby#versioning):
+      uses: ruby/setup-ruby@v1.70.1
+      with:
+        ruby-version: 2.7
+    - name: Show Ruby Version
+      run: ruby --version
+    - name: Install dependencies
+      run: bundle install
+    - name: Lint
+      run: bundle exec rake rubocop
+    - name: Run tests
+      run: bundle exec rake
+      
+  Test-Ruby-3-0:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Ruby
+    # To automatically get bug fixes and new Ruby versions for ruby/setup-ruby,
+    # change this to (see https://github.com/ruby/setup-ruby#versioning):
+      uses: ruby/setup-ruby@v1.70.1
+      with:
+        ruby-version: 3.0
+    - name: Show Ruby Version
+      run: ruby --version
     - name: Install dependencies
       run: bundle install
     - name: Lint

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # VirusTotal API Changelog
 
+## [0.5.5] - 2021-05-10
+* Add support for larger files
+  * [@Grandman](https://github.com/Grandman)
+
 ## [0.5.4] - 2020-12-10
 * Manage bad requests like not found
 * Use strict base64 encoding

data/README.md

@@ -65,7 +65,10 @@ require 'virustotal_api'
 file    = '/path/to/file'
 api_key = 'MY_API_KEY'
 
+# for upload file
 vtscan = VirustotalAPI::File.upload(file, api_key)
+# or large file (more than 32MB)
+vtscan = VirustotalAPI::File.upload_large(file, api_key)
 
 # Virustotal ID of file
 vtscan.id
@@ -233,10 +236,11 @@ vtgroup_report.report
 
 ## Contributors
 
-- [@postmodern](https://github.com/postmodern)
-- [@mkunkel](https://github.com/mkunkel)
 - [@jonnynux](https://github.com/jonnynux)
 - [@crondaemon](https://github.com/crondaemon/)
+- [@postmodern](https://github.com/postmodern)
+- [@mkunkel](https://github.com/mkunkel)
+- [@Grandman](https://github.com/Grandman)
 
 ## Contributing
 

data/lib/virustotal_api/base.rb

@@ -12,9 +12,9 @@ module VirustotalAPI
     attr_reader :report, :report_url, :id
 
     def initialize(report)
-      @report = report
+      @report     = report
       @report_url = report&.dig('data', 'links', 'self')
-      @id = report&.dig('data', 'id')
+      @id         = report&.dig('data', 'id')
     end
 
     # @return [String] string of API URI class method
@@ -22,6 +22,14 @@ module VirustotalAPI
       VirustotalAPI::URI
     end
 
+    def self.perform(path, api_key, method = :get, options = {})
+      base_perform(api_uri + path, api_key, method, options)
+    end
+
+    def self.perform_absolute(url, api_key, method = :get, options = {})
+      base_perform(url, api_key, method, options)
+    end
+
     # The actual method performing a call to Virustotal
     #
     # @param [String] url The url of the API
@@ -29,10 +37,10 @@ module VirustotalAPI
     # @param [String] method The HTTP method to use
     # @param [Hash] options Options to pass as payload
     # @return [VirustotalAPI::Domain] Report Search Result
-    def self.perform(url, api_key, method = :get, options = {})
+    def self.base_perform(url, api_key, method = :get, options = {})
       response = RestClient::Request.execute(
         method: method,
-        url: api_uri + url,
+        url: url,
         headers: { 'x-apikey': api_key },
         payload: options
       )
@@ -49,6 +57,8 @@ module VirustotalAPI
       raise VirustotalAPI::RateLimitError
     end
 
+    private_class_method :base_perform
+
     # @return [String] string of API URI instance method
     def api_uri
       self.class.api_uri

data/lib/virustotal_api/file.rb

@@ -23,7 +23,20 @@ module VirustotalAPI
     # @return [VirusotalAPI::File] Report
     def self.upload(file_path, api_key, opts = {})
       filename = opts.fetch('filename') { ::File.basename(file_path) }
-      report = perform('/files', api_key, :post, filename: filename, file: ::File.open(file_path, 'r'))
+      report   = perform('/files', api_key, :post, filename: filename, file: ::File.open(file_path, 'r'))
+      new(report)
+    end
+
+    # Upload a new file with size more than 32MB.
+    #
+    # @param [String] file_path for file to be sent for scan
+    # @param [String] api_key The key for virustotal
+    # @param [Hash] opts hash for additional options
+    # @return [VirusotalAPI::File] Report
+    def self.upload_large(file_path, api_key, opts = {})
+      filename = opts.fetch('filename') { ::File.basename(file_path) }
+      url      = upload_url(api_key)
+      report   = perform_absolute(url, api_key, :post, filename: filename, file: ::File.open(file_path, 'r'))
       new(report)
     end
 
@@ -37,6 +50,12 @@ module VirustotalAPI
       new(report)
     end
 
+    # @return [String] url for upload file
+    def self.upload_url(api_key)
+      data = perform('/files/upload_url', api_key)
+      data&.dig('data')
+    end
+
     # Check if the submitted hash is detected by an AV engine.
     #
     # @param [String] engine The engine to check.

data/lib/virustotal_api/version.rb

@@ -2,5 +2,5 @@
 
 module VirustotalAPI
   # The GEM version
-  VERSION = '0.5.4'
+  VERSION = '0.5.5'
 end

data/test/file_test.rb

@@ -51,6 +51,16 @@ class VirustotalAPIFileTest < Minitest::Test
     end
   end
 
+  def test_upload_large
+    VCR.use_cassette('large_file_upload') do
+      vt_file_upload = VirustotalAPI::File.upload_large(@file_path, @api_key)
+
+      assert vt_file_upload.exists?
+      assert vt_file_upload.report.is_a?(Hash)
+      assert vt_file_upload.id.is_a?(String)
+    end
+  end
+
   def test_analyse
     VCR.use_cassette('file_analyse') do
       vt_file_analyse = VirustotalAPI::File.analyse(@sha256, @api_key)

data/test/fixtures/file_upload.yml

@@ -49,6 +49,6 @@ http_interactions:
                 "type": "analysis"
             }
         }
-    http_version: 
+    http_version:
   recorded_at: Wed, 02 Sep 2020 14:02:37 GMT
 recorded_with: VCR 5.0.0

data/test/fixtures/large_file_upload.yml

@@ -0,0 +1,99 @@
+---
+http_interactions:
+- request:
+    method: get
+    uri: https://www.virustotal.com/api/v3/files/upload_url
+    body:
+    headers:
+      Accept:
+      - "*/*"
+      User-Agent:
+      - rest-client/2.1.0 (linux-gnu x86_64) ruby/2.5.1p57
+      X-Apikey:
+      - testapikey
+      Content-Length:
+      - '282'
+      Content-Type:
+      - multipart/form-data; boundary=----RubyFormBoundaryjv5FxFNLpwqDoUQA
+      Accept-Encoding:
+      - gzip;q=1.0,deflate;q=0.6,identity;q=0.3
+      Host:
+      - www.virustotal.com
+  response:
+    status:
+      code: 200
+      message: OK
+    headers:
+      Cache-Control:
+      - no-cache
+      Content-Type:
+      - application/json; charset=utf-8
+      X-Cloud-Trace-Context:
+      - 9c3d1ae8f345c5ca8060a4b793174891
+      Date:
+      - Wed, 02 Sep 2020 14:02:36 GMT
+      Server:
+      - Google Frontend
+      Content-Length:
+      - '128'
+    body:
+      encoding: UTF-8
+      string: |-
+        {
+            "data": "https://www.virustotal.com/_ah/upload/AMmfu6b-_DXUeFe36Sb3b0F4B8mH9Nb-CHbRoUNVOPwG/"
+        }
+    http_version:
+  recorded_at: Wed, 02 Sep 2020 14:02:37 GMT
+
+- request:
+    method: post
+    uri: https://www.virustotal.com/_ah/upload/AMmfu6b-_DXUeFe36Sb3b0F4B8mH9Nb-CHbRoUNVOPwG/
+    body:
+      encoding: ASCII-8BIT
+      string: "------RubyFormBoundaryjv5FxFNLpwqDoUQA\r\nContent-Disposition: form-data;
+        name=\"filename\"\r\n\r\nnull_file\r\n------RubyFormBoundaryjv5FxFNLpwqDoUQA\r\nContent-Disposition:
+        form-data; name=\"file\"; filename=\"null_file\"\r\nContent-Type: text/plain\r\n\r\n\n\r\n------RubyFormBoundaryjv5FxFNLpwqDoUQA--\r\n"
+    headers:
+      Accept:
+      - "*/*"
+      User-Agent:
+      - rest-client/2.1.0 (linux-gnu x86_64) ruby/2.5.1p57
+      X-Apikey:
+      - testapikey
+      Content-Length:
+      - '282'
+      Content-Type:
+      - multipart/form-data; boundary=----RubyFormBoundaryjv5FxFNLpwqDoUQA
+      Accept-Encoding:
+      - gzip;q=1.0,deflate;q=0.6,identity;q=0.3
+      Host:
+      - www.virustotal.com
+  response:
+    status:
+      code: 200
+      message: OK
+    headers:
+      Cache-Control:
+      - no-cache
+      Content-Type:
+      - application/json; charset=utf-8
+      X-Cloud-Trace-Context:
+      - 9c3d1ae8f345c5ca8060a4b793174891
+      Date:
+      - Wed, 02 Sep 2020 14:02:36 GMT
+      Server:
+      - Google Frontend
+      Content-Length:
+      - '128'
+    body:
+      encoding: UTF-8
+      string: |-
+        {
+            "data": {
+                "id": "NjhiMzI5ZGE5ODkzZTM0MDk5YzdkOGFkNWNiOWM5NDA6MTU5OTA1NTM1Ng==",
+                "type": "analysis"
+            }
+        }
+    http_version:
+  recorded_at: Wed, 02 Sep 2020 14:02:37 GMT
+recorded_with: VCR 5.0.0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: virustotal_api
 version: !ruby/object:Gem::Version
-  version: 0.5.4
+  version: 0.5.5
 platform: ruby
 authors:
 - pwelch
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-01-08 00:00:00.000000000 Z
+date: 2021-05-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -183,7 +183,9 @@ extra_rdoc_files: []
 files:
 - ".github/CODE_OF_CONDUCT.md"
 - ".github/CONTRIBUTING.md"
-- ".github/ISSUE_TEMPLATE.md"
+- ".github/ISSUE_TEMPLATE/bug_report.md"
+- ".github/ISSUE_TEMPLATE/feature_request.md"
+- ".github/pull_request_template.md"
 - ".github/workflows/ruby.yml"
 - ".gitignore"
 - ".rubocop.yml"
@@ -220,6 +222,7 @@ files:
 - test/fixtures/file_upload.yml
 - test/fixtures/group_find.yml
 - test/fixtures/ip.yml
+- test/fixtures/large_file_upload.yml
 - test/fixtures/null_file
 - test/fixtures/unscanned_url_find.yml
 - test/fixtures/url_analyse.yml
@@ -273,6 +276,7 @@ test_files:
 - test/fixtures/file_upload.yml
 - test/fixtures/group_find.yml
 - test/fixtures/ip.yml
+- test/fixtures/large_file_upload.yml
 - test/fixtures/null_file
 - test/fixtures/unscanned_url_find.yml
 - test/fixtures/url_analyse.yml

data/.github/ISSUE_TEMPLATE.md

@@ -1,15 +0,0 @@
-## Expected Behavior
-
-
-## Actual Behavior
-
-
-## Steps to Reproduce the Problem
-
-  1.
-
-## Specifications
-
-  - Version:
-  - Platform:
-  - Subsystem: