summaryrefslogtreecommitdiff
path: root/src/leap/gui/firstrun/regvalidation.py
diff options
context:
space:
mode:
authorkali <kali@leap.se>2012-11-09 11:21:40 +0900
committerkali <kali@leap.se>2012-11-09 11:21:40 +0900
commit8118056a244ca74d16380ad26a70e3da40e7e401 (patch)
tree4363948d180a667a545807b99b2d3199e30b5950 /src/leap/gui/firstrun/regvalidation.py
parentb84007d8fec8c949ba4ac1d26695c710a210d797 (diff)
connect page merged into regvalidation.
Flow nearly working with fake provider, except for authentication.
Diffstat (limited to 'src/leap/gui/firstrun/regvalidation.py')
-rw-r--r--src/leap/gui/firstrun/regvalidation.py186
1 files changed, 137 insertions, 49 deletions
diff --git a/src/leap/gui/firstrun/regvalidation.py b/src/leap/gui/firstrun/regvalidation.py
index 6cf150b6..f6d69a60 100644
--- a/src/leap/gui/firstrun/regvalidation.py
+++ b/src/leap/gui/firstrun/regvalidation.py
@@ -5,16 +5,16 @@ used if First Run Wizard
import logging
import json
import socket
-import time
from PyQt4 import QtGui
import requests
from leap.gui.progress import ValidationPage
+from leap.util.web import get_https_domain_and_port
from leap.base import auth
-from leap.gui.constants import APP_LOGO
+from leap.gui.constants import APP_LOGO, pause_for_user
logger = logging.getLogger(__name__)
@@ -22,6 +22,10 @@ logger = logging.getLogger(__name__)
class RegisterUserValidationPage(ValidationPage):
def __init__(self, parent=None):
+ # XXX TODO:
+ # We should check if we come from signup
+ # or login, and change title / first step
+ # accordingly.
super(RegisterUserValidationPage, self).__init__(parent)
self.setTitle("User Creation")
@@ -39,25 +43,40 @@ class RegisterUserValidationPage(ValidationPage):
we initialize the srp protocol register
and try to register user.
"""
- print 'register user checks'
-
wizard = self.wizard()
- domain = self.field('provider_domain')
- username = self.field('userName')
- password = self.field('userPassword')
+ full_domain = self.field('provider_domain')
+ domain, port = get_https_domain_and_port(full_domain)
- # XXX use pause_for_user from providerinfo
- update_signal.emit("head_sentinel", 0)
- update_signal.emit("registering with provider", 40)
- time.sleep(0.5)
- update_signal.emit("registering 2", 60)
- time.sleep(1)
- update_signal.emit("end_sentinel", 100)
- time.sleep(0.5)
- return
+ # FIXME #BUG 638 FIXME FIXME FIXME
+ verify = False # !!!!!!!!!!!!!!!!
+ # FIXME #BUG 638 FIXME FIXME FIXME
+
+ ###########################################
+ # Set Credentials.
+ # username and password are in different fields
+ # if they were stored in log_in or sign_up pages.
+
+ from_login = self.wizard().from_login
+ unamek_base = 'userName'
+ passwk_base = 'userPassword'
+ unamek = 'login_%s' % unamek_base if from_login else unamek_base
+ passwk = 'login_%s' % passwk_base if from_login else passwk_base
+
+ username = self.field(unamek)
+ password = self.field(passwk)
+ credentials = username, password
+
+ eipconfigchecker = wizard.eipconfigchecker()
+ pCertChecker = wizard.providercertchecker(
+ domain=domain)
+
+ ###########################################
+ # XXX this only should be setup
+ # if not from_login.
if wizard and wizard.debug_server:
# We're debugging
+ # XXX remove this branch?
dbgsrv = wizard.debug_server
schema = dbgsrv.scheme
netloc = dbgsrv.netloc
@@ -71,71 +90,140 @@ class RegisterUserValidationPage(ValidationPage):
signup = auth.LeapSRPRegister(
scheme=schema,
provider=provider,
- port=port)
+ port=port,
+ verify=verify)
else:
# this is the real thing
signup = auth.LeapSRPRegister(
- # XXX FIXME FIXME FIXME FIXME
- # XXX FIXME 0 Force HTTPS !!!
- # XXX FIXME FIXME FIXME FIXME
- #schema="https",
- schema="http",
- provider=domain)
+ schema="https",
+ port=port,
+ provider=domain,
+ verify=verify)
+
+ update_signal.emit("head_sentinel", 0)
+
+ ##################################################
+ # 1) register user
+ ##################################################
+ # XXX this only should be DONE
+ # if NOT from_login.
+
+ step = "register"
+ update_signal.emit("registering with provider", 40)
+ logger.debug('registering user')
+
try:
- ok, req = signup.register_user(username, password)
+ ok, req = signup.register_user(
+ username, password)
+
except socket.timeout:
- self.set_validation_status(
+ self.set_error(
+ step,
"Error connecting to provider (timeout)")
+ pause_for_user()
return False
except requests.exceptions.ConnectionError as exc:
- logger.error(exc)
- self.set_validation_status(
+ logger.error(exc.message)
+ self.set_error(
+ step,
"Error connecting to provider "
"(connection error)")
+ # XXX we should signal a BAD step
+ pause_for_user()
+ update_signal.emit("connection error!", 50)
+ pause_for_user()
return False
- if ok:
- return True
-
- # something went wrong.
- # not registered, let's catch what.
- # get timeout
- # ...
- if req.status_code == 500:
- self.set_validation_status(
- "Error during registration (500)")
- return False
+ # XXX check for != OK instead???
- if req.status_code == 404:
- self.set_validation_status(
- "Error during registration (404)")
+ if req.status_code in (404, 500):
+ self.set_error(
+ step,
+ "Error during registration (%s)" % req.status_code)
+ pause_for_user()
return False
validation_msgs = json.loads(req.content)
- logger.debug('validation errors: %s' % validation_msgs)
errors = validation_msgs.get('errors', None)
+ logger.debug('validation errors: %s' % validation_msgs)
+
if errors and errors.get('login', None):
# XXX this sometimes catch the blank username
# but we're not allowing that (soon)
- self.set_validation_status(
+ self.set_error(
+ step,
'Username not available.')
- else:
- self.set_validation_status(
- "Error during sign up")
- return False
+ pause_for_user()
+ return False
+
+ pause_for_user()
+
+ ##################################################
+ # 2) fetching eip service config
+ ##################################################
+
+ step = "fetch_eipconf"
+ fetching_eipconf_msg = "Fetching eip service configuration"
+ update_signal.emit(fetching_eipconf_msg, 60)
+ try:
+ eipconfigchecker.fetch_eip_service_config(
+ domain=full_domain)
+
+ # XXX get specific exception
+ except:
+ self.set_error(
+ step,
+ 'Could not download eip config.')
+ pause_for_user()
+ return False
+ pause_for_user()
+
+ ##################################################
+ # 3) getting client certificate
+ ##################################################
+
+ step = "fetch_eipcert"
+ fetching_clientcert_msg = "Fetching eip certificate"
+ update_signal.emit(fetching_clientcert_msg, 80)
+
+ try:
+ pCertChecker.download_new_client_cert(
+ credentials=credentials,
+ verify=verify)
+
+ except auth.SRPAuthenticationError as exc:
+ self.set_error(
+ step,
+ "Authentication error: %s" % exc.message)
+ return False
+
+ pause_for_user()
+
+ ################
+ # end !
+ ################
+
+ update_signal.emit("end_sentinel", 100)
+ pause_for_user()
def _do_validation(self):
"""
called after _do_checks has finished
(connected to checker thread finished signal)
"""
+ is_signup = self.field("is_signup")
+ prevpage = "signup" if is_signup else "login"
+
wizard = self.wizard()
if self.errors:
print 'going back with errors'
+ logger.error(self.errors)
+ name, first_error = self.pop_first_error()
wizard.set_validation_error(
- 'signup', 'that name is taken')
+ prevpage,
+ first_error)
self.go_back()
else:
print 'going next'
@@ -145,4 +233,4 @@ class RegisterUserValidationPage(ValidationPage):
wizard = self.wizard()
if not wizard:
return
- return wizard.get_page_index('connecting')
+ return wizard.get_page_index('lastpage')