Mercurial > pub > Impl
view Lib/Security/Auth.pm @ 84:e568c7c8b743
Minor changes to the test infrastructure
author | wizard |
---|---|
date | Wed, 14 Apr 2010 17:38:11 +0400 |
parents | 16ada169ca75 |
children |
line wrap: on
line source
package Security::Auth; use strict; use Common; use Security; use DateTime; use Digest::MD5 qw(md5_hex); our @ISA = qw(Object); our $Package; our $DataSource; srand time; BEGIN { DeclareProperty DS => ACCESS_READ; DeclareProperty SecPackage => ACCESS_READ; } { my $i = 0; sub GenSSID() { return md5_hex(time,rand,$i++); } } sub CTOR { my ($this,%args) = @_; $this->{$DS} = $args{'DS'} or die new Exception('A data source is required'); $this->{$SecPackage} = $args{'SecPackage'} or die new Exception('A security package is required'); } sub AuthenticateUser { my ($this,$Name,$SecData) = @_; my $User = $this->{$DS}->FindUser($Name); if (not $User or not $User->Active ) { return new Security::AuthResult ( State => Security::AUTH_FAILED, AuthModule => $this ); } else { if (my $StoredData = $this->{$DS}->GetUserAuthData($User,$this->{$SecPackage})) { my $AuthData = $this->{$SecPackage}->ConstructAuthData($StoredData->AuthData); if ((my $status = $AuthData->DoAuth($SecData)) != Security::AUTH_FAILED) { $AuthData = $this->{$SecPackage}->NewAuthData(GenSSID); return new Security::AuthResult ( State => $status, Session => $this->{$DS}->CreateSession(GenSSID,$User,$AuthData), ClientSecData => $AuthData->ClientAuthData, AuthModule => $this ) } else { return new Security::AuthResult ( State => Security::AUTH_FAILED, AuthModule => $this ); } } else { # the user isn't allowed to authenticate using this method return new Security::AuthResult ( State => Security::AUTH_FAILED, AuthModule => $this ); } } } sub AuthenticateSession { my ($this,$SSID,$SecData) = @_; my $Session = $this->{$DS}->LoadSession($SSID) or return new Security::AuthResult(State => Security::AUTH_FAILED); my $AuthData = $this->{$SecPackage}->ConstructAuthData($Session->SecData); if ((my $status = $AuthData->DoAuth($SecData)) != Security::AUTH_FAILED) { $Session->SecData($AuthData->SessionAuthData); $Session->LastUsage(DateTime->now()); return new Security::AuthResult(State => $status, Session => $Session, ClientSecData => $AuthData->ClientAuthData, AuthModule => $this); } else { $this->{$DS}->CloseSession($Session); return new Security::AuthResult(State => Security::AUTH_FAILED, AuthModule => $this); } } sub CreateUser { my ($this,$uname,$description,$active,$secData) = @_; my $user = $this->{$DS}->CreateUser($uname,$description,$active); $this->{$DS}->SetUserAuthData($user,$this->{$SecPackage},$this->{$SecPackage}->NewAuthData($secData)); return $user; } sub try_construct { my $package = shift; return $package->can('construct') ? $package->construct() : $package; } sub construct { $Package or die new Exception('A security package is reqiured'); $DataSource or die new Exception('A data source is required'); eval "require $DataSource;" or die new Exception('Failed to load the data source module',$@) if not ref $DataSource; eval "require $Package;" or die new Exception('Failed to load the security package module',$@) if not ref $Package; return __PACKAGE__->new(DS => try_construct($DataSource), SecPackage => try_construct($Package)); } 1;