kindle-drm 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/.project

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<projectDescription>
+	<name>kindle-drm</name>
+	<comment></comment>
+	<projects>
+	</projects>
+	<buildSpec>
+		<buildCommand>
+			<name>org.rubypeople.rdt.core.rubybuilder</name>
+			<arguments>
+			</arguments>
+		</buildCommand>
+	</buildSpec>
+	<natures>
+		<nature>org.rubypeople.rdt.core.rubynature</nature>
+	</natures>
+</projectDescription>

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Preston Lee
+
+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.rdoc

@@ -0,0 +1,26 @@
+= kindle-drm
+
+This code can calculate the PID (Personal ID) number of a Kindle eBook reader device based on it's serial number. To determine the serial number of your Kindle:
+
+1. Navigate to the device's "Settitngs" screen.
+1. Type "411" (without the quotes) on the keyboard.
+1. A secret page should now open. Write down the 16-character serial number of the device. (iPhone serial numbers are different but should also work.)
+
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+= History
+
+This code was originally ported Mobipocket PID calculator v0.2 by Igor Skochinsky. I have added support for Kindle 2 and Kindle DX (serial numbers starting with B003 and B004) in addition to the port.
+
+== Copyright
+
+Copyright (c) 2010 Preston Lee. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,53 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "kindle-drm"
+    gem.summary = %Q{Amazon Kindle .azw file manipulation library.}
+    gem.description = %Q{A Ruby library for performing DRM-related operations on .mobi files, know as DRM'd .azw files on the Amazon Kindle.}
+    gem.email = "conmotto@gmail.com"
+    gem.homepage = "http://github.com/preston/kindle-drm"
+    gem.authors = ["Preston Lee", "Igor Skochinsky"]
+    gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "kindle-drm #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/serial_to_pid.rb

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+
+current_dir = File.dirname(File.expand_path(__FILE__))
+lib_path = File.join(current_dir, '..', 'lib')
+$LOAD_PATH.unshift lib_path
+
+
+require 'zlib'
+require 'kindle-drm'
+
+
+include Kindle::DRM
+
+
+code = 0
+puts "\n"
+puts "Mobipocket PID calculator for Amazon Kindle. Refactored, ported and packaged for Ruby by Preston Lee. Original Python code by Igor Skochinsky."
+if ARGV.length == 1
+	serial = ARGV[0]
+	length = serial.length
+	type, size, pad = serialToDeviceTypeAndPidSize(serial)
+	if type.nil?
+	  puts "Warning: unrecognized serial number. Please recheck input."
+	  code = 1
+	else
+		puts "\n"
+		puts "Device Type:\t#{type}"
+		puts "PID:\t\t" + serialToPid(serial)
+	end
+else
+  puts "Usage: #{__FILE__} <Kindle Serial Number>/<iPhone/iPod Touch UDID>"
+end
+puts "\n"
+
+exit(code)
+

data/kindle-drm.gemspec

@@ -0,0 +1,59 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{kindle-drm}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Preston Lee", "Igor Skochinsky"]
+  s.date = %q{2010-02-20}
+  s.default_executable = %q{serial_to_pid.rb}
+  s.description = %q{A Ruby library for performing DRM-related operations on .mobi files, know as DRM'd .azw files on the Amazon Kindle.}
+  s.email = %q{conmotto@gmail.com}
+  s.executables = ["serial_to_pid.rb"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     ".project",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/serial_to_pid.rb",
+     "kindle-drm.gemspec",
+     "lib/kindle-drm.rb",
+     "lib/kindle/drm.rb",
+     "test/helper.rb",
+     "test/test_kindle-drm.rb"
+  ]
+  s.homepage = %q{http://github.com/preston/kindle-drm}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{Amazon Kindle .azw file manipulation library.}
+  s.test_files = [
+    "test/helper.rb",
+     "test/test_kindle-drm.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+  end
+end
+

data/lib/kindle-drm.rb

@@ -0,0 +1 @@
+require 'kindle/drm'

data/lib/kindle/drm.rb

@@ -0,0 +1,102 @@
+require 'zlib'
+
+module Kindle
+	
+	module DRM
+		
+		LETTERS = "ABCDEFGHIJKLMNPQRSTUVWXYZ123456789".split(//)
+
+		# Special cyclical redundancy check (checksumming) function.
+		def crc32(s)
+			(~Zlib.crc32(s, -1)) & 0xFFFFFFFF
+		end
+
+		# Generates the actual Personal ID (PID) needed to tie/untie content
+		# (such as .mobi/.azw files) to a specific device.		
+		def checksumPid(s)
+			tmp = crc32(s)
+			crc = tmp ^ (tmp >> 16)
+			res = s
+			l = LETTERS.length
+			for i in 0..1 do
+				b = crc & 0xFF
+				t = b.divmod(l)[0]
+				pos = t ^ (b % l)
+				res = "#{res}#{LETTERS[pos % l]}"
+				crc >>= 8
+			end
+			res
+		end
+
+		# Figures out the intermediary Personal ID (PID) of a device based on its
+		# serial number and expected length of the output. Apparently this
+		# varies by the type of device.
+		def serialToIntermediaryPid(s, l)
+			crc = crc32(s)  
+			arr1 = Array.new(l, 0)
+			for i in 0..(s.length - 1) do
+				code = s[i]
+				code = code.ord if code.class == String # Fix for Ruby 1.9+
+				arr1[i%l] ^= code
+			end
+
+			# Grab each CRC byte and OR with a portion of the 
+			crc_bytes = [crc >> 24 & 0xff, crc >> 16 & 0xff, crc >> 8 & 0xff, crc & 0xff]
+			for i in 0..(l - 1) do
+				arr1[i] ^= crc_bytes[i&3]
+			end
+
+			pid = ""
+			for i in 0..(l-1) do
+				b = arr1[i] & 0xff
+				pid += LETTERS[(b >> 7) + ((b >> 5 & 3) ^ (b & 0x1f))]
+			end
+			pid
+		end
+		
+		
+		def serialToDeviceTypeAndPidSize(serial)
+			type = nil
+			size = nil
+			pad = nil
+			
+			l = serial.length
+			if l == 16
+				case serial[0, 4]
+				when "B001"
+					type = "Kindle 1"
+				when "B002"
+					type = "Kindle 2"
+				when "B003"
+					type = "Kindle 2 International"
+				when "B004"
+					type = "Kindle DX"
+				end
+				size = 7
+				pad = '*'
+			elsif l == 40
+				type = "iPhone"
+				size = 8
+			end
+
+			return nil if type.nil? || size.nil?
+			return [type, size, pad]
+		end
+
+		
+		def serialToPid(serial)
+			pid = nil
+			type, size, pad = serialToDeviceTypeAndPidSize(serial)
+			if type.nil?
+				# No dice.. :(
+			else
+				ipid = serialToIntermediaryPid(serial, size)
+				ipid += pad unless pad.nil?
+				pid = checksumPid(ipid)
+			end
+			pid
+		end
+		
+	end
+	
+end

data/test/helper.rb

@@ -0,0 +1,12 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'kindle-drm'
+
+include Kindle::DRM
+
+class Test::Unit::TestCase
+end

data/test/test_kindle-drm.rb

@@ -0,0 +1,14 @@
+require 'helper'
+
+class TestKindleDrm < Test::Unit::TestCase
+
+  should "generate Personal ID (PID) strings correctly" do	
+	# Valid test output generated by original Python code.
+	assert_equal crc32('aoeuaoeuaoeuaoeu'), 593220825
+	assert_equal serialToIntermediaryPid('aoeuaoeuaoeuaoeu', 7), "RB5MXY5"
+	assert_equal checksumPid('aoeuaoeuaoeuaoeu'), "aoeuaoeuaoeuaoeu7P"
+	assert_equal checksumPid('RB5MXY5'), "RB5MXY5LR"
+	assert_equal serialToPid('B001DEADBABECAFE'), 'RB6IUSR*MU'
+  end
+
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification 
+name: kindle-drm
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Preston Lee
+- Igor Skochinsky
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-20 00:00:00 -07:00
+default_executable: serial_to_pid.rb
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: A Ruby library for performing DRM-related operations on .mobi files, know as DRM'd .azw files on the Amazon Kindle.
+email: conmotto@gmail.com
+executables: 
+- serial_to_pid.rb
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- .project
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/serial_to_pid.rb
+- kindle-drm.gemspec
+- lib/kindle-drm.rb
+- lib/kindle/drm.rb
+- test/helper.rb
+- test/test_kindle-drm.rb
+has_rdoc: true
+homepage: http://github.com/preston/kindle-drm
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Amazon Kindle .azw file manipulation library.
+test_files: 
+- test/helper.rb
+- test/test_kindle-drm.rb