root/python/TLSLite/dh/dh.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# 2006/06/30 --nakiki

# System
import tlslite.utils.cryptomath as cryptomath

# DHを利用した秘密鍵の安全な共有の仕方
if __name__ == '__main__':
  g = 5 # generator
  # pは128bitの素数を作るように指示している。
  p = cryptomath.getRandomPrime(128) # prime modulas 
  x = 0x640F7967334E9872 # xはAさんの秘密鍵（伝送路では流さない） 適当な番号
  y = 0x40F6BD5315A291DC # yはBさんの秘密鍵 (伝送路ではながさない) 適当な番号
  print 'g=%x' % g
  print 'p=%x' % p
  print 'x=%x' % x
  print 'y=%x' % y

  # Aさんはg^x mod pを計算する
  X = cryptomath.powMod(g, x, p)

  # Bさんはg^y mod pを計算する
  Y = cryptomath.powMod(g, y, p)

  # AさんはXをBさんに渡す
  print 'g^x mod p =%x' % X

  # BさんはYをAさんに渡す。
  print 'g^y mod p =%x' % Y

  secretA = cryptomath.powMod(X, y, p)
  secretB = cryptomath.powMod(Y, x, p)

  print 'secretA= (g^x mod p)^y =%x' % secretA
  print 'secretB=(g^y mod p)^x =%x' % secretB
  if secretA == secretB:
    print 'secretA==secretB ok'
  else:
    print 'secretA!=secretB ng'