Ruby SSL Server

openssl.sh

#!/bin/sh
#
# CA - wrapper around ca to make it easier to use ... basically ca requires
#      some setup stuff to be done before you can use it and this makes
#      things easier
#
# CA -newca ... will setup the right stuff
# CA -newreq ... will generate a certificate request
# CA -sign ... will sign the generated request and output
#
# At the end of that grab newreq.pem and newcert.pem (one has the key
# and the other the certificate) and cat them together and that is what
# you want/need ... I'll make even this a little cleaner later.
#


# default openssl.cnf file has setup as per the following
# demoCA ... where everything is stored
cp_pem() {
    infile=$1
    outfile=$2
    bound=$3
    flag=0
    exec <$infile;
    while read line; do
  if [ $flag -eq 1 ]; then
    echo $line|grep "^-----END.*$bound"  2>/dev/null 1>/dev/null
    if [ $? -eq 0 ] ; then
      echo $line >>$outfile
      break
    else
      echo $line >>$outfile
    fi
  fi

  echo $line|grep "^-----BEGIN.*$bound"  2>/dev/null 1>/dev/null
  if [ $? -eq 0 ]; then
    echo $line >$outfile
    flag=1
  fi
    done
}

usage() {
 echo "usage: $0 -newcert|-newreq|-newreq-nodes|-newca|-sign|-verify" >&2
}

if [ -z "$OPENSSL" ]; then OPENSSL=openssl; fi

if [ -z "$DAYS" ] ; then DAYS="-days 365" ; fi  # 1 year
CADAYS="-days 1095" # 3 years
REQ="$OPENSSL req $SSLEAY_CONFIG"
CA="$OPENSSL ca $SSLEAY_CONFIG"
VERIFY="$OPENSSL verify"
X509="$OPENSSL x509"
PKCS12="openssl pkcs12"

if [ -z "$CATOP" ] ; then CATOP=./demoCA ; fi
CAKEY=./cakey.pem
CAREQ=./careq.pem
CACERT=./cacert.pem

RET=0

while [ "$1" != "" ] ; do
case $1 in
-\?|-h|-help)
    usage
    exit 0
    ;;
-newcert)
    # create a certificate
    $REQ -new -x509 -keyout newkey.pem -out newcert.pem $DAYS
    RET=$?
    echo "Certificate is in newcert.pem, private key is in newkey.pem"
    ;;
-newreq)
    # create a certificate request
    $REQ -new -keyout newkey.pem -out newreq.pem $DAYS
    RET=$?
    echo "Request is in newreq.pem, private key is in newkey.pem"
    ;;
-newreq-nodes)
    # create a certificate request
    $REQ -new -nodes -keyout newreq.pem -out newreq.pem $DAYS
    RET=$?
    echo "Request (and private key) is in newreq.pem"
    ;;
-newca)
    # if explicitly asked for or it doesn't exist then setup the directory
    # structure that Eric likes to manage things
    NEW="1"
    if [ "$NEW" -o ! -f ${CATOP}/serial ]; then
  # create the directory hierarchy
  mkdir -p ${CATOP}
  mkdir -p ${CATOP}/certs
  mkdir -p ${CATOP}/crl
  mkdir -p ${CATOP}/newcerts
  mkdir -p ${CATOP}/private
  touch ${CATOP}/index.txt
    fi
    if [ ! -f ${CATOP}/private/$CAKEY ]; then
  echo "CA certificate filename (or enter to create)"
  read FILE

  # ask user for existing CA certificate
  if [ "$FILE" ]; then
      cp_pem $FILE ${CATOP}/private/$CAKEY PRIVATE
      cp_pem $FILE ${CATOP}/$CACERT CERTIFICATE
      RET=$?
      if [ ! -f "${CATOP}/serial" ]; then
    $X509 -in ${CATOP}/$CACERT -noout -next_serial \
          -out ${CATOP}/serial
      fi
  else
      echo "Making CA certificate ..."
      $REQ -new -keyout ${CATOP}/private/$CAKEY \
         -out ${CATOP}/$CAREQ
      $CA -create_serial -out ${CATOP}/$CACERT $CADAYS -batch \
         -keyfile ${CATOP}/private/$CAKEY -selfsign \
         -extensions v3_ca \
         -infiles ${CATOP}/$CAREQ
      RET=$?
  fi
    fi
    ;;
-xsign)
    $CA -policy policy_anything -infiles newreq.pem
    RET=$?
    ;;
-pkcs12)
    if [ -z "$2" ] ; then
  CNAME="My Certificate"
    else
  CNAME="$2"
    fi
    $PKCS12 -in newcert.pem -inkey newreq.pem -certfile ${CATOP}/$CACERT \
      -out newcert.p12 -export -name "$CNAME"
    RET=$?
    exit $RET
    ;;
-sign|-signreq)
    $CA -policy policy_anything -out newcert.pem -infiles newreq.pem
    RET=$?
    cat newcert.pem
    echo "Signed certificate is in newcert.pem"
    ;;
-signCA)
    $CA -policy policy_anything -out newcert.pem -extensions v3_ca -infiles newreq.pem
    RET=$?
    echo "Signed CA certificate is in newcert.pem"
    ;;
-signcert)
    echo "Cert passphrase will be requested twice - bug?"
    $X509 -x509toreq -in newreq.pem -signkey newreq.pem -out tmp.pem
    $CA -policy policy_anything -out newcert.pem -infiles tmp.pem
    RET=$?
    cat newcert.pem
    echo "Signed certificate is in newcert.pem"
    ;;
-verify)
    shift
    if [ -z "$1" ]; then
      $VERIFY -CAfile $CATOP/$CACERT newcert.pem
      RET=$?
    else
  for j
  do
      $VERIFY -CAfile $CATOP/$CACERT $j
      if [ $? != 0 ]; then
        RET=$?
      fi
  done
    fi
    exit $RET
    ;;
*)
    echo "Unknown arg $i" >&2
    usage
    exit 1
    ;;
esac
shift
done
exit $RET

ssl_client.rb

require 'socket'
require 'openssl'
require 'thread'

unless ARFV.length == 2
  puts "Usage: ruby #{__FILE__} port ssl_cert_file"
  exit 1
end

client  = TCPSocket.new '127.0.0.1', ARGV[0]
context = OpenSSL::SSL::SSLContext.new
context.verify_mode = OpenSSL::SSL::VERIFY_PEER | OpenSSL::SSL::VERIFY_FAIL_IF_NO_PEER_CERT
context.ca_file = ARGV[1]

secure = OpenSSL::SSL::SSLSocket.new(client, context)
secure.sync_close = true
secure.connect

Thread.new do
  begin
    while response = secure.gets
      $stdout.puts response
    end
  rescue
    $stderr.puts "Error from client: #{$!}"
  end
end

while request = $stdin.gets
  request = request.chomp
  secure.puts request
end

ssl_server.rb

require 'socket'
require 'openssl'
require 'thread'

if ARGV.length < 3
  puts "Usage: ruby #{__FILE__} port ssl_cert_path ssl_key_path"
  exit 1
end

server  = TCPServer.new ARGV[0]
context = OpenSSL::SSL::SSLContext.new

context.cert = OpenSSL::X509::Certificate.new(File.open(ARGV[1]))
context.key  = OpenSSL::PKey::RSA.new(File.open(ARGV[2]), ARGV[3])

secure = OpenSSL::SSL::SSLServer.new(server, context)

puts "Listening securely on port #{ARGV[0]}..."

loop do
  Thread.new(secure.accept) do |conn|
    begin
      while request = conn.gets
        $stdout.puts '=> ' + request
        response = "You said: #{request}"
        $stdout.puts '<= ' + response
        conn.puts response
      end
    rescue
      $stderr.puts $!
    end
  end
end

Usage.md

Sample TCP server-client implementation using openssl in ruby

Use the wrapper shell script to create ssl certificates ./openssl.sh -newca, enter the details and the required files will be generated to ./demoCA dir from where you run the script. cacert.pem will be your ssl-certificate and private/cakey.pem will be your ssl-key

Start SSL Server using:

ruby ssl_server.rb 9099 demoCA/cacert.pem demoCA/private/cakey.pem

Start SSL client using:

ruby ssl_client.rb 9099 demoCA/cacert.pem

https://gist.github.com/ashrithr/5820320#file-ssl_client-rb-L5 It is ARGV.