MovableInkAWS 1.0.18 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c67a1debb22324f3f567c390c5a5baf61c714b56567712dd6269dd044b2ed9dd
-  data.tar.gz: 9b02840d52d5ee4a4f99060ec274f9938617247e1df9468efbb352fb2ff2285f
+  metadata.gz: 88f1e4859005fa8150bd22e1c712a404ae26e79d39af2812ba5771b17c24fc70
+  data.tar.gz: b9611d91105694d9da25bc8551928f18e9561ef139ec7a291c0f88df5af93c87
 SHA512:
-  metadata.gz: f547de79011e7fc83bb33db7d905da20c55d8a5308ef6e539193bfabacbcf1a5f0cde3a417ec454b6c24f0361ec1bef5b0d05ebbba5730f1cc5427643be28550
-  data.tar.gz: 87ad9c72f123ef2e70d3ff35d89eeef5dac3948144668e88554de402eff5b71e733144d2d7dc9fd51473662a3e36fc479525c1ff01a63ac0d34c1046d1d2d2c7
+  metadata.gz: 9ff752b86badeb543f248e9b81781404ed35336b9a28bf0b9e59903945a26ba7c4a7323f77ec99a81c74421058c100fe5d7c82b172f65322433bc20bc6b3931a
+  data.tar.gz: 201f767f782b5d675834a1539e9e336b25b06773c7fa5e9162dc37d4c21fda0efbadec142bfd52f0ff7ef3082b5f0a2c2a7bc7833a110fe1c99f7f2ac26df079

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    MovableInkAWS (1.0.18)
+    MovableInkAWS (1.1.0)
       aws-sdk-athena (~> 1)
       aws-sdk-autoscaling (~> 1)
       aws-sdk-cloudwatch (~> 1)
@@ -20,8 +20,8 @@ GEM
   remote: https://rubygems.org/
   specs:
     aws-eventstream (1.1.0)
-    aws-partitions (1.322.0)
-    aws-sdk-athena (1.26.0)
+    aws-partitions (1.325.0)
+    aws-sdk-athena (1.27.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
     aws-sdk-autoscaling (1.37.0)
@@ -30,18 +30,18 @@ GEM
     aws-sdk-cloudwatch (1.38.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-core (3.97.0)
+    aws-sdk-core (3.97.1)
       aws-eventstream (~> 1, >= 1.0.2)
       aws-partitions (~> 1, >= 1.239.0)
       aws-sigv4 (~> 1.1)
       jmespath (~> 1.0)
-    aws-sdk-ec2 (1.164.0)
+    aws-sdk-ec2 (1.165.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-elasticache (1.35.0)
+    aws-sdk-elasticache (1.36.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-kms (1.32.0)
+    aws-sdk-kms (1.33.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
     aws-sdk-rds (1.85.0)
@@ -50,14 +50,14 @@ GEM
     aws-sdk-route53 (1.35.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-s3 (1.67.0)
+    aws-sdk-s3 (1.67.1)
       aws-sdk-core (~> 3, >= 3.96.1)
       aws-sdk-kms (~> 1)
       aws-sigv4 (~> 1.1)
     aws-sdk-sns (1.24.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-ssm (1.79.0)
+    aws-sdk-ssm (1.80.0)
       aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
     aws-sigv4 (1.1.4)

data/lib/movable_ink/aws.rb

@@ -1,4 +1,5 @@
 require_relative 'aws/errors'
+require_relative 'aws/metadata'
 require_relative 'aws/ec2'
 require_relative 'aws/sns'
 require_relative 'aws/autoscaling'
@@ -13,6 +14,7 @@ require 'aws-sdk-cloudwatch'
 
 module MovableInk
   class AWS
+    include Metadata
     include EC2
     include SNS
     include Autoscaling
@@ -93,14 +95,6 @@ module MovableInk
       self.class.regions
     end
 
-    def availability_zone
-      @availability_zone ||= begin
-        az = `ec2metadata --availability-zone 2>/dev/null`.chomp
-        raise(MovableInk::AWS::Errors::EC2Required) if az.empty?
-        az
-      end
-    end
-
     def my_region
       @my_region ||= availability_zone.chop
     end

data/lib/movable_ink/aws/ec2.rb

@@ -63,14 +63,6 @@ module MovableInk
         end
       end
 
-      def instance_id
-        @instance_id ||= begin
-          id = `ec2metadata --instance-id 2>/dev/null`.chomp
-          raise(MovableInk::AWS::Errors::EC2Required) if id.empty?
-          id
-        end
-      end
-
       def instance_tags
         @instance_tags ||= run_with_backoff(quiet: true) do
           ec2.describe_tags({
@@ -79,14 +71,6 @@ module MovableInk
         end
       end
 
-      def private_ipv4
-        @ipv4 ||= begin
-          ipv4 = `ec2metadata --local-ipv4 2>/dev/null`.chomp
-          raise(MovableInk::AWS::Errors::EC2Required) if ipv4.empty?
-          ipv4
-        end
-      end
-
       def me
         @me ||= all_instances.select{|instance| instance.instance_id == instance_id}.first rescue nil
       end

data/lib/movable_ink/aws/metadata.rb

@@ -0,0 +1,62 @@
+require 'net/http'
+
+module MovableInk
+  class AWS
+    module Metadata
+      def http
+        @http ||= begin
+          http = Net::HTTP.new("169.254.169.254", 80)
+          http.open_timeout = 1
+          http.read_timeout = 1
+          http
+        end
+      end
+
+      def retrieve_metadata(key)
+        request = Net::HTTP::Get.new("/latest/meta-data/#{key}")
+        request['X-aws-ec2-metadata-token'] = imds_token
+        response = http.request(request)
+        response.body
+      rescue
+        ""
+      end
+
+      def availability_zone
+        @availability_zone ||= begin
+          az = retrieve_metadata('placement/availability-zone')
+          raise(MovableInk::AWS::Errors::EC2Required) if az.empty?
+          az
+        end
+      end
+
+      def instance_id
+        @instance_id ||= begin
+          id = retrieve_metadata('instance-id')
+          raise(MovableInk::AWS::Errors::EC2Required) if id.empty?
+          id
+        end
+      end
+
+      def private_ipv4
+        @ipv4 ||= begin
+          ipv4 = retrieve_metadata('local-ipv4')
+          raise(MovableInk::AWS::Errors::EC2Required) if ipv4.empty?
+          ipv4
+        end
+      end
+
+      private
+
+      def imds_token
+        begin
+          request = Net::HTTP::Put.new('/latest/api/token')
+          request['X-aws-ec2-metadata-token-ttl-seconds'] = 120
+          response = http.request(request)
+          response.body
+        rescue
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/movable_ink/version.rb

@@ -1,5 +1,5 @@
 module MovableInk
   class AWS
-    VERSION = '1.0.18'
+    VERSION = '1.1.0'
   end
 end

data/spec/aws_spec.rb

@@ -2,15 +2,6 @@ require_relative '../lib/movable_ink/aws'
 
 describe MovableInk::AWS do
   context "outside EC2" do
-    it "should raise an error if EC2 is required" do
-      aws = MovableInk::AWS.new
-      allow(aws).to receive(:`).with('ec2metadata --instance-id 2>/dev/null').and_return("")
-      allow(aws).to receive(:`).with('ec2metadata --availability-zone 2>/dev/null').and_return("")
-
-      expect{ aws.instance_id }.to raise_error(MovableInk::AWS::Errors::EC2Required)
-      expect{ aws.availability_zone }.to raise_error(MovableInk::AWS::Errors::EC2Required)
-    end
-
     it 'doesnt raise an error if instance_id is set' do
       aws = MovableInk::AWS.new(instance_id: 'i-987654321')
       expect(aws.instance_id).to eq('i-987654321')
@@ -18,21 +9,9 @@ describe MovableInk::AWS do
   end
 
   context "inside EC2" do
-    it "should call ec2metadata to get the instance ID" do
-      aws = MovableInk::AWS.new
-      expect(aws).to receive(:`).with('ec2metadata --instance-id 2>/dev/null').and_return("i-12345\n")
-      expect(aws.instance_id).to eq('i-12345')
-    end
-
-    it "should call ec2metadata to get the availability zone" do
-      aws = MovableInk::AWS.new
-      expect(aws).to receive(:`).with('ec2metadata --availability-zone 2>/dev/null').and_return("us-east-1a\n")
-      expect(aws.availability_zone).to eq('us-east-1a')
-    end
-
     it "should find the datacenter by region" do
       aws = MovableInk::AWS.new
-      expect(aws).to receive(:`).with('ec2metadata --availability-zone 2>/dev/null').and_return("us-east-1a\n")
+      expect(aws).to receive(:retrieve_metadata).with('placement/availability-zone').and_return("us-east-1a")
       expect(aws.datacenter).to eq('iad')
     end
 

data/spec/ec2_spec.rb

@@ -4,17 +4,11 @@ describe MovableInk::AWS::EC2 do
   context "outside EC2" do
     it "should raise an error if trying to load mi_env outside of EC2" do
       aws = MovableInk::AWS.new
-      allow(aws).to receive(:`).with('ec2metadata --instance-id 2>/dev/null').and_return("")
-      allow(aws).to receive(:`).with('ec2metadata --availability-zone 2>/dev/null').and_return("")
+      allow(aws).to receive(:retrieve_metadata).with('instance-id').and_return("")
+      allow(aws).to receive(:retrieve_metadata).with('placement/availability-zone').and_return("")
       expect{ aws.mi_env }.to raise_error(MovableInk::AWS::Errors::EC2Required)
     end
 
-    it 'should raise an error if trying to load private_ipv4 outside of EC2' do
-      aws = MovableInk::AWS.new
-      allow(aws).to receive(:`).with('ec2metadata --local-ipv4 2>/dev/null').and_return('')
-      expect{ aws.private_ipv4 }.to raise_error(MovableInk::AWS::Errors::EC2Required)
-    end
-
     it "should use the provided environment" do
       aws = MovableInk::AWS.new(environment: 'test')
       expect(aws.mi_env).to eq('test')
@@ -67,12 +61,6 @@ describe MovableInk::AWS::EC2 do
       expect(aws.mi_env).to eq('test')
     end
 
-    it 'calls ec2metadata to get the private ipv4 address of the instance' do
-      aws = MovableInk::AWS.new
-      allow(aws).to receive(:`).with('ec2metadata --local-ipv4 2>/dev/null').and_return('10.0.0.1')
-      expect(aws.private_ipv4).to eq('10.0.0.1')
-    end
-
     context 'instance_tags' do
       it 'returns the tags of the current instance' do
         ec2.stub_responses(:describe_tags, tag_data)

data/spec/metadata_spec.rb

@@ -0,0 +1,40 @@
+require_relative '../lib/movable_ink/aws'
+
+describe MovableInk::AWS::Metadata do
+  context 'outside ec2' do
+    it 'should raise an error if EC2 is required' do
+      aws = MovableInk::AWS.new
+      allow(aws).to receive(:retrieve_metadata).with('instance-id').and_return("")
+      allow(aws).to receive(:retrieve_metadata).with('placement/availability-zone').and_return("")
+
+      expect{ aws.instance_id }.to raise_error(MovableInk::AWS::Errors::EC2Required)
+      expect{ aws.availability_zone }.to raise_error(MovableInk::AWS::Errors::EC2Required)
+    end
+
+    it 'should raise an error if trying to load private_ipv4 outside of EC2' do
+      aws = MovableInk::AWS.new
+      allow(aws).to receive(:retrieve_metadata).with('local-ipv4').and_return('')
+      expect{ aws.private_ipv4 }.to raise_error(MovableInk::AWS::Errors::EC2Required)
+    end
+  end
+
+  context 'inside ec2' do
+    it 'calls the EC2 metadata service to get the private ipv4 address of the instance' do
+      aws = MovableInk::AWS.new
+      allow(aws).to receive(:retrieve_metadata).with('local-ipv4').and_return('10.0.0.1')
+      expect(aws.private_ipv4).to eq('10.0.0.1')
+    end
+
+    it 'calls the EC2 metadata service to get the instance ID' do
+      aws = MovableInk::AWS.new
+      expect(aws).to receive(:retrieve_metadata).with('instance-id').and_return("i-12345")
+      expect(aws.instance_id).to eq('i-12345')
+    end
+
+    it 'calls the EC2 metadata service to get the availability zone' do
+      aws = MovableInk::AWS.new
+      expect(aws).to receive(:retrieve_metadata).with('placement/availability-zone').and_return("us-east-1a")
+      expect(aws.availability_zone).to eq('us-east-1a')
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: MovableInkAWS
 version: !ruby/object:Gem::Version
-  version: 1.0.18
+  version: 1.1.0
 platform: ruby
 authors:
 - Matt Chesler
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-30 00:00:00.000000000 Z
+date: 2020-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -212,6 +212,7 @@ files:
 - lib/movable_ink/aws/ec2.rb
 - lib/movable_ink/aws/elasticache.rb
 - lib/movable_ink/aws/errors.rb
+- lib/movable_ink/aws/metadata.rb
 - lib/movable_ink/aws/route53.rb
 - lib/movable_ink/aws/s3.rb
 - lib/movable_ink/aws/sns.rb
@@ -221,6 +222,7 @@ files:
 - spec/aws_spec.rb
 - spec/ec2_spec.rb
 - spec/elasticache_spec.rb
+- spec/metadata_spec.rb
 - spec/route53_spec.rb
 - spec/s3_spec.rb
 - spec/sns_spec.rb