NTRU-1.1.0.0: examples/sampleGeneralNTRU.hs

{-
To compile: ghc sampleGeneralNTRU.hs
To run: ./sampleGeneralNTRU
-}

import Math.NTRU
import Data.Char
import Data.Maybe

main :: IO ()
main = do

  --  let params = genParams "EES401EP1"
  let params = ParamSet {getN =  401, getP = 3, getQ = 2048, getDf = 113, getDg = 133, getLLen = 1, getDb = 112, getMaxMsgLenBytes = 60, getBufferLenBits = 600, getBufferLenTrits = 400, getDm0 = 113, getShaLvl = 1, getDr = 113, getC = 11, getMinCallsR = 32, getMinCallsMask = 9, getOID = [0,2,4], getPkLen = 114, getBitLvl = 112}

  (publicKey, privateKey) <- keyGen params

  let msg = map (fromIntegral . ord) "Hello, World" :: [Integer]

  ct <- encrypt params msg publicKey

  let unencrypted = decrypt params privateKey publicKey ct

  if (isNothing unencrypted)
  then print "Encryption Failed"
  else print $ map (chr . fromIntegral) (fromJust unencrypted)