kindle_strip 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Ueda Satoshi
+
+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,29 @@
+# KindleStrip
+
+Strip SRCS records from .mobi
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'kindle_strip'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install kindle_strip
+
+## Usage
+
+    $ kindle_strip input.mobi output.mobi
+
+## Contributing
+
+1. Fork it
+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 new Pull Request

data/Rakefile

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

data/bin/kindle_strip

@@ -0,0 +1,18 @@
+#!/usr/bin/env ruby
+# coding: utf-8
+$LOAD_PATH << File.expand_path("../lib", File.dirname(__FILE__))
+
+require "kindle_strip"
+
+include KindleStrip
+
+if ARGV.length == 2
+  File.open(ARGV[0], "rb") do |src|
+    buf = src.read
+    File.open(ARGV[1], "wb") do |dest|
+      dest.write(strip_srcs(buf))
+    end
+  end
+else
+  puts "Usage: #{File.basename(__FILE__)} input.mobi output.mobi"
+end

data/kindle_strip.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'kindle_strip/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "kindle_strip"
+  gem.version       = KindleStrip::VERSION
+  gem.authors       = ["Ueda Satoshi"]
+  gem.email         = ["gunyoki@gmail.com"]
+  gem.description   = %q{Strip SRCS records from .mobi}
+  gem.summary       = %q{Amazon kindlegen generates .mobi file and adds a copy of the source files. It doubles the file size. This script strips unwanted payload.}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+end

data/lib/kindle_strip/version.rb

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

data/lib/kindle_strip.rb

@@ -0,0 +1,130 @@
+require "kindle_strip/version"
+
+module KindleStrip
+  OFFSET_UNIQUE_ID_SEED = 68
+  OFFSET_NUMBER_OF_RECORDS = 76
+  OFFSET_RECORD_INFO = 78
+
+  def strip_srcs(input)
+    if input[0x3c, 8] != "BOOKMOBI"
+      raise "MobiPocket marker not found"
+    end
+    num_records = uint16_be(input, OFFSET_NUMBER_OF_RECORDS)
+    record0 = get_record0(input)
+    srcs_start = uint32_be(record0, 0xe0)
+    srcs_count = uint32_be(record0, 0xe4)
+    if srcs_start == 0xffffffff || srcs_count == 0
+      raise "File doesn't contain SRCS"
+    end
+    srcs_offset = uint32_be(input, OFFSET_RECORD_INFO + srcs_start * 8)
+    srcs_length = uint32_be(input, OFFSET_RECORD_INFO + (srcs_start + srcs_count) * 8) - srcs_offset
+    if input[srcs_offset, 4] != "SRCS"
+      raise "SRCS section num does not point to SRCS"
+    end
+
+    output = new_header(input, num_records - srcs_count)
+    output += new_record_infos(input, num_records, srcs_start, srcs_count, srcs_length)
+    output += make_padding(output)
+    output += new_records(input, srcs_offset, srcs_length)
+    fix_record0(record0, srcs_start, srcs_count)
+    set_record0(output, record0)
+    output
+  end
+
+  def get_record0(buf)
+    record0_offset = uint32_be(buf, OFFSET_RECORD_INFO)
+    record1_offset = uint32_be(buf, OFFSET_RECORD_INFO + 8)
+    buf[record0_offset ... record1_offset]
+  end
+  private :get_record0
+
+  def set_record0(buf, record0)
+    record0_offset = uint32_be(buf, OFFSET_RECORD_INFO)
+    record1_offset = uint32_be(buf, OFFSET_RECORD_INFO + 8)
+    buf[record0_offset ... record1_offset] = record0
+  end
+  private :set_record0
+
+  def new_header(input, num_records)
+    header = input[0 ... OFFSET_RECORD_INFO]
+    header[OFFSET_UNIQUE_ID_SEED, 4] = [num_records * 2 + 1].pack("N")
+    header[OFFSET_NUMBER_OF_RECORDS, 2] = [num_records].pack("n")
+    header
+  end
+  private :new_header
+
+  def new_record_infos(input, num_records, srcs_start, srcs_count, srcs_length)
+    srcs_end = srcs_start + srcs_count
+    record_infos = ""
+    num_records.times do |i|
+      next if (srcs_start ... srcs_end) === i
+      offset = uint32_be(input, OFFSET_RECORD_INFO + i * 8)
+      attr_and_id = uint32_be(input, OFFSET_RECORD_INFO + i * 8 + 4)
+      offset -= srcs_count * 8
+      if i >= srcs_end
+        offset -= srcs_length
+        attr_and_id = (i - srcs_count) * 2 | (attr_and_id & 0xff000000)
+      end
+      record_infos += [offset, attr_and_id].pack("NN")
+    end
+    record_infos
+  end
+  private :new_record_infos
+
+  def make_padding(output)
+    record_offset = uint32_be(output, OFFSET_RECORD_INFO)
+    [].pack("x#{record_offset - output.length}")  # padding by NUL
+  end
+  private :make_padding
+
+  def new_records(input, srcs_offset, srcs_length)
+    record_offset = uint32_be(input, OFFSET_RECORD_INFO)
+    records = input[record_offset .. -1]
+    records[srcs_offset - record_offset, srcs_length] = ""  # remove SRCS
+    records
+  end
+  private :new_records
+
+  def fix_record0(record0, srcs_start, srcs_count)
+    record0[0xe0, 8] = [0xffffffff, 0].pack("NN")  # set no SRCS
+    fix_exth(record0, srcs_start, srcs_count)
+  end
+  private :fix_record0
+
+  def fix_exth(record0, srcs_start, srcs_count)
+    unless uint32_be(record0, 0x80) & 0x40
+      # no EXTH header
+      return
+    end
+    exth_offset = 16 + uint32_be(record0, 0x14)
+    exth = record0[exth_offset .. -1]
+    if exth[0, 4] != "EXTH"
+      # marker not found
+      return
+    end
+    num_records = uint32_be(exth, 8)
+    pos = 12
+    num_records.times do |i|
+      type = uint32_be(exth, pos)
+      size = uint32_be(exth, pos + 4)
+      if type == 121
+        boundary = uint32_be(exth, pos + 8)
+        if srcs_start <= boundary
+          record0[exth_offset + pos + 8, 4] = [boundary - srcs_count].pack("N")
+        end
+      end
+      pos += size
+    end
+  end
+  private :fix_exth
+
+  def uint16_be(buf, offset = 0)
+    buf.unpack("@#{offset}n").first
+  end
+  private :uint16_be
+
+  def uint32_be(buf, offset = 0)
+    buf.unpack("@#{offset}N").first
+  end
+  private :uint32_be
+end

metadata

@@ -0,0 +1,56 @@
+--- !ruby/object:Gem::Specification
+name: kindle_strip
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Ueda Satoshi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-02 00:00:00.000000000 Z
+dependencies: []
+description: Strip SRCS records from .mobi
+email:
+- gunyoki@gmail.com
+executables:
+- kindle_strip
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/kindle_strip
+- kindle_strip.gemspec
+- lib/kindle_strip.rb
+- lib/kindle_strip/version.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Amazon kindlegen generates .mobi file and adds a copy of the source files.
+  It doubles the file size. This script strips unwanted payload.
+test_files: []