From 6b308f0d1a3932cde5346cef509fdd49a8b626a4 Mon Sep 17 00:00:00 2001
From: Evan Prodromou <evan@prodromou.name>
Date: Sat, 17 May 2008 15:30:30 -0400
Subject: [PATCH] check results of setting current user

darcs-hash:20080517193030-84dde-ecd85676afd34bd9eb4b42b84bb3b6a6b9ea6a9e.gz
---
 actions/login.php    | 7 ++++++-
 actions/register.php | 6 +++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/actions/login.php b/actions/login.php
index 60c95343f0..293d07faed 100644
--- a/actions/login.php
+++ b/actions/login.php
@@ -38,7 +38,12 @@ class LoginAction extends Action {
 		$nickname = $this->arg('nickname');
 		$password = $this->arg('password');
 		if (common_check_user($nickname, $password)) {
-			common_set_user($nickname);
+			# success!
+			if (!common_set_user($nickname)) {
+				common_server_error(_t('Error setting user.'));
+				return;
+			}
+			# success!			
 			common_redirect(common_local_url('all',
 											 array('nickname' =>
 												   $nickname)));
diff --git a/actions/register.php b/actions/register.php
index 29a7a70f7e..096eded79d 100644
--- a/actions/register.php
+++ b/actions/register.php
@@ -51,7 +51,11 @@ class RegisterAction extends Action {
 		} else if ($password != $confirm) {
 			$this->show_form(_t('Passwords don\'t match.'));
 		} else if ($this->register_user($nickname, $password, $email)) {
-			common_set_user($nickname);
+			# success!
+			if (!common_set_user($nickname)) {
+				common_server_error(_t('Error setting user.'));
+				return;
+			}
 			common_redirect(common_local_url('profilesettings'));
 		} else {
 			$this->show_form(_t('Invalid username or password.'));