Fixes

+ Null Check for wallet get.
+ Key check while login.

src/app/DALs/WalletDAL.php

@@ -42,10 +42,10 @@ public static function createWallet(String $address, int $timestamp, int $bcHeig
 
     public static function getWallet(String $address) {
         $wallet = Wallet::where('address', $address)->first();
-        if ($wallet->transfers[0] == 'W') {
+        if ($wallet && $wallet->transfers[0] == 'W') {
             $wallet->transfers = self::getFile($wallet->transfers);
         }
-        if ($wallet->keyImages[0] == 'W') {
+        if ($wallet && $wallet->keyImages[0] == 'W') {
             $wallet->keyImages = self::getFile($wallet->keyImages);
         }
         Log::debug($wallet);

src/app/Services/WalletService.php

@@ -96,6 +96,12 @@ public function setWallet(Request $request) {
             throw error::getBadRequestException(error::WALLET_NOT_FOUND);
         }
 
+        // Check if wallet address matches the key
+        $this->rpcService->setWallet(new SetWalletRequest(
+            $address, $viewKey, $wallet->createTime, $wallet->bcHeight,
+            $wallet->transfers, $wallet->keyImages
+        ));
+
         // Generate a new session.
         $request->session()->regenerate();
 

src/app/Utils/error.php

@@ -26,6 +26,7 @@ class error
     const SESSION_EXPIRED = "sessionExpired";
     const REFRESH_LOCKED = "refreshLocked";
     const WALLET_ALREADY_RESET = "walletAlreadyReset";
+    const WALLET_KEY_MISMATCH = "walletKeyMismatch";
 
     /**
      * Contains all error code and description for each key.
@@ -46,6 +47,7 @@ class error
 
         // Wallet erros
         self::WALLET_ALREADY_RESET  => array("code" => 1200, "message" => "Wallet can be reset only once. Contact admins"),
+        self::WALLET_KEY_MISMATCH   => array("code" => 1201, "message" => "Provided key didn't match the address."),
 
         // These errors should never happen!
         // Its over 9000! :P

src/public/js/controllers/registerCtrl.js

@@ -72,6 +72,9 @@ angular.module('aeonPocket').controller('registerCtrl', [
             userService.create($scope.viewWallet).then(function(data) {
                 $mdToast.showSimple("Account Created");
                 $state.go('public.login');
+            }, function (data) {
+                $scope.viewWalletForm.address.$setValidity('validation', false);
+                $scope.errorMessage = data.message;
             });
         }
     }

src/public/templates/views/public/register.html

@@ -82,13 +82,14 @@ <h1 class="text-center" style="color: #fff;">
     </md-card>
 
     <md-card class="loginBox" ng-show="step==3" style="overflow-y: auto;">
-        <form name="viewWalletForm" layout="column" ng-submit="registerViewOnlyWallet()">
+        <form name="viewWalletForm" layout="column" ng-submit="registerViewOnlyWallet()" autocomplete="off">
             <div layout="column">
                 <md-input-container>
                     <label>Your Wallet Address</label>
                     <input ng-disabled="inProgress" required name="address" ng-model="viewWallet.address" />
                     <div class="validation-messages" ng-messages="viewWalletForm.address.$error">
                         <div ng-message="required" ng-show="viewWalletForm.address.$error.required">Required</div>
+                        <div ng-message="validation" ng-show="viewWalletForm.address.$error.validation">{{errorMessage}}</div>
                     </div>
                 </md-input-container>
                 <md-input-container>