summaryrefslogtreecommitdiff
path: root/users/test/integration/api/python/flow_with_srp.py
blob: ea630f25101e5bc6942c2d4bc77ac92b32253bec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/usr/bin/env python

# under development

import requests
import json
import string
import random
import srp
import binascii

# let's have some random name
def id_generator(size=6, chars=string.ascii_uppercase + string.digits):
  return ''.join(random.choice(chars) for x in range(size))

# using globals for a start
server = 'http://localhost:3000'
login = id_generator()
password = id_generator() + id_generator()

# log the server communication
def print_and_parse(response):
  print response.request.method + ': ' + response.url
  print "    " + json.dumps(response.request.data)
  print " -> " + response.text
  return json.loads(response.text)

def signup(session):
  salt, vkey = srp.create_salted_verification_key( login, password, srp.SHA256, srp.NG_1024 )
  user_params = {
      'user[login]': login,
      'user[password_verifier]': binascii.hexlify(vkey),
      'user[password_salt]': binascii.hexlify(salt)
      }
  return session.post(server + '/users.json', data = user_params)

usr = srp.User( login, password, srp.SHA256, srp.NG_1024 )

def authenticate(session, login):
  uname, A = usr.start_authentication()
  params = {
      'login': uname,
      'A': binascii.hexlify(A)
      }
  init = print_and_parse(session.post(server + '/sessions', data = params))
  M = usr.process_challenge( binascii.unhexlify(init['salt']), binascii.unhexlify(init['B']) )
  return session.put(server + '/sessions/' + login, 
      data = {'client_auth': binascii.hexlify(M)})

session = requests.session()
user = print_and_parse(signup(session))

# SRP signup would happen here and calculate M hex
auth = print_and_parse(authenticate(session, user['login']))
usr.verify_session( binascii.unhexlify(auth["M2"]) )

# At this point the authentication process is complete.
assert usr.authenticated()