mirror of https://github.com/E-Almqvist/hsf
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
862 B
52 lines
862 B
#!/usr/bin/ruby
|
|
require_relative "rsa"
|
|
|
|
Primes = File.read("smallprimes.txt").chomp.split(",").map do |pstr|
|
|
pstr.to_i
|
|
end
|
|
|
|
def get_random_primes
|
|
puts "Fetching random primes..."
|
|
p1, p2 = Primes.sample, Primes.sample
|
|
return p1, p2
|
|
end
|
|
|
|
def choose_primes
|
|
print "1st prime: "
|
|
p1 = gets.chomp.to_i
|
|
print "2nd prime: "
|
|
p2 = gets.chomp.to_i
|
|
|
|
return p1, p2
|
|
end
|
|
|
|
puts "[1] Random primes"
|
|
puts "[2] Choose primes"
|
|
ch = gets.chomp.to_i
|
|
case ch
|
|
when 1
|
|
p1, p2 = get_random_primes
|
|
when 2
|
|
p1, p2 = choose_primes
|
|
else
|
|
p1, p2 = get_random_primes
|
|
end
|
|
|
|
puts "p1=#{p1}, p2=#{p2}"
|
|
puts
|
|
|
|
key = RSA::Key.new(p1, p2)
|
|
print "Message to be encrypted: "
|
|
msg = gets.chomp
|
|
|
|
data = RSA::Data.new(msg)
|
|
puts
|
|
|
|
data.encrypt!(key.pubkey)
|
|
puts "Encrypted bytes: #{data.raw}"
|
|
puts
|
|
|
|
data.decrypt!(key.privkey)
|
|
puts "Decrypted bytes: #{data.raw}"
|
|
puts "Decrypted msg: #{data.inspect}"
|
|
|
|
|