""" Select Provider Page, used in First Run Wizard """ from PyQt4 import QtCore from PyQt4 import QtGui from leap.base import exceptions as baseexceptions from leap.crypto import certs from leap.eip import exceptions as eipexceptions from leap.gui.constants import APP_LOGO from leap.gui.styles import ErrorLabelStyleSheet class SelectProviderPage(QtGui.QWizardPage): def __init__(self, parent=None, providers=None): super(SelectProviderPage, self).__init__(parent) self.setTitle("Enter Provider") self.setSubTitle( "Please enter the domain of the provider you want " "to use for your connection." ) self.setPixmap( QtGui.QWizard.LogoPixmap, QtGui.QPixmap(APP_LOGO)) self.did_cert_check = False providerNameLabel = QtGui.QLabel("h&ttps://") # note that we expect the bare domain name # we will add the scheme later providerNameEdit = QtGui.QLineEdit() providerNameEdit.cursorPositionChanged.connect( self.reset_validation_status) providerNameLabel.setBuddy(providerNameEdit) # add regex validator providerDomainRe = QtCore.QRegExp(r"^[a-z\d_-.]+$") providerNameEdit.setValidator( QtGui.QRegExpValidator(providerDomainRe, self)) self.providerNameEdit = providerNameEdit # Eventually we will seed a list of # well known providers here. #providercombo = QtGui.QComboBox() #if providers: #for provider in providers: #providercombo.addItem(provider) #providerNameSelect = providercombo self.registerField('provider_domain*', self.providerNameEdit) #self.registerField('provider_name_index', providerNameSelect) validationMsg = QtGui.QLabel("") validationMsg.setStyleSheet(ErrorLabelStyleSheet) self.validationMsg = validationMsg # cert info # this is used in the callback # for the checkbox changes. # tricky, since the first time came # from the exception message. # should get string from exception too! self.bad_cert_status = "Server certificate could not be verified." self.certInfo = QtGui.QLabel("") self.certInfo.setWordWrap(True) self.certWarning = QtGui.QLabel("") self.trustProviderCertCheckBox = QtGui.QCheckBox( "&Trust this provider certificate.") self.trustProviderCertCheckBox.stateChanged.connect( self.onTrustCheckChanged) layout = QtGui.QGridLayout() layout.addWidget(validationMsg, 0, 2) layout.addWidget(providerNameLabel, 1, 1) layout.addWidget(providerNameEdit, 1, 2) # XXX get a groupbox or something.... certinfoGroup = QtGui.QGroupBox("Certificate validation") certinfoLayout = QtGui.QVBoxLayout() certinfoLayout.addWidget(self.certInfo) certinfoLayout.addWidget(self.certWarning) certinfoLayout.addWidget(self.trustProviderCertCheckBox) certinfoGroup.setLayout(certinfoLayout) layout.addWidget(certinfoGroup, 4, 1, 4, 2) self.certinfoGroup = certinfoGroup self.certinfoGroup.hide() self.setLayout(layout) def is_insecure_cert_trusted(self): return self.trustProviderCertCheckBox.isChecked() def onTrustCheckChanged(self, state): checked = False if state == 2: checked = True if checked: self.reset_validation_status() else: self.set_validation_status(self.bad_cert_status) # trigger signal to redraw next button self.completeChanged.emit() def reset_validation_status(self): """ empty the validation msg """ self.validationMsg.setText('') def set_validation_status(self, status): self.validationMsg.setText(status) def add_cert_info(self, certinfo): self.certWarning.setText( "Do you want to trust this provider certificate?") self.certInfo.setText( 'SHA-256 fingerprint: %s
' % certinfo) self.certInfo.setWordWrap(True) self.certinfoGroup.show() # pagewizard methods def isComplete(self): if not self.did_cert_check: return True if self.is_insecure_cert_trusted(): return True return False def initializePage(self): self.certinfoGroup.hide() def validatePage(self): ################################## # XXX FIXME! ################################## ################################## ################################## ################################## ##### validation skipped !!! ##### ################################## ################################## return True ################################## ################################## ################################## # XXX move to ProviderInfo... wizard = self.wizard() netchecker = wizard.netchecker() providercertchecker = wizard.providercertchecker() eipconfigchecker = wizard.eipconfigchecker() domain = self.providerNameEdit.text() # try name resolution try: netchecker.check_name_resolution( domain) except baseexceptions.LeapException as exc: self.set_validation_status(exc.usermessage) return False # try https connection try: providercertchecker.is_https_working( "https://%s" % domain, verify=True) except eipexceptions.HttpsBadCertError as exc: if self.trustProviderCertCheckBox.isChecked(): pass else: self.set_validation_status(exc.usermessage) fingerprint = certs.get_cert_fingerprint( domain=domain, sep=" ") # it's ok if we've trusted this fgprt before trustedcrts = self.wizard().trusted_certs if trustedcrts and fingerprint.replace(' ', '') in trustedcrts: pass else: # let your user face panick :P self.add_cert_info(fingerprint) self.did_cert_check = True self.completeChanged.emit() return False except baseexceptions.LeapException as exc: self.set_validation_status(exc.usermessage) return False # try download provider info... eipconfigchecker.fetch_definition(domain=domain) wizard.set_providerconfig( eipconfigchecker.defaultprovider.config) # all ok, go on... return True def nextId(self): wizard = self.wizard() if not wizard: return return wizard.get_page_index('providerinfo')