--- loncom/auth/migrateuser.pm	2006/04/05 22:39:55	1.5
+++ loncom/auth/migrateuser.pm	2018/11/24 16:19:04	1.35
@@ -1,7 +1,7 @@
 # The LearningOnline Network
 # Starts a user off based of an existing token.
 #
-# $Id: migrateuser.pm,v 1.5 2006/04/05 22:39:55 albertel Exp $
+# $Id: migrateuser.pm,v 1.35 2018/11/24 16:19:04 raeburn Exp $
 #
 # Copyright Michigan State University Board of Trustees
 #
@@ -29,58 +29,478 @@
 package Apache::migrateuser;
 
 use strict;
+use LONCAPA qw(:DEFAULT :match);
 use Apache::Constants qw(:common :http :methods);
 use Apache::lonauth;
 use Apache::lonnet;
+use Apache::lonlocal;
+use Apache::lonlogin();
+use Apache::ltiauth;
+use CGI::Cookie;
 
 sub goto_login {
-    my ($r) = @_;
+    my ($r,$domain,$data) = @_;
+    if ((ref($data) eq 'HASH') && ($data->{'lti.login'})) {
+        &Apache::ltiauth::invalid_request($r,'22');
+    } else {
+        &Apache::loncommon::content_type($r,'text/html');
+        $r->send_http_header;
+        my $url = '/adm/login';
+        if ($domain) {
+            $url .= '?domain='.$domain;
+        }
+        $r->print(&Apache::loncommon::start_page('Going to login',undef,
+					         {'redirect' => [0,$url],}).
+	          '<h1>'.&mt('One moment please...').'</h1>'.
+	          '<p>'.&mt('Transferring to login page.').'</p>'.
+	          &Apache::loncommon::end_page());
+    }
+    return OK;
+}
+
+sub sso_check {
+    my ($data) = @_;
+    my %extra_env;
+    if (ref($data) eq 'HASH') {
+        if ($data->{'sso.login'}) {
+            $extra_env{'request.sso.login'} = $data->{'sso.login'};
+        }
+        if ($data->{'sso.reloginserver'}) {
+            $extra_env{'request.sso.reloginserver'} = 
+                $data->{'sso.reloginserver'};
+        }
+        if (($data->{'balancer'}) && ($data->{'server'}) && ($data->{'balcookie'})) {
+            $extra_env{'request.balancercookie'} = $data->{'server'}.':'.$data->{'balcookie'};
+        }
+    }
+    return \%extra_env;
+}
+
+sub lti_check {
+    my ($data) = @_;
+    my %lti_env;
+    if (ref($data) eq 'HASH') {
+        if ($data->{'lti.login'}) {
+            $lti_env{'request.lti.login'} = $data->{'lti.login'};
+            if ($data->{'lti.reqcrs'}) {
+                $lti_env{'request.lti.reqcrs'} = $data->{'lti.reqcrs'};
+            }
+            if ($data->{'lti.reqrole'}) {
+                $lti_env{'request.lti.reqrole'} = $data->{'lti.reqrole'};
+            }
+            if ($data->{'lti.selfenrollrole'}) {
+                $lti_env{'request.lti.selfenrollrole'} = $data->{'lti.selfenrollrole'};
+            }
+            if ($data->{'lti.uri'}) {
+                $lti_env{'request.lti.uri'} = $data->{'lti.uri'};
+            }
+            if ($data->{'lti.target'}) {
+                $lti_env{'request.lti.target'} = $data->{'lti.target'};
+            }
+            if ($data->{'lti.sourcecrs'}) {
+                $lti_env{'request.lti.sourcecrs'} = $data->{'lti.sourcecrs'};
+            }
+        }
+        if ($data->{'lti.passbackid'}) {
+            $lti_env{'request.lti.passbackid'} = $data->{'lti.passbackid'};
+        }
+        if ($data->{'lti.passbackurl'}) {
+            $lti_env{'request.lti.passbackurl'} = $data->{'lti.passbackurl'};
+        }
+        if ($data->{'lti.rosterid'}) {
+            $lti_env{'request.lti.rosterid'} = $data->{'lti.rosterid'};
+        }
+        if ($data->{'lti.rosterurl'}) {
+            $lti_env{'request.lti.rosterurl'} = $data->{'lti.rosterurl'};
+        }
+    }
+    return \%lti_env;
+}
+
+sub ip_changed {
+    my ($r,$udom,$camefrom,$dataref) = @_;
     &Apache::loncommon::content_type($r,'text/html');
     $r->send_http_header;
-    $r->print(&Apache::loncommon::start_page('Going to login',undef,
-					     {'redirect' =>
-						  [0,'/adm/login'],}).
-	      '<h1>'.&mt('One moment please...').'</h1>'.
-	      '<p>'.&mt('Transferring to login page.').'</p>'.
-	      &Apache::loncommon::end_page());
+    if (ref($dataref) eq 'HASH') {
+        my $title = 'LON-CAPA Session redirected';
+        my $message = &mt('Your internet address has changed since you logged in.');  
+        my $rule_in_effect;
+        if ($dataref->{'balancer'}) {
+            my $baldom = &Apache::lonnet::host_domain($camefrom);
+            my $balprimaryid = &Apache::lonnet::domain($baldom,'primary');
+            my $balintdom = &Apache::lonnet::internet_dom($balprimaryid);
+            my $uprimaryid = &Apache::lonnet::domain($udom,'primary'); 
+            my $uintdom = &Apache::lonnet::internet_dom($uprimaryid);
+            my $dom_in_use;
+            if (($uintdom ne '') && ($uintdom eq $balintdom)) {
+                $dom_in_use = $udom;
+            } else {
+                $dom_in_use = $baldom;
+            }
+            my ($result,$cached)=&Apache::lonnet::is_cached_new('loadbalancing',$dom_in_use);
+            unless (defined($cached)) {
+                my $cachetime = 60*60*24; 
+                my %domconfig =
+                    &Apache::lonnet::get_dom('configuration',['loadbalancing'],$dom_in_use);
+                if (ref($domconfig{'loadbalancing'}) eq 'HASH') {
+                    $result = &Apache::lonnet::do_cache_new('loadbalancing',$dom_in_use,
+                                                            $domconfig{'loadbalancing'},$cachetime);
+                }
+            }
+            if (ref($result) eq 'HASH') {
+                (undef,my $currtargets,my $currrules) =
+                    &Apache::lonnet::check_balancer_result($result,$dataref->{'server'});
+                if (ref($currrules) eq 'HASH') {
+                    if ($dataref->{'sso.login'}) {
+                        if ($currrules->{'_LC_ipchangesso'} ne '') {
+                            $rule_in_effect = $currrules->{'_LC_ipchangesso'};
+                        }
+                    } else {
+                        if ($currrules->{'_LC_ipchange'} ne '') {
+                            $rule_in_effect = $currrules->{'_LC_ipchange'};
+                        }
+                    }
+                }
+            }
+        }
+        my $url;
+        my $lonhost= $r->dir_config('lonHostID');
+        my $switchto = $lonhost;
+        if ($rule_in_effect ne 'offloadedto') {
+            my $hosthere;
+            my @ids=&Apache::lonnet::current_machine_ids();
+            unless ($rule_in_effect eq 'balancer') {
+                if (grep(/^\Q$rule_in_effect\E$/,@ids)) {
+                    $hosthere = 1;
+                }
+            }
+            unless ($hosthere) {
+                if ($dataref->{'role'}) {
+                    my ($adom,$aname);
+                    if ($dataref->{'role'} =~ m{^au\./($match_domain)/$}) {
+                        $adom = $1;
+                        $aname = $dataref->{'username'};
+                    } elsif ($dataref->{'role'} =~ m{^(?:ca|aa)\./($match_domain)/($match_username)$}) {
+                        $adom = $1;
+                        $aname = $2;
+                    }
+                    if ($adom ne '' && $aname ne '') {
+                        my $ahome = &Apache::lonnet::homeserver($aname,$adom);
+                        unless ($ahome eq 'no_host') {
+                            if ($ahome && grep(/^\Q$ahome\E$/,@ids)) {
+                                $hosthere = 1;
+                            }
+                        }
+                    }
+                }
+            }
+            unless ($hosthere) {
+                my $hostname;
+                if ($rule_in_effect eq 'balancer') {
+                    $hostname = &Apache::lonnet::hostname($dataref->{'server'});
+                    if ($hostname) {
+                        $switchto = $dataref->{'server'};
+                    }
+                } else {
+                    $hostname = &Apache::lonnet::hostname($rule_in_effect);
+                    if ($hostname) {
+                        $switchto = $rule_in_effect;
+                    }
+                }
+                if ($hostname) {
+                    my $protocol = $Apache::lonnet::protocol{$switchto};
+                    $protocol = 'http' if ($protocol ne 'https');
+                    $url = $protocol.'://'.$hostname;
+                    if ($rule_in_effect eq 'balancer') {
+                        $message .= '<br />'.
+                                    &mt('As a result, your LON-CAPA session is being redirected to the server where you originally logged in.');
+                    } else {
+                        $message .= '<br />'.
+                                    &mt('As a result, your LON-CAPA session is being redirected.');
+                    }
+                }
+            }
+            unless ($hosthere) {
+                if (($dataref->{'balancer'}) && ($dataref->{'balcookie'})) {
+                    &Apache::lonnet::delbalcookie($dataref->{'balcookie'},$dataref->{'balancer'});
+                }
+            }
+        }
+        if ($dataref->{'sso.login'}) {
+            $url .= '/adm/roles';
+        } else {
+            $url .= '/adm/login';
+            if ($udom) {
+                $url .= '?domain='.$udom;
+            }
+            $message .= '<br />'.&mt('You will need to provide your password one more time.');
+        }
+        my %info= (
+                    'domain'          => $udom,
+                    'username'        => $dataref->{'username'},
+                    'role'            => $dataref->{'role'},
+                    'sessionserver'   => $lonhost,
+                  );
+        if ($dataref->{'origurl'}) {
+            $info{'origurl'} = $dataref->{'origurl'};
+        }
+        if ($dataref->{'symb'}) {
+            $info{'symb'} = $dataref->{'symb'};
+        }
+        my $iptoken = &Apache::lonnet::tmpput(\%info,$switchto);
+        unless ($iptoken eq 'conlost') {
+            $url .= ($url =~ /\?/) ? '&' : '?';
+            $url .= 'iptoken='.$iptoken;
+        }
+        $r->print(&Apache::loncommon::start_page($title,undef,
+                                                 {'redirect' =>
+                                                  [2,$url],}).
+                  '<h1>'.&mt('One moment please...').'</h1>'.
+                  '<p class="LC_warning">'.$message.'</p>'.
+                  &Apache::loncommon::end_page());
+    } else {
+        return &goto_login($r);
+    }
     return OK;
 }
 
+sub logout {
+    my ($r,$handle,$data,$lti_env) = @_;
+    my $lonidsdir=$r->dir_config('lonIDsDir');
+    if (unlink("$lonidsdir/$handle.id")) {
+        if (($env{'user.linkedenv'} =~ /^[a-f0-9]+_linked$/) &&
+            (-l "$lonidsdir/$env{'user.linkedenv'}.id") &&
+            (readlink("$lonidsdir/$env{'user.linkedenv'}.id") eq "$lonidsdir/$handle.id")) {
+            unlink("$lonidsdir/$env{'user.linkedenv'}.id");
+        }
+    }
+    my %temp=('logout' => time);
+    &Apache::lonnet::put('email_status',\%temp);
+    &Apache::lonnet::log($env{'user.domain'},
+                         $env{'user.name'},
+                         $env{'user.home'},
+                         "Logout $ENV{'REMOTE_ADDR'}");
+
+    &Apache::loncommon::content_type($r,'text/html');
+
+    #expire the cookies
+    my %cookies=CGI::Cookie->parse($r->header_in('Cookie'));
+    foreach my $name (keys(%cookies)) {
+        next unless ($name =~ /^lon(|S|Link|Pub)ID$/);
+        my $c = new CGI::Cookie(-name    => $name,
+                                -value   => '',
+                                -expires => '-10y',);
+        $r->headers_out->add('Set-cookie' => $c);
+    }
+    my (%info,%user_info,%lti_info);
+    if (ref($lti_env) eq 'HASH') {
+        %lti_info = %{$lti_env};
+    }
+    my $lonhost = $r->dir_config('lonHostID');
+    if (ref($data) eq 'HASH') {
+        %user_info=('ip'       => $ENV{'REMOTE_ADDR'},
+                    'domain'   => $data->{'domain'},
+                    'username' => $data->{'username'},
+                    'role'     => $data->{'role'},
+                    'origurl'  => $data->{'origurl'},
+                    'symb'     => $data->{'symb'},
+                    'server'   => $lonhost);
+    }
+    %info = (%user_info,%lti_info);
+    my $token = &Apache::lonnet::tmpput(\%info,$lonhost);
+    my $url = '/adm/migrateuser?token='.$token;
+    $r->send_http_header;
+    $r->print(
+        &Apache::loncommon::start_page('Updating Session ...',undef,
+                                       {'redirect'       => [0.1,$url],
+                                        'only_body'      => 1,}).
+        &Apache::loncommon::end_page());
+    $r->register_cleanup(\&flush_course_logs);
+    return;
+}
+
+sub flush_course_logs {
+    &Apache::lonnet::flushcourselogs();
+    return OK;
+}
 
 sub handler {
     my ($r) = @_;
     
     &Apache::loncommon::get_unprocessed_cgi($ENV{'QUERY_STRING'},['token']);
     my %data =   &Apache::lonnet::tmpget($env{'form.token'});
+    if (keys(%data) == 0) {
+        return &goto_login($r);
+    }
     my $delete = &Apache::lonnet::tmpdel($env{'form.token'});
 
+    &Apache::lonlocal::get_language_handle($r);
+
     if ($delete ne 'ok') {
-	return &goto_login($r);
+	return &goto_login($r,undef,\%data);
     }
 
-    if ($data{'ip'} ne $ENV{'REMOTE_ADDR'} || !defined($data{'username'}) ||
-	!defined($data{'domain'}) ) {
-	return &goto_login($r);
+    if (!defined($data{'username'}) || !defined($data{'domain'})) {
+        return &goto_login($r,undef,\%data);
+    }
+    if ($data{'ip'} ne $ENV{'REMOTE_ADDR'}) {
+        &Apache::lonnet::logthis('IP change when session migration requested -- was: '.
+                 $data{'ip'}.'; now: '.$ENV{'REMOTE_ADDR'}.' for '.$data{'username'}.':'.$data{'domain'});
+	return &ip_changed($r,$data{'domain'},$data{'server'},\%data);
     }
 
-    &Apache::lonnet::logthis("Allowing access for $data{'username'}\@$data{'domain'} to $data{'role'}");
+    &Apache::lonnet::logthis("Allowing access for $data{'username'}:$data{'domain'} to $data{'role'}");
     my $home=&Apache::lonnet::homeserver($data{'username'},$data{'domain'});
-    if ($home =~ /(con_lost|no_such_host)/) { return &goto_login($r); }
+    my $udom;
+    if (&Apache::lonnet::domain($data{'domain'})) {
+        $udom=$data{'domain'};
+    }
+    if ($home =~ /(con_lost|no_such_host)/) { return &goto_login($r,$udom,\%data); }
+
+    my $sso_env = &sso_check(\%data);
+    my $lti_env = &lti_check(\%data);
+
+    my $extra_env;
+    if ((ref($sso_env) eq 'HASH') && (keys(%{$sso_env}))) {
+        $extra_env = $sso_env;
+    } elsif ((ref($lti_env) eq 'HASH') && (keys(%{$lti_env}))) {
+        $extra_env = $lti_env;
+    }
+
+    my %form;
+    if ($data{'symb'} ne '') {
+        $form{'symb'} = $data{'symb'};
+    }
+    if ($data{'iptoken'} ne '') {
+        $form{'iptoken'} = $data{'iptoken'};
+    }
+    if ($data{'noloadbalance'} ne '') {
+        $form{'noloadbalance'} = $data{'noloadbalance'};
+    }
 
     if (!$data{'role'}) {
-	&Apache::lonauth::success($r,$data{'username'},$data{'domain'},
-				  $home,'/adm/roles');
+	my $handle = &Apache::lonnet::check_for_valid_session($r);
+	if ($handle) {
+	    &Apache::lonnet::transfer_profile_to_env($r->dir_config('lonIDsDir'),
+						     $handle);
+            if ($data{'lti.login'}) {
+                my $needslogout;
+                if ($env{'request.lti.login'}) {
+                    if (($env{'user.name'} ne $data{'username'}) ||
+                        ($env{'user.domain'} ne $data{'domain'})) {
+                        $needslogout = 1;
+                    }
+                } else {
+                    $needslogout = 1;
+                }
+# If access is via LTI, and user already has a non-LTI session cookie 
+# (and session) or has an LTI session cookie for a different username,
+# logout the existing session, and start a new one
+                if ($needslogout) {
+                    &logout($r,$handle,\%data,$lti_env);
+                } elsif (($data{'lti.reqcrs'}) && ($data{'lti.reqrole'} eq 'cc')) {
+                    $form{'lti.reqcrs'} = $data{'lti.reqcrs'};
+                    $form{'lti.reqrole'} = $data{'lti.reqrole'};
+                    $form{'lti.sourcecrs'} = $data{'lti.sourcecrs'};
+                    $form{'lti.uri'} = $data{'lti.uri'};
+                    if ($data{'lti.passbackid'}) {
+                        $form{'lti.passbackid'} = $data{'lti.passbackid'};
+                    }
+                    if ($data{'lti.passbackurl'}) {
+                        $form{'lti.passbackurl'} = $data{'lti.passbackurl'};
+                    }
+                    if ($data{'lti.rosterid'}) {
+                        $form{'lti.rosterid'} = $data{'lti.rosterid'};
+                    }
+                    if ($data{'lti.rosterurl'}) {
+                        $form{'lti.rosterurl'} = $data{'lti.rosterurl'};
+                    }
+                    if ($data{'lti.target'}) {
+                        $form{'lti.target'} = $data{'lti.target'};
+                    }
+                    &Apache::loncommon::content_type($r,'text/html');
+                    $r->send_http_header;
+                    &Apache::ltiauth::lti_reqcrs($r,$data{'domain'},\%form,$data{'username'},$data{'domain'});
+                } else {
+                    if (ref($lti_env) eq 'HASH') {
+                        delete($lti_env->{'reqcrs'});
+                        delete($lti_env->{'reqrole'});
+                        delete($lti_env->{'selfenrollrole'});
+                    }
+                    if ($data{'lti.selfenrollrole'}) {
+                        if (&Apache::ltiauth::lti_enroll($data{'username'},$data{'domain'},
+                                                         $data{'lti.selfenrollrole'}) eq 'ok') {
+                            my $url = '/adm/roles?selectrole=1&'.
+                                      &escape($data{'lti.selfenrollrole'}).'=1';
+                            if ($data{'origurl'} =~ m{/default_\d+\.sequence$}) {
+                                $url .= '&orgurl='.$data{'origurl'}.'&navmap=1';
+                            } elsif ($data{'origurl'} ne '') {
+                                $url .= '&orgurl='.$data{'origurl'};
+                            }
+                            if (ref($lti_env) eq 'HASH') {
+                                &Apache::lonnet::appenv($lti_env);
+                            }
+                            $r->internal_redirect($url);
+                        } else {
+                            &Apache::ltiauth::invalid_request($r,23);
+                        }
+                    } elsif ($data{'origurl'} ne '') {
+                        my $url = $data{'origurl'};
+                        if ($url =~ m{/default_\d+\.sequence$}) {
+                            $url .= (($url =~/\?/)?'&':'?').'navmap=1';
+                        }
+                        if (ref($lti_env) eq 'HASH') {
+                            &Apache::lonnet::appenv($lti_env);
+                        }
+                        $r->internal_redirect($url);
+                    } else {
+                        if (ref($lti_env) eq 'HASH') {
+                            &Apache::lonnet::appenv($lti_env);
+                        }
+                    }
+                }
+            } elsif ($data{'origurl'} ne '') {
+                $r->internal_redirect($data{'origurl'});
+            } elsif ($env{'request.course.id'}) {
+                $r->internal_redirect('/adm/navmaps');
+	    } else {
+		$r->internal_redirect('/adm/roles');
+	    }
+	} else {
+            my $desturl = '/adm/roles';
+            if ($data{'origurl'} ne '') {
+                $desturl = $data{'origurl'};
+                if ($data{'lti.login'}) {
+                    $desturl = $data{'origurl'};
+                    if ($desturl =~ m{/default_\d+\.sequence$}) {
+                        $desturl .= (($desturl =~/\?/)?'&':'?').'navmap=1';
+                    }
+                }
+            }
+            my $skipcritical;
+            if (($data{'lti.login'}) && ($data{'lti.reqcrs'}) &&
+                ($data{'lti.reqrole'} eq 'cc')) {
+                $skipcritical = 1;
+            }
+	    &Apache::lonauth::success($r,$data{'username'},$data{'domain'},
+				      $home,$desturl,$extra_env,\%form,$skipcritical);
+	}
 	return OK;
     }
-    
-    my $cookie=&Apache::lonauth::success($r,$data{'username'},$data{'domain'},
-					 $home,'noredirect');
-    $r->header_out('Set-cookie',"lonID=$cookie; path=/");
-    &Apache::lonnet::transfer_profile_to_env($r->dir_config('lonIDsDir'),
-					     $cookie);
-    $env{'form.selectrole'}='1';
-    $env{'form.'.$data{'role'}}='1';
-    return &Apache::lonroles::handler($r);
+
+    my $next_url='/adm/roles?selectrole=1&'.&escape($data{'role'}).'=1';
+    if ($data{'origurl'} ne '') {
+        $next_url .= '&orgurl='.&escape($data{'origurl'});
+        if ($data{'lti.login'}) {
+            if (($data{'origurl'} =~ m{/default_\d+\.sequence$}) ||
+                ($data{'origurl'} =~ m{^/res/.+\.sequence$})) {
+                $next_url .= '&navmap=1';
+            }
+        }
+    }
+    &Apache::lonauth::success($r,$data{'username'},$data{'domain'},$home,
+			      $next_url,$extra_env,\%form);
+    return OK;
 }
 
 1;