newrelic_security 0.1.0

data/.simplecov

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+
+if ENV['CI']
+  random = SecureRandom.uuid
+  SimpleCov.command_name(random)
+  SimpleCov.coverage_dir("coverage_#{random}")
+end
+
+SimpleCov.start do
+  enable_coverage(:branch)
+  SimpleCov.root(File.join(File.dirname(__FILE__), '/lib'))
+  track_files('**/*.rb')
+  formatter(SimpleCov::Formatter::SimpleFormatter) if ENV['CI']
+end

data/CHANGELOG.md

@@ -0,0 +1,69 @@
+# New Relic Ruby Security Agent Release Notes
+
+## v0.1.0
+
+Version 0.1.0 introduces `newrelic_security` agent for public preview under Newrelic pre-release software notice.
+
+- json_version: 1.2.0
+
+[New Relic Interactive Application Security Testing (IAST)](https://docs.newrelic.com/docs/iast/introduction/) can help you prevent cyberattacks and breaches on your applications by probing your running code for exploitable vulnerabilities.
+
+The `newrelic_security` gem provides this feature for Ruby. It depends on `newrelic_rpm` gem version v9.12.0 or above.
+
+At this time, the security agent is intended for use only within a dedicated security testing environment(or pre production) with data that can tolerate modification or deletion. The security agent is available as a separate Ruby gem, `newrelic_security`. It is recommended that this separate gem only be introduced to a security testing environment by leveraging Bundler grouping like so:
+
+```ruby
+  # Gemfile
+  gem 'newrelic_rpm'               # New Relic APM observability agent
+
+  group :security do
+    gem 'newrelic_security', require: false        # New Relic security agent
+  end
+```
+
+In order to run the security agent, make sure `newrelic_security` is not loaded by bundler but `newrelic_rpm` only by adding `require: false` in Gemfile. To run the security agent by newrelic_rpm, you need to update your configuration in newrelic.yml. At a minimum, `security.agent.enabled` and `security.enabled` must be set to `true`. They are `false` by default. Similar to the gem installation, we recommend you set these configurations for a special security testing environment only.
+
+Here's an example using `newrelic.yml`:
+
+```yaml
+  common: &default_settings
+    license_key: <%= ENV['NEW_RELIC_LICENSE_KEY'] %>
+    app_name: "Example app"
+
+  development:
+    <<: *default_settings
+    app_name: <%= app_name %> (Development)
+
+  security:
+    <<: *default_settings
+    security.enabled: true
+    security.agent.enabled: true
+
+  production:
+    <<: *default_settings
+```
+
+The following configuration relate to the `newrelic_security` gem:
+
+| Configuration name | Default | Behavior |
+| ------------------ | ------- |----------|
+| security.agent.enabled | `false` | If `true`, the security agent is loaded (a Ruby 'require' is performed) |
+| security.enabled | `false` |  If `true`, the security agent is started (the agent runs in its event loop) |
+| security.mode | `'IAST'` | Defines the mode for the security agent to operate in. Currently only 'IAST' is supported |
+| security.validator_service_url | `'wss://csec.nr-data.net'` | Defines the endpoint URL for posting security related data |
+| security.detection.rci.enabled | `true` | If `true`, enables RCI (remote code injection) detection |
+| security.detection.rxss.enabled | `true` | If `true`, enables RXSS (reflected cross-site scripting) detection |
+| security.detection.deserialization.enabled | `true` |  If `true`, enables deserialization detection |
+| security.application_info.port | `nil` | An Integer representing the port the application is listening on. This setting is mandatory for Passenger servers. Other servers should be detected by default. |
+
+## v0.0.3
+
+This is pre released test version.
+
+## v0.0.2
+
+This is pre released test version.
+
+## v0.0.1
+
+This is pre released test version.

data/CONTRIBUTING.md

@@ -0,0 +1,22 @@
+# New Relic CSEC Agent Contributor License Agreement
+
+Any feedback provided to New Relic about the pre-release csec-ruby-agent, including feedback provided as source code, comments, or other copyrightable or patentable material, is provided to New Relic under the terms of the Apache Software License, version 2. If you do not provide attribution information or a copy of the license with your feedback, you waive the performance of those requirements of the Apache License with respect to New Relic. The license grant regarding any feedback is irrevocable and persists past the termination of this license.
+
+Keep in mind that you will need to sign the CSEC Agent CLA via the click-through using CLA-Assistant. If you have any questions, please drop our New Relic GitHub team an email at opensource@newrelic.com.
+
+For more information about CLAs, please check out Alex Russell’s excellent post,
+[“Why Do I Need to Sign This?”](https://infrequently.org/2008/06/why-do-i-need-to-sign-this/).
+
+## Contributing Feedback
+
+Contributions are always welcome. Before contributing please read the
+[code of conduct](https://github.com/newrelic/.github/blob/main/CODE_OF_CONDUCT.md) and [search the issue tracker](issues); your issue may have already been discussed or fixed in `main`. To contribute,
+[fork](https://help.github.com/articles/fork-a-repo/) this repository, commit your changes, and [send a Pull Request](https://help.github.com/articles/using-pull-requests/).
+
+Note that our [code of conduct](https://github.com/newrelic/.github/blob/main/CODE_OF_CONDUCT.md) applies to all platforms and venues related to this project; please follow it in all your interactions with the project and its participants.
+
+## Feedback on Feature Requests
+
+Feature requests should be submitted in the [Issue tracker](../../issues), with a description of the expected behavior & use case, where they’ll remain closed until sufficient interest, [e.g. :+1: reactions](https://help.github.com/articles/about-discussions-in-issues-and-pull-requests/), has been [shown by the community](../../issues?q=label%3A%22votes+needed%22+sort%3Areactions-%2B1-desc).
+Before submitting an Issue, please search for similar ones in the
+[closed issues](../../issues?q=is%3Aissue+is%3Aclosed+label%3Aenhancement).

data/Gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in newrelic_security.gemspec
+gemspec
+
+gem "rake", "~> 12.0"

data/Gemfile_test

@@ -0,0 +1,58 @@
+source 'https://rubygems.org'
+gem 'excon'
+gem 'faraday'
+gem 'httpclient'
+gem 'net-http-persistent'
+gem 'net-ldap'
+gem 'nokogiri'
+gem 'minitest'
+gem 'rake'
+gem 'rack'
+gem 'rackup'
+gem 'rubocop'
+gem 'rubocop-minitest'
+gem 'rubocop-rake'
+gem 'simplecov'
+gem 'railties'
+if RUBY_VERSION >= '3.0.0'
+  gem 'rails', '~>6.0.0'
+elsif RUBY_VERSION < '2.5.0'
+  gem 'rails', '~>4'
+else
+  gem 'rails'
+end
+gem 'loofah', '~> 2.19.0'
+gem 'sinatra'
+gem 'padrino'
+gem 'grape'
+gem 'roda'
+gem 'docker-api'
+gem 'http'
+gem 'httpx'
+gem 'typhoeus'
+gem 'async-http'
+gem 'ethon'
+if RUBY_ENGINE == 'jruby'
+  gem 'activerecord-jdbc-adapter'
+  gem 'jdbc-sqlite3'
+  gem 'activerecord-jdbcsqlite3-adapter'
+  gem 'jdbc-mysql'
+  gem 'activerecord-jdbcmysql-adapter'
+  gem 'jdbc-postgres'
+  gem 'activerecord-jdbcpostgresql-adapter'
+  gem 'mongo', '2.19'
+  gem 'nio4r', '<= 2.5.4'
+  gem 'async', '1.32.1'
+else
+  if RUBY_VERSION < '2.5.0'
+    gem 'sqlite3', '~> 1.3.0'
+    gem 'pg', '~>0.5'
+  else
+    gem 'sqlite3', '~> 1.4.0'
+    gem 'pg'
+  end
+  gem 'mysql2'
+  gem 'mongo'
+  gem 'curb'
+  gem 'patron'
+end

data/LICENSE

@@ -0,0 +1,43 @@
+## New Relic Pre-Release Software Notice
+
+### Acceptance
+
+This software license applies to the repositories, directories, or files that it has been associated with (referred to here as the “Pre-Release Software”). In order to get any license under these terms, you must agree to them as both strict obligations and conditions to all your licenses.
+License
+
+New Relic, Inc. (“New Relic”) grants you a license under its copyrights and patents to do everything necessary for you to use the Pre-Release Software for the Permitted Purposes and subject to your compliance with the Confidentiality terms and all other Restrictions on Use.  All other uses are prohibited.
+
+### Permitted Purposes
+
+The Pre-Release Software is solely for use with New Relic’s proprietary SaaS service (“New Relic Service”).   To use the Pre-Release Software you must have a valid and active  subscription agreement to the New Relic Service. You may not use the Pre-Release Software in conjunction with any other similar service. 
+
+Use of the Pre-Release Software is limited to evaluation and feedback, including pull requests, of the pre-release features and capabilities. The use of the Pre-Release Software requires a signed New Relic Pre-Release Agreement.  The New Relic Pre-Release Agreement will control in event of a conflict with this license.  The Pre-Release Software is also subject to the New Relic pre-release policy.  A copy of the New Relic Pre-release policy available at https://docs.newrelic.com/docs/licenses/license-information/referenced-policies/new-relic-pre-release-policy/.
+
+### Confidentiality
+
+Any Pre-Release Software not posted publicly contains New Relic Trade Secrets. You may not disclose the features and capabilities of the Pre-Release Software to any third party by any means. You will use commercially reasonable measures to maintain the secrecy of the features and capabilities of the Pre-Release Software. Without limiting the permissions you receive under any open source software (“OSS”) license, you may not disclose the source code of any OSS included as part of the Pre-Release Software in a way that discloses the New Relic Trade Secrets.
+
+### Restrictions on Use 
+
+1. The use of the Pre-Release Software is limited to you and your organization. You cannot expose the Pre-Release Software to third parties via the network or use the Pre-Release Software on behalf of anyone else.
+2. You cannot modify the Pre-Release Software. You can compile or link the Pre-Release Software into your programs as described in the New Relic Documentation, but you cannot otherwise make any derivative works of the Pre-Release Software.
+3. You cannot reverse engineer or decompile the Pre-Release Software. 
+4. You cannot use the Pre-Release Software for competitive analysis, benchmarking, or any purposes that compete with or are intended to compete with New Relic.
+5. You cannot remove or obscure any proprietary notices in the Pre-Release Software.
+6. You can only copy the Software to the extent needed to use the Pre-Release Software within your organization. You cannot distribute the Software to any third parties.
+7. You can only use the New Relic Services and the Pre-Release Software consistent with New Relic’s product Documentation and Acceptable Use Policy.
+
+### Intellectual Property
+
+Any feedback provided to New Relic about the Pre-Release Software, including feedback provided as source code, comments, or other copyrightable or patentable material, is provided to New Relic under the terms of the Apache Software License, version 2. If you do not provide attribution information or a copy of the license with your feedback, you waive the performance of those requirements of the Apache License with respect to New Relic. The license grant regarding any feedback is irrevocable and persists past the termination of this license.
+
+### Termination
+
+This license terminates when the Pre-Release Software stops being provided by New Relic or when described by the New Relic Pre-Release Agreement, whichever comes first. When this license terminates, you have a commercially reasonable time to either remove the Pre-Release Software from your systems or to upgrade to a publicly released version of the Software.
+
+### Other Terms
+
+1. The Pre-Release Software is provided “AS IS.” New Relic disclaims all warranties, whether express, implied, statutory or otherwise, including warranties of merchantability, fitness for a particular purpose, title and noninfringement.
+2. The Pre-Release Software has known flaws.  You use the Pre-Release Software at your own risk.
+3. To the full extent permitted by law, New Relic will have no liability arising from or related to the Pre-Release Software or from this license for any direct, indirect, special, incidental, or consequential damages of any kind, even if advised of their possibility in advance, and regardless of legal theory (whether contract, tort, negligence, strict liability or otherwise).
+4. The Pre-Release Software may contain third-party software, including open source software. The third party software is governed by its own license and subject to its own terms, disclosed in the provided licensing information associated with the third party components.   If a source code disclosure is required under the terms of the license covering any OSS component, the source code can be requested of New Relic by emailing opensource@newrelic.com. 

data/README.md

@@ -0,0 +1,133 @@
+# New Relic Ruby security agent
+
+The New Relic security agent for Ruby is in public preview and is not generally available. This module enables instrumentation of Ruby applications for interactive application security analysis (IAST) and exposes exploitable vulnerabilities.
+
+**Note:** The IAST capability should only be used in pre-production environments and never in production. 
+
+[![Gem Version](https://badge.fury.io/rb/newrelic_security.svg)](https://badge.fury.io/rb/newrelic_security)
+
+## Installation
+
+The software is meant to be used along with the [New Relic Ruby Agent](https://github.com/newrelic/newrelic-ruby-agent). You can see New Relic ruby agent install instructions [here](https://github.com/newrelic/newrelic-ruby-agent#installing-and-using).
+
+#### With Bundler
+For using with Bundler, add the Ruby agent to your project's Gemfile.
+
+```
+gem 'newrelic_security', require: false
+```
+
+and run `bundle install` to activate the new gem.
+
+#### Without Bundler
+If you are not using Bundler, install the gem with:
+
+```
+gem install newrelic_security
+```
+
+## Getting Started
+The newrelic_security must be explicitly enabled in order to perform IAST analysis of the application. In the newrelic.yml, set the following parameters:
+
+```
+ security:
+   agent:
+     enabled: true
+   enabled: true
+```
+
+## Support Matrix
+### Ruby Versions
+- CRuby: 2.4 or higher
+- JRuby: 9.0 or higher
+### Web frameworks
+- Rails: 4.0 or higher
+- Sinatra: 2.0 or higher
+- Padrino: 0.15 or higher
+- Grape: 1.2 or higher
+- Roda: 3.19 or higher
+- gRPC: 1 or higher
+### Web servers
+- Puma: 3 or higher
+- Unicorn: 4 or higher
+- Thin: 1 or higher
+- Passenger: 5 or higher
+- Falcon: 0.29 or higher
+- Webrick: Supported for all agent-supported versions of Ruby
+### Databases
+- Active Record: 4.0 or higher
+- Sequel: 4.45 or higher
+- MongoDB: 2.4 or higher
+- Sqlite3 
+- Mysql2
+- PostgreSql
+
+### HTTP / network clients
+- Async::HTTP: 0.59.0 or higher
+- Curb: 0.8.1 or higher
+- Ethon: 0.12.0 or higher
+- Excon: 0.19.0 or higher
+- gRPC: 1.0.0 or higher
+- HttpClient: 2.2.0 or higher
+- HttpRb: 0.9.9 or higher
+- HTTPX: 1.0.0 or higher
+- Net::HTTP: Supported for all agent-supported versions of Ruby.
+- Typhoeus: 0.5.3 or higher
+- Patron: 0.10 or higher
+
+### Other
+- nokogiri
+- net-ldap
+
+### Supported Vulnerabilities
+- Remote Code Execution
+- SQL Injection
+- NoSQL Injection
+- Stored XSS
+- Reflected XSS
+- Reverse Shell attack
+- File Access
+- SSRF
+- Application Integrity Violation
+- LDAP Injection
+- XPath Injection
+
+## Testing
+We use Minitest for the Ruby Security agent.
+#### Prerequisite
+```
+rake test_bundle
+```
+#### Running All Unit tests
+The following command runs all the unit tests:
+```
+BUNDLE_GEMFILE=Gemfile_test bundle exec rake test
+```
+#### Running Specific Tests
+To run a single unit test file use the command like:
+```
+BUNDLE_GEMFILE=Gemfile_test bundle exec ruby test/newrelic_security/instrumentation-security/kernel/kernel_test.rb
+```
+
+## Feedback or Contribute
+
+Any feedback provided to New Relic about the New Relic csec-ruby-agent, including feedback provided as source code, comments, or other copyrightable or patentable material, is provided to New Relic under the terms of the Apache Software License, version 2. If you do not provide attribution information or a copy of the license with your feedback, you waive the performance of those requirements of the Apache License with respect to New Relic. The license grant regarding any feedback is irrevocable and persists past the termination of the preview license.
+
+Keep in mind that when you submit a pull request or other feedback, you'll need to sign the New Relic CSEC Agent CLA via the click-through using CLA-Assistant. You only have to sign this CLA one time per project.
+
+If you have any questions, or to execute our corporate CLA (which is required if your contribution is on behalf of a company), drop us an email at opensource@newrelic.com.
+
+**A note about vulnerabilities**
+
+As noted in our [security policy](../../security/policy), New Relic is committed to the privacy and security of our customers and their data. We believe that providing coordinated disclosure by security researchers and engaging with the security community are important means to achieve our security goals.
+
+If you believe you have found a security vulnerability in this project or any of New Relic's products or websites, we welcome and greatly appreciate you reporting it to New Relic through [HackerOne](https://hackerone.com/newrelic).
+
+If you would like to contribute to this project, review [these guidelines](./CONTRIBUTING.md).
+
+To all contributors, we thank you!  Without your contribution, this project would not be what it is today.  We also host a community project page dedicated to [Project Name](<LINK TO https://opensource.newrelic.com/projects/... PAGE>).
+
+## License
+The New Relic csec-ruby-agent is licensed under the New Relic Pre-Release Software Notice.
+
+The New Relic csec-ruby-agent also uses source code from third-party libraries. You can find full details on which libraries are used and the terms under which they are licensed in the [third-party notices document](./THIRD_PARTY_NOTICES.md).

data/README_agent.md

@@ -0,0 +1,44 @@
+<a href="https://opensource.newrelic.com/oss-category/#community-project"><picture><source media="(prefers-color-scheme: dark)" srcset="https://github.com/newrelic/opensource-website/raw/main/src/images/categories/dark/Community_Project.png"><source media="(prefers-color-scheme: light)" srcset="https://github.com/newrelic/opensource-website/raw/main/src/images/categories/Community_Project.png"><img alt="New Relic Open Source community project banner." src="https://github.com/newrelic/opensource-website/raw/main/src/images/categories/Community_Project.png"></picture></a>
+
+# [Name of Project] [build badges go here when available]
+
+>[Brief description - what is the software and what value does it provide? How often should users expect to get releases? How is versioning set up? What are some next phases for the project or how will it evolve?]
+
+## Installing and using [project name]
+
+> [Link to the relevant information for this agent on docs.newrelic.com. Create a bulleted list with links to install, usage, and getting started info on docs. Avoid duplicating information from docs in the open source content to ensure there's no inconsistency between the two.]
+
+
+## Building
+
+>[**Optional** - Include this section if users will need to follow specific instructions to build the software from source. Be sure to include any third-party build dependencies that need to be installed separately. As mentioned, link to docs for install info that's already included there. Remove this section if it's not needed.]
+
+## Testing
+
+>[**Optional** - Include instructions on how to run tests if we include tests with the codebase. Remove this section if it's not needed.]
+
+## Support
+
+New Relic hosts and moderates an online forum where customers can interact with New Relic employees as well as other customers to get help and share best practices. Like all official New Relic open source projects, there's a related Community topic in the New Relic Explorers Hub. You can find this project's topic/threads here:
+
+>[Add the url for the support thread here: discuss.newrelic.com]
+
+## Contribute
+
+We encourage your contributions to improve [project name]! Keep in mind that when you submit your pull request, you'll need to sign the CLA via the click-through using CLA-Assistant. You only have to sign the CLA one time per project.
+
+If you have any questions, or to execute our corporate CLA (which is required if your contribution is on behalf of a company), drop us an email at opensource@newrelic.com.
+
+**A note about vulnerabilities**
+
+As noted in our [security policy](../../security/policy), New Relic is committed to the privacy and security of our customers and their data. We believe that providing coordinated disclosure by security researchers and engaging with the security community are important means to achieve our security goals.
+
+If you believe you have found a security vulnerability in this project or any of New Relic's products or websites, we welcome and greatly appreciate you reporting it to New Relic through [HackerOne](https://hackerone.com/newrelic).
+
+If you would like to contribute to this project, review [these guidelines](./CONTRIBUTING.md).
+
+To all contributors, we thank you!  Without your contribution, this project would not be what it is today.  We also host a community project page dedicated to [Project Name](<LINK TO https://opensource.newrelic.com/projects/... PAGE>).
+
+## License
+[Project name] is licensed under the [Apache 2.0](http://apache.org/licenses/LICENSE-2.0.txt) License.
+>[If applicable: The [project name] also uses source code from third-party libraries. Full details on which libraries are used and the terms under which they are licensed can be found in the third-party notices document.]

data/Rakefile

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+require "#{File.dirname(__FILE__)}/lib/tasks/all.rb"
+
+desc 'Install dependencies needed to run tests'
+task :test_bundle do
+  if RUBY_VERSION < '2.5.0'
+    sh 'gem install bundler -v 1.17.2'
+    exec('bundle _1.17.2_ install --gemfile Gemfile_test')
+  else
+    exec('bundle install --gemfile Gemfile_test')
+  end
+end
+
+Rake::TestTask.new(:test) do |t|
+  # ENV['BUNDLE_GEMFILE'] = 'Gemfile_test'
+  t.libs << 'test'
+  t.libs << 'lib'
+  ENV['TESTOPTS'] = '--verbose'
+  t.test_files = FileList['test/**/*_test.rb']
+end
+
+task :rubocop do
+  require 'rubocop/rake_task'
+  RuboCop::RakeTask.new
+end

data/THIRD_PARTY_NOTICES.md

@@ -0,0 +1,36 @@
+# Third Party Notices
+
+The New Relic Ruby Security Agent uses source code from third party libraries which carry
+their own copyright notices and license terms. These notices are provided
+below.
+
+In the event that a required notice is missing or incorrect, please notify us
+by e-mailing [support@newrelic.com](mailto:support@newrelic.com).
+
+
+## [websocket-client-simple](https://rubygems.org/gems/websocket-client-simple)
+
+Copyright (c) 2013-2014 Sho Hashimoto
+
+Distributed under the following license(s):
+
+  * [The MIT License](http://opensource.org/licenses/MIT)
+
+
+## [event_emitter](https://rubygems.org/gems/event_emitter)
+
+Copyright (c) 2012 Sho Hashimoto
+
+Distributed under the following license(s):
+
+  * [The MIT License](http://opensource.org/licenses/MIT)
+
+
+## [websocket-ruby](https://rubygems.org/gems/websocket)
+
+Copyright (c) 2012 Bernard Potocki
+
+Distributed under the following license(s):
+
+  * [The MIT License](http://opensource.org/licenses/MIT)
+

data/lib/newrelic_security/agent/agent.rb

@@ -0,0 +1,109 @@
+require 'newrelic_security/agent/control/event_subscriber'
+require 'newrelic_security/agent/control/websocket_client'
+require 'newrelic_security/agent/control/event_processor'
+require 'newrelic_security/agent/control/iast_client'
+require 'newrelic_security/agent/control/iast_data_transfer_request'
+require 'newrelic_security/agent/control/control_command'
+require 'newrelic_security/agent/control/fuzz_request'
+require 'newrelic_security/agent/control/reflected_xss'
+require 'newrelic_security/agent/control/http_context'
+require 'newrelic_security/agent/control/grpc_context'
+require 'newrelic_security/agent/control/collector'
+require 'newrelic_security/agent/control/app_info'
+require 'newrelic_security/agent/control/application_url_mappings'
+require 'newrelic_security/agent/control/health_check'
+require 'newrelic_security/agent/control/event'
+require 'newrelic_security/agent/control/critical_message'
+require 'newrelic_security/agent/control/event_counter'
+require 'newrelic_security/agent/control/event_stats'
+require 'newrelic_security/agent/control/exit_event'
+require 'newrelic_security/instrumentation-security/instrumentation_loader'
+
+module NewRelic::Security
+  module Agent
+    class Agent
+
+      attr_accessor :websocket_client, :event_processor, :iast_client, :http_request_count, :event_processed_count, :event_sent_count, :event_drop_count, :route_map, :iast_event_stats, :rasp_event_stats, :exit_event_stats
+
+      def initialize
+        NewRelic::Security::Agent.config
+        create_agent_home
+        NewRelic::Security::Agent::Utils.enable_object_space_in_jruby
+        NewRelic::Security::Agent.config.save_uuid
+        @started = false
+        @event_subscriber = NewRelic::Security::Agent::Control::EventSubscriber.new
+        @started = true
+        @route_map = ::Set.new
+        @http_request_count = NewRelic::Security::Agent::Control::EventCounter.new
+        @event_processed_count = NewRelic::Security::Agent::Control::EventCounter.new
+        @event_sent_count = NewRelic::Security::Agent::Control::EventCounter.new
+        @event_drop_count = NewRelic::Security::Agent::Control::EventCounter.new
+        @iast_event_stats = NewRelic::Security::Agent::Control::EventStats.new
+        @rasp_event_stats = NewRelic::Security::Agent::Control::EventStats.new
+        @exit_event_stats = NewRelic::Security::Agent::Control::EventStats.new
+      end
+
+      def init
+        NewRelic::Security::Agent.logger.info "Initializing Security Agent with config : #{NewRelic::Security::Agent::Utils.filtered_log(NewRelic::Security::Agent.config.inspect)}\n"
+        @ready = false
+        start_event_processor
+        start_websocket_client
+        NewRelic::Security::Instrumentation::InstrumentationLoader.add_instrumentation()
+        NewRelic::Security::Agent.logger.info "Security Agent ready.\n"
+        NewRelic::Security::Agent.init_logger.info "Security Agent ready.\n"
+        @ready = true
+      rescue Exception => exception
+        NewRelic::Security::Agent.logger.error "Exception in security agent init: #{exception.inspect} #{exception.backtrace}\n"
+      end
+
+      def start_websocket_client
+        NewRelic::Security::Agent::Control::WebsocketClient.instance.close(false) if NewRelic::Security::Agent::Control::WebsocketClient.instance.is_open?
+        @websocket_client = NewRelic::Security::Agent::Control::WebsocketClient.instance.connect
+      end
+
+      def start_event_processor
+        @event_processor&.event_dequeue_thread&.kill 
+        @event_processor&.healthcheck_thread&.kill
+        @event_processor = nil
+        @event_processor = NewRelic::Security::Agent::Control::EventProcessor.new
+      end
+
+      def start_iast_client
+        @iast_client&.iast_dequeue_threads&.each { |t| t.kill if t }
+        @iast_client&.iast_data_transfer_request_processor_thread&.kill
+        @iast_client = nil
+        @iast_client = NewRelic::Security::Agent::Control::IASTClient.new
+      end
+
+      def self.config
+        ::NewRelic::Security::Agent.config
+      end
+
+      def add_instrumentation
+        
+      end
+
+      def create_agent_home
+        log_dir = ::File.join(NewRelic::Security::Agent.config[:log_file_path], SEC_HOME_PATH, LOGS_DIR)
+        find_or_create_file_path(log_dir)
+        tmp_dir = ::File.join(NewRelic::Security::Agent.config[:log_file_path], SEC_HOME_PATH, TMP_DIR)
+        find_or_create_file_path(tmp_dir)
+      end
+
+      def find_or_create_file_path(path)
+        ::FileUtils.mkdir_p(path) unless ::File.directory?(path)
+        ::File.directory?(path)
+      rescue
+        return false
+      end
+
+      def reconnect(sleep_time = 15)
+        NewRelic::Security::Agent::Control::WebsocketClient.instance.close(false) if NewRelic::Security::Agent::Control::WebsocketClient.instance.is_open?
+        NewRelic::Security::Agent.logger.info "Trying to reconnect to websocket connection in #{sleep_time} sec..."
+        sleep sleep_time
+        NewRelic::Security::Agent.agent.start_websocket_client
+      end
+
+    end
+  end
+end

data/lib/newrelic_security/agent/configuration/default_source.rb

@@ -0,0 +1,8 @@
+module NewRelic::Security
+  module Agent
+    module Configuration
+      class DefaultSource
+      end
+    end
+  end
+end

data/lib/newrelic_security/agent/configuration/environment_source.rb

@@ -0,0 +1,8 @@
+module NewRelic::Security
+  module Agent
+    module Configuration
+      class EnvironmentSource
+      end
+    end
+  end
+end