zabbix_receiver 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1db84526dbaff0fd80bb4cef61161ceb0074ba7a
+  data.tar.gz: c303aa0a685791c419d8a77610f8408974075701
+SHA512:
+  metadata.gz: d15a574d116ca65dee81029ab96f8d92f747654bf2625e77935f5b384ce1e29ae7e87472a7afb78365f98859c450f47f8944fe50c04421e0585d4eced63ba096
+  data.tar.gz: de862f50067ed77859ddb88751c00526cfca9c368b31e7a878d749137a92d8a9f2619fd26a8fa4feb9ac3bfea640bba194d876fb918e733cf00171fb327fc48e

data/.gitignore

@@ -0,0 +1,14 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--require spec_helper

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in zabbix_receiver.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Ryota Arai
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,31 @@
+# ZabbixReceiver
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'zabbix_receiver'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install zabbix_receiver
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/zabbix_receiver/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/lib/zabbix_receiver.rb

@@ -0,0 +1,6 @@
+require "zabbix_receiver/server"
+require "zabbix_receiver/version"
+
+module ZabbixReceiver
+  # Your code goes here...
+end

data/lib/zabbix_receiver/server.rb

@@ -0,0 +1,111 @@
+require 'json'
+require 'logger'
+
+module ZabbixReceiver
+  class Server
+    ZABBIX_HEADER = "ZBXD\x01"
+
+    def initialize(options = {})
+      @options = {
+        proxy_to: {
+          port: 10051,
+        },
+      }.merge(options)
+
+      validate_options
+
+      @logger = options[:logger] || Logger.new(StringIO.new)
+
+      @blocks_on_receive_sender_data = []
+    end
+
+    def on_receive_sender_data(&block)
+      @blocks_on_receive_sender_data << block
+    end
+
+    def start(address, port)
+      server = TCPServer.open(address, port)
+
+      while true
+        Thread.start(server.accept) do |f|
+          accept(f)
+        end
+      end
+    end
+
+    def accept(f)
+      request = parse(f)
+
+      @logger.debug "Request: #{request}"
+
+      request_type = request['request']
+
+      case request_type
+      when 'active checks'
+        proxy_request(f)
+      when 'sender data'
+        @blocks_on_receive_sender_data.each do |block|
+          block.call(request)
+        end
+
+        count = request['data'].size
+
+        respond_with(f, {
+          "response" => "success",
+          "info" => "Processed #{count} Failed 0 Total #{count} Seconds spent 0.000000"
+        })
+      else
+        @logger.error "Unknown request type (#{request_type})"
+      end
+    ensure
+      f && f.close
+    end
+
+    def proxy_request(f)
+      f.seek(0)
+
+      socket = TCPSocket.open(
+        @options[:proxy_to][:host],
+        @options[:proxy_to][:port],
+      )
+      socket.write(f.read)
+      f.write(socket.read)
+    ensure
+      socket && socket.close
+    end
+
+    private
+
+    def parse(f)
+      f.seek(0)
+
+      unless f.read(5) == ZABBIX_HEADER
+        @logger.error "Invalid Zabbix request"
+        return
+      end
+
+      length = f.read(8).unpack('q').first
+      body = f.read
+      unless body.size == length
+        @logger.error "Length mismatch"
+        return
+      end
+
+      JSON.parse(body)
+    end
+
+    def respond_with(f, payload)
+      payload = payload.to_json
+      f.write(ZABBIX_HEADER + [payload.bytesize].pack('q') + payload)
+    end
+
+    def validate_options
+      [:host, :port].each do |k|
+        unless @options[:proxy_to][k]
+          raise "options[:proxy_to][:#{k}] is required."
+        end
+      end
+    end
+  end
+end
+

data/lib/zabbix_receiver/version.rb

@@ -0,0 +1,3 @@
+module ZabbixReceiver
+  VERSION = "0.0.1"
+end

data/spec/spec_helper.rb

@@ -0,0 +1,91 @@
+require 'zabbix_receiver'
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# The generated `.rspec` file contains `--require spec_helper` which will cause this
+# file to always be loaded, without a need to explicitly require it in any files.
+#
+# Given that it is always loaded, you are encouraged to keep this file as
+# light-weight as possible. Requiring heavyweight dependencies from this file
+# will add to the boot time of your test suite on EVERY test run, even for an
+# individual file that may not need all of that loaded. Instead, consider making
+# a separate helper file that requires the additional dependencies and performs
+# the additional setup, and require it from the spec files that actually need it.
+#
+# The `.rspec` file also contains a few flags that are not defaults but that
+# users commonly want.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  # rspec-expectations config goes here. You can use an alternate
+  # assertion/expectation library such as wrong or the stdlib/minitest
+  # assertions if you prefer.
+  config.expect_with :rspec do |expectations|
+    # This option will default to `true` in RSpec 4. It makes the `description`
+    # and `failure_message` of custom matchers include text for helper methods
+    # defined using `chain`, e.g.:
+    # be_bigger_than(2).and_smaller_than(4).description
+    #   # => "be bigger than 2 and smaller than 4"
+    # ...rather than:
+    #   # => "be bigger than 2"
+    expectations.include_chain_clauses_in_custom_matcher_descriptions = true
+  end
+
+  # rspec-mocks config goes here. You can use an alternate test double
+  # library (such as bogus or mocha) by changing the `mock_with` option here.
+  config.mock_with :rspec do |mocks|
+    # Prevents you from mocking or stubbing a method that does not exist on
+    # a real object. This is generally recommended, and will default to
+    # `true` in RSpec 4.
+    mocks.verify_partial_doubles = true
+  end
+
+# The settings below are suggested to provide a good initial experience
+# with RSpec, but feel free to customize to your heart's content.
+=begin
+  # These two settings work together to allow you to limit a spec run
+  # to individual examples or groups you care about by tagging them with
+  # `:focus` metadata. When nothing is tagged with `:focus`, all examples
+  # get run.
+  config.filter_run :focus
+  config.run_all_when_everything_filtered = true
+
+  # Limits the available syntax to the non-monkey patched syntax that is recommended.
+  # For more details, see:
+  #   - http://myronmars.to/n/dev-blog/2012/06/rspecs-new-expectation-syntax
+  #   - http://teaisaweso.me/blog/2013/05/27/rspecs-new-message-expectation-syntax/
+  #   - http://myronmars.to/n/dev-blog/2014/05/notable-changes-in-rspec-3#new__config_option_to_disable_rspeccore_monkey_patching
+  config.disable_monkey_patching!
+
+  # This setting enables warnings. It's recommended, but in some cases may
+  # be too noisy due to issues in dependencies.
+  config.warnings = true
+
+  # Many RSpec users commonly either run the entire suite or an individual
+  # file, and it's useful to allow more verbose output when running an
+  # individual spec file.
+  if config.files_to_run.one?
+    # Use the documentation formatter for detailed output,
+    # unless a formatter has already been configured
+    # (e.g. via a command-line flag).
+    config.default_formatter = 'doc'
+  end
+
+  # Print the 10 slowest examples and example groups at the
+  # end of the spec run, to help surface which specs are running
+  # particularly slow.
+  config.profile_examples = 10
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = :random
+
+  # Seed global randomization in this process using the `--seed` CLI option.
+  # Setting this allows you to use `--seed` to deterministically reproduce
+  # test failures related to randomization by passing the same `--seed` value
+  # as the one that triggered the failure.
+  Kernel.srand config.seed
+=end
+end

data/spec/zabbix_receiver/server_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+
+describe ZabbixReceiver::Server do
+  let(:options) { {
+    proxy_to: {
+      host: 'zabbix-server',
+      port: 'port',
+    }
+  } }
+  subject(:server) { described_class.new(options) }
+
+  describe '#accept' do
+    context 'with sender data' do
+      let(:request) { "ZBXD\u0001b\u0000\u0000\u0000\u0000\u0000\u0000\u0000{\n\t\"request\":\"sender data\",\n\t\"data\":[\n\t\t{\n\t\t\t\"host\":\"hello\",\n\t\t\t\"key\":\"key\",\n\t\t\t\"value\":\"value\"}]}" }
+
+      it 'receives sender data' do
+        io = StringIO.new(request.dup)
+
+        received_data = nil
+        server.on_receive_sender_data do |data|
+          received_data = data
+        end
+
+        server.accept(io)
+        expected_response = "ZBXD\u0001S\u0000\u0000\u0000\u0000\u0000\u0000\u0000{\"response\":\"success\",\"info\":\"Processed 1 Failed 0 Total 1 Seconds spent 0.000000\"}"
+        expect(io.string).to eq(request + expected_response)
+        expect(received_data).to eq({
+          'request' => 'sender data',
+          'data' => [{
+            'host'  => 'hello',
+            'key'   => 'key',
+            'value' => 'value',
+          }],
+        })
+      end
+    end
+
+    context 'with active checks' do
+      let(:socket_to_zabbix_server) { double(:socket) }
+      let(:request) { "ZBXD\u0001)\x00\x00\x00\x00\x00\x00\x00{\"request\":\"active checks\",\"key\":\"value\"}" }
+      let(:response) { "response_from_zabbix_server" }
+
+      it 'proxies a request to zabbix server' do
+        allow(socket_to_zabbix_server).to receive(:seek)
+
+        expect(TCPSocket).to receive(:open).with('zabbix-server', 'port').and_return(socket_to_zabbix_server)
+        expect(socket_to_zabbix_server).to receive(:close)
+        expect(socket_to_zabbix_server).to receive(:write).with(request)
+        expect(socket_to_zabbix_server).to receive(:read).and_return(response)
+
+        io = StringIO.new(request.dup)
+        server.accept(io)
+        expect(io.string).to eq(request + response)
+      end
+    end
+  end
+end

data/zabbix_receiver.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'zabbix_receiver/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "zabbix_receiver"
+  spec.version       = ZabbixReceiver::VERSION
+  spec.authors       = ["Ryota Arai"]
+  spec.email         = ["ryota-arai@cookpad.com"]
+  spec.summary       = %q{Server to receive sender data from zabbix-agent.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec"
+end

metadata

@@ -0,0 +1,100 @@
+--- !ruby/object:Gem::Specification
+name: zabbix_receiver
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Ryota Arai
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-01-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- ryota-arai@cookpad.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/zabbix_receiver.rb
+- lib/zabbix_receiver/server.rb
+- lib/zabbix_receiver/version.rb
+- spec/spec_helper.rb
+- spec/zabbix_receiver/server_spec.rb
+- zabbix_receiver.gemspec
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Server to receive sender data from zabbix-agent.
+test_files:
+- spec/spec_helper.rb
+- spec/zabbix_receiver/server_spec.rb