video_converter 0.7.8 → 0.7.9

data/lib/video_converter.rb

@@ -14,6 +14,7 @@ require "video_converter/mp4frag"
 require "video_converter/object"
 require "video_converter/output"
 require "video_converter/version"
+require "video_converter/openssl"
 require "video_screenshoter"
 
 module VideoConverter

data/lib/video_converter/base.rb

@@ -58,10 +58,14 @@ module VideoConverter
     end
 
     def encrypt(options = {})
-      outputs.each do |output| 
-        if output.drm == 'adobe'
+      outputs.each do |output|
+        case output.drm
+        when 'adobe'
           output.options.merge!(options)
           F4fpackager.run(output) or return false
+        when 'hls'
+          output.options.merge!(options)
+          OpenSSL.run(output) or return false
         end
       end
       true

data/lib/video_converter/live_segmenter.rb

@@ -58,7 +58,9 @@ module VideoConverter
         res = File.join(File.basename(output.chunks_dir), File.basename(chunk)) + "\n" + res
         res = "#EXTINF:%0.2f,\n" % duration + res
       end
-      res = "#EXTM3U\n#EXT-X-VERSION:3\n#EXT-X-TARGETDURATION:#{durations.max.to_f.ceil}\n#EXT-X-MEDIA-SEQUENCE:0\n" + res + "#EXT-X-ENDLIST"
+      encryption_info = OpenSSL.get_encryption_key_path(output)
+      encryption_info = "#EXT-X-KEY:METHOD=AES-128,URI=\"#{encryption_info}\"\n" if encryption_info
+      res = "#EXTM3U\n#{encryption_info}#EXT-X-VERSION:3\n#EXT-X-TARGETDURATION:#{durations.max.to_f.ceil}\n#EXT-X-MEDIA-SEQUENCE:0\n" + res + "#EXT-X-ENDLIST"
       !!File.open(File.join(output.work_dir, output.filename), 'w') { |f| f.write res }
     end
 

data/lib/video_converter/openssl.rb

@@ -0,0 +1,70 @@
+module VideoConverter
+  class OpenSSL
+    class << self
+      attr_accessor :bin, :command, :encryption_type
+    end
+
+    #TODO locate openssl
+    self.bin = 'openssl'
+    self.encryption_type = 'aes-128-cbc'
+    self.command = '%{bin} %{encryption_type} -e -in %{chunk} -out %{encrypted_chunk} -nosalt -iv %{initialization_vector} -K %{encryption_key}'
+
+    def self.run(output)
+      success = true
+      encryption_dir = FileUtils.mkdir_p("#{output.chunks_dir}_encrypted").first
+      chunks = Dir::glob(File.join(output.chunks_dir, "#{LiveSegmenter.chunk_prefix}-*[0-9].ts")).sort do |c1, c2|
+        File.basename(c2).match(/\d+/).to_s.to_i <=> File.basename(c1).match(/\d+/).to_s.to_i
+      end
+      chunks.each_with_index do |chunk, index|
+        initialization_vector = "%032x" % index
+        success &&= Command.new(command, prepare_params( :chunk => chunk, :encrypted_chunk => File.join(encryption_dir,File.basename(chunk)),
+                                                         :initialization_vector => initialization_vector, :encryption_key => OpenSSL.get_encryption_key(output) )).execute
+      end
+      remove_old_dir(output) if success
+      success
+    end
+
+    def self.get_encryption_key_path(output)
+      if output.encryption_key
+        File.open(File.join(output.work_dir,'video.key'), 'wb') { |f| f.puts output.encryption_key }
+        'video.key'
+      elsif output.encryption_key_url
+        uri = URI(output.encryption_key_url)
+        File.open(File.join(output.work_dir,'url.video.key'), 'wb') { |f| f.puts Net::HTTP.get(uri.host, uri.path) }
+        output.encryption_key_url
+      else
+        nil
+      end
+    end
+
+    def self.get_encryption_key(output)
+      if output.encryption_key
+        File.join(output.work_dir, "video.key")
+      elsif output.encryption_key_url
+        File.join(output.work_dir, "url.video.key")
+      else
+        nil
+      end
+    end
+
+    private
+
+    def self.prepare_params(params)
+      {
+        :bin => bin,
+        :encryption_type => encryption_type,
+        :chunk => params[:chunk],
+        :encrypted_chunk => params[:encrypted_chunk],
+        :initialization_vector => params[:initialization_vector],
+        :encryption_key => `hexdump -e '16/1 "%02x"' #{params[:encryption_key]}`
+      }
+    end
+
+    def self.remove_old_dir(output)
+      FileUtils.rm_r(output.chunks_dir)
+      FileUtils.mv("#{output.chunks_dir}_encrypted", output.chunks_dir)
+      FileUtils.rm (File.join(output.work_dir,'url.video.key')) if File.exists?(File.join(output.work_dir,'url.video.key'))
+    end
+
+  end
+end

data/lib/video_converter/output.rb

@@ -9,7 +9,7 @@ module VideoConverter
     self.log = 'converter.log'
     self.keyframe_interval_in_seconds = 4
 
-    attr_accessor :chunks_dir, :crop, :drm, :faststart, :ffmpeg_output, :filename, :group, :height, :log, :mkdir_mode, :no_fragments, :one_pass, :options, :path, :rotate, :streams, :thumbnails, :type, :uid, :volume, :watermarks, :width, :work_dir
+    attr_accessor :chunks_dir, :crop, :drm, :faststart, :ffmpeg_output, :filename, :group, :height, :log, :mkdir_mode, :no_fragments, :one_pass, :options, :path, :rotate, :streams, :thumbnails, :type, :uid, :volume, :watermarks, :width, :work_dir, :encryption_key, :encryption_key_url
 
     def initialize params = {}
       self.work_dir = File.join(self.class.work_dir, params[:uid])
@@ -35,7 +35,7 @@ module VideoConverter
         self.ffmpeg_output = File.join(work_dir, filename)
       end
       raise ArgumentError.new('Invalid type') unless %w(default segmented playlist).include?(type)
-      [:path, :streams, :width, :height, :one_pass, :rotate, :faststart, :thumbnails, :group, :drm, :no_fragments, :crop, :watermarks, :volume].each { |attr| self.send("#{attr}=", params[attr]) }
+      [:path, :streams, :width, :height, :one_pass, :rotate, :faststart, :thumbnails, :group, :drm, :no_fragments, :crop, :watermarks, :volume, :encryption_key, :encryption_key_url].each { |attr| self.send("#{attr}=", params[attr]) }
       [:video_bitrate, :audio_bitrate].each { |bitrate| params[bitrate] = "#{params[bitrate]}k" if params[bitrate].is_a?(Numeric) }
 
       # options will be substituted to convertation commands

data/lib/video_converter/version.rb

@@ -1,3 +1,3 @@
 module VideoConverter
-  VERSION = "0.7.8"
+  VERSION = "0.7.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: video_converter
 version: !ruby/object:Gem::Version
-  version: 0.7.8
+  version: 0.7.9
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-19 00:00:00.000000000 Z
+date: 2014-10-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: video_screenshoter
@@ -117,6 +117,7 @@ files:
 - lib/video_converter/live_segmenter.rb
 - lib/video_converter/mp4frag.rb
 - lib/video_converter/object.rb
+- lib/video_converter/openssl.rb
 - lib/video_converter/output.rb
 - lib/video_converter/version.rb
 - test/fixtures/chunk0.ts
@@ -141,7 +142,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2912131658363711804
+      hash: 646644403417642151
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -150,7 +151,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2912131658363711804
+      hash: 646644403417642151
 requirements:
 - ffmpeg, version 1.2 or greated configured with libx264 and libfaac
 - live_segmenter to convert to hls